mirrors/wayland
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/wayland/wayland.git synced 2025-11-07 13:29:55 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

os: wrap F_DUPFD_CLOEXEC

Browse source 
Some system C libraries do not have F_DUPFD_CLOEXEC. Provide a fallback.

Add tests for the new wl_os_dupfd_cloexec() wrapper.

Add per-wrapper call counters in os_wrappers-test.c. Makes it easier to
determine the minimum required number of wrapped calls.

Signed-off-by: Pekka Paalanen <ppaalanen@gmail.com>
This commit is contained in:
Pekka Paalanen 2012-04-23 13:55:55 +03:00
parent 3b29783dc8
commit 1463a41f89
4 changed files with 91 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

70
tests/os-wrappers-test.c
View file

@ -29,26 +29,32 @@
#include <unistd.h>
#include <dlfcn.h>
#include <errno.h>
#include <stdarg.h>
#include <fcntl.h>
#include "test-runner.h"
#include "../src/wayland-os.h"
static int fall_back;
static int wrapped_calls;
static int (*real_socket)(int, int, int);
static int wrapped_calls_socket;
static int (*real_fcntl)(int, int, ...);
static int wrapped_calls_fcntl;
static void
init_fallbacks(int do_fallbacks)
{
fall_back = do_fallbacks;
real_socket = dlsym(RTLD_NEXT, "socket");
real_fcntl = dlsym(RTLD_NEXT, "fcntl");
}
__attribute__ ((visibility("default"))) int
socket(int domain, int type, int protocol)
{
wrapped_calls++;
wrapped_calls_socket++;
if (fall_back && (type & SOCK_CLOEXEC)) {
errno = EINVAL;
@ -58,6 +64,26 @@ socket(int domain, int type, int protocol)
return real_socket(domain, type, protocol);
}
__attribute__ ((visibility("default"))) int
fcntl(int fd, int cmd, ...)
{
va_list ap;
void *arg;
wrapped_calls_fcntl++;
if (fall_back && (cmd == F_DUPFD_CLOEXEC)) {
errno = EINVAL;
return -1;
}
va_start(ap, cmd);
arg = va_arg(ap, void*);
va_end(ap);
return real_fcntl(fd, cmd, arg);
}
static void
do_os_wrappers_socket_cloexec(int n)
{
@ -74,7 +100,7 @@ do_os_wrappers_socket_cloexec(int n)
* Must have 2 calls if falling back, but must also allow
* falling back without a forced fallback.
*/
assert(wrapped_calls > n);
assert(wrapped_calls_socket > n);
exec_fd_leak_check(nr_fds);
}
@ -92,3 +118,41 @@ TEST(os_wrappers_socket_cloexec_fallback)
init_fallbacks(1);
do_os_wrappers_socket_cloexec(1);
}
static void
do_os_wrappers_dupfd_cloexec(int n)
{
int base_fd;
int fd;
int nr_fds;
nr_fds = count_open_fds();
base_fd = socket(PF_LOCAL, SOCK_STREAM, 0);
assert(base_fd >= 0);
fd = wl_os_dupfd_cloexec(base_fd, 13);
assert(fd >= 13);
close(base_fd);
/*
* Must have 4 calls if falling back, but must also allow
* falling back without a forced fallback.
*/
assert(wrapped_calls_fcntl > n);
exec_fd_leak_check(nr_fds);
}
TEST(os_wrappers_dupfd_cloexec)
{
init_fallbacks(0);
do_os_wrappers_dupfd_cloexec(0);
}
TEST(os_wrappers_dupfd_cloexec_fallback)
{
init_fallbacks(1);
do_os_wrappers_dupfd_cloexec(3);
}