programs/reg/delete.c

   1 /*
   2  * Copyright 2016-2017, 2021 Hugh McMaster
   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 "reg.h"
  20
  21 static BOOL op_delete_key = TRUE;
  22
  23 static void output_error(LONG rc)
  24 {
  25     if (rc == ERROR_FILE_NOT_FOUND)
  26     {
  27         if (op_delete_key)
  28             output_message(STRING_KEY_NONEXIST);
  29         else
  30             output_message(STRING_VALUE_NONEXIST);
  31     }
  32     else
  33     {
  34         output_message(STRING_ACCESS_DENIED);
  35     }
  36 }
  37
  38 static int run_delete(HKEY root, WCHAR *path, WCHAR *key_name, WCHAR *value_name,
  39                       BOOL value_empty, BOOL value_all, BOOL force)
  40 {
  41     LONG rc;
  42     HKEY hkey;
  43
  44     if (!force)
  45     {
  46         BOOL ret;
  47
  48         if (value_name || value_empty)
  49             ret = ask_confirm(STRING_DELETE_VALUE, value_name);
  50         else if (value_all)
  51             ret = ask_confirm(STRING_DELETE_VALUEALL, key_name);
  52         else
  53             ret = ask_confirm(STRING_DELETE_SUBKEY, key_name);
  54
  55         if (!ret)
  56         {
  57             output_message(STRING_CANCELLED);
  58             return 0;
  59         }
  60     }
  61
  62     /* Delete registry key if no /v* option is given */
  63     if (!value_name && !value_empty && !value_all)
  64     {
  65         if ((rc = RegDeleteTreeW(root, path)))
  66         {
  67             output_error(rc);
  68             return 1;
  69         }
  70
  71         output_message(STRING_SUCCESS);
  72         return 0;
  73     }
  74
  75     if ((rc = RegOpenKeyExW(root, path, 0, KEY_READ|KEY_SET_VALUE, &hkey)))
  76     {
  77         output_error(rc);
  78         return 1;
  79     }
  80
  81     op_delete_key = FALSE;
  82
  83     if (value_all)
  84     {
  85         DWORD max_value_len = 256, value_len;
  86         WCHAR *value_name;
  87
  88         value_name = malloc(max_value_len * sizeof(WCHAR));
  89
  90         while (1)
  91         {
  92             value_len = max_value_len;
  93             rc = RegEnumValueW(hkey, 0, value_name, &value_len, NULL, NULL, NULL, NULL);
  94             if (rc == ERROR_SUCCESS)
  95             {
  96                 rc = RegDeleteValueW(hkey, value_name);
  97                 if (rc != ERROR_SUCCESS)
  98                 {
  99                     free(value_name);
 100                     RegCloseKey(hkey);
 101                     output_message(STRING_VALUEALL_FAILED, key_name);
 102                     output_error(rc);
 103                     return 1;
 104                 }
 105             }
 106             else if (rc == ERROR_MORE_DATA)
 107             {
 108                 max_value_len *= 2;
 109                 value_name = realloc(value_name, max_value_len * sizeof(WCHAR));
 110             }
 111             else break;
 112         }
 113         free(value_name);
 114     }
 115     else if (value_name || value_empty)
 116     {
 117         if ((rc = RegDeleteValueW(hkey, value_name)))
 118         {
 119             RegCloseKey(hkey);
 120             output_error(rc);
 121             return 1;
 122         }
 123     }
 124
 125     RegCloseKey(hkey);
 126     output_message(STRING_SUCCESS);
 127     return 0;
 128 }
 129
 130 int reg_delete(int argc, WCHAR *argvW[])
 131 {
 132     HKEY root;
 133     WCHAR *path, *key_name, *value_name = NULL;
 134     BOOL value_all = FALSE, value_empty = FALSE, force = FALSE;
 135     int i;
 136
 137     if (!parse_registry_key(argvW[2], &root, &path))
 138         return 1;
 139
 140     for (i = 3; i < argc; i++)
 141     {
 142         WCHAR *str;
 143
 144         if (argvW[i][0] != '/' && argvW[i][0] != '-')
 145             goto invalid;
 146
 147         str = &argvW[i][1];
 148
 149         if (!lstrcmpiW(str, L"va"))
 150         {
 151             if (value_all) goto invalid;
 152             value_all = TRUE;
 153             continue;
 154         }
 155         else if (!lstrcmpiW(str, L"ve"))
 156         {
 157             if (value_empty) goto invalid;
 158             value_empty = TRUE;
 159             continue;
 160         }
 161         else if (!lstrcmpiW(str, L"reg:32") || !lstrcmpiW(str, L"reg:64"))
 162             continue;
 163         else if (!str[0] || str[1])
 164             goto invalid;
 165
 166         switch (towlower(*str))
 167         {
 168         case 'v':
 169             if (value_name || !(value_name = argvW[++i]))
 170                 goto invalid;
 171             break;
 172         case 'f':
 173             if (force) goto invalid;
 174             force = TRUE;
 175             break;
 176         default:
 177             goto invalid;
 178         }
 179     }
 180
 181     if ((value_name && value_empty) || (value_name && value_all) || (value_empty && value_all))
 182         goto invalid;
 183
 184     key_name = get_long_key(root, path);
 185
 186     return run_delete(root, path, key_name, value_name, value_empty, value_all, force);
 187
 188 invalid:
 189     output_message(STRING_INVALID_SYNTAX);
 190     output_message(STRING_FUNC_HELP, wcsupr(argvW[1]));
 191     return 1;
 192 }