2018-09-06 02:53:29 +00:00
|
|
|
package vmess
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"crypto/aes"
|
|
|
|
"crypto/cipher"
|
|
|
|
"crypto/hmac"
|
|
|
|
"crypto/md5"
|
|
|
|
"encoding/binary"
|
|
|
|
"errors"
|
|
|
|
"hash/fnv"
|
|
|
|
"io"
|
|
|
|
"math/rand"
|
|
|
|
"net"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"golang.org/x/crypto/chacha20poly1305"
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Conn wrapper a net.Conn with vmess protocol
|
|
|
|
type Conn struct {
|
|
|
|
net.Conn
|
|
|
|
reader io.Reader
|
|
|
|
writer io.Writer
|
|
|
|
dst *DstAddr
|
|
|
|
id *ID
|
|
|
|
reqBodyIV []byte
|
|
|
|
reqBodyKey []byte
|
|
|
|
respBodyIV []byte
|
|
|
|
respBodyKey []byte
|
|
|
|
respV byte
|
|
|
|
security byte
|
|
|
|
|
|
|
|
received bool
|
|
|
|
}
|
|
|
|
|
|
|
|
func (vc *Conn) Write(b []byte) (int, error) {
|
|
|
|
return vc.writer.Write(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (vc *Conn) Read(b []byte) (int, error) {
|
|
|
|
if vc.received {
|
|
|
|
return vc.reader.Read(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := vc.recvResponse(); err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
vc.received = true
|
|
|
|
return vc.reader.Read(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (vc *Conn) sendRequest() error {
|
2018-11-06 09:34:19 +00:00
|
|
|
timestamp := time.Now()
|
2018-09-06 02:53:29 +00:00
|
|
|
|
|
|
|
h := hmac.New(md5.New, vc.id.UUID.Bytes())
|
2018-11-06 09:34:19 +00:00
|
|
|
binary.Write(h, binary.BigEndian, uint64(timestamp.Unix()))
|
2018-09-06 02:53:29 +00:00
|
|
|
_, err := vc.Conn.Write(h.Sum(nil))
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
buf := &bytes.Buffer{}
|
|
|
|
|
|
|
|
// Ver IV Key V Opt
|
|
|
|
buf.WriteByte(Version)
|
|
|
|
buf.Write(vc.reqBodyIV[:])
|
|
|
|
buf.Write(vc.reqBodyKey[:])
|
|
|
|
buf.WriteByte(vc.respV)
|
|
|
|
buf.WriteByte(OptionChunkStream)
|
|
|
|
|
|
|
|
p := rand.Intn(16)
|
|
|
|
// P Sec Reserve Cmd
|
|
|
|
buf.WriteByte(byte(p<<4) | byte(vc.security))
|
|
|
|
buf.WriteByte(0)
|
2019-04-25 08:32:15 +00:00
|
|
|
if vc.dst.UDP {
|
|
|
|
buf.WriteByte(CommandUDP)
|
|
|
|
} else {
|
|
|
|
buf.WriteByte(CommandTCP)
|
|
|
|
}
|
2018-09-06 02:53:29 +00:00
|
|
|
|
|
|
|
// Port AddrType Addr
|
|
|
|
binary.Write(buf, binary.BigEndian, uint16(vc.dst.Port))
|
|
|
|
buf.WriteByte(vc.dst.AddrType)
|
|
|
|
buf.Write(vc.dst.Addr)
|
|
|
|
|
|
|
|
// padding
|
|
|
|
if p > 0 {
|
|
|
|
padding := make([]byte, p)
|
|
|
|
rand.Read(padding)
|
|
|
|
buf.Write(padding)
|
|
|
|
}
|
|
|
|
|
|
|
|
fnv1a := fnv.New32a()
|
|
|
|
fnv1a.Write(buf.Bytes())
|
|
|
|
buf.Write(fnv1a.Sum(nil))
|
|
|
|
|
|
|
|
block, err := aes.NewCipher(vc.id.CmdKey)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2018-11-06 09:34:19 +00:00
|
|
|
stream := cipher.NewCFBEncrypter(block, hashTimestamp(timestamp))
|
2018-09-06 02:53:29 +00:00
|
|
|
stream.XORKeyStream(buf.Bytes(), buf.Bytes())
|
|
|
|
_, err = vc.Conn.Write(buf.Bytes())
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (vc *Conn) recvResponse() error {
|
|
|
|
block, err := aes.NewCipher(vc.respBodyKey[:])
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
stream := cipher.NewCFBDecrypter(block, vc.respBodyIV[:])
|
|
|
|
buf := make([]byte, 4)
|
|
|
|
_, err = io.ReadFull(vc.Conn, buf)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
stream.XORKeyStream(buf, buf)
|
|
|
|
|
|
|
|
if buf[0] != vc.respV {
|
|
|
|
return errors.New("unexpected response header")
|
|
|
|
}
|
|
|
|
|
|
|
|
if buf[2] != 0 {
|
|
|
|
return errors.New("dynamic port is not supported now")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func hashTimestamp(t time.Time) []byte {
|
|
|
|
md5hash := md5.New()
|
|
|
|
ts := make([]byte, 8)
|
2018-11-06 09:34:19 +00:00
|
|
|
binary.BigEndian.PutUint64(ts, uint64(t.Unix()))
|
2018-09-06 02:53:29 +00:00
|
|
|
md5hash.Write(ts)
|
|
|
|
md5hash.Write(ts)
|
|
|
|
md5hash.Write(ts)
|
|
|
|
md5hash.Write(ts)
|
|
|
|
return md5hash.Sum(nil)
|
|
|
|
}
|
|
|
|
|
|
|
|
// newConn return a Conn instance
|
2019-02-15 13:55:15 +00:00
|
|
|
func newConn(conn net.Conn, id *ID, dst *DstAddr, security Security) (*Conn, error) {
|
2018-09-06 02:53:29 +00:00
|
|
|
randBytes := make([]byte, 33)
|
|
|
|
rand.Read(randBytes)
|
|
|
|
reqBodyIV := make([]byte, 16)
|
|
|
|
reqBodyKey := make([]byte, 16)
|
|
|
|
copy(reqBodyIV[:], randBytes[:16])
|
|
|
|
copy(reqBodyKey[:], randBytes[16:32])
|
|
|
|
respV := randBytes[32]
|
|
|
|
|
|
|
|
respBodyKey := md5.Sum(reqBodyKey[:])
|
|
|
|
respBodyIV := md5.Sum(reqBodyIV[:])
|
|
|
|
|
|
|
|
var writer io.Writer
|
|
|
|
var reader io.Reader
|
|
|
|
switch security {
|
|
|
|
case SecurityNone:
|
|
|
|
reader = newChunkReader(conn)
|
|
|
|
writer = newChunkWriter(conn)
|
|
|
|
case SecurityAES128GCM:
|
|
|
|
block, _ := aes.NewCipher(reqBodyKey[:])
|
|
|
|
aead, _ := cipher.NewGCM(block)
|
|
|
|
writer = newAEADWriter(conn, aead, reqBodyIV[:])
|
|
|
|
|
|
|
|
block, _ = aes.NewCipher(respBodyKey[:])
|
|
|
|
aead, _ = cipher.NewGCM(block)
|
|
|
|
reader = newAEADReader(conn, aead, respBodyIV[:])
|
|
|
|
case SecurityCHACHA20POLY1305:
|
|
|
|
key := make([]byte, 32)
|
|
|
|
t := md5.Sum(reqBodyKey[:])
|
|
|
|
copy(key, t[:])
|
|
|
|
t = md5.Sum(key[:16])
|
|
|
|
copy(key[16:], t[:])
|
|
|
|
aead, _ := chacha20poly1305.New(key)
|
|
|
|
writer = newAEADWriter(conn, aead, reqBodyIV[:])
|
|
|
|
|
|
|
|
t = md5.Sum(respBodyKey[:])
|
|
|
|
copy(key, t[:])
|
|
|
|
t = md5.Sum(key[:16])
|
|
|
|
copy(key[16:], t[:])
|
|
|
|
aead, _ = chacha20poly1305.New(key)
|
|
|
|
reader = newAEADReader(conn, aead, respBodyIV[:])
|
|
|
|
}
|
|
|
|
|
2019-02-15 13:55:15 +00:00
|
|
|
c := &Conn{
|
2018-09-06 02:53:29 +00:00
|
|
|
Conn: conn,
|
|
|
|
id: id,
|
|
|
|
dst: dst,
|
|
|
|
reqBodyIV: reqBodyIV,
|
|
|
|
reqBodyKey: reqBodyKey,
|
|
|
|
respV: respV,
|
|
|
|
respBodyIV: respBodyIV[:],
|
|
|
|
respBodyKey: respBodyKey[:],
|
|
|
|
reader: reader,
|
|
|
|
writer: writer,
|
|
|
|
security: security,
|
|
|
|
}
|
2019-02-15 13:55:15 +00:00
|
|
|
if err := c.sendRequest(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return c, nil
|
2018-09-06 02:53:29 +00:00
|
|
|
}
|