summaryrefslogtreecommitdiffstats
path: root/src/Common/ModuleManager.cpp
blob: 74c424dd4f4a59bd67f7276ed50f6bcccba6e599 (plain)
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
 * ModuleManager.cpp
 *
 * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de>
 *
 * 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/>.
 */


#include <config.h>

#include "ModuleManager.h"
#include "Application.h"

#include <Core/ConfigEntry.h>
#include <Core/ConfigManager.h>

#include <dlfcn.h>


namespace Mad {
namespace Common {

ModuleManager::ModuleManager(Application *application0) : application(application0) {
  application->getConfigManager()->registerConfigurable(this);
}

ModuleManager::~ModuleManager() {
  application->getConfigManager()->unregisterConfigurable(this);

  boost::shared_lock<boost::shared_mutex> lock(mutex);
  while(!moduleOrder.empty()) {
    std::string mod = moduleOrder.top();

    lock.unlock();
    unloadModule(mod);
    lock.lock();

    moduleOrder.pop();
  }
}

bool ModuleManager::handleConfigEntry(const Core::ConfigEntry &entry, bool handled) {
  if(handled)
    return false;

  if(entry[0].getKey().matches("LoadModule")) {
    if(!loadModule(entry[0][0].c_str()))
      application->logf(Core::LoggerBase::ERROR, "Can't load module '%s'.", entry[0][0].c_str());

    return true;
  }

  return false;
}

bool ModuleManager::loadModule(const std::string &name) {
  {
    boost::shared_lock<boost::shared_mutex> lock(mutex);

    if(modules.find(name) != modules.end())
      return true;
  }

  ModuleHandle handle = dlopen((name + MODULE_SUFFIX).c_str(), RTLD_NOW | RTLD_GLOBAL);

  if(!handle) {
    application->log(Core::LoggerBase::VERBOSE, std::string("loadModule: Can't open module: ") + std::string(dlerror()));
    return false;
  }

  ModuleLoadFunc loader;
  *(void**)&loader = dlsym(handle, (name + "_create").c_str());

  if(!loader) {
    application->log(Core::LoggerBase::VERBOSE, "loadModule: Can't open module: Invalid module");
    return false;
  }

  Module *mod = loader(application);

  if(!mod) {
    application->log(Core::LoggerBase::VERBOSE, "loadModule: Can't open module: Internal module error");
    return false;
  }

  modules.insert(std::make_pair(name, std::make_pair(handle, mod))).first;
  moduleOrder.push(name);

  return true;
}

void ModuleManager::unloadModule(const std::string &name) {
  boost::upgrade_lock<boost::shared_mutex> lock(mutex);

  std::map<std::string, std::pair<ModuleHandle, Module*> >::iterator mod = modules.find(name);
  if(mod == modules.end())
    return;

  boost::upgrade_to_unique_lock<boost::shared_mutex> upgradeLock(lock);

  delete mod->second.second;
  dlclose(mod->second.first);

  modules.erase(mod);
}

}
}