huwcampbell.com
Reactos
1/*
2 * Copyright 2016-2017, 2021 Hugh McMaster
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19#include "reg.h"
20
21static BOOL op_delete_key = TRUE;
22
23static void output_error(LONG rc)
24{
25 if (rc == ERROR_FILE_NOT_FOUND)
26 {
27 if (op_delete_key)
28 output_message(STRING_KEY_NONEXIST);
29 else
30 output_message(STRING_VALUE_NONEXIST);
31 }
32 else
33 {
34 output_message(STRING_ACCESS_DENIED);
35 }
36}
37
38static int run_delete(HKEY root, WCHAR *path, REGSAM sam, WCHAR *key_name, WCHAR *value_name,
39 BOOL value_empty, BOOL value_all, BOOL force)
40{
41 LONG rc;
42 HKEY hkey;
43
44 if (!force)
45 {
46 BOOL ret;
47
48 if (value_name || value_empty)
49 ret = ask_confirm(STRING_DELETE_VALUE, value_name);
50 else if (value_all)
51 ret = ask_confirm(STRING_DELETE_VALUEALL, key_name);
52 else
53 ret = ask_confirm(STRING_DELETE_SUBKEY, key_name);
54
55 if (!ret)
56 {
57 output_message(STRING_CANCELLED);
58 return 0;
59 }
60 }
61
62 if ((rc = RegOpenKeyExW(root, path, 0, KEY_READ|KEY_SET_VALUE|sam, &hkey)))
63 {
64 output_error(rc);
65 return 1;
66 }
67
68 /* Delete registry key if no /v* option is given */
69 if (!value_name && !value_empty && !value_all)
70 {
71 if ((rc = RegDeleteTreeW(hkey, NULL)))
72 {
73 RegCloseKey(hkey);
74 output_error(rc);
75 return 1;
76 }
77
78 RegDeleteKeyW(hkey, L"");
79 RegCloseKey(hkey);
80
81 output_message(STRING_SUCCESS);
82 return 0;
83 }
84
85 op_delete_key = FALSE;
86
87 if (value_all)
88 {
89 DWORD max_value_len = 256, value_len;
90 WCHAR *value_name;
91
92 value_name = malloc(max_value_len * sizeof(WCHAR));
93
94 while (1)
95 {
96 value_len = max_value_len;
97 rc = RegEnumValueW(hkey, 0, value_name, &value_len, NULL, NULL, NULL, NULL);
98 if (rc == ERROR_SUCCESS)
99 {
100 rc = RegDeleteValueW(hkey, value_name);
101 if (rc != ERROR_SUCCESS)
102 {
103 free(value_name);
104 RegCloseKey(hkey);
105 output_message(STRING_VALUEALL_FAILED, key_name);
106 output_error(rc);
107 return 1;
108 }
109 }
110 else if (rc == ERROR_MORE_DATA)
111 {
112 max_value_len *= 2;
113 value_name = realloc(value_name, max_value_len * sizeof(WCHAR));
114 }
115 else break;
116 }
117 free(value_name);
118 }
119 else if (value_name || value_empty)
120 {
121 if ((rc = RegDeleteValueW(hkey, value_name)))
122 {
123 RegCloseKey(hkey);
124 output_error(rc);
125 return 1;
126 }
127 }
128
129 RegCloseKey(hkey);
130 output_message(STRING_SUCCESS);
131 return 0;
132}
133
134int reg_delete(int argc, WCHAR *argvW[])
135{
136 HKEY root;
137 WCHAR *path, *key_name, *value_name = NULL;
138 BOOL value_all = FALSE, value_empty = FALSE, force = FALSE;
139 REGSAM sam = 0;
140 int i;
141
142 if (!parse_registry_key(argvW[2], &root, &path))
143 return 1;
144
145 for (i = 3; i < argc; i++)
146 {
147 WCHAR *str;
148
149 if (argvW[i][0] != '/' && argvW[i][0] != '-')
150 goto invalid;
151
152 str = &argvW[i][1];
153
154 if (!lstrcmpiW(str, L"va"))
155 {
156 if (value_all) goto invalid;
157 value_all = TRUE;
158 continue;
159 }
160 else if (!lstrcmpiW(str, L"ve"))
161 {
162 if (value_empty) goto invalid;
163 value_empty = TRUE;
164 continue;
165 }
166 else if (!lstrcmpiW(str, L"reg:32"))
167 {
168 if (sam & KEY_WOW64_32KEY) goto invalid;
169 sam |= KEY_WOW64_32KEY;
170 continue;
171 }
172 else if (!lstrcmpiW(str, L"reg:64"))
173 {
174 if (sam & KEY_WOW64_64KEY) goto invalid;
175 sam |= KEY_WOW64_64KEY;
176 continue;
177 }
178 else if (!str[0] || str[1])
179 goto invalid;
180
181 switch (towlower(*str))
182 {
183 case 'v':
184 if (value_name || !(value_name = argvW[++i]))
185 goto invalid;
186 break;
187 case 'f':
188 if (force) goto invalid;
189 force = TRUE;
190 break;
191 default:
192 goto invalid;
193 }
194 }
195
196 if ((value_name && value_empty) || (value_name && value_all) || (value_empty && value_all))
197 goto invalid;
198
199 if (sam == (KEY_WOW64_32KEY|KEY_WOW64_64KEY))
200 goto invalid;
201
202 key_name = get_long_key(root, path);
203
204 return run_delete(root, path, sam, key_name, value_name, value_empty, value_all, force);
205
206invalid:
207 output_message(STRING_INVALID_SYNTAX);
208 output_message(STRING_FUNC_HELP, _wcsupr(argvW[1]));
209 return 1;
210}