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