reactos/base/applications/cmdutils/reg/reg.c

   1 /*
   2  * Copyright 2008 Andrew Riedi
   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 <stdarg.h>
  20 #include <windef.h>
  21 #include <winbase.h>
  22 #include <winuser.h>
  23 #include <winreg.h>
  24 #include <wincon.h>
  25 #include <wine/unicode.h>
  26
  27 #include "reg.h"
  28
  29 static int reg_printfW(const WCHAR *msg, ...)
  30 {
  31     va_list va_args;
  32     int wlen;
  33     DWORD count, ret;
  34     WCHAR msg_buffer[8192];
  35
  36     va_start(va_args, msg);
  37     vsprintfW(msg_buffer, msg, va_args);
  38     va_end(va_args);
  39
  40     wlen = lstrlenW(msg_buffer);
  41     ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), msg_buffer, wlen, &count, NULL);
  42     if (!ret)
  43     {
  44         DWORD len;
  45         char  *msgA;
  46
  47         /* On Windows WriteConsoleW() fails if the output is redirected. So fall
  48          * back to WriteFile(), assuming the console encoding is still the right
  49          * one in that case.
  50          */
  51         len = WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen,
  52             NULL, 0, NULL, NULL);
  53         msgA = HeapAlloc(GetProcessHeap(), 0, len * sizeof(char));
  54         if (!msgA)
  55             return 0;
  56
  57         WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen, msgA, len,
  58             NULL, NULL);
  59         WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE);
  60         HeapFree(GetProcessHeap(), 0, msgA);
  61     }
  62
  63     return count;
  64 }
  65
  66 static int reg_message(int msg)
  67 {
  68     static const WCHAR formatW[] = {'%','s',0};
  69     WCHAR msg_buffer[8192];
  70
  71     LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer,
  72         sizeof(msg_buffer)/sizeof(WCHAR));
  73     return reg_printfW(formatW, msg_buffer);
  74 }
  75
  76 static HKEY get_rootkey(LPWSTR key)
  77 {
  78     static const WCHAR szHKLM[] = {'H','K','L','M',0};
  79     static const WCHAR szHKEY_LOCAL_MACHINE[] = {'H','K','E','Y','_','L','O','C','A','L','_','M','A','C','H','I','N','E',0};
  80     static const WCHAR szHKCU[] = {'H','K','C','U',0};
  81     static const WCHAR szHKEY_CURRENT_USER[] = {'H','K','E','Y','_','C','U','R','R','E','N','T','_','U','S','E','R',0};
  82     static const WCHAR szHKCR[] = {'H','K','C','R',0};
  83     static const WCHAR szHKEY_CLASSES_ROOT[] = {'H','K','E','Y','_','C','L','A','S','S','E','S','_','R','O','O','T',0};
  84     static const WCHAR szHKU[] = {'H','K','U',0};
  85     static const WCHAR szHKEY_USERS[] = {'H','K','E','Y','_','U','S','E','R','S',0};
  86     static const WCHAR szHKCC[] = {'H','K','C','C',0};
  87     static const WCHAR szHKEY_CURRENT_CONFIG[] = {'H','K','E','Y','_','C','U','R','R','E','N','T','_','C','O','N','F','I','G',0};
  88
  89     if (CompareStringW(CP_ACP,NORM_IGNORECASE,key,4,szHKLM,4)==CSTR_EQUAL ||
  90         CompareStringW(CP_ACP,NORM_IGNORECASE,key,18,szHKEY_LOCAL_MACHINE,18)==CSTR_EQUAL)
  91         return HKEY_LOCAL_MACHINE;
  92     else if (CompareStringW(CP_ACP,NORM_IGNORECASE,key,4,szHKCU,4)==CSTR_EQUAL ||
  93              CompareStringW(CP_ACP,NORM_IGNORECASE,key,17,szHKEY_CURRENT_USER,17)==CSTR_EQUAL)
  94         return HKEY_CURRENT_USER;
  95     else if (CompareStringW(CP_ACP,NORM_IGNORECASE,key,4,szHKCR,4)==CSTR_EQUAL ||
  96              CompareStringW(CP_ACP,NORM_IGNORECASE,key,17,szHKEY_CLASSES_ROOT,17)==CSTR_EQUAL)
  97         return HKEY_CLASSES_ROOT;
  98     else if (CompareStringW(CP_ACP,NORM_IGNORECASE,key,3,szHKU,3)==CSTR_EQUAL ||
  99              CompareStringW(CP_ACP,NORM_IGNORECASE,key,10,szHKEY_USERS,10)==CSTR_EQUAL)
 100         return HKEY_USERS;
 101     else if (CompareStringW(CP_ACP,NORM_IGNORECASE,key,4,szHKCC,4)==CSTR_EQUAL ||
 102              CompareStringW(CP_ACP,NORM_IGNORECASE,key,19,szHKEY_CURRENT_CONFIG,19)==CSTR_EQUAL)
 103         return HKEY_CURRENT_CONFIG;
 104     else return NULL;
 105 }
 106
 107 static DWORD get_regtype(LPWSTR type)
 108 {
 109     static const WCHAR szREG_SZ[] = {'R','E','G','_','S','Z',0};
 110     static const WCHAR szREG_MULTI_SZ[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0};
 111     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};
 112     static const WCHAR szREG_DWORD[] = {'R','E','G','_','D','W','O','R','D',0};
 113     static const WCHAR szREG_BINARY[] = {'R','E','G','_','B','I','N','A','R','Y',0};
 114     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};
 115     static const WCHAR szREG_NONE[] = {'R','E','G','_','N','O','N','E',0};
 116     static const WCHAR szREG_EXPAND_SZ[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0};
 117
 118     if (!type)
 119         return REG_SZ;
 120
 121     if (lstrcmpiW(type,szREG_SZ)==0) return REG_SZ;
 122     if (lstrcmpiW(type,szREG_DWORD)==0) return REG_DWORD;
 123     if (lstrcmpiW(type,szREG_MULTI_SZ)==0) return REG_MULTI_SZ;
 124     if (lstrcmpiW(type,szREG_EXPAND_SZ)==0) return REG_EXPAND_SZ;
 125     if (lstrcmpiW(type,szREG_DWORD_BIG_ENDIAN)==0) return REG_DWORD_BIG_ENDIAN;
 126     if (lstrcmpiW(type,szREG_DWORD_LITTLE_ENDIAN)==0) return REG_DWORD_LITTLE_ENDIAN;
 127     if (lstrcmpiW(type,szREG_BINARY)==0) return REG_BINARY;
 128     if (lstrcmpiW(type,szREG_NONE)==0) return REG_NONE;
 129
 130     return -1;
 131 }
 132
 133 static LPBYTE get_regdata(LPWSTR data, DWORD reg_type, WCHAR separator, DWORD *reg_count)
 134 {
 135     LPBYTE out_data = NULL;
 136     *reg_count = 0;
 137
 138     switch (reg_type)
 139     {
 140         case REG_SZ:
 141         {
 142             *reg_count = (lstrlenW(data) + 1) * sizeof(WCHAR);
 143             out_data = HeapAlloc(GetProcessHeap(),0,*reg_count);
 144             lstrcpyW((LPWSTR)out_data,data);
 145             break;
 146         }
 147         case REG_DWORD:
 148         {
 149             LPWSTR rest;
 150             DWORD val;
 151             val = strtolW(data, &rest, 0);
 152             if (rest == data) {
 153                 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};
 154                 reg_printfW(nonnumber);
 155                 break;
 156             }
 157             *reg_count = sizeof(DWORD);
 158             out_data = HeapAlloc(GetProcessHeap(),0,*reg_count);
 159             ((LPDWORD)out_data)[0] = val;
 160             break;
 161         }
 162         default:
 163         {
 164             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};
 165             reg_printfW(unhandled, reg_type,data);
 166         }
 167     }
 168
 169     return out_data;
 170 }
 171
 172 static int reg_add(WCHAR *key_name, WCHAR *value_name, BOOL value_empty,
 173     WCHAR *type, WCHAR separator, WCHAR *data, BOOL force)
 174 {
 175     static const WCHAR stubW[] = {'A','D','D',' ','-',' ','%','s',
 176         ' ','%','s',' ','%','d',' ','%','s',' ','%','s',' ','%','d','\n',0};
 177     LPWSTR p;
 178     HKEY root,subkey;
 179
 180     reg_printfW(stubW, key_name, value_name, value_empty, type, data, force);
 181
 182     if (key_name[0]=='\\' && key_name[1]=='\\')
 183     {
 184         reg_message(STRING_NO_REMOTE);
 185         return 1;
 186     }
 187
 188     p = strchrW(key_name,'\\');
 189     if (!p)
 190     {
 191         reg_message(STRING_INVALID_KEY);
 192         return 1;
 193     }
 194     p++;
 195
 196     root = get_rootkey(key_name);
 197     if (!root)
 198     {
 199         reg_message(STRING_INVALID_KEY);
 200         return 1;
 201     }
 202
 203     if(RegCreateKeyW(root,p,&subkey)!=ERROR_SUCCESS)
 204     {
 205         reg_message(STRING_INVALID_KEY);
 206         return 1;
 207     }
 208
 209     if (value_name || data)
 210     {
 211         DWORD reg_type;
 212         DWORD reg_count = 0;
 213         BYTE* reg_data = NULL;
 214
 215         if (!force)
 216         {
 217             if (RegQueryValueW(subkey,value_name,NULL,NULL)==ERROR_SUCCESS)
 218             {
 219                 /* FIXME:  Prompt for overwrite */
 220             }
 221         }
 222
 223         reg_type = get_regtype(type);
 224         if (reg_type == -1)
 225         {
 226             RegCloseKey(subkey);
 227             reg_message(STRING_INVALID_CMDLINE);
 228             return 1;
 229         }
 230
 231         if (data)
 232             reg_data = get_regdata(data,reg_type,separator,&reg_count);
 233
 234         RegSetValueExW(subkey,value_name,0,reg_type,reg_data,reg_count);
 235         HeapFree(GetProcessHeap(),0,reg_data);
 236     }
 237
 238     RegCloseKey(subkey);
 239     reg_message(STRING_SUCCESS);
 240
 241     return 0;
 242 }
 243
 244 static int reg_delete(WCHAR *key_name, WCHAR *value_name, BOOL value_empty,
 245     BOOL value_all, BOOL force)
 246 {
 247     LPWSTR p;
 248     HKEY root,subkey;
 249
 250     static const WCHAR stubW[] = {'D','E','L','E','T','E',
 251         ' ','-',' ','%','s',' ','%','s',' ','%','d',' ','%','d',' ','%','d','\n'
 252         ,0};
 253     reg_printfW(stubW, key_name, value_name, value_empty, value_all, force);
 254
 255     if (key_name[0]=='\\' && key_name[1]=='\\')
 256     {
 257         reg_message(STRING_NO_REMOTE);
 258         return 1;
 259     }
 260
 261     p = strchrW(key_name,'\\');
 262     if (!p)
 263     {
 264         reg_message(STRING_INVALID_KEY);
 265         return 1;
 266     }
 267     p++;
 268
 269     root = get_rootkey(key_name);
 270     if (!root)
 271     {
 272         reg_message(STRING_INVALID_KEY);
 273         return 1;
 274     }
 275
 276     if (value_name && value_empty)
 277     {
 278         reg_message(STRING_INVALID_CMDLINE);
 279         return 1;
 280     }
 281
 282     if (value_empty && value_all)
 283     {
 284         reg_message(STRING_INVALID_CMDLINE);
 285         return 1;
 286     }
 287
 288     if (!force)
 289     {
 290         /* FIXME:  Prompt for delete */
 291     }
 292
 293     /* Delete subtree only if no /v* option is given */
 294     if (!value_name && !value_empty && !value_all)
 295     {
 296         if (RegDeleteTreeW(root,p)!=ERROR_SUCCESS)
 297         {
 298             reg_message(STRING_CANNOT_FIND);
 299             return 1;
 300         }
 301         reg_message(STRING_SUCCESS);
 302         return 0;
 303     }
 304
 305     if(RegOpenKeyW(root,p,&subkey)!=ERROR_SUCCESS)
 306     {
 307         reg_message(STRING_CANNOT_FIND);
 308         return 1;
 309     }
 310
 311     if (value_all)
 312     {
 313         LPWSTR szValue;
 314         DWORD maxValue;
 315         DWORD count;
 316         LONG rc;
 317
 318         rc = RegQueryInfoKeyW(subkey, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
 319             &maxValue, NULL, NULL, NULL);
 320         if (rc != ERROR_SUCCESS)
 321         {
 322             /* FIXME: failure */
 323             RegCloseKey(subkey);
 324             return 1;
 325         }
 326         maxValue++;
 327         szValue = HeapAlloc(GetProcessHeap(),0,maxValue*sizeof(WCHAR));
 328
 329         while (1)
 330         {
 331             count = maxValue;
 332             rc = RegEnumValueW(subkey, 0, szValue, &count, NULL, NULL, NULL, NULL);
 333             if (rc == ERROR_SUCCESS)
 334             {
 335                 rc = RegDeleteValueW(subkey, szValue);
 336                 if (rc != ERROR_SUCCESS)
 337                     break;
 338             }
 339             else break;
 340         }
 341         if (rc != ERROR_SUCCESS)
 342         {
 343             /* FIXME  delete failed */
 344         }
 345     }
 346     else if (value_name)
 347     {
 348         if (RegDeleteValueW(subkey,value_name) != ERROR_SUCCESS)
 349         {
 350             RegCloseKey(subkey);
 351             reg_message(STRING_CANNOT_FIND);
 352             return 1;
 353         }
 354     }
 355     else if (value_empty)
 356     {
 357         RegSetValueExW(subkey,NULL,0,REG_SZ,NULL,0);
 358     }
 359
 360     RegCloseKey(subkey);
 361     reg_message(STRING_SUCCESS);
 362     return 0;
 363 }
 364
 365 static int reg_query(WCHAR *key_name, WCHAR *value_name, BOOL value_empty,
 366     BOOL subkey)
 367 {
 368     static const WCHAR stubW[] = {'S','T','U','B',' ','Q','U','E','R','Y',' ',
 369         '-',' ','%','s',' ','%','s',' ','%','d',' ','%','d','\n',0};
 370     reg_printfW(stubW, key_name, value_name, value_empty, subkey);
 371
 372     return 1;
 373 }
 374
 375 int wmain(int argc, WCHAR *argvW[])
 376 {
 377     int i;
 378
 379     static const WCHAR addW[] = {'a','d','d',0};
 380     static const WCHAR deleteW[] = {'d','e','l','e','t','e',0};
 381     static const WCHAR queryW[] = {'q','u','e','r','y',0};
 382     static const WCHAR slashDW[] = {'/','d',0};
 383     static const WCHAR slashFW[] = {'/','f',0};
 384     static const WCHAR slashHW[] = {'/','h',0};
 385     static const WCHAR slashSW[] = {'/','s',0};
 386     static const WCHAR slashTW[] = {'/','t',0};
 387     static const WCHAR slashVW[] = {'/','v',0};
 388     static const WCHAR slashVAW[] = {'/','v','a',0};
 389     static const WCHAR slashVEW[] = {'/','v','e',0};
 390     static const WCHAR slashHelpW[] = {'/','?',0};
 391
 392     if (argc < 2 || !lstrcmpW(argvW[1], slashHelpW)
 393                  || !lstrcmpiW(argvW[1], slashHW))
 394     {
 395         reg_message(STRING_USAGE);
 396         return 0;
 397     }
 398
 399     if (!lstrcmpiW(argvW[1], addW))
 400     {
 401         WCHAR *key_name, *value_name = NULL, *type = NULL, separator = '\0', *data = NULL;
 402         BOOL value_empty = FALSE, force = FALSE;
 403
 404         if (argc < 3)
 405         {
 406             reg_message(STRING_INVALID_CMDLINE);
 407             return 1;
 408         }
 409         else if (argc == 3 && (!lstrcmpW(argvW[2], slashHelpW) ||
 410                                !lstrcmpiW(argvW[2], slashHW)))
 411         {
 412             reg_message(STRING_ADD_USAGE);
 413             return 0;
 414         }
 415
 416         key_name = argvW[2];
 417         for (i = 1; i < argc; i++)
 418         {
 419             if (!lstrcmpiW(argvW[i], slashVW))
 420                 value_name = argvW[++i];
 421             else if (!lstrcmpiW(argvW[i], slashVEW))
 422                 value_empty = TRUE;
 423             else if (!lstrcmpiW(argvW[i], slashTW))
 424                 type = argvW[++i];
 425             else if (!lstrcmpiW(argvW[i], slashSW))
 426                 separator = argvW[++i][0];
 427             else if (!lstrcmpiW(argvW[i], slashDW))
 428                 data = argvW[++i];
 429             else if (!lstrcmpiW(argvW[i], slashFW))
 430                 force = TRUE;
 431         }
 432         return reg_add(key_name, value_name, value_empty, type, separator,
 433             data, force);
 434     }
 435     else if (!lstrcmpiW(argvW[1], deleteW))
 436     {
 437         WCHAR *key_name, *value_name = NULL;
 438         BOOL value_empty = FALSE, value_all = FALSE, force = FALSE;
 439
 440         if (argc < 3)
 441         {
 442             reg_message(STRING_INVALID_CMDLINE);
 443             return 1;
 444         }
 445         else if (argc == 3 && (!lstrcmpW(argvW[2], slashHelpW) ||
 446                                !lstrcmpiW(argvW[2], slashHW)))
 447         {
 448             reg_message(STRING_DELETE_USAGE);
 449             return 0;
 450         }
 451
 452         key_name = argvW[2];
 453         for (i = 1; i < argc; i++)
 454         {
 455             if (!lstrcmpiW(argvW[i], slashVW))
 456                 value_name = argvW[++i];
 457             else if (!lstrcmpiW(argvW[i], slashVEW))
 458                 value_empty = TRUE;
 459             else if (!lstrcmpiW(argvW[i], slashVAW))
 460                 value_all = TRUE;
 461             else if (!lstrcmpiW(argvW[i], slashFW))
 462                 force = TRUE;
 463         }
 464         return reg_delete(key_name, value_name, value_empty, value_all, force);
 465     }
 466     else if (!lstrcmpiW(argvW[1], queryW))
 467     {
 468         WCHAR *key_name, *value_name = NULL;
 469         BOOL value_empty = FALSE, subkey = FALSE;
 470
 471         if (argc < 3)
 472         {
 473             reg_message(STRING_INVALID_CMDLINE);
 474             return 1;
 475         }
 476         else if (argc == 3 && (!lstrcmpW(argvW[2], slashHelpW) ||
 477                                !lstrcmpiW(argvW[2], slashHW)))
 478         {
 479             reg_message(STRING_QUERY_USAGE);
 480             return 0;
 481         }
 482
 483         key_name = argvW[2];
 484         for (i = 1; i < argc; i++)
 485         {
 486             if (!lstrcmpiW(argvW[i], slashVW))
 487                 value_name = argvW[++i];
 488             else if (!lstrcmpiW(argvW[i], slashVEW))
 489                 value_empty = TRUE;
 490             else if (!lstrcmpiW(argvW[i], slashSW))
 491                 subkey = TRUE;
 492         }
 493         return reg_query(key_name, value_name, value_empty, subkey);
 494     }
 495     else
 496     {
 497         reg_message(STRING_INVALID_CMDLINE);
 498         return 1;
 499     }
 500 }