lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-02-20 23:47:21 -05:00
Code Issues Releases Wiki Activity

remove AddressFamily.Either()

Browse Source
This commit is contained in:
Darien Raymond 2018-11-16 11:13:36 +01:00
parent a01fdc29a6
commit 2364f5f280
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
4 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/router/condition.go
View File

@ -144,7 +144,7 @@ func (m *MultiGeoIPMatcher) Apply(ctx context.Context) bool {
dest := m.destFunc(ctx) dest := m.destFunc(ctx)
if dest.IsValid() && (dest.Address.Family().IsIPv4() || dest.Address.Family().IsIPv6()) { if dest.IsValid() && dest.Address.Family().IsIP() {
ips = append(ips, dest.Address.IP()) ips = append(ips, dest.Address.IP())
} else if resolver, ok := ResolvedIPsFromContext(ctx); ok { } else if resolver, ok := ResolvedIPsFromContext(ctx); ok {
resolvedIPs := resolver.Resolve() resolvedIPs := resolver.Resolve()

15
common/net/address.go
View File

@ -35,16 +35,6 @@ const (
AddressFamilyDomain = AddressFamily(2) AddressFamilyDomain = AddressFamily(2)
) )
// Either returns true if current AddressFamily matches any of the AddressFamilies provided.
func (af AddressFamily) Either(fs ...AddressFamily) bool {
for _, f := range fs {
if af == f {
return true
}
}
return false
}
// IsIPv4 returns true if current AddressFamily is IPv4. // IsIPv4 returns true if current AddressFamily is IPv4.
func (af AddressFamily) IsIPv4() bool { func (af AddressFamily) IsIPv4() bool {
return af == AddressFamilyIPv4 return af == AddressFamilyIPv4
@ -55,6 +45,11 @@ func (af AddressFamily) IsIPv6() bool {
return af == AddressFamilyIPv6 return af == AddressFamilyIPv6
} }
// IsIP returns true if current AddressFamily is IPv6 or IPv4.
func (af AddressFamily) IsIP() bool {
return af == AddressFamilyIPv4 || af == AddressFamilyIPv6
}
// IsDomain returns true if current AddressFamily is Domain. // IsDomain returns true if current AddressFamily is Domain.
func (af AddressFamily) IsDomain() bool { func (af AddressFamily) IsDomain() bool {
return af == AddressFamilyDomain return af == AddressFamilyDomain

2
common/protocol/address.go
View File

@ -213,7 +213,7 @@ func (p *addressParser) readAddress(b *buf.Buffer, reader io.Reader) (net.Addres
domain := string(b.BytesFrom(-domainLength)) domain := string(b.BytesFrom(-domainLength))
if maybeIPPrefix(domain[0]) { if maybeIPPrefix(domain[0]) {
addr := net.ParseAddress(domain) addr := net.ParseAddress(domain)
if addr.Family().IsIPv4() || addr.Family().IsIPv6() { if addr.Family().IsIP() {
return addr, nil return addr, nil
} }
} }

2
transport/internet/websocket/hub.go
View File

@ -41,7 +41,7 @@ func (h *requestHandler) ServeHTTP(writer http.ResponseWriter, request *http.Req
forwardedAddrs := http_proto.ParseXForwardedFor(request.Header) forwardedAddrs := http_proto.ParseXForwardedFor(request.Header)
remoteAddr := conn.RemoteAddr() remoteAddr := conn.RemoteAddr()
if len(forwardedAddrs) > 0 && forwardedAddrs[0].Family().Either(net.AddressFamilyIPv4, net.AddressFamilyIPv6) { if len(forwardedAddrs) > 0 && forwardedAddrs[0].Family().IsIP() {
remoteAddr.(*net.TCPAddr).IP = forwardedAddrs[0].IP() remoteAddr.(*net.TCPAddr).IP = forwardedAddrs[0].IP()
} }