reg: Parse the registry operation using an enum.

Signed-off-by: Hugh McMaster <hugh.mcmaster@outlook.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Hugh McMaster 2016-06-03 05:23:27 +00:00 committed by Alexandre Julliard
parent 6ce8524271
commit 6a927ef70c
1 changed files with 31 additions and 8 deletions

View File

@ -888,13 +888,34 @@ static BOOL is_help_switch(const WCHAR *s)
return FALSE; return FALSE;
} }
enum operations {
REG_ADD,
REG_DELETE,
REG_QUERY,
REG_INVALID
};
static const WCHAR addW[] = {'a','d','d',0};
static const WCHAR deleteW[] = {'d','e','l','e','t','e',0};
static const WCHAR queryW[] = {'q','u','e','r','y',0};
static enum operations get_operation(const WCHAR *str)
{
if (!lstrcmpiW(str, addW))
return REG_ADD;
if (!lstrcmpiW(str, deleteW))
return REG_DELETE;
if (!lstrcmpiW(str, queryW))
return REG_QUERY;
return REG_INVALID;
}
int wmain(int argc, WCHAR *argvW[]) int wmain(int argc, WCHAR *argvW[])
{ {
int i; int i, op;
static const WCHAR addW[] = {'a','d','d',0};
static const WCHAR deleteW[] = {'d','e','l','e','t','e',0};
static const WCHAR queryW[] = {'q','u','e','r','y',0};
static const WCHAR slashDW[] = {'/','d',0}; static const WCHAR slashDW[] = {'/','d',0};
static const WCHAR slashFW[] = {'/','f',0}; static const WCHAR slashFW[] = {'/','f',0};
static const WCHAR slashSW[] = {'/','s',0}; static const WCHAR slashSW[] = {'/','s',0};
@ -916,7 +937,9 @@ int wmain(int argc, WCHAR *argvW[])
return 0; return 0;
} }
if (!lstrcmpiW(argvW[1], addW)) op = get_operation(argvW[1]);
if (op == REG_ADD)
{ {
WCHAR *key_name, *value_name = NULL, *type = NULL, separator = '\0', *data = NULL; WCHAR *key_name, *value_name = NULL, *type = NULL, separator = '\0', *data = NULL;
BOOL value_empty = FALSE, force = FALSE; BOOL value_empty = FALSE, force = FALSE;
@ -973,7 +996,7 @@ int wmain(int argc, WCHAR *argvW[])
return reg_add(key_name, value_name, value_empty, type, separator, return reg_add(key_name, value_name, value_empty, type, separator,
data, force); data, force);
} }
else if (!lstrcmpiW(argvW[1], deleteW)) else if (op == REG_DELETE)
{ {
WCHAR *key_name, *value_name = NULL; WCHAR *key_name, *value_name = NULL;
BOOL value_empty = FALSE, value_all = FALSE, force = FALSE; BOOL value_empty = FALSE, value_all = FALSE, force = FALSE;
@ -1010,7 +1033,7 @@ int wmain(int argc, WCHAR *argvW[])
} }
return reg_delete(key_name, value_name, value_empty, value_all, force); return reg_delete(key_name, value_name, value_empty, value_all, force);
} }
else if (!lstrcmpiW(argvW[1], queryW)) else if (op == REG_QUERY)
{ {
WCHAR *key_name, *value_name = NULL; WCHAR *key_name, *value_name = NULL;
BOOL value_empty = FALSE, recurse = FALSE; BOOL value_empty = FALSE, recurse = FALSE;