github/gosocks5
4
0
Fork 0
mirror of https://github.com/ginuerzh/gosocks5.git synced 2024-08-11 17:54:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix addr encode

Browse Source
This commit is contained in:
rui.zheng
2016-09-03 09:06:34 +08:00
parent 89ddeaf64d
commit bc931b305d
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
socks5.go
+10 -2
View File
@@ -302,13 +302,21 @@ func (addr *Addr) Encode(b []byte) (int, error) {
pos := 1
switch addr.Type {
case AddrIPv4:
pos += copy(b[pos:], net.ParseIP(addr.Host).To4())
ip4 := net.ParseIP(addr.Host).To4()
if ip4 == nil {
ip4 = net.IPv4zero.To4()
}
pos += copy(b[pos:], ip4)
case AddrDomain:
b[pos] = byte(len(addr.Host))
pos++
pos += copy(b[pos:], []byte(addr.Host))
case AddrIPv6:
pos += copy(b[pos:], net.ParseIP(addr.Host).To16())
ip16 := net.ParseIP(addr.Host).To16()
if ip16 == nil {
ip16 = net.IPv6zero.To16()
}
pos += copy(b[pos:], ip16)
default:
b[0] = AddrIPv4
copy(b[pos:pos+4], net.IPv4zero.To4())