94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package inbound
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"net/http"
|
|
"net/netip"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/Dreamacro/clash/common/nnip"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/transport/socks5"
|
|
)
|
|
|
|
func parseSocksAddr(target socks5.Addr) *C.Metadata {
|
|
metadata := &C.Metadata{}
|
|
|
|
switch target[0] {
|
|
case socks5.AtypDomainName:
|
|
// trim for FQDN
|
|
metadata.Host = strings.TrimRight(string(target[2:2+target[1]]), ".")
|
|
metadata.DstPort = uint16((int(target[2+target[1]]) << 8) | int(target[2+target[1]+1]))
|
|
case socks5.AtypIPv4:
|
|
metadata.DstIP = nnip.IpToAddr(net.IP(target[1 : 1+net.IPv4len]))
|
|
metadata.DstPort = uint16((int(target[1+net.IPv4len]) << 8) | int(target[1+net.IPv4len+1]))
|
|
case socks5.AtypIPv6:
|
|
ip6, _ := netip.AddrFromSlice(target[1 : 1+net.IPv6len])
|
|
metadata.DstIP = ip6.Unmap()
|
|
metadata.DstPort = uint16((int(target[1+net.IPv6len]) << 8) | int(target[1+net.IPv6len+1]))
|
|
}
|
|
|
|
return metadata
|
|
}
|
|
|
|
func parseHTTPAddr(request *http.Request) *C.Metadata {
|
|
host := request.URL.Hostname()
|
|
port := request.URL.Port()
|
|
if port == "" {
|
|
port = "80"
|
|
}
|
|
|
|
// trim FQDN (#737)
|
|
host = strings.TrimRight(host, ".")
|
|
|
|
var uint16Port uint16
|
|
if port, err := strconv.ParseUint(port, 10, 16); err == nil {
|
|
uint16Port = uint16(port)
|
|
}
|
|
|
|
metadata := &C.Metadata{
|
|
NetWork: C.TCP,
|
|
Host: host,
|
|
DstIP: netip.Addr{},
|
|
DstPort: uint16Port,
|
|
}
|
|
|
|
ip, err := netip.ParseAddr(host)
|
|
if err == nil {
|
|
metadata.DstIP = ip
|
|
}
|
|
|
|
return metadata
|
|
}
|
|
|
|
func parseAddr(addr net.Addr) (netip.AddrPort, error) {
|
|
// Filter when net.Addr interface is nil
|
|
if addr == nil {
|
|
return netip.AddrPort{}, errors.New("nil addr")
|
|
}
|
|
if rawAddr, ok := addr.(interface{ RawAddr() net.Addr }); ok {
|
|
if addrPort, err := parseAddr(rawAddr.RawAddr()); err == nil {
|
|
return addrPort, nil
|
|
}
|
|
}
|
|
if addr, ok := addr.(interface{ AddrPort() netip.AddrPort }); ok {
|
|
if addrPort := addr.AddrPort(); addrPort.IsValid() {
|
|
return addrPort, nil
|
|
}
|
|
}
|
|
addrStr := addr.String()
|
|
host, port, err := net.SplitHostPort(addrStr)
|
|
if err != nil {
|
|
return netip.AddrPort{}, err
|
|
}
|
|
|
|
var uint16Port uint16
|
|
if port, err := strconv.ParseUint(port, 10, 16); err == nil {
|
|
uint16Port = uint16(port)
|
|
}
|
|
|
|
ip, err := netip.ParseAddr(host)
|
|
return netip.AddrPortFrom(ip, uint16Port), err
|
|
}
|