]> Sergey Matveev's repositories - tofuproxy.git/blob - fifos/restricted.go
76c5f0b053eaa3b1c68b0900d00dc6b0c00534c2
[tofuproxy.git] / 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 }