qaac/dl.h at master · nu774/qaac · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#ifndef DL_H
#define DL_H

#include <string>
#include <stdexcept>
#include <memory>
#ifdef _WIN32
#include "platformutil.h"
#else
#include <dlfcn.h>
#endif
#include "strutil.h"

class AutoCast {
#ifdef _WIN32
    FARPROC m_pointer;
public:
    AutoCast(FARPROC p): m_pointer(p) {}
#else
    void *m_pointer;
public:
    AutoCast(void *p): m_pointer(p) {}
#endif
    template <typename U>
    operator U*() { return reinterpret_cast<U*>(m_pointer); }
};

class DL {
#ifdef _WIN32
    std::shared_ptr<HINSTANCE__> m_module;
public:
    DL() {}
    DL(HMODULE handle, bool own=true)
    {
        if (own)
            m_module.reset(handle, FreeLibrary);
        else
            m_module.reset(handle, [](HMODULE){});
    }
    bool load(const std::string &path)
    {
        HMODULE handle = LoadLibraryW(strutil::us2w(path).c_str());
        if (handle) m_module.reset(handle, FreeLibrary);
        return loaded();
    }
    bool loaded() const { return m_module.get() != 0; }
    void reset() { m_module.reset(); }
    AutoCast fetch(const char *name)
    {
        return AutoCast(GetProcAddress(m_module.get(), name));
    }
#else
    std::shared_ptr<void> m_module;
public:
    DL() {}
    DL(void *handle, bool own=true)
    {
        if (own)
            m_module.reset(handle, dlclose);
        else
            m_module.reset(handle, [](void*){});
    }
    bool load(const std::string &path)
    {
        void *handle = dlopen(path.c_str(), RTLD_NOW);
        if (handle) m_module.reset(handle, dlclose);
        return loaded();
    }
    bool loaded() const { return m_module.get() != 0; }
    void reset() { m_module.reset(); }
    AutoCast fetch(const char *name)
    {
        return AutoCast(dlsym(m_module.get(), name));
    }
#endif
};

#endif