fifos/spies.go

   1 /*
   2 tofuproxy -- flexible HTTP/WARC proxy with TLS certificates management
   3 Copyright (C) 2021 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
  18 package fifos
  19
  20 import (
  21         "bufio"
  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                 fd, err := os.OpenFile(p, os.O_RDONLY, os.FileMode(0666))
  48                 if err != nil {
  49                         log.Fatalln(err)
  50                 }
  51                 hosts := make(map[string]struct{})
  52                 scanner := bufio.NewScanner(fd)
  53                 for scanner.Scan() {
  54                         t := scanner.Text()
  55                         if len(t) > 0 {
  56                                 hosts[t] = struct{}{}
  57                         }
  58                 }
  59                 fd.Close()
  60                 for host := range hosts {
  61                         log.Printf("%s: adding host %s\n", p, host)
  62                 }
  63                 caches.SpiesM.Lock()
  64                 for _, spy := range caches.Spies {
  65                         hosts[spy] = struct{}{}
  66                 }
  67                 caches.Spies = caches.Spies[:0]
  68                 for host := range hosts {
  69                         caches.Spies = append(caches.Spies, host)
  70                 }
  71                 caches.SpiesM.Unlock()
  72         }
  73 }