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