fifos/spies.go

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