reactos/reactos/lib/userenv/setup.c

/* $Id: setup.c,v 1.5 2004/06/24 14:48:33 ekohl Exp $ 
 *
 * COPYRIGHT:       See COPYING in the top level directory
 * PROJECT:         ReactOS system libraries
 * FILE:            lib/userenv/setup.c
 * PURPOSE:         Profile setup functions
 * PROGRAMMER:      Eric Kohl
 */

#include <ntos.h>
#include <windows.h>
#include <string.h>
#include <stdarg.h>
#include <stdio.h>

#include <userenv.h>

#include "internal.h"

typedef struct _DIRDATA
{
  BOOL Hidden;
  LPWSTR DirName;
} DIRDATA, *PDIRDATA;


static DIRDATA
DefaultUserDirectories[] =
{
  {TRUE,  L"Application Data"},
  {FALSE, L"Desktop"},
  {FALSE, L"Favorites"},
  {FALSE, L"My Documents"},
  {TRUE,  L"PrintHood"},
  {TRUE,  L"Recent"},
  {FALSE, L"Start Menu"},
  {FALSE, L"Start Menu\\Programs"},
  {FALSE, L"Start Menu\\Programs\\Startup"},
  {TRUE,  L"Local Settings"},
  {TRUE,  L"Local Settings\\Application Data"},
  {FALSE, L"Local Settings\\Temp"},
  {FALSE, NULL}
};


static DIRDATA
AllUsersDirectories[] =
{
  {TRUE,  L"Application Data"},
  {FALSE, L"Desktop"},
  {FALSE, L"Favorites"},
  {FALSE, L"My Documents"},
  {FALSE, L"Start Menu"},
  {FALSE, L"Start Menu\\Programs"},
  {FALSE, L"Start Menu\\Programs\\Startup"},
  {FALSE, L"Start Menu\\Programs\\Administrative Tools"},
  {TRUE,  L"Templates"},
  {FALSE, NULL}
};


void
DebugPrint(char* fmt,...)
{
  char buffer[512];
  va_list ap;

  va_start(ap, fmt);
  vsprintf(buffer, fmt, ap);
  va_end(ap);

  OutputDebugStringA(buffer);
}


