[SNDVOL32] Add the small line dialog
[reactos.git] / 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 static const LPCWSTR usage =
24 L"Usage:\n"
25 L" regedit filenames\n"
26 L" regedit /E filename [regpath]\n"
27 L" regedit /D regpath\n"
28 L"\n"
29 L"filenames - List of registry files names\n"
30 L"filename - Registry file name\n"
31 L"regpath - Name of the registry key\n"
32 L"\n"
33 L"When is called without any switches adds contents of the specified\n"
34 L"registry files to the registry.\n"
35 L"\n"
36 L"Switches:\n"
37 L" /E - Exports contents of the specified registry key to the specified\n"
38 L" file. Exports the whole registry if no key is specified.\n"
39 L" /D - Deletes specified registry key\n"
40 L" /S - Silent execution, can be used with any other switch.\n"
41 L" The only existing mode, exists for compatibility with Windows regedit.\n"
42 L" /V - Advanced mode, can be used with any other switch.\n"
43 L" Ignored, exists for compatibility with Windows regedit.\n"
44 L" /L - Location of system.dat file. Can be used with any other switch.\n"
45 L" Ignored. Exists for compatibility with Windows regedit.\n"
46 L" /R - Location of user.dat file. Can be used with any other switch.\n"
47 L" Ignored. Exists for compatibility with Windows regedit.\n"
48 L" /? - Print this help. Any other switches are ignored.\n"
49 L" /C - Create registry from. Not implemented.\n"
50 L"\n"
51 L"The switches are case-insensitive, can be prefixed either by '-' or '/'.\n"
52 L"This program is command-line compatible with Microsoft Windows\n"
53 L"regedit.\n";
54
55 typedef enum
56 {
57 ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE
58 } REGEDIT_ACTION;
59
60
61 LPCWSTR getAppName(void)
62 {
63 return L"regedit";
64 }
65
66 /******************************************************************************
67 * Copies file name from command line string to the buffer.
68 * Rewinds the command line string pointer to the next non-space character
69 * after the file name.
70 * Buffer contains an empty string if no filename was found;
71 *
72 * params:
73 * command_line - command line current position pointer
74 * where *s[0] is the first symbol of the file name.
75 * file_name - buffer to write the file name to.
76 */
77 void get_file_name(LPWSTR *command_line, LPWSTR file_name)
78 {
79 WCHAR *s = *command_line;
80 size_t pos = 0; /* position of pointer "s" in *command_line */
81 file_name[0] = 0;
82
83 if (!s[0])
84 {
85 return;
86 }
87
88 if (s[0] == L'"')
89 {
90 s++;
91 (*command_line)++;
92 while(s[0] != L'"')
93 {
94 if (!s[0])
95 {
96 fwprintf(stderr, L"%s: Unexpected end of file name!\n", getAppName());
97 exit(1);
98 }
99 s++;
100 pos++;
101 }
102 }
103 else
104 {
105 while(s[0] && !iswspace(s[0]))
106 {
107 s++;
108 pos++;
109 }
110 }
111 memcpy(file_name, *command_line, pos * sizeof(WCHAR));
112 /* remove the last backslash */
113 if (file_name[pos - 1] == L'\\')
114 {
115 file_name[pos - 1] = L'\0';
116 }
117 else
118 {
119 file_name[pos] = L'\0';
120 }
121
122 if (s[0])
123 {
124 s++;
125 pos++;
126 }
127 while(s[0] && iswspace(s[0]))
128 {
129 s++;
130 pos++;
131 }
132 (*command_line) += pos;
133 }
134
135 BOOL PerformRegAction(REGEDIT_ACTION action, LPWSTR s, BOOL silent)
136 {
137 switch (action)
138 {
139 case ACTION_ADD:
140 {
141 WCHAR szText[512];
142 WCHAR filename[MAX_PATH];
143 FILE *fp;
144
145 get_file_name(&s, filename);
146 if (!filename[0])
147 {
148 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No file name is specified.");
149 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
150 exit(4);
151 }
152
153 while (filename[0])
154 {
155 /* Request import confirmation */
156 if (!silent)
157 {
158 int choice;
159
160 LoadStringW(hInst, IDS_IMPORT_PROMPT, szText, COUNT_OF(szText));
161
162 choice = InfoMessageBox(NULL, MB_YESNOCANCEL | MB_ICONWARNING, szTitle, szText, filename);
163
164 switch (choice)
165 {
166 case IDNO:
167 goto cont;
168 case IDCANCEL:
169 /* The cancel case is useful if the user is importing more than one registry file
170 at a time, and wants to back out anytime during the import process. This way, the
171 user doesn't have to resort to ending the regedit process abruptly just to cancel
172 the operation. */
173 return TRUE;
174 default:
175 break;
176 }
177 }
178
179 /* Open the file */
180 fp = _wfopen(filename, L"r");
181
182 /* Import it */
183 if (fp == NULL || !import_registry_file(fp))
184 {
185 /* Error opening the file */
186 if (!silent)
187 {
188 LoadStringW(hInst, IDS_IMPORT_ERROR, szText, COUNT_OF(szText));
189 InfoMessageBox(NULL, MB_OK | MB_ICONERROR, szTitle, szText, filename);
190 }
191 }
192 else
193 {
194 /* Show successful import */
195 if (!silent)
196 {
197 LoadStringW(hInst, IDS_IMPORT_OK, szText, COUNT_OF(szText));
198 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, szText, filename);
199 }
200 }
201
202 /* Close the file */
203 if (fp) fclose(fp);
204
205 cont:
206 get_file_name(&s, filename);
207 }
208 break;
209 }
210
211 case ACTION_DELETE:
212 {
213 WCHAR reg_key_name[KEY_MAX_LEN];
214 get_file_name(&s, reg_key_name);
215 if (!reg_key_name[0])
216 {
217 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No registry key is specified for removal.");
218 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
219 exit(6);
220 }
221 delete_registry_key(reg_key_name);
222 break;
223 }
224
225 case ACTION_EXPORT:
226 {
227 WCHAR filename[MAX_PATH];
228
229 filename[0] = L'\0';
230 get_file_name(&s, filename);
231 if (!filename[0])
232 {
233 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, NULL, L"No file name is specified.");
234 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
235 exit(7);
236 }
237
238 if (s[0])
239 {
240 WCHAR reg_key_name[KEY_MAX_LEN];
241 get_file_name(&s, reg_key_name);
242 export_registry_key(filename, reg_key_name, REG_FORMAT_4);
243 }
244 else
245 {
246 export_registry_key(filename, NULL, REG_FORMAT_4);
247 }
248 break;
249 }
250
251 default:
252 fwprintf(stderr, L"%s: Unhandled action!\n", getAppName());
253 exit(8);
254 break;
255 }
256
257 return TRUE;
258 }
259
260 /**
261 * Process unknown switch.
262 *
263 * Params:
264 * chu - the switch character in upper-case.
265 * s - the command line string where s points to the switch character.
266 */
267 static void error_unknown_switch(WCHAR chu, LPWSTR s)
268 {
269 if (iswalpha(chu))
270 {
271 fwprintf(stderr, L"%s: Undefined switch /%c!\n", getAppName(), chu);
272 }
273 else
274 {
275 fwprintf(stderr, L"%s: Alphabetic character is expected after '%c' "
276 L"in switch specification\n", getAppName(), *(s - 1));
277 }
278 exit(1);
279 }
280
281 BOOL ProcessCmdLine(LPWSTR lpCmdLine)
282 {
283 BOOL silent = FALSE;
284 REGEDIT_ACTION action = ACTION_UNDEF;
285 LPWSTR s = lpCmdLine; /* command line pointer */
286 WCHAR ch = *s; /* current character */
287
288 while (ch && ((ch == L'-') || (ch == L'/')))
289 {
290 WCHAR chu;
291 WCHAR ch2;
292
293 s++;
294 ch = *s;
295 ch2 = *(s + 1);
296 chu = towupper(ch);
297 if (!ch2 || iswspace(ch2))
298 {
299 if (chu == L'S')
300 {
301 /* Silence dialogs */
302 silent = TRUE;
303 }
304 else if (chu == L'V')
305 {
306 /* Ignore this switch */
307 }
308 else
309 {
310 switch (chu)
311 {
312 case L'D':
313 action = ACTION_DELETE;
314 break;
315 case L'E':
316 action = ACTION_EXPORT;
317 break;
318 case L'?':
319 InfoMessageBox(NULL, MB_OK | MB_ICONINFORMATION, szTitle, usage);
320 exit(3);
321 break;
322 default:
323 error_unknown_switch(chu, s);
324 break;
325 }
326 }
327 s++;
328 }
329 else
330 {
331 if (ch2 == L':')
332 {
333 switch (chu)
334 {
335 case L'L':
336 /* fall through */
337 case L'R':
338 s += 2;
339 while (*s && !iswspace(*s))
340 {
341 s++;
342 }
343 break;
344 default:
345 error_unknown_switch(chu, s);
346 break;
347 }
348 }
349 else
350 {
351 /* this is a file name, starting from '/' */
352 s--;
353 break;
354 }
355 }
356 /* skip spaces to the next parameter */
357 ch = *s;
358 while (ch && iswspace(ch))
359 {
360 s++;
361 ch = *s;
362 }
363 }
364
365 if (*s && action == ACTION_UNDEF)
366 action = ACTION_ADD;
367
368 if (action != ACTION_UNDEF)
369 return PerformRegAction(action, s, silent);
370 else
371 return FALSE;
372 }