Dll.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef __WINNT__
    #include <windows.h>
#else
    #include <dcl/_dlfcn.h>
#endif
 
#include <dcl/Object.h>
#include <dcl/Charset.h>
 
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#include <dcl/Files.h>
#include <dcl/Dll.h>
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/Dll.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
IMPLEMENT_CLASSINFO(DllException, SysError)
 
DllException::DllException(const String& _name, Exception* _cause)
: SysError(_cause)
{
    __message = _name;
}
 
DllException::DllException(const String& _name, uint32_t _errorNo)
    : SysError(_errorNo)
{
    __message = _name;
}
 
DllException::DllException(const String& _name, const String& _message)
{
    __message = _name + __T(" ") + _message;
}
 
String DllException::toString() const
{
    StringBuilder r = __message;
    if (SysError::errorNo() != 0)
        r += __T(" ") + SysError::toString();
 
    return r;
}
 
IMPLEMENT_CLASSINFO(Dll, Object)
 
Dll::Dll()
{
    __handle = NULL;
}
 
Dll::~Dll()
{
    if (__handle) {
#ifdef __DCL_DEBUG
        try {
            close();
        }
        catch (DllException* e) {
            __DCL_TRACE1(L"Warning Dll::close fail! %ls\n",
                    e->toString().data());
            e->destroy();
        }
#else
        close();
#endif
    }
}
 
void Dll::open(
    const String& _filename, int _flags /* = DEFAULT */
) __DCL_THROWS1(DllException*)
{
    __DCL_ASSERT(__handle == NULL);
    __DCL_ASSERT(!_filename.isEmpty());
 
#ifdef __WINNT__
    HINSTANCE handle = LoadLibraryExW(
        _filename,
        NULL,
        LOAD_WITH_ALTERED_SEARCH_PATH
        );
    if (handle == NULL) {
        DWORD lastError = GetLastError();
    //  __DCL_TRACE1(L"LoadLibraryEx Failed [%d]\n", lastError);
        throw new DllException(_filename, WINAPI_ERROR(lastError));
    }
 
#else
    if (_filename[(size_t) 0] == L'/') {
        if(!Files::exists(_filename))
            throw new DllException(_filename, ENOENT);
    }
 
    void* handle = __dlopen(_filename, _flags);
    if (handle == NULL) {
        String msg;
        __DCL_VERIFY(__dlerror(msg) == 0);
        throw new DllException(_filename, msg);
    }
 
#endif
    __filename = _filename;
    __handle = (void*)handle;
}
 
void Dll::close() 
    __DCL_THROWS1(DllException*)
{
    if (__handle) {
        void* handle = __handle;
        __handle = NULL;
 
#ifdef __WINNT__
        if (::FreeLibrary((HINSTANCE) handle) == FALSE)
            throw new DllException(__filename, WINAPI_ERROR(GetLastError()));
 
#else
        if (dlclose(handle) != 0) {
            String msg;
            __DCL_VERIFY(__dlerror(msg) == 0);
            throw new DllException(__filename, msg);
        }
 
#endif
    }
}
 
void* Dll::getAddress(const char* _symbol)
    __DCL_THROWS1(DllException*)
{
    __DCL_ASSERT(__handle != NULL);
    __DCL_ASSERT(_symbol != NULL && *_symbol != '\0');
 
    void* r = NULL;
 
#ifdef __WINNT__
    r = (void*) GetProcAddress((HINSTANCE)__handle, _symbol);
    if (r == NULL) {
        String symbol = AsciiDecoder::decode(_symbol);
        throw new DllException(
            __filename + L"::" + symbol,
            WINAPI_ERROR(GetLastError()));
    }
 
#else
    r = dlsym(__handle, _symbol);
    if (r == NULL) {
        String symbol = AsciiDecoder::decode(_symbol);
        String msg;
        __DCL_VERIFY(__dlerror(msg) == 0);
 
        throw new DllException(
            __filename + L"::" + symbol,
            msg
            );
    }
 
#endif
    return r;
}
 
__DCL_END_NAMESPACE