BOOL WINAPI
InitializeProfiles (VOID)
{
  WCHAR szProfilesPath[MAX_PATH];
  WCHAR szProfilePath[MAX_PATH];
  WCHAR szBuffer[MAX_PATH];
  LPWSTR lpszPtr;
  DWORD dwLength;
  PDIRDATA lpDirData;

  HKEY hKey;

  if (RegOpenKeyExW (HKEY_LOCAL_MACHINE,
		     L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList",
		     0,
		     KEY_ALL_ACCESS,
		     &hKey))
    {
      DPRINT1("Error: %lu\n", GetLastError());
      return FALSE;
    }

  /* Get profiles path */
  dwLength = MAX_PATH * sizeof(WCHAR);
  if (RegQueryValueExW (hKey,
			L"ProfilesDirectory",
			NULL,
			NULL,
			(LPBYTE)szBuffer,
			&dwLength))
    {
      DPRINT1("Error: %lu\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  /* Expand it */
  if (!ExpandEnvironmentStringsW (szBuffer,
				  szProfilesPath,
				  MAX_PATH))
    {
      DPRINT1("Error: %lu\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  /* Create profiles directory */
  if (!CreateDirectoryW (szProfilesPath, NULL))
    {
      if (GetLastError () != ERROR_ALREADY_EXISTS)
	{
	  DPRINT1("Error: %lu\n", GetLastError());
	  RegCloseKey (hKey);
	  return FALSE;
	}
    }

  /* Set 'DefaultUserProfile' value */
  wcscpy (szBuffer, L"Default User");
  if (!AppendSystemPostfix (szBuffer, MAX_PATH))
    {
      DPRINT1("AppendSystemPostfix() failed\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  dwLength = (wcslen (szBuffer) + 1) * sizeof(WCHAR);
  if (RegSetValueExW (hKey,
		      L"DefaultUserProfile",
		      0,
		      REG_SZ,
		      (LPBYTE)szBuffer,
		      dwLength))
    {
      DPRINT1("Error: %lu\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  /* Create 'Default User' profile directory */
  wcscpy (szProfilePath, szProfilesPath);
  wcscat (szProfilePath, L"\\");
  wcscat (szProfilePath, szBuffer);
  if (!CreateDirectoryW (szProfilePath, NULL))
    {
      if (GetLastError () != ERROR_ALREADY_EXISTS)
	{
	  DPRINT1("Error: %lu\n", GetLastError());
	  RegCloseKey (hKey);
	  return FALSE;
	}
    }

  /* Set current user profile */
  SetEnvironmentVariableW (L"USERPROFILE", szProfilePath);

  /* Create 'Default User' subdirectories */
  /* FIXME: Get these paths from the registry */
  lpszPtr = AppendBackslash (szProfilePath);
  lpDirData = &DefaultUserDirectories[0];
  while (lpDirData->DirName != NULL)
    {
      wcscpy (lpszPtr, lpDirData->DirName);

      if (!CreateDirectoryW (szProfilePath, NULL))
	{
	  if (GetLastError () != ERROR_ALREADY_EXISTS)
	    {
	      DPRINT1("Error: %lu\n", GetLastError());
	      RegCloseKey (hKey);
	      return FALSE;
	    }
	}

      if (lpDirData->Hidden == TRUE)
	{
	  SetFileAttributesW (szProfilePath,
			      FILE_ATTRIBUTE_HIDDEN);
	}

      lpDirData++;
    }

  /* Set 'AllUsersProfile' value */
  wcscpy (szBuffer, L"All Users");
  if (!AppendSystemPostfix (szBuffer, MAX_PATH))
    {
      DPRINT1("AppendSystemPostfix() failed\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  dwLength = (wcslen (szBuffer) + 1) * sizeof(WCHAR);
  if (RegSetValueExW (hKey,
		      L"AllUsersProfile",
		      0,
		      REG_SZ,
		      (LPBYTE)szBuffer,
		      dwLength))
    {
      DPRINT1("Error: %lu\n", GetLastError());
      RegCloseKey (hKey);
      return FALSE;
    }

  /* Create 'All Users' profile directory */
  wcscpy (szProfilePath, szProfilesPath);
  wcscat (szProfilePath, L"\\");
  wcscat (szProfilePath, szBuffer);
  if (!CreateDirectoryW (szProfilePath, NULL))
    {
      if (GetLastError () != ERROR_ALREADY_EXISTS)
	{
	  DPRINT1("Error: %lu\n", GetLastError());
	  RegCloseKey (hKey);
	  return FALSE;
	}
    }

  /* Set 'All Users' profile */
  SetEnvironmentVariableW (L"ALLUSERSPROFILE", szProfilePath);

  /* Create 'All Users' subdirectories */
  /* FIXME: Take these paths from the registry */
  lpszPtr = AppendBackslash (szProfilePath);
  lpDirData = &AllUsersDirectories[0];
  while (lpDirData->DirName != NULL)
    {
      wcscpy (lpszPtr, lpDirData->DirName);

      if (!CreateDirectoryW (szProfilePath, NULL))
	{
	  if (GetLastError () != ERROR_ALREADY_EXISTS)
	    {
	      DPRINT1("Error: %lu\n", GetLastError());
	      RegCloseKey (hKey);
	      return FALSE;
	    }
	}

      if (lpDirData->Hidden == TRUE)
	{
	  SetFileAttributesW (szProfilePath,
			      FILE_ATTRIBUTE_HIDDEN);
	}

      lpDirData++;
    }

  RegCloseKey (hKey);

  DPRINT1("Success\n");

  return TRUE;
}