mststs

import "github.com/TheManticoreProject/Manticore/windows/protocols/ms-tsts"

const TSVIP_MAX_ADAPTER_ADDRESS_LENGTH = 16

type BOUNDED_ULONG ndr.DWORD

type ENUM_HANDLE [20]byte

type EXECENVDATA struct {
    Level ndr.DWORD
    Data  ExecEnvData
}

type EXECENVDATAEX struct {
    Level ndr.DWORD
    Data  ExecEnvDataEx
}

type EXECENVDATAEX_LEVEL1 struct {
    ExecEnvId      int32
    State          int32
    AbsSessionId   int32
    PszSessionName *ndr.WSTR `ndr:"unique"`
    PszHostName    *ndr.WSTR `ndr:"unique"`
    PszUserName    *ndr.WSTR `ndr:"unique"`
    PszDomainName  *ndr.WSTR `ndr:"unique"`
    PszFarmName    *ndr.WSTR `ndr:"unique"`
}

type EXECENVDATA_LEVEL1 struct {
    ExecEnvId   int32
    State       int32
    SessionName [33]uint16
}

type EXECENVDATA_LEVEL2 struct {
    ExecEnvId    int32
    State        int32
    SessionName  [33]uint16
    AbsSessionId int32
    HostName     [33]uint16
    UserName     [33]uint16
    DomainName   [33]uint16
    FarmName     [33]uint16
}

type ExecEnvData struct {
    Tag                ndr.DWORD          `ndr:"switch"`
    ExecEnvEnum_Level1 EXECENVDATA_LEVEL1 `ndr:"case=1"`
    ExecEnvEnum_Level2 EXECENVDATA_LEVEL2 `ndr:"case=2"`
}

type ExecEnvDataEx struct {
    Tag                ndr.DWORD            `ndr:"switch"`
    ExecEnvEnum_Level1 EXECENVDATAEX_LEVEL1 `ndr:"case=1"`
}

type HLISTENER [20]byte

type LISTENERENUM struct {
    Level ndr.DWORD
    Data  ListenerInfo
}

type LISTENERENUM_LEVEL1 struct {
    Id         int32
    BListening ndr.BOOL
    Name       [33]uint16
}

type LISTENER_NAME uint16

type LSMSESSIONINFORMATION struct {
    PszUserName     *ndr.WSTR `ndr:"unique"`
    PszDomain       *ndr.WSTR `ndr:"unique"`
    PszTerminalName *ndr.WSTR `ndr:"unique"`
    SessionState    int32
    DesktopLocked   ndr.BOOL
    ConnectTime     int64
    DisconnectTime  int64
    LogonTime       int64
}

type LSMSESSIONINFORMATION_EX struct {
    Level ndr.DWORD
    Data  LSM_SESSIONINFO_EX
}

type LSM_SESSIONINFO_EX struct {
    Tag                    ndr.DWORD                 `ndr:"switch"`
    LSM_SessionInfo_Level1 LSM_SESSIONINFO_EX_LEVEL1 `ndr:"case=1"`
}

type LSM_SESSIONINFO_EX_LEVEL1 struct {
    SessionState     int32
    SessionFlags     int32
    SessionName      [33]uint16
    DomainName       [18]uint16
    UserName         [21]uint16
    ConnectTime      int64
    DisconnectTime   int64
    LogonTime        int64
    LastInputTime    int64
    ProtocolDataSize ndr.DWORD
    ProtocolData     []uint8 `ndr:"unique,size_is=ProtocolDataSize"`
}

type ListenerInfo struct {
    Tag                 ndr.DWORD           `ndr:"switch"`
    ListenerEnum_Level1 LISTENERENUM_LEVEL1 `ndr:"case=1"`
}

type NOTIFY_HANDLE [20]byte

type NT6_TS_UNICODE_STRING struct {
    Length        uint16
    MaximumLength uint16
    Buffer        []uint16 `ndr:"unique,size_is=MaximumLength/2,length_is=Length/2"`
}

type PROTOCOLSTATUS_INFO_TYPE uint16

