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
|
/* Copyright (c) 2003-2018 Dovecot authors, see the included COPYING file */
#include "auth-common.h"
#include "userdb.h"
#include "auth-cache.h"
#if defined(BUILTIN_LUA) || defined(PLUGIN_BUILD)
#include "db-lua.h"
struct dlua_userdb_module {
struct userdb_module module;
struct dlua_script *script;
const char *file;
};
static void userdb_lua_lookup(struct auth_request *auth_request,
userdb_callback_t *callback)
{
struct userdb_module *_module = auth_request->userdb->userdb;
struct dlua_userdb_module *module =
(struct dlua_userdb_module *)_module;
const char *error;
enum userdb_result result =
auth_lua_call_userdb_lookup(module->script, auth_request, &error);
if (result == USERDB_RESULT_INTERNAL_FAILURE)
e_error(authdb_event(auth_request),
"userdb-lua: %s", error);
callback(result, auth_request);
}
static struct userdb_module *
userdb_lua_preinit(pool_t pool, const char *args)
{
struct dlua_userdb_module *module;
const char *cache_key = "%u";
bool blocking = TRUE;
module = p_new(pool, struct dlua_userdb_module, 1);
const char *const *fields = t_strsplit_spaces(args, " ");
while(*fields != NULL) {
if (str_begins(*fields, "file=")) {
module->file = p_strdup(pool, (*fields)+5);
} else if (str_begins(*fields, "blocking=")) {
const char *value = (*fields)+9;
if (strcmp(value, "yes") == 0) {
blocking = TRUE;
} else if (strcmp(value, "no") == 0) {
blocking = FALSE;
} else {
i_fatal("Invalid value %s. "
"Field blocking must be yes or no",
value);
}
} else if (str_begins(*fields, "cache_key=")) {
if (*((*fields)+10) != '\0')
cache_key = (*fields)+10;
else /* explicitly disable auth caching for lua */
cache_key = NULL;
} else {
i_fatal("Unsupported parameter %s", *fields);
}
fields++;
}
if (module->file == NULL)
i_fatal("userdb-lua: Missing mandatory file= parameter");
module->module.blocking = blocking;
if (cache_key != NULL) {
module->module.default_cache_key =
auth_cache_parse_key(pool, cache_key);
}
return &module->module;
}
static void userdb_lua_init(struct userdb_module *_module)
{
struct dlua_userdb_module *module =
(struct dlua_userdb_module *)_module;
const char *error;
if (dlua_script_create_file(module->file, &module->script, auth_event, &error) < 0 ||
auth_lua_script_init(module->script, &error) < 0)
i_fatal("userdb-lua: initialization failed: %s", error);
}
static void userdb_lua_deinit(struct userdb_module *_module)
{
struct dlua_userdb_module *module =
(struct dlua_userdb_module *)_module;
dlua_script_unref(&module->script);
}
static struct userdb_iterate_context *
userdb_lua_iterate_init(struct auth_request *auth_request,
userdb_iter_callback_t *callback,
void *context)
{
struct userdb_module *_module = auth_request->userdb->userdb;
struct dlua_userdb_module *module =
(struct dlua_userdb_module *)_module;
return auth_lua_call_userdb_iterate_init(module->script, auth_request,
callback, context);
}
static void userdb_lua_iterate_next(struct userdb_iterate_context *ctx)
{
auth_lua_userdb_iterate_next(ctx);
}
static int userdb_lua_iterate_deinit(struct userdb_iterate_context *ctx)
{
return auth_lua_userdb_iterate_deinit(ctx);
}
#ifndef PLUGIN_BUILD
struct userdb_module_interface userdb_lua =
#else
struct userdb_module_interface userdb_lua_plugin =
#endif
{
"lua",
userdb_lua_preinit,
userdb_lua_init,
userdb_lua_deinit,
userdb_lua_lookup,
userdb_lua_iterate_init,
userdb_lua_iterate_next,
userdb_lua_iterate_deinit
};
#else
struct userdb_module_interface userdb_lua = {
.name = "lua"
};
#endif
|
