83 lines
1.4 KiB
Go
83 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/ginuerzh/gosocks5"
|
|
"github.com/gorilla/websocket"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type WSConn struct {
|
|
*websocket.Conn
|
|
rb []byte
|
|
}
|
|
|
|
func NewWSConn(conn *websocket.Conn) *WSConn {
|
|
c := &WSConn{
|
|
Conn: conn,
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
func (conn *WSConn) Read(b []byte) (n int, err error) {
|
|
if len(conn.rb) == 0 {
|
|
_, conn.rb, err = conn.ReadMessage()
|
|
}
|
|
n = copy(b, conn.rb)
|
|
conn.rb = conn.rb[n:]
|
|
|
|
//log.Println("ws r:", n)
|
|
|
|
return
|
|
}
|
|
|
|
func (conn *WSConn) Write(b []byte) (n int, err error) {
|
|
err = conn.WriteMessage(websocket.BinaryMessage, b)
|
|
n = len(b)
|
|
//log.Println("ws w:", n)
|
|
|
|
return
|
|
}
|
|
|
|
func (conn *WSConn) SetDeadline(t time.Time) error {
|
|
if err := conn.SetReadDeadline(t); err != nil {
|
|
return err
|
|
}
|
|
return conn.SetWriteDeadline(t)
|
|
}
|
|
|
|
type WSServer struct {
|
|
Addr string
|
|
}
|
|
|
|
var upgrader = websocket.Upgrader{
|
|
ReadBufferSize: 8192,
|
|
WriteBufferSize: 8192,
|
|
CheckOrigin: func(r *http.Request) bool { return true },
|
|
}
|
|
|
|
func (s *WSServer) handle(w http.ResponseWriter, r *http.Request) {
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
//defer conn.Close()
|
|
|
|
c := gosocks5.ServerConn(NewWSConn(conn), serverConfig)
|
|
/*
|
|
if err := c.Handleshake(); err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
*/
|
|
serveSocks5(c)
|
|
}
|
|
|
|
func (s *WSServer) ListenAndServe() error {
|
|
http.HandleFunc("/", s.handle)
|
|
return http.ListenAndServe(s.Addr, nil)
|
|
}
|