rounds/spy.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 rounds
  18
  19 import (
  20         "fmt"
  21         "net/http"
  22         "strings"
  23
  24         "go.stargrave.org/tofuproxy/caches"
  25         "go.stargrave.org/tofuproxy/fifos"
  26 )
  27
  28 func IsSpy(host string) bool {
  29         caches.SpiesM.RLock()
  30         defer caches.SpiesM.RUnlock()
  31         for _, spy := range caches.Spies {
  32                 if strings.HasSuffix(host, spy) {
  33                         return true
  34                 }
  35         }
  36         return false
  37 }
  38
  39 func RoundDenySpy(
  40         host string,
  41         resp *http.Response,
  42         w http.ResponseWriter,
  43         req *http.Request,
  44 ) (bool, error) {
  45         if IsSpy(host) {
  46                 http.NotFound(w, req)
  47                 fifos.LogVarious <- fmt.Sprintf("%s %s\tdeny spy", req.Method, req.URL)
  48                 return false, nil
  49         }
  50         return true, nil
  51 }