#include <lttng/ust.h>
+struct user_desc;
+
pid_t fork(void)
{
static pid_t (*plibc_func)(void) = NULL;
return retval;
}
-#ifdef __linux__
+int daemon(int nochdir, int noclose)
+{
+ static int (*plibc_func)(int nochdir, int noclose) = NULL;
+ sigset_t sigset;
+ int retval;
-struct user_desc;
+ if (plibc_func == NULL) {
+ plibc_func = dlsym(RTLD_NEXT, "daemon");
+ if (plibc_func == NULL) {
+ fprintf(stderr, "libustfork: unable to find \"daemon\" symbol\n");
+ errno = ENOSYS;
+ return -1;
+ }
+ }
+
+ ust_before_fork(&sigset);
+ /* Do the real daemon call */
+ retval = plibc_func(nochdir, noclose);
+ if (retval == 0) {
+ /* child, parent called _exit() directly */
+ ust_after_fork_child(&sigset);
+ } else {
+ /* on error in the parent */
+ ust_after_fork_parent(&sigset);
+ }
+ return retval;
+}
struct ustfork_clone_info {
int (*fn)(void *);
tls, ctid);
} else {
/* Creating a real process, we need to intervene. */
- struct ustfork_clone_info info = { fn: fn, arg: arg };
+ struct ustfork_clone_info info = { .fn = fn, .arg = arg };
ust_before_fork(&info.sigset);
retval = plibc_func(clone_fn, child_stack, flags, &info,
}
return retval;
}
-
-#elif defined (__FreeBSD__)
-
-pid_t rfork(int flags)
-{
- static pid_t (*plibc_func)(void) = NULL;
- sigset_t sigset;
- pid_t retval;
-
- if (plibc_func == NULL) {
- plibc_func = dlsym(RTLD_NEXT, "rfork");
- if (plibc_func == NULL) {
- fprintf(stderr, "libustfork: unable to find \"rfork\" symbol\n");
- return -1;
- }
- }
-
- ust_before_fork(&sigset);
- /* Do the real rfork */
- retval = plibc_func();
- if (retval == 0) {
- /* child */
- ust_after_fork_child(&sigset);
- } else {
- ust_after_fork_parent(&sigset);
- }
- return retval;
-}
-
-/*
- * On BSD, no need to override vfork, because it runs in the context of
- * the parent, with parent waiting until execve or exit is executed in
- * the child.
- */
-
-#else
-#warning "Unknown OS. You might want to ensure that fork/clone/vfork/fork handling is complete."
-#endif