github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-09-06 07:06:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

module

Browse Source
This commit is contained in:
unknown
2019-10-15 17:12:27 +08:00
parent f4ae503982
commit 6771816be7
8 changed files with 153 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
server/socks5/socks5_read_request_handle.go
View File

@@ -32,7 +32,7 @@ const (
addrTypeNotSupported = 8
)
func (request *Request) Run(ctx context.Context, config map[string]string) error {
func (request *Request) Run(ctx context.Context, config map[string]string) (context.Context, error) {
request.clientConn = request.GetClientConn(ctx)
request.ctx = ctx
@@ -49,23 +49,23 @@ func (request *Request) Run(ctx context.Context, config map[string]string) error
_, err := io.ReadFull(request.clientConn, header)
if err != nil {
return errors.New("illegal request" + err.Error())
return request.ctx, errors.New("illegal request" + err.Error())
}
switch header[1] {
case connectMethod:
context.WithValue(request.ctx, core.PROXY_CONNECTION_TYPE, "tcp")
return request.doConnect()
request.ctx = context.WithValue(request.ctx, core.PROXY_CONNECTION_TYPE, "tcp")
return request.ctx, request.doConnect()
case bindMethod:
return request.handleBind()
return request.ctx, request.handleBind()
case associateMethod:
context.WithValue(request.ctx, core.PROXY_CONNECTION_TYPE, "udp")
return request.handleUDP()
request.ctx = context.WithValue(request.ctx, core.PROXY_CONNECTION_TYPE, "udp")
return request.ctx, request.handleUDP()
default:
request.sendReply(commandNotSupported)
return errors.New("command not supported")
return request.ctx, errors.New("command not supported")
}
return nil
return request.ctx, nil
}
func (request *Request) sendReply(rep uint8) error {
@@ -116,8 +116,8 @@ func (request *Request) doConnect() error {
var port uint16
binary.Read(request.clientConn, binary.BigEndian, &port)
context.WithValue(request.ctx, core.PROXY_CONNECTION_ADDR, host)
context.WithValue(request.ctx, core.PROXY_CONNECTION_PORT, port)
request.ctx = context.WithValue(request.ctx, core.PROXY_CONNECTION_ADDR, host)
request.ctx = context.WithValue(request.ctx, core.PROXY_CONNECTION_PORT, port)
return nil
}