Plugin symbol resolving fix

[babeltrace.git] / plugins / plugin.c

/*
 * plugin.c
 *
 * Babeltrace Plugin
 * 
 * Copyright 2015 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <babeltrace/compiler.h>
#include <babeltrace/ref.h>
#include <babeltrace/plugin/plugin-internal.h>
#include <glib.h>

#define PLUGIN_SYMBOL_NAME	"__bt_plugin_name"
#define PLUGIN_SYMBOL_AUTHOR	"__bt_plugin_author"
#define PLUGIN_SYMBOL_LICENSE	"__bt_plugin_license"
#define PLUGIN_SYMBOL_INIT	"__bt_plugin_init"
#define PLUGIN_SYMBOL_EXIT	"__bt_plugin_exit"

static
void bt_plugin_destroy(struct bt_object *obj)
{
	struct bt_plugin *plugin;

	assert(obj);
	plugin = container_of(obj, struct bt_plugin, base);

	if (plugin->module) {
		if (!g_module_close(plugin->module)) {
				printf_error("Module close error: %s",
					g_module_error());

		}
	}
	g_free(plugin);
}

BT_HIDDEN
struct bt_plugin *bt_plugin_create(GModule *module)
{
	struct bt_plugin *plugin = NULL;
	gpointer symbol = NULL;

	if (!module) {
		goto error;
	}

	plugin = g_new0(struct bt_plugin, 1);
	if (!plugin) {
		goto error;
	}

	bt_object_init(plugin, bt_plugin_destroy);
	if (!g_module_symbol(module, PLUGIN_SYMBOL_NAME,
			(gpointer *) &plugin->name)) {
		printf_error("Unable to resolve plugin symbol %s from %s",
				PLUGIN_SYMBOL_NAME, g_module_name(module));
		goto error;
	}

	if (!g_module_symbol(module, PLUGIN_SYMBOL_LICENSE,
			(gpointer *) &plugin->license)) {
		printf_error("Unable to resolve plugin symbol %s from %s",
				PLUGIN_SYMBOL_LICENSE, g_module_name(module));
		goto error;
	}
	if (!g_module_symbol(module, PLUGIN_SYMBOL_INIT, &symbol)) {
		printf_error("Unable to resolve plugin symbol %s from %s",
				PLUGIN_SYMBOL_INIT, g_module_name(module));
		goto error;
	} else {
		plugin->init = *((bt_plugin_init_func *) symbol);
		if (!plugin->init) {
			printf_error("NULL %s symbol target",
					PLUGIN_SYMBOL_INIT);
			goto error;
		}
	}

	/* Optional */
	if (g_module_symbol(module, PLUGIN_SYMBOL_EXIT,
			(gpointer *) &symbol)) {
		plugin->exit = *((bt_plugin_exit_func *) symbol);
	}
	g_module_symbol(module, PLUGIN_SYMBOL_AUTHOR,
			(gpointer *) &plugin->author);

	return plugin;
error:
        BT_PUT(plugin);
	return plugin;
}

BT_HIDDEN
enum bt_component_status bt_plugin_register_component_classes(
		struct bt_plugin *plugin, struct bt_component_factory *factory)
{
	assert(plugin && factory);
	return plugin->init(factory);
}