/*
+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
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
-package main
+package tofuproxy
import (
"bytes"
"sync"
"go.cypherpunks.ru/ucspi"
+ "go.stargrave.org/tofuproxy/caches"
+ "go.stargrave.org/tofuproxy/fifos"
)
var (
CmdCerttool = "certtool"
- CmdWish = "wish8.7"
-
- certs *string
- accepted = make(map[string]string)
- acceptedM sync.RWMutex
- rejected = make(map[string]string)
- rejectedM sync.RWMutex
- VerifyM sync.Mutex
+ CmdWish = "wish8.6"
+
+ Certs string
+ VerifyM sync.Mutex
)
func spkiHash(cert *x509.Certificate) string {
return hex.EncodeToString(hsh[:])
}
-func acceptedAdd(addr, h string) {
- acceptedM.Lock()
- accepted[addr] = h
- acceptedM.Unlock()
-}
-
-func rejectedAdd(addr, h string) {
- rejectedM.Lock()
- rejected[addr] = h
- rejectedM.Unlock()
-}
-
type ErrRejected struct {
addr string
}
certTheirHash := spkiHash(certTheir)
VerifyM.Lock()
defer VerifyM.Unlock()
- acceptedM.RLock()
- certOurHash := accepted[host]
- acceptedM.RUnlock()
+ caches.AcceptedM.RLock()
+ certOurHash := caches.Accepted[host]
+ caches.AcceptedM.RUnlock()
if certTheirHash == certOurHash {
return nil
}
- rejectedM.RLock()
- certOurHash = rejected[host]
- rejectedM.RUnlock()
+ caches.RejectedM.RLock()
+ certOurHash = caches.Rejected[host]
+ caches.RejectedM.RUnlock()
if certTheirHash == certOurHash {
return ErrRejected{host}
}
daneExists, daneMatched := dane(host, certTheir)
if daneExists {
if daneMatched {
- sinkCert <- fmt.Sprintf("DANE\t%s\tmatched", host)
+ fifos.LogDANE <- fmt.Sprintf("%s\tACK", host)
} else {
- sinkErr <- fmt.Sprintf("DANE\t%s\tnot matched", host)
+ fifos.LogDANE <- fmt.Sprintf("%s\tNAK", host)
}
}
- fn := filepath.Join(*certs, host)
+ fn := filepath.Join(Certs, host)
certsOur, _, err := ucspi.CertPoolFromFile(fn)
if err == nil || dialErr != nil || (daneExists && !daneMatched) {
if certsOur != nil && certTheirHash == spkiHash(certsOur[0]) {
- acceptedAdd(host, certTheirHash)
+ caches.AcceptedM.Lock()
+ caches.Accepted[host] = certTheirHash
+ caches.AcceptedM.Unlock()
if bytes.Compare(certsOur[0].Raw, rawCerts[0]) != 0 {
- sinkCert <- fmt.Sprintf("Refresh\t%s\t%s", host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("Refresh\t%s\t%s", host, certTheirHash)
goto CertUpdate
}
return nil
if daneMatched {
b.WriteString("label .lDANE -bg green -text \"DANE matched\"\n")
} else {
- b.WriteString("label .lDANE -bg red -text \"DANE not matched!\"\n")
+ b.WriteString("label .lDANE -bg red -text \"DANE NOT matched\"\n")
}
b.WriteString("grid .lDANE\n")
}
proc doOnce {} { exit 11 }
proc doReject {} { exit 12 }
button .bAccept -text "Accept" -bg green -command doAccept
-button .bOnce -text "Once" -command doOnce
+button .bOnce -text "Once" -bg green -command doOnce
button .bReject -text "Reject" -bg red -command doReject
grid .bAccept .bOnce .bReject
grid rowconfigure . 0 -weight 1
err = cmd.Run()
exitError, ok := err.(*exec.ExitError)
if !ok {
- sinkCert <- fmt.Sprintf("DENY\t%s\t%s", host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("Reject\t%s\t%s", host, certTheirHash)
return ErrRejected{host}
}
switch exitError.ExitCode() {
case 10:
- sinkCert <- fmt.Sprintf("ADD\t%s\t%s", host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("Accept\t%s\t%s", host, certTheirHash)
goto CertUpdate
case 11:
- sinkCert <- fmt.Sprintf("ONCE\t%s\t%s", host, certTheirHash)
- acceptedAdd(host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("Once\t%s\t%s", host, certTheirHash)
+ caches.AcceptedM.Lock()
+ caches.Accepted[host] = certTheirHash
+ caches.AcceptedM.Unlock()
return nil
case 12:
- rejectedAdd(host, certTheirHash)
+ caches.RejectedM.Lock()
+ caches.Rejected[host] = certTheirHash
+ caches.RejectedM.Unlock()
fallthrough
default:
- sinkCert <- fmt.Sprintf("DENY\t%s\t%s", host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("Reject\t%s\t%s", host, certTheirHash)
return ErrRejected{host}
}
} else {
if !os.IsNotExist(err) {
return err
}
- sinkCert <- fmt.Sprintf("TOFU\t%s\t%s", host, certTheirHash)
+ fifos.LogCert <- fmt.Sprintf("TOFU\t%s\t%s", host, certTheirHash)
}
CertUpdate:
- tmp, err := os.CreateTemp(*certs, "")
+ tmp, err := os.CreateTemp(Certs, "")
if err != nil {
log.Fatalln(err)
}
}
tmp.Close()
os.Rename(tmp.Name(), fn)
- acceptedAdd(host, certTheirHash)
+ caches.AcceptedM.Lock()
+ caches.Accepted[host] = certTheirHash
+ caches.AcceptedM.Unlock()
return nil
}