2 * Copyright 2008 Andrew Riedi
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.
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.
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
25 #include <wine/unicode.h>
29 static int reg_printfW(const WCHAR
*msg
, ...)
34 WCHAR msg_buffer
[8192];
36 va_start(va_args
, msg
);
37 vsprintfW(msg_buffer
, msg
, va_args
);
40 wlen
= lstrlenW(msg_buffer
);
41 ret
= WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE
), msg_buffer
, wlen
, &count
, NULL
);
47 len
= WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer
, wlen
,
49 msgA
= HeapAlloc(GetProcessHeap(), 0, len
* sizeof(char));
53 WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer
, wlen
, msgA
, len
,
55 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE
), msgA
, len
, &count
, FALSE
);
56 HeapFree(GetProcessHeap(), 0, msgA
);
62 static int reg_message(int msg
)
64 static const WCHAR formatW
[] = {'%','s',0};
65 WCHAR msg_buffer
[8192];
67 LoadStringW(GetModuleHandleW(NULL
), msg
, msg_buffer
,
68 sizeof(msg_buffer
)/sizeof(WCHAR
));
69 return reg_printfW(formatW
, msg_buffer
);
72 static HKEY
get_rootkey(LPWSTR key
)
74 static const WCHAR szHKLM
[] = {'H','K','L','M',0};
75 static const WCHAR szHKCU
[] = {'H','K','C','U',0};
76 static const WCHAR szHKCR
[] = {'H','K','C','R',0};
77 static const WCHAR szHKU
[] = {'H','K','U',0};
78 static const WCHAR szHKCC
[] = {'H','K','C','C',0};
80 if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKLM
,4)==2)
81 return HKEY_LOCAL_MACHINE
;
82 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCU
,4)==2)
83 return HKEY_CURRENT_USER
;
84 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCR
,4)==2)
85 return HKEY_CLASSES_ROOT
;
86 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,3,szHKU
,3)==2)
88 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCC
,4)==2)
89 return HKEY_CURRENT_CONFIG
;
93 static DWORD
get_regtype(LPWSTR type
)
95 static const WCHAR szREG_SZ
[] = {'R','E','G','_','S','Z',0};
96 static const WCHAR szREG_MULTI_SZ
[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0};
97 static const WCHAR szREG_DWORD_BIG_ENDIAN
[] = {'R','E','G','_','D','W','O','R','D','_','B','I','G','_','E','N','D','I','A','N',0};
98 static const WCHAR szREG_DWORD
[] = {'R','E','G','_','D','W','O','R','D',0};
99 static const WCHAR szREG_BINARY
[] = {'R','E','G','_','B','I','N','A','R','Y',0};
100 static const WCHAR szREG_DWORD_LITTLE_ENDIAN
[] = {'R','E','G','_','D','W','O','R','D','_','L','I','T','T','L','E','_','E','N','D','I','A','N',0};
101 static const WCHAR szREG_NONE
[] = {'R','E','G','_','N','O','N','E',0};
102 static const WCHAR szREG_EXPAND_SZ
[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0};
107 if (lstrcmpiW(type
,szREG_SZ
)==0) return REG_SZ
;
108 if (lstrcmpiW(type
,szREG_DWORD
)==0) return REG_DWORD
;
109 if (lstrcmpiW(type
,szREG_MULTI_SZ
)==0) return REG_MULTI_SZ
;
110 if (lstrcmpiW(type
,szREG_EXPAND_SZ
)==0) return REG_EXPAND_SZ
;
111 if (lstrcmpiW(type
,szREG_DWORD_BIG_ENDIAN
)==0) return REG_DWORD_BIG_ENDIAN
;
112 if (lstrcmpiW(type
,szREG_DWORD_LITTLE_ENDIAN
)==0) return REG_DWORD_LITTLE_ENDIAN
;
113 if (lstrcmpiW(type
,szREG_BINARY
)==0) return REG_BINARY
;
114 if (lstrcmpiW(type
,szREG_NONE
)==0) return REG_NONE
;
119 static LPBYTE
get_regdata(LPWSTR data
, DWORD reg_type
, WCHAR separator
, DWORD
*reg_count
)
121 LPBYTE out_data
= NULL
;
128 *reg_count
= (lstrlenW(data
) + 1) * sizeof(WCHAR
);
129 out_data
= HeapAlloc(GetProcessHeap(),0,*reg_count
);
130 lstrcpyW((LPWSTR
)out_data
,data
);
137 val
= strtolW(data
, &rest
, 0);
139 static const WCHAR nonnumber
[] = {'E','r','r','o','r',':',' ','/','d',' ','r','e','q','u','i','r','e','s',' ','n','u','m','b','e','r','.','\n',0};
140 reg_printfW(nonnumber
);
143 *reg_count
= sizeof(DWORD
);
144 out_data
= HeapAlloc(GetProcessHeap(),0,*reg_count
);
145 ((LPDWORD
)out_data
)[0] = val
;
150 static const WCHAR unhandled
[] = {'U','n','h','a','n','d','l','e','d',' ','T','y','p','e',' ','0','x','%','x',' ',' ','d','a','t','a',' ','%','s','\n',0};
151 reg_printfW(unhandled
, reg_type
,data
);
158 static int reg_add(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
159 WCHAR
*type
, WCHAR separator
, WCHAR
*data
, BOOL force
)
161 static const WCHAR stubW
[] = {'A','D','D',' ','-',' ','%','s',
162 ' ','%','s',' ','%','d',' ','%','s',' ','%','s',' ','%','d','\n',0};
166 reg_printfW(stubW
, key_name
, value_name
, value_empty
, type
, data
, force
);
168 if (key_name
[0]=='\\' && key_name
[1]=='\\')
170 reg_message(STRING_NO_REMOTE
);
174 p
= strchrW(key_name
,'\\');
177 reg_message(STRING_INVALID_KEY
);
182 root
= get_rootkey(key_name
);
185 reg_message(STRING_INVALID_KEY
);
189 if(RegCreateKeyW(root
,p
,&subkey
)!=ERROR_SUCCESS
)
191 reg_message(STRING_INVALID_KEY
);
195 if (value_name
|| data
)
199 BYTE
* reg_data
= NULL
;
203 if (RegQueryValueW(subkey
,value_name
,NULL
,NULL
)==ERROR_SUCCESS
)
205 /* FIXME: Prompt for overwrite */
209 reg_type
= get_regtype(type
);
213 reg_message(STRING_INVALID_CMDLINE
);
218 reg_data
= get_regdata(data
,reg_type
,separator
,®_count
);
220 RegSetValueExW(subkey
,value_name
,0,reg_type
,reg_data
,reg_count
);
221 HeapFree(GetProcessHeap(),0,reg_data
);
225 reg_message(STRING_SUCCESS
);
230 static int reg_delete(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
231 BOOL value_all
, BOOL force
)
236 static const WCHAR stubW
[] = {'D','E','L','E','T','E',
237 ' ','-',' ','%','s',' ','%','s',' ','%','d',' ','%','d',' ','%','d','\n'
239 reg_printfW(stubW
, key_name
, value_name
, value_empty
, value_all
, force
);
241 if (key_name
[0]=='\\' && key_name
[1]=='\\')
243 reg_message(STRING_NO_REMOTE
);
247 p
= strchrW(key_name
,'\\');
250 reg_message(STRING_INVALID_KEY
);
255 root
= get_rootkey(key_name
);
258 reg_message(STRING_INVALID_KEY
);
262 if (value_name
&& value_empty
)
264 reg_message(STRING_INVALID_CMDLINE
);
268 if (value_empty
&& value_all
)
270 reg_message(STRING_INVALID_CMDLINE
);
276 /* FIXME: Prompt for delete */
279 /* Delete subtree only if no /v* option is given */
280 if (!value_name
&& !value_empty
&& !value_all
)
282 if (RegDeleteTreeW(root
,p
)!=ERROR_SUCCESS
)
284 reg_message(STRING_CANNOT_FIND
);
287 reg_message(STRING_SUCCESS
);
291 if(RegOpenKeyW(root
,p
,&subkey
)!=ERROR_SUCCESS
)
293 reg_message(STRING_CANNOT_FIND
);
304 rc
= RegQueryInfoKeyW(subkey
, NULL
, NULL
, NULL
, NULL
, NULL
, NULL
, NULL
,
305 &maxValue
, NULL
, NULL
, NULL
);
306 if (rc
!= ERROR_SUCCESS
)
313 szValue
= HeapAlloc(GetProcessHeap(),0,maxValue
*sizeof(WCHAR
));
318 rc
= RegEnumValueW(subkey
, 0, szValue
, &count
, NULL
, NULL
, NULL
, NULL
);
319 if (rc
== ERROR_SUCCESS
)
321 rc
= RegDeleteValueW(subkey
, szValue
);
322 if (rc
!= ERROR_SUCCESS
)
327 if (rc
!= ERROR_SUCCESS
)
329 /* FIXME delete failed */
334 if (RegDeleteValueW(subkey
,value_name
) != ERROR_SUCCESS
)
337 reg_message(STRING_CANNOT_FIND
);
341 else if (value_empty
)
343 RegSetValueExW(subkey
,NULL
,0,REG_SZ
,NULL
,0);
347 reg_message(STRING_SUCCESS
);
351 static int reg_query(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
354 static const WCHAR stubW
[] = {'S','T','U','B',' ','Q','U','E','R','Y',' ',
355 '-',' ','%','s',' ','%','s',' ','%','d',' ','%','d','\n',0};
356 reg_printfW(stubW
, key_name
, value_name
, value_empty
, subkey
);
361 int wmain(int argc
, WCHAR
*argvW
[])
365 static const WCHAR addW
[] = {'a','d','d',0};
366 static const WCHAR deleteW
[] = {'d','e','l','e','t','e',0};
367 static const WCHAR queryW
[] = {'q','u','e','r','y',0};
368 static const WCHAR slashDW
[] = {'/','d',0};
369 static const WCHAR slashFW
[] = {'/','f',0};
370 static const WCHAR slashHW
[] = {'/','h',0};
371 static const WCHAR slashSW
[] = {'/','s',0};
372 static const WCHAR slashTW
[] = {'/','t',0};
373 static const WCHAR slashVW
[] = {'/','v',0};
374 static const WCHAR slashVAW
[] = {'/','v','a',0};
375 static const WCHAR slashVEW
[] = {'/','v','e',0};
376 static const WCHAR slashHelpW
[] = {'/','?',0};
378 if (argc
< 2 || !lstrcmpW(argvW
[1], slashHelpW
)
379 || !lstrcmpiW(argvW
[1], slashHW
))
381 reg_message(STRING_USAGE
);
385 if (!lstrcmpiW(argvW
[1], addW
))
387 WCHAR
*key_name
, *value_name
= NULL
, *type
= NULL
, separator
= '\0', *data
= NULL
;
388 BOOL value_empty
= FALSE
, force
= FALSE
;
392 reg_message(STRING_INVALID_CMDLINE
);
395 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
396 !lstrcmpiW(argvW
[2], slashHW
)))
398 reg_message(STRING_ADD_USAGE
);
403 for (i
= 1; i
< argc
; i
++)
405 if (!lstrcmpiW(argvW
[i
], slashVW
))
406 value_name
= argvW
[++i
];
407 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
409 else if (!lstrcmpiW(argvW
[i
], slashTW
))
411 else if (!lstrcmpiW(argvW
[i
], slashSW
))
412 separator
= argvW
[++i
][0];
413 else if (!lstrcmpiW(argvW
[i
], slashDW
))
415 else if (!lstrcmpiW(argvW
[i
], slashFW
))
418 return reg_add(key_name
, value_name
, value_empty
, type
, separator
,
421 else if (!lstrcmpiW(argvW
[1], deleteW
))
423 WCHAR
*key_name
, *value_name
= NULL
;
424 BOOL value_empty
= FALSE
, value_all
= FALSE
, force
= FALSE
;
428 reg_message(STRING_INVALID_CMDLINE
);
431 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
432 !lstrcmpiW(argvW
[2], slashHW
)))
434 reg_message(STRING_DELETE_USAGE
);
439 for (i
= 1; i
< argc
; i
++)
441 if (!lstrcmpiW(argvW
[i
], slashVW
))
442 value_name
= argvW
[++i
];
443 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
445 else if (!lstrcmpiW(argvW
[i
], slashVAW
))
447 else if (!lstrcmpiW(argvW
[i
], slashFW
))
450 return reg_delete(key_name
, value_name
, value_empty
, value_all
, force
);
452 else if (!lstrcmpiW(argvW
[1], queryW
))
454 WCHAR
*key_name
, *value_name
= NULL
;
455 BOOL value_empty
= FALSE
, subkey
= FALSE
;
459 reg_message(STRING_INVALID_CMDLINE
);
462 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
463 !lstrcmpiW(argvW
[2], slashHW
)))
465 reg_message(STRING_QUERY_USAGE
);
470 for (i
= 1; i
< argc
; i
++)
472 if (!lstrcmpiW(argvW
[i
], slashVW
))
473 value_name
= argvW
[++i
];
474 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
476 else if (!lstrcmpiW(argvW
[i
], slashSW
))
479 return reg_query(key_name
, value_name
, value_empty
, subkey
);
483 reg_message(STRING_INVALID_CMDLINE
);