You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

163 lines
5.4 KiB

/*==LICENSE==*
CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011 Cyan Worlds, Inc.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
You can contact Cyan Worlds, Inc. by email legal@cyan.com
or by snail mail at:
Cyan Worlds, Inc.
14617 N Newport Hwy
Mead, WA 99021
*==LICENSE==*/
//////////////////////////////////////////////////////////////////////////////
//
// plWinRegistryTools
// Utility class for doing various usefull things in Win32
// Written by Mathew Burrack
// 4.23.2002
//
//////////////////////////////////////////////////////////////////////////////
#include "hsTypes.h"
#include "plWinRegistryTools.h"
#include "hsWindows.h"
//////////////////////////////////////////////////////////////////////////////
//// Static Utility Functions ////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//// ISetRegKey //////////////////////////////////////////////////////////////
// Sets the given registry key to the given string value. If valueName = nil,
// sets the (default) value
static hsBool ISetRegKey( const char *keyName, const char *value, const char *valueName = nil )
{
HKEY regKey;
DWORD result;
// Create the key (just opens if it already exists)
if( ::RegCreateKeyEx( HKEY_CLASSES_ROOT, keyName, 0, nil, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS,
nil, &regKey, &result ) != ERROR_SUCCESS )
{
hsStatusMessageF( "Warning: Registry database open failed for key '%s'.\n", keyName );
return false;
}
// Assign the "default" subkey value
LONG lResult = ::RegSetValueEx( regKey, valueName, 0, REG_SZ, (const BYTE *)value, ( lstrlen( value ) + 1 ) * sizeof( TCHAR ) );
if( ::RegCloseKey( regKey ) == ERROR_SUCCESS && lResult == ERROR_SUCCESS )
return true;
hsStatusMessageF( "Warning: Registry database update failed for key '%s'.\n", keyName );
return false;
}
//////////////////////////////////////////////////////////////////////////////
//// Public Utility Functions ////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//// AssociateFileType ///////////////////////////////////////////////////////
// Associates a given file type in the Win32 registry with the given
// application. Also assigns a default icon if iconIndex != -1
//
// To do this, we create a set of keys in the registry under CLASSES_ROOT that
// looks like this:
// fileTypeID (value = fileTypeName)
// |
// |--- DefaultIcon (value = path,index) [omit this one if you don't
// | want a default icon]
// |
// |--- shell
// |
// |--- open
// |
// |--- command (value = command line)
//
hsBool plWinRegistryTools::AssociateFileType( const char *fileTypeID, const char *fileTypeName, const char *appPath, int iconIndex )
{
char keyName[ 512 ], keyValue[ 512 ];
// Root key
if( !ISetRegKey( fileTypeID, fileTypeName ) )
return false;
// DefaultIcon key, if we want one
if( iconIndex != -1 )
{
sprintf( keyName, "%s\\DefaultIcon", fileTypeID );
sprintf( keyValue, "%s,%d", appPath, iconIndex );
if( !ISetRegKey( keyName, keyValue ) )
return false;
}
// shell/open/command key
sprintf( keyName, "%s\\shell\\open\\command", fileTypeID );
sprintf( keyValue, "\"%s\" \"%%1\"", appPath );
if( !ISetRegKey( keyName, keyValue ) )
return false;
// Success!
return true;
}
//// AssociateFileExtension //////////////////////////////////////////////////
// Assigns a given file extension to a previously registered Win32 file type
// (using the above function)
//
// We do this by creating a key entry under CLASSES_ROOT of the following
// structure:
//
// fileExtension (value = fileTypeID)
//
// where fileExtension includes the leading . and fileTypeID is the same
// typeID registered with the above function
hsBool plWinRegistryTools::AssociateFileExtension( const char *fileExtension, const char *fileTypeID )
{
return ISetRegKey( fileExtension, fileTypeID );
}
//// GetCurrentFileExtensionAssociation //////////////////////////////////////
// Obtains the current fileTypeID associated with the given file extension,
// or a null string if it isn't yet associated.
hsBool plWinRegistryTools::GetCurrentFileExtensionAssociation( const char *extension, char *buffer, int bufferLen )
{
long dataLen;
buffer[ 0 ] = 0;
dataLen = bufferLen;
LONG retVal = ::RegQueryValue( HKEY_CLASSES_ROOT, extension, buffer, &dataLen );
if( retVal != ERROR_SUCCESS )
{
char msg[ 512 ];
FormatMessage( FORMAT_MESSAGE_FROM_SYSTEM, 0, retVal, 0, msg, sizeof( msg ), nil );
hsStatusMessageF( "Error querying registry key '%s' : %s\n", extension, msg );
return false;
}
return true;
}