|
- // Copyright 2012 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
-
- package windows
-
- import (
- "syscall"
- "unsafe"
- )
-
- const (
- STANDARD_RIGHTS_REQUIRED = 0xf0000
- STANDARD_RIGHTS_READ = 0x20000
- STANDARD_RIGHTS_WRITE = 0x20000
- STANDARD_RIGHTS_EXECUTE = 0x20000
- STANDARD_RIGHTS_ALL = 0x1F0000
- )
-
- const (
- NameUnknown = 0
- NameFullyQualifiedDN = 1
- NameSamCompatible = 2
- NameDisplay = 3
- NameUniqueId = 6
- NameCanonical = 7
- NameUserPrincipal = 8
- NameCanonicalEx = 9
- NameServicePrincipal = 10
- NameDnsDomain = 12
- )
-
- // This function returns 1 byte BOOLEAN rather than the 4 byte BOOL.
- // http://blogs.msdn.com/b/drnick/archive/2007/12/19/windows-and-upn-format-credentials.aspx
- //sys TranslateName(accName *uint16, accNameFormat uint32, desiredNameFormat uint32, translatedName *uint16, nSize *uint32) (err error) [failretval&0xff==0] = secur32.TranslateNameW
- //sys GetUserNameEx(nameFormat uint32, nameBuffre *uint16, nSize *uint32) (err error) [failretval&0xff==0] = secur32.GetUserNameExW
-
- // TranslateAccountName converts a directory service
- // object name from one format to another.
- func TranslateAccountName(username string, from, to uint32, initSize int) (string, error) {
- u, e := UTF16PtrFromString(username)
- if e != nil {
- return "", e
- }
- n := uint32(50)
- for {
- b := make([]uint16, n)
- e = TranslateName(u, from, to, &b[0], &n)
- if e == nil {
- return UTF16ToString(b[:n]), nil
- }
- if e != ERROR_INSUFFICIENT_BUFFER {
- return "", e
- }
- if n <= uint32(len(b)) {
- return "", e
- }
- }
- }
-
- const (
- // do not reorder
- NetSetupUnknownStatus = iota
- NetSetupUnjoined
- NetSetupWorkgroupName
- NetSetupDomainName
- )
-
- type UserInfo10 struct {
- Name *uint16
- Comment *uint16
- UsrComment *uint16
- FullName *uint16
- }
-
- //sys NetUserGetInfo(serverName *uint16, userName *uint16, level uint32, buf **byte) (neterr error) = netapi32.NetUserGetInfo
- //sys NetGetJoinInformation(server *uint16, name **uint16, bufType *uint32) (neterr error) = netapi32.NetGetJoinInformation
- //sys NetApiBufferFree(buf *byte) (neterr error) = netapi32.NetApiBufferFree
-
- const (
- // do not reorder
- SidTypeUser = 1 + iota
- SidTypeGroup
- SidTypeDomain
- SidTypeAlias
- SidTypeWellKnownGroup
- SidTypeDeletedAccount
- SidTypeInvalid
- SidTypeUnknown
- SidTypeComputer
- SidTypeLabel
- )
-
- type SidIdentifierAuthority struct {
- Value [6]byte
- }
-
- var (
- SECURITY_NULL_SID_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 0}}
- SECURITY_WORLD_SID_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 1}}
- SECURITY_LOCAL_SID_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 2}}
- SECURITY_CREATOR_SID_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 3}}
- SECURITY_NON_UNIQUE_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 4}}
- SECURITY_NT_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 5}}
- SECURITY_MANDATORY_LABEL_AUTHORITY = SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 16}}
- )
-
- const (
- SECURITY_NULL_RID = 0
- SECURITY_WORLD_RID = 0
- SECURITY_LOCAL_RID = 0
- SECURITY_CREATOR_OWNER_RID = 0
- SECURITY_CREATOR_GROUP_RID = 1
- SECURITY_DIALUP_RID = 1
- SECURITY_NETWORK_RID = 2
- SECURITY_BATCH_RID = 3
- SECURITY_INTERACTIVE_RID = 4
- SECURITY_LOGON_IDS_RID = 5
- SECURITY_SERVICE_RID = 6
- SECURITY_LOCAL_SYSTEM_RID = 18
- SECURITY_BUILTIN_DOMAIN_RID = 32
- SECURITY_PRINCIPAL_SELF_RID = 10
- SECURITY_CREATOR_OWNER_SERVER_RID = 0x2
- SECURITY_CREATOR_GROUP_SERVER_RID = 0x3
- SECURITY_LOGON_IDS_RID_COUNT = 0x3
- SECURITY_ANONYMOUS_LOGON_RID = 0x7
- SECURITY_PROXY_RID = 0x8
- SECURITY_ENTERPRISE_CONTROLLERS_RID = 0x9
- SECURITY_SERVER_LOGON_RID = SECURITY_ENTERPRISE_CONTROLLERS_RID
- SECURITY_AUTHENTICATED_USER_RID = 0xb
- SECURITY_RESTRICTED_CODE_RID = 0xc
- SECURITY_NT_NON_UNIQUE_RID = 0x15
- )
-
- // Predefined domain-relative RIDs for local groups.
- // See https://msdn.microsoft.com/en-us/library/windows/desktop/aa379649(v=vs.85).aspx
- const (
- DOMAIN_ALIAS_RID_ADMINS = 0x220
- DOMAIN_ALIAS_RID_USERS = 0x221
- DOMAIN_ALIAS_RID_GUESTS = 0x222
- DOMAIN_ALIAS_RID_POWER_USERS = 0x223
- DOMAIN_ALIAS_RID_ACCOUNT_OPS = 0x224
- DOMAIN_ALIAS_RID_SYSTEM_OPS = 0x225
- DOMAIN_ALIAS_RID_PRINT_OPS = 0x226
- DOMAIN_ALIAS_RID_BACKUP_OPS = 0x227
- DOMAIN_ALIAS_RID_REPLICATOR = 0x228
- DOMAIN_ALIAS_RID_RAS_SERVERS = 0x229
- DOMAIN_ALIAS_RID_PREW2KCOMPACCESS = 0x22a
- DOMAIN_ALIAS_RID_REMOTE_DESKTOP_USERS = 0x22b
- DOMAIN_ALIAS_RID_NETWORK_CONFIGURATION_OPS = 0x22c
- DOMAIN_ALIAS_RID_INCOMING_FOREST_TRUST_BUILDERS = 0x22d
- DOMAIN_ALIAS_RID_MONITORING_USERS = 0x22e
- DOMAIN_ALIAS_RID_LOGGING_USERS = 0x22f
- DOMAIN_ALIAS_RID_AUTHORIZATIONACCESS = 0x230
- DOMAIN_ALIAS_RID_TS_LICENSE_SERVERS = 0x231
- DOMAIN_ALIAS_RID_DCOM_USERS = 0x232
- DOMAIN_ALIAS_RID_IUSERS = 0x238
- DOMAIN_ALIAS_RID_CRYPTO_OPERATORS = 0x239
- DOMAIN_ALIAS_RID_CACHEABLE_PRINCIPALS_GROUP = 0x23b
- DOMAIN_ALIAS_RID_NON_CACHEABLE_PRINCIPALS_GROUP = 0x23c
- DOMAIN_ALIAS_RID_EVENT_LOG_READERS_GROUP = 0x23d
- DOMAIN_ALIAS_RID_CERTSVC_DCOM_ACCESS_GROUP = 0x23e
- )
-
- //sys LookupAccountSid(systemName *uint16, sid *SID, name *uint16, nameLen *uint32, refdDomainName *uint16, refdDomainNameLen *uint32, use *uint32) (err error) = advapi32.LookupAccountSidW
- //sys LookupAccountName(systemName *uint16, accountName *uint16, sid *SID, sidLen *uint32, refdDomainName *uint16, refdDomainNameLen *uint32, use *uint32) (err error) = advapi32.LookupAccountNameW
- //sys ConvertSidToStringSid(sid *SID, stringSid **uint16) (err error) = advapi32.ConvertSidToStringSidW
- //sys ConvertStringSidToSid(stringSid *uint16, sid **SID) (err error) = advapi32.ConvertStringSidToSidW
- //sys GetLengthSid(sid *SID) (len uint32) = advapi32.GetLengthSid
- //sys CopySid(destSidLen uint32, destSid *SID, srcSid *SID) (err error) = advapi32.CopySid
- //sys AllocateAndInitializeSid(identAuth *SidIdentifierAuthority, subAuth byte, subAuth0 uint32, subAuth1 uint32, subAuth2 uint32, subAuth3 uint32, subAuth4 uint32, subAuth5 uint32, subAuth6 uint32, subAuth7 uint32, sid **SID) (err error) = advapi32.AllocateAndInitializeSid
- //sys createWellKnownSid(sidType WELL_KNOWN_SID_TYPE, domainSid *SID, sid *SID, sizeSid *uint32) (err error) = advapi32.CreateWellKnownSid
- //sys isWellKnownSid(sid *SID, sidType WELL_KNOWN_SID_TYPE) (isWellKnown bool) = advapi32.IsWellKnownSid
- //sys FreeSid(sid *SID) (err error) [failretval!=0] = advapi32.FreeSid
- //sys EqualSid(sid1 *SID, sid2 *SID) (isEqual bool) = advapi32.EqualSid
- //sys getSidIdentifierAuthority(sid *SID) (authority *SidIdentifierAuthority) = advapi32.GetSidIdentifierAuthority
- //sys getSidSubAuthorityCount(sid *SID) (count *uint8) = advapi32.GetSidSubAuthorityCount
- //sys getSidSubAuthority(sid *SID, index uint32) (subAuthority *uint32) = advapi32.GetSidSubAuthority
- //sys isValidSid(sid *SID) (isValid bool) = advapi32.IsValidSid
-
- // The security identifier (SID) structure is a variable-length
- // structure used to uniquely identify users or groups.
- type SID struct{}
-
- // StringToSid converts a string-format security identifier
- // SID into a valid, functional SID.
- func StringToSid(s string) (*SID, error) {
- var sid *SID
- p, e := UTF16PtrFromString(s)
- if e != nil {
- return nil, e
- }
- e = ConvertStringSidToSid(p, &sid)
- if e != nil {
- return nil, e
- }
- defer LocalFree((Handle)(unsafe.Pointer(sid)))
- return sid.Copy()
- }
-
- // LookupSID retrieves a security identifier SID for the account
- // and the name of the domain on which the account was found.
- // System specify target computer to search.
- func LookupSID(system, account string) (sid *SID, domain string, accType uint32, err error) {
- if len(account) == 0 {
- return nil, "", 0, syscall.EINVAL
- }
- acc, e := UTF16PtrFromString(account)
- if e != nil {
- return nil, "", 0, e
- }
- var sys *uint16
- if len(system) > 0 {
- sys, e = UTF16PtrFromString(system)
- if e != nil {
- return nil, "", 0, e
- }
- }
- n := uint32(50)
- dn := uint32(50)
- for {
- b := make([]byte, n)
- db := make([]uint16, dn)
- sid = (*SID)(unsafe.Pointer(&b[0]))
- e = LookupAccountName(sys, acc, sid, &n, &db[0], &dn, &accType)
- if e == nil {
- return sid, UTF16ToString(db), accType, nil
- }
- if e != ERROR_INSUFFICIENT_BUFFER {
- return nil, "", 0, e
- }
- if n <= uint32(len(b)) {
- return nil, "", 0, e
- }
- }
- }
-
- // String converts SID to a string format
- // suitable for display, storage, or transmission.
- func (sid *SID) String() (string, error) {
- var s *uint16
- e := ConvertSidToStringSid(sid, &s)
- if e != nil {
- return "", e
- }
- defer LocalFree((Handle)(unsafe.Pointer(s)))
- return UTF16ToString((*[256]uint16)(unsafe.Pointer(s))[:]), nil
- }
-
- // Len returns the length, in bytes, of a valid security identifier SID.
- func (sid *SID) Len() int {
- return int(GetLengthSid(sid))
- }
-
- // Copy creates a duplicate of security identifier SID.
- func (sid *SID) Copy() (*SID, error) {
- b := make([]byte, sid.Len())
- sid2 := (*SID)(unsafe.Pointer(&b[0]))
- e := CopySid(uint32(len(b)), sid2, sid)
- if e != nil {
- return nil, e
- }
- return sid2, nil
- }
-
- // IdentifierAuthority returns the identifier authority of the SID.
- func (sid *SID) IdentifierAuthority() SidIdentifierAuthority {
- return *getSidIdentifierAuthority(sid)
- }
-
- // SubAuthorityCount returns the number of sub-authorities in the SID.
- func (sid *SID) SubAuthorityCount() uint8 {
- return *getSidSubAuthorityCount(sid)
- }
-
- // SubAuthority returns the sub-authority of the SID as specified by
- // the index, which must be less than sid.SubAuthorityCount().
- func (sid *SID) SubAuthority(idx uint32) uint32 {
- if idx >= uint32(sid.SubAuthorityCount()) {
- panic("sub-authority index out of range")
- }
- return *getSidSubAuthority(sid, idx)
- }
-
- // IsValid returns whether the SID has a valid revision and length.
- func (sid *SID) IsValid() bool {
- return isValidSid(sid)
- }
-
- // Equals compares two SIDs for equality.
- func (sid *SID) Equals(sid2 *SID) bool {
- return EqualSid(sid, sid2)
- }
-
- // IsWellKnown determines whether the SID matches the well-known sidType.
- func (sid *SID) IsWellKnown(sidType WELL_KNOWN_SID_TYPE) bool {
- return isWellKnownSid(sid, sidType)
- }
-
- // LookupAccount retrieves the name of the account for this SID
- // and the name of the first domain on which this SID is found.
- // System specify target computer to search for.
- func (sid *SID) LookupAccount(system string) (account, domain string, accType uint32, err error) {
- var sys *uint16
- if len(system) > 0 {
- sys, err = UTF16PtrFromString(system)
- if err != nil {
- return "", "", 0, err
- }
- }
- n := uint32(50)
- dn := uint32(50)
- for {
- b := make([]uint16, n)
- db := make([]uint16, dn)
- e := LookupAccountSid(sys, sid, &b[0], &n, &db[0], &dn, &accType)
- if e == nil {
- return UTF16ToString(b), UTF16ToString(db), accType, nil
- }
- if e != ERROR_INSUFFICIENT_BUFFER {
- return "", "", 0, e
- }
- if n <= uint32(len(b)) {
- return "", "", 0, e
- }
- }
- }
-
- // Various types of pre-specified SIDs that can be synthesized and compared at runtime.
- type WELL_KNOWN_SID_TYPE uint32
-
- const (
- WinNullSid = 0
- WinWorldSid = 1
- WinLocalSid = 2
- WinCreatorOwnerSid = 3
- WinCreatorGroupSid = 4
- WinCreatorOwnerServerSid = 5
- WinCreatorGroupServerSid = 6
- WinNtAuthoritySid = 7
- WinDialupSid = 8
- WinNetworkSid = 9
- WinBatchSid = 10
- WinInteractiveSid = 11
- WinServiceSid = 12
- WinAnonymousSid = 13
- WinProxySid = 14
- WinEnterpriseControllersSid = 15
- WinSelfSid = 16
- WinAuthenticatedUserSid = 17
- WinRestrictedCodeSid = 18
- WinTerminalServerSid = 19
- WinRemoteLogonIdSid = 20
- WinLogonIdsSid = 21
- WinLocalSystemSid = 22
- WinLocalServiceSid = 23
- WinNetworkServiceSid = 24
- WinBuiltinDomainSid = 25
- WinBuiltinAdministratorsSid = 26
- WinBuiltinUsersSid = 27
- WinBuiltinGuestsSid = 28
- WinBuiltinPowerUsersSid = 29
- WinBuiltinAccountOperatorsSid = 30
- WinBuiltinSystemOperatorsSid = 31
- WinBuiltinPrintOperatorsSid = 32
- WinBuiltinBackupOperatorsSid = 33
- WinBuiltinReplicatorSid = 34
- WinBuiltinPreWindows2000CompatibleAccessSid = 35
- WinBuiltinRemoteDesktopUsersSid = 36
- WinBuiltinNetworkConfigurationOperatorsSid = 37
- WinAccountAdministratorSid = 38
- WinAccountGuestSid = 39
- WinAccountKrbtgtSid = 40
- WinAccountDomainAdminsSid = 41
- WinAccountDomainUsersSid = 42
- WinAccountDomainGuestsSid = 43
- WinAccountComputersSid = 44
- WinAccountControllersSid = 45
- WinAccountCertAdminsSid = 46
- WinAccountSchemaAdminsSid = 47
- WinAccountEnterpriseAdminsSid = 48
- WinAccountPolicyAdminsSid = 49
- WinAccountRasAndIasServersSid = 50
- WinNTLMAuthenticationSid = 51
- WinDigestAuthenticationSid = 52
- WinSChannelAuthenticationSid = 53
- WinThisOrganizationSid = 54
- WinOtherOrganizationSid = 55
- WinBuiltinIncomingForestTrustBuildersSid = 56
- WinBuiltinPerfMonitoringUsersSid = 57
- WinBuiltinPerfLoggingUsersSid = 58
- WinBuiltinAuthorizationAccessSid = 59
- WinBuiltinTerminalServerLicenseServersSid = 60
- WinBuiltinDCOMUsersSid = 61
- WinBuiltinIUsersSid = 62
- WinIUserSid = 63
- WinBuiltinCryptoOperatorsSid = 64
- WinUntrustedLabelSid = 65
- WinLowLabelSid = 66
- WinMediumLabelSid = 67
- WinHighLabelSid = 68
- WinSystemLabelSid = 69
- WinWriteRestrictedCodeSid = 70
- WinCreatorOwnerRightsSid = 71
- WinCacheablePrincipalsGroupSid = 72
- WinNonCacheablePrincipalsGroupSid = 73
- WinEnterpriseReadonlyControllersSid = 74
- WinAccountReadonlyControllersSid = 75
- WinBuiltinEventLogReadersGroup = 76
- WinNewEnterpriseReadonlyControllersSid = 77
- WinBuiltinCertSvcDComAccessGroup = 78
- WinMediumPlusLabelSid = 79
- WinLocalLogonSid = 80
- WinConsoleLogonSid = 81
- WinThisOrganizationCertificateSid = 82
- WinApplicationPackageAuthoritySid = 83
- WinBuiltinAnyPackageSid = 84
- WinCapabilityInternetClientSid = 85
- WinCapabilityInternetClientServerSid = 86
- WinCapabilityPrivateNetworkClientServerSid = 87
- WinCapabilityPicturesLibrarySid = 88
- WinCapabilityVideosLibrarySid = 89
- WinCapabilityMusicLibrarySid = 90
- WinCapabilityDocumentsLibrarySid = 91
- WinCapabilitySharedUserCertificatesSid = 92
- WinCapabilityEnterpriseAuthenticationSid = 93
- WinCapabilityRemovableStorageSid = 94
- WinBuiltinRDSRemoteAccessServersSid = 95
- WinBuiltinRDSEndpointServersSid = 96
- WinBuiltinRDSManagementServersSid = 97
- WinUserModeDriversSid = 98
- WinBuiltinHyperVAdminsSid = 99
- WinAccountCloneableControllersSid = 100
- WinBuiltinAccessControlAssistanceOperatorsSid = 101
- WinBuiltinRemoteManagementUsersSid = 102
- WinAuthenticationAuthorityAssertedSid = 103
- WinAuthenticationServiceAssertedSid = 104
- WinLocalAccountSid = 105
- WinLocalAccountAndAdministratorSid = 106
- WinAccountProtectedUsersSid = 107
- WinCapabilityAppointmentsSid = 108
- WinCapabilityContactsSid = 109
- WinAccountDefaultSystemManagedSid = 110
- WinBuiltinDefaultSystemManagedGroupSid = 111
- WinBuiltinStorageReplicaAdminsSid = 112
- WinAccountKeyAdminsSid = 113
- WinAccountEnterpriseKeyAdminsSid = 114
- WinAuthenticationKeyTrustSid = 115
- WinAuthenticationKeyPropertyMFASid = 116
- WinAuthenticationKeyPropertyAttestationSid = 117
- WinAuthenticationFreshKeyAuthSid = 118
- WinBuiltinDeviceOwnersSid = 119
- )
-
- // Creates a SID for a well-known predefined alias, generally using the constants of the form
- // Win*Sid, for the local machine.
- func CreateWellKnownSid(sidType WELL_KNOWN_SID_TYPE) (*SID, error) {
- return CreateWellKnownDomainSid(sidType, nil)
- }
-
- // Creates a SID for a well-known predefined alias, generally using the constants of the form
- // Win*Sid, for the domain specified by the domainSid parameter.
- func CreateWellKnownDomainSid(sidType WELL_KNOWN_SID_TYPE, domainSid *SID) (*SID, error) {
- n := uint32(50)
- for {
- b := make([]byte, n)
- sid := (*SID)(unsafe.Pointer(&b[0]))
- err := createWellKnownSid(sidType, domainSid, sid, &n)
- if err == nil {
- return sid, nil
- }
- if err != ERROR_INSUFFICIENT_BUFFER {
- return nil, err
- }
- if n <= uint32(len(b)) {
- return nil, err
- }
- }
- }
-
- const (
- // do not reorder
- TOKEN_ASSIGN_PRIMARY = 1 << iota
- TOKEN_DUPLICATE
- TOKEN_IMPERSONATE
- TOKEN_QUERY
- TOKEN_QUERY_SOURCE
- TOKEN_ADJUST_PRIVILEGES
- TOKEN_ADJUST_GROUPS
- TOKEN_ADJUST_DEFAULT
- TOKEN_ADJUST_SESSIONID
-
- TOKEN_ALL_ACCESS = STANDARD_RIGHTS_REQUIRED |
- TOKEN_ASSIGN_PRIMARY |
- TOKEN_DUPLICATE |
- TOKEN_IMPERSONATE |
- TOKEN_QUERY |
- TOKEN_QUERY_SOURCE |
- TOKEN_ADJUST_PRIVILEGES |
- TOKEN_ADJUST_GROUPS |
- TOKEN_ADJUST_DEFAULT |
- TOKEN_ADJUST_SESSIONID
- TOKEN_READ = STANDARD_RIGHTS_READ | TOKEN_QUERY
- TOKEN_WRITE = STANDARD_RIGHTS_WRITE |
- TOKEN_ADJUST_PRIVILEGES |
- TOKEN_ADJUST_GROUPS |
- TOKEN_ADJUST_DEFAULT
- TOKEN_EXECUTE = STANDARD_RIGHTS_EXECUTE
- )
-
- const (
- // do not reorder
- TokenUser = 1 + iota
- TokenGroups
- TokenPrivileges
- TokenOwner
- TokenPrimaryGroup
- TokenDefaultDacl
- TokenSource
- TokenType
- TokenImpersonationLevel
- TokenStatistics
- TokenRestrictedSids
- TokenSessionId
- TokenGroupsAndPrivileges
- TokenSessionReference
- TokenSandBoxInert
- TokenAuditPolicy
- TokenOrigin
- TokenElevationType
- TokenLinkedToken
- TokenElevation
- TokenHasRestrictions
- TokenAccessInformation
- TokenVirtualizationAllowed
- TokenVirtualizationEnabled
- TokenIntegrityLevel
- TokenUIAccess
- TokenMandatoryPolicy
- TokenLogonSid
- MaxTokenInfoClass
- )
-
- // Group attributes inside of Tokengroups.Groups[i].Attributes
- const (
- SE_GROUP_MANDATORY = 0x00000001
- SE_GROUP_ENABLED_BY_DEFAULT = 0x00000002
- SE_GROUP_ENABLED = 0x00000004
- SE_GROUP_OWNER = 0x00000008
- SE_GROUP_USE_FOR_DENY_ONLY = 0x00000010
- SE_GROUP_INTEGRITY = 0x00000020
- SE_GROUP_INTEGRITY_ENABLED = 0x00000040
- SE_GROUP_LOGON_ID = 0xC0000000
- SE_GROUP_RESOURCE = 0x20000000
- SE_GROUP_VALID_ATTRIBUTES = SE_GROUP_MANDATORY | SE_GROUP_ENABLED_BY_DEFAULT | SE_GROUP_ENABLED | SE_GROUP_OWNER | SE_GROUP_USE_FOR_DENY_ONLY | SE_GROUP_LOGON_ID | SE_GROUP_RESOURCE | SE_GROUP_INTEGRITY | SE_GROUP_INTEGRITY_ENABLED
- )
-
- // Privilege attributes
- const (
- SE_PRIVILEGE_ENABLED_BY_DEFAULT = 0x00000001
- SE_PRIVILEGE_ENABLED = 0x00000002
- SE_PRIVILEGE_REMOVED = 0x00000004
- SE_PRIVILEGE_USED_FOR_ACCESS = 0x80000000
- SE_PRIVILEGE_VALID_ATTRIBUTES = SE_PRIVILEGE_ENABLED_BY_DEFAULT | SE_PRIVILEGE_ENABLED | SE_PRIVILEGE_REMOVED | SE_PRIVILEGE_USED_FOR_ACCESS
- )
-
- // Token types
- const (
- TokenPrimary = 1
- TokenImpersonation = 2
- )
-
- // Impersonation levels
- const (
- SecurityAnonymous = 0
- SecurityIdentification = 1
- SecurityImpersonation = 2
- SecurityDelegation = 3
- )
-
- type LUID struct {
- LowPart uint32
- HighPart int32
- }
-
- type LUIDAndAttributes struct {
- Luid LUID
- Attributes uint32
- }
-
- type SIDAndAttributes struct {
- Sid *SID
- Attributes uint32
- }
-
- type Tokenuser struct {
- User SIDAndAttributes
- }
-
- type Tokenprimarygroup struct {
- PrimaryGroup *SID
- }
-
- type Tokengroups struct {
- GroupCount uint32
- Groups [1]SIDAndAttributes // Use AllGroups() for iterating.
- }
-
- // AllGroups returns a slice that can be used to iterate over the groups in g.
- func (g *Tokengroups) AllGroups() []SIDAndAttributes {
- return (*[(1 << 28) - 1]SIDAndAttributes)(unsafe.Pointer(&g.Groups[0]))[:g.GroupCount:g.GroupCount]
- }
-
- type Tokenprivileges struct {
- PrivilegeCount uint32
- Privileges [1]LUIDAndAttributes // Use AllPrivileges() for iterating.
- }
-
- // AllPrivileges returns a slice that can be used to iterate over the privileges in p.
- func (p *Tokenprivileges) AllPrivileges() []LUIDAndAttributes {
- return (*[(1 << 27) - 1]LUIDAndAttributes)(unsafe.Pointer(&p.Privileges[0]))[:p.PrivilegeCount:p.PrivilegeCount]
- }
-
- type Tokenmandatorylabel struct {
- Label SIDAndAttributes
- }
-
- func (tml *Tokenmandatorylabel) Size() uint32 {
- return uint32(unsafe.Sizeof(Tokenmandatorylabel{})) + GetLengthSid(tml.Label.Sid)
- }
-
- // Authorization Functions
- //sys checkTokenMembership(tokenHandle Token, sidToCheck *SID, isMember *int32) (err error) = advapi32.CheckTokenMembership
- //sys OpenProcessToken(process Handle, access uint32, token *Token) (err error) = advapi32.OpenProcessToken
- //sys OpenThreadToken(thread Handle, access uint32, openAsSelf bool, token *Token) (err error) = advapi32.OpenThreadToken
- //sys ImpersonateSelf(impersonationlevel uint32) (err error) = advapi32.ImpersonateSelf
- //sys RevertToSelf() (err error) = advapi32.RevertToSelf
- //sys SetThreadToken(thread *Handle, token Token) (err error) = advapi32.SetThreadToken
- //sys LookupPrivilegeValue(systemname *uint16, name *uint16, luid *LUID) (err error) = advapi32.LookupPrivilegeValueW
- //sys AdjustTokenPrivileges(token Token, disableAllPrivileges bool, newstate *Tokenprivileges, buflen uint32, prevstate *Tokenprivileges, returnlen *uint32) (err error) = advapi32.AdjustTokenPrivileges
- //sys AdjustTokenGroups(token Token, resetToDefault bool, newstate *Tokengroups, buflen uint32, prevstate *Tokengroups, returnlen *uint32) (err error) = advapi32.AdjustTokenGroups
- //sys GetTokenInformation(token Token, infoClass uint32, info *byte, infoLen uint32, returnedLen *uint32) (err error) = advapi32.GetTokenInformation
- //sys SetTokenInformation(token Token, infoClass uint32, info *byte, infoLen uint32) (err error) = advapi32.SetTokenInformation
- //sys DuplicateTokenEx(existingToken Token, desiredAccess uint32, tokenAttributes *SecurityAttributes, impersonationLevel uint32, tokenType uint32, newToken *Token) (err error) = advapi32.DuplicateTokenEx
- //sys GetUserProfileDirectory(t Token, dir *uint16, dirLen *uint32) (err error) = userenv.GetUserProfileDirectoryW
- //sys getSystemDirectory(dir *uint16, dirLen uint32) (len uint32, err error) = kernel32.GetSystemDirectoryW
- //sys getWindowsDirectory(dir *uint16, dirLen uint32) (len uint32, err error) = kernel32.GetWindowsDirectoryW
- //sys getSystemWindowsDirectory(dir *uint16, dirLen uint32) (len uint32, err error) = kernel32.GetSystemWindowsDirectoryW
-
- // An access token contains the security information for a logon session.
- // The system creates an access token when a user logs on, and every
- // process executed on behalf of the user has a copy of the token.
- // The token identifies the user, the user's groups, and the user's
- // privileges. The system uses the token to control access to securable
- // objects and to control the ability of the user to perform various
- // system-related operations on the local computer.
- type Token Handle
-
- // OpenCurrentProcessToken opens the access token
- // associated with current process. It is a real
- // token that needs to be closed, unlike
- // GetCurrentProcessToken.
- func OpenCurrentProcessToken() (Token, error) {
- p, e := GetCurrentProcess()
- if e != nil {
- return 0, e
- }
- var t Token
- e = OpenProcessToken(p, TOKEN_QUERY|TOKEN_DUPLICATE, &t)
- if e != nil {
- return 0, e
- }
- return t, nil
- }
-
- // GetCurrentProcessToken returns the access token associated with
- // the current process. It is a pseudo token that does not need
- // to be closed.
- func GetCurrentProcessToken() Token {
- return Token(^uintptr(4 - 1))
- }
-
- // GetCurrentThreadToken return the access token associated with
- // the current thread. It is a pseudo token that does not need
- // to be closed.
- func GetCurrentThreadToken() Token {
- return Token(^uintptr(5 - 1))
- }
-
- // GetCurrentThreadEffectiveToken returns the effective access token
- // associated with the current thread. It is a pseudo token that does
- // not need to be closed.
- func GetCurrentThreadEffectiveToken() Token {
- return Token(^uintptr(6 - 1))
- }
-
- // Close releases access to access token.
- func (t Token) Close() error {
- return CloseHandle(Handle(t))
- }
-
- // getInfo retrieves a specified type of information about an access token.
- func (t Token) getInfo(class uint32, initSize int) (unsafe.Pointer, error) {
- n := uint32(initSize)
- for {
- b := make([]byte, n)
- e := GetTokenInformation(t, class, &b[0], uint32(len(b)), &n)
- if e == nil {
- return unsafe.Pointer(&b[0]), nil
- }
- if e != ERROR_INSUFFICIENT_BUFFER {
- return nil, e
- }
- if n <= uint32(len(b)) {
- return nil, e
- }
- }
- }
-
- // GetTokenUser retrieves access token t user account information.
- func (t Token) GetTokenUser() (*Tokenuser, error) {
- i, e := t.getInfo(TokenUser, 50)
- if e != nil {
- return nil, e
- }
- return (*Tokenuser)(i), nil
- }
-
- // GetTokenGroups retrieves group accounts associated with access token t.
- func (t Token) GetTokenGroups() (*Tokengroups, error) {
- i, e := t.getInfo(TokenGroups, 50)
- if e != nil {
- return nil, e
- }
- return (*Tokengroups)(i), nil
- }
-
- // GetTokenPrimaryGroup retrieves access token t primary group information.
- // A pointer to a SID structure representing a group that will become
- // the primary group of any objects created by a process using this access token.
- func (t Token) GetTokenPrimaryGroup() (*Tokenprimarygroup, error) {
- i, e := t.getInfo(TokenPrimaryGroup, 50)
- if e != nil {
- return nil, e
- }
- return (*Tokenprimarygroup)(i), nil
- }
-
- // GetUserProfileDirectory retrieves path to the
- // root directory of the access token t user's profile.
- func (t Token) GetUserProfileDirectory() (string, error) {
- n := uint32(100)
- for {
- b := make([]uint16, n)
- e := GetUserProfileDirectory(t, &b[0], &n)
- if e == nil {
- return UTF16ToString(b), nil
- }
- if e != ERROR_INSUFFICIENT_BUFFER {
- return "", e
- }
- if n <= uint32(len(b)) {
- return "", e
- }
- }
- }
-
- // IsElevated returns whether the current token is elevated from a UAC perspective.
- func (token Token) IsElevated() bool {
- var isElevated uint32
- var outLen uint32
- err := GetTokenInformation(token, TokenElevation, (*byte)(unsafe.Pointer(&isElevated)), uint32(unsafe.Sizeof(isElevated)), &outLen)
- if err != nil {
- return false
- }
- return outLen == uint32(unsafe.Sizeof(isElevated)) && isElevated != 0
- }
-
- // GetLinkedToken returns the linked token, which may be an elevated UAC token.
- func (token Token) GetLinkedToken() (Token, error) {
- var linkedToken Token
- var outLen uint32
- err := GetTokenInformation(token, TokenLinkedToken, (*byte)(unsafe.Pointer(&linkedToken)), uint32(unsafe.Sizeof(linkedToken)), &outLen)
- if err != nil {
- return Token(0), err
- }
- return linkedToken, nil
- }
-
- // GetSystemDirectory retrieves the path to current location of the system
- // directory, which is typically, though not always, `C:\Windows\System32`.
- func GetSystemDirectory() (string, error) {
- n := uint32(MAX_PATH)
- for {
- b := make([]uint16, n)
- l, e := getSystemDirectory(&b[0], n)
- if e != nil {
- return "", e
- }
- if l <= n {
- return UTF16ToString(b[:l]), nil
- }
- n = l
- }
- }
-
- // GetWindowsDirectory retrieves the path to current location of the Windows
- // directory, which is typically, though not always, `C:\Windows`. This may
- // be a private user directory in the case that the application is running
- // under a terminal server.
- func GetWindowsDirectory() (string, error) {
- n := uint32(MAX_PATH)
- for {
- b := make([]uint16, n)
- l, e := getWindowsDirectory(&b[0], n)
- if e != nil {
- return "", e
- }
- if l <= n {
- return UTF16ToString(b[:l]), nil
- }
- n = l
- }
- }
-
- // GetSystemWindowsDirectory retrieves the path to current location of the
- // Windows directory, which is typically, though not always, `C:\Windows`.
- func GetSystemWindowsDirectory() (string, error) {
- n := uint32(MAX_PATH)
- for {
- b := make([]uint16, n)
- l, e := getSystemWindowsDirectory(&b[0], n)
- if e != nil {
- return "", e
- }
- if l <= n {
- return UTF16ToString(b[:l]), nil
- }
- n = l
- }
- }
-
- // IsMember reports whether the access token t is a member of the provided SID.
- func (t Token) IsMember(sid *SID) (bool, error) {
- var b int32
- if e := checkTokenMembership(t, sid, &b); e != nil {
- return false, e
- }
- return b != 0, nil
- }
-
- const (
- WTS_CONSOLE_CONNECT = 0x1
- WTS_CONSOLE_DISCONNECT = 0x2
- WTS_REMOTE_CONNECT = 0x3
- WTS_REMOTE_DISCONNECT = 0x4
- WTS_SESSION_LOGON = 0x5
- WTS_SESSION_LOGOFF = 0x6
- WTS_SESSION_LOCK = 0x7
- WTS_SESSION_UNLOCK = 0x8
- WTS_SESSION_REMOTE_CONTROL = 0x9
- WTS_SESSION_CREATE = 0xa
- WTS_SESSION_TERMINATE = 0xb
- )
-
- const (
- WTSActive = 0
- WTSConnected = 1
- WTSConnectQuery = 2
- WTSShadow = 3
- WTSDisconnected = 4
- WTSIdle = 5
- WTSListen = 6
- WTSReset = 7
- WTSDown = 8
- WTSInit = 9
- )
-
- type WTSSESSION_NOTIFICATION struct {
- Size uint32
- SessionID uint32
- }
-
- type WTS_SESSION_INFO struct {
- SessionID uint32
- WindowStationName *uint16
- State uint32
- }
-
- //sys WTSQueryUserToken(session uint32, token *Token) (err error) = wtsapi32.WTSQueryUserToken
- //sys WTSEnumerateSessions(handle Handle, reserved uint32, version uint32, sessions **WTS_SESSION_INFO, count *uint32) (err error) = wtsapi32.WTSEnumerateSessionsW
- //sys WTSFreeMemory(ptr uintptr) = wtsapi32.WTSFreeMemory
|