mirror of
https://github.com/reactos/reactos.git
synced 2024-08-19 18:02:55 +00:00
![Amine Khaldi](/assets/img/avatar_default.png)
* Create a branch for some evul shell experiments. svn path=/branches/shell-experiments/; revision=61927
259 lines
5.9 KiB
C
259 lines
5.9 KiB
C
/*
|
|
* Assoc.C - assoc internal command.
|
|
*
|
|
*
|
|
* History:
|
|
*
|
|
* 14-Mar-2009 Lee C. Baker
|
|
* - initial implementation
|
|
*
|
|
* 15-Mar-2009 Lee C. Baker
|
|
* - Don't write to (or use) HKEY_CLASSES_ROOT directly
|
|
* - Externalize strings
|
|
*
|
|
* TODO:
|
|
* - PrintAllAssociations might could be optimized to not fetch all registry subkeys under 'Classes', just the ones that start with '.'
|
|
* - Make sure that non-administrator users can list associations, and get appropriate error messages when they don't have sufficient
|
|
* priveleges to perform an operation
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
#ifdef INCLUDE_CMD_ASSOC
|
|
|
|
static INT
|
|
PrintAssociation(LPTSTR extension)
|
|
{
|
|
DWORD return_val;
|
|
HKEY hKey = NULL, hInsideKey = NULL;
|
|
|
|
DWORD fileTypeLength = 0;
|
|
LPTSTR fileType = NULL;
|
|
|
|
return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
return -1;
|
|
}
|
|
|
|
return_val = RegOpenKeyEx(hKey, extension, 0, KEY_READ, &hInsideKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
RegCloseKey(hInsideKey);
|
|
return 0;
|
|
}
|
|
|
|
/* obtain string length */
|
|
return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, NULL, &fileTypeLength);
|
|
|
|
if (return_val == ERROR_FILE_NOT_FOUND) /* no default value, don't display */
|
|
{
|
|
RegCloseKey(hInsideKey);
|
|
RegCloseKey(hKey);
|
|
return 0;
|
|
}
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hInsideKey);
|
|
RegCloseKey(hKey);
|
|
return -2;
|
|
}
|
|
|
|
fileType = cmd_alloc(fileTypeLength * sizeof(TCHAR));
|
|
|
|
/* obtain actual file type */
|
|
return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, (LPBYTE) fileType, &fileTypeLength);
|
|
|
|
RegCloseKey(hInsideKey);
|
|
RegCloseKey(hKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
cmd_free(fileType);
|
|
return -2;
|
|
}
|
|
|
|
if (fileTypeLength != 0) /* if there is a default key, display relevant information */
|
|
{
|
|
ConOutPrintf(_T("%s=%s\n"), extension, fileType);
|
|
}
|
|
|
|
if (fileTypeLength)
|
|
cmd_free(fileType);
|
|
|
|
return 1;
|
|
}
|
|
|
|
static INT
|
|
PrintAllAssociations()
|
|
{
|
|
DWORD return_val = 0;
|
|
HKEY hKey = NULL;
|
|
DWORD numKeys = 0;
|
|
|
|
DWORD extLength = 0;
|
|
LPTSTR extName = NULL;
|
|
DWORD keyCtr = 0;
|
|
|
|
return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
return -1;
|
|
}
|
|
|
|
return_val = RegQueryInfoKey(hKey, NULL, NULL, NULL, &numKeys, &extLength, NULL, NULL, NULL, NULL, NULL, NULL);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
return -2;
|
|
}
|
|
|
|
extName = cmd_alloc(extLength * sizeof(TCHAR));
|
|
|
|
for(keyCtr = 0; keyCtr < numKeys; keyCtr++)
|
|
{
|
|
DWORD buffer_size = extLength;
|
|
return_val = RegEnumKeyEx(hKey, keyCtr, extName, &buffer_size, NULL, NULL, NULL, NULL);
|
|
|
|
if (return_val == ERROR_SUCCESS || return_val == ERROR_MORE_DATA)
|
|
{
|
|
if (*extName == _T('.'))
|
|
PrintAssociation(extName);
|
|
}
|
|
else
|
|
{
|
|
cmd_free(extName);
|
|
RegCloseKey(hKey);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
RegCloseKey(hKey);
|
|
|
|
if (extName)
|
|
cmd_free(extName);
|
|
|
|
return numKeys;
|
|
}
|
|
|
|
static INT
|
|
AddAssociation(LPTSTR extension, LPTSTR type)
|
|
{
|
|
DWORD return_val;
|
|
HKEY hKey = NULL, insideKey = NULL;
|
|
|
|
return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
return -1;
|
|
|
|
return_val = RegCreateKeyEx(hKey, extension, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &insideKey, NULL);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
return -1;
|
|
}
|
|
|
|
return_val = RegSetValueEx(insideKey, NULL, 0, REG_SZ, (LPBYTE)type, (_tcslen(type) + 1) * sizeof(TCHAR));
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(insideKey);
|
|
RegCloseKey(hKey);
|
|
return -2;
|
|
}
|
|
|
|
RegCloseKey(insideKey);
|
|
RegCloseKey(hKey);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static int
|
|
RemoveAssociation(LPTSTR extension)
|
|
{
|
|
DWORD return_val;
|
|
HKEY hKey;
|
|
|
|
return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
return -1;
|
|
|
|
return_val = RegDeleteKey(hKey, extension);
|
|
|
|
if (return_val != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hKey);
|
|
return -2;
|
|
}
|
|
|
|
RegCloseKey(hKey);
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
INT CommandAssoc (LPTSTR param)
|
|
{
|
|
/* print help */
|
|
if (!_tcsncmp (param, _T("/?"), 2))
|
|
{
|
|
ConOutResPaging(TRUE,STRING_ASSOC_HELP);
|
|
return 0;
|
|
}
|
|
|
|
nErrorLevel = 0;
|
|
|
|
if (_tcslen(param) == 0)
|
|
PrintAllAssociations();
|
|
else
|
|
{
|
|
LPTSTR lpEqualSign = _tcschr(param, _T('='));
|
|
if (lpEqualSign != NULL)
|
|
{
|
|
LPTSTR fileType = lpEqualSign + 1;
|
|
LPTSTR extension = cmd_alloc((lpEqualSign - param + 1) * sizeof(TCHAR));
|
|
|
|
_tcsncpy(extension, param, lpEqualSign - param);
|
|
extension[lpEqualSign - param] = (TCHAR)0;
|
|
|
|
if (_tcslen(fileType) == 0)
|
|
/* if the equal sign is the last character
|
|
in the string, then delete the key */
|
|
{
|
|
RemoveAssociation(extension);
|
|
}
|
|
else
|
|
/* otherwise, add the key and print out the association*/
|
|
{
|
|
AddAssociation( extension, fileType);
|
|
PrintAssociation(extension);
|
|
}
|
|
|
|
cmd_free(extension);
|
|
}
|
|
else
|
|
{
|
|
/* no equal sign, print all associations */
|
|
INT retval = PrintAssociation(param);
|
|
|
|
if (retval == 0) /* if nothing printed out */
|
|
ConOutResPrintf(STRING_ASSOC_ERROR, param);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#endif /* INCLUDE_CMD_ASSOC */
|