reactos/base/shell/cmd/assoc.c

   1 /*
   2  *  Assoc.C - assoc internal command.
   3  *
   4  *
   5  *  History:
   6  *
   7  * 14-Mar-2009 Lee C. Baker
   8  * - initial implementation
   9  *
  10  * 15-Mar-2009 Lee C. Baker
  11  * - Don't write to (or use) HKEY_CLASSES_ROOT directly
  12  * - Externalize strings
  13  *
  14  * TODO:
  15  * - PrintAllAssociations might could be optimized to not fetch all registry subkeys under 'Classes', just the ones that start with '.'
  16  * - Make sure that non-administrator users can list associations, and get appropriate error messages when they don't have sufficient
  17  *   priveleges to perform an operation
  18  */
  19
  20 #include "precomp.h"
  21
  22 #ifdef INCLUDE_CMD_ASSOC
  23
  24 static INT
  25 PrintAssociation(LPTSTR extension)
  26 {
  27     DWORD return_val;
  28     HKEY hKey = NULL, hInsideKey = NULL;
  29
  30     DWORD fileTypeLength = 0;
  31     LPTSTR fileType = NULL;
  32
  33     return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
  34
  35     if (return_val != ERROR_SUCCESS)
  36     {
  37         RegCloseKey(hKey);
  38         return -1;
  39     }
  40
  41     return_val = RegOpenKeyEx(hKey, extension, 0, KEY_READ, &hInsideKey);
  42
  43     if (return_val != ERROR_SUCCESS)
  44     {
  45         RegCloseKey(hKey);
  46         RegCloseKey(hInsideKey);
  47         return 0;
  48     }
  49
  50     /* obtain string length */
  51     return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, NULL, &fileTypeLength);
  52
  53     if (return_val == ERROR_FILE_NOT_FOUND)     /* no default value, don't display */
  54     {
  55         RegCloseKey(hInsideKey);
  56         RegCloseKey(hKey);
  57         return 0;
  58     }
  59
  60     if (return_val != ERROR_SUCCESS)
  61     {
  62         RegCloseKey(hInsideKey);
  63         RegCloseKey(hKey);
  64         return -2;
  65     }
  66
  67     fileType = cmd_alloc(fileTypeLength * sizeof(TCHAR));
  68
  69     /* obtain actual file type */
  70     return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, (LPBYTE) fileType, &fileTypeLength);
  71
  72     RegCloseKey(hInsideKey);
  73     RegCloseKey(hKey);
  74
  75     if (return_val != ERROR_SUCCESS)
  76     {
  77         cmd_free(fileType);
  78         return -2;
  79     }
  80
  81     if (fileTypeLength != 0)    /* if there is a default key, display relevant information */
  82     {
  83         ConOutPrintf(_T("%s=%s\n"), extension, fileType);
  84     }
  85
  86     if (fileTypeLength)
  87         cmd_free(fileType);
  88
  89     return 1;
  90 }
  91
  92 static INT
  93 PrintAllAssociations()
  94 {
  95     DWORD return_val = 0;
  96     HKEY hKey = NULL;
  97     DWORD numKeys = 0;
  98
  99     DWORD extLength = 0;
 100     LPTSTR extName = NULL;
 101     DWORD keyCtr = 0;
 102
 103     return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
 104
 105     if (return_val != ERROR_SUCCESS)
 106     {
 107         RegCloseKey(hKey);
 108         return -1;
 109     }
 110
 111     return_val = RegQueryInfoKey(hKey, NULL, NULL, NULL, &numKeys, &extLength, NULL, NULL, NULL, NULL, NULL, NULL);
 112
 113     if (return_val != ERROR_SUCCESS)
 114     {
 115         RegCloseKey(hKey);
 116         return -2;
 117     }
 118
 119     extLength++;
 120     extName = cmd_alloc(extLength * sizeof(TCHAR));
 121
 122     for(keyCtr = 0; keyCtr < numKeys; keyCtr++)
 123     {
 124         DWORD buffer_size = extLength;
 125         return_val = RegEnumKeyEx(hKey, keyCtr, extName, &buffer_size, NULL, NULL, NULL, NULL);
 126
 127         if (return_val == ERROR_SUCCESS || return_val == ERROR_MORE_DATA)
 128         {
 129             if (*extName == _T('.'))
 130                 PrintAssociation(extName);
 131         }
 132         else
 133         {
 134             cmd_free(extName);
 135             RegCloseKey(hKey);
 136             return -1;
 137         }
 138     }
 139
 140     RegCloseKey(hKey);
 141
 142     if (extName)
 143         cmd_free(extName);
 144
 145     return numKeys;
 146 }
 147
 148 static INT
 149 AddAssociation(LPTSTR extension, LPTSTR type)
 150 {
 151     DWORD return_val;
 152     HKEY hKey = NULL, insideKey = NULL;
 153
 154     return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
 155
 156     if (return_val != ERROR_SUCCESS)
 157         return -1;
 158
 159     return_val = RegCreateKeyEx(hKey, extension, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &insideKey, NULL);
 160
 161     if (return_val != ERROR_SUCCESS)
 162     {
 163         RegCloseKey(hKey);
 164         return -1;
 165     }
 166
 167     return_val = RegSetValueEx(insideKey, NULL, 0, REG_SZ, (LPBYTE)type, (_tcslen(type) + 1) * sizeof(TCHAR));
 168
 169     if (return_val != ERROR_SUCCESS)
 170     {
 171         RegCloseKey(insideKey);
 172         RegCloseKey(hKey);
 173         return -2;
 174     }
 175
 176     RegCloseKey(insideKey);
 177     RegCloseKey(hKey);
 178     return 0;
 179 }
 180
 181
 182 static int
 183 RemoveAssociation(LPTSTR extension)
 184 {
 185     DWORD return_val;
 186     HKEY hKey;
 187
 188     return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
 189
 190     if (return_val != ERROR_SUCCESS)
 191         return -1;
 192
 193     return_val = RegDeleteKey(hKey, extension);
 194
 195     if (return_val != ERROR_SUCCESS)
 196     {
 197         RegCloseKey(hKey);
 198         return -2;
 199     }
 200
 201     RegCloseKey(hKey);
 202     return 0;
 203 }
 204
 205
 206
 207 INT CommandAssoc (LPTSTR param)
 208 {
 209     /* print help */
 210     if (!_tcsncmp (param, _T("/?"), 2))
 211     {
 212         ConOutResPaging(TRUE,STRING_ASSOC_HELP);
 213         return 0;
 214     }
 215
 216     nErrorLevel = 0;
 217
 218     if (_tcslen(param) == 0)
 219         PrintAllAssociations();
 220     else
 221     {
 222         LPTSTR lpEqualSign = _tcschr(param, _T('='));
 223         if (lpEqualSign != NULL)
 224         {
 225             LPTSTR fileType = lpEqualSign + 1;
 226             LPTSTR extension = cmd_alloc((lpEqualSign - param + 1) * sizeof(TCHAR));
 227
 228             _tcsncpy(extension, param, lpEqualSign - param);
 229             extension[lpEqualSign - param] = (TCHAR)0;
 230
 231             if (_tcslen(fileType) == 0)
 232             /* if the equal sign is the last character
 233             in the string, then delete the key */
 234             {
 235                 RemoveAssociation(extension);
 236             }
 237             else
 238             /* otherwise, add the key and print out the association*/
 239             {
 240                 AddAssociation( extension, fileType);
 241                 PrintAssociation(extension);
 242             }
 243
 244             cmd_free(extension);
 245         }
 246         else
 247         {
 248             /* no equal sign, print all associations */
 249             INT retval = PrintAssociation(param);
 250
 251             if (retval == 0)    /* if nothing printed out */
 252                 ConOutResPrintf(STRING_ASSOC_ERROR, param);
 253         }
 254     }
 255
 256     return 0;
 257 }
 258
 259 #endif /* INCLUDE_CMD_ASSOC */