mirror of
https://github.com/necronicle/z2k.git
synced 2026-04-28 11:30:30 +00:00
e688fb1963
- Build all 9 arch binaries with Go 1.22.12 to fix MIPS crash (Go issue #71591) - Remove dead MTProxy/transparent mode code (relay.go, secret.go, transparent.go, dcmap.go) - Drop gotd/td dependency — only gorilla/websocket + stdlib remain - Tunnel is now the only mode, --tunnel flag removed - Transparent WS reconnect: keep client TCP alive during CF Worker WS drops - Re-CONNECT surviving streams after WS reconnects — seamless for clients - streamReadLoop waits for WS instead of dying on disconnect - New connections wait up to 5s for WS during reconnect instead of dropping - Drain connect semaphore on WS disconnect to prevent deadlock - Worker: MAX_STREAMS 200→100, improved logging Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
var (
|
|
listenAddr = flag.String("listen", ":1443", "Local listen address")
|
|
tunnelURL = flag.String("tunnel-url", "wss://z2k-tunnel.necronicle.workers.dev/ws", "Cloudflare Worker WebSocket URL")
|
|
tunnelSecret = flag.String("tunnel-secret", "d01f72f9543b29da4e3724b1530c0d11cb30a6f8db15bc0adfe8f2d37b5844b2", "Shared secret for tunnel auth")
|
|
verbose = flag.Bool("v", false, "Verbose logging")
|
|
connTimeout = flag.Duration("timeout", 5*time.Minute, "Idle connection timeout")
|
|
maxConns = flag.Int("max-conns", 1024, "Maximum concurrent connections")
|
|
)
|
|
|
|
// connSemaphore limits concurrent connections
|
|
var connSemaphore chan struct{}
|
|
|
|
// wsWriter serializes all writes to a WebSocket connection.
|
|
// gorilla/websocket supports only one concurrent writer.
|
|
type wsWriter struct {
|
|
ws *websocket.Conn
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func (w *wsWriter) WriteMessage(messageType int, data []byte) error {
|
|
w.mu.Lock()
|
|
defer w.mu.Unlock()
|
|
w.ws.SetWriteDeadline(time.Now().Add(10 * time.Second))
|
|
return w.ws.WriteMessage(messageType, data)
|
|
}
|
|
|
|
func (w *wsWriter) WriteControl(messageType int, data []byte, deadline time.Time) error {
|
|
w.mu.Lock()
|
|
defer w.mu.Unlock()
|
|
return w.ws.WriteControl(messageType, data, deadline)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if err := runTunnel(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|