120 lines
2.7 KiB
Go
120 lines
2.7 KiB
Go
package common
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"errors"
|
|
"fmt"
|
|
"regexp"
|
|
|
|
"net"
|
|
"strconv"
|
|
)
|
|
|
|
type Addr struct {
|
|
scheam string
|
|
user, passwd string
|
|
ip string
|
|
port int
|
|
httpAuthorizationHeader string
|
|
}
|
|
|
|
func ParseAddr(str string) (cfg *Addr, err error) {
|
|
defer func() {
|
|
if cfg != nil && cfg.user != "" && cfg.passwd != "" {
|
|
cfg.httpAuthorizationHeader = fmt.Sprintf("Proxy-Authorization: Basic %s", base64.URLEncoding.EncodeToString([]byte(cfg.user+":"+cfg.passwd)))
|
|
|
|
}
|
|
|
|
}()
|
|
r, _ := regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)@(\S+):(\d+)`)
|
|
m := r.FindAllStringSubmatch(str, 1)
|
|
|
|
if m != nil {
|
|
|
|
addr, err := net.ResolveTCPAddr("tcp", m[0][4]+":"+m[0][5])
|
|
if err != nil {
|
|
return nil, errors.New("配置解析错误 " + m[0][4] + ":" + m[0][5] + " 不是有效的 地址:端口")
|
|
}
|
|
return &Addr{
|
|
scheam: m[0][1],
|
|
user: m[0][2],
|
|
passwd: m[0][3],
|
|
ip: m[0][4],
|
|
port: addr.Port,
|
|
}, nil
|
|
}
|
|
r, _ = regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)@(\d+)`)
|
|
m = r.FindAllStringSubmatch(str, 1)
|
|
|
|
if m != nil {
|
|
|
|
port, _ := strconv.Atoi(m[0][4])
|
|
return &Addr{
|
|
scheam: m[0][1],
|
|
user: m[0][2],
|
|
passwd: m[0][3],
|
|
ip: "",
|
|
port: port,
|
|
}, nil
|
|
}
|
|
r, _ = regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)$`)
|
|
m = r.FindAllStringSubmatch(str, 1)
|
|
|
|
if m != nil {
|
|
|
|
addr, err := net.ResolveTCPAddr("tcp", m[0][2]+":"+m[0][3])
|
|
if err != nil {
|
|
return nil, errors.New("配置解析错误 " + m[0][1] + ":" + m[0][2] + " 不是有效的 地址:端口")
|
|
}
|
|
|
|
return &Addr{
|
|
scheam: m[0][1],
|
|
user: "",
|
|
passwd: "",
|
|
ip: m[0][2],
|
|
port: addr.Port,
|
|
}, nil
|
|
}
|
|
|
|
port, err := strconv.Atoi(str)
|
|
if err != nil {
|
|
return nil, errors.New("配置解析错误，请按照 用户名:密码@地址:端口 的方式填写，或者 用户名:密码@端口 或者 ip:端口 或者 只有端口")
|
|
}
|
|
return &Addr{port: port}, nil
|
|
}
|
|
func (c *Addr) IP() string {
|
|
return c.ip
|
|
}
|
|
func (c *Addr) Addr() string {
|
|
return fmt.Sprintf("%s:%d", c.ip, c.port)
|
|
}
|
|
func (c *Addr) Port() string {
|
|
return fmt.Sprintf("%d", c.port)
|
|
}
|
|
func (c *Addr) String() string {
|
|
if c.user == "" && c.passwd == "" {
|
|
if c.ip == "" {
|
|
return fmt.Sprintf("%d", c.port)
|
|
}
|
|
|
|
return fmt.Sprintf("%s%s:%d", c.scheam, c.ip, c.port)
|
|
}
|
|
|
|
return fmt.Sprintf("%s%s:%s@%s:%d", c.scheam, c.user, c.passwd, c.ip, c.port)
|
|
}
|
|
func (c *Addr) GetHttpAuthorizationHeader() string {
|
|
return c.httpAuthorizationHeader
|
|
}
|
|
func (c *Addr) User() string {
|
|
return c.user
|
|
}
|
|
func (c *Addr) Password() string {
|
|
return c.passwd
|
|
}
|
|
func (c *Addr) Scheam() string {
|
|
return c.scheam
|
|
}
|
|
func (c *Addr) HttpUrl() string {
|
|
return "http://" + c.Addr()
|
|
}
|