internal.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_DYN_FUNCS
struct	_DYN_MODULE

Typedefs
typedef struct _DYN_FUNCS	DYN_FUNCS
typedef struct _DYN_FUNCS *	PDYN_FUNCS
typedef struct _DYN_MODULE	DYN_MODULE
typedef struct _DYN_MODULE *	PDYN_MODULE

Functions
BOOL	CopyDirectory (LPCWSTR lpDestinationPath, LPCWSTR lpSourcePath)
BOOL	CreateDirectoryPath (LPCWSTR lpPathName, LPSECURITY_ATTRIBUTES lpSecurityAttributes)
BOOL	RemoveDirectoryPath (LPCWSTR lpPathName)
BOOL	LoadDynamicImports (PDYN_MODULE Module, PDYN_FUNCS DynFuncs)
VOID	UnloadDynamicImports (PDYN_FUNCS DynFuncs)
LPWSTR	AppendBackslash (LPWSTR String)
PSECURITY_DESCRIPTOR	CreateDefaultSecurityDescriptor (VOID)
BOOL	AppendSystemPostfix (LPWSTR lpName, DWORD dwMaxLength)
BOOL	CreateUserHive (LPCWSTR lpKeyName, LPCWSTR lpProfilePath)
BOOL	UpdateUsersShellFolderSettings (LPCWSTR lpUserProfilePath, HKEY hUserKey)
BOOL	GetUserSidStringFromToken (HANDLE hToken, PUNICODE_STRING SidString)
VOID	InitializeGPNotifications (VOID)
VOID	UninitializeGPNotifications (VOID)

Variables
SID_IDENTIFIER_AUTHORITY	LocalSystemAuthority
SID_IDENTIFIER_AUTHORITY	WorldAuthority
DYN_MODULE	DynOle32
HINSTANCE	hInstance

Typedef Documentation

DYN_FUNCS

typedef struct _DYN_FUNCS DYN_FUNCS

DYN_MODULE

typedef struct _DYN_MODULE DYN_MODULE

PDYN_FUNCS

typedef struct _DYN_FUNCS * PDYN_FUNCS

PDYN_MODULE

typedef struct _DYN_MODULE * PDYN_MODULE

Function Documentation

AppendBackslash()

LPWSTR AppendBackslash

(

LPWSTR

String

)

Definition at line 40 of file misc.c.

{
    ULONG Length;
 
    Length = lstrlenW(String);
    if (String[Length - 1] != L'\\')
    {
        String[Length] = L'\\';
        Length++;
        String[Length] = (WCHAR)0;
    }
 
    return &String[Length];
}

Referenced by CopyDirectory().

AppendSystemPostfix()

BOOL AppendSystemPostfix	(	LPWSTR	lpName,
		DWORD	dwMaxLength
	)

Definition at line 20 of file profile.c.

{
    WCHAR szSystemRoot[MAX_PATH];
    LPWSTR lpszPostfix;
    LPWSTR lpszPtr;
 
    /* Build profile name postfix */
    if (!ExpandEnvironmentStringsW(L"%SystemRoot%",
                                   szSystemRoot,
                                   ARRAYSIZE(szSystemRoot)))
    {
        DPRINT1("Error: %lu\n", GetLastError());
        return FALSE;
    }
 
    _wcsupr(szSystemRoot);
 
    /* Get name postfix */
    szSystemRoot[2] = L'.';
    lpszPostfix = &szSystemRoot[2];
    lpszPtr = lpszPostfix;
    while (*lpszPtr != (WCHAR)0)
    {
        if (*lpszPtr == L'\\')
            *lpszPtr = L'_';
        lpszPtr++;
    }
 
    if (wcslen(lpName) + wcslen(lpszPostfix) + 1 >= dwMaxLength)
    {
        DPRINT1("Error: buffer overflow\n");
        SetLastError(ERROR_BUFFER_OVERFLOW);
        return FALSE;
    }
 
    wcscat(lpName, lpszPostfix);
 
    return TRUE;
}

Referenced by CreateStandardProfile().

CopyDirectory()

BOOL CopyDirectory	(	LPCWSTR	lpDestinationPath,
		LPCWSTR	lpSourcePath
	)

Definition at line 82 of file directory.c.

