This version should work but has issues

Author: simsor

.gitignore

@@ -0,0 +1 @@
+*.exe

cmd/main.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+"flag"
+
+"github.com/simsor/quakejs-proxy/proxy"
+"github.com/sirupsen/logrus"
+)
+
+var (
+wsServer = flag.String("ws", "127.0.0.1:27961", "Hostname of the WebSocket server")
+listen = flag.String("listen", "", "Host to listen on")
+)
+
+func main() {
+flag.Parse()
+
+server := proxy.New(*listen, *wsServer)
+
+logrus.WithFields(logrus.Fields{
+"listen": *listen,
+"dest": *wsServer,
+}).Info("Proxy server listening")
+
+err := server.Start()
+if err != nil {
+panic(err)
+}
+server.Close()
+}

go.mod

@@ -1,3 +1,8 @@
 module github.com/simsor/quakejs-proxy
 
 go 1.14
+
+require (
+github.com/gorilla/websocket v1.4.2
+github.com/sirupsen/logrus v1.7.0
+)

go.sum

@@ -0,0 +1,9 @@
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
+github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/sirupsen/logrus v1.7.0 h1:ShrD1U9pZB12TX0cVy0DtePoCH97K8EtX+mg7ZARUtM=
+github.com/sirupsen/logrus v1.7.0/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+golang.org/x/sys v0.0.0-20191026070338-33540a1f6037 h1:YyJpGZS1sBuBCzLAR1VEpK193GlqGZbnPFnPV/5Rsb4=
+golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=

proxy/agent.go

@@ -0,0 +1,55 @@
+package proxy
+
+import (
+"encoding/hex"
+"fmt"
+"net"
+
+"github.com/gorilla/websocket"
+"github.com/sirupsen/logrus"
+)
+
+type agent struct {
+remote net.Addr
+sock net.PacketConn
+ws *websocket.Conn
+}
+
+func (a *agent) ws2sock() {
+log := logrus.WithFields(logrus.Fields{
+"remote": a.remote.String(),
+"func": "ws2sock",
+})
+defer a.ws.Close()
+
+for {
+t, p, err := a.ws.ReadMessage()
+if err != nil {
+log.WithField("err", err).Error("Could not read WebSocket message")
+return
+}
+log.Info("Got WS message")
+
+if t != websocket.BinaryMessage {
+log.Error("Wrong message type")
+return
+}
+
+n, err := a.sock.WriteTo(p, a.remote)
+if err != nil {
+log.WithError(err).Error("Could not write to socket")
+return
+}
+
+if n != len(p) {
+log.WithFields(logrus.Fields{
+"written": n,
+"length": len(p),
+}).Error("Did not write full message")
+return
+}
+log.Info("Successfully sent WS message to socket")
+
+fmt.Println(hex.Dump(p))
+}
+}

proxy/server.go

@@ -0,0 +1,105 @@
+package proxy
+
+import (
+"encoding/hex"
+"fmt"
+"net"
+"net/url"
+
+"github.com/gorilla/websocket"
+"github.com/sirupsen/logrus"
+)
+
+// SocketServer is the proxy server
+type SocketServer struct {
+ListenAddress string
+ListenPort int
+Destination string
+
+listener net.PacketConn
+agents map[string]*agent
+}
+
+// New creates a new socket to websocket proxy server listening on the given address, on port 27960
+func New(listen, dest string) SocketServer {
+return SocketServer{
+ListenAddress: listen,
+ListenPort: 27960,
+Destination: dest,
+agents: make(map[string]*agent),
+}
+}
+
+func (s *SocketServer) listen() error {
+// Socket server
+listener, err := net.ListenPacket("udp4", fmt.Sprintf("%s:%d", s.ListenAddress, s.ListenPort))
+if err != nil {
+return err
+}
+
+s.listener = listener
+return nil
+}
+
+// Start is a blocking call which starts the proxy server
+func (s *SocketServer) Start() error {
+err := s.listen()
+if err != nil {
+return err
+}
+
+p := make([]byte, 65535)
+
+for {
+n, addr, err := s.listener.ReadFrom(p)
+if err != nil {
+logrus.WithFields(logrus.Fields{
+"err": err,
+}).Error("Could not ReadFrom() a connection")
+continue
+}
+//logrus.WithField("remote", addr.String()).Info("Accepted UDP connection")
+
+a := s.getAgent(addr)
+err = a.ws.WriteMessage(websocket.BinaryMessage, p[:n])
+if err != nil {
+logrus.WithFields(logrus.Fields{
+"err": err,
+"remote": addr.String(),
+}).Error("Could not WriteMessage()")
+continue
+}
+logrus.WithField("remote", addr.String()).Info("Successfully sent UDP message to WebSocket")
+fmt.Println(hex.Dump(p[:n]))
+}
+
+}
+
+// Close closes both sockets
+func (s *SocketServer) Close() error {
+return s.listener.Close()
+}
+
+func (s *SocketServer) getAgent(ip net.Addr) *agent {
+a, ok := s.agents[ip.String()]
+if ok {
+return a
+}
+
+// WebSocket client
+u := url.URL{Scheme: "ws", Host: s.Destination, Path: "/"}
+ws, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
+if err != nil {
+panic(err)
+}
+
+a = &agent{
+ws: ws,
+sock: s.listener,
+remote: ip,
+}
+go a.ws2sock()
+s.agents[ip.String()] = a
+
+return a
+}