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
26 #include <wine/unicode.h>
30 static int reg_printfW(const WCHAR
*msg
, ...)
35 WCHAR msg_buffer
[8192];
37 va_start(va_args
, msg
);
38 vsnprintfW(msg_buffer
, 8192, msg
, va_args
);
41 wlen
= lstrlenW(msg_buffer
);
42 ret
= WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE
), msg_buffer
, wlen
, &count
, NULL
);
48 /* On Windows WriteConsoleW() fails if the output is redirected. So fall
49 * back to WriteFile(), assuming the console encoding is still the right
52 len
= WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer
, wlen
,
54 msgA
= HeapAlloc(GetProcessHeap(), 0, len
* sizeof(char));
58 WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer
, wlen
, msgA
, len
,
60 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE
), msgA
, len
, &count
, FALSE
);
61 HeapFree(GetProcessHeap(), 0, msgA
);
67 static int reg_message(int msg
)
69 static const WCHAR formatW
[] = {'%','s',0};
70 WCHAR msg_buffer
[8192];
72 LoadStringW(GetModuleHandleW(NULL
), msg
, msg_buffer
,
73 sizeof(msg_buffer
)/sizeof(WCHAR
));
74 return reg_printfW(formatW
, msg_buffer
);
77 static HKEY
get_rootkey(LPWSTR key
)
79 static const WCHAR szHKLM
[] = {'H','K','L','M',0};
80 static const WCHAR szHKEY_LOCAL_MACHINE
[] = {'H','K','E','Y','_','L','O','C','A','L','_','M','A','C','H','I','N','E',0};
81 static const WCHAR szHKCU
[] = {'H','K','C','U',0};
82 static const WCHAR szHKEY_CURRENT_USER
[] = {'H','K','E','Y','_','C','U','R','R','E','N','T','_','U','S','E','R',0};
83 static const WCHAR szHKCR
[] = {'H','K','C','R',0};
84 static const WCHAR szHKEY_CLASSES_ROOT
[] = {'H','K','E','Y','_','C','L','A','S','S','E','S','_','R','O','O','T',0};
85 static const WCHAR szHKU
[] = {'H','K','U',0};
86 static const WCHAR szHKEY_USERS
[] = {'H','K','E','Y','_','U','S','E','R','S',0};
87 static const WCHAR szHKCC
[] = {'H','K','C','C',0};
88 static const WCHAR szHKEY_CURRENT_CONFIG
[] = {'H','K','E','Y','_','C','U','R','R','E','N','T','_','C','O','N','F','I','G',0};
90 if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKLM
,4)==CSTR_EQUAL
||
91 CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,18,szHKEY_LOCAL_MACHINE
,18)==CSTR_EQUAL
)
92 return HKEY_LOCAL_MACHINE
;
93 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCU
,4)==CSTR_EQUAL
||
94 CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,17,szHKEY_CURRENT_USER
,17)==CSTR_EQUAL
)
95 return HKEY_CURRENT_USER
;
96 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCR
,4)==CSTR_EQUAL
||
97 CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,17,szHKEY_CLASSES_ROOT
,17)==CSTR_EQUAL
)
98 return HKEY_CLASSES_ROOT
;
99 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,3,szHKU
,3)==CSTR_EQUAL
||
100 CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,10,szHKEY_USERS
,10)==CSTR_EQUAL
)
102 else if (CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,4,szHKCC
,4)==CSTR_EQUAL
||
103 CompareStringW(CP_ACP
,NORM_IGNORECASE
,key
,19,szHKEY_CURRENT_CONFIG
,19)==CSTR_EQUAL
)
104 return HKEY_CURRENT_CONFIG
;
108 static DWORD
get_regtype(LPWSTR type
)
110 static const WCHAR szREG_SZ
[] = {'R','E','G','_','S','Z',0};
111 static const WCHAR szREG_MULTI_SZ
[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0};
112 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};
113 static const WCHAR szREG_DWORD
[] = {'R','E','G','_','D','W','O','R','D',0};
114 static const WCHAR szREG_BINARY
[] = {'R','E','G','_','B','I','N','A','R','Y',0};
115 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};
116 static const WCHAR szREG_NONE
[] = {'R','E','G','_','N','O','N','E',0};
117 static const WCHAR szREG_EXPAND_SZ
[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0};
122 if (lstrcmpiW(type
,szREG_SZ
)==0) return REG_SZ
;
123 if (lstrcmpiW(type
,szREG_DWORD
)==0) return REG_DWORD
;
124 if (lstrcmpiW(type
,szREG_MULTI_SZ
)==0) return REG_MULTI_SZ
;
125 if (lstrcmpiW(type
,szREG_EXPAND_SZ
)==0) return REG_EXPAND_SZ
;
126 if (lstrcmpiW(type
,szREG_DWORD_BIG_ENDIAN
)==0) return REG_DWORD_BIG_ENDIAN
;
127 if (lstrcmpiW(type
,szREG_DWORD_LITTLE_ENDIAN
)==0) return REG_DWORD_LITTLE_ENDIAN
;
128 if (lstrcmpiW(type
,szREG_BINARY
)==0) return REG_BINARY
;
129 if (lstrcmpiW(type
,szREG_NONE
)==0) return REG_NONE
;
134 static LPBYTE
get_regdata(LPWSTR data
, DWORD reg_type
, WCHAR separator
, DWORD
*reg_count
)
136 LPBYTE out_data
= NULL
;
143 *reg_count
= (lstrlenW(data
) + 1) * sizeof(WCHAR
);
144 out_data
= HeapAlloc(GetProcessHeap(),0,*reg_count
);
145 lstrcpyW((LPWSTR
)out_data
,data
);
152 val
= strtolW(data
, &rest
, 0);
154 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};
155 reg_printfW(nonnumber
);
158 *reg_count
= sizeof(DWORD
);
159 out_data
= HeapAlloc(GetProcessHeap(),0,*reg_count
);
160 ((LPDWORD
)out_data
)[0] = val
;
165 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};
166 reg_printfW(unhandled
, reg_type
,data
);
173 static int reg_add(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
174 WCHAR
*type
, WCHAR separator
, WCHAR
*data
, BOOL force
)
176 static const WCHAR stubW
[] = {'A','D','D',' ','-',' ','%','s',
177 ' ','%','s',' ','%','d',' ','%','s',' ','%','s',' ','%','d','\n',0};
181 reg_printfW(stubW
, key_name
, value_name
, value_empty
, type
, data
, force
);
183 if (key_name
[0]=='\\' && key_name
[1]=='\\')
185 reg_message(STRING_NO_REMOTE
);
189 p
= strchrW(key_name
,'\\');
192 reg_message(STRING_INVALID_KEY
);
197 root
= get_rootkey(key_name
);
200 reg_message(STRING_INVALID_KEY
);
204 if(RegCreateKeyW(root
,p
,&subkey
)!=ERROR_SUCCESS
)
206 reg_message(STRING_INVALID_KEY
);
210 if (value_name
|| data
)
214 BYTE
* reg_data
= NULL
;
218 if (RegQueryValueW(subkey
,value_name
,NULL
,NULL
)==ERROR_SUCCESS
)
220 /* FIXME: Prompt for overwrite */
224 reg_type
= get_regtype(type
);
228 reg_message(STRING_INVALID_CMDLINE
);
233 reg_data
= get_regdata(data
,reg_type
,separator
,®_count
);
235 RegSetValueExW(subkey
,value_name
,0,reg_type
,reg_data
,reg_count
);
236 HeapFree(GetProcessHeap(),0,reg_data
);
240 reg_message(STRING_SUCCESS
);
245 static int reg_delete(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
246 BOOL value_all
, BOOL force
)
251 static const WCHAR stubW
[] = {'D','E','L','E','T','E',
252 ' ','-',' ','%','s',' ','%','s',' ','%','d',' ','%','d',' ','%','d','\n'
254 reg_printfW(stubW
, key_name
, value_name
, value_empty
, value_all
, force
);
256 if (key_name
[0]=='\\' && key_name
[1]=='\\')
258 reg_message(STRING_NO_REMOTE
);
262 p
= strchrW(key_name
,'\\');
265 reg_message(STRING_INVALID_KEY
);
270 root
= get_rootkey(key_name
);
273 reg_message(STRING_INVALID_KEY
);
277 if (value_name
&& value_empty
)
279 reg_message(STRING_INVALID_CMDLINE
);
283 if (value_empty
&& value_all
)
285 reg_message(STRING_INVALID_CMDLINE
);
291 /* FIXME: Prompt for delete */
294 /* Delete subtree only if no /v* option is given */
295 if (!value_name
&& !value_empty
&& !value_all
)
297 if (SHDeleteKeyW(root
, p
) != ERROR_SUCCESS
)
299 reg_message(STRING_CANNOT_FIND
);
302 reg_message(STRING_SUCCESS
);
306 if(RegOpenKeyW(root
,p
,&subkey
)!=ERROR_SUCCESS
)
308 reg_message(STRING_CANNOT_FIND
);
319 rc
= RegQueryInfoKeyW(subkey
, NULL
, NULL
, NULL
, NULL
, NULL
, NULL
, NULL
,
320 &maxValue
, NULL
, NULL
, NULL
);
321 if (rc
!= ERROR_SUCCESS
)
328 szValue
= HeapAlloc(GetProcessHeap(),0,maxValue
*sizeof(WCHAR
));
333 rc
= RegEnumValueW(subkey
, 0, szValue
, &count
, NULL
, NULL
, NULL
, NULL
);
334 if (rc
== ERROR_SUCCESS
)
336 rc
= RegDeleteValueW(subkey
, szValue
);
337 if (rc
!= ERROR_SUCCESS
)
342 if (rc
!= ERROR_SUCCESS
)
344 /* FIXME delete failed */
349 if (RegDeleteValueW(subkey
,value_name
) != ERROR_SUCCESS
)
352 reg_message(STRING_CANNOT_FIND
);
356 else if (value_empty
)
358 RegSetValueExW(subkey
,NULL
,0,REG_SZ
,NULL
,0);
362 reg_message(STRING_SUCCESS
);
366 static int reg_query(WCHAR
*key_name
, WCHAR
*value_name
, BOOL value_empty
,
369 static const WCHAR stubW
[] = {'S','T','U','B',' ','Q','U','E','R','Y',' ',
370 '-',' ','%','s',' ','%','s',' ','%','d',' ','%','d','\n',0};
371 reg_printfW(stubW
, key_name
, value_name
, value_empty
, subkey
);
376 int wmain(int argc
, WCHAR
*argvW
[])
380 static const WCHAR addW
[] = {'a','d','d',0};
381 static const WCHAR deleteW
[] = {'d','e','l','e','t','e',0};
382 static const WCHAR queryW
[] = {'q','u','e','r','y',0};
383 static const WCHAR slashDW
[] = {'/','d',0};
384 static const WCHAR slashFW
[] = {'/','f',0};
385 static const WCHAR slashHW
[] = {'/','h',0};
386 static const WCHAR slashSW
[] = {'/','s',0};
387 static const WCHAR slashTW
[] = {'/','t',0};
388 static const WCHAR slashVW
[] = {'/','v',0};
389 static const WCHAR slashVAW
[] = {'/','v','a',0};
390 static const WCHAR slashVEW
[] = {'/','v','e',0};
391 static const WCHAR slashHelpW
[] = {'/','?',0};
393 if (argc
< 2 || !lstrcmpW(argvW
[1], slashHelpW
)
394 || !lstrcmpiW(argvW
[1], slashHW
))
396 reg_message(STRING_USAGE
);
400 if (!lstrcmpiW(argvW
[1], addW
))
402 WCHAR
*key_name
, *value_name
= NULL
, *type
= NULL
, separator
= '\0', *data
= NULL
;
403 BOOL value_empty
= FALSE
, force
= FALSE
;
407 reg_message(STRING_INVALID_CMDLINE
);
410 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
411 !lstrcmpiW(argvW
[2], slashHW
)))
413 reg_message(STRING_ADD_USAGE
);
418 for (i
= 1; i
< argc
; i
++)
420 if (!lstrcmpiW(argvW
[i
], slashVW
))
421 value_name
= argvW
[++i
];
422 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
424 else if (!lstrcmpiW(argvW
[i
], slashTW
))
426 else if (!lstrcmpiW(argvW
[i
], slashSW
))
427 separator
= argvW
[++i
][0];
428 else if (!lstrcmpiW(argvW
[i
], slashDW
))
430 else if (!lstrcmpiW(argvW
[i
], slashFW
))
433 return reg_add(key_name
, value_name
, value_empty
, type
, separator
,
436 else if (!lstrcmpiW(argvW
[1], deleteW
))
438 WCHAR
*key_name
, *value_name
= NULL
;
439 BOOL value_empty
= FALSE
, value_all
= FALSE
, force
= FALSE
;
443 reg_message(STRING_INVALID_CMDLINE
);
446 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
447 !lstrcmpiW(argvW
[2], slashHW
)))
449 reg_message(STRING_DELETE_USAGE
);
454 for (i
= 1; i
< argc
; i
++)
456 if (!lstrcmpiW(argvW
[i
], slashVW
))
457 value_name
= argvW
[++i
];
458 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
460 else if (!lstrcmpiW(argvW
[i
], slashVAW
))
462 else if (!lstrcmpiW(argvW
[i
], slashFW
))
465 return reg_delete(key_name
, value_name
, value_empty
, value_all
, force
);
467 else if (!lstrcmpiW(argvW
[1], queryW
))
469 WCHAR
*key_name
, *value_name
= NULL
;
470 BOOL value_empty
= FALSE
, subkey
= FALSE
;
474 reg_message(STRING_INVALID_CMDLINE
);
477 else if (argc
== 3 && (!lstrcmpW(argvW
[2], slashHelpW
) ||
478 !lstrcmpiW(argvW
[2], slashHW
)))
480 reg_message(STRING_QUERY_USAGE
);
485 for (i
= 1; i
< argc
; i
++)
487 if (!lstrcmpiW(argvW
[i
], slashVW
))
488 value_name
= argvW
[++i
];
489 else if (!lstrcmpiW(argvW
[i
], slashVEW
))
491 else if (!lstrcmpiW(argvW
[i
], slashSW
))
494 return reg_query(key_name
, value_name
, value_empty
, subkey
);
498 reg_message(STRING_INVALID_CMDLINE
);