quakeforge/libs/util/plugin.c

408 lines
9.3 KiB
C
Raw Normal View History

/*
plugin.c
QuakeForge plugin API functions
Copyright (C) 2001 Jeff Teunissen <deek@quakeforge.net>
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 2
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, write to:
Free Software Foundation, Inc.
59 Temple Place - Suite 330
Boston, MA 02111-1307, USA
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
static __attribute__ ((unused)) const char rcsid[] =
"$Id$";
#ifdef HAVE_STDIO_H
# include <stdio.h>
#endif
#ifdef HAVE_STRING_H
# include <string.h>
#endif
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
#ifdef HAVE_WINDOWS_H
# include "winquake.h"
#endif
#ifdef HAVE_DLFCN_H
# include <dlfcn.h>
#endif
#ifndef RTLD_LAZY
# ifdef DL_LAZY
# define RTLD_LAZY DL_LAZY
# else
# define RTLD_LAZY 0
# endif
#endif
2001-05-31 03:41:35 +00:00
#include "QF/cvar.h"
2001-10-10 16:22:41 +00:00
#include "QF/hash.h"
#include "QF/plugin.h"
#include "QF/sys.h"
#include "QF/cmd.h"
#include "compat.h"
// loaded_plugins is for plugins loaded from scripts only - not system
// plugins
typedef struct loaded_plugin_s {
char *name;
plugin_t *plugin;
} loaded_plugin_t;
2001-10-10 16:22:41 +00:00
cvar_t *fs_pluginpath;
hashtab_t *registered_plugins, *loaded_plugins;
2001-10-10 16:22:41 +00:00
static const char *pi_error = "";
2001-10-10 16:22:41 +00:00
static const char *
plugin_get_key (void *pl, void *unused)
2001-10-10 16:22:41 +00:00
{
return ((plugin_list_t *) pl)->name;
2001-10-10 16:22:41 +00:00
}
static const char *
loaded_plugin_get_key (void *lp, void *unusued)
{
return ((loaded_plugin_t *) lp)->name;
}
static void
loaded_plugin_delete (void *lp, void *unused)
{
free (((loaded_plugin_t *) lp)->name);
free ((loaded_plugin_t *) lp);
}
static int
pi_close_lib (void *handle)
{
#if defined(HAVE_DLOPEN)
return (dlclose (handle) == 0);
#elif defined (_WIN32)
return (FreeLibrary (handle) == 0);
#endif
}
static void *
pi_get_symbol (void *handle, const char *name)
{
#if defined(HAVE_DLOPEN)
return dlsym (handle, name);
#elif defined (_WIN32)
return GetProcAddress (handle, name);
#endif
}
static void *
pi_open_lib (const char *name, int global_syms)
{
void *dlhand;
2003-06-05 17:51:49 +00:00
#ifndef _WIN32
# ifdef __OpenBSD__
2003-06-05 16:08:10 +00:00
int flags = RTLD_LAZY;
2003-06-05 17:51:49 +00:00
# else
int flags = RTLD_NOW;
2003-06-05 17:51:49 +00:00
# endif
2003-06-05 16:08:10 +00:00
#endif
#if defined(HAVE_DLOPEN)
# if defined(RTLD_GLOBAL)
if (global_syms)
flags |= RTLD_GLOBAL;
# endif
if (!(dlhand = dlopen (name, flags))) {
pi_error = dlerror ();
return 0;
}
#elif defined (_WIN32)
if (!(dlhand = LoadLibrary (name))) { // lib not found
pi_error = "LoadLibrary failed";
return 0;
}
#endif
pi_error = "";
return dlhand;
}
static void
pi_realname (char *realname, int size, const char *type, const char *name)
{
#if defined(HAVE_DLOPEN)
const char *format = "%s/%s_%s.so";
#elif defined(_WIN32)
const char *format = "%s/QF%s_%s.dll";
#else
# error "No shared library support. FIXME"
#endif
snprintf (realname, size, format, fs_pluginpath->string, type, name);
}
static void
pi_info_name (char *info_name, int size, const char *type, const char *name)
{
if (type && name)
snprintf (info_name, size, "%s_%s_PluginInfo", type, name);
else if (type)
snprintf (info_name, size, "%s_PluginInfo", type);
else
snprintf (info_name, size, "PluginInfo");
}
static void
PI_InitCvars (void)
{
fs_pluginpath = Cvar_Get ("fs_pluginpath", FS_PLUGINPATH, CVAR_ROM, NULL,
"Location of your plugin directory");
}
static void
PI_Plugin_Load_f (void)
{
plugin_t *pi;
const char *type, *name;
if (Cmd_Argc() != 3) {
Sys_Printf ("Usage: plugin_load <type> <name>\n");
return;
}
type = Cmd_Argv(1);
name = Cmd_Argv(2);
pi = PI_LoadPlugin (type, name);
if (!pi)
Sys_Printf ("Error loading plugin %s %s\n", type, name);
else if (pi->functions && pi->functions->general &&
pi->functions->general->p_Init)
pi->functions->general->p_Init ();
}
static void
PI_Plugin_Unload_f (void)
{
char plugin_name[1024];
loaded_plugin_t *lp;
plugin_t *pi;
if (Cmd_Argc() != 3) {
Sys_Printf ("Usage: plugin_unload <type> <name>\n");
return;
}
// try to locate the plugin
snprintf (plugin_name, sizeof (plugin_name), "%s_%s", Cmd_Argv(1),
Cmd_Argv(2));
lp = Hash_Find (loaded_plugins, plugin_name);
if (lp) {
pi = lp->plugin;
} else {
Sys_Printf ("Plugin %s not loaded\n", plugin_name);
return;
}
PI_UnloadPlugin (pi);
}
/*
PI_Init
Eventually this function will likely initialize libltdl. It doesn't, yet,
since we aren't using libltdl yet.
*/
void
PI_Init (void)
{
PI_InitCvars ();
registered_plugins = Hash_NewTable (253, plugin_get_key, 0, 0);
loaded_plugins = Hash_NewTable(253, loaded_plugin_get_key,
loaded_plugin_delete, 0);
Cmd_AddCommand("plugin_load", PI_Plugin_Load_f,
"load the plugin of the given type name and name");
Cmd_AddCommand("plugin_unload", PI_Plugin_Unload_f,
"unload the plugin of the given type name and name");
}
void
PI_Shutdown (void)
{
void **elems, **cur;
// unload all "loaded" plugins and free the hash
elems = Hash_GetList (loaded_plugins);
for (cur = elems; *cur; ++cur)
PI_UnloadPlugin (((loaded_plugin_t *) *cur)->plugin);
free (elems);
Hash_DelTable (loaded_plugins);
}
plugin_t *
2001-07-15 07:04:17 +00:00
PI_LoadPlugin (const char *type, const char *name)
{
char realname[4096];
2001-10-10 16:22:41 +00:00
char plugin_name[1024];
char plugin_info_name[1024];
char *tmpname;
void *dlhand = NULL;
plugin_t *plugin = NULL;
P_PluginInfo plugin_info = NULL;
plugin_list_t *pl;
loaded_plugin_t *lp;
if (!name)
return NULL;
tmpname = strrchr (name, '/'); // Get the base name, don't allow paths
2001-10-10 16:22:41 +00:00
// Build the plugin name
snprintf (plugin_name, sizeof (plugin_name), "%s_%s", type, name);
// make sure we're not already loaded
lp = Hash_Find (loaded_plugins, plugin_name);
if (lp) {
Sys_Printf ("Plugin %s already loaded\n", plugin_name);
return NULL;
}
2001-10-10 16:22:41 +00:00
pl = Hash_Find (registered_plugins, plugin_name);
if (pl) {
plugin_info = pl->info;
}
if (!plugin_info) {
// Build the path to the file to load
pi_realname (realname, sizeof (realname), type,
2001-10-10 16:22:41 +00:00
(tmpname ? tmpname + 1 : name));
if (!(dlhand = pi_open_lib (realname, 0))) {
2001-10-10 16:22:41 +00:00
// lib not found
Sys_Printf ("Could not load plugin \"%s\".\n", realname);
Sys_DPrintf ("Reason: \"%s\".\n", pi_error);
2001-10-10 16:22:41 +00:00
return NULL;
}
// Build the plugin info name as $type_$name_PluginInfo
pi_info_name (plugin_info_name, sizeof (plugin_info_name), type, name);
if (!(plugin_info = pi_get_symbol (dlhand, plugin_info_name))) {
// Build the plugin info name as $type_PluginInfo
pi_info_name (plugin_info_name, sizeof (plugin_info_name),
type, 0);
if (!(plugin_info = pi_get_symbol (dlhand, plugin_info_name))) {
// Build the plugin info name as PluginInfo
pi_info_name (plugin_info_name, sizeof (plugin_info_name),
0, 0);
if (!(plugin_info = pi_get_symbol (dlhand, plugin_info_name))) {
// info function not found
pi_close_lib (dlhand);
Sys_Printf ("Plugin info function not found\n");
return NULL;
}
}
2001-10-10 16:22:41 +00:00
}
// get the plugin data structure
if (!(plugin = plugin_info ())) {
pi_close_lib (dlhand);
Sys_Printf ("Something went badly wrong.\n");
return NULL;
}
#if defined(HAVE_DLOPEN) && defined(RTLD_GLOBAL)
// check if it wants its syms to be global
if (plugin->data->general->flag & PIF_GLOBAL) {
// do the whole thing over again with global syms
pi_close_lib (dlhand);
// try to reopen
if (!(dlhand = pi_open_lib (realname, 1))) {
Sys_Printf ("Error reopening plugin \"%s\".\n", realname);
Sys_DPrintf ("Reason: \"%s\".\n", pi_error);
return NULL;
}
// get the plugin_info func pointer
if (!(plugin_info = pi_get_symbol (dlhand, plugin_info_name))) {
pi_close_lib (dlhand);
Sys_Printf ("Plugin info function missing on reload\n");
return NULL;
}
// get the plugin data structure
if (!(plugin = plugin_info ())) {
pi_close_lib (dlhand);
Sys_Printf ("Something went badly wrong on module reload\n");
return NULL;
}
}
#endif
} else if (!(plugin = plugin_info ())) { // Something went badly wrong
pi_close_lib (dlhand);
Sys_Printf ("Something went badly wrong.\n");
return NULL;
}
// add the plugin to the loaded_plugins hashtable
lp = malloc (sizeof (loaded_plugin_t));
lp->name = strdup (plugin_name);
lp->plugin = plugin;
Hash_Add (loaded_plugins, lp);
plugin->full_name = lp->name;
plugin->handle = dlhand;
return plugin;
}
qboolean
PI_UnloadPlugin (plugin_t *plugin)
{
if (plugin
&& plugin->functions
&& plugin->functions->general
&& plugin->functions->general->p_Shutdown) {
plugin->functions->general->p_Shutdown ();
} else {
Sys_DPrintf ("Warning: No shutdown function for type %d plugin!\n",
plugin->type);
}
// remove from the table of loaded plugins
Hash_Free (loaded_plugins, Hash_Del (loaded_plugins, plugin->full_name));
2001-10-10 16:22:41 +00:00
if (!plugin->handle) // we didn't load it
return true;
return pi_close_lib (plugin->handle);
}
2001-10-10 16:22:41 +00:00
void
PI_RegisterPlugins (plugin_list_t *plugins)
{
while (plugins->name)
Hash_Add (registered_plugins, plugins++);
}