aboutsummaryrefslogtreecommitdiffhomepage
path: root/plugin.c
blob: fc642ac1a22a7dbc9edae2734c249c0d8c8e74c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#include <stdio.h>
#include <string.h>
#include <dlfcn.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include "plugin.h"

static size_t plugins = 0;
static void** plugin_handle = NULL;

static int plugin_attach(char* path, char* file){
	plugin_init init = NULL;
	void* handle = NULL;

	char* lib = calloc(strlen(path) + strlen(file) + 1, sizeof(char));
	if(!lib){
		fprintf(stderr, "Failed to allocate memory\n");
		return 1;
	}

	snprintf(lib, strlen(path) + strlen(file) + 1, "%s%s", path, file);

	handle = dlopen(lib, RTLD_NOW);
	if(!handle){
		fprintf(stderr, "Failed to load plugin %s: %s\n", lib, dlerror());
		free(lib);
		return 0;
	}

	init = (plugin_init) dlsym(handle, "init");
	if(init){
		if(init()){
			fprintf(stderr, "Plugin %s failed to initialize\n", lib);
			dlclose(handle);
			free(lib);
			return 1;
		}
	}
	else{
		dlclose(handle);
		free(lib);
		return 0;
	}
	free(lib);

	plugin_handle = realloc(plugin_handle, (plugins + 1) * sizeof(void*));
	if(!plugin_handle){
		fprintf(stderr, "Failed to allocate memory\n");
		dlclose(handle);
		return 1;
	}

	plugin_handle[plugins] = handle;
	plugins++;

	return 0;
}

int plugins_load(char* path){
	int rv = -1;

	struct dirent* entry;
	struct stat file_stat;
	DIR* directory = opendir(path);
	if(!directory){
		fprintf(stderr, "Failed to open plugin search path %s: %s\n", path, strerror(errno));
		return 1;
	}

	for(entry = readdir(directory); entry; entry = readdir(directory)){
		if(strlen(entry->d_name) < 4 || strncmp(".so", entry->d_name + (strlen(entry->d_name) - 3), 3)){
			continue;
		}

		if(fstatat(dirfd(directory), entry->d_name, &file_stat, 0) < 0){
			fprintf(stderr, "Failed to stat %s: %s\n", entry->d_name, strerror(errno));
			continue;
		}

		if(!S_ISREG(file_stat.st_mode)){
			continue;
		}

		if(!(file_stat.st_mode & S_IXUSR)){
			continue;
		}

		if(plugin_attach(path, entry->d_name)){
			goto load_done;
		}
	}
	rv = 0;

load_done:
	if(closedir(directory) < 0){
		fprintf(stderr, "Failed to close plugin directory %s: %s\n", path, strerror(errno));
		return -1;
	}
	return rv;
}

int plugins_close(){
	size_t u;
	for(u = 0; u < plugins; u++){
		if(dlclose(plugin_handle[u])){
			fprintf(stderr, "Failed to unload plugin: %s\n", dlerror());
		}
	}

	free(plugin_handle);
	plugins = 0;
	return 0;
}