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