-
Notifications
You must be signed in to change notification settings - Fork 49
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
245 additions
and
39 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,154 @@ | ||
| package socks5 | ||
|
|
||
| import ( | ||
| "encoding/binary" | ||
| "io" | ||
| "net" | ||
|
|
||
| "github.com/pkg/errors" | ||
| ) | ||
|
|
||
| func ToSocks5(c net.Conn, domain, port string) net.Conn { | ||
| switch port { | ||
| case "80": | ||
| return &conn{init: make(chan struct{}), Conn: c, domain: domain, port: []byte{0x00, 0x50}} | ||
| case "443": | ||
| return &conn{init: make(chan struct{}), Conn: c, domain: domain, port: []byte{0x01, 0xbb}} | ||
| default: | ||
| panic("invalid port: " + port) | ||
| } | ||
| } | ||
|
|
||
| type conn struct { | ||
| init chan struct{} | ||
| domain string | ||
| port []byte | ||
| net.Conn | ||
| } | ||
|
|
||
| func (c *conn) Read(b []byte) (n int, err error) { | ||
| <-c.init | ||
| return c.Conn.Read(b) | ||
| } | ||
|
|
||
| func (c *conn) Write(b []byte) (n int, err error) { | ||
| select { | ||
| case <-c.init: | ||
| return c.Conn.Write(b) | ||
| default: | ||
| } | ||
|
|
||
| { | ||
| req := &authReq{ | ||
| VER: 5, | ||
| NMETHODS: 1, | ||
| METHODS: [1]byte{0}, // NO AUTHENTICATION REQUIRED | ||
| } | ||
| if err := binary.Write(c.Conn, binary.BigEndian, req); err != nil { | ||
| return 0, err | ||
| } | ||
| } | ||
| { | ||
| resp := &authResp{} | ||
| if err := binary.Read(c.Conn, binary.BigEndian, resp); err != nil { | ||
| return 0, err | ||
| } | ||
| } | ||
| { | ||
| req := &request{ | ||
| req: req{ | ||
| VER: 5, // socks5 | ||
| CMD: 1, // CONNECT | ||
| RSV: 0, // RESERVED | ||
| ATYP: 3, // DOMAINNAME | ||
| }, | ||
| DST_ADDR: append([]byte{byte(len(c.domain))}, []byte(c.domain)...), | ||
| DST_PORT: c.port, | ||
| } | ||
|
|
||
| if _, err := c.Conn.Write(req.Bytes()); err != nil { | ||
| return 0, err | ||
| } | ||
| } | ||
| { | ||
| resp := &response{} | ||
| if err := binary.Read(c.Conn, binary.BigEndian, &(resp.resp)); err != nil { | ||
| return 0, err | ||
| } | ||
|
|
||
| switch resp.REP { | ||
| case 0x00: | ||
| default: | ||
| return 0, errors.Errorf("socks5 handshake fail, return code: %d", resp.REP) | ||
| } | ||
|
|
||
| switch resp.ATYP { | ||
| case 0x01: // IPv4 | ||
| resp.DST_ADDR = make([]byte, net.IPv4len) | ||
| if _, err := io.ReadFull(c.Conn, resp.DST_ADDR); err != nil { | ||
| return 0, err | ||
| } | ||
| case 0x03: // domain name | ||
| if _, err := io.ReadFull(c.Conn, resp.DST_ADDR[:1]); err != nil { | ||
| return 0, err | ||
| } | ||
| if _, err := io.ReadFull(c.Conn, resp.DST_ADDR[1:1+int(resp.DST_ADDR[0])]); err != nil { | ||
| return 0, err | ||
| } | ||
| case 0x04: | ||
| resp.DST_ADDR = make([]byte, net.IPv6len) | ||
| if _, err := io.ReadFull(c.Conn, resp.DST_ADDR); err != nil { | ||
| return 0, err | ||
| } | ||
| } | ||
|
|
||
| resp.DST_PORT = make([]byte, 2) | ||
| if _, err := io.ReadFull(c.Conn, resp.DST_PORT); err != nil { | ||
| return 0, err | ||
| } | ||
| } | ||
|
|
||
| close(c.init) | ||
| return c.Conn.Write(b) | ||
| } | ||
|
|
||
| type authReq struct { | ||
| VER byte | ||
| NMETHODS byte | ||
| METHODS [1]byte // 1 to 255, fix to no authentication | ||
| } | ||
|
|
||
| type authResp struct { | ||
| VER byte | ||
| METHOD byte | ||
| } | ||
|
|
||
| type request struct { | ||
| req | ||
| DST_ADDR []byte // first byte is length | ||
| DST_PORT []byte // two bytes | ||
| } | ||
| type req struct { | ||
| VER byte | ||
| CMD byte | ||
| RSV byte | ||
| ATYP byte | ||
| } | ||
|
|
||
| func (r *request) Bytes() []byte { | ||
| out := []byte{r.VER, r.CMD, r.RSV, r.ATYP} | ||
| out = append(out, r.DST_ADDR...) | ||
| return append(out, r.DST_PORT...) | ||
| } | ||
|
|
||
| type response struct { | ||
| resp | ||
| DST_ADDR []byte // first byte is length | ||
| DST_PORT []byte // two bytes | ||
| } | ||
| type resp struct { | ||
| VER byte | ||
| REP byte | ||
| RSV byte | ||
| ATYP byte | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.