update socks5 udp

This commit is contained in:
刘河
2019-12-03 00:46:30 +08:00
parent 8d27a17cdb
commit f1cb45146f
3 changed files with 14 additions and 4 deletions

View File

@@ -225,7 +225,12 @@ func (s *TRPClient) handleUdp(serverConn net.Conn) {
buf := bytes.Buffer{}
dgram := common.NewUDPDatagram(common.NewUDPHeader(0, 0, common.ToSocksAddr(raddr)), b[:n])
dgram.Write(&buf)
if _, err := serverConn.Write(buf.Bytes()); err != nil {
b, err := conn.GetLenBytes(buf.Bytes())
if err != nil {
logs.Warn("get len bytes error", err.Error())
continue
}
if _, err := serverConn.Write(b); err != nil {
logs.Error("write data to remote error", err.Error())
return
}