view module.c @ 38:f69408c0441a

Modified FOREACH function for more security
author David Demelier <markand@malikania.fr>
date Sun, 02 Oct 2011 10:53:48 +0200
parents 23a3ebcbf08e
children
line wrap: on
line source

/*
 * module.c -- portable functions to manipulate dynamic libraries
 *
 * Copyright (c) 2011, David Demelier <markand@malikania.fr>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <stdio.h>
#include <stdlib.h>

#include "module.h"

struct module *
module_load(const char *path, int flags)
{
	struct module *mod;

	if ((mod = malloc(sizeof (struct module))) == NULL)
		return NULL;

	/* Flags are not supported on Windows */
#if defined(_WIN32) || defined(__WIN32__)
	(void) flags;
	mod->handler = LoadLibrary(path);
#else
	mod->handler = dlopen(path, flags);
#endif

	if (!(mod->handler)) {
		module_free(mod);
		return NULL;
	}

	return mod;
}

int
module_find(struct module *mod, const char *sym)
{
#if defined(_WIN32) || defined(__WIN32__)
	mod->sym = GetProcAddress(mod->handler, sym);
#else
	mod->sym = dlsym(mod->handler, sym);
#endif

	if (!(mod->sym))
		return -1;

	return 0;
}

void
module_free(struct module *mod)
{
#if defined(_WIN32) || defined(__WIN32__)
	FreeLibrary(mod->handler);
#else
	dlclose(mod->handler);
#endif

	free(mod);
}