mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-07 04:57:35 +00:00
773b9b85b3
This fixes a leak when there's more than one plugin.
379 lines
8.6 KiB
C
379 lines
8.6 KiB
C
/* radare - LGPL - Copyright 2008-2015 - pancake */
|
|
|
|
#include "r_types.h"
|
|
#include "r_util.h"
|
|
#include "r_lib.h"
|
|
#include <stdio.h>
|
|
#include <dirent.h>
|
|
|
|
R_LIB_VERSION(r_lib);
|
|
|
|
/* TODO: support for nested plugins ?? here */
|
|
|
|
#if __UNIX__
|
|
#include <dlfcn.h>
|
|
#define DLOPEN(x) dlopen(x, RTLD_GLOBAL | RTLD_NOW)
|
|
#define DLSYM(x,y) dlsym(x,y)
|
|
#define DLCLOSE(x) dlclose(x)
|
|
#elif __WINDOWS__
|
|
#include <windows.h>
|
|
#define DLOPEN(x) LoadLibrary(x)
|
|
#define DLSYM(x,y) GetProcAddress(x,y)
|
|
#define DLCLOSE(x) 0//(x)
|
|
//CloseLibrary(x)
|
|
#else
|
|
#define DLOPEN(x) NULL
|
|
#define DLSYM(x,y) NULL
|
|
#define DLCLOSE(x) NULL
|
|
#endif
|
|
|
|
/* XXX : this must be registered in runtime */
|
|
static const char *r_lib_types[] = {
|
|
"io", "dbg", "lang", "asm", "anal", "parse", "bin", //"bininfo",
|
|
"bp", "syscall", "fastcall", "crypto", "cmd", "egg", NULL
|
|
};
|
|
|
|
static int __has_debug = 0;
|
|
|
|
#define IFDBG if(__has_debug)
|
|
|
|
/* XXX: Rename this helper function */
|
|
R_API const char *r_lib_types_get(int idx) {
|
|
if (idx < 0 || idx > R_LIB_TYPE_LAST-1)
|
|
return "unk";
|
|
return r_lib_types[idx];
|
|
}
|
|
|
|
R_API int r_lib_types_get_i(const char *str) {
|
|
int i;
|
|
for (i=0; r_lib_types[i]; i++) {
|
|
if (!strcmp (str, r_lib_types[i]))
|
|
return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
R_API void *r_lib_dl_open(const char *libname) {
|
|
void *ret;
|
|
if (!libname || !*libname)
|
|
return NULL;
|
|
ret = DLOPEN (libname);
|
|
if (__has_debug && ret == NULL)
|
|
#if __UNIX__
|
|
eprintf ("dlerror(%s): %s\n", libname, dlerror ());
|
|
#else
|
|
eprintf ("r_lib_dl_open: Cannot open '%s'\n", libname);
|
|
#endif
|
|
return ret;
|
|
}
|
|
|
|
R_API void *r_lib_dl_sym(void *handler, const char *name) {
|
|
return DLSYM (handler, name);
|
|
}
|
|
|
|
R_API int r_lib_dl_close(void *handler) {
|
|
return DLCLOSE (handler);
|
|
}
|
|
|
|
/* ---- */
|
|
|
|
R_API char *r_lib_path(const char *libname) {
|
|
char *next, *path0, libpath[1024];
|
|
#if __APPLE__
|
|
char *env = r_sys_getenv ("DYLD_LIBRARY_PATH");
|
|
const char *ext = ".dylib";
|
|
env = r_str_concat (env, ":/lib:/usr/lib:/usr/local/lib");
|
|
#elif __UNIX__
|
|
char *env = r_sys_getenv ("LD_LIBRARY_PATH");
|
|
const char *ext = ".so";
|
|
env = r_str_concat (env, ":/lib:/usr/lib:/usr/local/lib");
|
|
#else
|
|
char *env = strdup (".:../../../../../../../windows/system32");
|
|
const char *ext = ".dll";
|
|
#endif
|
|
if (!env) env = strdup (".");
|
|
path0 = env;
|
|
do {
|
|
next = strchr (path0, ':');
|
|
if (next) *next = 0;
|
|
snprintf (libpath, sizeof (libpath), "%s/%s%s", path0, libname, ext);
|
|
//eprintf ("--> %s\n", libpath);
|
|
if (r_file_exists (libpath)) {
|
|
free (env);
|
|
return strdup (libpath);
|
|
}
|
|
path0 = next+1;
|
|
} while (next);
|
|
free (env);
|
|
return NULL;
|
|
}
|
|
|
|
R_API RLib *r_lib_new(const char *symname) {
|
|
RLib *lib = R_NEW (RLib);
|
|
char *env_debug;
|
|
if (lib) {
|
|
env_debug = r_sys_getenv ("R_DEBUG");
|
|
__has_debug = env_debug ? R_TRUE : R_FALSE;
|
|
if (env_debug) {
|
|
free (env_debug);
|
|
}
|
|
lib->handlers = r_list_newf (free);
|
|
lib->plugins = r_list_newf (free);
|
|
strncpy (lib->symname, symname, sizeof (lib->symname)-1);
|
|
}
|
|
return lib;
|
|
}
|
|
|
|
R_API RLib *r_lib_free(RLib *lib) {
|
|
if (!lib) return NULL;
|
|
r_lib_close (lib, NULL);
|
|
r_list_free (lib->handlers);
|
|
r_list_free (lib->plugins);
|
|
free (lib);
|
|
return NULL;
|
|
}
|
|
|
|
/* THIS IS WRONG */
|
|
R_API int r_lib_dl_check_filename(const char *file) {
|
|
if (strstr (file, "."R_LIB_EXT))
|
|
return R_TRUE;
|
|
return R_FALSE;
|
|
}
|
|
|
|
/* high level api */
|
|
|
|
R_API int r_lib_run_handler(RLib *lib, RLibPlugin *plugin, RLibStruct *symbol) {
|
|
RLibHandler *h = plugin->handler;
|
|
if (h && h->constructor) {
|
|
IFDBG eprintf ("PLUGIN HANDLER %p %p\n", h, h->constructor);
|
|
return h->constructor (plugin, h->user, symbol->data);
|
|
} else IFDBG eprintf ("Cannot find plugin constructor\n");
|
|
return R_FAIL;
|
|
}
|
|
|
|
R_API RLibHandler *r_lib_get_handler(RLib *lib, int type) {
|
|
RLibHandler *h;
|
|
RListIter *iter;
|
|
r_list_foreach (lib->handlers, iter, h) {
|
|
if (h->type == type)
|
|
return h;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
R_API R_API int r_lib_close(RLib *lib, const char *file) {
|
|
RLibPlugin *p;
|
|
RListIter *iter, *iter_tmp;
|
|
r_list_foreach_safe (lib->plugins, iter, iter_tmp, p) {
|
|
if ((file==NULL || (!strcmp (file, p->file)))) {
|
|
int ret = 0;
|
|
if (p->handler && p->handler->constructor) {
|
|
ret = p->handler->destructor (p,
|
|
p->handler->user, p->data);
|
|
}
|
|
free (p->file);
|
|
r_list_delete (lib->plugins, iter);
|
|
if (file != NULL) {
|
|
return ret;
|
|
}
|
|
}
|
|
}
|
|
if (file == NULL) {
|
|
return 0;
|
|
}
|
|
// delete similar plugin name
|
|
r_list_foreach (lib->plugins, iter, p) {
|
|
if (strstr (p->file, file)) {
|
|
int ret = 0;
|
|
if (p->handler && p->handler->constructor) {
|
|
ret = p->handler->destructor (p,
|
|
p->handler->user, p->data);
|
|
}
|
|
eprintf ("Unloaded %s\n", p->file);
|
|
free (p->file);
|
|
r_list_delete (lib->plugins, iter);
|
|
return ret;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
// XXX ugly hack ?
|
|
static int samefile(const char *a, const char *b) {
|
|
char *sa = strdup (a);
|
|
char *sb = strdup (b);
|
|
char *ptr;
|
|
int len, ret = R_FALSE;
|
|
|
|
if (sa != NULL && sb != NULL) {
|
|
do {
|
|
ptr = strstr (sa, "//");
|
|
if (ptr) {
|
|
len = strlen (ptr+1) + 1;
|
|
memmove (ptr, ptr+1, len);
|
|
}
|
|
} while (ptr);
|
|
do {
|
|
ptr = strstr (sb, "//");
|
|
if (ptr) {
|
|
len = strlen (ptr+1) + 1;
|
|
memmove (ptr, ptr+1, len);
|
|
}
|
|
} while (ptr);
|
|
ret = strcmp (sa,sb)? R_FALSE: R_TRUE;
|
|
}
|
|
|
|
free (sa);
|
|
free (sb);
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_lib_open(RLib *lib, const char *file) {
|
|
RLibStruct *stru;
|
|
void *handler;
|
|
|
|
/* ignored by filename */
|
|
if (!r_lib_dl_check_filename (file)) {
|
|
eprintf ("Invalid library extension: %s\n", file);
|
|
return R_FAIL;
|
|
}
|
|
|
|
handler = r_lib_dl_open (file);
|
|
if (handler == NULL) {
|
|
IFDBG eprintf ("Cannot open library: '%s'\n", file);
|
|
return R_FAIL;
|
|
}
|
|
|
|
stru = (RLibStruct *) r_lib_dl_sym (handler, lib->symname);
|
|
if (stru == NULL) {
|
|
IFDBG eprintf ("Cannot find symbol '%s' in library '%s'\n",
|
|
lib->symname, file);
|
|
r_lib_dl_close (handler);
|
|
return R_FAIL;
|
|
}
|
|
|
|
return r_lib_open_ptr (lib, file, handler, stru);
|
|
}
|
|
|
|
R_API int r_lib_open_ptr (RLib *lib, const char *file, void *handler, RLibStruct *stru) {
|
|
RLibPlugin *p;
|
|
RListIter *iter;
|
|
int ret = R_FALSE;
|
|
|
|
if (stru->version) {
|
|
if (strcmp (stru->version, R2_VERSION)) {
|
|
eprintf ("Invalid version for module %s\n", file);
|
|
return R_FAIL;
|
|
}
|
|
}
|
|
// TODO: Use Sdb here. just a single line
|
|
r_list_foreach (lib->plugins, iter, p) {
|
|
if (samefile (file, p->file)) {
|
|
IFDBG eprintf ("Dupped\n");
|
|
// TODO: reload if opening again?
|
|
// TODO: store timestamp of file
|
|
// TODO: autoreload plugins if updated \o/
|
|
r_lib_dl_close (handler);
|
|
return R_FAIL;
|
|
}
|
|
}
|
|
|
|
p = R_NEW0 (RLibPlugin);
|
|
p->type = stru->type;
|
|
p->data = stru->data;
|
|
p->file = strdup (file);
|
|
p->dl_handler = handler;
|
|
p->handler = r_lib_get_handler (lib, p->type);
|
|
|
|
ret = r_lib_run_handler (lib, p, stru);
|
|
if (ret == R_FAIL) {
|
|
IFDBG eprintf ("Library handler has failed for '%s'\n", file);
|
|
free (p->file);
|
|
free (p);
|
|
r_lib_dl_close (handler);
|
|
} else r_list_append (lib->plugins, p);
|
|
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_lib_opendir(RLib *lib, const char *path) {
|
|
char file[1024];
|
|
struct dirent *de;
|
|
DIR *dh;
|
|
|
|
#ifdef LIBR_PLUGINS
|
|
if (path == NULL)
|
|
path = LIBR_PLUGINS;
|
|
#endif
|
|
if (path == NULL)
|
|
return R_FALSE;
|
|
|
|
dh = opendir (path);
|
|
if (dh == NULL) {
|
|
IFDBG eprintf ("Cannot open directory '%s'\n", path);
|
|
return R_FALSE;
|
|
}
|
|
while ((de = (struct dirent *)readdir (dh))) {
|
|
snprintf (file, sizeof (file), "%s/%s", path, de->d_name);
|
|
if (r_lib_dl_check_filename (file))
|
|
r_lib_open (lib, file);
|
|
}
|
|
closedir (dh);
|
|
return R_TRUE;
|
|
}
|
|
|
|
R_API int r_lib_add_handler(RLib *lib,
|
|
int type, const char *desc,
|
|
int (*cb)(RLibPlugin *, void *, void *), /* constructor */
|
|
int (*dt)(RLibPlugin *, void *, void *), /* destructor */
|
|
void *user)
|
|
{
|
|
RLibHandler *h;
|
|
RListIter *iter;
|
|
RLibHandler *handler = NULL;
|
|
|
|
r_list_foreach (lib->handlers, iter, h) {
|
|
if (type == h->type) {
|
|
IFDBG eprintf ("Redefining library handler constructor for %d\n", type);
|
|
handler = h;
|
|
break;
|
|
}
|
|
}
|
|
if (handler == NULL) {
|
|
handler = R_NEW (RLibHandler);
|
|
if (handler == NULL)
|
|
return R_FALSE;
|
|
handler->type = type;
|
|
r_list_append (lib->handlers, handler);
|
|
}
|
|
strncpy (handler->desc, desc, sizeof (handler->desc)-1);
|
|
handler->user = user;
|
|
handler->constructor = cb;
|
|
handler->destructor = dt;
|
|
|
|
return R_TRUE;
|
|
}
|
|
|
|
R_API int r_lib_del_handler(RLib *lib, int type) {
|
|
RLibHandler *h;
|
|
RListIter *iter;
|
|
// TODO: remove all handlers for that type? or only one?
|
|
/* No _safe loop necessary because we return immediately after the delete. */
|
|
r_list_foreach (lib->handlers, iter, h) {
|
|
if (type == h->type) {
|
|
r_list_delete (lib->handlers, iter);
|
|
return R_TRUE;
|
|
}
|
|
}
|
|
return R_FALSE;
|
|
}
|
|
|
|
R_API void r_lib_list(RLib *lib) {
|
|
RListIter *iter;
|
|
RLibPlugin *p;
|
|
r_list_foreach (lib->plugins, iter, p) {
|
|
printf (" %5s %p %s \n", r_lib_types_get (p->type),
|
|
p->dl_handler, p->file);
|
|
}
|
|
}
|