go-socks5/_example/testconnect/main.go

70 lines
1.4 KiB
Go

package main
import (
"io"
"log"
"net"
"os"
"time"
"git.tcp.direct/kayos/go-socks5"
"git.tcp.direct/kayos/go-socks5/ccsocks5"
)
func handleErr(err error) {
if err != nil {
panic(err)
}
}
func main() {
// Create a local listener
l, err := net.Listen("tcp", "127.0.0.1:0")
handleErr(err)
go func() {
conn, err := l.Accept()
handleErr(err)
defer conn.Close()
for {
buf := make([]byte, 4)
_, err = io.ReadAtLeast(conn, buf, 4)
handleErr(err)
log.Printf("server: %+v", string(buf))
conn.Write([]byte("pong")) // nolint: errcheck
}
}()
lAddr := l.Addr().(*net.TCPAddr)
go func() {
time.Sleep(time.Second * 1)
client := ccsocks5.NewClient("127.0.0.1:10808")
con, err := client.Dial("tcp", lAddr.String())
handleErr(err)
for {
con.Write([]byte("ping")) // nolint: errcheck
out := make([]byte, 4)
con.SetDeadline(time.Now().Add(time.Second)) // nolint: errcheck
_, err = io.ReadFull(con, out)
con.SetDeadline(time.Time{}) // nolint: errcheck
if err != nil {
log.Printf("client: %+v", err)
} else {
log.Printf("client: %+v", string(out))
}
time.Sleep(time.Second * 2)
}
}()
// Create a SOCKS5 server
server := socks5.NewServer(socks5.WithLogger(socks5.NewLogger(log.New(os.Stdout, "socks5: ", log.LstdFlags))))
// Create SOCKS5 proxy on localhost port 8000
if err := server.ListenAndServe("tcp", "127.0.0.1:10808"); err != nil {
panic(err)
}
}