fix older version sms code

[bugfix] use thread-local statistic vars

core/protocol.go

@@ -168,7 +168,7 @@ func BlockTXStream(server string, token *[48]byte, ipRev *[4]byte, ep *EasyConne
 	errCh := make(chan error)
 
 	ep.OnRecv = func(buf []byte) {
-		n, err = conn.Write(buf)
+		var n, err = conn.Write(buf)
 		if err != nil {
 			errCh <- err
 			return

core/web_login.go

@@ -104,7 +104,7 @@ func WebLogin(server string, username string, password string) (string, error) {
 	// log.Printf("First stage login response: %s", string(buf[:n]))
 
 	// SMS Code Process
-	if strings.Contains(string(buf[:n]), "<NextService>auth/sms</NextService>") {
+	if strings.Contains(string(buf[:n]), "<NextService>auth/sms</NextService>") || strings.Contains(string(buf[:n]), "<NextAuth>2</NextAuth>") {
 		log.Print("SMS code required.")
 
 		addr = server + "/por/login_sms.csp?apiversion=1"
@@ -121,7 +121,7 @@ func WebLogin(server string, username string, password string) (string, error) {
 		n, _ := resp.Body.Read(buf)
 		defer resp.Body.Close()
 
-		if !strings.Contains(string(buf[:n]), "验证码已发送到您的手机") {
+		if !strings.Contains(string(buf[:n]), "验证码已发送到您的手机") && !strings.Contains(string(buf[:n]), "<USER_PHONE>") {
 			debug.PrintStack()
 			return "", errors.New("unexpected sms resp: " + string(buf[:n]))
 		}

main.go

@@ -5,6 +5,7 @@ import (
 	"flag"
 	"fmt"
 	"log"
+	"runtime"
 )
 
 func main() {
@@ -51,4 +52,6 @@ func main() {
 	log.Printf("Login success, your IP: %d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3])
 
 	client.ServeSocks5(socksBind, debugDump)
+
+	runtime.KeepAlive(client)
 }