39a443b32a359e17d1613b106301f1cbdacefaa5
[reactos.git] / 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 #include <limits.h>
30 #include <string.h>
31 #include <stdio.h>
32
33 #include "mkhive.h"
34
35 #ifdef _MSC_VER
36 #include <stdlib.h>
37 #define PATH_MAX _MAX_PATH
38 #endif // _MSC_VER
39
40 #ifndef _WIN32
41 #ifndef PATH_MAX
42 #define PATH_MAX 260
43 #endif
44 #define DIR_SEPARATOR_CHAR '/'
45 #define DIR_SEPARATOR_STRING "/"
46 #else
47 #define DIR_SEPARATOR_CHAR '\\'
48 #define DIR_SEPARATOR_STRING "\\"
49 #endif
50
51
52 void usage(void)
53 {
54 printf("Usage: mkhive -h:hive1[,hiveN...] -d:<dstdir> <inffiles>\n\n"
55 " -h:hiveN - Comma-separated list of hives to create. Possible values are:\n"
56 " SETUPREG, SYSTEM, SOFTWARE, DEFAULT, SAM, SECURITY, BCD.\n"
57 " -d:dstdir - The binary hive files are created in this directory.\n"
58 " inffiles - List of INF files with full path.\n");
59 }
60
61 void convert_path(char *dst, char *src)
62 {
63 int i;
64
65 i = 0;
66 while (src[i] != 0)
67 {
68 #ifdef _WIN32
69 if (src[i] == '/')
70 {
71 dst[i] = '\\';
72 }
73 #else
74 if (src[i] == '\\')
75 {
76 dst[i] = '/';
77 }
78 #endif
79 else
80 {
81 dst[i] = src[i];
82 }
83
84 i++;
85 }
86 dst[i] = 0;
87 }
88
89 int main(int argc, char *argv[])
90 {
91 INT ret;
92 UINT i;
93 PCSTR HiveList = NULL;
94 CHAR DestPath[PATH_MAX] = "";
95 CHAR FileName[PATH_MAX];
96
97 if (argc < 4)
98 {
99 usage();
100 return -1;
101 }
102
103 printf("Binary hive maker\n");
104
105 /* Read the options */
106 for (i = 1; i < argc && *argv[i] == '-'; i++)
107 {
108 if (argv[i][1] == 'h' && (argv[i][2] == ':' || argv[i][2] == '='))
109 {
110 HiveList = argv[i] + 3;
111 }
112 else if (argv[i][1] == 'd' && (argv[i][2] == ':' || argv[i][2] == '='))
113 {
114 convert_path(DestPath, argv[i] + 3);
115 }
116 else
117 {
118 fprintf(stderr, "Unrecognized option: %s\n", argv[i]);
119 return -1;
120 }
121 }
122
123 /* Check whether we have all the parameters needed */
124 if (!HiveList || !*HiveList)
125 {
126 fprintf(stderr, "The mandatory list of hives is missing.\n");
127 return -1;
128 }
129 if (!*DestPath)
130 {
131 fprintf(stderr, "The mandatory output directory is missing.\n");
132 return -1;
133 }
134 if (i >= argc)
135 {
136 fprintf(stderr, "Not enough parameters, or the list of INF files is missing.\n");
137 return -1;
138 }
139
140 /* Initialize the registry */
141 RegInitializeRegistry(HiveList);
142
143 /* Default to failure */
144 ret = -1;
145
146 /* Now we should have the list of INF files: parse it */
147 for (; i < argc; ++i)
148 {
149 convert_path(FileName, argv[i]);
150 if (!ImportRegistryFile(FileName))
151 goto Quit;
152 }
153
154 for (i = 0; i < MAX_NUMBER_OF_REGISTRY_HIVES; ++i)
155 {
156 /* Skip this registry hive if it's not in the list */
157 if (!strstr(HiveList, RegistryHives[i].HiveName))
158 continue;
159
160 strcpy(FileName, DestPath);
161 strcat(FileName, DIR_SEPARATOR_STRING);
162 strcat(FileName, RegistryHives[i].HiveName);
163
164 /* Exception for the special setup registry hive */
165 // if (strcmp(RegistryHives[i].HiveName, "SETUPREG") == 0)
166 if (i == 0)
167 strcat(FileName, ".HIV");
168
169 if (!ExportBinaryHive(FileName, RegistryHives[i].CmHive))
170 goto Quit;
171
172 /* If we happen to deal with the special setup registry hive, stop there */
173 // if (strcmp(RegistryHives[i].HiveName, "SETUPREG") == 0)
174 if (i == 0)
175 break;
176 }
177
178 /* Success */
179 ret = 0;
180
181 Quit:
182 /* Shut down the registry */
183 RegShutdownRegistry();
184
185 if (ret == 0)
186 printf(" Done.\n");
187
188 return ret;
189 }
190
191 /* EOF */