mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
984 B
58 lines
984 B
package vmess
|
|
|
|
import (
|
|
"github.com/v2ray/v2ray-core/common/dice"
|
|
)
|
|
|
|
type UserLevel byte
|
|
|
|
const (
|
|
UserLevelAdmin = UserLevel(255)
|
|
UserLevelUntrusted = UserLevel(0)
|
|
)
|
|
|
|
type User struct {
|
|
ID *ID
|
|
AlterIDs []*ID
|
|
Level UserLevel
|
|
}
|
|
|
|
func NewUser(id *ID, level UserLevel, alterIdCount uint16) *User {
|
|
u := &User{
|
|
ID: id,
|
|
Level: level,
|
|
}
|
|
if alterIdCount > 0 {
|
|
u.AlterIDs = make([]*ID, alterIdCount)
|
|
prevId := id.UUID()
|
|
for idx, _ := range u.AlterIDs {
|
|
newid := prevId.Next()
|
|
// TODO: check duplicate
|
|
u.AlterIDs[idx] = NewID(newid)
|
|
prevId = newid
|
|
}
|
|
}
|
|
return u
|
|
}
|
|
|
|
func (this *User) AnyValidID() *ID {
|
|
if len(this.AlterIDs) == 0 {
|
|
return this.ID
|
|
}
|
|
return this.AlterIDs[dice.Roll(len(this.AlterIDs))]
|
|
}
|
|
|
|
type UserSettings struct {
|
|
PayloadReadTimeout int
|
|
}
|
|
|
|
func GetUserSettings(level UserLevel) UserSettings {
|
|
settings := UserSettings{
|
|
PayloadReadTimeout: 120,
|
|
}
|
|
if level > 0 {
|
|
settings.PayloadReadTimeout = 0
|
|
}
|
|
return settings
|
|
}
|