odyssey/sources/module.c

/*
 * Odyssey.
 *
 * Scalable PostgreSQL connection pooler.
 */

#include "module.h"
#include "od_dlsym.h"
#include "list.h"
#include "id.h"
#include "logger.h"

void
od_modules_init(od_module_t *module)
{
	od_list_init(&module->link);
}

int
od_target_module_add(od_logger_t *logger,
                     od_module_t *modules,
                     char *target_module_path)
{
	od_list_t *i;
	od_list_foreach(&modules->link, i)
	{
		od_module_t *m;
		m = od_container_of(i, od_module_t, link);
		if (strcmp(m->path, target_module_path) == 0) {
			goto module_exists;
		}
	}

	void *handle;
	od_module_t *module_ptr;
	char *err;

	handle = od_dlopen(target_module_path);
	if (!handle) {
		goto error;
	}
	module_ptr = od_load_module(handle);
	if (module_ptr == NULL) {
		goto error;
	}

	module_ptr->handle = handle;
	od_list_init(&module_ptr->link);
	od_list_append(&modules->link, &module_ptr->link);
	strcat(module_ptr->path, target_module_path);

	return OD_MODULE_CB_OK_RETCODE;

module_exists:
	if (logger == NULL) {
		printf("od_load_module: failed to load module %s: was already loaded!",
		       target_module_path);
	} else {
		od_log(logger,
		       "od_load_module",
		       NULL,
		       NULL,
		       "od_load_module: failed to load module %s: was already loaded!",
		       target_module_path);
	}
	return OD_MODULE_CB_FAIL_RETCODE;
error:
	err = od_dlerror();
	if (logger == NULL) {
		printf("od_load_module: failed to load module %s", err);
	} else {
		od_log(logger,
		       "od_load_module",
		       NULL,
		       NULL,
		       "od_load_module: failed to load module %s",
		       err);
	}
	return OD_MODULE_CB_FAIL_RETCODE;
}

static inline void
od_module_free(od_module_t *module)
{
	od_list_unlink(&module->link);
}

int
od_target_module_unload(od_logger_t *logger,
                        od_module_t *modules,
                        char *target_module)
{
	char *err;
	od_list_t *i;
	od_list_foreach(&modules->link, i)
	{
		od_module_t *m;
		m = od_container_of(i, od_module_t, link);
		if (strcmp(m->path, target_module) == 0) {
			int rc;
			rc = m->unload_cb();
			if (rc != OD_MODULE_CB_OK_RETCODE)
				return -1;
			void *h   = m->handle;
			m->handle = NULL;
			od_module_free(m);
			// because we cannot access handle after calling dlclose
			if (od_dlclose(h)) {
				goto error;
			}

			return OD_MODULE_CB_OK_RETCODE;
		}
	}

	od_log(logger,
	       "od target module unload failed",
	       NULL,
	       NULL,
	       "od_module_unload: failed to find specified module to unload %s",
	       target_module);

	return OD_MODULE_CB_FAIL_RETCODE;
error:
	err = od_dlerror();
	od_log(logger,
	       "od unload module error",
	       NULL,
	       NULL,
	       "od_module_unload: %s",
	       err);

	return OD_MODULE_CB_FAIL_RETCODE;
}

int
od_modules_unload(od_logger_t *logger, od_module_t *modules)
{
	char *err;
	od_list_t *i, *n;
	od_list_foreach_safe(&modules->link, i, n)
	{
		od_module_t *m;
		m = od_container_of(i, od_module_t, link);
		int rc;
		rc = m->unload_cb();
		if (rc != OD_MODULE_CB_OK_RETCODE)
			return -1;
		void *h   = m->handle;
		m->handle = NULL;
		od_module_free(m);
		// because we cannot access handle after calling dlclose
		if (od_dlclose(h)) {
			goto error;
		}
	}
	return OD_MODULE_CB_OK_RETCODE;
error:
	err = od_dlerror();
	od_log(logger,
	       "od unload module error",
	       NULL,
	       NULL,
	       "od_module_unload: %s",
	       err);

	return OD_MODULE_CB_FAIL_RETCODE;
}

int
od_modules_unload_fast(od_module_t *modules)
{
	od_list_t *i;
	od_list_foreach(&modules->link, i)
	{
		od_module_t *m;
		m         = od_container_of(i, od_module_t, link);
		void *h   = m->handle;
		m->handle = NULL;
		od_module_free(m);
		// because we cannot access handle after calling dlclose
		if (od_dlclose(h)) {
			return OD_MODULE_CB_FAIL_RETCODE;
		}
	}
	return OD_MODULE_CB_OK_RETCODE;
}