Refactor lookup argument shuffling
[exim.git] / src / src / search.c
1 /*************************************************
2 *     Exim - an Internet mail transport agent    *
3 *************************************************/
4
5 /* Copyright (c) University of Cambridge 1995 - 2015 */
6 /* Copyright (c) The Exim Maintainers 2020 */
7 /* See the file NOTICE for conditions of use and distribution. */
8
9 /* A set of functions to search databases in various formats. An open
10 database is represented by a void * value which is returned from a lookup-
11 specific "open" function. These are now all held in individual modules in the
12 lookups subdirectory and the functions here form a generic interface.
13
14 Caching is used to improve performance. Open files are cached until a tidyup
15 function is called, and for each file the result of the last lookup is cached.
16 However, if too many files are opened, some of those that are not in use have
17 to be closed. Those open items that use real files are kept on a LRU chain to
18 help with this.
19
20 All the data is held in permanent store so as to be independent of the stacking
21 pool that is reset from time to time. In fact, we use malloc'd store so that it
22 can be freed when the caches are tidied up. It isn't actually clear whether
23 this is a benefit or not, to be honest. */
24
25 #include "exim.h"
26
27
28 /* Tree in which to cache open files until tidyup called. */
29
30 static tree_node *search_tree = NULL;
31
32 /* Two-way chain of open databases that use real files. This is maintained in
33 recently-used order for the purposes of closing the least recently used when
34 too many files are open. */
35
36 static tree_node *open_top = NULL;
37 static tree_node *open_bot = NULL;
38
39 /* Count of open databases that use real files */
40
41 static int open_filecount = 0;
42
43 /* Allow us to reset store used for lookups and lookup caching */
44
45 static rmark search_reset_point = NULL;
46
47
48
49 /*************************************************
50 *      Validate a plain lookup type name         *
51 *************************************************/
52
53 /* Only those names that are recognized and whose code is included in the
54 binary give an OK response. Use a binary chop search now that the list has got
55 so long.
56
57 Arguments:
58   name       lookup type name - not necessarily zero terminated (e.g. dbm*)
59   len        length of the name
60
61 Returns:     +ve => valid lookup name; value is offset in lookup_list
62              -ve => invalid name; message in search_error_message.
63 */
64
65 int
66 search_findtype(const uschar *name, int len)
67 {
68 int bot = 0;
69 int top = lookup_list_count;
70 while (top > bot)
71   {
72   int mid = (top + bot)/2;
73   int c = Ustrncmp(name, lookup_list[mid]->name, len);
74
75   /* If c == 0 we have matched the incoming name with the start of the search
76   type name. However, some search types are substrings of others (e.g. nis and
77   nisplus) so we need to check that the lengths are the same. The length of the
78   type name cannot be shorter (else c would not be 0); if it is not equal it
79   must be longer, and in that case, the incoming name comes before the name we
80   are testing. By leaving c == 0 when the lengths are different, and doing a
81   > 0 test below, this all falls out correctly. */
82
83   if (c == 0 && Ustrlen(lookup_list[mid]->name) == len)
84     {
85     if (lookup_list[mid]->find != NULL) return mid;
86     search_error_message  = string_sprintf("lookup type \"%.*s\" is not "
87       "available (not in the binary - check buildtime LOOKUP configuration)",
88       len, name);
89     return -1;
90     }
91
92   if (c > 0) bot = mid + 1; else top = mid;
93   }
94
95 search_error_message = string_sprintf("unknown lookup type \"%.*s\"",len,name);
96 return -1;
97 }
98
99
100
101 /*************************************************
102 *       Validate a full lookup type name         *
103 *************************************************/
104
105 /* This function recognizes the "partial-" prefix and also terminating * and *@
106 suffixes.
107
108 Arguments:
109   name         the full lookup type name
110   ptypeptr     where to put the partial type
111                  after subtraction of 1024 or 2048:
112                    negative     => no partial matching
113                    non-negative => minimum number of non-wild components
114   ptypeaff     where to put a pointer to the affix
115                  the affix is within name if supplied therein
116                  otherwise it's a literal string
117   afflen       the length of the affix
118   starflags    where to put the SEARCH_STAR and SEARCH_STARAT flags
119   opts         where to put the options
120
121 Returns:     +ve => valid lookup name; value is offset in lookup_list
122              -ve => invalid name; message in search_error_message.
123 */
124
125 int
126 search_findtype_partial(const uschar *name, int *ptypeptr, const uschar **ptypeaff,
127   int *afflen, int *starflags, const uschar ** opts)
128 {
129 int len, stype;
130 int pv = -1;
131 const uschar *ss = name;
132 const uschar * t;
133
134 *starflags = 0;
135 *ptypeaff = NULL;
136
137 /* Check for a partial matching type. It must start with "partial", optionally
138 followed by a sequence of digits. If this is followed by "-", the affix is the
139 default "*." string. Otherwise we expect an affix in parentheses. Affixes are a
140 limited number of characters, not including parens. */
141
142 if (Ustrncmp(name, "partial", 7) == 0)
143   {
144   ss += 7;
145   if (isdigit (*ss))
146     {
147     pv = 0;
148     while (isdigit(*ss)) pv = pv*10 + *ss++ - '0';
149     }
150   else pv = 2;         /* Default number of wild components */
151
152   if (*ss == '(')
153     {
154     *ptypeaff = ++ss;
155     while (ispunct(*ss) && *ss != ')') ss++;
156     if (*ss != ')') goto BAD_TYPE;
157     *afflen = ss++ - *ptypeaff;
158     }
159   else if (*ss++ == '-')
160     {
161     *ptypeaff = US "*.";
162     *afflen = 2;
163     }
164   else
165     {
166     BAD_TYPE:
167     search_error_message = string_sprintf("format error in lookup type \"%s\"",
168       name);
169     return -1;
170     }
171   }
172
173 /* Now we are left with a lookup name, possibly followed by * or *@,
174 and then by options starting with a "," */
175
176 len = Ustrlen(ss);
177 if ((t = Ustrchr(ss, '*')))
178   {
179   len = t - ss;
180   *starflags |= (t[1] == '@' ? SEARCH_STARAT : SEARCH_STAR);
181   }
182 else
183   t = ss;
184
185 if ((t = Ustrchr(t, ',')))
186   {
187   int l = t - ss;
188   if (l < len) len = l;
189   *opts = string_copy(t+1);
190   }
191 else
192   *opts = NULL;
193
194 /* Check for the individual search type. Only those that are actually in the
195 binary are valid. For query-style types, "partial" and default types are
196 erroneous. */
197
198 stype = search_findtype(ss, len);
199 if (stype >= 0 && mac_islookup(stype, lookup_querystyle))
200   {
201   if (pv >= 0)
202     {
203     search_error_message = string_sprintf("\"partial\" is not permitted "
204       "for lookup type \"%s\"", ss);
205     return -1;
206     }
207   if ((*starflags & (SEARCH_STAR|SEARCH_STARAT)) != 0)
208     {
209     search_error_message = string_sprintf("defaults using \"*\" or \"*@\" are "
210       "not permitted for lookup type \"%s\"", ss);
211     return -1;
212     }
213   }
214
215 *ptypeptr = pv;
216 return stype;
217 }
218
219
220 /* Set the parameters for the three different kinds of lookup.
221 Arguments:
222  search_type    the search-type code
223  search         the search-type string
224  query          argument for the search; filename or query
225  fnamep         pointer to return filename
226
227 Return: keyquery        the search-type (for single-key) or query (for query-type)
228  */
229 uschar *
230 search_args(int search_type, uschar * search, uschar * query, uschar ** fnamep)
231 {
232 Uskip_whitespace(&query);
233 if (mac_islookup(search_type, lookup_absfilequery))
234   {                                     /* query-style but with file (sqlite) */
235   uschar * s = query;
236   if (*query == '/')
237     {
238     while (*query && !isspace(*query)) query++;
239     *fnamep = string_copyn(s, query - s);
240     Uskip_whitespace(&query);
241     }
242   else
243     *fnamep = NULL;
244   return query;         /* remainder after file skipped */
245   }
246 if (!mac_islookup(search_type, lookup_querystyle))
247   {                                     /* single-key */
248   *fnamep = query;
249   return search;        /* modifiers important so use "keyquery" for them */
250   }
251 *fnamep = NULL;                         /* else query-style */
252 return query;
253 }
254
255
256
257 /*************************************************
258 *               Release cached resources         *
259 *************************************************/
260
261 /* When search_open is called it caches the "file" that it opens in
262 search_tree. The name of the tree node is a concatenation of the search type
263 with the file name. For query-style lookups, the file name is empty. Real files
264 are normally closed only when this tidyup routine is called, typically at the
265 end of sections of code where a number of lookups might occur. However, if too
266 many files are open simultaneously, some get closed beforehand. They can't be
267 removed from the tree. There is also a general tidyup function which is called
268 for the lookup driver, if it exists.
269
270 First, there is an internal, recursive subroutine.
271
272 Argument:    a pointer to a search_openfile tree node
273 Returns:     nothing
274 */
275
276 static void
277 tidyup_subtree(tree_node *t)
278 {
279 search_cache * c = (search_cache *)(t->data.ptr);
280 if (t->left)  tidyup_subtree(t->left);
281 if (t->right) tidyup_subtree(t->right);
282 if (c && c->handle && lookup_list[c->search_type]->close)
283   lookup_list[c->search_type]->close(c->handle);
284 }
285
286
287 /* The external entry point
288
289 Argument: none
290 Returns:  nothing
291 */
292
293 void
294 search_tidyup(void)
295 {
296 int old_pool = store_pool;
297
298 DEBUG(D_lookup) debug_printf_indent("search_tidyup called\n");
299
300 /* Close individually each cached open file. */
301
302 store_pool = POOL_SEARCH;
303 if (search_tree)
304   {
305   tidyup_subtree(search_tree);
306   search_tree = NULL;
307   }
308 open_top = open_bot = NULL;
309 open_filecount = 0;
310
311 /* Call the general tidyup entry for any drivers that have one. */
312
313 for (int i = 0; i < lookup_list_count; i++) if (lookup_list[i]->tidy)
314   (lookup_list[i]->tidy)();
315
316 if (search_reset_point) search_reset_point = store_reset(search_reset_point);
317 store_pool = old_pool;
318 }
319
320
321
322
323 /*************************************************
324 *             Open search database               *
325 *************************************************/
326
327 /* A mode, and lists of owners and groups, are passed over for checking in
328 the cases where the database is one or more files. Return NULL, with a message
329 pointed to by message, in cases of error.
330
331 For search types that use a file or files, check up on the mode after
332 opening. It is tempting to do a stat before opening the file, and use it as
333 an existence check. However, doing that opens a small security loophole in
334 that the status could be changed before the file is opened. Can't quite see
335 what problems this might lead to, but you can't be too careful where security
336 is concerned. Fstat() on an open file can normally be expected to succeed,
337 but there are some NFS states where it does not.
338
339 There are two styles of query: (1) in the "single-key+file" style, a single
340 key string and a file name are given, for example, for linear searches, DBM
341 files, or for NIS. (2) In the "query" style, no "filename" is given; instead
342 just a single query string is passed. This applies to multiple-key lookup
343 types such as NIS+.
344
345 Before opening, scan the tree of cached files to see if this file is already
346 open for the correct search type. If so, return the saved handle. If not, put
347 the handle in the tree for possible subsequent use. See search_tidyup above for
348 closing all the cached files.
349
350 A count of open databases which use real files is maintained, and if this
351 gets too large, we have to close a cached file. Its entry remains in the tree,
352 but is marked closed.
353
354 Arguments:
355   filename       the name of the file for single-key+file style lookups,
356                  NULL for query-style lookups
357   search_type    the type of search required
358   modemask       if a real single file is used, this specifies mode bits that
359                  must not be set; otherwise it is ignored
360   owners         if a real single file is used, this specifies the possible
361                  owners of the file; otherwise it is ignored
362   owngroups      if a real single file is used, this specifies the possible
363                  group owners of the file; otherwise it is ignored
364
365 Returns:         an identifying handle for the open database;
366                  this is the pointer to the tree block in the
367                  cache of open files; return NULL on open failure, with
368                  a message in search_error_message
369 */
370
371 void *
372 search_open(const uschar * filename, int search_type, int modemask,
373   uid_t * owners, gid_t * owngroups)
374 {
375 void *handle;
376 tree_node *t;
377 search_cache *c;
378 lookup_info *lk = lookup_list[search_type];
379 uschar keybuffer[256];
380 int old_pool = store_pool;
381
382 if (filename && is_tainted(filename))
383   {
384   log_write(0, LOG_MAIN|LOG_PANIC,
385     "Tainted filename for search: '%s'", filename);
386   return NULL;
387   }
388
389 /* Change to the search store pool and remember our reset point */
390
391 store_pool = POOL_SEARCH;
392 if (!search_reset_point) search_reset_point = store_mark();
393
394 DEBUG(D_lookup) debug_printf_indent("search_open: %s \"%s\"\n", lk->name,
395   filename ? filename : US"NULL");
396
397 /* See if we already have this open for this type of search, and if so,
398 pass back the tree block as the handle. The key for the tree node is the search
399 type plus '0' concatenated with the file name. There may be entries in the tree
400 with closed files if a lot of files have been opened. */
401
402 sprintf(CS keybuffer, "%c%.254s", search_type + '0',
403   filename ? filename : US"");
404
405 if ((t = tree_search(search_tree, keybuffer)))
406   {
407   if ((c = (search_cache *)t->data.ptr)->handle)
408     {
409     DEBUG(D_lookup) debug_printf_indent("  cached open\n");
410     store_pool = old_pool;
411     return t;
412     }
413   DEBUG(D_lookup) debug_printf_indent("  cached closed\n");
414   }
415
416 /* Otherwise, we need to open the file or database - each search type has its
417 own code, which is now split off into separately compiled modules. Before doing
418 this, if the search type is one that uses real files, check on the number that
419 we are holding open in the cache. If the limit is reached, close the least
420 recently used one. */
421
422 if (lk->type == lookup_absfile && open_filecount >= lookup_open_max)
423   if (!open_bot)
424     log_write(0, LOG_MAIN|LOG_PANIC, "too many lookups open, but can't find "
425       "one to close");
426   else
427     {
428     search_cache *c = (search_cache *)(open_bot->data.ptr);
429     DEBUG(D_lookup) debug_printf_indent("Too many lookup files open\n  closing %s\n",
430       open_bot->name);
431     if ((open_bot = c->up))
432       ((search_cache *)(open_bot->data.ptr))->down = NULL;
433     else
434       open_top = NULL;
435     ((lookup_list[c->search_type])->close)(c->handle);
436     c->handle = NULL;
437     open_filecount--;
438     }
439
440 /* If opening is successful, call the file-checking function if there is one,
441 and if all is still well, enter the open database into the tree. */
442
443 if (!(handle = (lk->open)(filename, &search_error_message)))
444   {
445   store_pool = old_pool;
446   return NULL;
447   }
448
449 if (  lk->check
450    && !lk->check(handle, filename, modemask, owners, owngroups,
451          &search_error_message))
452   {
453   lk->close(handle);
454   store_pool = old_pool;
455   return NULL;
456   }
457
458 /* If this is a search type that uses real files, keep count. */
459
460 if (lk->type == lookup_absfile) open_filecount++;
461
462 /* If we found a previously opened entry in the tree, re-use it; otherwise
463 insert a new entry. On re-use, leave any cached lookup data and the lookup
464 count alone. */
465
466 if (!t)
467   {
468   t = store_get(sizeof(tree_node) + Ustrlen(keybuffer), FALSE);
469   t->data.ptr = c = store_get(sizeof(search_cache), FALSE);
470   c->item_cache = NULL;
471   Ustrcpy(t->name, keybuffer);
472   tree_insertnode(&search_tree, t);
473   }
474 else c = t->data.ptr;
475
476 c->handle = handle;
477 c->search_type = search_type;
478 c->up = c->down = NULL;
479
480 store_pool = old_pool;
481 return t;
482 }
483
484
485
486
487
488 /*************************************************
489 *  Internal function: Find one item in database  *
490 *************************************************/
491
492 /* The answer is always put into dynamic store. The last lookup for each handle
493 is cached.
494
495 Arguments:
496   handle       the handle from search_open; points to tree node
497   filename     the filename that was handed to search_open, or
498                NULL for query-style searches
499   keystring    the keystring for single-key+file lookups, or
500                the querystring for query-style lookups
501   cache_rd     FALSE to avoid lookup in cache layer
502   opts         type-specific options
503
504 Returns:       a pointer to a dynamic string containing the answer,
505                or NULL if the query failed or was deferred; in the
506                latter case, search_find_defer is set TRUE; after an unusual
507                failure, there may be a message in search_error_message.
508 */
509
510 static uschar *
511 internal_search_find(void * handle, const uschar * filename, uschar * keystring,
512   BOOL cache_rd, const uschar * opts)
513 {
514 tree_node * t = (tree_node *)handle;
515 search_cache * c = (search_cache *)(t->data.ptr);
516 expiring_data * e = NULL;       /* compiler quietening */
517 uschar * data = NULL;
518 int search_type = t->name[0] - '0';
519 int old_pool = store_pool;
520
521 /* Lookups that return DEFER may not always set an error message. So that
522 the callers don't have to test for NULL, set an empty string. */
523
524 search_error_message = US"";
525 f.search_find_defer = FALSE;
526
527 DEBUG(D_lookup) debug_printf_indent("internal_search_find: file=\"%s\"\n  "
528   "type=%s key=\"%s\" opts=%s%s%s\n", filename,
529   lookup_list[search_type]->name, keystring,
530   opts ? "\"" : "", opts, opts ? "\"" : "");
531
532 /* Insurance. If the keystring is empty, just fail. */
533
534 if (keystring[0] == 0) return NULL;
535
536 /* Use the special store pool for search data */
537
538 store_pool = POOL_SEARCH;
539
540 /* Look up the data for the key, unless it is already in the cache for this
541 file. No need to check c->item_cache for NULL, tree_search will do so. Check
542 whether we want to use the cache entry last so that we can always replace it. */
543
544 if (  (t = tree_search(c->item_cache, keystring))
545    && (!(e = t->data.ptr)->expiry || e->expiry > time(NULL))
546    && (!opts && !e->opts  ||  opts && e->opts && Ustrcmp(opts, e->opts) == 0)
547    && cache_rd
548    )
549   { /* Data was in the cache already; set the pointer from the tree node */
550   data = e->data.ptr;
551   DEBUG(D_lookup) debug_printf_indent("cached data used for lookup of %s%s%s\n",
552     keystring,
553     filename ? US"\n  in " : US"", filename ? filename : US"");
554   }
555 else
556   {
557   uint do_cache = UINT_MAX;
558   int keylength = Ustrlen(keystring);
559
560   DEBUG(D_lookup)
561     {
562     if (t)
563       debug_printf_indent("cached data found but %s; ",
564         e->expiry && e->expiry <= time(NULL) ? "out-of-date"
565         : cache_rd ? "wrong opts" : "no_rd option set");
566     debug_printf_indent("%s lookup required for %s%s%s\n",
567       filename ? US"file" : US"database",
568       keystring,
569       filename ? US"\n  in " : US"", filename ? filename : US"");
570     }
571
572   /* Call the code for the different kinds of search. DEFER is handled
573   like FAIL, except that search_find_defer is set so the caller can
574   distinguish if necessary. */
575
576   if (lookup_list[search_type]->find(c->handle, filename, keystring, keylength,
577       &data, &search_error_message, &do_cache, opts) == DEFER)
578     f.search_find_defer = TRUE;
579
580   /* A record that has been found is now in data, which is either NULL
581   or points to a bit of dynamic store. Cache the result of the lookup if
582   caching is permitted. Lookups can disable caching, when they did something
583   that changes their data. The mysql and pgsql lookups do this when an
584   UPDATE/INSERT query was executed.  Lookups can also set a TTL for the
585   cache entry; the dnsdb lookup does.
586   Finally, the caller can request no caching by setting an option. */
587
588   else if (do_cache)
589     {
590     DEBUG(D_lookup) debug_printf_indent("%s cache entry\n",
591       t ? "replacing old" : "creating new");
592     if (!t)     /* No existing entry.  Create new one. */
593       {
594       int len = keylength + 1;
595       e = store_get(sizeof(expiring_data) + sizeof(tree_node) + len,
596                     is_tainted(keystring));
597       t = (tree_node *)(e+1);
598       memcpy(t->name, keystring, len);
599       t->data.ptr = e;
600       tree_insertnode(&c->item_cache, t);
601       }
602       /* Else previous, out-of-date cache entry.  Update with the */
603       /* new result and forget the old one */
604     e->expiry = do_cache == UINT_MAX ? 0 : time(NULL)+do_cache;
605     e->opts = opts ? string_copy(opts) : NULL;
606     e->data.ptr = data;
607     }
608
609   /* If caching was disabled, empty the cache tree. We just set the cache
610   pointer to NULL here, because we cannot release the store at this stage. */
611
612   else
613     {
614     DEBUG(D_lookup) debug_printf_indent("lookup forced cache cleanup\n");
615     c->item_cache = NULL;       /* forget all lookups on this connection */
616     }
617   }
618
619 DEBUG(D_lookup)
620   {
621   if (data)
622     debug_printf_indent("lookup yielded: %s\n", data);
623   else if (f.search_find_defer)
624     debug_printf_indent("lookup deferred: %s\n", search_error_message);
625   else debug_printf_indent("lookup failed\n");
626   }
627
628 /* Return it in new dynamic store in the regular pool */
629
630 store_pool = old_pool;
631 return data ? string_copy(data) : NULL;
632 }
633
634
635
636
637 /*************************************************
638 * Find one item in database, possibly wildcarded *
639 *************************************************/
640
641 /* This function calls the internal function above; once only if there
642 is no partial matching, but repeatedly when partial matching is requested.
643
644 Arguments:
645   handle         the handle from search_open
646   filename       the filename that was handed to search_open, or
647                    NULL for query-style searches
648   keystring      the keystring for single-key+file lookups, or
649                    the querystring for query-style lookups
650   partial        -1 means no partial matching;
651                    otherwise it's the minimum number of components;
652   affix          the affix string for partial matching
653   affixlen       the length of the affix string
654   starflags      SEARCH_STAR and SEARCH_STARAT flags
655   expand_setup   pointer to offset for setting up expansion strings;
656                  don't do any if < 0
657   opts           type-specific options
658
659 Returns:         a pointer to a dynamic string containing the answer,
660                  or NULL if the query failed or was deferred; in the
661                  latter case, search_find_defer is set TRUE
662 */
663
664 uschar *
665 search_find(void * handle, const uschar * filename, uschar * keystring,
666   int partial, const uschar * affix, int affixlen, int starflags,
667   int * expand_setup, const uschar * opts)
668 {
669 tree_node * t = (tree_node *)handle;
670 BOOL set_null_wild = FALSE, cache_rd = TRUE, ret_key = FALSE;
671 uschar * yield;
672
673 DEBUG(D_lookup)
674   {
675   if (partial < 0) affixlen = 99;   /* So that "NULL" prints */
676   debug_printf_indent("search_find: file=\"%s\"\n  key=\"%s\" "
677     "partial=%d affix=%.*s starflags=%x opts=%s%s%s\n",
678     filename ? filename : US"NULL",
679     keystring, partial, affixlen, affix, starflags,
680     opts ? "\"" : "", opts, opts ? "\"" : "");
681
682   }
683
684 /* Parse global lookup options. Also, create a new options list with
685 the global options dropped so that the cache-modifiers are not
686 used in the cache key. */
687
688 if (opts)
689   {
690   int sep = ',';
691   gstring * g = NULL;
692
693   for (uschar * ele; ele = string_nextinlist(&opts, &sep, NULL, 0); )
694     if (Ustrcmp(ele, "ret=key") == 0) ret_key = TRUE;
695     else if (Ustrcmp(ele, "cache=no_rd") == 0) cache_rd = FALSE;
696     else g = string_append_listele(g, ',', ele);
697
698   opts = string_from_gstring(g);
699   }
700
701 /* Arrange to put this database at the top of the LRU chain if it is a type
702 that opens real files. */
703
704 if (  open_top != (tree_node *)handle 
705    && lookup_list[t->name[0]-'0']->type == lookup_absfile)
706   {
707   search_cache *c = (search_cache *)(t->data.ptr);
708   tree_node *up = c->up;
709   tree_node *down = c->down;
710
711   /* Cut it out of the list. A newly opened file will a NULL up pointer.
712   Otherwise there will be a non-NULL up pointer, since we checked above that
713   this block isn't already at the top of the list. */
714
715   if (up)
716     {
717     ((search_cache *)(up->data.ptr))->down = down;
718     if (down)
719       ((search_cache *)(down->data.ptr))->up = up;
720     else
721       open_bot = up;
722     }
723
724   /* Now put it at the head of the list. */
725
726   c->up = NULL;
727   c->down = open_top;
728   if (!open_top) open_bot = t;
729   else ((search_cache *)(open_top->data.ptr))->up = t;
730   open_top = t;
731   }
732
733 DEBUG(D_lookup)
734   {
735   debug_printf_indent("LRU list:\n");
736   for (tree_node *t = open_top; t; )
737     {
738     search_cache *c = (search_cache *)(t->data.ptr);
739     debug_printf_indent("  %s\n", t->name);
740     if (t == open_bot) debug_printf_indent("  End\n");
741     t = c->down;
742     }
743   }
744
745 /* First of all, try to match the key string verbatim. If matched a complete
746 entry but could have been partial, flag to set up variables. */
747
748 yield = internal_search_find(handle, filename, keystring, cache_rd, opts);
749 if (f.search_find_defer) return NULL;
750
751 if (yield) { if (partial >= 0) set_null_wild = TRUE; }
752
753 /* Not matched a complete entry; handle partial lookups, but only if the full
754 search didn't defer. Don't use string_sprintf() to construct the initial key,
755 just in case the original key is too long for the string_sprintf() buffer (it
756 *has* happened!). The case of a zero-length affix has to be treated specially.
757 */
758
759 else if (partial >= 0)
760   {
761   int len = Ustrlen(keystring);
762   uschar *keystring2;
763
764   /* Try with the affix on the front, except for a zero-length affix */
765
766   if (affixlen == 0) keystring2 = keystring; else
767     {
768     keystring2 = store_get(len + affixlen + 1,
769                         is_tainted(keystring) || is_tainted(affix));
770     Ustrncpy(keystring2, affix, affixlen);
771     Ustrcpy(keystring2 + affixlen, keystring);
772     DEBUG(D_lookup) debug_printf_indent("trying partial match %s\n", keystring2);
773     yield = internal_search_find(handle, filename, keystring2, cache_rd, opts);
774     if (f.search_find_defer) return NULL;
775     }
776
777   /* The key in its entirety did not match a wild entry; try chopping off
778   leading components. */
779
780   if (!yield)
781     {
782     int dotcount = 0;
783     uschar *keystring3 = keystring2 + affixlen;
784     uschar *s = keystring3;
785     while (*s != 0) if (*s++ == '.') dotcount++;
786
787     while (dotcount-- >= partial)
788       {
789       while (*keystring3 != 0 && *keystring3 != '.') keystring3++;
790
791       /* If we get right to the end of the string (which will be the last time
792       through this loop), we've failed if the affix is null. Otherwise do one
793       last lookup for the affix itself, but if it is longer than 1 character,
794       remove the last character if it is ".". */
795
796       if (*keystring3 == 0)
797         {
798         if (affixlen < 1) break;
799         if (affixlen > 1 && affix[affixlen-1] == '.') affixlen--;
800         Ustrncpy(keystring2, affix, affixlen);
801         keystring2[affixlen] = 0;
802         keystring3 = keystring2;
803         }
804       else
805         {
806         keystring3 -= affixlen - 1;
807         if (affixlen > 0) Ustrncpy(keystring3, affix, affixlen);
808         }
809
810       DEBUG(D_lookup) debug_printf_indent("trying partial match %s\n", keystring3);
811       yield = internal_search_find(handle, filename, keystring3,
812                 cache_rd, opts);
813       if (f.search_find_defer) return NULL;
814       if (yield)
815         {
816         /* First variable is the wild part; second is the fixed part. Take care
817         to get it right when keystring3 is just "*". */
818
819         if (expand_setup && *expand_setup >= 0)
820           {
821           int fixedlength = Ustrlen(keystring3) - affixlen;
822           int wildlength = Ustrlen(keystring) - fixedlength - 1;
823           *expand_setup += 1;
824           expand_nstring[*expand_setup] = keystring;
825           expand_nlength[*expand_setup] = wildlength;
826           *expand_setup += 1;
827           expand_nstring[*expand_setup] = keystring + wildlength + 1;
828           expand_nlength[*expand_setup] = (fixedlength < 0)? 0 : fixedlength;
829           }
830         break;
831         }
832       keystring3 += affixlen;
833       }
834     }
835
836   else set_null_wild = TRUE; /* Matched a wild entry without any wild part */
837   }
838
839 /* If nothing has been matched, but the option to look for "*@" is set, try
840 replacing everything to the left of @ by *. After a match, the wild part
841 is set to the string to the left of the @. */
842
843 if (!yield  &&  starflags & SEARCH_STARAT)
844   {
845   uschar *atat = Ustrrchr(keystring, '@');
846   if (atat != NULL && atat > keystring)
847     {
848     int savechar;
849     savechar = *(--atat);
850     *atat = '*';
851
852     DEBUG(D_lookup) debug_printf_indent("trying default match %s\n", atat);
853     yield = internal_search_find(handle, filename, atat, cache_rd, opts);
854     *atat = savechar;
855     if (f.search_find_defer) return NULL;
856
857     if (yield && expand_setup && *expand_setup >= 0)
858       {
859       *expand_setup += 1;
860       expand_nstring[*expand_setup] = keystring;
861       expand_nlength[*expand_setup] = atat - keystring + 1;
862       *expand_setup += 1;
863       expand_nstring[*expand_setup] = keystring;
864       expand_nlength[*expand_setup] = 0;
865       }
866     }
867   }
868
869 /* If we still haven't matched anything, and the option to look for "*" is set,
870 try that. If we do match, the first variable (the wild part) is the whole key,
871 and the second is empty. */
872
873 if (!yield  &&  starflags & (SEARCH_STAR|SEARCH_STARAT))
874   {
875   DEBUG(D_lookup) debug_printf_indent("trying to match *\n");
876   yield = internal_search_find(handle, filename, US"*", cache_rd, opts);
877   if (yield && expand_setup && *expand_setup >= 0)
878     {
879     *expand_setup += 1;
880     expand_nstring[*expand_setup] = keystring;
881     expand_nlength[*expand_setup] = Ustrlen(keystring);
882     *expand_setup += 1;
883     expand_nstring[*expand_setup] = keystring;
884     expand_nlength[*expand_setup] = 0;
885     }
886   }
887
888 /* If this was a potentially partial lookup, and we matched either a
889 complete non-wild domain entry, or we matched a wild-carded entry without
890 chopping off any of the domain components, set up the expansion variables
891 (if required) so that the first one is empty, and the second one is the
892 fixed part of the domain. The set_null_wild flag is set only when yield is not
893 NULL. */
894
895 if (set_null_wild && expand_setup && *expand_setup >= 0)
896   {
897   *expand_setup += 1;
898   expand_nstring[*expand_setup] = keystring;
899   expand_nlength[*expand_setup] = 0;
900   *expand_setup += 1;
901   expand_nstring[*expand_setup] = keystring;
902   expand_nlength[*expand_setup] = Ustrlen(keystring);
903   }
904
905 /* If we have a result, check the options to see if the key was wanted rather
906 than the result.  Return a de-tainted version of the key on the grounds that
907 it have been validated by the lookup. */
908
909 if (yield && ret_key)
910   yield = string_copy_taint(keystring, FALSE);
911
912 return yield;
913 }
914
915 /* End of search.c */