aboutsummaryrefslogtreecommitdiffstats
path: root/test/rspamd_upstream_test.c
diff options
context:
space:
mode:
authorVsevolod Stakhov <vsevolod@rspamd.com>2023-07-26 10:49:23 +0100
committerVsevolod Stakhov <vsevolod@rspamd.com>2023-07-26 10:49:23 +0100
commit537a7180a0d5132c11636c4fd8b1450cd99d352c (patch)
treefb9f8c84955a411bdffbd6371ea32f2716fb3687 /test/rspamd_upstream_test.c
parent5fd7a90fdaa33f52c59bdb0ca84451e5c1e22365 (diff)
downloadrspamd-537a7180a0d5132c11636c4fd8b1450cd99d352c.tar.gz
rspamd-537a7180a0d5132c11636c4fd8b1450cd99d352c.zip
[Rework] Use clang-format to unify formatting in all sources
No meaningful changes.
Diffstat (limited to 'test/rspamd_upstream_test.c')
-rw-r--r--test/rspamd_upstream_test.c171
1 files changed, 85 insertions, 86 deletions
diff --git a/test/rspamd_upstream_test.c b/test/rspamd_upstream_test.c
index 263a28e75..1f4e27daa 100644
--- a/test/rspamd_upstream_test.c
+++ b/test/rspamd_upstream_test.c
@@ -26,34 +26,33 @@ char test_key[32];
extern struct ev_loop *event_loop;
static void
-rspamd_upstream_test_method (struct upstream_list *ls,
- enum rspamd_upstream_rotation rot, const gchar *expected)
+rspamd_upstream_test_method(struct upstream_list *ls,
+ enum rspamd_upstream_rotation rot, const gchar *expected)
{
struct upstream *up;
if (rot != RSPAMD_UPSTREAM_HASHED) {
- up = rspamd_upstream_get (ls, rot, NULL, 0);
- g_assert (up != NULL);
- g_assert (strcmp (rspamd_upstream_name (up), expected) == 0);
+ up = rspamd_upstream_get(ls, rot, NULL, 0);
+ g_assert(up != NULL);
+ g_assert(strcmp(rspamd_upstream_name(up), expected) == 0);
}
else {
- up = rspamd_upstream_get (ls, RSPAMD_UPSTREAM_HASHED, test_key,
- sizeof (test_key));
- g_assert (up != NULL);
- g_assert (strcmp (rspamd_upstream_name (up), expected) == 0);
+ up = rspamd_upstream_get(ls, RSPAMD_UPSTREAM_HASHED, test_key,
+ sizeof(test_key));
+ g_assert(up != NULL);
+ g_assert(strcmp(rspamd_upstream_name(up), expected) == 0);
}
}
static void
-rspamd_upstream_timeout_handler (EV_P_ ev_timer *w, int revents)
+rspamd_upstream_timeout_handler(EV_P_ ev_timer *w, int revents)
{
- struct rspamd_dns_resolver *resolver = (struct rspamd_dns_resolver *)w->data;
+ struct rspamd_dns_resolver *resolver = (struct rspamd_dns_resolver *) w->data;
- rdns_resolver_release (resolver->r);
+ rdns_resolver_release(resolver->r);
}
-void
-rspamd_upstream_test_func (void)
+void rspamd_upstream_test_func(void)
{
struct upstream_list *ls, *nls;
struct upstream *up, *upn;
@@ -65,119 +64,119 @@ rspamd_upstream_test_func (void)
static ev_timer ev;
rspamd_inet_addr_t *addr, *next_addr, *paddr;
- cfg = rspamd_config_new (RSPAMD_CONFIG_INIT_SKIP_LUA);
+ cfg = rspamd_config_new(RSPAMD_CONFIG_INIT_SKIP_LUA);
cfg->dns_retransmits = 2;
cfg->dns_timeout = 0.5;
cfg->upstream_max_errors = 1;
cfg->upstream_revive_time = 0.5;
cfg->upstream_error_time = 2;
- resolver = rspamd_dns_resolver_init (NULL, event_loop, cfg);
- rspamd_upstreams_library_config (cfg, cfg->ups_ctx, event_loop, resolver->r);
+ resolver = rspamd_dns_resolver_init(NULL, event_loop, cfg);
+ rspamd_upstreams_library_config(cfg, cfg->ups_ctx, event_loop, resolver->r);
/*
* Test v4/v6 priorities
*/
- nls = rspamd_upstreams_create (cfg->ups_ctx);
- g_assert (rspamd_upstreams_add_upstream (nls, "127.0.0.1", 0,
- RSPAMD_UPSTREAM_PARSE_DEFAULT,
- NULL));
- up = rspamd_upstream_get (nls, RSPAMD_UPSTREAM_RANDOM, NULL, 0);
- rspamd_parse_inet_address (&paddr, "127.0.0.2", strlen ("127.0.0.2"),
- RSPAMD_INET_ADDRESS_PARSE_DEFAULT);
- g_assert (rspamd_upstream_add_addr (up, paddr));
- rspamd_parse_inet_address (&paddr, "::1", strlen ("::1"),
- RSPAMD_INET_ADDRESS_PARSE_DEFAULT);
- g_assert (rspamd_upstream_add_addr (up, paddr));
+ nls = rspamd_upstreams_create(cfg->ups_ctx);
+ g_assert(rspamd_upstreams_add_upstream(nls, "127.0.0.1", 0,
+ RSPAMD_UPSTREAM_PARSE_DEFAULT,
+ NULL));
+ up = rspamd_upstream_get(nls, RSPAMD_UPSTREAM_RANDOM, NULL, 0);
+ rspamd_parse_inet_address(&paddr, "127.0.0.2", strlen("127.0.0.2"),
+ RSPAMD_INET_ADDRESS_PARSE_DEFAULT);
+ g_assert(rspamd_upstream_add_addr(up, paddr));
+ rspamd_parse_inet_address(&paddr, "::1", strlen("::1"),
+ RSPAMD_INET_ADDRESS_PARSE_DEFAULT);
+ g_assert(rspamd_upstream_add_addr(up, paddr));
/* Rewind to start */
- addr = rspamd_upstream_addr_next (up);
- addr = rspamd_upstream_addr_next (up);
+ addr = rspamd_upstream_addr_next(up);
+ addr = rspamd_upstream_addr_next(up);
/* cur should be zero here */
- addr = rspamd_upstream_addr_next (up);
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (addr) == AF_INET);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET);
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET6);
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET);
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET);
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET6);
+ addr = rspamd_upstream_addr_next(up);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(addr) == AF_INET);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET6);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET6);
/* Test errors with IPv6 */
- rspamd_upstream_fail (up, TRUE, NULL);
+ rspamd_upstream_fail(up, TRUE, NULL);
/* Now we should have merely IPv4 addresses in rotation */
- addr = rspamd_upstream_addr_next (up);
+ addr = rspamd_upstream_addr_next(up);
for (i = 0; i < 256; i++) {
- next_addr = rspamd_upstream_addr_next (up);
- g_assert (rspamd_inet_address_get_af (addr) == AF_INET);
- g_assert (rspamd_inet_address_get_af (next_addr) == AF_INET);
- g_assert (rspamd_inet_address_compare (addr, next_addr, FALSE) != 0);
+ next_addr = rspamd_upstream_addr_next(up);
+ g_assert(rspamd_inet_address_get_af(addr) == AF_INET);
+ g_assert(rspamd_inet_address_get_af(next_addr) == AF_INET);
+ g_assert(rspamd_inet_address_compare(addr, next_addr, FALSE) != 0);
addr = next_addr;
}
- rspamd_upstreams_destroy (nls);
+ rspamd_upstreams_destroy(nls);
- ls = rspamd_upstreams_create (cfg->ups_ctx);
- g_assert (rspamd_upstreams_parse_line (ls, test_upstream_list, 443, NULL));
- g_assert (rspamd_upstreams_count (ls) == 3);
+ ls = rspamd_upstreams_create(cfg->ups_ctx);
+ g_assert(rspamd_upstreams_parse_line(ls, test_upstream_list, 443, NULL));
+ g_assert(rspamd_upstreams_count(ls) == 3);
/* Test master-slave rotation */
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org");
/* Test round-robin rotation */
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "google.com");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "google.com");
- rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "microsoft.com");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "google.com");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "google.com");
+ rspamd_upstream_test_method(ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "microsoft.com");
/* Test stable hashing */
- nls = rspamd_upstreams_create (cfg->ups_ctx);
- g_assert (rspamd_upstreams_parse_line (nls, test_upstream_list, 443, NULL));
- g_assert (rspamd_upstreams_parse_line (nls, new_upstream_list, 443, NULL));
- for (i = 0; i < assumptions; i ++) {
- ottery_rand_bytes (test_key, sizeof (test_key));
- up = rspamd_upstream_get (ls, RSPAMD_UPSTREAM_HASHED, test_key,
- sizeof (test_key));
- upn = rspamd_upstream_get (nls, RSPAMD_UPSTREAM_HASHED, test_key,
- sizeof (test_key));
-
- if (strcmp (rspamd_upstream_name (up), rspamd_upstream_name (upn)) == 0) {
- success ++;
+ nls = rspamd_upstreams_create(cfg->ups_ctx);
+ g_assert(rspamd_upstreams_parse_line(nls, test_upstream_list, 443, NULL));
+ g_assert(rspamd_upstreams_parse_line(nls, new_upstream_list, 443, NULL));
+ for (i = 0; i < assumptions; i++) {
+ ottery_rand_bytes(test_key, sizeof(test_key));
+ up = rspamd_upstream_get(ls, RSPAMD_UPSTREAM_HASHED, test_key,
+ sizeof(test_key));
+ upn = rspamd_upstream_get(nls, RSPAMD_UPSTREAM_HASHED, test_key,
+ sizeof(test_key));
+
+ if (strcmp(rspamd_upstream_name(up), rspamd_upstream_name(upn)) == 0) {
+ success++;
}
}
- p = 1.0 - fabs (3.0 / 4.0 - (gdouble)success / (gdouble)assumptions);
+ p = 1.0 - fabs(3.0 / 4.0 - (gdouble) success / (gdouble) assumptions);
/*
* P value is calculated as following:
* when we add/remove M upstreams from the list, the probability of hash
* miss should be close to the relation N / (N + M), where N is the size of
* the previous upstreams list.
*/
- msg_debug ("p value for hash consistency: %.6f", p);
- g_assert (p > 0.9);
+ msg_debug("p value for hash consistency: %.6f", p);
+ g_assert(p > 0.9);
- rspamd_upstreams_destroy (nls);
+ rspamd_upstreams_destroy(nls);
/* Upstream fail test */
ev.data = resolver;
- ev_timer_init (&ev, rspamd_upstream_timeout_handler, 2.0, 0.0);
+ ev_timer_init(&ev, rspamd_upstream_timeout_handler, 2.0, 0.0);
- up = rspamd_upstream_get (ls, RSPAMD_UPSTREAM_MASTER_SLAVE, NULL, 0);
- for (i = 0; i < 100; i ++) {
- rspamd_upstream_fail (up, TRUE, NULL);
+ up = rspamd_upstream_get(ls, RSPAMD_UPSTREAM_MASTER_SLAVE, NULL, 0);
+ for (i = 0; i < 100; i++) {
+ rspamd_upstream_fail(up, TRUE, NULL);
}
- g_assert (rspamd_upstreams_alive (ls) == 2);
+ g_assert(rspamd_upstreams_alive(ls) == 2);
- ev_timer_start (event_loop, &ev);
+ ev_timer_start(event_loop, &ev);
- ev_run (event_loop, 0);
- g_assert (rspamd_upstreams_alive (ls) == 3);
+ ev_run(event_loop, 0);
+ g_assert(rspamd_upstreams_alive(ls) == 3);
- rspamd_upstreams_destroy (ls);
- REF_RELEASE (cfg);
+ rspamd_upstreams_destroy(ls);
+ REF_RELEASE(cfg);
}