clash/adapters/inbound/util.go

73 lines
1.5 KiB
Go
Raw Normal View History

package adapters
import (
"net"
2018-08-12 08:18:58 +00:00
"net/http"
"strconv"
2019-04-25 05:48:47 +00:00
"github.com/Dreamacro/clash/component/socks5"
C "github.com/Dreamacro/clash/constant"
)
2019-04-25 05:48:47 +00:00
func parseSocksAddr(target socks5.Addr) *C.Metadata {
2018-12-05 13:13:29 +00:00
metadata := &C.Metadata{
AddrType: int(target[0]),
}
switch target[0] {
2019-04-25 05:48:47 +00:00
case socks5.AtypDomainName:
2018-12-05 13:13:29 +00:00
metadata.Host = string(target[2 : 2+target[1]])
2019-05-09 13:00:29 +00:00
metadata.DstPort = strconv.Itoa((int(target[2+target[1]]) << 8) | int(target[2+target[1]+1]))
2019-04-25 05:48:47 +00:00
case socks5.AtypIPv4:
2018-12-05 13:13:29 +00:00
ip := net.IP(target[1 : 1+net.IPv4len])
2019-05-09 13:00:29 +00:00
metadata.DstIP = &ip
metadata.DstPort = strconv.Itoa((int(target[1+net.IPv4len]) << 8) | int(target[1+net.IPv4len+1]))
2019-04-25 05:48:47 +00:00
case socks5.AtypIPv6:
2018-12-05 13:13:29 +00:00
ip := net.IP(target[1 : 1+net.IPv6len])
2019-05-09 13:00:29 +00:00
metadata.DstIP = &ip
metadata.DstPort = strconv.Itoa((int(target[1+net.IPv6len]) << 8) | int(target[1+net.IPv6len+1]))
}
2018-12-05 13:13:29 +00:00
return metadata
}
2018-09-30 04:25:52 +00:00
func parseHTTPAddr(request *http.Request) *C.Metadata {
2018-08-26 16:06:40 +00:00
host := request.URL.Hostname()
port := request.URL.Port()
if port == "" {
port = "80"
}
2018-12-05 13:13:29 +00:00
metadata := &C.Metadata{
NetWork: C.TCP,
2019-05-09 13:00:29 +00:00
Type: C.HTTP,
2018-12-05 13:13:29 +00:00
AddrType: C.AtypDomainName,
Host: host,
2019-05-09 13:00:29 +00:00
DstIP: nil,
DstPort: port,
}
2018-12-05 13:13:29 +00:00
ip := net.ParseIP(host)
if ip != nil {
switch {
case ip.To4() == nil:
metadata.AddrType = C.AtypIPv6
default:
metadata.AddrType = C.AtypIPv4
}
2019-05-09 13:00:29 +00:00
metadata.DstIP = &ip
2018-12-05 13:13:29 +00:00
}
return metadata
}
2019-05-09 13:00:29 +00:00
func parseAddr(addr string) (*net.IP, string, error) {
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, "", err
}
2019-05-09 13:00:29 +00:00
ip := net.ParseIP(host)
return &ip, port, nil
}