1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
|
#include "torture.h"
#include <cmocka.h>
#include <unistd.h>
#include <errno.h>
#ifdef HAVE___CLOSE_NOCANCEL
extern int __close_nocancel(int fd);
#endif
static int setup(void **state)
{
torture_setup_socket_dir(state);
return 0;
}
static int teardown(void **state)
{
torture_teardown_socket_dir(state);
return 0;
}
static void test_dup2_existing_open_fd(void **state)
{
int s, dup_s;
int rc;
(void) state; /* unused */
s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
assert_int_not_equal(s, -1);
/*
* Here we try to duplicate the existing socket fd to itself
* and as per man page for dup2() it must return the already
* open fd without any failure.
*/
dup_s = dup2(s, s);
assert_int_equal(dup_s, s);
#ifdef HAVE___CLOSE_NOCANCEL
rc = __close_nocancel(s);
assert_return_code(rc, errno);
rc = close(s);
assert_int_equal(rc, -1);
assert_int_equal(errno, EBADF);
rc = __close_nocancel(s);
assert_int_equal(rc, -1);
assert_int_equal(errno, EBADF);
#else
rc = close(s);
assert_return_code(rc, errno);
#endif
}
int main(void) {
int rc;
const struct CMUnitTest tcp_dup2_tests[] = {
cmocka_unit_test(test_dup2_existing_open_fd),
};
rc = cmocka_run_group_tests(tcp_dup2_tests, setup, teardown);
return rc;
}
|