X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=unixsocketwatcher.ml;h=32316e2f0218c3e3c4abb3f798e69b8514a46375;hb=53b9ff33939d35b1bd35a790ca5758ea6386a590;hp=6ef359ab552551d9b2ef5c52ffcc07a55ac02257;hpb=9ec649e8f697ffba675a6a2b7b959e89f80f4eab;p=vsys.git diff --git a/unixsocketwatcher.ml b/unixsocketwatcher.ml index 6ef359a..32316e2 100644 --- a/unixsocketwatcher.ml +++ b/unixsocketwatcher.ml @@ -10,148 +10,77 @@ open Unix open Globals -open Dirwatcher +open Fdwatcher open Printf let close_if_open fd = (try (ignore(close fd);) with _ -> ()) -type in_pathname = string -type directory = string -type base_pathname = string +type control_path_name = string +type exec_path_name = string type slice_name = string -let unix_socket_table: (in_pathname,(directory*base_pathname*slice_name*Unix.file_descr) option) Hashtbl.t = +let unix_socket_table: (control_path_name,(exec_path_name*slice_name*Unix.file_descr) option) Hashtbl.t = Hashtbl.create 1024 -let pidmap: (int,in_pathname * Unix.file_descr) Hashtbl.t = Hashtbl.create 1024 - -let move_gate fname = - let tmpfname=String.concat "." [fname;"tmp"] in - Unix.rename fname tmpfname; - tmpfname - -let move_ungate fname restore = - Unix.rename restore fname - -let list_check lst elt _ = - let rec list_check_rec lst = - match lst with - | [] -> false - | car::cdr -> - if (car==elt) then - true - else - list_check_rec cdr - in - list_check_rec lst - -let openentry_int fifoin = - let fdin = - try openfile fifoin [O_RDONLY;O_NONBLOCK] 0o777 with - e->logprint "Error opening and connecting FIFO: %s,%o\n" fifoin 0o777;raise e - in - fdin - -(** Open entry safely, by first masking out the file to be opened *) -let openentry_safe root_dir fqp_in backend_spec = - let restore = move_gate fqp_in in - let fd_in = openentry_int restore in - move_ungate fqp_in restore; - let (fqp,slice_name) = backend_spec in - Hashtbl.replace direct_fifo_table fqp_in (Some(root_dir,fqp,slice_name,fd_in)) - -let openentry root_dir fqp backend_spec = - let control_file = String.concat "." [fqp;"control"] in - openentry_safe root_dir fqp_in backend_spec - -let reopenentry fifoin = - let entry = try Hashtbl.find direct_fifo_table fifoin with _ -> None in - match entry with - | Some(dir, fqp,slice_name,fd) -> close_if_open fd;openentry_safe dir fifoin (fqp,slice_name) - | None -> () - -(* vsys is activated when a client opens an in file *) -let connect_file fqp_in = - (* Do we care about this file? *) - let entry_info = try - Hashtbl.find direct_fifo_table fqp_in with _ -> None in - match entry_info with - | Some(_,execpath,slice_name,fifo_fdin) -> - begin - let len = String.length fqp_in in - let fqp = String.sub fqp_in 0 (len-3) in - let fqp_out = String.concat "." [fqp;"out"] in - let fifo_fdout = - try openfile fqp_out [O_WRONLY;O_NONBLOCK] 0o777 with - _-> (* The client is opening the descriptor too fast *) - sleep 1;try openfile fqp_out [O_WRONLY;O_NONBLOCK] 0o777 with - _-> - logprint "%s Output pipe not open, using stdout in place of %s\n" slice_name fqp_out;stdout - in - ignore(sigprocmask SIG_BLOCK [Sys.sigchld]); - ( - clear_nonblock fifo_fdin; - let pid=try Some(create_process execpath [|execpath;slice_name|] fifo_fdin fifo_fdout fifo_fdout) with e -> None in - match pid with - | Some(pid) -> - if (fifo_fdout <> stdout) then close_if_open fifo_fdout; - Hashtbl.add pidmap pid (fqp_in,fifo_fdout) - | None ->logprint "Error executing service: %s\n" execpath;reopenentry fqp_in - ); - ignore(sigprocmask SIG_UNBLOCK [Sys.sigchld]); - end - | None -> () - - (** Make a pair of fifo entries *) -let mkentry fqp abspath perm uname = - logprint "Making control entry %s->%s\n" fqp abspath; +let mkentry fqp exec_fqp perm slice_name = + logprint "Making control entry %s->%s\n" fqp exec_fqp; let control_filename=sprintf "%s.control" fqp in try let listening_socket = socket PF_UNIX SOCK_STREAM 0 in (try Unix.unlink control_filename with _ -> ()); let socket_address = ADDR_UNIX(control_filename) in bind listening_socket socket_address; + listen listening_socket 10; ( (* Make the user the owner of the pipes in a non-chroot environment *) if (!Globals.nochroot) then - let pwentry = Unix.getpwnam uname in + let pwentry = Unix.getpwnam slice_name in Unix.chown control_filename pwentry.pw_uid pwentry.pw_gid ); + Hashtbl.replace unix_socket_table control_filename (Some(exec_fqp,slice_name,listening_socket)); Success with - e->logprint "Error creating FIFO: %s->%s. May be something wrong at the frontend.\n" fqp fifoout;Failed) - + e->logprint "Error creating FIFO: %s->%s. May be something wrong at the frontend.\n" fqp exec_fqp;Failed -(** Close fifos that just got removed *) +let receive_event (listening_socket_spec:fname_and_fd) (_:fname_and_fd) = + (* Do we care about this file? *) + try + let (_,listening_socket) = listening_socket_spec in + let (data_socket, addr) = accept listening_socket in + match addr with + | ADDR_UNIX(fname) -> + let entry_info = + try + Hashtbl.find unix_socket_table fname + with _ -> None in + match entry_info with + begin + | Some(_,execpath,slice_name,fd) -> + begin + + + + end + | None -> () + end + | _ -> logprint "Serious error! Got a non UNIX connection over a UNIX socket\n" + with e-> logprint "Error connecting service %s\n" execpath + +(** Close sockets that just got removed *) let closeentry fqp = let control_filename = String.concat "." [fqp;"control"] in - let entry = try Hashtbl.find direct_fifo_table fqp_in with Not_found -> None in + let entry = try Hashtbl.find unix_socket_table control_filename with Not_found -> None in match entry with | None -> () - | Some(_,_,_,fd) -> + | Some(_,_,fd) -> + shutdown fd SHUTDOWN_ALL; close_if_open fd; - Hashtbl.remove direct_fifo_table fqp_in + Fdwatcher.add_fd (None,fd) (None,fd) receive_event; + Hashtbl.remove unix_socket_table control_filename + -let sigchld_handle s = - let pid,_=Unix.waitpid [Unix.WNOHANG] 0 in - try - let fqp_in,fd_out = Hashtbl.find pidmap pid in - begin - reopenentry fqp_in - end - with _ -> () -let rec add_dir_watch fqp = - Dirwatcher.add_watch fqp [S_Open] direct_fifo_handler -and - direct_fifo_handler wd dirname evlist fname = - let is_event = list_check evlist in - if (is_event Open Attrib) then - let fqp_in = String.concat "/" [dirname;fname] in - connect_file fqp_in -let del_dir_watch fqp = - () let initialize () = - Sys.set_signal Sys.sigchld (Sys.Signal_handle sigchld_handle) + ()