mirror of
https://github.com/XTLS/Xray-core.git
synced 2025-06-17 21:59:47 +03:00
Add init check
This commit is contained in:
@ -328,13 +328,22 @@ func NewPacketWriter(conn net.Conn, h *Handler, ctx context.Context, UDPOverride
|
|||||||
counter = statConn.WriteCounter
|
counter = statConn.WriteCounter
|
||||||
}
|
}
|
||||||
if c, ok := iConn.(*internet.PacketConnWrapper); ok {
|
if c, ok := iConn.(*internet.PacketConnWrapper); ok {
|
||||||
|
// If target is a domain, it will be resolved in dialer
|
||||||
|
// check this behavior and add it to map
|
||||||
|
outbounds := session.OutboundsFromContext(ctx)
|
||||||
|
targetAddr := outbounds[len(outbounds)-1].Target.Address
|
||||||
|
resolvedUDPAddr := make(map[string]net.Address)
|
||||||
|
if targetAddr.Family().IsDomain() {
|
||||||
|
RemoteAddress, _, _ := net.SplitHostPort(conn.RemoteAddr().String())
|
||||||
|
resolvedUDPAddr[targetAddr.String()] = net.ParseAddress(RemoteAddress)
|
||||||
|
}
|
||||||
return &PacketWriter{
|
return &PacketWriter{
|
||||||
PacketConnWrapper: c,
|
PacketConnWrapper: c,
|
||||||
Counter: counter,
|
Counter: counter,
|
||||||
Handler: h,
|
Handler: h,
|
||||||
Context: ctx,
|
Context: ctx,
|
||||||
UDPOverride: UDPOverride,
|
UDPOverride: UDPOverride,
|
||||||
resolvedUDPAddr: make(map[string]net.Address),
|
resolvedUDPAddr: resolvedUDPAddr,
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user