mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-16 09:26:21 -05:00
139 lines
3.4 KiB
Go
139 lines
3.4 KiB
Go
// Package vmess contains the implementation of VMess protocol and transportation.
|
|
//
|
|
// VMess contains both inbound and outbound connections. VMess inbound is usually used on servers
|
|
// together with 'freedom' to talk to final destination, while VMess outbound is usually used on
|
|
// clients with 'socks' for proxying.
|
|
package vmess
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
"v2ray.com/core/common/protocol"
|
|
)
|
|
|
|
const (
|
|
updateIntervalSec = 10
|
|
cacheDurationSec = 120
|
|
)
|
|
|
|
type idEntry struct {
|
|
id *protocol.ID
|
|
userIdx int
|
|
lastSec protocol.Timestamp
|
|
lastSecRemoval protocol.Timestamp
|
|
}
|
|
|
|
type TimedUserValidator struct {
|
|
sync.RWMutex
|
|
ctx context.Context
|
|
validUsers []*protocol.User
|
|
userHash map[[16]byte]*indexTimePair
|
|
ids []*idEntry
|
|
hasher protocol.IDHash
|
|
}
|
|
|
|
type indexTimePair struct {
|
|
index int
|
|
timeSec protocol.Timestamp
|
|
}
|
|
|
|
func NewTimedUserValidator(ctx context.Context, hasher protocol.IDHash) protocol.UserValidator {
|
|
tus := &TimedUserValidator{
|
|
ctx: ctx,
|
|
validUsers: make([]*protocol.User, 0, 16),
|
|
userHash: make(map[[16]byte]*indexTimePair, 512),
|
|
ids: make([]*idEntry, 0, 512),
|
|
hasher: hasher,
|
|
}
|
|
go tus.updateUserHash(updateIntervalSec * time.Second)
|
|
return tus
|
|
}
|
|
|
|
func (v *TimedUserValidator) generateNewHashes(nowSec protocol.Timestamp, idx int, entry *idEntry) {
|
|
var hashValue [16]byte
|
|
var hashValueRemoval [16]byte
|
|
idHash := v.hasher(entry.id.Bytes())
|
|
for entry.lastSec <= nowSec {
|
|
idHash.Write(entry.lastSec.Bytes(nil))
|
|
idHash.Sum(hashValue[:0])
|
|
idHash.Reset()
|
|
|
|
idHash.Write(entry.lastSecRemoval.Bytes(nil))
|
|
idHash.Sum(hashValueRemoval[:0])
|
|
idHash.Reset()
|
|
|
|
v.userHash[hashValue] = &indexTimePair{idx, entry.lastSec}
|
|
delete(v.userHash, hashValueRemoval)
|
|
|
|
entry.lastSec++
|
|
entry.lastSecRemoval++
|
|
}
|
|
}
|
|
|
|
func (v *TimedUserValidator) updateUserHash(interval time.Duration) {
|
|
for {
|
|
select {
|
|
case now := <-time.After(interval):
|
|
nowSec := protocol.Timestamp(now.Unix() + cacheDurationSec)
|
|
v.Lock()
|
|
for _, entry := range v.ids {
|
|
v.generateNewHashes(nowSec, entry.userIdx, entry)
|
|
}
|
|
v.Unlock()
|
|
case <-v.ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (v *TimedUserValidator) Add(user *protocol.User) error {
|
|
v.Lock()
|
|
defer v.Unlock()
|
|
|
|
idx := len(v.validUsers)
|
|
v.validUsers = append(v.validUsers, user)
|
|
rawAccount, err := user.GetTypedAccount()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
account := rawAccount.(*InternalAccount)
|
|
|
|
nowSec := time.Now().Unix()
|
|
|
|
entry := &idEntry{
|
|
id: account.ID,
|
|
userIdx: idx,
|
|
lastSec: protocol.Timestamp(nowSec - cacheDurationSec),
|
|
lastSecRemoval: protocol.Timestamp(nowSec - cacheDurationSec*3),
|
|
}
|
|
v.generateNewHashes(protocol.Timestamp(nowSec+cacheDurationSec), idx, entry)
|
|
v.ids = append(v.ids, entry)
|
|
for _, alterid := range account.AlterIDs {
|
|
entry := &idEntry{
|
|
id: alterid,
|
|
userIdx: idx,
|
|
lastSec: protocol.Timestamp(nowSec - cacheDurationSec),
|
|
lastSecRemoval: protocol.Timestamp(nowSec - cacheDurationSec*3),
|
|
}
|
|
v.generateNewHashes(protocol.Timestamp(nowSec+cacheDurationSec), idx, entry)
|
|
v.ids = append(v.ids, entry)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *TimedUserValidator) Get(userHash []byte) (*protocol.User, protocol.Timestamp, bool) {
|
|
defer v.RUnlock()
|
|
v.RLock()
|
|
|
|
var fixedSizeHash [16]byte
|
|
copy(fixedSizeHash[:], userHash)
|
|
pair, found := v.userHash[fixedSizeHash]
|
|
if found {
|
|
return v.validUsers[pair.index], pair.timeSec, true
|
|
}
|
|
return nil, 0, false
|
|
}
|