registry.cpp

#include "registry.h"
#include <windows.h>

using namespace irr;
using namespace irr::core;

// we want the logger, declared somewhere else
// again, this is because we cannot inlucde "main.h" here
extern ILogger* logger;


// constructor
Registry::Registry()
{
        logger->log( "Registry ctor()" );
}

// destructor
Registry::~Registry()
{
        logger->log( "Registry dtor()" );
}


// get value (type void*)
ERR_TYPE Registry::queryValue( stringc key, stringc name, void* outBuffer, u32 bufferSize )
{
        LONG result = 0;
        HKEY hKey;

        result = RegOpenKeyEx(  HKEY_LOCAL_MACHINE,
                                                        TEXT(key.c_str()),
                                                        0,
                                                        KEY_READ,
                                                        &hKey );

        if ( ERROR_SUCCESS != result )
        {
                stringc err("queryKey() cannot open handle to key: ");
                err += key;
                err += " in HKEY_LOCAL_MACHINE. Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                return ERR_REGISTRY_OPEN_KEY_FAILED;
        }

        result = RegQueryValueEx(       hKey,
                                                                TEXT(name.c_str()),
                                                                NULL,
                                                                NULL,
                                                                (LPBYTE) outBuffer,
                                                                (LPDWORD) &bufferSize );
        if ( ERROR_FILE_NOT_FOUND == result )
        {
                stringc err("queryKey() name not found: ");
                err += name;
                err += " in HKEY_LOCAL_MACHINE\\";
                err += key;
                logger->log( err.c_str() );
                RegCloseKey( hKey );
                return ERR_REGISTRY_NAME_NOT_FOUND;
        }
        else if ( ERROR_SUCCESS != result )
        {
                stringc err("queryKey() cannot query name: ");
                err += name;
                err += " in HKEY_LOCAL_MACHINE\\";
                err += key;
                err += ". Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                RegCloseKey( hKey );
                return ERR_REGISTRY_NAME_NOT_FOUND;
        }

        RegCloseKey( hKey );
        return ERR_OK;
}


// get value (type REG_SZ)
ERR_TYPE Registry::queryValue( stringc key, stringc name, stringc* outValue )
{
        ERR_TYPE result = ERR_OK;
        stringc valueString("");
        c8 data[512];
        u32 dataSize = sizeof( data );

        result = this->queryValue( key, name, &data, dataSize );
        if ( result == ERR_OK )
        {
                outValue->append( data );
        }

        return result;
}


// get value (type REG_DWORD)
ERR_TYPE Registry::queryValue( stringc key, stringc name, u32* outValue )
{
        ERR_TYPE result = ERR_OK;
        u32 data = 0;
        u32 dataSize = sizeof( data );

        result = this->queryValue( key, name, &data, dataSize );
        if ( result == ERR_OK )
        {
                *outValue = data;
        }

        return result;
}


ERR_TYPE Registry::addValue( stringc key, stringc name, stringc value )
{
        LONG result = 0;
        HKEY hKey;
        DWORD dwDisp;

        result = RegCreateKeyEx(        HKEY_LOCAL_MACHINE,
                                                                (LPCTSTR) key.c_str(),
                                                                0,
                                                                NULL,
                                                                REG_OPTION_VOLATILE,    // only store in memory, do not write to disk
                                                                KEY_WRITE,
                                                                NULL,
                                                                &hKey,
                                                                &dwDisp );

        if ( ERROR_SUCCESS != result )
        {
                stringc err("addValue() cannot create key: ");
                err += key;
                err += " in HKEY_LOCAL_MACHINE. Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                return ERR_REGISTRY_CREATE_KEY_FAILED;
        }

        result = RegSetValueEx( hKey,
                                                        (LPCTSTR) name.c_str(),
                                                        0,
                                                        REG_SZ,
                                                        (LPBYTE) value.c_str(),
                                                        value.size() + 1 );                     // must include terminating null character
        if ( ERROR_SUCCESS != result )
        {
                stringc err("addValue() cannot create value: ");
                err += value;
                err += " with name ";
                err += name;
                err += " at key ";
                err += key;
                err += " in HKEY_LOCAL_MACHINE. Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                RegCloseKey( hKey );
                return ERR_REGISTRY_CREATE_KEY_FAILED;
        }

        RegCloseKey( hKey );
        return ERR_OK;
}


ERR_TYPE Registry::addValue( stringc key, stringc name, u32 value )
{
        LONG result = 0;
        HKEY hKey;
        DWORD dwDisp;

        result = RegCreateKeyEx(        HKEY_LOCAL_MACHINE,
                                                                (LPCTSTR) key.c_str(),
                                                                0,
                                                                NULL,
                                                                REG_OPTION_VOLATILE,    // only store in memory, do not write to disk
                                                                KEY_WRITE,
                                                                NULL,
                                                                &hKey,
                                                                &dwDisp );

        if ( ERROR_SUCCESS != result )
        {
                stringc err("addValue() cannot create key: ");
                err += key;
                err += " in HKEY_LOCAL_MACHINE. Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                return ERR_REGISTRY_CREATE_KEY_FAILED;
        }

        result = RegSetValueEx( hKey,
                                                        (LPCTSTR) name.c_str(),
                                                        0,
                                                        REG_DWORD,
                                                        (LPBYTE) &value,
                                                        sizeof( u32 ) );
        if ( ERROR_SUCCESS != result )
        {
                stringc err("addValue() cannot create value: ");
                err += value;
                err += " with name ";
                err += name;
                err += " at key ";
                err += key;
                err += " in HKEY_LOCAL_MACHINE. Error (see Winerror.h) is: ";
                err += result;
                logger->log( err.c_str() );
                RegCloseKey( hKey );
                return ERR_REGISTRY_CREATE_KEY_FAILED;
        }

        RegCloseKey( hKey );
        return ERR_OK;
}


ERR_TYPE Registry::removeKey( stringc key, stringc name )
{
        return ERR_OK;
}


ERR_TYPE Registry::getProcessPath( stringc* outPath )
{
        LONG result = 0;
        c8 buffer[1024];
        u32 bufferSize = sizeof( buffer );
        
        memset( buffer, 0, bufferSize );

        result = GetModuleFileName( NULL, buffer, bufferSize );
        if ( 0 == result )
        {
                stringc err("getProcessPath() cannot get module file name.");
                logger->log( err.c_str() );
                return ERR_REGISTRY_GET_EXE_PATH_FAILED;
        }
        
        // strip off exe filename
        stringc temp( buffer );
        s32 last = temp.findLast( (c8) 92 );    // 92 = '\'
        if ( -1 == last )
        {
                stringc err("getProcessPath() cannot get module file name");
                logger->log( err.c_str() );
                return ERR_REGISTRY_GET_EXE_PATH_FAILED;
        }
        
        buffer[last] = 0;

        // copy to caller
        outPath->append( buffer );

        return ERR_OK;
}

---