black-internet-days-ahead/Xray-core
1
0
Fork 0
mirror of https://github.com/XTLS/Xray-core.git synced 2024-11-25 18:11:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix freedom PacketReader/PacketWriter type assertion (#3734)

Browse source
This commit is contained in:
dyhkwong 2024-08-27 22:17:55 +08:00 committed by GitHub
parent 1cfed96aba
commit 815a959c96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 16 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
proxy/freedom/freedom.go
View file

@ -276,24 +276,24 @@ func NewPacketReader(conn net.Conn, UDPOverride net.Destination) buf.Reader {
if statConn != nil { if statConn != nil {
counter = statConn.ReadCounter counter = statConn.ReadCounter
} }
if c, ok := iConn.(*internet.PacketConnWrapper); ok && UDPOverride.Address == nil && UDPOverride.Port == 0 { if c, ok := iConn.(net.PacketConn); ok && UDPOverride.Address == nil && UDPOverride.Port == 0 {
return &PacketReader{ return &PacketReader{
PacketConnWrapper: c, PacketConn: c,
Counter: counter, Counter: counter,
} }
} }
return &buf.PacketReader{Reader: conn} return &buf.PacketReader{Reader: conn}
} }
type PacketReader struct { type PacketReader struct {
*internet.PacketConnWrapper net.PacketConn
stats.Counter stats.Counter
} }
func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) { func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
b := buf.New() b := buf.New()
b.Resize(0, buf.Size) b.Resize(0, buf.Size)
n, d, err := r.PacketConnWrapper.ReadFrom(b.Bytes()) n, d, err := r.PacketConn.ReadFrom(b.Bytes())
if err != nil { if err != nil {
b.Release() b.Release()
return nil, err return nil, err
@ -320,24 +320,26 @@ func NewPacketWriter(conn net.Conn, h *Handler, ctx context.Context, UDPOverride
if statConn != nil { if statConn != nil {
counter = statConn.WriteCounter counter = statConn.WriteCounter
} }
if c, ok := iConn.(*internet.PacketConnWrapper); ok { if c, ok := iConn.(net.PacketConn); ok {
return &PacketWriter{ return &PacketWriter{
PacketConnWrapper: c, PacketConn: c,
Counter: counter, Counter: counter,
Handler: h, Handler: h,
Context: ctx, Context: ctx,
UDPOverride: UDPOverride, UDPOverride: UDPOverride,
Conn: iConn,
} }
} }
return &buf.SequentialWriter{Writer: conn} return &buf.SequentialWriter{Writer: conn}
} }
type PacketWriter struct { type PacketWriter struct {
*internet.PacketConnWrapper net.PacketConn
stats.Counter stats.Counter
*Handler *Handler
context.Context context.Context
UDPOverride net.Destination UDPOverride net.Destination
Conn net.Conn
} }
func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error { func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
@ -367,9 +369,9 @@ func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
b.Release() b.Release()
continue continue
} }
n, err = w.PacketConnWrapper.WriteTo(b.Bytes(), destAddr) n, err = w.PacketConn.WriteTo(b.Bytes(), destAddr)
} else { } else {
n, err = w.PacketConnWrapper.Write(b.Bytes()) n, err = w.Conn.Write(b.Bytes())
} }
b.Release() b.Release()
if err != nil { if err != nil {