96665b6e4e68d7204b29db0e1fb16409a1b8623a
[exim.git] / src / src / lookups / dbmdb.c
1 /*************************************************
2 *     Exim - an Internet mail transport agent    *
3 *************************************************/
4
5 /* Copyright (c) The Exim Maintainers 2020 - 2022 */
6 /* Copyright (c) University of Cambridge 1995 - 2018 */
7 /* See the file NOTICE for conditions of use and distribution. */
8 /* SPDX-License-Identifier: GPL-2.0-or-later */
9
10 #include "../exim.h"
11 #include "lf_functions.h"
12
13
14 /*************************************************
15 *              Open entry point                  *
16 *************************************************/
17
18 /* See local README for interface description */
19
20 static void *
21 dbmdb_open(const uschar * filename, uschar ** errmsg)
22 {
23 uschar * dirname = string_copy(filename);
24 uschar * s;
25 EXIM_DB * yield = NULL;
26
27 if ((s = Ustrrchr(dirname, '/'))) *s = '\0';
28 if (!(yield = exim_dbopen(filename, dirname, O_RDONLY, 0)))
29   *errmsg = string_open_failed("%s as a %s file", filename, EXIM_DBTYPE);
30 return yield;
31 }
32
33
34
35 /*************************************************
36 *             Check entry point                  *
37 *************************************************/
38
39 /* This needs to know more about the underlying files than is good for it!
40 We need to know what the real file names are in order to check the owners and
41 modes. If USE_DB is set, we know it is Berkeley DB, which uses an unmodified
42 file name. If USE_TDB or USE_GDBM is set, we know it is tdb or gdbm, which do
43 the same. Otherwise, for safety, we have to check for x.db or x.dir and x.pag.
44 */
45
46 static BOOL
47 dbmdb_check(void *handle, const uschar *filename, int modemask, uid_t *owners,
48   gid_t *owngroups, uschar **errmsg)
49 {
50 int rc;
51
52 #if defined(USE_DB) || defined(USE_TDB) || defined(USE_GDBM)
53 rc = lf_check_file(-1, filename, S_IFREG, modemask, owners, owngroups,
54   "dbm", errmsg);
55 #else
56   {
57   uschar filebuffer[256];
58   (void)sprintf(CS filebuffer, "%.250s.db", filename);
59   rc = lf_check_file(-1, filebuffer, S_IFREG, modemask, owners, owngroups,
60     "dbm", errmsg);
61   if (rc < 0)        /* stat() failed */
62     {
63     (void)sprintf(CS filebuffer, "%.250s.dir", filename);
64     rc = lf_check_file(-1, filebuffer, S_IFREG, modemask, owners, owngroups,
65       "dbm", errmsg);
66     if (rc == 0)     /* x.dir was OK */
67       {
68       (void)sprintf(CS filebuffer, "%.250s.pag", filename);
69       rc = lf_check_file(-1, filebuffer, S_IFREG, modemask, owners, owngroups,
70         "dbm", errmsg);
71       }
72     }
73   }
74 #endif
75
76 return rc == 0;
77 }
78
79
80
81 /*************************************************
82 *              Find entry point                  *
83 *************************************************/
84
85 /* See local README for interface description. This function adds 1 to
86 the keylength in order to include the terminating zero. */
87
88 static int
89 dbmdb_find(void * handle, const uschar * filename, const uschar * keystring,
90   int length, uschar ** result, uschar ** errmsg, uint * do_cache,
91   const uschar * opts)
92 {
93 EXIM_DB *d = (EXIM_DB *)handle;
94 EXIM_DATUM key, data;
95
96 exim_datum_init(&key);               /* Some DBM libraries require datums to */
97 exim_datum_init(&data);              /* be cleared before use. */
98 length++;
99 exim_datum_data_set(&key,
100   memcpy(store_get(length, keystring), keystring, length)); /* key can have embedded NUL */
101 exim_datum_size_set(&key, length);
102
103 if (exim_dbget(d, &key, &data))
104   {
105   unsigned len = exim_datum_size_get(&data);
106   *result = len > 0 ? string_copyn(exim_datum_data_get(&data), len) : US"";
107   exim_datum_free(&data);            /* Some DBM libraries need a free() call */
108   return OK;
109   }
110 return FAIL;
111 }
112
113
114
115 /*************************************************
116 *      Find entry point - no zero on key         *
117 *************************************************/
118
119 /* See local README for interface description */
120
121 static int
122 dbmnz_find(void * handle, const uschar * filename, const uschar * keystring,
123   int length, uschar ** result, uschar ** errmsg, uint * do_cache,
124   const uschar * opts)
125 {
126 return dbmdb_find(handle, filename, keystring, length-1, result, errmsg,
127   do_cache, opts);
128 }
129
130
131
132 /*************************************************
133 *     Find entry point - zero-joined list key    *
134 *************************************************/
135
136 /*
137  * The parameter passed as a key is a list in normal Exim list syntax.
138  * The elements of that list are joined together on NUL, with no trailing
139  * NUL, to form the key.
140  */
141
142 static int
143 dbmjz_find(void * handle, const uschar * filename, const uschar * keystring,
144   int length, uschar ** result, uschar ** errmsg, uint * do_cache,
145   const uschar * opts)
146 {
147 uschar *key_item, *key_buffer, *key_p;
148 const uschar *key_elems = keystring;
149 int buflen, bufleft, key_item_len, sep = 0;
150
151 /* To a first approximation, the size of the lookup key needs to be about,
152 or less than, the length of the delimited list passed in + 1. */
153
154 buflen = length + 3;
155 key_buffer = store_get(buflen, keystring);
156
157 key_buffer[0] = '\0';
158
159 key_p = key_buffer;
160 bufleft = buflen;
161
162 /* In all cases of an empty list item, we can set 1 and advance by 1 and then
163 pick up the trailing NUL from the previous list item, EXCEPT when at the
164 beginning of the output string, in which case we need to supply that NUL
165 ourselves.  */
166 while ((key_item = string_nextinlist(&key_elems, &sep, key_p, bufleft)) != NULL)
167   {
168   key_item_len = Ustrlen(key_item) + 1;
169   if (key_item_len == 1)
170     {
171     key_p[0] = '\0';
172     if (key_p == key_buffer)
173       {
174       key_p[1] = '\0';
175       key_item_len += 1;
176       }
177     }
178
179   bufleft -= key_item_len;
180   if (bufleft <= 0)
181     {
182     /* The string_nextinlist() will stop at buffer size, but we should always
183     have at least 1 character extra, so some assumption has failed. */
184     *errmsg = string_copy(US"Ran out of buffer space for joining elements");
185     return DEFER;
186     }
187   key_p += key_item_len;
188   }
189
190 if (key_p == key_buffer)
191   {
192   *errmsg = string_copy(US"empty list key");
193   return FAIL;
194   }
195
196 /* We do not pass in the final NULL; if needed, the list should include an
197 empty element to put one in. Boundary: key length 1, is a NULL */
198 key_item_len = key_p - key_buffer - 1;
199
200 DEBUG(D_lookup) debug_printf_indent("NUL-joined key length: %d\n", key_item_len);
201
202 /* beware that dbmdb_find() adds 1 to length to get back terminating NUL, so
203 because we've calculated the real length, we need to subtract one more here */
204
205 return dbmdb_find(handle, filename, key_buffer, key_item_len - 1,
206     result, errmsg, do_cache, opts);
207 }
208
209
210
211 /*************************************************
212 *              Close entry point                 *
213 *************************************************/
214
215 /* See local README for interface description */
216
217 void
218 static dbmdb_close(void *handle)
219 {
220 exim_dbclose((EXIM_DB *)handle);
221 }
222
223
224
225 /*************************************************
226 *         Version reporting entry point          *
227 *************************************************/
228
229 /* See local README for interface description. */
230
231 #include "../version.h"
232
233 gstring *
234 dbm_version_report(gstring * g)
235 {
236 #ifdef DYNLOOKUP
237 g = string_fmt_append(g, "Library version: DBM: Exim version %s\n", EXIM_VERSION_STR);
238 #endif
239 return g;
240 }
241
242
243 lookup_info dbm_lookup_info = {
244   .name = US"dbm",                      /* lookup name */
245   .type = lookup_absfile,               /* uses absolute file name */
246   .open = dbmdb_open,                   /* open function */
247   .check = dbmdb_check,                 /* check function */
248   .find = dbmdb_find,                   /* find function */
249   .close = dbmdb_close,                 /* close function */
250   .tidy = NULL,                         /* no tidy function */
251   .quote = NULL,                        /* no quoting function */
252   .version_report = dbm_version_report             /* version reporting */
253 };
254
255 lookup_info dbmz_lookup_info = {
256   .name = US"dbmnz",                    /* lookup name */
257   .type = lookup_absfile,               /* uses absolute file name */
258   .open = dbmdb_open,                   /* sic */     /* open function */
259   .check = dbmdb_check,                 /* sic */     /* check function */
260   .find = dbmnz_find,                   /* find function */
261   .close = dbmdb_close,                 /* sic */     /* close function */
262   .tidy = NULL,                         /* no tidy function */
263   .quote = NULL,                        /* no quoting function */
264   .version_report = NULL                           /* no version reporting (redundant) */
265 };
266
267 lookup_info dbmjz_lookup_info = {
268   .name = US"dbmjz",                    /* lookup name */
269   .type = lookup_absfile,               /* uses absolute file name */
270   .open = dbmdb_open,                   /* sic */     /* open function */
271   .check = dbmdb_check,                 /* sic */     /* check function */
272   .find = dbmjz_find,                   /* find function */
273   .close = dbmdb_close,                 /* sic */     /* close function */
274   .tidy = NULL,                         /* no tidy function */
275   .quote = NULL,                        /* no quoting function */
276   .version_report = NULL                           /* no version reporting (redundant) */
277 };
278
279 #ifdef DYNLOOKUP
280 #define dbmdb_lookup_module_info _lookup_module_info
281 #endif
282
283 static lookup_info *_lookup_list[] = { &dbm_lookup_info, &dbmz_lookup_info, &dbmjz_lookup_info };
284 lookup_module_info dbmdb_lookup_module_info = { LOOKUP_MODULE_INFO_MAGIC, _lookup_list, 3 };
285
286 /* End of lookups/dbmdb.c */
287 /* vi: aw ai sw=2
288 */