fix: unmap 4in6 ip
This commit is contained in:
parent
1a232b7504
commit
9bd516bc62
4 changed files with 22 additions and 36 deletions
|
@ -38,27 +38,24 @@ func WithSpecialProxy(specialProxy string) Addition {
|
||||||
|
|
||||||
func WithSrcAddr(addr net.Addr) Addition {
|
func WithSrcAddr(addr net.Addr) Addition {
|
||||||
return func(metadata *C.Metadata) {
|
return func(metadata *C.Metadata) {
|
||||||
if addrPort, err := parseAddr(addr); err == nil {
|
addrPort := parseAddr(addr)
|
||||||
metadata.SrcIP = addrPort.Addr()
|
metadata.SrcIP = addrPort.Addr().Unmap()
|
||||||
metadata.SrcPort = addrPort.Port()
|
metadata.SrcPort = addrPort.Port()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func WithDstAddr(addr net.Addr) Addition {
|
func WithDstAddr(addr net.Addr) Addition {
|
||||||
return func(metadata *C.Metadata) {
|
return func(metadata *C.Metadata) {
|
||||||
if addrPort, err := parseAddr(addr); err == nil {
|
addrPort := parseAddr(addr)
|
||||||
metadata.DstIP = addrPort.Addr()
|
metadata.DstIP = addrPort.Addr().Unmap()
|
||||||
metadata.DstPort = addrPort.Port()
|
metadata.DstPort = addrPort.Port()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func WithInAddr(addr net.Addr) Addition {
|
func WithInAddr(addr net.Addr) Addition {
|
||||||
return func(metadata *C.Metadata) {
|
return func(metadata *C.Metadata) {
|
||||||
if addrPort, err := parseAddr(addr); err == nil {
|
addrPort := parseAddr(addr)
|
||||||
metadata.InIP = addrPort.Addr()
|
metadata.InIP = addrPort.Addr().Unmap()
|
||||||
metadata.InPort = addrPort.Port()
|
metadata.InPort = addrPort.Port()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
|
@ -16,9 +16,9 @@ func SkipAuthPrefixes() []netip.Prefix {
|
||||||
}
|
}
|
||||||
|
|
||||||
func SkipAuthRemoteAddr(addr net.Addr) bool {
|
func SkipAuthRemoteAddr(addr net.Addr) bool {
|
||||||
if addrPort, err := parseAddr(addr); err == nil {
|
if addrPort := parseAddr(addr); addrPort.IsValid() {
|
||||||
for _, prefix := range skipAuthPrefixes {
|
for _, prefix := range skipAuthPrefixes {
|
||||||
if prefix.Contains(addrPort.Addr()) {
|
if prefix.Contains(addrPort.Addr().Unmap()) {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
package inbound
|
package inbound
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
|
@ -63,25 +62,23 @@ func parseHTTPAddr(request *http.Request) *C.Metadata {
|
||||||
return metadata
|
return metadata
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseAddr(addr net.Addr) (netip.AddrPort, error) {
|
func parseAddr(addr net.Addr) netip.AddrPort {
|
||||||
// Filter when net.Addr interface is nil
|
// Filter when net.Addr interface is nil
|
||||||
if addr == nil {
|
if addr == nil {
|
||||||
return netip.AddrPort{}, errors.New("nil addr")
|
return netip.AddrPort{}
|
||||||
}
|
}
|
||||||
if rawAddr, ok := addr.(interface{ RawAddr() net.Addr }); ok {
|
if addr, ok := addr.(interface{ RawAddr() net.Addr }); ok {
|
||||||
if addrPort, err := parseAddr(rawAddr.RawAddr()); err == nil {
|
if rawAddr := addr.RawAddr(); rawAddr != nil {
|
||||||
return addrPort, nil
|
return parseAddr(rawAddr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if addr, ok := addr.(interface{ AddrPort() netip.AddrPort }); ok {
|
if addr, ok := addr.(interface{ AddrPort() netip.AddrPort }); ok {
|
||||||
if addrPort := addr.AddrPort(); addrPort.IsValid() {
|
return addr.AddrPort()
|
||||||
return addrPort, nil
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
addrStr := addr.String()
|
addrStr := addr.String()
|
||||||
host, port, err := net.SplitHostPort(addrStr)
|
host, port, err := net.SplitHostPort(addrStr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return netip.AddrPort{}, err
|
return netip.AddrPort{}
|
||||||
}
|
}
|
||||||
|
|
||||||
var uint16Port uint16
|
var uint16Port uint16
|
||||||
|
@ -89,6 +86,6 @@ func parseAddr(addr net.Addr) (netip.AddrPort, error) {
|
||||||
uint16Port = uint16(port)
|
uint16Port = uint16(port)
|
||||||
}
|
}
|
||||||
|
|
||||||
ip, err := netip.ParseAddr(host)
|
ip, _ := netip.ParseAddr(host)
|
||||||
return netip.AddrPortFrom(ip, uint16Port), err
|
return netip.AddrPortFrom(ip, uint16Port)
|
||||||
}
|
}
|
||||||
|
|
|
@ -93,12 +93,8 @@ func (h *ListenerHandler) NewConnection(ctx context.Context, conn net.Conn, meta
|
||||||
NetWork: C.TCP,
|
NetWork: C.TCP,
|
||||||
Type: h.Type,
|
Type: h.Type,
|
||||||
Host: metadata.Destination.Fqdn,
|
Host: metadata.Destination.Fqdn,
|
||||||
DstIP: metadata.Destination.Addr,
|
|
||||||
DstPort: metadata.Destination.Port,
|
|
||||||
SrcIP: metadata.Source.Addr,
|
|
||||||
SrcPort: metadata.Source.Port,
|
|
||||||
}
|
}
|
||||||
additions := combineAdditions(ctx, h.Additions, inbound.WithInAddr(conn.LocalAddr()))
|
additions := combineAdditions(ctx, h.Additions, inbound.WithDstAddr(metadata.Destination), inbound.WithSrcAddr(metadata.Source), inbound.WithInAddr(conn.LocalAddr()))
|
||||||
for _, addition := range additions {
|
for _, addition := range additions {
|
||||||
addition.Apply(cMetadata)
|
addition.Apply(cMetadata)
|
||||||
}
|
}
|
||||||
|
@ -160,12 +156,8 @@ func (h *ListenerHandler) NewPacketConnection(ctx context.Context, conn network.
|
||||||
NetWork: C.UDP,
|
NetWork: C.UDP,
|
||||||
Type: h.Type,
|
Type: h.Type,
|
||||||
Host: dest.Fqdn,
|
Host: dest.Fqdn,
|
||||||
DstIP: dest.Addr,
|
|
||||||
DstPort: dest.Port,
|
|
||||||
SrcIP: metadata.Source.Addr,
|
|
||||||
SrcPort: metadata.Source.Port,
|
|
||||||
}
|
}
|
||||||
additions := combineAdditions(ctx, h.Additions, inbound.WithInAddr(conn.LocalAddr()))
|
additions := combineAdditions(ctx, h.Additions, inbound.WithDstAddr(dest), inbound.WithSrcAddr(metadata.Source), inbound.WithInAddr(conn.LocalAddr()))
|
||||||
for _, addition := range additions {
|
for _, addition := range additions {
|
||||||
addition.Apply(cMetadata)
|
addition.Apply(cMetadata)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue