WARC

[tofuproxy.git] / cmd / tofuproxy / main.go

diff --git a/cmd/tofuproxy/main.go b/cmd/tofuproxy/main.go
index 5335ffb0f911358a647a23d562a97a554f190f74..d55233979ffceff872b4f19df283e20b0fe476d1 100644 (file)
--- a/cmd/tofuproxy/main.go
+++ b/cmd/tofuproxy/main.go
@@ -1,5 +1,5 @@
 /*
-tofuproxy -- HTTP proxy with TLS certificates management
+tofuproxy -- flexible HTTP/WARC 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
@@ -26,6 +26,7 @@ import (
        "go.cypherpunks.ru/ucspi"
        "go.stargrave.org/tofuproxy"
        "go.stargrave.org/tofuproxy/fifos"
+       "go.stargrave.org/tofuproxy/rounds"
 )
 
 func main() {
@@ -33,9 +34,11 @@ func main() {
        prvPath := flag.String("key", "prv.pem", "Path to server PKCS#8 private key")
        bind := flag.String("bind", "[::1]:8080", "Bind address")
        certs := flag.String("certs", "./certs", "Directory with pinned certificates")
+       ccerts := flag.String("ccerts", "./ccerts", "Directory with client certificates")
        dnsSrv := flag.String("dns", "[::1]:53", "DNS server")
        fifosDir := flag.String("fifos", "fifos", "Directory with FIFOs")
        notai := flag.Bool("notai", false, "Do not prepend TAI64N to logs")
+       warcOnly := flag.Bool("warc-only", false, "Server only WARC URIs")
        flag.Parse()
        log.SetFlags(log.Lshortfile)
 
@@ -50,12 +53,13 @@ func main() {
        }
 
        fifos.NoTAI = *notai
-       fifos.FIFOs = *fifosDir
-       fifos.Init()
+       fifos.Start(*fifosDir)
        tofuproxy.Certs = *certs
+       tofuproxy.CCerts = *ccerts
        tofuproxy.DNSSrv = *dnsSrv
        tofuproxy.CACert = caCert
        tofuproxy.CAPrv = caPrv
+       rounds.WARCOnly = *warcOnly
 
        ln, err := net.Listen("tcp", *bind)
        if err != nil {
@@ -65,7 +69,7 @@ func main() {
                Handler:      &tofuproxy.Handler{},
                TLSNextProto: tofuproxy.TLSNextProtoS,
        }
-       log.Println("listening:", *bind, "certs:", *certs)
+       log.Println("listening:", *bind, "dns:", *dnsSrv, "certs:", *certs, "ccerts:", *ccerts)
        if err := srv.Serve(ln); err != nil {
                log.Fatalln(err)
        }