mirrors/Xray-core
0
Fork 0
mirror of https://github.com/XTLS/Xray-core.git synced 2025-06-18 06:09:48 +03:00
Code Issues Releases Activity

Add and use new TypedSyncMap

Browse Source
This commit is contained in:
风扇滑翔翼
2025-06-13 08:30:58 +00:00
committed by GitHub
parent f4246e9314
commit 9436956e2a
2 changed files with 64 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
proxy/freedom/freedom.go
View File

@ -27,6 +27,7 @@ import (
"github.com/xtls/xray-core/transport/internet"
"github.com/xtls/xray-core/transport/internet/stat"
"github.com/xtls/xray-core/transport/internet/tls"
"github.com/xtls/xray-core/common/utils"
)
var useSplice bool
@ -343,7 +344,7 @@ func NewPacketWriter(conn net.Conn, h *Handler, ctx context.Context, UDPOverride
Handler: h,
Context: ctx,
UDPOverride: UDPOverride,
resolvedUDPAddr: resolvedUDPAddr,
resolvedUDPAddr: util.NewTypedSyncMap[string, net.Address](),
}
}
@ -361,7 +362,7 @@ type PacketWriter struct {
// But resolver will return a random one if the domain has many IPs
// Resulting in these packets being sent to many different IPs randomly
// So, cache and keep the resolve result
resolvedUDPAddr map[string]net.Address
resolvedUDPAddr *util.TypedSyncMap[string, net.Address]
}
func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
@ -381,13 +382,13 @@ func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
b.UDP.Port = w.UDPOverride.Port
}
if w.Handler.config.hasStrategy() && b.UDP.Address.Family().IsDomain() {
if ip := w.resolvedUDPAddr[b.UDP.Address.Domain()]; ip != nil {
if ip, ok := w.resolvedUDPAddr.Load(b.UDP.Address.Domain()); ok {
b.UDP.Address = ip
} else {
ip := w.Handler.resolveIP(w.Context, b.UDP.Address.Domain(), nil)
if ip != nil {
b.UDP.Address = ip
w.resolvedUDPAddr[b.UDP.Address.Domain()] = ip
w.resolvedUDPAddr.Store(b.UDP.Address.Domain(), ip)
}
}
}