a safer way to sync code on a running node - no Makefile.vsh anyway
[lxc-userspace.git] / setns.c
diff --git a/setns.c b/setns.c
index c40d52b..36aa360 100644 (file)
--- a/setns.c
+++ b/setns.c
@@ -1,8 +1,27 @@
 #include <Python.h>
 #include <fcntl.h>
 #include <stdio.h>
-#include <asm-generic/unistd.h>
+#include <asm/unistd.h>
 #include <sys/mount.h>
+#include <errno.h>
+
+static PyObject *
+proc_mount(PyObject *self, PyObject *args)
+{
+    int sts; 
+    sts = mount("none","/proc","proc",0,NULL);
+
+    return Py_BuildValue("i", sts);
+}
+
+static PyObject *
+proc_umount(PyObject *self, PyObject *args)
+{
+    int sts; 
+    sts = umount("/proc");
+
+    return Py_BuildValue("i", sts);
+}
 
 static PyObject *
 chfscontext(PyObject *self, PyObject *args)
@@ -15,29 +34,20 @@ chfscontext(PyObject *self, PyObject *args)
 
     int fd = open(filepath, O_RDONLY);
     if (fd < 0) {
-           //printf("Could not open ns file\n");
-        sts = -1;
+        sts = -errno;
         goto out;
     }
     
-    if (setns(fd, 666)) {
-        sts = -1;
+    if (setns(fd, 0)) {
+        sts = -errno;
     }
     close(fd);
+    sts = 0;
 
 out:
     return Py_BuildValue("i", sts);
 }
 
-static PyObject *
-proc_mount(PyObject *self, PyObject *args)
-{
-    int sts; 
-    sts = mount("none","/proc","proc",0,NULL);
-
-    return Py_BuildValue("i", sts);
-}
-
 static PyObject *
 chcontext(PyObject *self, PyObject *args)
 {
@@ -49,15 +59,15 @@ chcontext(PyObject *self, PyObject *args)
 
     int fd = open(filepath, O_RDONLY);
     if (fd < 0) {
-//     printf("Could not open ns file\n");
-        sts = -1;
+        sts = -errno;
         goto out;
     }
     
     if (setns(fd, 0)) {
-        sts = -1;
+        sts = -errno;
     }
     close(fd);
+    sts = 0;
 
 out:
     return Py_BuildValue("i", sts);
@@ -66,6 +76,7 @@ out:
 static PyMethodDef SetnsMethods[] =
 {
          {"proc_mount", proc_mount, METH_VARARGS, "Mount a volume via the mount system call."},
+         {"proc_umount", proc_umount, METH_VARARGS, "Umount a volume via the umount system call."},
          {"chcontext", chcontext, METH_VARARGS, "Switch into an lxc container."},
          {"chfscontext", chfscontext, METH_VARARGS, "Switch into an lxc container."},
               {NULL, NULL, 0, NULL}