Add userspace namespace contexts

[lttng-ust.git] / liblttng-ust-fork / ustfork.c

diff --git a/liblttng-ust-fork/ustfork.c b/liblttng-ust-fork/ustfork.c
index 8fade56e84c5413bd30c679d7f3fd5c30b4e3c70..25f9d4cc04c31d2589e36d148684bf61d846c5aa 100644 (file)
--- a/liblttng-ust-fork/ustfork.c
+++ b/liblttng-ust-fork/ustfork.c
@@ -1,6 +1,6 @@
 /*
  * Copyright (C) 2009  Pierre-Marc Fournier
- * Copyright (C) 2011  Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ * Copyright (C) 2011-2012  Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
@@ -18,13 +18,13 @@
  */
 
 #define _GNU_SOURCE
-#include <dlfcn.h>
+#include <lttng/ust-dlfcn.h>
 #include <unistd.h>
 #include <stdio.h>
 #include <signal.h>
 #include <sched.h>
 #include <stdarg.h>
-#include "usterr.h"
+#include <errno.h>
 
 #include <lttng/ust.h>
 
@@ -33,11 +33,13 @@ pid_t fork(void)
        static pid_t (*plibc_func)(void) = NULL;
        sigset_t sigset;
        pid_t retval;
+       int saved_errno;
 
        if (plibc_func == NULL) {
                plibc_func = dlsym(RTLD_NEXT, "fork");
                if (plibc_func == NULL) {
                        fprintf(stderr, "libustfork: unable to find \"fork\" symbol\n");
+                       errno = ENOSYS;
                        return -1;
                }
        }
@@ -45,12 +47,45 @@ pid_t fork(void)
        ust_before_fork(&sigset);
        /* Do the real fork */
        retval = plibc_func();
+       saved_errno = errno;
        if (retval == 0) {
                /* child */
                ust_after_fork_child(&sigset);
        } else {
                ust_after_fork_parent(&sigset);
        }
+       errno = saved_errno;
+       return retval;
+}
+
+int daemon(int nochdir, int noclose)
+{
+       static int (*plibc_func)(int nochdir, int noclose) = NULL;
+       sigset_t sigset;
+       int retval;
+       int saved_errno;
+
+       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);
+       saved_errno = errno;
+       if (retval == 0) {
+               /* child, parent called _exit() directly */
+               ust_after_fork_child(&sigset);
+       } else {
+               /* on error in the parent */
+               ust_after_fork_parent(&sigset);
+       }
+       errno = saved_errno;
        return retval;
 }
 
@@ -85,6 +120,7 @@ int clone(int (*fn)(void *), void *child_stack, int flags, void *arg, ...)
        /* end of var args */
        va_list ap;
        int retval;
+       int saved_errno;
 
        va_start(ap, arg);
        ptid = va_arg(ap, pid_t *);
@@ -96,6 +132,7 @@ int clone(int (*fn)(void *), void *child_stack, int flags, void *arg, ...)
                plibc_func = dlsym(RTLD_NEXT, "clone");
                if (plibc_func == NULL) {
                        fprintf(stderr, "libustfork: unable to find \"clone\" symbol.\n");
+                       errno = ENOSYS;
                        return -1;
                }
        }
@@ -107,16 +144,69 @@ int clone(int (*fn)(void *), void *child_stack, int flags, void *arg, ...)
                 */
                retval = plibc_func(fn, child_stack, flags, arg, ptid,
                                tls, ctid);
+               saved_errno = errno;
        } 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,
                                ptid, tls, ctid);
+               saved_errno = errno;
                /* The child doesn't get here. */
                ust_after_fork_parent(&info.sigset);
        }
+       errno = saved_errno;
+       return retval;
+}
+
+int setns(int fd, int nstype)
+{
+       static int (*plibc_func)(int fd, int nstype) = NULL;
+       int retval;
+       int saved_errno;
+
+       if (plibc_func == NULL) {
+               plibc_func = dlsym(RTLD_NEXT, "setns");
+               if (plibc_func == NULL) {
+                       fprintf(stderr, "libustfork: unable to find \"setns\" symbol\n");
+                       errno = ENOSYS;
+                       return -1;
+               }
+       }
+
+       /* Do the real setns */
+       retval = plibc_func(fd, nstype);
+       saved_errno = errno;
+
+       ust_after_setns();
+
+       errno = saved_errno;
+       return retval;
+}
+
+int unshare(int flags)
+{
+       static int (*plibc_func)(int flags) = NULL;
+       int retval;
+       int saved_errno;
+
+       if (plibc_func == NULL) {
+               plibc_func = dlsym(RTLD_NEXT, "unshare");
+               if (plibc_func == NULL) {
+                       fprintf(stderr, "libustfork: unable to find \"unshare\" symbol\n");
+                       errno = ENOSYS;
+                       return -1;
+               }
+       }
+
+       /* Do the real setns */
+       retval = plibc_func(flags);
+       saved_errno = errno;
+
+       ust_after_unshare();
+
+       errno = saved_errno;
        return retval;
 }
 
@@ -127,11 +217,13 @@ pid_t rfork(int flags)
        static pid_t (*plibc_func)(void) = NULL;
        sigset_t sigset;
        pid_t retval;
+       int saved_errno;
 
        if (plibc_func == NULL) {
                plibc_func = dlsym(RTLD_NEXT, "rfork");
                if (plibc_func == NULL) {
                        fprintf(stderr, "libustfork: unable to find \"rfork\" symbol\n");
+                       errno = ENOSYS;
                        return -1;
                }
        }
@@ -139,12 +231,14 @@ pid_t rfork(int flags)
        ust_before_fork(&sigset);
        /* Do the real rfork */
        retval = plibc_func();
+       saved_errno = errno;
        if (retval == 0) {
                /* child */
                ust_after_fork_child(&sigset);
        } else {
                ust_after_fork_parent(&sigset);
        }
+       errno = saved_errno;
        return retval;
 }