From 63f16e018851760367f1974125f5ca33361fb3e2 Mon Sep 17 00:00:00 2001
From: Dingjun <fangdingjun@gmail.com>
Date: Tue, 6 Dec 2016 11:00:18 +0800
Subject: [PATCH] add password auth for socks5

---
 socks5.go | 99 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 96 insertions(+), 3 deletions(-)

diff --git a/socks5.go b/socks5.go
index d624a4d..43b51d0 100644
--- a/socks5.go
+++ b/socks5.go
@@ -1,6 +1,7 @@
 package socks
 
 import (
+	"errors"
 	"fmt"
 	"io"
 	"log"
@@ -24,6 +25,16 @@ byte | 0  |  1  |
      |0x05| auth|
 
 
+username/password auth request
+
+byte | 0  |  1         |          |     1 byte   |          |
+     |0x01|username_len| username | password_len | password |
+
+username/password auth reponse
+
+byte | 0  | 1    |
+     |0x01|status|
+
 request
 
 byte | 0  | 1 | 2  |   3    | 4 | .. | n-2 | n-1| n |
@@ -34,6 +45,9 @@ byte |0   |  1   | 2  |   3    | 4 | .. | n-2 | n-1 | n |
      |0x05|status|0x00|addrtype|     addr     |  port   |
 
 */
+
+var Socks5AuthRequired bool
+
 type socks5Conn struct {
 	//addr        string
 	clientConn net.Conn
@@ -70,14 +84,93 @@ func (s5 *socks5Conn) handshake() error {
 	l := int(buf[0]) + 1
 	if n < l {
 		// read remains data
-		_, err := io.ReadFull(s5.clientConn, buf[n:l])
+		if n1, err := io.ReadFull(s5.clientConn, buf[n:l]); err != nil {
+			return err
+		} else {
+			n += n1
+		}
+	}
+
+	if !Socks5AuthRequired {
+		// no auth required
+		s5.clientConn.Write([]byte{0x05, 0x00})
+		return nil
+	}
+
+	hasPassAuth := false
+	var passAuth byte = 0x02
+
+	// check auth method
+	// only password(0x02) supported
+	for i := 1; i < n; i++ {
+		if buf[i] == passAuth {
+			hasPassAuth = true
+			break
+		}
+	}
+
+	if !hasPassAuth {
+		s5.clientConn.Write([]byte{0x05, 0xff})
+		return errors.New("no supported auth method")
+	}
+
+	err = s5.passwordAuth()
+	return err
+}
+
+func (s5 *socks5Conn) passwordAuth() error {
+	buf := make([]byte, 32)
+
+	// username/password required
+	s5.clientConn.Write([]byte{0x05, 0x02})
+	n, err := io.ReadAtLeast(s5.clientConn, buf, 2)
+	if err != nil {
+		return err
+	}
+
+	//log.Printf("%+v", buf[:n])
+
+	// check auth version
+	if buf[0] != 0x01 {
+		return errors.New("unsupported auth version")
+	}
+
+	username_len := int(buf[1])
+
+	p0 := 2
+	p1 := p0 + username_len
+
+	if n < p1 {
+		n1, err := s5.clientConn.Read(buf[n:])
+		if err != nil {
+			return err
+		}
+		n += n1
+	}
+
+	username := buf[p0:p1]
+	password_len := int(buf[p1])
+
+	p3 := p1 + 1
+	p4 := p3 + password_len
+
+	if n < p4 {
+		n1, err := s5.clientConn.Read(buf[n:])
 		if err != nil {
 			return err
 		}
+		n += n1
 	}
 
-	// no auth required
-	s5.clientConn.Write([]byte{0x05, 0x00})
+	password := buf[p3:p4]
+
+	log.Printf("get username: %s, password: %s", username, password)
+
+	if string(username) == "" && string(password) == "" {
+		s5.clientConn.Write([]byte{0x01, 0x01})
+	} else {
+		s5.clientConn.Write([]byte{0x01, 0x00})
+	}
 
 	return nil
 }