mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-06 18:28:00 -05:00
70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package protocol
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/md5"
|
|
"hash"
|
|
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/uuid"
|
|
)
|
|
|
|
const (
|
|
IDBytesLen = 16
|
|
)
|
|
|
|
type IDHash func(key []byte) hash.Hash
|
|
|
|
func DefaultIDHash(key []byte) hash.Hash {
|
|
return hmac.New(md5.New, key)
|
|
}
|
|
|
|
// The ID of en entity, in the form of an UUID.
|
|
type ID struct {
|
|
uuid uuid.UUID
|
|
cmdKey [IDBytesLen]byte
|
|
}
|
|
|
|
// Equals returns true if this ID equals to the other one.
|
|
func (id *ID) Equals(another *ID) bool {
|
|
return id.uuid.Equals(&(another.uuid))
|
|
}
|
|
|
|
func (id *ID) Bytes() []byte {
|
|
return id.uuid.Bytes()
|
|
}
|
|
|
|
func (id *ID) String() string {
|
|
return id.uuid.String()
|
|
}
|
|
|
|
func (id *ID) UUID() uuid.UUID {
|
|
return id.uuid
|
|
}
|
|
|
|
func (id ID) CmdKey() []byte {
|
|
return id.cmdKey[:]
|
|
}
|
|
|
|
// NewID returns an ID with given UUID.
|
|
func NewID(uuid uuid.UUID) *ID {
|
|
id := &ID{uuid: uuid}
|
|
md5hash := md5.New()
|
|
common.Must2(md5hash.Write(uuid.Bytes()))
|
|
common.Must2(md5hash.Write([]byte("c48619fe-8f02-49e0-b9e9-edf763e17e21")))
|
|
md5hash.Sum(id.cmdKey[:0])
|
|
return id
|
|
}
|
|
|
|
func NewAlterIDs(primary *ID, alterIDCount uint16) []*ID {
|
|
alterIDs := make([]*ID, alterIDCount)
|
|
prevID := primary.UUID()
|
|
for idx := range alterIDs {
|
|
newid := prevID.Next()
|
|
// TODO: check duplicates
|
|
alterIDs[idx] = NewID(newid)
|
|
prevID = newid
|
|
}
|
|
return alterIDs
|
|
}
|