97bff0a8c02528a8918e773a88f4247f2ebe8ff4
[users/jgh/exim.git] / src / src / routers / iplookup.c
1 /* $Cambridge: exim/src/src/routers/iplookup.c,v 1.10 2007/01/08 10:50:20 ph10 Exp $ */
2
3 /*************************************************
4 *     Exim - an Internet mail transport agent    *
5 *************************************************/
6
7 /* Copyright (c) University of Cambridge 1995 - 2007 */
8 /* See the file NOTICE for conditions of use and distribution. */
9
10
11 #include "../exim.h"
12 #include "rf_functions.h"
13 #include "iplookup.h"
14
15
16 /* IP connection types */
17
18 #define ip_udp 0
19 #define ip_tcp 1
20
21
22 /* Options specific to the iplookup router. */
23
24 optionlist iplookup_router_options[] = {
25   { "hosts",    opt_stringptr,
26       (void *)(offsetof(iplookup_router_options_block, hosts)) },
27   { "optional", opt_bool,
28       (void *)(offsetof(iplookup_router_options_block, optional)) },
29   { "port",     opt_int,
30       (void *)(offsetof(iplookup_router_options_block, port)) },
31   { "protocol", opt_stringptr,
32       (void *)(offsetof(iplookup_router_options_block, protocol_name)) },
33   { "query",    opt_stringptr,
34       (void *)(offsetof(iplookup_router_options_block, query)) },
35   { "reroute",  opt_stringptr,
36       (void *)(offsetof(iplookup_router_options_block, reroute)) },
37   { "response_pattern", opt_stringptr,
38       (void *)(offsetof(iplookup_router_options_block, response_pattern)) },
39   { "timeout",  opt_time,
40       (void *)(offsetof(iplookup_router_options_block, timeout)) }
41 };
42
43 /* Size of the options list. An extern variable has to be used so that its
44 address can appear in the tables drtables.c. */
45
46 int iplookup_router_options_count =
47   sizeof(iplookup_router_options)/sizeof(optionlist);
48
49 /* Default private options block for the iplookup router. */
50
51 iplookup_router_options_block iplookup_router_option_defaults = {
52   -1,       /* port */
53   ip_udp,   /* protocol */
54   5,        /* timeout */
55   NULL,     /* protocol_name */
56   NULL,     /* hosts */
57   NULL,     /* query; NULL => local_part@domain */
58   NULL,     /* response_pattern; NULL => don't apply regex */
59   NULL,     /* reroute; NULL => just used returned data */
60   NULL,     /* re_response_pattern; compiled pattern */
61   FALSE     /* optional */
62 };
63
64
65
66 /*************************************************
67 *          Initialization entry point            *
68 *************************************************/
69
70 /* Called for each instance, after its options have been read, to enable
71 consistency checks to be done, or anything else that needs to be set up. */
72
73 void
74 iplookup_router_init(router_instance *rblock)
75 {
76 iplookup_router_options_block *ob =
77   (iplookup_router_options_block *)(rblock->options_block);
78
79 /* A port and a host list must be given */
80
81 if (ob->port < 0)
82   log_write(0, LOG_PANIC_DIE|LOG_CONFIG_FOR, "%s router:\n  "
83     "a port must be specified", rblock->name);
84
85 if (ob->hosts == NULL)
86   log_write(0, LOG_PANIC_DIE|LOG_CONFIG_FOR, "%s router:\n  "
87     "a host list must be specified", rblock->name);
88
89 /* Translate protocol name into value */
90
91 if (ob->protocol_name != NULL)
92   {
93   if (Ustrcmp(ob->protocol_name, "udp") == 0) ob->protocol = ip_udp;
94   else if (Ustrcmp(ob->protocol_name, "tcp") == 0) ob->protocol = ip_tcp;
95   else log_write(0, LOG_PANIC_DIE|LOG_CONFIG_FOR, "%s router:\n  "
96     "protocol not specified as udp or tcp", rblock->name);
97   }
98
99 /* If a response pattern is given, compile it now to get the error early. */
100
101 if (ob->response_pattern != NULL)
102   ob->re_response_pattern =
103     regex_must_compile(ob->response_pattern, FALSE, TRUE);
104 }
105
106
107
108 /*************************************************
109 *              Main entry point                  *
110 *************************************************/
111
112 /* See local README for interface details. This router returns:
113
114 DECLINE
115   . pattern or identification match on returned data failed
116
117 DEFER
118   . failed to expand the query or rerouting string
119   . failed to create socket ("optional" not set)
120   . failed to find a host, failed to connect, timed out ("optional" not set)
121   . rerouting string is not in the form localpart@domain
122   . verifying the errors address caused a deferment or a big disaster such
123       as an expansion failure (rf_get_errors_address)
124   . expanding a headers_{add,remove} string caused a deferment or another
125       expansion error (rf_get_munge_headers)
126
127 PASS
128   . failed to create socket ("optional" set)
129   . failed to find a host, failed to connect, timed out ("optional" set)
130
131 OK
132   . new address added to addr_new
133 */
134
135 int
136 iplookup_router_entry(
137   router_instance *rblock,        /* data for this instantiation */
138   address_item *addr,             /* address we are working on */
139   struct passwd *pw,              /* passwd entry after check_local_user */
140   int verify,                     /* v_none/v_recipient/v_sender/v_expn */
141   address_item **addr_local,      /* add it to this if it's local */
142   address_item **addr_remote,     /* add it to this if it's remote */
143   address_item **addr_new,        /* put new addresses on here */
144   address_item **addr_succeed)    /* put old address here on success */
145 {
146 uschar *query = NULL;
147 uschar reply[256];
148 uschar *hostname, *reroute, *domain, *listptr;
149 uschar host_buffer[256];
150 host_item *host = store_get(sizeof(host_item));
151 address_item *new_addr;
152 iplookup_router_options_block *ob =
153   (iplookup_router_options_block *)(rblock->options_block);
154 const pcre *re = ob->re_response_pattern;
155 int count, query_len, rc;
156 int sep = 0;
157
158 addr_local = addr_local;    /* Keep picky compilers happy */
159 addr_remote = addr_remote;
160 addr_succeed = addr_succeed;
161 pw = pw;
162
163 DEBUG(D_route) debug_printf("%s router called for %s: domain = %s\n",
164   rblock->name, addr->address, addr->domain);
165
166 /* Build the query string to send. If not explicitly given, a default of
167 "user@domain user@domain" is used. */
168
169 if (ob->query == NULL)
170   query = string_sprintf("%s@%s %s@%s", addr->local_part, addr->domain,
171     addr->local_part, addr->domain);
172 else
173   {
174   query = expand_string(ob->query);
175   if (query == NULL)
176     {
177     addr->message = string_sprintf("%s router: failed to expand %s: %s",
178       rblock->name, ob->query, expand_string_message);
179     return DEFER;
180     }
181   }
182
183 query_len = Ustrlen(query);
184 DEBUG(D_route) debug_printf("%s router query is \"%s\"\n", rblock->name,
185   string_printing(query));
186
187 /* Now connect to the required port for each of the hosts in turn, until a
188 response it received. Initialization insists on the port being set and there
189 being a host list. */
190
191 listptr = ob->hosts;
192 while ((hostname = string_nextinlist(&listptr, &sep, host_buffer,
193        sizeof(host_buffer))) != NULL)
194   {
195   host_item *h;
196
197   DEBUG(D_route) debug_printf("calling host %s\n", hostname);
198
199   host->name = hostname;
200   host->address = NULL;
201   host->port = PORT_NONE;
202   host->mx = MX_NONE;
203   host->next = NULL;
204
205   if (string_is_ip_address(host->name, NULL) != 0)
206     host->address = host->name;
207   else
208     {
209     int rc = host_find_byname(host, NULL, HOST_FIND_QUALIFY_SINGLE, NULL, TRUE);
210     if (rc == HOST_FIND_FAILED || rc == HOST_FIND_AGAIN) continue;
211     }
212
213   /* Loop for possible multiple IP addresses for the given name. */
214
215   for (h = host; h != NULL; h = h->next)
216     {
217     int host_af, query_socket;
218
219     /* Skip any hosts for which we have no address */
220
221     if (h->address == NULL) continue;
222
223     /* Create a socket, for UDP or TCP, as configured. IPv6 addresses are
224     detected by checking for a colon in the address. */
225
226     host_af = (Ustrchr(h->address, ':') != NULL)? AF_INET6 : AF_INET;
227     query_socket = ip_socket((ob->protocol == ip_udp)? SOCK_DGRAM:SOCK_STREAM,
228       host_af);
229     if (query_socket < 0)
230       {
231       if (ob->optional) return PASS;
232       addr->message = string_sprintf("failed to create socket in %s router",
233         rblock->name);
234       return DEFER;
235       }
236
237     /* Connect to the remote host, under a timeout. In fact, timeouts can occur
238     here only for TCP calls; for a UDP socket, "connect" always works (the
239     router will timeout later on the read call). */
240
241     if (ip_connect(query_socket, host_af, h->address,ob->port, ob->timeout) < 0)
242       {
243       close(query_socket);
244       DEBUG(D_route)
245         debug_printf("connection to %s failed: %s\n", h->address,
246           strerror(errno));
247       continue;
248       }
249
250     /* Send the query. If it fails, just continue with the next address. */
251
252     if (send(query_socket, query, query_len, 0) < 0)
253       {
254       DEBUG(D_route) debug_printf("send to %s failed\n", h->address);
255       (void)close(query_socket);
256       continue;
257       }
258
259     /* Read the response and close the socket. If the read fails, try the
260     next IP address. */
261
262     count = ip_recv(query_socket, reply, sizeof(reply) - 1, ob->timeout);
263     (void)close(query_socket);
264     if (count <= 0)
265       {
266       DEBUG(D_route) debug_printf("%s from %s\n", (errno == ETIMEDOUT)?
267         "timed out" : "recv failed", h->address);
268       *reply = 0;
269       continue;
270       }
271
272     /* Success; break the loop */
273
274     reply[count] = 0;
275     DEBUG(D_route) debug_printf("%s router received \"%s\" from %s\n",
276       rblock->name, string_printing(reply), h->address);
277     break;
278     }
279
280   /* If h == NULL we have tried all the IP addresses and failed on all of them,
281   so we must continue to try more host names. Otherwise we have succeeded. */
282
283   if (h != NULL) break;
284   }
285
286
287 /* If hostname is NULL, we have failed to find any host, or failed to
288 connect to any of the IP addresses, or timed out while reading or writing to
289 those we have connected to. In all cases, we must pass if optional and
290 defer otherwise. */
291
292 if (hostname == NULL)
293   {
294   DEBUG(D_route) debug_printf("%s router failed to get anything\n", rblock->name);
295   if (ob->optional) return PASS;
296   addr->message = string_sprintf("%s router: failed to communicate with any "
297     "host", rblock->name);
298   return DEFER;
299   }
300
301
302 /* If a response pattern was supplied, match the returned string against it. A
303 failure to match causes the router to decline. After a successful match, the
304 numerical variables for expanding the rerouted address are set up. */
305
306 if (re != NULL)
307   {
308   if (!regex_match_and_setup(re, reply, 0, -1))
309     {
310     DEBUG(D_route) debug_printf("%s router: %s failed to match response %s\n",
311       rblock->name, ob->response_pattern, reply);
312     return DECLINE;
313     }
314   }
315
316
317 /* If no response pattern was supplied, set up $0 as the response up to the
318 first white space (if any). Also, if no query was specified, check that what
319 follows the white space matches user@domain. */
320
321 else
322   {
323   int n = 0;
324   while (reply[n] != 0 && !isspace(reply[n])) n++;
325   expand_nmax = 0;
326   expand_nstring[0] = reply;
327   expand_nlength[0] = n;
328
329   if (ob->query == NULL)
330     {
331     int nn = n;
332     while (isspace(reply[nn])) nn++;
333     if (Ustrcmp(query + query_len/2 + 1, reply+nn) != 0)
334       {
335       DEBUG(D_route) debug_printf("%s router: failed to match identification "
336         "in response %s\n", rblock->name, reply);
337       return DECLINE;
338       }
339     }
340
341   reply[n] = 0;  /* Terminate for the default case */
342   }
343
344 /* If an explicit rerouting string is specified, expand it. Otherwise, use
345 what was sent back verbatim. */
346
347 if (ob->reroute != NULL)
348   {
349   reroute = expand_string(ob->reroute);
350   expand_nmax = -1;
351   if (reroute == NULL)
352     {
353     addr->message = string_sprintf("%s router: failed to expand %s: %s",
354       rblock->name, ob->reroute, expand_string_message);
355     return DEFER;
356     }
357   }
358 else reroute = reply;
359
360 /* We should now have a new address in the form user@domain. */
361
362 domain = Ustrchr(reroute, '@');
363 if (domain == NULL)
364   {
365   log_write(0, LOG_MAIN, "%s router: reroute string %s is not of the form "
366     "user@domain", rblock->name, reroute);
367   addr->message = string_sprintf("%s router: reroute string %s is not of the "
368     "form user@domain", rblock->name, reroute);
369   return DEFER;
370   }
371
372 /* Create a child address with the old one as parent. Put the new address on
373 the chain of new addressess. */
374
375 new_addr = deliver_make_addr(reroute, TRUE);
376 new_addr->parent = addr;
377
378 copyflag(new_addr, addr, af_propagate);
379 new_addr->p = addr->p;
380
381 if (addr->child_count == SHRT_MAX)
382   log_write(0, LOG_MAIN|LOG_PANIC_DIE, "%s router generated more than %d "
383     "child addresses for <%s>", rblock->name, SHRT_MAX, addr->address);
384 addr->child_count++;
385 new_addr->next = *addr_new;
386 *addr_new = new_addr;
387
388 /* Set up the errors address, if any, and the additional and removeable headers
389 for this new address. */
390
391 rc = rf_get_errors_address(addr, rblock, verify, &(new_addr->p.errors_address));
392 if (rc != OK) return rc;
393
394 rc = rf_get_munge_headers(addr, rblock, &(new_addr->p.extra_headers),
395   &(new_addr->p.remove_headers));
396 if (rc != OK) return rc;
397
398 return OK;
399 }
400
401 /* End of routers/iplookup.c */