[REGEDIT]
[reactos.git] / reactos / base / applications / regedit / regedit.c
1 /*
2 * Windows regedit.exe registry editor implementation.
3 *
4 * Copyright (C) 2002 Andriy Palamarchuk
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include <regedit.h>
22
23
24 static const char *usage =
25 "Usage:\n"
26 " regedit filenames\n"
27 " regedit /E filename [regpath]\n"
28 " regedit /D regpath\n"
29 "\n"
30 "filenames - List of registry files names\n"
31 "filename - Registry file name\n"
32 "regpath - Name of the registry key\n"
33 "\n"
34 "When is called without any switches adds contents of the specified\n"
35 "registry files to the registry.\n"
36 "\n"
37 "Switches:\n"
38 " /E - Exports contents of the specified registry key to the specified\n"
39 " file. Exports the whole registry if no key is specified.\n"
40 " /D - Deletes specified registry key\n"
41 " /S - Silent execution, can be used with any other switch.\n"
42 " The only existing mode, exists for compatibility with Windows regedit.\n"
43 " /V - Advanced mode, can be used with any other switch.\n"
44 " Ignored, exists for compatibility with Windows regedit.\n"
45 " /L - Location of system.dat file. Can be used with any other switch.\n"
46 " Ignored. Exists for compatibility with Windows regedit.\n"
47 " /R - Location of user.dat file. Can be used with any other switch.\n"
48 " Ignored. Exists for compatibility with Windows regedit.\n"
49 " /? - Print this help. Any other switches are ignored.\n"
50 " /C - Create registry from. Not implemented.\n"
51 "\n"
52 "The switches are case-insensitive, can be prefixed either by '-' or '/'.\n"
53 "This program is command-line compatible with Microsoft Windows\n"
54 "regedit.\n";
55
56 typedef enum
57 {
58 ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE
59 } REGEDIT_ACTION;
60
61
62 const CHAR *getAppName(void)
63 {
64 return "regedit";
65 }
66
67 /******************************************************************************
68 * Copies file name from command line string to the buffer.
69 * Rewinds the command line string pointer to the next non-space character
70 * after the file name.
71 * Buffer contains an empty string if no filename was found;
72 *
73 * params:
74 * command_line - command line current position pointer
75 * where *s[0] is the first symbol of the file name.
76 * file_name - buffer to write the file name to.
77 */
78 void get_file_name(LPTSTR *command_line, LPTSTR file_name)
79 {
80 TCHAR *s = *command_line;
81 int pos = 0; /* position of pointer "s" in *command_line */
82 file_name[0] = 0;
83
84 if (!s[0])
85 {
86 return;
87 }
88
89 if (s[0] == _T('"'))
90 {
91 s++;
92 (*command_line)++;
93 while(s[0] != _T('"'))
94 {
95 if (!s[0])
96 {
97 fprintf(stderr, "%s: Unexpected end of file name!\n", getAppName());
98 exit(1);
99 }
100 s++;
101 pos++;
102 }
103 }
104 else
105 {
106 while(s[0] && !_istspace(s[0]))
107 {
108 s++;
109 pos++;
110 }
111 }
112 memcpy(file_name, *command_line, pos * sizeof(WCHAR));
113 /* remove the last backslash */
114 if (file_name[pos - 1] == _T('\\'))
115 {
116 file_name[pos - 1] = _T('\0');
117 }
118 else
119 {
120 file_name[pos] = _T('\0');
121 }
122
123 if (s[0])
124 {
125 s++;
126 pos++;
127 }
128 while(s[0] && _istspace(s[0]))
129 {
130 s++;
131 pos++;
132 }
133 (*command_line) += pos;
134 }
135
136 BOOL PerformRegAction(REGEDIT_ACTION action, LPTSTR s, BOOL silent)
137 {
138 switch (action)
139 {
140 case ACTION_ADD:
141 {
142 TCHAR szTitle[512], szText[512];
143 TCHAR filename[MAX_PATH];
144 FILE *fp;
145
146 get_file_name(&s, filename);
147 if (!filename[0])
148 {
149 fprintf(stderr, "%s: No file name is specified\n", getAppName());
150 fprintf(stderr, usage);
151 exit(4);
152 }
153
154 LoadString(hInst, IDS_APP_TITLE, szTitle, COUNT_OF(szTitle));
155
156 while (filename[0])
157 {
158 /* Request import confirmation */
159 if (!silent)
160 {
161 LoadString(hInst, IDS_IMPORT_PROMPT, szText, COUNT_OF(szText));
162
163 if (InfoMessageBox(NULL, MB_YESNO | MB_ICONWARNING, szTitle, szText, filename) != IDYES)
164 goto cont;
165 }
166
167 fp = _tfopen(filename, _T("r"));
168 if (fp != NULL)
169 {
170 import_registry_file(fp);
171
172 /* Show successful import */
173 if (!silent)
174 {
175 LoadString(hInst, IDS_IMPORT_OK, szText, COUNT_OF(szText));
176 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, szText, filename);
177 }
178 }
179 else
180 {
181 LPSTR p = GetMultiByteString(filename);
182 perror("");
183 fprintf(stderr, "%s: Can't open file \"%s\"\n", getAppName(), p);
184 HeapFree(GetProcessHeap(), 0, p);
185
186 /* Error opening the file */
187 if (!silent)
188 {
189 LoadString(hInst, IDS_IMPORT_ERROR, szText, COUNT_OF(szText));
190 InfoMessageBox(NULL, MB_OK | MB_ICONERROR, szTitle, szText, filename);
191 }
192 }
193
194 cont:
195 get_file_name(&s, filename);
196 }
197 break;
198 }
199
200 case ACTION_DELETE:
201 {
202 TCHAR reg_key_name[KEY_MAX_LEN];
203 get_file_name(&s, reg_key_name);
204 if (!reg_key_name[0])
205 {
206 fprintf(stderr, "%s: No registry key is specified for removal\n", getAppName());
207 fprintf(stderr, usage);
208 exit(6);
209 }
210 delete_registry_key(reg_key_name);
211 break;
212 }
213
214 case ACTION_EXPORT:
215 {
216 TCHAR filename[MAX_PATH];
217
218 filename[0] = _T('\0');
219 get_file_name(&s, filename);
220 if (!filename[0])
221 {
222 fprintf(stderr, "%s: No file name is specified\n", getAppName());
223 fprintf(stderr, usage);
224 exit(7);
225 }
226
227 if (s[0])
228 {
229 TCHAR reg_key_name[KEY_MAX_LEN];
230 get_file_name(&s, reg_key_name);
231 export_registry_key(filename, reg_key_name, REG_FORMAT_4);
232 }
233 else
234 {
235 export_registry_key(filename, NULL, REG_FORMAT_4);
236 }
237 break;
238 }
239
240 default:
241 fprintf(stderr, "%s: Unhandled action!\n", getAppName());
242 exit(8);
243 break;
244 }
245
246 return TRUE;
247 }
248
249 /**
250 * Process unknown switch.
251 *
252 * Params:
253 * chu - the switch character in upper-case.
254 * s - the command line string where s points to the switch character.
255 */
256 static void error_unknown_switch(WCHAR chu, LPWSTR s)
257 {
258 if (iswalpha(chu))
259 {
260 fprintf(stderr, "%s: Undefined switch /%c!\n", getAppName(), chu);
261 }
262 else
263 {
264 fprintf(stderr, "%s: Alphabetic character is expected after '%c' "
265 "in switch specification\n", getAppName(), *(s - 1));
266 }
267 exit(1);
268 }
269
270 BOOL ProcessCmdLine(LPTSTR lpCmdLine)
271 {
272 BOOL silent = FALSE;
273 REGEDIT_ACTION action = ACTION_UNDEF;
274 LPTSTR s = lpCmdLine; /* command line pointer */
275 TCHAR ch = *s; /* current character */
276
277 while (ch && ((ch == _T('-')) || (ch == _T('/'))))
278 {
279 TCHAR chu;
280 TCHAR ch2;
281
282 s++;
283 ch = *s;
284 ch2 = *(s + 1);
285 chu = _totupper(ch);
286 if (!ch2 || _istspace(ch2))
287 {
288 if (chu == _T('S'))
289 {
290 /* Silence dialogs */
291 silent = TRUE;
292 }
293 else if (chu == _T('V'))
294 {
295 /* Ignore this switch */
296 }
297 else
298 {
299 switch (chu)
300 {
301 case _T('D'):
302 action = ACTION_DELETE;
303 break;
304 case _T('E'):
305 action = ACTION_EXPORT;
306 break;
307 case _T('?'):
308 fprintf(stderr, usage);
309 exit(3);
310 break;
311 default:
312 error_unknown_switch(chu, s);
313 break;
314 }
315 }
316 s++;
317 }
318 else
319 {
320 if (ch2 == _T(':'))
321 {
322 switch (chu)
323 {
324 case _T('L'):
325 /* fall through */
326 case _T('R'):
327 s += 2;
328 while (*s && !_istspace(*s))
329 {
330 s++;
331 }
332 break;
333 default:
334 error_unknown_switch(chu, s);
335 break;
336 }
337 }
338 else
339 {
340 /* this is a file name, starting from '/' */
341 s--;
342 break;
343 }
344 }
345 /* skip spaces to the next parameter */
346 ch = *s;
347 while (ch && _istspace(ch))
348 {
349 s++;
350 ch = *s;
351 }
352 }
353
354 if (*s && action == ACTION_UNDEF)
355 action = ACTION_ADD;
356
357 if (action != ACTION_UNDEF)
358 return PerformRegAction(action, s, silent);
359 else
360 return FALSE;
361 }