1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-18 13:35:23 +00:00

Adjust VLESS & Trojan Validator

This commit is contained in:
RPRX 2020-10-22 04:53:27 +00:00 committed by GitHub
parent 24f688cc20
commit 3e87b187c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions

View File

@ -1,3 +1,5 @@
// +build !confonly
package trojan
import (
@ -7,14 +9,14 @@ import (
"v2ray.com/core/common/protocol"
)
// Validator stores valid trojan users
// Validator stores valid trojan users.
type Validator struct {
// Considering email's usage here, map + sync.Mutex/RWMutex may have better performance.
email sync.Map
users sync.Map
}
// Add a trojan user
// Add a trojan user, Email must be empty or unique.
func (v *Validator) Add(u *protocol.MemoryUser) error {
if u.Email != "" {
_, loaded := v.email.LoadOrStore(strings.ToLower(u.Email), u)
@ -26,7 +28,7 @@ func (v *Validator) Add(u *protocol.MemoryUser) error {
return nil
}
// Del a trojan user
// Del a trojan user with a non-empty Email.
func (v *Validator) Del(e string) error {
if e == "" {
return newError("Email must not be empty.")
@ -41,7 +43,7 @@ func (v *Validator) Del(e string) error {
return nil
}
// Get user with hashed key, nil if user doesn't exist.
// Get a trojan user with hashed key, nil if user doesn't exist.
func (v *Validator) Get(hash string) *protocol.MemoryUser {
u, _ := v.users.Load(hash)
if u != nil {

View File

@ -10,12 +10,14 @@ import (
"v2ray.com/core/common/uuid"
)
// Validator stores valid VLESS users.
type Validator struct {
// Considering email's usage here, map + sync.Mutex/RWMutex may have better performance.
email sync.Map
users sync.Map
}
// Add a VLESS user, Email must be empty or unique.
func (v *Validator) Add(u *protocol.MemoryUser) error {
if u.Email != "" {
_, loaded := v.email.LoadOrStore(strings.ToLower(u.Email), u)
@ -27,6 +29,7 @@ func (v *Validator) Add(u *protocol.MemoryUser) error {
return nil
}
// Del a VLESS user with a non-empty Email.
func (v *Validator) Del(e string) error {
if e == "" {
return newError("Email must not be empty.")
@ -41,6 +44,7 @@ func (v *Validator) Del(e string) error {
return nil
}
// Get a VLESS user with UUID, nil if user doesn't exist.
func (v *Validator) Get(id uuid.UUID) *protocol.MemoryUser {
u, _ := v.users.Load(id)
if u != nil {