You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.0 KiB
Go
59 lines
1.0 KiB
Go
package socks
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
socks4Version = 0x04
|
|
socks5Version = 0x05
|
|
cmdConnect = 0x01
|
|
addrTypeIPv4 = 0x01
|
|
addrTypeDomain = 0x03
|
|
addrTypeIPv6 = 0x04
|
|
)
|
|
|
|
type dialFunc func(network, addr string) (net.Conn, error)
|
|
|
|
// Conn present a client connection
|
|
type Conn struct {
|
|
net.Conn
|
|
// the function to dial to upstream server
|
|
// when nil, use net.Dial
|
|
Dial dialFunc
|
|
// username for socks5 server
|
|
Username string
|
|
// password
|
|
Password string
|
|
}
|
|
|
|
// Serve serve the client
|
|
func (s *Conn) Serve() {
|
|
buf := make([]byte, 1)
|
|
|
|
// read version
|
|
io.ReadFull(s.Conn, buf)
|
|
|
|
dial := s.Dial
|
|
if s.Dial == nil {
|
|
d := net.Dialer{Timeout: 10 * time.Second}
|
|
dial = d.Dial
|
|
}
|
|
|
|
switch buf[0] {
|
|
case socks4Version:
|
|
s4 := socks4Conn{clientConn: s.Conn, dial: dial}
|
|
s4.Serve()
|
|
case socks5Version:
|
|
s5 := socks5Conn{clientConn: s.Conn, dial: dial,
|
|
username: s.Username, password: s.Password}
|
|
s5.Serve()
|
|
default:
|
|
log.Printf("error version %d", buf[0])
|
|
s.Conn.Close()
|
|
}
|
|
}
|