update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package udp 
 
import ( 
    "context" 
 
    "github.com/v2fly/v2ray-core/v5/common/buf" 
    "github.com/v2fly/v2ray-core/v5/common/net" 
    "github.com/v2fly/v2ray-core/v5/common/net/packetaddr" 
    "github.com/v2fly/v2ray-core/v5/common/protocol/udp" 
    "github.com/v2fly/v2ray-core/v5/features/routing" 
) 
 
type PacketAddrDispatcher struct { 
    conn     net.PacketConn 
    callback ResponseCallback 
    ctx      context.Context 
} 
 
func (p PacketAddrDispatcher) Close() error { 
    return p.conn.Close() 
} 
 
func (p PacketAddrDispatcher) Dispatch(ctx context.Context, destination net.Destination, payload *buf.Buffer) { 
    if destination.Network != net.Network_UDP { 
        return 
    } 
    p.conn.WriteTo(payload.Bytes(), &net.UDPAddr{IP: destination.Address.IP(), Port: int(destination.Port.Value())}) 
} 
 
func (p PacketAddrDispatcher) readWorker() { 
    for { 
        readBuf := buf.New() 
        n, addr, err := p.conn.ReadFrom(readBuf.Extend(2048)) 
        if err != nil { 
            return 
        } 
        readBuf.Resize(0, int32(n)) 
        p.callback(p.ctx, &udp.Packet{Payload: readBuf, Source: net.DestinationFromAddr(addr)}) 
    } 
} 
 
type PacketAddrDispatcherCreator struct { 
    ctx context.Context 
} 
 
func NewPacketAddrDispatcherCreator(ctx context.Context) PacketAddrDispatcherCreator { 
    return PacketAddrDispatcherCreator{ctx: ctx} 
} 
 
func (pdc *PacketAddrDispatcherCreator) NewPacketAddrDispatcher( 
    dispatcher routing.Dispatcher, callback ResponseCallback, 
) DispatcherI { 
    packetConn, _ := packetaddr.CreatePacketAddrConn(pdc.ctx, dispatcher, false) 
    pd := &PacketAddrDispatcher{conn: packetConn, callback: callback, ctx: pdc.ctx} 
    go pd.readWorker() 
    return pd 
}