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