fifos/spies.go

   1 /*
   2 tofuproxy -- flexible HTTP proxy, TLS terminator, X.509 certificates
   3              manager, WARC/Gemini browser
   4 Copyright (C) 2021 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         "bufio"
  23         "log"
  24         "os"
  25
  26         "go.stargrave.org/tofuproxy/caches"
  27 )
  28
  29 func listSpies(p string) {
  30         for {
  31                 fd, err := os.OpenFile(p, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
  32                 if err != nil {
  33                         log.Fatalln(err)
  34                 }
  35                 caches.SpiesM.RLock()
  36                 for _, spy := range caches.Spies {
  37                         if _, err = fd.WriteString(spy + "\n"); err != nil {
  38                                 break
  39                         }
  40                 }
  41                 caches.SpiesM.RUnlock()
  42                 fd.Close()
  43         }
  44 }
  45
  46 func addSpy(p string) {
  47         for {
  48                 fd, err := os.OpenFile(p, os.O_RDONLY, os.FileMode(0666))
  49                 if err != nil {
  50                         log.Fatalln(err)
  51                 }
  52                 hosts := make(map[string]struct{})
  53                 scanner := bufio.NewScanner(fd)
  54                 for scanner.Scan() {
  55                         t := scanner.Text()
  56                         if len(t) > 0 {
  57                                 hosts[t] = struct{}{}
  58                         }
  59                 }
  60                 fd.Close()
  61                 for host := range hosts {
  62                         log.Printf("%s: adding host %s\n", p, host)
  63                 }
  64                 caches.SpiesM.Lock()
  65                 for _, spy := range caches.Spies {
  66                         hosts[spy] = struct{}{}
  67                 }
  68                 caches.Spies = caches.Spies[:0]
  69                 for host := range hosts {
  70                         caches.Spies = append(caches.Spies, host)
  71                 }
  72                 caches.SpiesM.Unlock()
  73         }
  74 }