Reroutemount library for transparently mounting fuse partitions.

[vsys-wrappers.git] / fuse / reroutemount.c

diff --git a/fuse/reroutemount.c b/fuse/reroutemount.c
index 4ded3bb..9bb8b60 100644 (file)
--- a/fuse/reroutemount.c
+++ b/fuse/reroutemount.c
@@ -6,9 +6,18 @@
 #include <sys/un.h>
 #include "stolen_from_fuse.h"
 
-char *socket_name = "/vsys/local_fusemount.control";
-
-int rrm_connect_socket() {
+char *socket_name = "/vsys/fd_fusemount.control";
+unsigned int arg_length = 128;
+
+void send_argument(int control_channel_fd, char *source) {
+    int sent;
+    sent=send(control_channel_fd, source, arg_length, 0);
+    if (sent<arg_length) {
+        printf("Error receiving arguments over the control buffer\n");
+        exit(1);
+    }
+}
+int connect_socket() {
   int fd = socket( AF_UNIX, SOCK_STREAM, 0 );
   struct sockaddr_un addr;
   addr.sun_family = AF_UNIX;
@@ -18,7 +27,7 @@ int rrm_connect_socket() {
   return fd;
 }
 
-void rrm_do_umount( char *const argv[], int n, int fd ) {
+void do_umount( char *const argv[], int n, int fd ) {
 
   // write the length
   char buf[1024];
@@ -43,53 +52,70 @@ void rrm_do_umount( char *const argv[], int n, int fd ) {
 
 int umount2( const char *mnt, int flags ) {
  
-  int fd = rrm_connect_socket();
+  int fd = connect_socket();
 
   const char *argv[3];
   argv[0] = "fusermount";
   argv[1] = "-u";
   argv[2] = mnt;
 
-  rrm_do_umount( (char **const) argv, 3, fd );
+  do_umount( (char **const) argv, 3, fd );
 
   close(fd);
  
 }
 
+int get_magic_fd (char *data) {
+    char *ptr;
+    int fd;
+
+    data[arg_length-1]='\0';
+    ptr = strstr(data,"fd=");
+    if (!ptr)
+        return -1;
+
+    // Found two fd= expressions
+    if (strstr(ptr+3,"fd="))
+        return -1;
+
+    if (*(ptr+3)!='\0') {
+        sscanf(ptr+3,"%d",&fd);
+        return fd;
+    }
+    else
+        return -1;
+}
 int mount(const char *source, const char *target, const char *filesystemtype,
         unsigned long mountflags, const void *data) {
+  int fd = connect_socket();
+  int old_fuse_fd, new_fuse_fd;
 
-  int fd = rrm_connect_socket();
-  
   char buf[1024];
-  sprintf( buf, "%08x\n", 0 );
-  write( fd, buf, strlen(buf) );
 
-  sprintf( buf, "%s\n%s\n%s\n%ld\n%s\n", source, target, filesystemtype,
-          mountflags, data );
-  write( fd, buf, strlen(buf) );
+  send_argument(fd, source);
+  send_argument(fd, target );
+  send_argument(fd, filesystemtype );
+  send_argument(fd, data );
 
-  char inbuf[10];
-  int n = read( fd, inbuf, 9 );
-  inbuf[n] = '\0';
-
-  int r;
-  assert( sscanf( inbuf, "%08x\n", &r ) == 1);
-
-  int fuse_fd = 0;
-  if( r < 0 ) {
-    errno = r;
-    return -1;
-  } else if( r > 0 ) {
-    // get the fd
-    fuse_fd = rrm_receive_fd(fd);
-
-    // what was the old fd?
-    int old_fd;
-    char extra[1024];
-    int s = sscanf( data, "fd=%d,%s", &old_fd, extra );
-    assert( dup2( fuse_fd, old_fd ) == old_fd );
+  old_fuse_fd = get_magic_fd (data);
+
+  send_fd(fd, old_fuse_fd);
+
+  if (fuse_fd == -1) {
+      printf ("Reroutemount: Could not identify FUSE fd: %d\n", fuse_fd);
+      exit(1);
+  }
+
+  new_fuse_fd=receive_fd(fd);
+
+  if (new_fuse_fd == -1) {
+      printf ("Reroutemount: Fusemount returned bad fd: %d\n", fuse_fd);
+      exit(1);
+  }
 
+  if( dup2(new_fuse_fd, old_fuse_fd ) != new_fuse_fd ) {
+      printf ("Could not duplicate returned file descriptor\n");
+      exit(1);
   }
 
   close(fd);
@@ -120,9 +146,9 @@ int execv( const char *path, char *const argv[] ) {
   }
 
   // Have root do any fusermounts we need done
-  int fd = rrm_connect_socket();
+  int fd = connect_socket();
 
-  rrm_do_umount( argv, n, fd );
+  do_umount( argv, n, fd );
 
   exit(0);