-
Notifications
You must be signed in to change notification settings - Fork 271
/
doveadm-util.c
151 lines (128 loc) · 3.55 KB
/
doveadm-util.c
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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
/* Copyright (c) 2009-2016 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "array.h"
#include "net.h"
#include "time-util.h"
#include "master-service.h"
#include "module-dir.h"
#include "doveadm-settings.h"
#include "doveadm-mail.h"
#include "doveadm-util.h"
#include <time.h>
#include <dirent.h>
#include <sys/stat.h>
bool doveadm_verbose = FALSE, doveadm_debug = FALSE, doveadm_server = FALSE;
static struct module *modules = NULL;
void doveadm_load_modules(void)
{
struct module_dir_load_settings mod_set;
/* some doveadm plugins have dependencies to mail plugins. we can load
only those whose dependencies have been loaded earlier, the rest are
ignored. */
memset(&mod_set, 0, sizeof(mod_set));
mod_set.abi_version = DOVECOT_ABI_VERSION;
mod_set.require_init_funcs = TRUE;
mod_set.debug = doveadm_debug;
mod_set.ignore_dlopen_errors = TRUE;
modules = module_dir_load_missing(modules, DOVEADM_MODULEDIR,
NULL, &mod_set);
module_dir_init(modules);
}
void doveadm_unload_modules(void)
{
module_dir_unload(&modules);
}
bool doveadm_has_unloaded_plugin(const char *name)
{
struct module *module;
DIR *dir;
struct dirent *d;
const char *plugin_name;
unsigned int name_len = strlen(name);
bool found = FALSE;
/* first check that it's not actually loaded */
for (module = modules; module != NULL; module = module->next) {
if (strcmp(module_get_plugin_name(module), name) == 0)
return FALSE;
}
dir = opendir(DOVEADM_MODULEDIR);
if (dir == NULL)
return FALSE;
while ((d = readdir(dir)) != NULL) {
plugin_name = module_file_get_name(d->d_name);
if (strncmp(plugin_name, "doveadm_", 8) == 0)
plugin_name += 8;
if (strncmp(plugin_name, name, name_len) == 0 &&
(plugin_name[name_len] == '\0' ||
strcmp(plugin_name + name_len, "_plugin") == 0)) {
found = TRUE;
break;
}
}
(void)closedir(dir);
return found;
}
const char *unixdate2str(time_t timestamp)
{
return t_strflocaltime("%Y-%m-%d %H:%M:%S", timestamp);
}
const char *doveadm_plugin_getenv(const char *name)
{
const char *const *envs;
unsigned int i, count;
if (!array_is_created(&doveadm_settings->plugin_envs))
return NULL;
envs = array_get(&doveadm_settings->plugin_envs, &count);
for (i = 0; i < count; i += 2) {
if (strcmp(envs[i], name) == 0)
return envs[i+1];
}
return NULL;
}
static int
doveadm_tcp_connect_port(const char *host, in_port_t port)
{
struct ip_addr *ips;
unsigned int ips_count;
int ret, fd;
ret = net_gethostbyname(host, &ips, &ips_count);
if (ret != 0) {
i_fatal("Lookup of host %s failed: %s",
host, net_gethosterror(ret));
}
fd = net_connect_ip_blocking(&ips[0], port, NULL);
if (fd == -1) {
i_fatal("connect(%s:%u) failed: %m",
net_ip2addr(&ips[0]), port);
}
return fd;
}
int doveadm_tcp_connect(const char *target, in_port_t default_port)
{
const char *host;
in_port_t port;
if (net_str2hostport(target, default_port, &host, &port) < 0) {
i_fatal("Port not known for %s. Either set proxy_port "
"or use %s:port", target, target);
}
return doveadm_tcp_connect_port(host, port);
}
int doveadm_connect_with_default_port(const char *path,
in_port_t default_port)
{
int fd;
/* we'll assume UNIX sockets typically have an absolute path,
or at the very least '/' somewhere. */
if (strchr(path, '/') == NULL)
fd = doveadm_tcp_connect(path, default_port);
else {
fd = net_connect_unix(path);
if (fd == -1)
i_fatal("net_connect_unix(%s) failed: %m", path);
}
return fd;
}
int doveadm_connect(const char *path)
{
return doveadm_connect_with_default_port(path, 0);
}