Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
//go:build !confonly
// +build !confonly
 
package vmess
 
import (
    "strings"
 
    "github.com/v2fly/v2ray-core/v5/common/dice"
    "github.com/v2fly/v2ray-core/v5/common/protocol"
    "github.com/v2fly/v2ray-core/v5/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
 
    AuthenticatedLengthExperiment bool
    NoTerminationSignal           bool
}
 
// 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)
    var AuthenticatedLength, NoTerminationSignal bool
    if strings.Contains(a.TestsEnabled, "AuthenticatedLength") {
        AuthenticatedLength = true
    }
    if strings.Contains(a.TestsEnabled, "NoTerminationSignal") {
        NoTerminationSignal = true
    }
    return &MemoryAccount{
        ID:                            protoID,
        AlterIDs:                      protocol.NewAlterIDs(protoID, uint16(a.AlterId)),
        Security:                      a.SecuritySettings.GetSecurityType(),
        AuthenticatedLengthExperiment: AuthenticatedLength,
        NoTerminationSignal:           NoTerminationSignal,
    }, nil
}