123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
package prox5
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/miekg/dns"
|
|
"net/netip"
|
|
)
|
|
|
|
func filterv6(in string) (filtered string, ok bool) {
|
|
split := strings.Split(in, "]:")
|
|
if len(split) < 2 {
|
|
|
|
return "", false
|
|
}
|
|
split2 := strings.Split(split[1], ":")
|
|
switch len(split2) {
|
|
case 0:
|
|
combo, err := netip.ParseAddrPort(buildProxyString("", "", split[0], split2[0], true))
|
|
if err == nil {
|
|
return combo.String(), true
|
|
}
|
|
case 1:
|
|
concat := buildProxyString("", "", split[0], split2[0], true)
|
|
combo, err := netip.ParseAddrPort(concat)
|
|
if err == nil {
|
|
return combo.String(), true
|
|
}
|
|
default:
|
|
_, err := netip.ParseAddrPort(buildProxyString("", "", split[0], split2[0], true))
|
|
if err == nil {
|
|
return buildProxyString(split2[1], split2[2], split[0], split2[0], true), true
|
|
}
|
|
}
|
|
return "", true
|
|
}
|
|
|
|
func isNumber(s string) bool {
|
|
_, err := strconv.Atoi(s)
|
|
return err == nil
|
|
}
|
|
|
|
func buildProxyString(username, password, address, port string, v6 bool) (result string) {
|
|
builder := strs.Get()
|
|
defer strs.MustPut(builder)
|
|
if username != "" && password != "" {
|
|
builder.MustWriteString(username)
|
|
builder.MustWriteString(":")
|
|
builder.MustWriteString(password)
|
|
builder.MustWriteString("@")
|
|
}
|
|
builder.MustWriteString(address)
|
|
if v6 {
|
|
builder.MustWriteString("]")
|
|
}
|
|
builder.MustWriteString(":")
|
|
builder.MustWriteString(port)
|
|
return builder.String()
|
|
}
|
|
|
|
func filter(in string) (filtered string, ok bool) { //nolint:cyclop
|
|
if !strings.Contains(in, ":") {
|
|
return "", false
|
|
}
|
|
split := strings.Split(in, ":")
|
|
|
|
if len(split) < 2 {
|
|
return "", false
|
|
}
|
|
switch len(split) {
|
|
case 2:
|
|
_, isDomain := dns.IsDomainName(split[0])
|
|
if isDomain && isNumber(split[1]) {
|
|
return in, true
|
|
}
|
|
combo, err := netip.ParseAddrPort(in)
|
|
if err != nil {
|
|
return "", false
|
|
}
|
|
return combo.String(), true
|
|
case 3:
|
|
if !strings.Contains(in, "@") {
|
|
return "", false
|
|
}
|
|
split := strings.Split(in, "@")
|
|
if !strings.Contains(split[0], ":") {
|
|
return "", false
|
|
}
|
|
splitAuth := strings.Split(split[0], ":")
|
|
splitServ := strings.Split(split[1], ":")
|
|
_, isDomain := dns.IsDomainName(splitServ[0])
|
|
if isDomain && isNumber(splitServ[1]) {
|
|
return buildProxyString(splitAuth[0], splitAuth[1],
|
|
splitServ[0], splitServ[1], false), true
|
|
}
|
|
if _, err := netip.ParseAddrPort(split[1]); err == nil {
|
|
return buildProxyString(splitAuth[0], splitAuth[1],
|
|
splitServ[0], splitServ[1], false), true
|
|
}
|
|
case 4:
|
|
_, isDomain := dns.IsDomainName(split[0])
|
|
if isDomain && isNumber(split[1]) {
|
|
return buildProxyString(split[2], split[3], split[0], split[1], false), true
|
|
}
|
|
_, isDomain = dns.IsDomainName(split[2])
|
|
if isDomain && isNumber(split[3]) {
|
|
return buildProxyString(split[0], split[1], split[2], split[3], false), true
|
|
}
|
|
if _, err := netip.ParseAddrPort(split[2] + ":" + split[3]); err == nil {
|
|
return buildProxyString(split[0], split[1], split[2], split[3], false), true
|
|
}
|
|
if _, err := netip.ParseAddrPort(split[0] + ":" + split[1]); err == nil {
|
|
return buildProxyString(split[2], split[3], split[0], split[1], false), true
|
|
}
|
|
default:
|
|
if !strings.Contains(in, "[") || !strings.Contains(in, "]:") {
|
|
return "", false
|
|
}
|
|
}
|
|
return filterv6(in)
|
|
}
|