fifos/restricted.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         "fmt"
  21         "log"
  22         "os"
  23         "strings"
  24
  25         "go.stargrave.org/tofuproxy/caches"
  26 )
  27
  28 func listRestricted(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.RestrictedM.RLock()
  35                 for spki, hosts := range caches.Restricted {
  36                         for _, host := range hosts {
  37                                 if _, err = fmt.Fprintf(fd, "%s\t%s\n", spki, host); err != nil {
  38                                         break
  39                                 }
  40                         }
  41                 }
  42                 caches.RestrictedM.RUnlock()
  43                 fd.Close()
  44         }
  45 }
  46
  47 func addRestricted(p string) {
  48         for {
  49                 neu := make(map[string][]string)
  50                 for _, line := range readLinesFromFIFO(p) {
  51                         cols := strings.Fields(line)
  52                         if len(cols) != 2 {
  53                                 continue
  54                         }
  55                         spki, host := cols[0], cols[1]
  56                         log.Printf("%s: adding %s: %s\n", p, spki, host)
  57                         neu[spki] = append(neu[spki], host)
  58                 }
  59                 caches.RestrictedM.Lock()
  60                 for spki, hosts := range neu {
  61                         caches.Restricted[spki] = append(caches.Restricted[spki], hosts...)
  62                 }
  63                 caches.RestrictedM.Unlock()
  64         }
  65 }