/*
* ReactOS kernel
- * Copyright (C) 2003 ReactOS Team
+ * Copyright (C) 2003, 2006 ReactOS Team
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-/* $Id$
- * COPYRIGHT: See COPYING in the top level directory
+/* COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS hive maker
* FILE: tools/mkhive/reginf.h
* PURPOSE: Inf file import code
* PROGRAMMER: Eric Kohl
+ * Hervé Poussineau
*/
/* INCLUDES *****************************************************************/
#include <stdlib.h>
#include <stdio.h>
+#define NDEBUG
#include "mkhive.h"
-#include "registry.h"
-#include "infcache.h"
-
-
#define FLG_ADDREG_BINVALUETYPE 0x00000001
#define FLG_ADDREG_NOCLOBBER 0x00000002
static BOOL
GetRootKey (PCHAR Name)
{
- if (!strcasecmp (Name, "HKCR"))
- {
- strcpy (Name, "\\Registry\\Machine\\SOFTWARE\\Classes\\");
- return TRUE;
- }
-
- if (!strcasecmp (Name, "HKCU"))
- {
- strcpy (Name, "\\Registry\\User\\.DEFAULT\\");
- return TRUE;
- }
-
- if (!strcasecmp (Name, "HKLM"))
- {
- strcpy (Name, "\\Registry\\Machine\\");
- return TRUE;
- }
-
- if (!strcasecmp (Name, "HKU"))
- {
- strcpy (Name, "\\Registry\\User\\");
- return TRUE;
- }
+ if (!strcasecmp (Name, "HKCR"))
+ {
+ strcpy (Name, "\\Registry\\Machine\\SOFTWARE\\Classes\\");
+ return TRUE;
+ }
+
+ if (!strcasecmp (Name, "HKCU"))
+ {
+ strcpy (Name, "\\Registry\\User\\.DEFAULT\\");
+ return TRUE;
+ }
+
+ if (!strcasecmp (Name, "HKLM"))
+ {
+ strcpy (Name, "\\Registry\\Machine\\");
+ return TRUE;
+ }
+
+ if (!strcasecmp (Name, "HKU"))
+ {
+ strcpy (Name, "\\Registry\\User\\");
+ return TRUE;
+ }
#if 0
- if (!strcasecmp (Name, "HKR"))
- return FALSE;
+ if (!strcasecmp (Name, "HKR"))
+ return FALSE;
#endif
return FALSE;
* Append a multisz string to a multisz registry value.
*/
static VOID
-AppendMultiSzValue (HKEY KeyHandle,
- PCHAR ValueName,
- PCHAR Strings,
- ULONG StringSize)
+AppendMultiSzValue (
+ IN HKEY KeyHandle,
+ IN PCHAR ValueName,
+ IN PCHAR Strings,
+ IN SIZE_T StringSize)
{
- ULONG Size;
- ULONG Type;
- ULONG Total;
- PCHAR Buffer;
- PCHAR p;
- int len;
- LONG Error;
-
- Error = RegQueryValue (KeyHandle,
- ValueName,
- &Type,
- NULL,
- &Size);
- if ((Error != ERROR_SUCCESS) ||
- (Type != REG_MULTI_SZ))
- return;
-
- Buffer = malloc (Size + StringSize);
- if (Buffer == NULL)
- return;
-
- Error = RegQueryValue (KeyHandle,
- ValueName,
- NULL,
- (PUCHAR)Buffer,
- &Size);
- if (Error != ERROR_SUCCESS)
- goto done;
-
- /* compare each string against all the existing ones */
- Total = Size;
- while (*Strings != 0)
- {
- len = strlen (Strings) + 1;
-
- for (p = Buffer; *p != 0; p += strlen (p) + 1)
- if (!strcasecmp (p, Strings))
- break;
-
- if (*p == 0) /* not found, need to append it */
- {
- memcpy (p, Strings, len);
- p[len] = 0;
- Total += len;
- }
- Strings += len;
- }
-
- if (Total != Size)
- {
- DPRINT ("setting value %s to %s\n", ValueName, Buffer);
- RegSetValue (KeyHandle,
- ValueName,
- REG_MULTI_SZ,
- (PUCHAR)Buffer,
- Total);
- }
+ SIZE_T Size;
+ ULONG Type;
+ size_t Total;
+ PCHAR Buffer;
+ PCHAR p;
+ size_t len;
+ LONG Error;
+
+ Error = RegQueryValueExA (
+ KeyHandle,
+ ValueName,
+ NULL,
+ &Type,
+ NULL,
+ &Size);
+ if ((Error != ERROR_SUCCESS) ||
+ (Type != REG_MULTI_SZ))
+ return;
+
+ Buffer = malloc (Size + StringSize);
+ if (Buffer == NULL)
+ return;
+
+ Error = RegQueryValueExA (
+ KeyHandle,
+ ValueName,
+ NULL,
+ NULL,
+ (PUCHAR)Buffer,
+ &Size);
+ if (Error != ERROR_SUCCESS)
+ goto done;
+
+ /* compare each string against all the existing ones */
+ Total = Size;
+ while (*Strings != 0)
+ {
+ len = strlen (Strings) + 1;
+
+ for (p = Buffer; *p != 0; p += strlen (p) + 1)
+ if (!strcasecmp (p, Strings))
+ break;
+
+ if (*p == 0) /* not found, need to append it */
+ {
+ memcpy (p, Strings, len);
+ p[len] = 0;
+ Total += len;
+ }
+ Strings += len;
+ }
+
+ if (Total != Size)
+ {
+ DPRINT ("setting value %s to %s\n", ValueName, Buffer);
+ RegSetValueExA (
+ KeyHandle,
+ ValueName,
+ 0,
+ REG_MULTI_SZ,
+ (PUCHAR)Buffer,
+ (ULONG)Total);
+ }
done:
- free (Buffer);
+ free (Buffer);
}
* Perform an add/delete registry operation depending on the flags.
*/
static BOOL
-do_reg_operation(HKEY KeyHandle,
- PCHAR ValueName,
- PINFCONTEXT Context,
- ULONG Flags)
+do_reg_operation(
+ IN HKEY KeyHandle,
+ IN PCHAR ValueName,
+ IN PINFCONTEXT Context,
+ IN ULONG Flags)
{
- CHAR EmptyStr = (CHAR)0;
- ULONG Type;
- ULONG Size;
- LONG Error;
-
- if (Flags & FLG_ADDREG_DELVAL) /* deletion */
- {
- if (ValueName)
- {
- RegDeleteValue (KeyHandle,
- ValueName);
- }
- else
- {
- RegDeleteKey (KeyHandle,
- NULL);
- }
-
- return TRUE;
- }
-
- if (Flags & FLG_ADDREG_KEYONLY)
- return TRUE;
-
- if (Flags & (FLG_ADDREG_NOCLOBBER | FLG_ADDREG_OVERWRITEONLY))
- {
- Error = RegQueryValue (KeyHandle,
- ValueName,
- NULL,
- NULL,
- NULL);
- if ((Error == ERROR_SUCCESS) &&
- (Flags & FLG_ADDREG_NOCLOBBER))
- return TRUE;
-
- if ((Error != ERROR_SUCCESS) &&
- (Flags & FLG_ADDREG_OVERWRITEONLY))
- return TRUE;
- }
-
- switch (Flags & FLG_ADDREG_TYPE_MASK)
- {
- case FLG_ADDREG_TYPE_SZ:
- Type = REG_SZ;
- break;
-
- case FLG_ADDREG_TYPE_MULTI_SZ:
- Type = REG_MULTI_SZ;
- break;
-
- case FLG_ADDREG_TYPE_EXPAND_SZ:
- Type = REG_EXPAND_SZ;
- break;
-
- case FLG_ADDREG_TYPE_BINARY:
- Type = REG_BINARY;
- break;
-
- case FLG_ADDREG_TYPE_DWORD:
- Type = REG_DWORD;
- break;
-
- case FLG_ADDREG_TYPE_NONE:
- Type = REG_NONE;
- break;
-
- default:
- Type = Flags >> 16;
- break;
- }
-
- if (!(Flags & FLG_ADDREG_BINVALUETYPE) ||
- (Type == REG_DWORD && InfGetFieldCount (Context) == 5))
- {
- PCHAR Str = NULL;
+ CHAR EmptyStr = (CHAR)0;
+ ULONG Type;
+ ULONG Size;
+ LONG Error;
- if (Type == REG_MULTI_SZ)
+ if (Flags & FLG_ADDREG_DELVAL) /* deletion */
{
- if (!InfGetMultiSzField (Context, 5, NULL, 0, &Size))
- Size = 0;
+ if (ValueName)
+ {
+ RegDeleteValueA (KeyHandle, ValueName);
+ }
+ else
+ {
+ RegDeleteKeyA (KeyHandle, NULL);
+ }
- if (Size)
- {
- Str = malloc (Size);
- if (Str == NULL)
- return FALSE;
-
- InfGetMultiSzField (Context, 5, Str, Size, NULL);
- }
+ return TRUE;
+ }
- if (Flags & FLG_ADDREG_APPEND)
- {
- if (Str == NULL)
+ if (Flags & FLG_ADDREG_KEYONLY)
return TRUE;
- AppendMultiSzValue (KeyHandle,
- ValueName,
- Str,
- Size);
+ if (Flags & (FLG_ADDREG_NOCLOBBER | FLG_ADDREG_OVERWRITEONLY))
+ {
+ Error = RegQueryValueExA (
+ KeyHandle,
+ ValueName,
+ NULL,
+ NULL,
+ NULL,
+ NULL);
+ if ((Error == ERROR_SUCCESS) &&
+ (Flags & FLG_ADDREG_NOCLOBBER))
+ return TRUE;
- free (Str);
- return TRUE;
- }
- /* else fall through to normal string handling */
+ if ((Error != ERROR_SUCCESS) &&
+ (Flags & FLG_ADDREG_OVERWRITEONLY))
+ return TRUE;
}
- else
+
+ switch (Flags & FLG_ADDREG_TYPE_MASK)
{
- if (!InfGetStringField (Context, 5, NULL, 0, &Size))
- Size = 0;
+ case FLG_ADDREG_TYPE_SZ:
+ Type = REG_SZ;
+ break;
- if (Size)
- {
- Str = malloc (Size);
- if (Str == NULL)
- return FALSE;
+ case FLG_ADDREG_TYPE_MULTI_SZ:
+ Type = REG_MULTI_SZ;
+ break;
- InfGetStringField (Context, 5, Str, Size, NULL);
- }
- }
+ case FLG_ADDREG_TYPE_EXPAND_SZ:
+ Type = REG_EXPAND_SZ;
+ break;
- if (Type == REG_DWORD)
- {
- ULONG dw = Str ? strtoul (Str, NULL, 0) : 0;
+ case FLG_ADDREG_TYPE_BINARY:
+ Type = REG_BINARY;
+ break;
- DPRINT("setting dword %s to %lx\n", ValueName, dw);
+ case FLG_ADDREG_TYPE_DWORD:
+ Type = REG_DWORD;
+ break;
- RegSetValue (KeyHandle,
- ValueName,
- Type,
- (PVOID)&dw,
- sizeof(ULONG));
+ case FLG_ADDREG_TYPE_NONE:
+ Type = REG_NONE;
+ break;
+
+ default:
+ Type = Flags >> 16;
+ break;
}
- else
+
+ if (!(Flags & FLG_ADDREG_BINVALUETYPE) ||
+ (Type == REG_DWORD && InfHostGetFieldCount (Context) == 5))
{
- DPRINT("setting value %s to %s\n", ValueName, Str);
-
- if (Str)
- {
- RegSetValue (KeyHandle,
- ValueName,
- Type,
- (PVOID)Str,
- Size);
- }
- else
- {
- RegSetValue (KeyHandle,
- ValueName,
- Type,
- (PVOID)&EmptyStr,
- sizeof(CHAR));
- }
+ CHAR *Str = NULL;
+
+ if (Type == REG_MULTI_SZ)
+ {
+ if (InfHostGetMultiSzField (Context, 5, NULL, 0, &Size) != 0)
+ Size = 0;
+
+ if (Size)
+ {
+ Str = malloc (Size);
+ if (Str == NULL)
+ return FALSE;
+
+ InfHostGetMultiSzField (Context, 5, Str, (ULONG)Size, NULL);
+ }
+
+ if (Flags & FLG_ADDREG_APPEND)
+ {
+ if (Str == NULL)
+ return TRUE;
+
+ AppendMultiSzValue (
+ KeyHandle,
+ ValueName,
+ Str,
+ Size);
+
+ free (Str);
+ return TRUE;
+ }
+ /* else fall through to normal string handling */
+ }
+ else
+ {
+ if (InfHostGetStringField (Context, 5, NULL, 0, &Size) != 0)
+ Size = 0;
+
+ if (Size)
+ {
+ Str = malloc (Size);
+ if (Str == NULL)
+ return FALSE;
+
+ InfHostGetStringField (Context, 5, Str, (ULONG)Size, NULL);
+ }
+ }
+
+ if (Type == REG_DWORD)
+ {
+ ULONG dw = Str ? strtoul (Str, NULL, 0) : 0;
+
+ DPRINT("setting dword %s to %x\n", ValueName, dw);
+
+ RegSetValueExA (
+ KeyHandle,
+ ValueName,
+ 0,
+ Type,
+ (const PUCHAR)&dw,
+ sizeof(ULONG));
+ }
+ else
+ {
+ DPRINT("setting value %s to %s\n", ValueName, Str);
+
+ if (Str)
+ {
+ RegSetValueExA (
+ KeyHandle,
+ ValueName,
+ 0,
+ Type,
+ (PVOID)Str,
+ (ULONG)Size);
+ }
+ else
+ {
+ RegSetValueExA (
+ KeyHandle,
+ ValueName,
+ 0,
+ Type,
+ (PVOID)&EmptyStr,
+ (ULONG)sizeof(CHAR));
+ }
+ }
+ free (Str);
}
- free (Str);
- }
- else /* get the binary data */
- {
- PUCHAR Data = NULL;
-
- if (!InfGetBinaryField (Context, 5, NULL, 0, &Size))
- Size = 0;
-
- if (Size)
+ else /* get the binary data */
{
- Data = malloc (Size);
- if (Data == NULL)
- return FALSE;
-
- DPRINT("setting binary data %s len %lu\n", ValueName, Size);
- InfGetBinaryField (Context, 5, Data, Size, NULL);
+ PUCHAR Data = NULL;
+
+ if (InfHostGetBinaryField (Context, 5, NULL, 0, &Size) != 0)
+ Size = 0;
+
+ if (Size)
+ {
+ Data = malloc (Size);
+ if (Data == NULL)
+ return FALSE;
+
+ DPRINT("setting binary data %s len %d\n", ValueName, Size);
+ InfHostGetBinaryField (Context, 5, Data, Size, NULL);
+ }
+
+ RegSetValueExA (
+ KeyHandle,
+ ValueName,
+ 0,
+ Type,
+ (PVOID)Data,
+ (ULONG)Size);
+
+ free (Data);
}
- RegSetValue (KeyHandle,
- ValueName,
- Type,
- (PVOID)Data,
- Size);
-
- free (Data);
- }
-
- return TRUE;
+ return TRUE;
}
-
/***********************************************************************
* registry_callback
*
static BOOL
registry_callback (HINF hInf, PCHAR Section, BOOL Delete)
{
- CHAR Buffer[MAX_INF_STRING_LENGTH];
- PCHAR ValuePtr;
- ULONG Flags;
- ULONG Length;
-
- INFCONTEXT Context;
- HKEY KeyHandle;
- BOOL Ok;
-
-
- Ok = InfFindFirstLine (hInf, Section, NULL, &Context);
- if (!Ok)
- return TRUE; /* Don't fail if the section isn't present */
-
- for (;Ok; Ok = InfFindNextLine (&Context, &Context))
- {
- /* get root */
- if (!InfGetStringField (&Context, 1, Buffer, MAX_INF_STRING_LENGTH, NULL))
- continue;
- if (!GetRootKey (Buffer))
- continue;
-
- /* get key */
- Length = strlen (Buffer);
- if (!InfGetStringField (&Context, 2, Buffer + Length, MAX_INF_STRING_LENGTH - Length, NULL))
- *Buffer = 0;
-
- DPRINT("KeyName: <%s>\n", Buffer);
-
- if (Delete)
- {
- Flags = FLG_ADDREG_DELVAL;
- }
- else
- {
- /* get flags */
- if (!InfGetIntField (&Context, 4, (PLONG)&Flags))
- Flags = 0;
- }
-
- DPRINT("Flags: %lx\n", Flags);
-
- if (Delete || (Flags & FLG_ADDREG_OVERWRITEONLY))
- {
- if (RegOpenKey (NULL, Buffer, &KeyHandle) != ERROR_SUCCESS)
- {
- DPRINT("RegOpenKey(%s) failed\n", Buffer);
- continue; /* ignore if it doesn't exist */
- }
- }
- else
- {
- if (RegCreateKey (NULL, Buffer, &KeyHandle) != ERROR_SUCCESS)
- {
- DPRINT("RegCreateKey(%s) failed\n", Buffer);
- continue;
- }
- }
+ CHAR Buffer[MAX_INF_STRING_LENGTH];
+ PCHAR ValuePtr;
+ ULONG Flags;
+ size_t Length;
+
+ PINFCONTEXT Context = NULL;
+ HKEY KeyHandle;
+ BOOL Ok;
- /* get value name */
- if (InfGetStringField (&Context, 3, Buffer, MAX_INF_STRING_LENGTH, NULL))
- {
- ValuePtr = Buffer;
- }
- else
- {
- ValuePtr = NULL;
- }
- /* and now do it */
- if (!do_reg_operation (KeyHandle, ValuePtr, &Context, Flags))
+ Ok = InfHostFindFirstLine (hInf, Section, NULL, &Context) == 0;
+ if (!Ok)
+ return TRUE; /* Don't fail if the section isn't present */
+
+ for (;Ok; Ok = (InfHostFindNextLine (Context, Context) == 0))
{
- return FALSE;
+ /* get root */
+ if (InfHostGetStringField (Context, 1, Buffer, MAX_INF_STRING_LENGTH, NULL) != 0)
+ continue;
+ if (!GetRootKey (Buffer))
+ continue;
+
+ /* get key */
+ Length = strlen (Buffer);
+ if (InfHostGetStringField (Context, 2, Buffer + Length, MAX_INF_STRING_LENGTH - (ULONG)Length, NULL) != 0)
+ *Buffer = 0;
+
+ DPRINT("KeyName: <%s>\n", Buffer);
+
+ if (Delete)
+ {
+ Flags = FLG_ADDREG_DELVAL;
+ }
+ else
+ {
+ /* get flags */
+ if (InfHostGetIntField (Context, 4, &Flags) != 0)
+ Flags = 0;
+ }
+
+ DPRINT("Flags: 0x%x\n", Flags);
+
+ if (Delete || (Flags & FLG_ADDREG_OVERWRITEONLY))
+ {
+ if (RegOpenKeyA (NULL, Buffer, &KeyHandle) != ERROR_SUCCESS)
+ {
+ DPRINT("RegOpenKey(%s) failed\n", Buffer);
+ continue; /* ignore if it doesn't exist */
+ }
+ }
+ else
+ {
+ if (RegCreateKeyA (NULL, Buffer, &KeyHandle) != ERROR_SUCCESS)
+ {
+ DPRINT("RegCreateKey(%s) failed\n", Buffer);
+ continue;
+ }
+ }
+
+ /* get value name */
+ if (InfHostGetStringField (Context, 3, Buffer, MAX_INF_STRING_LENGTH, NULL) == 0)
+ {
+ ValuePtr = Buffer;
+ }
+ else
+ {
+ ValuePtr = NULL;
+ }
+
+ /* and now do it */
+ if (!do_reg_operation (KeyHandle, ValuePtr, Context, Flags))
+ {
+ return FALSE;
+ }
}
- }
- return TRUE;
+ InfHostFreeContext(Context);
+
+ return TRUE;
}
BOOL
-ImportRegistryFile(PCHAR FileName,
- PCHAR Section,
- BOOL Delete)
+ImportRegistryFile(PCHAR FileName)
{
- HINF hInf;
- ULONG ErrorLine;
+ HINF hInf;
+ ULONG ErrorLine;
- /* Load inf file from install media. */
- if (!InfOpenFile(&hInf, FileName, &ErrorLine))
- {
- DPRINT1 ("InfOpenFile() failed\n");
- return FALSE;
- }
+ /* Load inf file from install media. */
+ if (InfHostOpenFile(&hInf, FileName, &ErrorLine) != 0)
+ {
+ DPRINT1 ("InfHostOpenFile(%s) failed\n", FileName);
+ return FALSE;
+ }
- if (!registry_callback (hInf, "DelReg", TRUE))
- {
- DPRINT1 ("registry_callback() failed\n");
- }
+ if (!registry_callback (hInf, "DelReg", TRUE))
+ {
+ DPRINT1 ("registry_callback() for DelReg failed\n");
+ }
- if (!registry_callback (hInf, "AddReg", FALSE))
- {
- DPRINT1 ("registry_callback() failed\n");
- }
+ if (!registry_callback (hInf, "AddReg", FALSE))
+ {
+ DPRINT1 ("registry_callback() for AddReg failed\n");
+ }
- InfCloseFile (hInf);
+ InfHostCloseFile (hInf);
- return TRUE;
+ return TRUE;
}
/* EOF */