cuberite/libevent
1
0
Fork 0
mirror of https://github.com/cuberite/libevent.git synced 2025-09-18 08:49:57 -04:00
Code Issues Packages Projects Releases Wiki Activity

test/dns: verify bufferevent_socket_connect() errorcb invoking if socket() fails

Browse Source 
Refs: #600
(cherry picked from commit a3d8f2e09302536c6c0ffc3ccb1fc6f43b51dc4f)
This commit is contained in:
Azat Khuzhin 2018-04-24 01:45:00 +03:00 committed by Azat Khuzhin
parent 1dde74efa8
commit f3eb294f79
No known key found for this signature in database
GPG Key ID: B86086848EF8686D
1 changed files with 43 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
test/regress_dns.c
View File

@ -59,6 +59,10 @@
#include <string.h> #include <string.h>
#include <errno.h> #include <errno.h>
#ifdef EVENT__HAVE_SYS_RESOURCE_H
#include <sys/resource.h>
#endif
#include "event2/dns.h" #include "event2/dns.h"
#include "event2/dns_compat.h" #include "event2/dns_compat.h"
#include "event2/dns_struct.h" #include "event2/dns_struct.h"
@ -1162,15 +1166,22 @@ be_connect_hostname_event_cb(struct bufferevent *bev, short what, void *ctx)
got->what = what; got->what = what;
if ((what & BEV_EVENT_CONNECTED) || (what & BEV_EVENT_ERROR)) { if ((what & BEV_EVENT_CONNECTED) || (what & BEV_EVENT_ERROR)) {
int r; int expected = 3;
if ((r = bufferevent_socket_get_dns_error(bev))) { int r = bufferevent_socket_get_dns_error(bev);
if (r) {
got->dnserr = r; got->dnserr = r;
TT_BLATHER(("DNS error %d: %s", r, TT_BLATHER(("DNS error %d: %s", r,
evutil_gai_strerror(r))); evutil_gai_strerror(r)));
} ++total_connected_or_failed; } ++total_connected_or_failed;
TT_BLATHER(("Got %d connections or errors.", total_connected_or_failed)); TT_BLATHER(("Got %d connections or errors.", total_connected_or_failed));
if (total_n_accepted >= 3 && total_connected_or_failed >= 5) /** emfile test */
if (errno == EMFILE) {
expected = 0;
}
if (total_n_accepted >= expected && total_connected_or_failed >= 5)
event_base_loopexit(be_connect_hostname_base, event_base_loopexit(be_connect_hostname_base,
NULL); NULL);
} }
@ -1195,7 +1206,15 @@ test_bufferevent_connect_hostname(void *arg)
ev_uint16_t dns_port=0; ev_uint16_t dns_port=0;
int n_accept=0, n_dns=0; int n_accept=0, n_dns=0;
char buf[128]; char buf[128];
int emfile = data->setup_data && !strcmp(data->setup_data, "emfile");
int success = BEV_EVENT_CONNECTED;
int default_error = 0;
if (emfile) {
success = BEV_EVENT_ERROR;
default_error = EVUTIL_EAI_SYSTEM;
}
be_connect_hostname_base = data->base; be_connect_hostname_base = data->base;
/* Bind an address and figure out what port it's on. */ /* Bind an address and figure out what port it's on. */
@ -1221,6 +1240,13 @@ test_bufferevent_connect_hostname(void *arg)
evutil_snprintf(buf, sizeof(buf), "127.0.0.1:%d", (int)dns_port); evutil_snprintf(buf, sizeof(buf), "127.0.0.1:%d", (int)dns_port);
evdns_base_nameserver_ip_add(dns, buf); evdns_base_nameserver_ip_add(dns, buf);
#ifdef EVENT__HAVE_SETRLIMIT
if (emfile) {
struct rlimit file = { 1, 1 };
setrlimit(RLIMIT_NOFILE, &file);
}
#endif
/* Now, finally, at long last, launch the bufferevents. One should do /* Now, finally, at long last, launch the bufferevents. One should do
* a failing lookup IP, one should do a successful lookup by IP, * a failing lookup IP, one should do a successful lookup by IP,
* and one should do a successful lookup by hostname. */ * and one should do a successful lookup by hostname. */
@ -1264,18 +1290,22 @@ test_bufferevent_connect_hostname(void *arg)
tt_int_op(be_outcome[0].what, ==, BEV_EVENT_ERROR); tt_int_op(be_outcome[0].what, ==, BEV_EVENT_ERROR);
tt_int_op(be_outcome[0].dnserr, ==, EVUTIL_EAI_NONAME); tt_int_op(be_outcome[0].dnserr, ==, EVUTIL_EAI_NONAME);
tt_int_op(be_outcome[1].what, ==, BEV_EVENT_CONNECTED); tt_int_op(be_outcome[1].what, ==, success);
tt_int_op(be_outcome[1].dnserr, ==, 0); tt_int_op(be_outcome[1].dnserr, ==, 0);
tt_int_op(be_outcome[2].what, ==, BEV_EVENT_CONNECTED); tt_int_op(be_outcome[2].what, ==, success);
tt_int_op(be_outcome[2].dnserr, ==, 0); tt_int_op(be_outcome[2].dnserr, ==, 0);
tt_int_op(be_outcome[3].what, ==, BEV_EVENT_CONNECTED); tt_int_op(be_outcome[3].what, ==, success);
tt_int_op(be_outcome[3].dnserr, ==, 0); tt_int_op(be_outcome[3].dnserr, ==, default_error);
if (expect_err) { if (expect_err) {
tt_int_op(be_outcome[4].what, ==, BEV_EVENT_ERROR); tt_int_op(be_outcome[4].what, ==, BEV_EVENT_ERROR);
tt_int_op(be_outcome[4].dnserr, ==, expect_err); tt_int_op(be_outcome[4].dnserr, ==, expect_err);
} }
tt_int_op(n_accept, ==, 3); if (emfile) {
tt_int_op(n_accept, ==, 0);
} else {
tt_int_op(n_accept, ==, 3);
}
tt_int_op(n_dns, ==, 2); tt_int_op(n_dns, ==, 2);
end: end:
@ -2113,6 +2143,10 @@ struct testcase_t dns_testcases[] = {
{ "inflight", dns_inflight_test, TT_FORK|TT_NEED_BASE, &basic_setup, NULL }, { "inflight", dns_inflight_test, TT_FORK|TT_NEED_BASE, &basic_setup, NULL },
{ "bufferevent_connect_hostname", test_bufferevent_connect_hostname, { "bufferevent_connect_hostname", test_bufferevent_connect_hostname,
TT_FORK|TT_NEED_BASE, &basic_setup, NULL }, TT_FORK|TT_NEED_BASE, &basic_setup, NULL },
#ifdef EVENT__HAVE_SETRLIMIT
{ "bufferevent_connect_hostname_emfile", test_bufferevent_connect_hostname,
TT_FORK|TT_NEED_BASE, &basic_setup, (char*)"emfile" },
#endif
{ "disable_when_inactive", dns_disable_when_inactive_test, { "disable_when_inactive", dns_disable_when_inactive_test,
TT_FORK|TT_NEED_BASE, &basic_setup, NULL }, TT_FORK|TT_NEED_BASE, &basic_setup, NULL },
{ "disable_when_inactive_no_ns", dns_disable_when_inactive_no_ns_test, { "disable_when_inactive_no_ns", dns_disable_when_inactive_no_ns_test,