const (
    PROTOCOLSTATUS_INFO_BASIC    PROTOCOLSTATUS_INFO_TYPE = 0
    PROTOCOLSTATUS_INFO_EXTENDED PROTOCOLSTATUS_INFO_TYPE = 1
)

type QUERY_SESSION_DATA_TYPE uint16

const (
    QUERY_SESSION_DATA_MODULE             QUERY_SESSION_DATA_TYPE = 0
    QUERY_SESSION_DATA_WDCONFIG           QUERY_SESSION_DATA_TYPE = 1
    QUERY_SESSION_DATA_VIRTUALDATA        QUERY_SESSION_DATA_TYPE = 2
    QUERY_SESSION_DATA_LICENSE            QUERY_SESSION_DATA_TYPE = 3
    QUERY_SESSION_DATA_DEVICEID           QUERY_SESSION_DATA_TYPE = 4
    QUERY_SESSION_DATA_LICENSE_VALIDATION QUERY_SESSION_DATA_TYPE = 5
)

type RCM_REMOTEADDRESS struct {
    SinFamily uint16                 `ndr:"switch"`
    Ipv4      RCM_REMOTEADDRESS_IPV4 `ndr:"case=2"`
    Ipv6      RCM_REMOTEADDRESS_IPV6 `ndr:"case=23"`
}

type RCM_REMOTEADDRESS_IPV4 struct {
    SinPort uint16
    InAddr  ndr.DWORD
    SinZero [8]uint8
}

type RCM_REMOTEADDRESS_IPV6 struct {
    Sin6Port     uint16
    Sin6Flowinfo ndr.DWORD
    Sin6Addr     [8]uint16
    Sin6ScopeId  ndr.DWORD
}

type SERVER_HANDLE [20]byte

type SESSIONENUM struct {
    Level ndr.DWORD
    Data  SessionInfo
}

type SESSIONENUM_EX struct {
    Level ndr.DWORD
    Data  SessionInfo_Ex
}

type SESSIONENUM_LEVEL1 struct {
    SessionId int32
    State     int32
    Name      [33]uint16
}

type SESSIONENUM_LEVEL2 struct {
    SessionId    int32
    State        int32
    Name         [33]uint16
    Source       ndr.DWORD
    BFullDesktop ndr.BOOL
    SessionType  guid.GUID
}

type SESSIONENUM_LEVEL3 struct {
    SessionId     int32
    State         int32
    Name          [33]uint16
    Source        ndr.DWORD
    BFullDesktop  ndr.BOOL
    SessionType   guid.GUID
    ProtoDataSize ndr.DWORD
    PProtocolData []uint8 `ndr:"unique,size_is=ProtoDataSize"`
}

type SESSION_CHANGE struct {
    SessionId      int32
    NotificationId TNotificationId
}

type SESSION_FILTER uint16

const (
    // SF_SERVICES_SESSION_POPUP requests the sessions that can display a services message box.
    SF_SERVICES_SESSION_POPUP SESSION_FILTER = 0
)

type SESSION_HANDLE [20]byte

type SHADOW_CONTROL_REQUEST uint16

const (
    SHADOW_CONTROL_REQUEST_VIEW        SHADOW_CONTROL_REQUEST = 0
    SHADOW_CONTROL_REQUEST_TAKECONTROL SHADOW_CONTROL_REQUEST = 1
    SHADOW_CONTROL_REQUEST_Count       SHADOW_CONTROL_REQUEST = 2
)

type SHADOW_PERMISSION_REQUEST uint16

const (
    SHADOW_PERMISSION_REQUEST_SILENT            SHADOW_PERMISSION_REQUEST = 0
    SHADOW_PERMISSION_REQUEST_REQUESTPERMISSION SHADOW_PERMISSION_REQUEST = 1
    SHADOW_PERMISSION_REQUEST_Count             SHADOW_PERMISSION_REQUEST = 2
)

type SHADOW_REQUEST_RESPONSE uint16

