diff options
Diffstat (limited to 'test/rspamd_dns_test.c')
-rw-r--r-- | test/rspamd_dns_test.c | 83 |
1 files changed, 41 insertions, 42 deletions
diff --git a/test/rspamd_dns_test.c b/test/rspamd_dns_test.c index 336e58bfb..d77919342 100644 --- a/test/rspamd_dns_test.c +++ b/test/rspamd_dns_test.c @@ -12,93 +12,92 @@ extern struct ev_loop *event_loop; struct rspamd_dns_resolver *resolver; gboolean -session_fin (gpointer unused) +session_fin(gpointer unused) { - ev_break (event_loop, EVBREAK_ALL); + ev_break(event_loop, EVBREAK_ALL); return TRUE; } static void -test_dns_cb (struct rdns_reply *reply, gpointer arg) +test_dns_cb(struct rdns_reply *reply, gpointer arg) { struct rdns_reply_entry *cur; const struct rdns_request_name *name = - rdns_request_get_name (reply->request, NULL); + rdns_request_get_name(reply->request, NULL); - msg_debug ("got reply with code %s for request %s", - rdns_strerror (reply->code), name->name); + msg_debug("got reply with code %s for request %s", + rdns_strerror(reply->code), name->name); if (reply->code == RDNS_RC_NOERROR) { cur = reply->entries; while (cur) { switch (cur->type) { case RDNS_REQUEST_A: - msg_debug ("got ip: %s", inet_ntoa (cur->content.a.addr)); + msg_debug("got ip: %s", inet_ntoa(cur->content.a.addr)); break; case RDNS_REQUEST_PTR: - msg_debug ("got name %s", cur->content.ptr.name); + msg_debug("got name %s", cur->content.ptr.name); break; case RDNS_REQUEST_TXT: - msg_debug ("got txt %s", cur->content.txt.data); + msg_debug("got txt %s", cur->content.txt.data); break; case RDNS_REQUEST_SPF: - msg_debug ("got spf %s", cur->content.txt.data); + msg_debug("got spf %s", cur->content.txt.data); break; case RDNS_REQUEST_SRV: - msg_debug ("got srv pri: %d, weight: %d, port: %d, target: %s", cur->content.srv.weight, - cur->content.srv.priority, cur->content.srv.port, cur->content.srv.target); + msg_debug("got srv pri: %d, weight: %d, port: %d, target: %s", cur->content.srv.weight, + cur->content.srv.priority, cur->content.srv.port, cur->content.srv.target); break; case RDNS_REQUEST_MX: - msg_debug ("got mx %s:%d", cur->content.mx.name, cur->content.mx.priority); + msg_debug("got mx %s:%d", cur->content.mx.name, cur->content.mx.priority); break; } cur = cur->next; } } - if (-- requests == 0) { - session_fin (NULL); + if (--requests == 0) { + session_fin(NULL); } } -void -rspamd_dns_test_func (void) +void rspamd_dns_test_func(void) { struct rspamd_config *cfg; rspamd_mempool_t *pool; struct rspamd_async_session *s; - cfg = (struct rspamd_config *)g_malloc (sizeof (struct rspamd_config)); - bzero (cfg, sizeof (struct rspamd_config)); - cfg->cfg_pool = rspamd_mempool_new (rspamd_mempool_suggest_size (), NULL, 0); + cfg = (struct rspamd_config *) g_malloc(sizeof(struct rspamd_config)); + bzero(cfg, sizeof(struct rspamd_config)); + cfg->cfg_pool = rspamd_mempool_new(rspamd_mempool_suggest_size(), NULL, 0); cfg->dns_retransmits = 2; cfg->dns_timeout = 0.5; - pool = rspamd_mempool_new (rspamd_mempool_suggest_size (), NULL, 0); + pool = rspamd_mempool_new(rspamd_mempool_suggest_size(), NULL, 0); - s = rspamd_session_create (pool, session_fin, NULL, NULL, NULL); + s = rspamd_session_create(pool, session_fin, NULL, NULL, NULL); - resolver = rspamd_dns_resolver_init (NULL, event_loop, cfg); + resolver = rspamd_dns_resolver_init(NULL, event_loop, cfg); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_A, "google.com")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_PTR, "81.19.70.3")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_MX, "rambler.ru")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "rambler.ru")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "google.com")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_SPF, "rambler.ru")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_SRV, - "_xmpp-server._tcp.jabber.org")); - requests ++; - g_assert (rspamd_dns_resolver_request (resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "non-existent.arpa")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_A, "google.com")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_PTR, "81.19.70.3")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_MX, "rambler.ru")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "rambler.ru")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "google.com")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_SPF, "rambler.ru")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_SRV, + "_xmpp-server._tcp.jabber.org")); + requests++; + g_assert(rspamd_dns_resolver_request(resolver, s, pool, test_dns_cb, NULL, RDNS_REQUEST_TXT, "non-existent.arpa")); - g_assert (resolver != NULL); + g_assert(resolver != NULL); - ev_run (event_loop, 0); + ev_run(event_loop, 0); } |