{
    WCHAR szFileName[MAX_PATH];
    WCHAR szFullSrcName[MAX_PATH];
    WCHAR szFullDstName[MAX_PATH];
    WIN32_FIND_DATAW FindFileData;
    LPWSTR lpSrcPtr;
    LPWSTR lpDstPtr;
    HANDLE hFind;
 
    DPRINT("CopyDirectory (%S, %S) called\n",
            lpDestinationPath, lpSourcePath);
 
    wcscpy(szFileName, lpSourcePath);
    wcscat(szFileName, L"\\*.*");
 
    hFind = FindFirstFileW(szFileName,
                           &FindFileData);
    if (hFind == INVALID_HANDLE_VALUE)
    {
        DPRINT1("Error: %lu\n", GetLastError());
        return FALSE;
    }
 
    wcscpy(szFullSrcName, lpSourcePath);
    lpSrcPtr = AppendBackslash(szFullSrcName);
 
    wcscpy(szFullDstName, lpDestinationPath);
    lpDstPtr = AppendBackslash(szFullDstName);
 
    for (;;)
    {
        if (wcscmp(FindFileData.cFileName, L".") &&
            wcscmp(FindFileData.cFileName, L".."))
        {
            wcscpy(lpSrcPtr, FindFileData.cFileName);
            wcscpy(lpDstPtr, FindFileData.cFileName);
 
            if (FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
            {
                DPRINT("Create directory: %S\n", szFullDstName);
                if (!CreateDirectoryExW(szFullSrcName, szFullDstName, NULL))
                {
                    if (GetLastError() != ERROR_ALREADY_EXISTS)
                    {
                        DPRINT1("Error: %lu\n", GetLastError());
 
                        FindClose(hFind);
                        return FALSE;
                    }
                }
 
                if (!CopyDirectory(szFullDstName, szFullSrcName))
                {
                    DPRINT1("Error: %lu\n", GetLastError());
 
                    FindClose(hFind);
                    return FALSE;
                }
            }
            else
            {
                DPRINT("Copy file: %S -> %S\n", szFullSrcName, szFullDstName);
                if (!CopyFileW(szFullSrcName, szFullDstName, FALSE))
                {
                    DPRINT1("Error: %lu\n", GetLastError());
 
                    FindClose(hFind);
                    return FALSE;
                }
            }
        }
 
        if (!FindNextFileW(hFind, &FindFileData))
        {
            if (GetLastError() != ERROR_NO_MORE_FILES)
            {
                DPRINT1("Error: %lu\n", GetLastError());
            }
 
            break;
        }
    }
 
    FindClose(hFind);
 
    DPRINT("CopyDirectory() done\n");
 
    return TRUE;
}

Referenced by CopyDirectory(), CopyProfileDirectoryW(), CopySystemProfile(), and CreateUserProfileExW().

CreateDefaultSecurityDescriptor()

PSECURITY_DESCRIPTOR CreateDefaultSecurityDescriptor

(

VOID

)

Definition at line 57 of file misc.c.

{
    PSID LocalSystemSid = NULL;
    PSID AdministratorsSid = NULL;
    PSID EveryoneSid = NULL;
    PACL Dacl;
    DWORD DaclSize;
    PSECURITY_DESCRIPTOR pSD = NULL;
 
    /* create the SYSTEM, Administrators and Everyone SIDs */
    if (!AllocateAndInitializeSid(&LocalSystemAuthority,
                                  1,
                                  SECURITY_LOCAL_SYSTEM_RID,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  &LocalSystemSid) ||
        !AllocateAndInitializeSid(&LocalSystemAuthority,
                                  2,
                                  SECURITY_BUILTIN_DOMAIN_RID,
                                  DOMAIN_ALIAS_RID_ADMINS,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  &AdministratorsSid) ||
        !AllocateAndInitializeSid(&WorldAuthority,
                                  1,
                                  SECURITY_WORLD_RID,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  0,
                                  &EveryoneSid))
    {
        DPRINT1("Failed initializing the SIDs for the default security descriptor (0x%p, 0x%p, 0x%p)\n",
                LocalSystemSid, AdministratorsSid, EveryoneSid);
        goto Cleanup;
    }
 
    /* allocate the security descriptor and DACL */
    DaclSize = sizeof(ACL) +
               ((GetLengthSid(LocalSystemSid) +
                 GetLengthSid(AdministratorsSid) +
                 GetLengthSid(EveryoneSid)) +
                (3 * FIELD_OFFSET(ACCESS_ALLOWED_ACE,
                                  SidStart)));
 
    pSD = (PSECURITY_DESCRIPTOR)LocalAlloc(LMEM_FIXED,
                                           (SIZE_T)DaclSize + sizeof(SECURITY_DESCRIPTOR));
    if (pSD == NULL)
    {
        DPRINT1("Failed to allocate the default security descriptor and ACL\n");
        goto Cleanup;
    }
 
    if (!InitializeSecurityDescriptor(pSD,
                                      SECURITY_DESCRIPTOR_REVISION))
    {
        DPRINT1("Failed to initialize the default security descriptor\n");
        goto Cleanup;
    }
 
    /* initialize and build the DACL */
    Dacl = (PACL)((ULONG_PTR)pSD + sizeof(SECURITY_DESCRIPTOR));
    if (!InitializeAcl(Dacl,
                       (DWORD)DaclSize,
                       ACL_REVISION))
    {
        DPRINT1("Failed to initialize the DACL of the default security descriptor\n");
        goto Cleanup;
    }
 
    /* add the SYSTEM Ace */
    if (!AddAccessAllowedAce(Dacl,
                             ACL_REVISION,
                             GENERIC_ALL,
                             LocalSystemSid))
    {
        DPRINT1("Failed to add the SYSTEM ACE\n");
        goto Cleanup;
    }
 
    /* add the Administrators Ace */
    if (!AddAccessAllowedAce(Dacl,
                             ACL_REVISION,
                             GENERIC_ALL,
                             AdministratorsSid))
    {
        DPRINT1("Failed to add the Administrators ACE\n");
        goto Cleanup;
    }
 
    /* add the Everyone Ace */
    if (!AddAccessAllowedAce(Dacl,
                             ACL_REVISION,
                             GENERIC_EXECUTE,
                             EveryoneSid))
    {
        DPRINT1("Failed to add the Everyone ACE\n");
        goto Cleanup;
    }
 
    /* set the DACL */
    if (!SetSecurityDescriptorDacl(pSD,
                                   TRUE,
                                   Dacl,
                                   FALSE))
    {
        DPRINT1("Failed to set the DACL of the default security descriptor\n");
 
Cleanup:
        if (pSD != NULL)
        {
            LocalFree((HLOCAL)pSD);
            pSD = NULL;
        }
    }
 
    if (LocalSystemSid != NULL)
    {
        FreeSid(LocalSystemSid);
    }
    if (AdministratorsSid != NULL)
    {
        FreeSid(AdministratorsSid);
    }
    if (EveryoneSid != NULL)
    {
        FreeSid(EveryoneSid);
    }
 
    return pSD;
}

Referenced by EnterCriticalPolicySection(), and RegisterGPNotification().

CreateDirectoryPath()

BOOL CreateDirectoryPath	(	LPCWSTR	lpPathName,
		LPSECURITY_ATTRIBUTES	lpSecurityAttributes
	)

Definition at line 176 of file directory.c.

{
    WCHAR szPath[MAX_PATH];
    LPWSTR Ptr;
    DWORD dwError;
 
    DPRINT("CreateDirectoryPath() called\n");
 
    if (lpPathName == NULL || *lpPathName == 0)
        return TRUE;
 
    if (CreateDirectoryW(lpPathName,
                         lpSecurityAttributes))
        return TRUE;
 
    dwError = GetLastError();
    if (dwError == ERROR_ALREADY_EXISTS)
        return TRUE;
 
    wcscpy(szPath, lpPathName);
 
    if (wcslen(szPath) > 3 && szPath[1] == ':' && szPath[2] == '\\')
    {
        Ptr = &szPath[3];
    }
    else
    {
        Ptr = szPath;
    }
 
    while (Ptr != NULL)
    {
        Ptr = wcschr(Ptr, L'\\');
        if (Ptr != NULL)
            *Ptr = 0;
 
        DPRINT("CreateDirectory(%S)\n", szPath);
        if (!CreateDirectoryW(szPath,
                              lpSecurityAttributes))
        {
            dwError = GetLastError();
            if (dwError != ERROR_ALREADY_EXISTS)
                return FALSE;
        }
 
        if (Ptr != NULL)
        {
            *Ptr = L'\\';
            Ptr++;
        }
    }
 
    DPRINT("CreateDirectoryPath() done\n");
 
    return TRUE;
}

Referenced by AddDesktopItemW(), and CreateGroupW().

CreateUserHive()

BOOL CreateUserHive	(	LPCWSTR	lpKeyName,
		LPCWSTR	lpProfilePath
	)

Definition at line 279 of file registry.c.

{
    HKEY hDefaultKey = NULL;
    HKEY hUserKey = NULL;
    LONG Error;
    BOOL Ret = FALSE;
 
    DPRINT("CreateUserHive(%S) called\n", lpKeyName);
 
    Error = RegOpenKeyExW(HKEY_USERS,
                          L".Default",
                          0,
                          KEY_READ,
                          &hDefaultKey);
    if (Error != ERROR_SUCCESS)
    {
        SetLastError((DWORD)Error);
        goto Cleanup;
    }
 
    Error = RegOpenKeyExW(HKEY_USERS,
                          lpKeyName,
                          0,
                          KEY_ALL_ACCESS,
                          &hUserKey);
    if (Error != ERROR_SUCCESS)
    {
        SetLastError((DWORD)Error);
        goto Cleanup;
    }
 
    if (!CopyKey(hUserKey, hDefaultKey))
    {
        goto Cleanup;
    }
 
    if (!UpdateUsersShellFolderSettings(lpProfilePath,
                                        hUserKey))
    {
        goto Cleanup;
    }
 
    RegFlushKey(hUserKey);
    Ret = TRUE;
 
Cleanup:
    if (hUserKey != NULL)
        RegCloseKey (hUserKey);
 
    if (hDefaultKey != NULL)
        RegCloseKey (hDefaultKey);
 
    return Ret;
}

Referenced by CreateUserProfileExW().

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().

InitializeGPNotifications()

VOID InitializeGPNotifications

(

VOID

)

Definition at line 65 of file gpolicy.c.

{
    InitializeCriticalSection(&GPNotifyLock);
}

Referenced by DllMain().

LoadDynamicImports()

BOOL LoadDynamicImports	(	PDYN_MODULE	Module,
		PDYN_FUNCS	DynFuncs
	)

Definition at line 223 of file misc.c.

{
    LPSTR *fname;
    PVOID *fn;
 
    ZeroMemory(DynFuncs, sizeof(DYN_FUNCS));
 
    DynFuncs->hModule = LoadLibraryW(Module->Library);
    if (!DynFuncs->hModule)
    {
        return FALSE;
    }
 
    fn = &DynFuncs->fn.foo;
 
    /* load the imports */
    for (fname = Module->Functions; *fname != NULL; fname++)
    {
        *fn = GetProcAddress(DynFuncs->hModule, *fname);
        if (*fn == NULL)
        {
            FreeLibrary(DynFuncs->hModule);
            DynFuncs->hModule = (HMODULE)0;
 
            return FALSE;
        }
 
        fn++;
    }
 
    return TRUE;
}

Referenced by AddDesktopItemW(), and AddItemW().

RemoveDirectoryPath()

BOOL RemoveDirectoryPath

(

LPCWSTR

lpPathName

)

Definition at line 324 of file directory.c.

{
    if (!RecursiveRemoveDir(lpPathName))
        return FALSE;
 
    DPRINT("Delete directory: '%S'\n", lpPathName);
    return RemoveDirectoryW(lpPathName);
}

Referenced by DeleteGroupW(), and DeleteProfileW().

UninitializeGPNotifications()

VOID UninitializeGPNotifications

(

VOID

)

Definition at line 71 of file gpolicy.c.

{
    EnterCriticalSection(&GPNotifyLock);
 
    /* rundown the notification thread */
    if (hNotificationThread != NULL)
    {
        ASSERT(hNotificationThreadEvent != NULL);
 
        /* notify the thread */
        GPNotificationAction = gpaTerminate;
        SetEvent(hNotificationThreadEvent);
 
        LeaveCriticalSection(&GPNotifyLock);
 
        /* wait for the thread to terminate itself */
        WaitForSingleObject(hNotificationThread,
                            INFINITE);
 
        EnterCriticalSection(&GPNotifyLock);
 
        if (hNotificationThread != NULL)
        {
            /* the handle should be closed by the thread,
               just in case that didn't happen for an unknown reason */
            CloseHandle(hNotificationThread);
            hNotificationThread = NULL;
        }
    }
 
    if (hNotificationThreadEvent != NULL)
    {
        CloseHandle(hNotificationThreadEvent);
        hNotificationThreadEvent = NULL;
    }
 
    LeaveCriticalSection(&GPNotifyLock);
 
    DeleteCriticalSection(&GPNotifyLock);
}

Referenced by DllMain().

UnloadDynamicImports()

VOID UnloadDynamicImports

(

PDYN_FUNCS

DynFuncs

)

Definition at line 259 of file misc.c.

{
    if (DynFuncs->hModule)
    {
        FreeLibrary(DynFuncs->hModule);
        DynFuncs->hModule = (HMODULE)0;
    }
}

Referenced by AddDesktopItemW(), and AddItemW().

UpdateUsersShellFolderSettings()

BOOL UpdateUsersShellFolderSettings	(	LPCWSTR	lpUserProfilePath,
		HKEY	hUserKey
	)

Definition at line 434 of file setup.c.

{
    WCHAR szBuffer[MAX_PATH];
    DWORD dwLength;
    PFOLDERDATA lpFolderData;
    HKEY hFoldersKey;
    LONG Error;
 
    DPRINT("UpdateUsersShellFolderSettings() called\n");
 
    DPRINT("User profile path: %S\n", lpUserProfilePath);
 
    Error = RegOpenKeyExW(hUserKey,
                          L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders",
                          0,
                          KEY_SET_VALUE,
                          &hFoldersKey);
    if (Error != ERROR_SUCCESS)
    {
        DPRINT1("Error: %lu\n", Error);
        SetLastError((DWORD)Error);
        return FALSE;
    }
 
    lpFolderData = &UserShellFolders[0];
    while (lpFolderData->lpValueName != NULL)
    {
        if (lpFolderData->bShellFolder)
        {
            StringCbCopyW(szBuffer, sizeof(szBuffer), lpUserProfilePath);
            StringCbCatW(szBuffer, sizeof(szBuffer), L"\\");
 
            /* Append the folder name */
            dwLength = wcslen(szBuffer);
            if (!LoadStringW(hInstance,
                             lpFolderData->uId,
                             &szBuffer[dwLength],
                             ARRAYSIZE(szBuffer) - dwLength))
            {
                /* Use the default name instead */
                StringCbCatW(szBuffer, sizeof(szBuffer), lpFolderData->lpPath);
            }
 
            DPRINT("%S: %S\n", lpFolderData->lpValueName, szBuffer);
 
            dwLength = (wcslen(szBuffer) + 1) * sizeof(WCHAR);
            Error = RegSetValueExW(hFoldersKey,
                                   lpFolderData->lpValueName,
                                   0,
                                   REG_SZ,
                                   (LPBYTE)szBuffer,
                                   dwLength);
            if (Error != ERROR_SUCCESS)
            {
                DPRINT1("Error: %lu\n", Error);
                RegCloseKey(hFoldersKey);
                SetLastError((DWORD)Error);
                return FALSE;
            }
        }
 
        lpFolderData++;
    }
 
    RegCloseKey(hFoldersKey);
 
    DPRINT("UpdateUsersShellFolderSettings() done\n");
 
    return TRUE;
}

Referenced by CreateUserHive().

Variable Documentation

DynOle32

DYN_MODULE DynOle32

extern

Definition at line 204 of file misc.c.

Referenced by AddDesktopItemW(), and AddItemW().

hInstance

HINSTANCE hInstance

extern

Definition at line 19 of file charmap.c.

LocalSystemAuthority

SID_IDENTIFIER_AUTHORITY LocalSystemAuthority

extern

Definition at line 34 of file misc.c.

Referenced by CheckForGuestsAndAdmins(), and CreateDefaultSecurityDescriptor().

WorldAuthority

SID_IDENTIFIER_AUTHORITY WorldAuthority

extern

Definition at line 35 of file misc.c.

Referenced by CreateDefaultSecurityDescriptor().