const (
    SHADOW_REQUEST_RESPONSE_ALLOW                                SHADOW_REQUEST_RESPONSE = 0
    SHADOW_REQUEST_RESPONSE_DECLINE                              SHADOW_REQUEST_RESPONSE = 1
    SHADOW_REQUEST_RESPONSE_POLICY_PERMISSION_REQUIRED           SHADOW_REQUEST_RESPONSE = 2
    SHADOW_REQUEST_RESPONSE_POLICY_DISABLED                      SHADOW_REQUEST_RESPONSE = 3
    SHADOW_REQUEST_RESPONSE_POLICY_VIEW_ONLY                     SHADOW_REQUEST_RESPONSE = 4
    SHADOW_REQUEST_RESPONSE_POLICY_VIEW_ONLY_PERMISSION_REQUIRED SHADOW_REQUEST_RESPONSE = 5
    SHADOW_REQUEST_RESPONSE_SESSION_ALREADY_CONTROLLED           SHADOW_REQUEST_RESPONSE = 6
)

type SessionInfo struct {
    Tag                ndr.DWORD          `ndr:"switch"`
    SessionEnum_Level1 SESSIONENUM_LEVEL1 `ndr:"case=1"`
    SessionEnum_Level2 SESSIONENUM_LEVEL2 `ndr:"case=2"`
}

type SessionInfo_Ex struct {
    Tag                ndr.DWORD          `ndr:"switch"`
    SessionEnum_Level1 SESSIONENUM_LEVEL1 `ndr:"case=1"`
    SessionEnum_Level2 SESSIONENUM_LEVEL2 `ndr:"case=2"`
    SessionEnum_Level3 SESSIONENUM_LEVEL3 `ndr:"case=3"`
}

type TNotificationId = ndr.DWORD

const (
    WTS_NOTIFY_NONE               TNotificationId = 0x00000000
    WTS_NOTIFY_CREATE             TNotificationId = 0x00000001
    WTS_NOTIFY_CONNECT            TNotificationId = 0x00000002
    WTS_NOTIFY_DISCONNECT         TNotificationId = 0x00000004
    WTS_NOTIFY_LOGON              TNotificationId = 0x00000008
    WTS_NOTIFY_LOGOFF             TNotificationId = 0x00000010
    WTS_NOTIFY_SHADOW_START       TNotificationId = 0x00000020
    WTS_NOTIFY_SHADOW_STOP        TNotificationId = 0x00000040
    WTS_NOTIFY_TERMINATE          TNotificationId = 0x00000080
    WTS_NOTIFY_CONSOLE_CONNECT    TNotificationId = 0x00000100
    WTS_NOTIFY_CONSOLE_DISCONNECT TNotificationId = 0x00000200
    WTS_NOTIFY_LOCK               TNotificationId = 0x00000400
    WTS_NOTIFY_UNLOCK             TNotificationId = 0x00000800
    WTS_NOTIFY_ALL                TNotificationId = 0xffffffff
)

type TSVIPAddress struct {
    DwVersion             ndr.DWORD
    IPAddress             TSVIP_SOCKADDR
    PrefixOrSubnetMask    ndr.DWORD
    PhysicalAddressLength uint32
    PhysicalAddress       [TSVIP_MAX_ADAPTER_ADDRESS_LENGTH]uint8
    LeaseExpires          ndr.DWORD
    T1                    ndr.DWORD
    T2                    ndr.DWORD
}

type TSVIPSession struct {
    DwVersion ndr.DWORD
    SessionId ndr.DWORD
    SessionIP TSVIPAddress
}

type TSVIP_SOCKADDR struct {
    SinFamily uint16              `ndr:"switch"`
    Ipv4      TSVIP_SOCKADDR_IPV4 `ndr:"case=2"`
    Ipv6      TSVIP_SOCKADDR_IPV6 `ndr:"case=23"`
}

type TSVIP_SOCKADDR_IPV4 struct {
    SinPort uint16
    InAddr  ndr.DWORD
    SinZero [8]uint8
}

type TSVIP_SOCKADDR_IPV6 struct {
    Sin6Port     uint16
    Sin6Flowinfo ndr.DWORD
    Sin6Addr     [8]uint16
    Sin6ScopeId  ndr.DWORD
}

