Add basic tests for os_calls routines
This commit is contained in:
matt335672
parent
a5034e3572
commit
2c6419475c
@ -4,6 +4,10 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
#include <sys/time.h>
|
||||||
|
#include <sys/resource.h>
|
||||||
|
#include <poll.h>
|
||||||
|
|
||||||
#include "os_calls.h"
|
#include "os_calls.h"
|
||||||
|
|
||||||
#include "test_common.h"
|
#include "test_common.h"
|
||||||
@ -12,6 +16,55 @@
|
|||||||
#define TOP_SRCDIR "."
|
#define TOP_SRCDIR "."
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
/******************************************************************************/
|
||||||
|
/***
|
||||||
|
* Gets the number of open file descriptors for the current process */
|
||||||
|
static unsigned int
|
||||||
|
get_open_fd_count(void)
|
||||||
|
{
|
||||||
|
unsigned int i;
|
||||||
|
unsigned int rv;
|
||||||
|
|
||||||
|
// What's the max number of file descriptors?
|
||||||
|
struct rlimit nofile;
|
||||||
|
if (getrlimit(RLIMIT_NOFILE, &nofile) < 0)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
ck_abort_msg("Can't create socketpair [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
struct pollfd *fds =
|
||||||
|
(struct pollfd *)g_malloc(sizeof(struct pollfd) * nofile.rlim_cur, 0);
|
||||||
|
ck_assert_ptr_nonnull(fds);
|
||||||
|
|
||||||
|
for (i = 0 ; i < nofile.rlim_cur; ++i)
|
||||||
|
{
|
||||||
|
fds[i].fd = i;
|
||||||
|
fds[i].events = 0;
|
||||||
|
fds[i].revents = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (poll(fds, nofile.rlim_cur, 0) < 0)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
ck_abort_msg("Can't poll fds [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
rv = nofile.rlim_cur;
|
||||||
|
for (i = 0 ; i < nofile.rlim_cur; ++i)
|
||||||
|
{
|
||||||
|
if (fds[i].revents == POLLNVAL)
|
||||||
|
{
|
||||||
|
--rv;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
g_free(fds);
|
||||||
|
|
||||||
|
return rv;
|
||||||
|
}
|
||||||
|
|
||||||
|
/******************************************************************************/
|
||||||
START_TEST(test_g_file_get_size__returns_file_size)
|
START_TEST(test_g_file_get_size__returns_file_size)
|
||||||
{
|
{
|
||||||
unsigned long long size;
|
unsigned long long size;
|
||||||
@ -145,6 +198,172 @@ START_TEST(test_g_file_get_size__5GiB)
|
|||||||
END_TEST
|
END_TEST
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
/******************************************************************************/
|
||||||
|
START_TEST(test_g_sck_fd_passing)
|
||||||
|
{
|
||||||
|
int sck[2];
|
||||||
|
char buff[16];
|
||||||
|
int istatus;
|
||||||
|
unsigned int fdcount;
|
||||||
|
|
||||||
|
int devzerofd = g_file_open("/dev/zero");
|
||||||
|
ck_assert(devzerofd >= 0);
|
||||||
|
|
||||||
|
if (g_sck_local_socketpair(sck) != 0)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
ck_abort_msg("Can't create socketpair [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pass the fd for /dev/zero to sck[0]...
|
||||||
|
istatus = g_sck_send_fd_set(sck[0], "?", 1, &devzerofd, 1);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
ck_abort_msg("Can't send fd set [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// We can now close the fd for /dev/zero, as it's "in flight"
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
devzerofd = -1;
|
||||||
|
|
||||||
|
// Read the fd for /dev/zero from sck[1]
|
||||||
|
fdcount = -1;
|
||||||
|
istatus = g_sck_recv_fd_set(sck[1], buff, sizeof(buff),
|
||||||
|
&devzerofd, 1, &fdcount);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
ck_abort_msg("Can't receive fd set [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Don't need the socket pair any more
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
|
||||||
|
// We should have got 1 fd back, and received a data byte of '?'
|
||||||
|
if (fdcount != 1)
|
||||||
|
{
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
ck_abort_msg("Should have 1 fd, got %u", fdcount);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (buff[0] != '?')
|
||||||
|
{
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
ck_abort_msg("Should have received '?' in buffer");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Does the fd for /dev/zero work?
|
||||||
|
istatus = g_file_read(devzerofd, buff, 1);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
ck_abort_msg("Can't read from /dev/zero fd %d [%s]", devzerofd, errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
g_file_close(devzerofd);
|
||||||
|
|
||||||
|
ck_assert_int_eq(buff[0], '\0');
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
|
START_TEST(test_g_sck_fd_overflow)
|
||||||
|
{
|
||||||
|
int sck[2];
|
||||||
|
char buff[16];
|
||||||
|
int istatus;
|
||||||
|
unsigned int fdcount;
|
||||||
|
int devzerofd[2];
|
||||||
|
|
||||||
|
// Count the number of file descriptors for the process
|
||||||
|
unsigned int base_fd_count = get_open_fd_count();
|
||||||
|
unsigned int proc_fd_count;
|
||||||
|
|
||||||
|
// Open a couple of file descriptors to /dev/zero
|
||||||
|
devzerofd[0] = g_file_open("/dev/zero");
|
||||||
|
devzerofd[1] = g_file_open("/dev/zero");
|
||||||
|
ck_assert(devzerofd[0] >= 0);
|
||||||
|
ck_assert(devzerofd[1] >= 0);
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count + 2);
|
||||||
|
|
||||||
|
if (g_sck_local_socketpair(sck) != 0)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd[0]);
|
||||||
|
g_file_close(devzerofd[1]);
|
||||||
|
ck_abort_msg("Can't create socketpair [%s]", errstr);
|
||||||
|
}
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count + 4);
|
||||||
|
|
||||||
|
// Pass the /dev/zero fds to sck[0]...
|
||||||
|
istatus = g_sck_send_fd_set(sck[0], "?", 1, devzerofd, 2);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd[0]);
|
||||||
|
g_file_close(devzerofd[1]);
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
ck_abort_msg("Can't send fd set [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// We can now close fds for /dev/zero, as they are "in flight"
|
||||||
|
g_file_close(devzerofd[0]);
|
||||||
|
g_file_close(devzerofd[1]);
|
||||||
|
devzerofd[0] = -1;
|
||||||
|
devzerofd[1] = -1;
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count + 2);
|
||||||
|
|
||||||
|
// Read one fd for /dev/zero from sck[1]
|
||||||
|
fdcount = -1;
|
||||||
|
istatus = g_sck_recv_fd_set(sck[1], buff, sizeof(buff),
|
||||||
|
&devzerofd[0], 1, &fdcount);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
ck_abort_msg("Can't receive fd set [%s]", errstr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// We should now have just ONE more file descriptor
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count + 3);
|
||||||
|
|
||||||
|
// Don't need the socket pair any more
|
||||||
|
g_file_close(sck[0]);
|
||||||
|
g_file_close(sck[1]);
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count + 1);
|
||||||
|
|
||||||
|
// Does the fd for /dev/zero work?
|
||||||
|
istatus = g_file_read(devzerofd[0], buff, 1);
|
||||||
|
if (istatus != 1)
|
||||||
|
{
|
||||||
|
const char *errstr = g_get_strerror();
|
||||||
|
g_file_close(devzerofd[0]);
|
||||||
|
ck_abort_msg("Can't read from /dev/zero fd %d [%s]",
|
||||||
|
devzerofd[0], errstr);
|
||||||
|
}
|
||||||
|
ck_assert_int_eq(buff[0], '\0');
|
||||||
|
|
||||||
|
g_file_close(devzerofd[0]);
|
||||||
|
proc_fd_count = get_open_fd_count();
|
||||||
|
ck_assert_int_eq(proc_fd_count, base_fd_count);
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
/******************************************************************************/
|
/******************************************************************************/
|
||||||
Suite *
|
Suite *
|
||||||
make_suite_test_os_calls(void)
|
make_suite_test_os_calls(void)
|
||||||
@ -163,6 +382,8 @@ make_suite_test_os_calls(void)
|
|||||||
tcase_add_test(tc_os_calls, test_g_file_get_size__2GiB);
|
tcase_add_test(tc_os_calls, test_g_file_get_size__2GiB);
|
||||||
tcase_add_test(tc_os_calls, test_g_file_get_size__5GiB);
|
tcase_add_test(tc_os_calls, test_g_file_get_size__5GiB);
|
||||||
#endif
|
#endif
|
||||||
|
tcase_add_test(tc_os_calls, test_g_sck_fd_passing);
|
||||||
|
tcase_add_test(tc_os_calls, test_g_sck_fd_overflow);
|
||||||
|
|
||||||
return s;
|
return s;
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user