sdk/tools/mkhive/mkhive.c

   1 /*
   2  *  ReactOS kernel
   3  *  Copyright (C) 2003, 2006 ReactOS Team
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License along
  16  *  with this program; if not, write to the Free Software Foundation, Inc.,
  17  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19 /*
  20  * COPYRIGHT:       See COPYING in the top level directory
  21  * PROJECT:         ReactOS hive maker
  22  * FILE:            tools/mkhive/mkhive.c
  23  * PURPOSE:         Hive maker
  24  * PROGRAMMERS:     Eric Kohl
  25  *                  Hervé Poussineau
  26  *                  Hermès Bélusca-Maïto
  27  */
  28
  29 /* INCLUDES *****************************************************************/
  30
  31 #include <limits.h>
  32 #include <string.h>
  33 #include <stdio.h>
  34
  35 #include "mkhive.h"
  36
  37 #ifdef _MSC_VER
  38 #include <stdlib.h>
  39 #define PATH_MAX _MAX_PATH
  40 #endif // _MSC_VER
  41
  42 #ifndef _WIN32
  43 #ifndef PATH_MAX
  44 #define PATH_MAX 260
  45 #endif
  46 #define DIR_SEPARATOR_CHAR '/'
  47 #define DIR_SEPARATOR_STRING "/"
  48 #else
  49 #define DIR_SEPARATOR_CHAR '\\'
  50 #define DIR_SEPARATOR_STRING "\\"
  51 #endif
  52
  53 /* FUNCTIONS ****************************************************************/
  54
  55 void usage(void)
  56 {
  57     printf("Usage: mkhive [-?] -h:hive1[,hiveN...] [-u] -d:<dstdir> <inffiles>\n\n"
  58            "  -h:hiveN  - Comma-separated list of hives to create. Possible values are:\n"
  59            "              SETUPREG, SYSTEM, SOFTWARE, DEFAULT, SAM, SECURITY, BCD.\n"
  60            "  -u        - Generate file names in uppercase (default: lowercase) (TEMPORARY FLAG!).\n"
  61            "  -d:dstdir - The binary hive files are created in this directory.\n"
  62            "  inffiles  - List of INF files with full path.\n"
  63            "  -?        - Displays this help screen.\n");
  64 }
  65
  66 void convert_path(char *dst, char *src)
  67 {
  68     int i;
  69
  70     i = 0;
  71     while (src[i] != 0)
  72     {
  73 #ifdef _WIN32
  74         if (src[i] == '/')
  75         {
  76             dst[i] = '\\';
  77         }
  78 #else
  79         if (src[i] == '\\')
  80         {
  81             dst[i] = '/';
  82         }
  83 #endif
  84         else
  85         {
  86             dst[i] = src[i];
  87         }
  88
  89         i++;
  90     }
  91     dst[i] = 0;
  92 }
  93
  94 int main(int argc, char *argv[])
  95 {
  96     INT ret;
  97     UINT i;
  98     PSTR ptr;
  99     BOOL UpperCaseFileName = FALSE;
 100     PCSTR HiveList = NULL;
 101     CHAR DestPath[PATH_MAX] = "";
 102     CHAR FileName[PATH_MAX];
 103
 104     if (argc < 4)
 105     {
 106         usage();
 107         return -1;
 108     }
 109
 110     printf("Binary hive maker\n");
 111
 112     /* Read the options */
 113     for (i = 1; i < argc && *argv[i] == '-'; i++)
 114     {
 115         if (argv[i][1] == '?' && argv[i][2] == 0)
 116         {
 117             usage();
 118             return 0;
 119         }
 120
 121         if (argv[i][1] == 'u' && argv[i][2] == 0)
 122         {
 123             UpperCaseFileName = TRUE;
 124         }
 125         else
 126         if (argv[i][1] == 'h' && (argv[i][2] == ':' || argv[i][2] == '='))
 127         {
 128             HiveList = argv[i] + 3;
 129         }
 130         else if (argv[i][1] == 'd' && (argv[i][2] == ':' || argv[i][2] == '='))
 131         {
 132             convert_path(DestPath, argv[i] + 3);
 133         }
 134         else
 135         {
 136             fprintf(stderr, "Unrecognized option: %s\n", argv[i]);
 137             return -1;
 138         }
 139     }
 140
 141     /* Check whether we have all the parameters needed */
 142     if (!HiveList || !*HiveList)
 143     {
 144         fprintf(stderr, "The mandatory list of hives is missing.\n");
 145         return -1;
 146     }
 147     if (!*DestPath)
 148     {
 149         fprintf(stderr, "The mandatory output directory is missing.\n");
 150         return -1;
 151     }
 152     if (i >= argc)
 153     {
 154         fprintf(stderr, "Not enough parameters, or the list of INF files is missing.\n");
 155         return -1;
 156     }
 157
 158     /* Initialize the registry */
 159     RegInitializeRegistry(HiveList);
 160
 161     /* Default to failure */
 162     ret = -1;
 163
 164     /* Now we should have the list of INF files: parse it */
 165     for (; i < argc; ++i)
 166     {
 167         convert_path(FileName, argv[i]);
 168         if (!ImportRegistryFile(FileName))
 169             goto Quit;
 170     }
 171
 172     for (i = 0; i < MAX_NUMBER_OF_REGISTRY_HIVES; ++i)
 173     {
 174         /* Skip this registry hive if it's not in the list */
 175         if (!strstr(HiveList, RegistryHives[i].HiveName))
 176             continue;
 177
 178         strcpy(FileName, DestPath);
 179         strcat(FileName, DIR_SEPARATOR_STRING);
 180
 181         ptr = FileName + strlen(FileName);
 182
 183         strcat(FileName, RegistryHives[i].HiveName);
 184
 185         /* Exception for the special setup registry hive */
 186         // if (strcmp(RegistryHives[i].HiveName, "SETUPREG") == 0)
 187         if (i == 0)
 188             strcat(FileName, ".HIV");
 189
 190         /* Adjust file name case if needed */
 191         if (UpperCaseFileName)
 192         {
 193             for (; *ptr; ++ptr)
 194                 *ptr = toupper(*ptr);
 195         }
 196         else
 197         {
 198             for (; *ptr; ++ptr)
 199                 *ptr = tolower(*ptr);
 200         }
 201
 202         if (!ExportBinaryHive(FileName, RegistryHives[i].CmHive))
 203             goto Quit;
 204
 205         /* If we happen to deal with the special setup registry hive, stop there */
 206         // if (strcmp(RegistryHives[i].HiveName, "SETUPREG") == 0)
 207         if (i == 0)
 208             break;
 209     }
 210
 211     /* Success */
 212     ret = 0;
 213
 214 Quit:
 215     /* Shut down the registry */
 216     RegShutdownRegistry();
 217
 218     if (ret == 0)
 219         printf("  Done.\n");
 220
 221     return ret;
 222 }
 223
 224 /* EOF */