aboutsummaryrefslogtreecommitdiff
path: root/src/dynamic.cpp
blob: 20db9bac862b54c578f15da64c237b1a608678fd (about) (plain) (blame)
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
121
122
123
124
125
126
127
128
/*
 * InspIRCd -- Internet Relay Chat Daemon
 *
 *   Copyright (C) 2020 Matt Schatz <genius3000@g3k.solutions>
 *   Copyright (C) 2017-2020 Sadie Powell <sadie@witchery.services>
 *   Copyright (C) 2014 Attila Molnar <attilamolnar@hush.com>
 *   Copyright (C) 2012 Robby <robby@chatbelgie.be>
 *   Copyright (C) 2012 ChrisTX <xpipe@hotmail.de>
 *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
 *   Copyright (C) 2009 Uli Schlachter <psychon@inspircd.org>
 *   Copyright (C) 2007 Robin Burchell <robin+git@viroteck.net>
 *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
 *   Copyright (C) 2003, 2006, 2010 Craig Edwards <brain@inspircd.org>
 *
 * This file is part of InspIRCd.  InspIRCd 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, version 2.
 *
 * 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 "inspircd.h"
#include "dynamic.h"

#ifndef _WIN32
# include <dlfcn.h>
#endif

DLLManager::DLLManager(const std::string& name)
	: libname(name)
{
	const static size_t extlen = strlen(DLL_EXTENSION);
	if (name.length() <= extlen || name.compare(name.length() - extlen, name.length(), DLL_EXTENSION))
	{
		err.assign(name + " is not a module (no " DLL_EXTENSION " extension)");
		return;
	}

#ifdef _WIN32
	lib = LoadLibraryA(name.c_str());
#else
	lib = dlopen(name.c_str(), RTLD_NOW|RTLD_LOCAL);
#endif

	if (!lib)
		RetrieveLastError();
}

DLLManager::~DLLManager()
{
	if (!lib)
		return;

#ifdef _WIN32
	FreeLibrary(lib);
#else
	dlclose(lib);
#endif
}

Module* DLLManager::CallInit()
{
	if (!lib)
		return nullptr;

	const unsigned long* abi = GetSymbol<const unsigned long>(MODULE_STR_ABI);
	if (!abi)
	{
		err.assign(libname + " is not a module (no ABI symbol)");
		return nullptr;
	}
	else if (*abi != MODULE_ABI)
	{
		const char* version = GetVersion();
		err.assign(InspIRCd::Format("%s was built against %s (%lu) which is too %s to use with %s (%lu).",
			libname.c_str(), version ? version : "an unknown version", *abi,
			*abi < MODULE_ABI ? "old" : "new", INSPIRCD_VERSION, MODULE_ABI));
		return nullptr;
	}

	union
	{
		void* vptr;
		Module* (*fptr)();
	};

	vptr = GetSymbol(MODULE_STR_INIT);
	if (!vptr)
	{
		err.assign(libname + " is not a module (no init symbol)");
		return nullptr;
	}

	return (*fptr)();
}

void* DLLManager::GetSymbol(const char* name) const
{
	if (!lib)
		return nullptr;

#if defined _WIN32
	return GetProcAddress(lib, name);
#else
	return dlsym(lib, name);
#endif
}

void DLLManager::RetrieveLastError()
{
#ifdef _WIN32
	err = GetErrorMessage(GetLastError());
	SetLastError(ERROR_SUCCESS);
#else
	const char* errmsg = dlerror();
	err = errmsg ? errmsg : "Unknown error";
#endif

	for (size_t pos = 0; ((pos = err.find_first_of("\r\n", pos)) != std::string::npos); )
		err[pos] = ' ';
}