tlsrp

commit e084c3ba526d43b65d2fdbe2fd041d5522aa4cc1
parent e8c36f4afc94b27c6a22d433f1474aa9885e33e9
Author: Robert Russell <robertrussell.72001@gmail.com>
Date:   Sun, 14 Jul 2024 22:20:04 -0700

Fix cast bug

Diffstat:
M
tlsrp.go
 | 
9
+++++++--

1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/tlsrp.go b/tlsrp.go
@@ -314,7 +314,12 @@ func handshake(conn *tls.Conn) error {
 	return err
 }
 
-func splice(a, b *tls.Conn) error {
+type conn interface {
+	io.ReadWriteCloser
+	CloseWrite() error
+}
+
+func splice(a, b conn) error {
 	a2bErr := make(chan error, 1)
 	go func() {
 		_, err := io.Copy(b, a)
@@ -383,7 +388,7 @@ func proxy(client *tls.Conn) {
 	}
 	defer sink.Close()
 
-	err = splice(client, sink.(*tls.Conn))
+	err = splice(client, sink.(conn))
 	if err != nil {
 		logf("splice error: %s", err)
 		return