3b6bb15e86
d9b8fa6098d fix: Fake broadcast address for 127.x.x.x aa649165a57 chore: Add code for future netprof TCP testing 9e5693de5ac chore: add to_string functions for netprof enums 52d915e6a90 cleanup: Heap allocate network profile objects 80fabd4a729 feat: Implement Tox network profiler 05abe083cb6 cleanup: Some random cleanups, mostly related to mem. 5cca24513b8 cleanup: Check that onion IP/Port packing worked. e092ecd1244 cleanup: Use tox memory allocator in some more places. 3cfe41c7587 fix: Avoid `memcpy`-ing structs into onion ping id data. e32ac001938 fix: Add more information on why the frame was not sent. ab887003687 fix: Allow TCP connections to fail `connect` calls. 7603170e663 refactor: Use tox memory in group connection allocations. 5bd8a85eb89 cleanup: Align internal logger with external on type of source line. e9bf524d9e1 cleanup: Add missing `#include` to sort_test.cc. d10c966b998 feat: Add `to_string` functions for toxencryptsave errors. 7bfd0dc8003 docs: Update the docs for group join functions 380dde9f2ae test: Add more logging to TCP connection constructor. 0f12f384c8c cleanup: Reduce stack frame sizes to below 4096 bytes. bc43cec0626 chore: Happy new year! fbe78f1702e cleanup: Add a `TOX_HIDE_DEPRECATED` check to hide deprecated symbols. 44d9da07e77 refactor: Use tox memory for group moderation/pack allocations. 7f26d520168 refactor: Use tox memory in group chats allocations. 2f62f3d0e77 refactor: Use tox Memory for group allocations. 8a968162041 chore: Add dispatch/events headers to bazel export. 2bbfb35abf6 docs: Output the error code string instead of int. in toxav logging d55d0e4eaef cleanup: Remove redundant code for checking if group exists 2a6dc643338 chore: Upgrade dependencies for websockify. fc0650601c1 fix: Allow peers to reconnect to group chats using a password git-subtree-dir: external/toxcore/c-toxcore git-subtree-split: d9b8fa6098de6c074038b6664d2572627540b148
116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
// A Go version WebSocket to TCP socket proxy
|
|
//
|
|
// This is a heavily modified version of this file:
|
|
// https://github.com/novnc/websockify-other/blob/master/golang/websockify.go
|
|
//
|
|
// Changes include:
|
|
// - Fix infinite loop on error.
|
|
// - Proper logging.
|
|
// - Proper error handling in general.
|
|
// - Support both websocket and regular GET requests on /.
|
|
//
|
|
// Copyright 2022-2025 The TokTok team.
|
|
// Copyright 2021 Michael.liu.
|
|
// See LICENSE for licensing conditions.
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
var (
|
|
sourceAddr = flag.String("l", "127.0.0.1:8080", "http service address")
|
|
targetAddr = flag.String("t", "127.0.0.1:5900", "tcp service address")
|
|
)
|
|
|
|
var upgrader = websocket.Upgrader{
|
|
// Should be enough to fit any Tox TCP packets.
|
|
ReadBufferSize: 2048,
|
|
WriteBufferSize: 2048,
|
|
Subprotocols: []string{"binary"},
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true
|
|
},
|
|
}
|
|
|
|
func forwardTCP(wsconn *websocket.Conn, conn net.Conn) {
|
|
var tcpbuffer [2048]byte
|
|
defer wsconn.Close()
|
|
defer conn.Close()
|
|
for {
|
|
n, err := conn.Read(tcpbuffer[0:])
|
|
if err != nil {
|
|
log.Println("TCP READ :", err)
|
|
break
|
|
}
|
|
log.Println("TCP READ :", n, hex.EncodeToString(tcpbuffer[0:n]))
|
|
|
|
if err := wsconn.WriteMessage(websocket.BinaryMessage, tcpbuffer[0:n]); err != nil {
|
|
log.Println("WS WRITE :", err)
|
|
break
|
|
}
|
|
log.Println("WS WRITE :", n)
|
|
}
|
|
}
|
|
|
|
func forwardWeb(wsconn *websocket.Conn, conn net.Conn) {
|
|
defer wsconn.Close()
|
|
defer conn.Close()
|
|
for {
|
|
_, buffer, err := wsconn.ReadMessage()
|
|
if err != nil {
|
|
log.Println("WS READ :", err)
|
|
break
|
|
}
|
|
log.Println("WS READ :", len(buffer), hex.EncodeToString(buffer))
|
|
|
|
m, err := conn.Write(buffer)
|
|
if err != nil {
|
|
log.Println("TCP WRITE:", err)
|
|
break
|
|
}
|
|
log.Println("TCP WRITE:", m)
|
|
}
|
|
}
|
|
|
|
func serveWs(w http.ResponseWriter, r *http.Request) {
|
|
ws, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
log.Println("upgrade:", err)
|
|
return
|
|
}
|
|
vnc, err := net.Dial("tcp", *targetAddr)
|
|
if err != nil {
|
|
log.Println("dial:", err)
|
|
return
|
|
}
|
|
go forwardTCP(ws, vnc)
|
|
go forwardWeb(ws, vnc)
|
|
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
log.Println("Starting up websockify endpoint")
|
|
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Header.Get("Upgrade") == "websocket" {
|
|
serveWs(w, r)
|
|
} else {
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
w.WriteHeader(http.StatusNotFound)
|
|
|
|
fmt.Fprintf(w, "404 Not Found")
|
|
}
|
|
})
|
|
log.Fatal(http.ListenAndServe(*sourceAddr, nil))
|
|
}
|