cy/goproxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Signed-off-by: arraykeys@gmail.com <arraykeys@gmail.com>

Browse Source
This commit is contained in:
arraykeys@gmail.com
2018-03-09 17:16:21 +08:00
parent 7bb8f19b90
commit 1cf4313d12
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
config.go
View File

@ -264,6 +264,12 @@ func initConfig() (err error) {
*kcpArgs.Crypt = "aes" *kcpArgs.Crypt = "aes"
kcpArgs.Block, _ = kcp.NewAESBlockCrypt(pass) kcpArgs.Block, _ = kcp.NewAESBlockCrypt(pass)
} }
//attach kcp config
tcpArgs.KCP = kcpArgs
httpArgs.KCP = kcpArgs
socksArgs.KCP = kcpArgs
spsArgs.KCP = kcpArgs
flags := log.Ldate flags := log.Ldate
if *debug { if *debug {
flags |= log.Lshortfile | log.Lmicroseconds flags |= log.Lshortfile | log.Lmicroseconds

10
utils/serve-channel.go
View File

@ -142,15 +142,19 @@ func (sc *ServerChannel) ListenUDP(fn func(packet []byte, localAddr, srcAddr *ne
func (sc *ServerChannel) ListenKCP(config kcpcfg.KCPConfigArgs, fn func(conn net.Conn)) (err error) { func (sc *ServerChannel) ListenKCP(config kcpcfg.KCPConfigArgs, fn func(conn net.Conn)) (err error) {
lis, err := kcp.ListenWithOptions(fmt.Sprintf("%s:%d", sc.ip, sc.port), config.Block, *config.DataShard, *config.ParityShard) lis, err := kcp.ListenWithOptions(fmt.Sprintf("%s:%d", sc.ip, sc.port), config.Block, *config.DataShard, *config.ParityShard)
if err == nil { if err == nil {
if err := lis.SetDSCP(*config.DSCP); err != nil { if err = lis.SetDSCP(*config.DSCP); err != nil {
log.Println("SetDSCP:", err) log.Println("SetDSCP:", err)
return
} }
if err := lis.SetReadBuffer(*config.SockBuf); err != nil { if err = lis.SetReadBuffer(*config.SockBuf); err != nil {
log.Println("SetReadBuffer:", err) log.Println("SetReadBuffer:", err)
return
} }
if err := lis.SetWriteBuffer(*config.SockBuf); err != nil { if err = lis.SetWriteBuffer(*config.SockBuf); err != nil {
log.Println("SetWriteBuffer:", err) log.Println("SetWriteBuffer:", err)
return
} }
sc.Listener = new(net.Listener)
*sc.Listener = lis *sc.Listener = lis
go func() { go func() {
defer func() { defer func() {