mirror of
https://github.com/nadoo/glider.git
synced 2025-02-23 17:35:40 +08:00
112 lines
2.1 KiB
Go
112 lines
2.1 KiB
Go
package vmess
|
|
|
|
import (
|
|
"net"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/nadoo/glider/log"
|
|
"github.com/nadoo/glider/proxy"
|
|
)
|
|
|
|
// VMess struct.
|
|
type VMess struct {
|
|
dialer proxy.Dialer
|
|
addr string
|
|
|
|
uuid string
|
|
aead bool
|
|
alterID int
|
|
security string
|
|
|
|
client *Client
|
|
}
|
|
|
|
func init() {
|
|
proxy.RegisterDialer("vmess", NewVMessDialer)
|
|
}
|
|
|
|
// NewVMess returns a vmess proxy.
|
|
func NewVMess(s string, d proxy.Dialer) (*VMess, error) {
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
log.F("parse url err: %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
addr := u.Host
|
|
security := u.User.Username()
|
|
uuid, ok := u.User.Password()
|
|
if !ok {
|
|
// no security type specified, vmess://uuid@server
|
|
uuid = security
|
|
security = ""
|
|
}
|
|
|
|
query := u.Query()
|
|
aid := query.Get("alterID")
|
|
if aid == "" {
|
|
aid = "0"
|
|
}
|
|
|
|
alterID, err := strconv.ParseUint(aid, 10, 32)
|
|
if err != nil {
|
|
log.F("parse alterID err: %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
aead := alterID == 0
|
|
client, err := NewClient(uuid, security, int(alterID), aead)
|
|
if err != nil {
|
|
log.F("create vmess client err: %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
p := &VMess{
|
|
dialer: d,
|
|
addr: addr,
|
|
uuid: uuid,
|
|
alterID: int(alterID),
|
|
security: security,
|
|
client: client,
|
|
}
|
|
|
|
return p, nil
|
|
}
|
|
|
|
// NewVMessDialer returns a vmess proxy dialer.
|
|
func NewVMessDialer(s string, dialer proxy.Dialer) (proxy.Dialer, error) {
|
|
return NewVMess(s, dialer)
|
|
}
|
|
|
|
// Addr returns forwarder's address.
|
|
func (s *VMess) Addr() string {
|
|
if s.addr == "" {
|
|
return s.dialer.Addr()
|
|
}
|
|
return s.addr
|
|
}
|
|
|
|
// Dial connects to the address addr on the network net via the proxy.
|
|
func (s *VMess) Dial(network, addr string) (net.Conn, error) {
|
|
rc, err := s.dialer.Dial("tcp", s.addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return s.client.NewConn(rc, addr, CmdTCP)
|
|
}
|
|
|
|
// DialUDP connects to the given address via the proxy.
|
|
func (s *VMess) DialUDP(network, addr string) (net.PacketConn, net.Addr, error) {
|
|
rc, err := s.dialer.Dial("tcp", s.addr)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
rc, err = s.client.NewConn(rc, addr, CmdUDP)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return NewPktConn(rc), nil, err
|
|
}
|