sid.c File Reference

#include "precomp.h"
#include <ndk/sefuncs.h>
#include <debug.h>

Include dependency graph for sid.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
PSID WINAPI	GetUserSid (HANDLE hToken)
BOOL	GetUserSidStringFromToken (HANDLE hToken, PUNICODE_STRING SidString)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 31 of file sid.c.

Function Documentation

GetUserSid()

PSID WINAPI GetUserSid

(

HANDLE

hToken

)

Definition at line 36 of file sid.c.

{
    PTOKEN_USER UserBuffer;
    PTOKEN_USER TempBuffer;
    NTSTATUS Status;
    ULONG Length;
    PSID pSid;
 
    Length = 256;
    UserBuffer = LocalAlloc(LPTR, Length);
    if (UserBuffer == NULL)
    {
        return NULL;
    }
 
    Status = NtQueryInformationToken(hToken,
                                     TokenUser,
                                     (PVOID)UserBuffer,
                                     Length,
                                     &Length);
    if (Status == STATUS_BUFFER_TOO_SMALL)
    {
        TempBuffer = LocalReAlloc(UserBuffer, Length, LMEM_MOVEABLE);
        if (TempBuffer == NULL)
        {
            LocalFree(UserBuffer);
            return NULL;
        }
 
        UserBuffer = TempBuffer;
        Status = NtQueryInformationToken(hToken,
                                         TokenUser,
                                         (PVOID)UserBuffer,
                                         Length,
                                         &Length);
    }
 
    if (!NT_SUCCESS(Status))
    {
        LocalFree(UserBuffer);
        return NULL;
    }
 
        Length = RtlLengthSid(UserBuffer->User.Sid);
 
    pSid = LocalAlloc(LPTR, Length);
    if (pSid == NULL)
    {
        LocalFree(UserBuffer);
        return NULL;
    }
 
    Status = RtlCopySid(Length, pSid, UserBuffer->User.Sid);
 
    LocalFree(UserBuffer);
 
    if (!NT_SUCCESS(Status))
    {
        LocalFree(pSid);
        return NULL;
    }
 
    return pSid;
}

Referenced by CopySystemProfile(), and GetUserAndDomainName().

GetUserSidStringFromToken()

BOOL GetUserSidStringFromToken	(	HANDLE	hToken,
		PUNICODE_STRING	SidString
	)

Definition at line 102 of file sid.c.

{
    PTOKEN_USER UserBuffer, nsb;
    ULONG Length;
    NTSTATUS Status;
 
    Length = 256;
    UserBuffer = LocalAlloc(LPTR, Length);
    if (UserBuffer == NULL)
        return FALSE;
 
    Status = NtQueryInformationToken(hToken,
                                     TokenUser,
                                     (PVOID)UserBuffer,
                                     Length,
                                     &Length);
    if (Status == STATUS_BUFFER_TOO_SMALL)
    {
        nsb = LocalReAlloc(UserBuffer,
                           Length,
                           LMEM_MOVEABLE);
        if (nsb == NULL)
        {
            LocalFree(UserBuffer);
            return FALSE;
        }
 
        UserBuffer = nsb;
        Status = NtQueryInformationToken(hToken,
                                         TokenUser,
                                         (PVOID)UserBuffer,
                                         Length,
                                         &Length);
    }
 
    if (!NT_SUCCESS (Status))
    {
        LocalFree(UserBuffer);
        SetLastError(RtlNtStatusToDosError(Status));
        return FALSE;
    }
 
    DPRINT("SidLength: %lu\n", RtlLengthSid (UserBuffer->User.Sid));
 
    Status = RtlConvertSidToUnicodeString(SidString,
                                          UserBuffer->User.Sid,
                                          TRUE);
 
    LocalFree(UserBuffer);
 
    if (!NT_SUCCESS(Status))
    {
        SetLastError(RtlNtStatusToDosError(Status));
        return FALSE;
    }
 
    DPRINT("SidString.Length: %lu\n", SidString->Length);
    DPRINT("SidString.MaximumLength: %lu\n", SidString->MaximumLength);
    DPRINT("SidString: '%wZ'\n", SidString);
 
    return TRUE;
}

Referenced by CheckForLoadedProfile(), CopySystemProfile(), GetCurrentUserKey(), GetProfileType(), GetUserProfileDirectoryW(), LoadUserProfileW(), and UnloadUserProfile().