1 /******************************************************************************/
2 /* pfixtools: a collection of postfix related tools */
4 /* ________________________________________________________________________ */
6 /* Redistribution and use in source and binary forms, with or without */
7 /* modification, are permitted provided that the following conditions */
10 /* 1. Redistributions of source code must retain the above copyright */
11 /* notice, this list of conditions and the following disclaimer. */
12 /* 2. Redistributions in binary form must reproduce the above copyright */
13 /* notice, this list of conditions and the following disclaimer in the */
14 /* documentation and/or other materials provided with the distribution. */
15 /* 3. The names of its contributors may not be used to endorse or promote */
16 /* products derived from this software without specific prior written */
19 /* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND */
20 /* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE */
21 /* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR */
22 /* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS */
23 /* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR */
24 /* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF */
25 /* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS */
26 /* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN */
27 /* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) */
28 /* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF */
29 /* THE POSSIBILITY OF SUCH DAMAGE. */
30 /******************************************************************************/
33 * Copyright © 2008 Florent Bruneau
43 typedef struct rbl_context_t {
45 rbl_result_callback_t call;
50 static struct ub_ctx *ctx = NULL;
51 static server_t *async_event = NULL;
52 static PA(rbl_context_t) ctx_pool = ARRAY_INIT;
54 static rbl_context_t *rbl_context_new(void)
56 return p_new(rbl_context_t, 1);
59 static void rbl_context_delete(rbl_context_t **context)
66 static void rbl_context_wipe(rbl_context_t *context)
71 static rbl_context_t *rbl_context_acquire(void)
73 if (array_len(ctx_pool) > 0) {
74 return array_pop_last(ctx_pool);
76 return rbl_context_new();
80 static void rbl_context_release(rbl_context_t *context)
82 rbl_context_wipe(context);
83 array_add(ctx_pool, context);
86 static void rbl_exit(void)
92 if (async_event != NULL) {
94 server_release(async_event);
97 array_deep_wipe(ctx_pool, rbl_context_delete);
99 module_exit(rbl_exit);
101 static void rbl_callback(void *arg, int err, struct ub_result *result)
103 rbl_context_t *context = arg;
105 debug("asynchronous request led to an error");
106 *context->result = RBL_ERROR;
107 } else if (result->nxdomain) {
108 debug("asynchronous request done, %s NOT FOUND", result->qname);
109 *context->result = RBL_NOTFOUND;
111 debug("asynchronous request done, %s FOUND", result->qname);
112 *context->result = RBL_FOUND;
114 if (context->call != NULL) {
115 debug("calling callback");
116 context->call(context->result, context->data);
118 debug("no callback defined");
120 ub_resolve_free(result);
121 rbl_context_release(context);
124 static int rbl_handler(server_t *event, void *config)
127 debug("rbl_handler called: ub_fd triggered");
129 if ((retval = ub_process(ctx)) != 0) {
130 err("error in DNS resolution: %s", ub_strerror(retval));
136 static inline bool rbl_dns_check(const char *hostname, rbl_result_t *result,
137 rbl_result_callback_t callback, void *data)
140 ctx = ub_ctx_create();
141 ub_ctx_async(ctx, true);
142 if ((async_event = server_register(ub_fd(ctx), rbl_handler, NULL)) == NULL) {
143 crit("cannot register asynchronous DNS event handler");
147 rbl_context_t *context = rbl_context_acquire();
148 context->result = result;
149 context->call = callback;
150 context->data = data;
151 debug("running dns resolution on %s", hostname);
152 if (ub_resolve_async(ctx, (char*)hostname, 1, 1, context, rbl_callback, NULL) == 0) {
157 rbl_context_release(context);
162 bool rbl_check(const char *rbl, uint32_t ip, rbl_result_t *result,
163 rbl_result_callback_t callback, void *data)
168 len = snprintf(host, 257, "%d.%d.%d.%d.%s.",
169 ip & 0xff, (ip >> 8) & 0xff, (ip >> 16) & 0xff, (ip >> 24) & 0xff,
171 if (len >= (int)sizeof(host))
173 if (host[len - 2] == '.')
174 host[len - 1] = '\0';
175 return rbl_dns_check(host, result, callback, data);
178 bool rhbl_check(const char *rhbl, const char *hostname, rbl_result_t *result,
179 rbl_result_callback_t callback, void *data)
184 len = snprintf(host, 257, "%s.%s.", hostname, rhbl);
185 if (len >= (int)sizeof(host))
187 if (host[len - 2] == '.')
188 host[len - 1] = '\0';
189 return rbl_dns_check(host, result, callback, data);