mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-21 09:36:34 -05:00
124 lines
2.1 KiB
Go
124 lines
2.1 KiB
Go
package core
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
mrand "math/rand"
|
|
"time"
|
|
|
|
"github.com/v2ray/v2ray-core/log"
|
|
)
|
|
|
|
const (
|
|
IDBytesLen = 16
|
|
)
|
|
|
|
// The ID of en entity, in the form of an UUID.
|
|
type ID struct {
|
|
String string
|
|
Bytes []byte
|
|
cmdKey []byte
|
|
}
|
|
|
|
func NewID(id string) (ID, error) {
|
|
idBytes, err := UUIDToID(id)
|
|
if err != nil {
|
|
return ID{}, log.Error("Failed to parse id %s", id)
|
|
}
|
|
|
|
md5hash := md5.New()
|
|
md5hash.Write(idBytes)
|
|
md5hash.Write([]byte("c48619fe-8f02-49e0-b9e9-edf763e17e21"))
|
|
cmdKey := md5.Sum(nil)
|
|
|
|
return ID{
|
|
String: id,
|
|
Bytes: idBytes,
|
|
cmdKey: cmdKey[:],
|
|
}, nil
|
|
}
|
|
|
|
func (v ID) TimeRangeHash(rangeSec int) ([]byte, int64) {
|
|
nowSec := time.Now().UTC().Unix()
|
|
delta := mrand.Intn(rangeSec*2) - rangeSec
|
|
|
|
targetSec := nowSec + int64(delta)
|
|
return v.TimeHash(targetSec), targetSec
|
|
}
|
|
|
|
func (v ID) TimeHash(timeSec int64) []byte {
|
|
buffer := []byte{
|
|
byte(timeSec >> 56),
|
|
byte(timeSec >> 48),
|
|
byte(timeSec >> 40),
|
|
byte(timeSec >> 32),
|
|
byte(timeSec >> 24),
|
|
byte(timeSec >> 16),
|
|
byte(timeSec >> 8),
|
|
byte(timeSec),
|
|
}
|
|
return v.Hash(buffer)
|
|
}
|
|
|
|
func (v ID) Hash(data []byte) []byte {
|
|
hasher := hmac.New(md5.New, v.Bytes)
|
|
hasher.Write(data)
|
|
return hasher.Sum(nil)
|
|
}
|
|
|
|
func (v ID) CmdKey() []byte {
|
|
return v.cmdKey
|
|
}
|
|
|
|
func TimestampHash(timeSec int64) []byte {
|
|
md5hash := md5.New()
|
|
buffer := []byte{
|
|
byte(timeSec >> 56),
|
|
byte(timeSec >> 48),
|
|
byte(timeSec >> 40),
|
|
byte(timeSec >> 32),
|
|
byte(timeSec >> 24),
|
|
byte(timeSec >> 16),
|
|
byte(timeSec >> 8),
|
|
byte(timeSec),
|
|
}
|
|
md5hash.Write(buffer)
|
|
md5hash.Write(buffer)
|
|
md5hash.Write(buffer)
|
|
md5hash.Write(buffer)
|
|
return md5hash.Sum(nil)
|
|
}
|
|
|
|
var byteGroups = []int{8, 4, 4, 4, 12}
|
|
|
|
// TODO: leverage a full functional UUID library
|
|
func UUIDToID(uuid string) (v []byte, err error) {
|
|
v = make([]byte, 16)
|
|
|
|
text := []byte(uuid)
|
|
if len(text) < 32 {
|
|
err = log.Error("uuid: invalid UUID string: %s", text)
|
|
return
|
|
}
|
|
|
|
b := v[:]
|
|
|
|
for _, byteGroup := range byteGroups {
|
|
if text[0] == '-' {
|
|
text = text[1:]
|
|
}
|
|
|
|
_, err = hex.Decode(b[:byteGroup/2], text[:byteGroup])
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
text = text[byteGroup:]
|
|
b = b[byteGroup/2:]
|
|
}
|
|
|
|
return
|
|
}
|