type TS_ALL_PROCESSES_INFO struct {
    PTsProcessInfo *TS_SYS_PROCESS_INFORMATION `ndr:"unique"`
    SizeOfSid      ndr.DWORD
    PSid           []byte `ndr:"unique,size_is=SizeOfSid"`
}

type TS_ALL_PROCESSES_INFO_NT6 struct {
    PTsProcessInfo *TS_SYS_PROCESS_INFORMATION_NT6 `ndr:"unique"`
    SizeOfSid      ndr.DWORD
    PSid           []byte `ndr:"unique,size_is=SizeOfSid"`
}

type TS_COUNTER struct {
    CounterHead TS_COUNTER_HEADER
    DwValue     ndr.DWORD
    StartTime   msdtyp.LARGE_INTEGER
}

type TS_COUNTER_HEADER struct {
    DwCounterID ndr.DWORD
    BResult     ndr.BOOLEAN
}

type TS_SYS_PROCESS_INFORMATION struct {
    NextEntryOffset              ndr.DWORD
    NumberOfThreads              ndr.DWORD
    SpareLi1                     msdtyp.LARGE_INTEGER
    SpareLi2                     msdtyp.LARGE_INTEGER
    SpareLi3                     msdtyp.LARGE_INTEGER
    CreateTime                   msdtyp.LARGE_INTEGER
    UserTime                     msdtyp.LARGE_INTEGER
    KernelTime                   msdtyp.LARGE_INTEGER
    ImageName                    TS_UNICODE_STRING
    BasePriority                 int32
    UniqueProcessId              ndr.DWORD
    InheritedFromUniqueProcessId ndr.DWORD
    HandleCount                  ndr.DWORD
    SessionId                    ndr.DWORD
    SpareUl3                     ndr.DWORD
    PeakVirtualSize              uint64
    VirtualSize                  uint64
    PageFaultCount               ndr.DWORD
    PeakWorkingSetSize           ndr.DWORD
    WorkingSetSize               ndr.DWORD
    QuotaPeakPagedPoolUsage      uint64
    QuotaPagedPoolUsage          uint64
    QuotaPeakNonPagedPoolUsage   uint64
    QuotaNonPagedPoolUsage       uint64
    PagefileUsage                uint64
    PeakPagefileUsage            uint64
    PrivatePageCount             uint64
}

type TS_SYS_PROCESS_INFORMATION_NT6 struct {
    NextEntryOffset              ndr.DWORD
    NumberOfThreads              ndr.DWORD
    SpareLi1                     msdtyp.LARGE_INTEGER
    SpareLi2                     msdtyp.LARGE_INTEGER
    SpareLi3                     msdtyp.LARGE_INTEGER
    CreateTime                   msdtyp.LARGE_INTEGER
    UserTime                     msdtyp.LARGE_INTEGER
    KernelTime                   msdtyp.LARGE_INTEGER
    ImageName                    NT6_TS_UNICODE_STRING
    BasePriority                 int32
    UniqueProcessId              ndr.DWORD
    InheritedFromUniqueProcessId ndr.DWORD
    HandleCount                  ndr.DWORD
    SessionId                    ndr.DWORD
    SpareUl3                     ndr.DWORD
    PeakVirtualSize              uint64
    VirtualSize                  uint64
    PageFaultCount               ndr.DWORD
    PeakWorkingSetSize           ndr.DWORD
    WorkingSetSize               ndr.DWORD
    QuotaPeakPagedPoolUsage      uint64
    QuotaPagedPoolUsage          uint64
    QuotaPeakNonPagedPoolUsage   uint64
    QuotaNonPagedPoolUsage       uint64
    PagefileUsage                uint64
    PeakPagefileUsage            uint64
    PrivatePageCount             uint64
}

type TS_UNICODE_STRING struct {
    Length        uint16
    MaximumLength uint16
    Buffer        []uint16 `ndr:"unique,size_is=MaximumLength,length_is=Length"`
}