基于go手动写个转发代理服务的代码实现

2019-11-10 12:16:17王冬梅

由于公司经常需要异地办公,在调试的时候需要用到内网环境,因此手动写了个代理转发服务器給兄弟们用:socks5proxy。

选型上,语言上就选择了Go,简单清晰,转发协议选择了socks5。

SOCKS5协议介绍

SOCKS是一种网络传输协议,主要用于客户端与外网服务器之间通讯的中间传递,SOCKS是"SOCKetS"的缩写。 SOCKS5是SOCKS4的升级版,其主要多了鉴定、IPv6、UDP支持。

SOCKS5协议可以分为三个部分:

(1) 协议版本及认证方式
(2) 根据认证方式执行对应的认证
(3) 请求信息

(1)协议版本及认证方式

创建与SOCKS5服务器的TCP连接后客户端需要先发送请求来协议版本及认证方式,

VER NMETHODS METHODS
1 1 1-255

VER是SOCKS版本,这里应该是0x05; NMETHODS是METHODS部分的长度; METHODS是客户端支持的认证方式列表,每个方法占1字节。当前的定义是: 0x00 不需要认证 0x01 GSSAPI 0x02 用户名、密码认证 0x03 - 0x7F由IANA分配(保留) 0x80 - 0xFE为私人方法保留 0xFF 无可接受的方法

服务器回复客户端可用方法:

VER METHOD
1 1

VER是SOCKS版本,这里应该是0x05; METHOD是服务端选中的方法。如果返回0xFF表示没有一个认证方法被选中,客户端需要关闭连接。

代码实现:

type ProtocolVersion struct {
  VER uint8
  NMETHODS uint8
  METHODS []uint8
}


func (s *ProtocolVersion) handshake(conn net.Conn) error {
  b := make([]byte, 255)
  n, err := conn.Read(b)
  if err != nil {
    log.Println(err)
    return err
  }
  s.VER = b[0] //ReadByte reads and returns a single byte,第一个参数为socks的版本号
  s.NMETHODS = b[1] //nmethods是记录methods的长度的。nmethods的长度是1个字节
  if n != int(2+s.NMETHODS) {
    return errors.New("协议错误, sNMETHODS不对")
  }
  s.METHODS = b[2:2+s.NMETHODS] //读取指定长度信息,读取正好len(buf)长度的字节。如果字节数不是指定长度,则返回错误信息和正确的字节数

  if s.VER != 5 {
    return errors.New("该协议不是socks5协议")
  }

  //服务器回应客户端消息:
  //第一个参数表示版本号为5,即socks5协议,
  // 第二个参数表示服务端选中的认证方法,0即无需密码访问, 2表示需要用户名和密码进行验证。 
  resp :=[]byte{5, 0} 
  conn.Write(resp)
  return nil
} 

(2)根据认证方式执行对应的认证

SOCKS5协议提供5种认证方式:

    0x00 不需要认证 0x01 GSSAPI 0x02 用户名、密码认证 0x03 - 0x7F由IANA分配(保留) 0x80 - 0xFE为私人方法保留