1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-10-31 00:08:08 -04:00
v2fly/common/protocol/server_spec.go

117 lines
2.0 KiB
Go
Raw Normal View History

2016-07-24 17:22:46 -04:00
package protocol
import (
"sync"
"time"
2016-08-20 14:55:45 -04:00
"v2ray.com/core/common/dice"
2017-01-13 18:27:45 -05:00
"v2ray.com/core/common/net"
2016-07-24 17:22:46 -04:00
)
2016-07-25 10:48:09 -04:00
type ValidationStrategy interface {
IsValid() bool
Invalidate()
}
type AlwaysValidStrategy struct{}
func AlwaysValid() ValidationStrategy {
return AlwaysValidStrategy{}
}
2016-11-27 15:39:09 -05:00
func (v AlwaysValidStrategy) IsValid() bool {
2016-07-25 10:48:09 -04:00
return true
}
2016-11-27 15:39:09 -05:00
func (v AlwaysValidStrategy) Invalidate() {}
2016-07-25 10:48:09 -04:00
type TimeoutValidStrategy struct {
until time.Time
}
func BeforeTime(t time.Time) ValidationStrategy {
2016-07-25 15:31:16 -04:00
return &TimeoutValidStrategy{
2016-07-25 10:48:09 -04:00
until: t,
}
}
2016-11-27 15:39:09 -05:00
func (v *TimeoutValidStrategy) IsValid() bool {
return v.until.After(time.Now())
2016-07-25 10:48:09 -04:00
}
2016-11-27 15:39:09 -05:00
func (v *TimeoutValidStrategy) Invalidate() {
v.until = time.Time{}
2016-07-25 10:48:09 -04:00
}
2016-07-24 17:22:46 -04:00
type ServerSpec struct {
sync.RWMutex
2017-01-13 18:27:45 -05:00
dest net.Destination
2016-11-02 11:33:04 -04:00
users []*User
valid ValidationStrategy
2016-07-24 17:22:46 -04:00
}
2017-01-13 18:27:45 -05:00
func NewServerSpec(dest net.Destination, valid ValidationStrategy, users ...*User) *ServerSpec {
2016-07-24 17:22:46 -04:00
return &ServerSpec{
2016-11-02 11:33:04 -04:00
dest: dest,
users: users,
valid: valid,
2016-07-24 17:22:46 -04:00
}
}
2016-11-02 11:33:04 -04:00
func NewServerSpecFromPB(spec ServerEndpoint) *ServerSpec {
2017-01-13 18:27:45 -05:00
dest := net.TCPDestination(spec.Address.AsAddress(), net.Port(spec.Port))
2016-11-02 11:33:04 -04:00
return NewServerSpec(dest, AlwaysValid(), spec.User...)
2016-09-17 18:41:21 -04:00
}
2017-01-13 18:27:45 -05:00
func (v *ServerSpec) Destination() net.Destination {
2016-11-27 15:39:09 -05:00
return v.dest
2016-07-25 10:48:09 -04:00
}
2016-11-27 15:39:09 -05:00
func (v *ServerSpec) HasUser(user *User) bool {
v.RLock()
defer v.RUnlock()
2016-07-24 17:22:46 -04:00
2016-10-16 08:22:21 -04:00
accountA, err := user.GetTypedAccount()
2016-09-17 18:41:21 -04:00
if err != nil {
return false
}
2016-11-27 15:39:09 -05:00
for _, u := range v.users {
2016-10-16 08:22:21 -04:00
accountB, err := u.GetTypedAccount()
2016-09-17 18:41:21 -04:00
if err == nil && accountA.Equals(accountB) {
2016-07-24 17:22:46 -04:00
return true
}
}
return false
}
2016-11-27 15:39:09 -05:00
func (v *ServerSpec) AddUser(user *User) {
if v.HasUser(user) {
2016-07-24 17:22:46 -04:00
return
}
2016-11-27 15:39:09 -05:00
v.Lock()
defer v.Unlock()
2016-07-24 17:22:46 -04:00
2016-11-27 15:39:09 -05:00
v.users = append(v.users, user)
2016-07-24 17:22:46 -04:00
}
2016-11-27 15:39:09 -05:00
func (v *ServerSpec) PickUser() *User {
userCount := len(v.users)
2017-01-07 19:06:35 -05:00
switch userCount {
case 0:
return nil
case 1:
return v.users[0]
default:
return v.users[dice.Roll(userCount)]
}
2016-07-24 17:22:46 -04:00
}
2016-11-27 15:39:09 -05:00
func (v *ServerSpec) IsValid() bool {
return v.valid.IsValid()
2016-07-24 17:22:46 -04:00
}
2016-11-27 15:39:09 -05:00
func (v *ServerSpec) Invalidate() {
v.valid.Invalidate()
2016-07-24 17:22:46 -04:00
}