Starting commits for fd passing support

[vsys.git] / frontend.ml

diff --git a/frontend.ml b/frontend.ml
index 335782a..9341633 100644 (file)
--- a/frontend.ml
+++ b/frontend.ml
@@ -10,8 +10,12 @@ open Directfifowatcher
   @param slice_name actual slice name - set with care, since the acl functionality refers to these names *)
 class frontendHandler (root_dir,slice_name) = 
 object(this)
-  method get_slice_name () = slice_name
 
+  (** regex indicating that the script passes fds around *)
+  val fd_regex = Str.regexp "^fd_"
+
+  method is_fd_passer fname = Str.string_match fd_regex fname 0
+  method get_slice_name () = slice_name
   (** A new script was copied into the backend, make a corresponding entry in
     the frontend.
     @param rp Relative path of the entry in the backend
@@ -22,10 +26,25 @@ object(this)
       match rp with Relpath(rel) ->
         let fqp = String.concat "/" [root_dir;rel] in
         let res = Directfifowatcher.mkentry fqp abspath realperm slice_name in
-          match res with 
-            | Success ->
-                Directfifowatcher.openentry fqp (abspath,slice_name)
-            | _ -> ()
+          begin
+            match res with 
+              | Success ->
+                  Directfifowatcher.openentry root_dir fqp (abspath,slice_name)
+              | _ -> 
+                  logprint "Could not create entry %s" abspath
+          end;
+          if (is_fd_passer rel) then
+            let res = Unixsocketwatcher.mkentry fqp abspath realperm slice_name in
+              begin
+                match res with
+                  | Success ->
+                      Unixsocketwatcher.openentry root_dir fqp (abspath, slice_name)
+                  | _ -> 
+                      logprint "Could not create entry %s" abspath
+              end
+
+
+
 
   (** A new directory was created at the backend, make a corresponding directory
     at the frontend. Refer to mkentry for parameters *)
@@ -46,11 +65,7 @@ object(this)
               end;
         with Unix.Unix_error(_,_,_) ->
           Unix.mkdir fqp perm;
-        Directfifowatcher.add_dir_watch fqp
-
-
-
-            
+          Directfifowatcher.add_dir_watch fqp
 
   (** Functions corresponding to file deletion/directory removal *)
 
@@ -65,7 +80,7 @@ object(this)
           Unix.unlink fqp_in;
           Unix.unlink fqp_out
         with _ ->
-          fprintf logfd "Hm. %s disappeared. Looks like slice %s shot itself in the foot\n" fqp (this#get_slice_name ());flush logfd
+          logprint "Hm. %s disappeared. Looks like slice %s shot itself in the foot\n" fqp (this#get_slice_name ())
 
   method rmdir rp =
     match rp with Relpath(rel) ->
@@ -74,8 +89,27 @@ object(this)
         try
           Unix.rmdir fqp
         with _ ->
-          fprintf logfd "Hm. %s disappeared or not empty. Looks like slice %s shot itself in the foot\n" fqp (this#get_slice_name ());flush logfd
+          logprint "Hm. %s disappeared or not empty. Looks like slice %s shot itself in the foot\n" fqp (this#get_slice_name ())
 
   initializer 
-    Directfifowatcher.add_dir_watch root_dir
+    (
+      try 
+        let s = Unix.stat root_dir in
+          if (s.st_kind<>S_DIR) then
+            begin
+              Unix.unlink root_dir;
+              Unix.mkdir root_dir 0o700
+            end
+          else if (s.st_perm <> 0o700) then
+            begin
+              Unix.rmdir root_dir;
+              Unix.mkdir root_dir 0o700
+            end;
+      with Unix.Unix_error(_,_,_) ->
+        begin
+          try 
+            Unix.mkdir root_dir 0o700;
+          with _ -> ();
+        end);
+          Directfifowatcher.add_dir_watch root_dir
 end