mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-08 20:19:10 -04:00
41 lines
846 B
Go
41 lines
846 B
Go
package core
|
|
|
|
import (
|
|
"encoding/base64"
|
|
)
|
|
|
|
type VUserSet struct {
|
|
validUserIds []VID
|
|
userIdsAskHash map[string]int
|
|
}
|
|
|
|
func NewVUserSet() *VUserSet {
|
|
vuSet := new(VUserSet)
|
|
vuSet.validUserIds = make([]VID, 0, 16)
|
|
vuSet.userIdsAskHash = make(map[string]int)
|
|
return vuSet
|
|
}
|
|
|
|
func hashBytesToString(hash []byte) string {
|
|
return base64.StdEncoding.EncodeToString(hash)
|
|
}
|
|
|
|
func (us *VUserSet) AddUser(user VUser) error {
|
|
id := user.Id
|
|
us.validUserIds = append(us.validUserIds, id)
|
|
|
|
idBase64 := hashBytesToString(id.Hash([]byte("ASK")))
|
|
us.userIdsAskHash[idBase64] = len(us.validUserIds) - 1
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us VUserSet) IsValidUserId(askHash []byte) (*VID, bool) {
|
|
askBase64 := hashBytesToString(askHash)
|
|
idIndex, found := us.userIdsAskHash[askBase64]
|
|
if found {
|
|
return &us.validUserIds[idIndex], true
|
|
}
|
|
return nil, false
|
|
}
|