Initial implementation: copy structure from "binder" project

1 files changed, 112 insertions, 0 deletions

diff --git a/src/untls.go b/src/untls.go
new file mode 100644
index 0000000..545ee2e
--- /dev/null
+++ b/src/untls.go
@@ -0,0 +1,112 @@
+package untls
+
+import (
+	"crypto/tls"
+	"fmt"
+	"io"
+	"net"
+	"os"
+
+	g "gobang"
+)
+
+
+
+type _CLIArgs struct {
+	certFile string
+	keyFile  string
+	fromAddr string
+	toAddr   string
+}
+
+
+
+var emitActiveConnection = g.MakeGauge("active-connections")
+
+const X = 1
+
+
+
+func parseArgs(args []string) _CLIArgs {
+	if len(args) != 5 {
+		fmt.Fprintf(
+			os.Stderr,
+			"Usage: %s CERT.pem KEY.pem FROM.socket TO.socket\n",
+			args[0],
+		)
+		os.Exit(2)
+	}
+	return _CLIArgs {
+		certFile: args[1],
+		keyFile:  args[2],
+		fromAddr: args[3],
+		toAddr:   args[4],
+	}
+}
+
+func listen(certFile string, keyFile string, fromAddr string) net.Listener {
+	certificate, err := tls.LoadX509KeyPair(certFile, keyFile)
+	g.FatalIf(err)
+
+	config := &tls.Config {
+		MinVersion: tls.VersionTLS13,
+		Certificates: []tls.Certificate {
+			certificate,
+		},
+	}
+
+	listener, err := tls.Listen("unix", fromAddr, config)
+	g.FatalIf(err)
+	g.Info("Started listening", "listen-start", "from-address", fromAddr)
+	return listener
+}
+
+func copyData(c chan struct {}, from io.Reader, to io.WriteCloser) {
+	io.Copy(to, from)
+	c <- struct {} {}
+	// connection is closed, send signal to stop proxy  FIXME
+}
+
+func start(toAddr string, listener net.Listener) {
+	for {
+		connFrom, err := listener.Accept()
+		if err != nil {
+			g.Warning(
+				"Error accepting connection",
+				"accept-connection-error",
+				"err", err,
+			)
+			continue
+		}
+		defer connFrom.Close()
+		emitActiveConnection.Inc()
+
+		connTo, err := net.Dial("unix", toAddr)
+		if err != nil {
+			g.Warning(
+				"Error dialing connection",
+				"dial-connection-error",
+				"err", err,
+			)
+			continue
+		}
+		defer connTo.Close()
+
+		c := make(chan struct {})
+		go copyData(c, connFrom, connTo)
+		go copyData(c, connTo, connFrom)
+		go func() {
+			<- c
+			emitActiveConnection.Dec()
+		}()
+	}
+}
+
+
+
+func Main() {
+	g.Init()
+	args := parseArgs(os.Args)
+	listener := listen(args.certFile, args.keyFile, args.fromAddr)
+	start(args.toAddr, listener)
+}