mirror of
https://github.com/nadoo/glider.git
synced 2025-02-24 01:45:39 +08:00
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package vmess
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/md5"
|
||
"encoding/binary"
|
||
"encoding/hex"
|
||
"errors"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
// User of vmess client
|
||
type User struct {
|
||
UUID [16]byte
|
||
CmdKey [16]byte
|
||
}
|
||
|
||
// NewUser .
|
||
func NewUser(uuid [16]byte) *User {
|
||
u := &User{UUID: uuid}
|
||
copy(u.CmdKey[:], GetKey(uuid))
|
||
return u
|
||
}
|
||
|
||
func nextID(oldID [16]byte) (newID [16]byte) {
|
||
md5hash := md5.New()
|
||
md5hash.Write(oldID[:])
|
||
md5hash.Write([]byte("16167dc8-16b6-4e6d-b8bb-65dd68113a81"))
|
||
for {
|
||
md5hash.Sum(newID[:0])
|
||
if !bytes.Equal(oldID[:], newID[:]) {
|
||
return
|
||
}
|
||
md5hash.Write([]byte("533eff8a-4113-4b10-b5ce-0f5d76b98cd2"))
|
||
}
|
||
}
|
||
|
||
// GenAlterIDUsers generates users according to primary user's id and alterID
|
||
func (u *User) GenAlterIDUsers(alterID int) []*User {
|
||
users := make([]*User, alterID)
|
||
preID := u.UUID
|
||
for i := 0; i < alterID; i++ {
|
||
newID := nextID(preID)
|
||
// NOTE: alterID user is a user which have a different uuid but a same cmdkey with the primary user.
|
||
users[i] = &User{UUID: newID, CmdKey: u.CmdKey}
|
||
preID = newID
|
||
}
|
||
|
||
return users
|
||
}
|
||
|
||
// StrToUUID converts string to uuid.
|
||
// s fomat: "6ba7b810-9dad-11d1-80b4-00c04fd430c8"
|
||
func StrToUUID(s string) (uuid [16]byte, err error) {
|
||
b := []byte(strings.Replace(s, "-", "", -1))
|
||
if len(b) != 32 {
|
||
return uuid, errors.New("invalid UUID: " + s)
|
||
}
|
||
_, err = hex.Decode(uuid[:], b)
|
||
return
|
||
}
|
||
|
||
// GetKey returns the key of AES-128-CFB encrypter
|
||
// Key：MD5(UUID + []byte('c48619fe-8f02-49e0-b9e9-edf763e17e21'))
|
||
func GetKey(uuid [16]byte) []byte {
|
||
md5hash := md5.New()
|
||
md5hash.Write(uuid[:])
|
||
md5hash.Write([]byte("c48619fe-8f02-49e0-b9e9-edf763e17e21"))
|
||
return md5hash.Sum(nil)
|
||
}
|
||
|
||
// TimestampHash returns the iv of AES-128-CFB encrypter
|
||
// IV：MD5(X + X + X + X)，X = []byte(timestamp.now) (8 bytes, Big Endian)
|
||
func TimestampHash(t time.Time) []byte {
|
||
md5hash := md5.New()
|
||
|
||
ts := make([]byte, 8)
|
||
binary.BigEndian.PutUint64(ts, uint64(t.UTC().Unix()))
|
||
md5hash.Write(ts)
|
||
md5hash.Write(ts)
|
||
md5hash.Write(ts)
|
||
md5hash.Write(ts)
|
||
return md5hash.Sum(nil)
|
||
}
|