1 files changed, 79 insertions, 0 deletions
diff --git a/rudp/cmd/proxy/proxy.go b/rudp/cmd/proxy/proxy.go
new file mode 100644
index 0000000..b8ca9f4
--- /dev/null
+++ b/rudp/cmd/proxy/proxy.go
@@ -0,0 +1,79 @@
+/*
+Proxy is a Minetest RUDP proxy server
+supporting multiple concurrent connections.
+
+Usage:
+	proxy dial:port listen:port
+where dial:port is the server address
+and listen:port is the address to listen on.
+*/
+package main
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"os"
+
+	"github.com/anon55555/mt/rudp"
+)
+
+func main() {
+	if len(os.Args) != 3 {
+		fmt.Fprintln(os.Stderr, "usage: proxy dial:port listen:port")
+		os.Exit(1)
+	}
+
+	pc, err := net.ListenPacket("udp", os.Args[2])
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer pc.Close()
+
+	l := rudp.Listen(pc)
+	for {
+		clt, err := l.Accept()
+		if err != nil {
+			log.Print("accept: ", err)
+			continue
+		}
+
+		log.Print(clt.ID(), ": connected")
+
+		conn, err := net.Dial("udp", os.Args[1])
+		if err != nil {
+			log.Print(err)
+			continue
+		}
+		srv := rudp.Connect(conn)
+
+		go proxy(clt, srv)
+		go proxy(srv, clt)
+	}
+}
+
+func proxy(src, dest *rudp.Conn) {
+	s := fmt.Sprint(src.ID(), " (", src.RemoteAddr(), "): ")
+
+	for {
+		pkt, err := src.Recv()
+		if err != nil {
+			if errors.Is(err, net.ErrClosed) {
+				if err := src.WhyClosed(); err != nil {
+					log.Print(s, "disconnected: ", err)
+				} else {
+					log.Print(s, "disconnected")
+				}
+				break
+			}
+
+			log.Print(s, err)
+			continue
+		}
+
+		dest.Send(pkt)
+	}
+
+	dest.Close()
+}