Ability to dynamically control list of spying domains

[tofuproxy.git] / fifos / spies.go

diff --git a/fifos/spies.go b/fifos/spies.go
new file mode 100644 (file)
index 0000000..0b6aa57
--- /dev/null
+++ b/fifos/spies.go
@@ -0,0 +1,73 @@
+/*
+tofuproxy -- HTTP proxy with TLS certificates management
+Copyright (C) 2021 Sergey Matveev <stargrave@stargrave.org>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, version 3 of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package fifos
+
+import (
+       "bufio"
+       "log"
+       "os"
+
+       "go.stargrave.org/tofuproxy/caches"
+)
+
+func listSpies(p string) {
+       for {
+               fd, err := os.OpenFile(p, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
+               if err != nil {
+                       log.Fatalln(err)
+               }
+               caches.SpiesM.RLock()
+               for _, spy := range caches.Spies {
+                       if _, err = fd.WriteString(spy + "\n"); err != nil {
+                               break
+                       }
+               }
+               caches.SpiesM.RUnlock()
+               fd.Close()
+       }
+}
+
+func addSpy(p string) {
+       for {
+               fd, err := os.OpenFile(p, os.O_RDONLY, os.FileMode(0666))
+               if err != nil {
+                       log.Fatalln(err)
+               }
+               hosts := make(map[string]struct{})
+               scanner := bufio.NewScanner(fd)
+               for scanner.Scan() {
+                       t := scanner.Text()
+                       if len(t) > 0 {
+                               hosts[t] = struct{}{}
+                       }
+               }
+               fd.Close()
+               for host := range hosts {
+                       log.Printf("%s: adding host %s\n", p, host)
+               }
+               caches.SpiesM.Lock()
+               for _, spy := range caches.Spies {
+                       hosts[spy] = struct{}{}
+               }
+               caches.Spies = caches.Spies[:0]
+               for host := range hosts {
+                       caches.Spies = append(caches.Spies, host)
+               }
+               caches.SpiesM.Unlock()
+       }
+}