fifos/spies.go

   1 /*
   2 tofuproxy -- flexible HTTP/HTTPS proxy, TLS terminator, X.509 TOFU
   3              manager, WARC/geminispace browser
   4 Copyright (C) 2021-2023 Sergey Matveev <stargrave@stargrave.org>
   5
   6 This program is free software: you can redistribute it and/or modify
   7 it under the terms of the GNU General Public License as published by
   8 the Free Software Foundation, version 3 of the License.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package fifos
  20
  21 import (
  22         "log"
  23         "os"
  24
  25         "go.stargrave.org/tofuproxy/caches"
  26 )
  27
  28 func listSpies(p string) {
  29         for {
  30                 fd, err := os.OpenFile(p, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
  31                 if err != nil {
  32                         log.Fatalln(err)
  33                 }
  34                 caches.SpiesM.RLock()
  35                 for _, spy := range caches.Spies {
  36                         if _, err = fd.WriteString(spy + "\n"); err != nil {
  37                                 break
  38                         }
  39                 }
  40                 caches.SpiesM.RUnlock()
  41                 fd.Close()
  42         }
  43 }
  44
  45 func addSpy(p string) {
  46         for {
  47                 hosts := make(map[string]struct{})
  48                 for _, line := range readLinesFromFIFO(p) {
  49                         hosts[line] = struct{}{}
  50                 }
  51                 for host := range hosts {
  52                         log.Printf("%s: adding host %s\n", p, host)
  53                 }
  54                 caches.SpiesM.Lock()
  55                 for _, spy := range caches.Spies {
  56                         hosts[spy] = struct{}{}
  57                 }
  58                 caches.Spies = caches.Spies[:0]
  59                 for host := range hosts {
  60                         caches.Spies = append(caches.Spies, host)
  61                 }
  62                 caches.SpiesM.Unlock()
  63         }
  64 }