SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-04 04:17:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: AssociatePacketConn.ReadFrom, AssociatePacketConn.ReadPacket, AssociatePacketReadWaiter.WaitReadPacket remoteAddr被conn remoteAddr覆盖问题

Browse source
This commit is contained in:
pengjunwen 2024-05-10 19:53:20 +08:00
parent 4ab8cac5eb
commit de04c8b788
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
protocol/socks/packet.go
View file

@ -49,20 +49,20 @@ func (c *AssociatePacketConn) RemoteAddr() net.Addr {
//warn:unsafe
func (c *AssociatePacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
n, addr, err = c.NetPacketConn.ReadFrom(p)
n, _, err = c.NetPacketConn.ReadFrom(p)
if err != nil {
return
}
if n < 3 {
return 0, nil, ErrInvalidPacket
}
c.remoteAddr = M.SocksaddrFromNet(addr)
reader := bytes.NewReader(p[3:n])
destination, err := M.SocksaddrSerializer.ReadAddrPort(reader)
if err != nil {
return
}
addr = destination.UDPAddr()
c.remoteAddr = M.SocksaddrFromNet(addr)
index := 3 + int(reader.Size()) - reader.Len()
n = copy(p, p[index:n])
return
@ -95,19 +95,19 @@ func (c *AssociatePacketConn) Write(b []byte) (n int, err error) {
}
func (c *AssociatePacketConn) ReadPacket(buffer *buf.Buffer) (destination M.Socksaddr, err error) {
destination, err = c.NetPacketConn.ReadPacket(buffer)
_, err = c.NetPacketConn.ReadPacket(buffer)
if err != nil {
return M.Socksaddr{}, err
}
if buffer.Len() < 3 {
return M.Socksaddr{}, ErrInvalidPacket
}
c.remoteAddr = destination
buffer.Advance(3)
destination, err = M.SocksaddrSerializer.ReadAddrPort(buffer)
if err != nil {
return
}
c.remoteAddr = destination
return destination.Unwrap(), nil
}

4
protocol/socks/packet_wait.go
View file

@ -29,7 +29,7 @@ func (w *AssociatePacketReadWaiter) InitializeReadWaiter(options N.ReadWaitOptio
}
func (w *AssociatePacketReadWaiter) WaitReadPacket() (buffer *buf.Buffer, destination M.Socksaddr, err error) {
buffer, destination, err = w.readWaiter.WaitReadPacket()
buffer, _, err = w.readWaiter.WaitReadPacket()
if err != nil {
return
}
@ -37,12 +37,12 @@ func (w *AssociatePacketReadWaiter) WaitReadPacket() (buffer *buf.Buffer, destin
buffer.Release()
return nil, M.Socksaddr{}, ErrInvalidPacket
}
w.conn.remoteAddr = destination
buffer.Advance(3)
destination, err = M.SocksaddrSerializer.ReadAddrPort(buffer)
if err != nil {
buffer.Release()
return nil, M.Socksaddr{}, err
}
w.conn.remoteAddr = destination
return
}