func main() {
var (
outTraffic int64
inTraffic int64
conns int64
)
go util.RecordTraffic(&inTraffic, &outTraffic, &conns)
l, _ := net.Listen("tcp", "127.0.0.1:9091")
for {
rc, _ := l.Accept()
go func(conn net.Conn) {
// read addr.
b := make([]byte, 4096)
n, err := conn.Read(b)
if err != nil {
println(err.Error())
return
}
b = b[:n]
addrAndOther := b[1:]
addr := string(addrAndOther[:int(b[0])])
c, err := net.Dial("tcp", addr)
if err != nil {
println("connect err:", err.Error())
return
}
atomic.AddInt64(&conns, 1)
if len(addrAndOther[int(b[0]):]) > 0 {
println("I need write.", len(addrAndOther[int(b[0]):]))
c.Write(addrAndOther[int(b[0]):])
}
go util.Copy(c, conn, &outTraffic)
util.Copy(conn, c, &outTraffic)
atomic.AddInt64(&conns, -1)
}(rc)
}
}
func main() {
// record traffic.
var (
outTraffic int64
inTraffic int64
conns int64
// remote addr(host and port).
remote = "127.0.0.1:9091"
rand = -1
)
go util.RecordTraffic(&inTraffic, &outTraffic, &conns)
logger := log.New(os.Stdout, "", log.Ldate|log.Lshortfile)
listen, err := net.Listen("tcp", "127.0.0.1:9090")
if err != nil {
panic(err)
}
socks5 := proxy.Socks5Listen{
HandleConnect: func(addr string) (*net.TCPConn, error) {
// logger.Println("connet addr:=", addr)
rand *= -1
// rand = -1
if rand == 1 {
conn, err := net.Dial("tcp", addr)
if err != nil {
logger.Println("connect error:=", err)
return nil, err
}
return conn.(*net.TCPConn), nil
} else {
conn, err := net.Dial("tcp", remote)
if err != nil {
logger.Println("connect remote error:=", err)
return nil, err
}
data := []byte{byte(len(addr))}
_, err = conn.Write(append(data, []byte(addr)...))
if err != nil {
logger.Println("conn write error::", err)
}
return conn.(*net.TCPConn), nil
}
},
Transport: func(target net.Conn, client net.Conn) error {
atomic.AddInt64(&conns, 1)
defer atomic.AddInt64(&conns, -1)
go util.Copy(client, target, &inTraffic)
_, err := util.Copy(target, client, &outTraffic)
return err
},
Auth: func(id, pwd []byte) bool {
logger.Println(len(id), len(pwd))
logger.Printf("user(%s) pwd(%s)", id, pwd)
return true
},
HandleAssociate: proxy.DefaultHandleAssociate,
TransportUdp: proxy.DefaultTransportUdp,
AddrForClient: "127.0.0.1",
RawListen: listen,
}
socks5.Listen()
}