1
0
Fork 0

Use commandline options

logger
Ambrose Chua 2018-02-14 21:42:26 +08:00
parent a019fa6da5
commit 64b8ee7aaa
2 changed files with 24 additions and 17 deletions

View File

@ -1,17 +1,23 @@
package main package main
import ( import (
"flag"
"github.com/serverwentdown/bypass" "github.com/serverwentdown/bypass"
) )
var listen string
func main() { func main() {
flag.StringVar(&listen, "listen", ":8000", "listen on ip and port")
flag.Parse()
conf := &socks5.Config{} conf := &socks5.Config{}
server, err := socks5.New(conf) server, err := socks5.New(conf)
if err != nil { if err != nil {
panic(err) panic(err)
} }
if err := server.ListenAndServe("tcp", "127.0.0.1:8000"); err != nil { if err := server.ListenAndServe("tcp", listen); err != nil {
panic(err) panic(err)
} }
} }

View File

@ -1,13 +1,14 @@
package socks5 package socks5
import ( import (
"bytes"
"fmt" "fmt"
"io" "io"
"log"
"math/big"
"net" "net"
"strconv" "strconv"
"strings" "strings"
"math/big"
"bytes"
"golang.org/x/net/context" "golang.org/x/net/context"
) )
@ -372,8 +373,8 @@ func copyAndModify(dst io.Writer, src io.Reader) (written int64, err error) {
for { for {
nr, er := src.Read(buf) nr, er := src.Read(buf)
if written == 0 && nr > 0 { if written == 0 && nr > 0 {
if (buf[0] == 0x16 && buf[5] == 0x01) {// && buf[1] == 0x03 && buf[2] == 0x01) { if buf[0] == 0x16 && buf[5] == 0x01 { // && buf[1] == 0x03 && buf[2] == 0x01) {
modifyHello(buf); modifyHello(buf)
} }
} }
if nr > 0 { if nr > 0 {
@ -446,7 +447,7 @@ func modifyHello(buf []byte) {
extensionsEnd := extensionsStart + extensionsLength.Uint64() extensionsEnd := extensionsStart + extensionsLength.Uint64()
cur := extensionsStart cur := extensionsStart
for cur+4 < extensionsEnd { for cur+4 < extensionsEnd {
if (buf[cur] != 0x00 || buf[cur + 1] != 0x00) { if buf[cur] != 0x00 || buf[cur+1] != 0x00 {
extensionLength := big.NewInt(0) extensionLength := big.NewInt(0)
extensionLength.SetBytes(buf[cur+2 : cur+4]) extensionLength.SetBytes(buf[cur+2 : cur+4])
cur += 4 + extensionLength.Uint64() cur += 4 + extensionLength.Uint64()
@ -480,7 +481,7 @@ func modifyHello(buf []byte) {
list = append(list, name.String()) list = append(list, name.String())
lcur += 3 + nameLength.Uint64() lcur += 3 + nameLength.Uint64()
} }
fmt.Println(list) log.Println(list)
break break
} }