mirror of
https://github.com/reactos/reactos.git
synced 2024-11-10 16:48:16 +00:00
281 lines
8.4 KiB
C
281 lines
8.4 KiB
C
/*
|
|
* Windows regedit.exe registry editor implementation.
|
|
*
|
|
* Copyright 2002 Andriy Palamarchuk
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <regedit.h>
|
|
|
|
|
|
static const char *usage =
|
|
"Usage:\n"
|
|
" regedit filename\n"
|
|
" regedit /E filename [regpath]\n"
|
|
" regedit /D regpath\n"
|
|
"\n"
|
|
"filename - registry file name\n"
|
|
"regpath - name of the registry key\n"
|
|
"\n"
|
|
"When is called without any switches adds contents of the specified\n"
|
|
"registry file to the registry\n"
|
|
"\n"
|
|
"Switches:\n"
|
|
" /E - exports contents of the specified registry key to the specified\n"
|
|
" file. Exports the whole registry if no key is specified.\n"
|
|
" /D - deletes specified registry key\n"
|
|
" /S - silent execution, can be used with any other switch.\n"
|
|
" The only existing mode, exists for compatibility with Windows regedit.\n"
|
|
" /V - advanced mode, can be used with any other switch.\n"
|
|
" Ignored, exists for compatibility with Windows regedit.\n"
|
|
" /L - location of system.dat file. Can be used with any other switch.\n"
|
|
" Ignored. Exists for compatibility with Windows regedit.\n"
|
|
" /R - location of user.dat file. Can be used with any other switch.\n"
|
|
" Ignored. Exists for compatibility with Windows regedit.\n"
|
|
" /? - print this help. Any other switches are ignored.\n"
|
|
" /C - create registry from. Not implemented.\n"
|
|
"\n"
|
|
"The switches are case-insensitive, can be prefixed either by '-' or '/'.\n"
|
|
"This program is command-line compatible with Microsoft Windows\n"
|
|
"regedit.\n";
|
|
|
|
typedef enum {
|
|
ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE
|
|
} REGEDIT_ACTION;
|
|
|
|
|
|
const CHAR *getAppName(void)
|
|
{
|
|
return "regedit";
|
|
}
|
|
|
|
/******************************************************************************
|
|
* Copies file name from command line string to the buffer.
|
|
* Rewinds the command line string pointer to the next non-space character
|
|
* after the file name.
|
|
* Buffer contains an empty string if no filename was found;
|
|
*
|
|
* params:
|
|
* command_line - command line current position pointer
|
|
* where *s[0] is the first symbol of the file name.
|
|
* file_name - buffer to write the file name to.
|
|
*/
|
|
void get_file_name(LPWSTR *command_line, LPWSTR file_name)
|
|
{
|
|
WCHAR *s = *command_line;
|
|
int pos = 0; /* position of pointer "s" in *command_line */
|
|
file_name[0] = 0;
|
|
|
|
if (!s[0]) {
|
|
return;
|
|
}
|
|
|
|
if (s[0] == L'"') {
|
|
s++;
|
|
(*command_line)++;
|
|
while(s[0] != L'"') {
|
|
if (!s[0]) {
|
|
fprintf(stderr, "%s: Unexpected end of file name!\n", getAppName());
|
|
exit(1);
|
|
}
|
|
s++;
|
|
pos++;
|
|
}
|
|
} else {
|
|
while(s[0] && !iswspace(s[0])) {
|
|
s++;
|
|
pos++;
|
|
}
|
|
}
|
|
memcpy(file_name, *command_line, pos * sizeof((*command_line)[0]));
|
|
/* remove the last backslash */
|
|
if (file_name[pos - 1] == L'\\') {
|
|
file_name[pos - 1] = L'\0';
|
|
} else {
|
|
file_name[pos] = L'\0';
|
|
}
|
|
|
|
if (s[0]) {
|
|
s++;
|
|
pos++;
|
|
}
|
|
while(s[0] && iswspace(s[0])) {
|
|
s++;
|
|
pos++;
|
|
}
|
|
(*command_line) += pos;
|
|
}
|
|
|
|
BOOL PerformRegAction(REGEDIT_ACTION action, LPWSTR s)
|
|
{
|
|
switch (action) {
|
|
case ACTION_ADD: {
|
|
WCHAR filename[MAX_PATH];
|
|
FILE *fp;
|
|
|
|
get_file_name(&s, filename);
|
|
if (!filename[0]) {
|
|
fprintf(stderr, "%s: No file name is specified\n", getAppName());
|
|
fprintf(stderr, usage);
|
|
exit(4);
|
|
}
|
|
|
|
while(filename[0]) {
|
|
fp = _wfopen(filename, L"r");
|
|
if (fp == NULL)
|
|
{
|
|
LPSTR p = GetMultiByteString(filename);
|
|
perror("");
|
|
fprintf(stderr, "%s: Can't open file \"%s\"\n", getAppName(), p);
|
|
HeapFree(GetProcessHeap(), 0, p);
|
|
exit(5);
|
|
}
|
|
import_registry_file(fp);
|
|
get_file_name(&s, filename);
|
|
}
|
|
break;
|
|
}
|
|
case ACTION_DELETE: {
|
|
WCHAR reg_key_name[KEY_MAX_LEN];
|
|
get_file_name(&s, reg_key_name);
|
|
if (!reg_key_name[0]) {
|
|
fprintf(stderr, "%s: No registry key is specified for removal\n", getAppName());
|
|
fprintf(stderr, usage);
|
|
exit(6);
|
|
}
|
|
delete_registry_key(reg_key_name);
|
|
break;
|
|
}
|
|
case ACTION_EXPORT: {
|
|
WCHAR filename[MAX_PATH];
|
|
|
|
filename[0] = _T('\0');
|
|
get_file_name(&s, filename);
|
|
if (!filename[0]) {
|
|
fprintf(stderr, "%s: No file name is specified\n", getAppName());
|
|
fprintf(stderr, usage);
|
|
exit(7);
|
|
}
|
|
|
|
if (s[0]) {
|
|
WCHAR reg_key_name[KEY_MAX_LEN];
|
|
get_file_name(&s, reg_key_name);
|
|
export_registry_key(filename, reg_key_name, REG_FORMAT_4);
|
|
} else {
|
|
export_registry_key(filename, NULL, REG_FORMAT_4);
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
fprintf(stderr, "%s: Unhandled action!\n", getAppName());
|
|
exit(8);
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
/**
|
|
* Process unknown switch.
|
|
*
|
|
* Params:
|
|
* chu - the switch character in upper-case.
|
|
* s - the command line string where s points to the switch character.
|
|
*/
|
|
static void error_unknown_switch(WCHAR chu, LPWSTR s)
|
|
{
|
|
if (iswalpha(chu)) {
|
|
fprintf(stderr, "%s: Undefined switch /%c!\n", getAppName(), chu);
|
|
} else {
|
|
fprintf(stderr, "%s: Alphabetic character is expected after '%c' "
|
|
"in swit ch specification\n", getAppName(), *(s - 1));
|
|
}
|
|
exit(1);
|
|
}
|
|
|
|
BOOL ProcessCmdLine(LPWSTR lpCmdLine)
|
|
{
|
|
REGEDIT_ACTION action = ACTION_UNDEF;
|
|
LPWSTR s = lpCmdLine; /* command line pointer */
|
|
WCHAR ch = *s; /* current character */
|
|
|
|
while (ch && ((ch == L'-') || (ch == L'/')))
|
|
{
|
|
WCHAR chu;
|
|
WCHAR ch2;
|
|
|
|
s++;
|
|
ch = *s;
|
|
ch2 = *(s + 1);
|
|
chu = (WCHAR)towupper(ch);
|
|
if (!ch2 || iswspace(ch2)) {
|
|
if (chu == L'S' || chu == L'V')
|
|
{
|
|
/* ignore these switches */
|
|
} else {
|
|
switch (chu) {
|
|
case L'D':
|
|
action = ACTION_DELETE;
|
|
break;
|
|
case L'E':
|
|
action = ACTION_EXPORT;
|
|
break;
|
|
case L'?':
|
|
fprintf(stderr, usage);
|
|
exit(3);
|
|
break;
|
|
default:
|
|
error_unknown_switch(chu, s);
|
|
break;
|
|
}
|
|
}
|
|
s++;
|
|
} else {
|
|
if (ch2 == L':') {
|
|
switch (chu) {
|
|
case L'L':
|
|
/* fall through */
|
|
case L'R':
|
|
s += 2;
|
|
while (*s && !iswspace(*s)) {
|
|
s++;
|
|
}
|
|
break;
|
|
default:
|
|
error_unknown_switch(chu, s);
|
|
break;
|
|
}
|
|
} else {
|
|
/* this is a file name, starting from '/' */
|
|
s--;
|
|
break;
|
|
}
|
|
}
|
|
/* skip spaces to the next parameter */
|
|
ch = *s;
|
|
while (ch && iswspace(ch)) {
|
|
s++;
|
|
ch = *s;
|
|
}
|
|
}
|
|
|
|
if (*s && action == ACTION_UNDEF)
|
|
action = ACTION_ADD;
|
|
|
|
if (action == ACTION_UNDEF)
|
|
return FALSE;
|
|
|
|
return PerformRegAction(action, s);
|
|
}
|