mirror of https://github.com/XTLS/Xray-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.
49 lines
1.3 KiB
49 lines
1.3 KiB
package vmess |
|
|
|
import ( |
|
"github.com/xtls/xray-core/common/dice" |
|
"github.com/xtls/xray-core/common/protocol" |
|
"github.com/xtls/xray-core/common/uuid" |
|
) |
|
|
|
// MemoryAccount is an in-memory form of VMess account. |
|
type MemoryAccount struct { |
|
// ID is the main ID of the account. |
|
ID *protocol.ID |
|
// AlterIDs are the alternative IDs of the account. |
|
AlterIDs []*protocol.ID |
|
// Security type of the account. Used for client connections. |
|
Security protocol.SecurityType |
|
} |
|
|
|
// AnyValidID returns an ID that is either the main ID or one of the alternative IDs if any. |
|
func (a *MemoryAccount) AnyValidID() *protocol.ID { |
|
if len(a.AlterIDs) == 0 { |
|
return a.ID |
|
} |
|
return a.AlterIDs[dice.Roll(len(a.AlterIDs))] |
|
} |
|
|
|
// Equals implements protocol.Account. |
|
func (a *MemoryAccount) Equals(account protocol.Account) bool { |
|
vmessAccount, ok := account.(*MemoryAccount) |
|
if !ok { |
|
return false |
|
} |
|
// TODO: handle AlterIds difference |
|
return a.ID.Equals(vmessAccount.ID) |
|
} |
|
|
|
// AsAccount implements protocol.Account. |
|
func (a *Account) AsAccount() (protocol.Account, error) { |
|
id, err := uuid.ParseString(a.Id) |
|
if err != nil { |
|
return nil, newError("failed to parse ID").Base(err).AtError() |
|
} |
|
protoID := protocol.NewID(id) |
|
return &MemoryAccount{ |
|
ID: protoID, |
|
AlterIDs: protocol.NewAlterIDs(protoID, uint16(a.AlterId)), |
|
Security: a.SecuritySettings.GetSecurityType(), |
|
}, nil |
|
}
|
|
|