]> Sergey Matveev's repositories - tofuproxy.git/blob - fifos/spies.go
Raised copyright years
[tofuproxy.git] / fifos / spies.go
1 /*
2 tofuproxy -- flexible HTTP/HTTPS proxy, TLS terminator, X.509 TOFU
3              manager, WARC/geminispace browser
4 Copyright (C) 2021-2022 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         "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                 hosts := make(map[string]struct{})
48                 for _, line := range readLinesFromFIFO(p) {
49                         hosts[line] = struct{}{}
50                 }
51                 for host := range hosts {
52                         log.Printf("%s: adding host %s\n", p, host)
53                 }
54                 caches.SpiesM.Lock()
55                 for _, spy := range caches.Spies {
56                         hosts[spy] = struct{}{}
57                 }
58                 caches.Spies = caches.Spies[:0]
59                 for host := range hosts {
60                         caches.Spies = append(caches.Spies, host)
61                 }
62                 caches.SpiesM.Unlock()
63         }
64 }