[hcoop/domtool2.git] / src / main.sml

(* HCoop Domtool (http://hcoop.sourceforge.net/)
 * Copyright (c) 2006, Adam Chlipala
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *)

(* Main interface *)

structure Main :> MAIN = struct

open Ast MsgTypes Print

structure SM = StringMap

fun init () = Acl.read Config.aclFile
	    
fun check' G fname =
    let
	val prog = Parse.parse fname
    in
	if !ErrorMsg.anyErrors then
	    G
	else
	    Tycheck.checkFile G (Defaults.tInit ()) prog
    end

fun basis () =
    let
	val dir = Posix.FileSys.opendir Config.libRoot

	fun loop files =
	    case Posix.FileSys.readdir dir of
		NONE => (Posix.FileSys.closedir dir;
			 files)
	      | SOME fname =>
		if String.isSuffix ".dtl" fname then
		    loop (OS.Path.joinDirFile {dir = Config.libRoot,
					       file = fname}
			  :: files)
		else
		    loop files

	val files = loop []
	val (_, files) = Order.order NONE files
    in
	if !ErrorMsg.anyErrors then
	    Env.empty
	else
	    (Tycheck.allowExterns ();
	     foldl (fn (fname, G) => check' G fname) Env.empty files
	     before Tycheck.disallowExterns ())
    end

fun check fname =
    let
	val _ = ErrorMsg.reset ()
	val _ = Env.preTycheck ()

	val b = basis ()
    in
	if !ErrorMsg.anyErrors then
	    raise ErrorMsg.Error
	else
	    let
		val _ = Tycheck.disallowExterns ()
		val _ = ErrorMsg.reset ()
		val prog = Parse.parse fname
	    in
		if !ErrorMsg.anyErrors then
		    raise ErrorMsg.Error
		else
		    let
			val G' = Tycheck.checkFile b (Defaults.tInit ()) prog
		    in
			if !ErrorMsg.anyErrors then
			    raise ErrorMsg.Error
			else
			    (G', #3 prog)
		    end
	    end
    end

val notTmp = CharVector.all (fn ch => Char.isAlphaNum ch orelse ch = #"." orelse ch = #"_" orelse ch = #"-")

fun checkDir dname =
    let
	val b = basis ()

	val dir = Posix.FileSys.opendir dname

	fun loop files =
	    case Posix.FileSys.readdir dir of
		NONE => (Posix.FileSys.closedir dir;
			 files)
	      | SOME fname =>
		if notTmp fname then
		    loop (OS.Path.joinDirFile {dir = dname,
					       file = fname}
			  :: files)
		else
		    loop files

	val files = loop []
	val (_, files) = Order.order (SOME b) files
    in
	if !ErrorMsg.anyErrors then
	    false
	else
	    (foldl (fn (fname, G) => check' G fname) b files;
	     !ErrorMsg.anyErrors)
    end

fun reduce fname =
    let
	val (G, body) = check fname
    in
	if !ErrorMsg.anyErrors then
	    NONE
	else
	    case body of
		SOME body =>
		let
		    val body' = Reduce.reduceExp G body
		in
		    (*printd (PD.hovBox (PD.PPS.Rel 0,
					 [PD.string "Result:",
					  PD.space 1,
					  p_exp body']))*)
		    SOME body'
		end
	      | _ => NONE
    end

fun eval fname =
    case reduce fname of
	(SOME body') =>
	if !ErrorMsg.anyErrors then
	    raise ErrorMsg.Error
	else
	    Eval.exec (Defaults.eInit ()) body'
      | NONE => raise ErrorMsg.Error

val dispatcher =
    Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort

fun requestContext f =
    let
	val uid = Posix.ProcEnv.getuid ()
	val user = Posix.SysDB.Passwd.name (Posix.SysDB.getpwuid uid)
		   
	val () = Acl.read Config.aclFile
	val () = Domain.setUser user
		 
	val () = f ()

	val context = OpenSSL.context (Config.certDir ^ "/" ^ user ^ ".pem",
				       Config.keyDir ^ "/" ^ user ^ "/key.pem",
				       Config.trustStore)
    in
	(user, context)
    end

fun requestBio f =
    let
	val (user, context) = requestContext f
    in
	(user, OpenSSL.connect (context, dispatcher))
    end

fun request fname =
    let
	val (user, bio) = requestBio (fn () => ignore (check fname))

	val inf = TextIO.openIn fname

	fun loop lines =
	    case TextIO.inputLine inf of
		NONE => String.concat (List.rev lines)
	      | SOME line => loop (line :: lines)

	val code = loop []
    in
	TextIO.closeIn inf;
	Msg.send (bio, MsgConfig code);
	case Msg.recv bio of
	    NONE => print "Server closed connection unexpectedly.\n"
	  | SOME m =>
	    case m of
		MsgOk => print "Configuration succeeded.\n"
	      | MsgError s => print ("Configuration failed: " ^ s ^ "\n")
	      | _ => print "Unexpected server reply.\n";
	OpenSSL.close bio
    end
    handle ErrorMsg.Error => ()

fun requestDir dname =
    let
	val (user, bio) = requestBio (fn () => ignore (checkDir dname))

	val b = basis ()

	val dir = Posix.FileSys.opendir dname

	fun loop files =
	    case Posix.FileSys.readdir dir of
		NONE => (Posix.FileSys.closedir dir;
			 files)
	      | SOME fname =>
		if notTmp fname then
		    loop (OS.Path.joinDirFile {dir = dname,
					       file = fname}
			  :: files)
		else
		    loop files

	val files = loop []
	val (_, files) = Order.order (SOME b) files

	val _ = if !ErrorMsg.anyErrors then
		    raise ErrorMsg.Error
		else
		    ()

	val codes = map (fn fname =>
			    let	    
				val inf = TextIO.openIn fname

				fun loop lines =
				    case TextIO.inputLine inf of
					NONE => String.concat (rev lines)
				      | SOME line => loop (line :: lines)
			    in
				loop []
				before TextIO.closeIn inf
			    end) files
    in
	Msg.send (bio, MsgMultiConfig codes);
	case Msg.recv bio of
	    NONE => print "Server closed connection unexpectedly.\n"
	  | SOME m =>
	    case m of
		MsgOk => print "Configuration succeeded.\n"
	      | MsgError s => print ("Configuration failed: " ^ s ^ "\n")
	      | _ => print "Unexpected server reply.\n";
	OpenSSL.close bio
    end
    handle ErrorMsg.Error => ()

fun requestGrant acl =
    let
	val (user, bio) = requestBio (fn () => ())
    in
	Msg.send (bio, MsgGrant acl);
	case Msg.recv bio of
	    NONE => print "Server closed connection unexpectedly.\n"
	  | SOME m =>
	    case m of
		MsgOk => print "Grant succeeded.\n"
	      | MsgError s => print ("Grant failed: " ^ s ^ "\n")
	      | _ => print "Unexpected server reply.\n";
	OpenSSL.close bio
    end

fun requestRevoke acl =
    let
	val (user, bio) = requestBio (fn () => ())
    in
	Msg.send (bio, MsgRevoke acl);
	case Msg.recv bio of
	    NONE => print "Server closed connection unexpectedly.\n"
	  | SOME m =>
	    case m of
		MsgOk => print "Revoke succeeded.\n"
	      | MsgError s => print ("Revoke failed: " ^ s ^ "\n")
	      | _ => print "Unexpected server reply.\n";
	OpenSSL.close bio
    end

fun requestListPerms user =
    let
	val (_, bio) = requestBio (fn () => ())
    in
	Msg.send (bio, MsgListPerms user);
	(case Msg.recv bio of
	     NONE => (print "Server closed connection unexpectedly.\n";
		      NONE)
	   | SOME m =>
	     case m of
		 MsgPerms perms => SOME perms
	       | MsgError s => (print ("Listing failed: " ^ s ^ "\n");
				NONE)
	       | _ => (print "Unexpected server reply.\n";
		       NONE))
	before OpenSSL.close bio
    end

fun requestWhoHas perm =
    let
	val (_, bio) = requestBio (fn () => ())
    in
	Msg.send (bio, MsgWhoHas perm);
	(case Msg.recv bio of
	     NONE => (print "Server closed connection unexpectedly.\n";
		      NONE)
	   | SOME m =>
	     case m of
		 MsgWhoHasResponse users => SOME users
	       | MsgError s => (print ("whohas failed: " ^ s ^ "\n");
				NONE)
	       | _ => (print "Unexpected server reply.\n";
		       NONE))
	before OpenSSL.close bio
    end

fun service () =
    let
	val () = Acl.read Config.aclFile
	
	val context = OpenSSL.context (Config.serverCert,
				       Config.serverKey,
				       Config.trustStore)
	val _ = Domain.set_context context

	val sock = OpenSSL.listen (context, Config.dispatcherPort)

	fun loop () =
	    case OpenSSL.accept sock of
		NONE => ()
	      | SOME bio =>
		let
		    val user = OpenSSL.peerCN bio
		    val () = print ("\nConnection from " ^ user ^ "\n")
		    val () = Domain.setUser user

		    fun doConfig codes =
			let
			    val _ = print "Configuration:\n"
			    val _ = app (fn s => (print s; print "\n")) codes
			    val _ = print "\n"

			    val outname = OS.FileSys.tmpName ()

			    fun doOne code =
				let
				    val outf = TextIO.openOut outname
				in
				    TextIO.output (outf, code);
				    TextIO.closeOut outf;
				    eval outname
				end
			in
			    (app doOne codes;
			     Msg.send (bio, MsgOk))
			    handle ErrorMsg.Error =>
				   (print "Compilation error\n";
				    Msg.send (bio,
					      MsgError "Error during configuration evaluation"))
				 | OpenSSL.OpenSSL s =>
				   (print "OpenSSL error\n";
				    Msg.send (bio,
					      MsgError
						  ("Error during configuration evaluation: "
						   ^ s)));
			   OS.FileSys.remove outname;
			   (ignore (OpenSSL.readChar bio);
			    OpenSSL.close bio)
			   handle OpenSSL.OpenSSL _ => ();
			   loop ()
			end

		    fun cmdLoop () =
			case Msg.recv bio of
			    NONE => (OpenSSL.close bio
				     handle OpenSSL.OpenSSL _ => ();
				     loop ())
			  | SOME m =>
			    case m of
				MsgConfig code => doConfig [code]
			      | MsgMultiConfig codes => doConfig codes

			      | MsgGrant acl =>
				if Acl.query {user = user, class = "priv", value = "all"} then
				    ((Acl.grant acl;
				      Acl.write Config.aclFile;
				      Msg.send (bio, MsgOk);
				      print ("Granted permission " ^ #value acl ^ " to " ^ #user acl ^ " in " ^ #class acl ^ ".\n"))
				     handle OpenSSL.OpenSSL s =>
					    (print "OpenSSL error\n";
					     Msg.send (bio,
						       MsgError
							   ("Error during granting: "
							    ^ s)));
				    (ignore (OpenSSL.readChar bio);
				     OpenSSL.close bio)
				    handle OpenSSL.OpenSSL _ => ();
				    loop ())
				else
				    ((Msg.send (bio, MsgError "Not authorized to grant privileges");
				      print "Unauthorized user asked to grant a permission!\n";
				      ignore (OpenSSL.readChar bio);
				      OpenSSL.close bio)
				     handle OpenSSL.OpenSSL _ => ();
				     loop ())

			      | MsgRevoke acl =>
				if Acl.query {user = user, class = "priv", value = "all"} then
				    ((Acl.revoke acl;
				      Acl.write Config.aclFile;
				      Msg.send (bio, MsgOk);
				      print ("Revoked permission " ^ #value acl ^ " from " ^ #user acl ^ " in " ^ #class acl ^ ".\n"))
				     handle OpenSSL.OpenSSL s =>
					    (print "OpenSSL error\n";
					     Msg.send (bio,
						       MsgError
							   ("Error during revocation: "
							    ^ s)));
				    (ignore (OpenSSL.readChar bio);
				     OpenSSL.close bio)
				    handle OpenSSL.OpenSSL _ => ();
				    loop ())
				else
				    ((Msg.send (bio, MsgError "Not authorized to revoke privileges");
				      print "Unauthorized user asked to revoke a permission!\n";
				      ignore (OpenSSL.readChar bio);
				      OpenSSL.close bio)
				     handle OpenSSL.OpenSSL _ => ();
				     loop ())

			      | MsgListPerms user =>
				((Msg.send (bio, MsgPerms (Acl.queryAll user));
				  print ("Sent permission list for user " ^ user ^ ".\n"))
				 handle OpenSSL.OpenSSL s =>
					(print "OpenSSL error\n";
					 Msg.send (bio,
						   MsgError
						       ("Error during permission listing: "
							^ s)));
				(ignore (OpenSSL.readChar bio);
				 OpenSSL.close bio)
				handle OpenSSL.OpenSSL _ => ();
				loop ())

			      | MsgWhoHas perm =>
				((Msg.send (bio, MsgWhoHasResponse (Acl.whoHas perm));
				  print ("Sent whohas response for " ^ #class perm ^ " / " ^ #value perm ^ ".\n"))
				 handle OpenSSL.OpenSSL s =>
					(print "OpenSSL error\n";
					 Msg.send (bio,
						   MsgError
						       ("Error during whohas: "
							^ s)));
				(ignore (OpenSSL.readChar bio);
				 OpenSSL.close bio)
				handle OpenSSL.OpenSSL _ => ();
				loop ())

			      | _ =>
				(Msg.send (bio, MsgError "Unexpected command")
				 handle OpenSSL.OpenSSL _ => ();
				 OpenSSL.close bio
				 handle OpenSSL.OpenSSL _ => ();
				 loop ())
		in
		    cmdLoop ()
		end
		    handle OpenSSL.OpenSSL s =>
			   (print ("OpenSSL error: " ^ s ^ "\n");
			    OpenSSL.close bio
			    handle OpenSSL.OpenSSL _ => ();
			    loop ())
			 | OS.SysErr (s, _) =>
			   (print ("System error: " ^ s ^ "\n");
			    OpenSSL.close bio
			    handle OpenSSL.OpenSSL _ => ();
			    loop ())
    in
	print "Listening for connections....\n";
	loop ();
	OpenSSL.shutdown sock
    end

fun slave () =
    let
	val host = Slave.hostname ()

	val context = OpenSSL.context (Config.certDir ^ "/" ^ host ^ ".pem",
				       Config.keyDir ^ "/" ^ host ^ "/key.pem",
				       Config.trustStore)

	val sock = OpenSSL.listen (context, Config.slavePort)

	fun loop () =
	    case OpenSSL.accept sock of
		NONE => ()
	      | SOME bio =>
		let
		    val peer = OpenSSL.peerCN bio
		    val () = print ("\nConnection from " ^ peer ^ "\n")
		in
		    if peer <> Config.dispatcherName then
			(print "Not authorized!\n";
			 OpenSSL.close bio;
			 loop ())
		    else let
			    fun loop' files =
				case Msg.recv bio of
				    NONE => print "Dispatcher closed connection unexpectedly\n"
				  | SOME m =>
				    case m of
					MsgFile file => loop' (file :: files)
				      | MsgDoFiles => (Slave.handleChanges files;
						       Msg.send (bio, MsgOk))
				      | _ => (print "Dispatcher sent unexpected command\n";
					      Msg.send (bio, MsgError "Unexpected command"))
			in
			    loop' [];
			    ignore (OpenSSL.readChar bio);
			    OpenSSL.close bio;
			    loop ()
			end
		end handle OpenSSL.OpenSSL s =>
			   (print ("OpenSSL error: "^ s ^ "\n");
			    OpenSSL.close bio
					  handle OpenSSL.OpenSSL _ => ();
			    loop ())
			 | OS.SysErr (s, _) =>
			   (print ("System error: "^ s ^ "\n");
			    OpenSSL.close bio
			    handle OpenSSL.OpenSSL _ => ();
			    loop ())
    in
	loop ();
	OpenSSL.shutdown sock
    end

fun autodocBasis outdir =
    let
	val dir = Posix.FileSys.opendir Config.libRoot

	fun loop files =
	    case Posix.FileSys.readdir dir of
		NONE => (Posix.FileSys.closedir dir;
			 files)
	      | SOME fname =>
		if String.isSuffix ".dtl" fname then
		    loop (OS.Path.joinDirFile {dir = Config.libRoot,
					       file = fname}
			  :: files)
		else
		    loop files

	val files = loop []
    in
	Autodoc.autodoc {outdir = outdir, infiles = files}
    end

end
Commit	Line	Data
234b917a AC	1	(* HCoop Domtool (http://hcoop.sourceforge.net/)
	2	* Copyright (c) 2006, Adam Chlipala
	3	*
	4	* This program is free software; you can redistribute it and/or
	5	* modify it under the terms of the GNU General Public License
	6	* as published by the Free Software Foundation; either version 2
	7	* of the License, or (at your option) any later version.
	8	*
	9	* This program is distributed in the hope that it will be useful,
	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	12	* GNU General Public License for more details.
	13	*
	14	* You should have received a copy of the GNU General Public License
	15	* along with this program; if not, write to the Free Software
	16	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
dac62e84	17	*)
234b917a AC	18
	19	(* Main interface *)
	20
	21	structure Main :> MAIN = struct
	22
36e42cb8	23	open Ast MsgTypes Print
234b917a	24
6ae327f8 AC	25	structure SM = StringMap
6ae327f8 AC	26
aa56e112	27	fun init () = Acl.read Config.aclFile
234b917a	28
d189ec0e	29	fun check' G fname =
a3698041 AC	30	let
	31	val prog = Parse.parse fname
	32	in
	33	if !ErrorMsg.anyErrors then
d189ec0e	34	G
a3698041	35	else
aa56e112	36	Tycheck.checkFile G (Defaults.tInit ()) prog
a3698041 AC	37	end
a3698041 AC	38
d189ec0e	39	fun basis () =
234b917a	40	let
d189ec0e AC	41	val dir = Posix.FileSys.opendir Config.libRoot
	42
	43	fun loop files =
	44	case Posix.FileSys.readdir dir of
d612d62c AC	45	NONE => (Posix.FileSys.closedir dir;
d612d62c AC	46	files)
d189ec0e AC	47	\| SOME fname =>
d189ec0e AC	48	if String.isSuffix ".dtl" fname then
d612d62c AC	49	loop (OS.Path.joinDirFile {dir = Config.libRoot,
d612d62c AC	50	file = fname}
d189ec0e AC	51	:: files)
	52	else
	53	loop files
	54
	55	val files = loop []
c53e82e4	56	val (_, files) = Order.order NONE files
d189ec0e	57	in
6ae327f8 AC	58	if !ErrorMsg.anyErrors then
	59	Env.empty
	60	else
b3159a70 AC	61	(Tycheck.allowExterns ();
	62	foldl (fn (fname, G) => check' G fname) Env.empty files
	63	before Tycheck.disallowExterns ())
d189ec0e AC	64	end
	65
	66	fun check fname =
	67	let
	68	val _ = ErrorMsg.reset ()
12adf55a	69	val _ = Env.preTycheck ()
d189ec0e AC	70
d189ec0e AC	71	val b = basis ()
234b917a AC	72	in
234b917a AC	73	if !ErrorMsg.anyErrors then
36e42cb8	74	raise ErrorMsg.Error
234b917a AC	75	else
234b917a AC	76	let
b3159a70	77	val _ = Tycheck.disallowExterns ()
7f012ffd	78	val _ = ErrorMsg.reset ()
d189ec0e	79	val prog = Parse.parse fname
234b917a	80	in
492c1cff	81	if !ErrorMsg.anyErrors then
36e42cb8	82	raise ErrorMsg.Error
492c1cff	83	else
d189ec0e	84	let
aa56e112	85	val G' = Tycheck.checkFile b (Defaults.tInit ()) prog
d189ec0e	86	in
36e42cb8 AC	87	if !ErrorMsg.anyErrors then
	88	raise ErrorMsg.Error
	89	else
	90	(G', #3 prog)
d189ec0e	91	end
234b917a AC	92	end
	93	end
	94
c53e82e4 AC	95	val notTmp = CharVector.all (fn ch => Char.isAlphaNum ch orelse ch = #"." orelse ch = #"_" orelse ch = #"-")
	96
	97	fun checkDir dname =
	98	let
	99	val b = basis ()
	100
	101	val dir = Posix.FileSys.opendir dname
	102
	103	fun loop files =
	104	case Posix.FileSys.readdir dir of
	105	NONE => (Posix.FileSys.closedir dir;
	106	files)
	107	\| SOME fname =>
	108	if notTmp fname then
	109	loop (OS.Path.joinDirFile {dir = dname,
	110	file = fname}
	111	:: files)
	112	else
	113	loop files
	114
	115	val files = loop []
	116	val (_, files) = Order.order (SOME b) files
	117	in
	118	if !ErrorMsg.anyErrors then
	119	false
	120	else
	121	(foldl (fn (fname, G) => check' G fname) b files;
	122	!ErrorMsg.anyErrors)
	123	end
	124
d189ec0e	125	fun reduce fname =
a3698041	126	let
d189ec0e	127	val (G, body) = check fname
a3698041 AC	128	in
a3698041 AC	129	if !ErrorMsg.anyErrors then
d189ec0e	130	NONE
a3698041	131	else
d189ec0e AC	132	case body of
	133	SOME body =>
	134	let
	135	val body' = Reduce.reduceExp G body
	136	in
	137	(*printd (PD.hovBox (PD.PPS.Rel 0,
	138	[PD.string "Result:",
	139	PD.space 1,
	140	p_exp body']))*)
	141	SOME body'
	142	end
	143	\| _ => NONE
a3698041 AC	144	end
a3698041 AC	145
d189ec0e AC	146	fun eval fname =
	147	case reduce fname of
	148	(SOME body') =>
	149	if !ErrorMsg.anyErrors then
36e42cb8	150	raise ErrorMsg.Error
d189ec0e	151	else
aa56e112	152	Eval.exec (Defaults.eInit ()) body'
36e42cb8	153	\| NONE => raise ErrorMsg.Error
d189ec0e	154
3b267643 AC	155	val dispatcher =
3b267643 AC	156	Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort
559e89e9	157
5ee41dd0	158	fun requestContext f =
07cc384c	159	let
a56cc2c3 AC	160	val uid = Posix.ProcEnv.getuid ()
a56cc2c3 AC	161	val user = Posix.SysDB.Passwd.name (Posix.SysDB.getpwuid uid)
5ee41dd0	162
a56cc2c3 AC	163	val () = Acl.read Config.aclFile
a56cc2c3 AC	164	val () = Domain.setUser user
5ee41dd0 AC	165
5ee41dd0 AC	166	val () = f ()
aa56e112	167
aa56e112	168	val context = OpenSSL.context (Config.certDir ^ "/" ^ user ^ ".pem",
a088cea6	169	Config.keyDir ^ "/" ^ user ^ "/key.pem",
3b267643	170	Config.trustStore)
5ee41dd0 AC	171	in
	172	(user, context)
	173	end
07cc384c	174
5ee41dd0 AC	175	fun requestBio f =
	176	let
	177	val (user, context) = requestContext f
	178	in
	179	(user, OpenSSL.connect (context, dispatcher))
	180	end
	181
	182	fun request fname =
	183	let
	184	val (user, bio) = requestBio (fn () => ignore (check fname))
559e89e9	185
3b267643 AC	186	val inf = TextIO.openIn fname
3b267643 AC	187
36e42cb8	188	fun loop lines =
3b267643	189	case TextIO.inputLine inf of
36e42cb8 AC	190	NONE => String.concat (List.rev lines)
	191	\| SOME line => loop (line :: lines)
	192
	193	val code = loop []
559e89e9	194	in
3b267643	195	TextIO.closeIn inf;
36e42cb8 AC	196	Msg.send (bio, MsgConfig code);
	197	case Msg.recv bio of
	198	NONE => print "Server closed connection unexpectedly.\n"
	199	\| SOME m =>
	200	case m of
	201	MsgOk => print "Configuration succeeded.\n"
	202	\| MsgError s => print ("Configuration failed: " ^ s ^ "\n")
	203	\| _ => print "Unexpected server reply.\n";
3b267643	204	OpenSSL.close bio
559e89e9	205	end
aa56e112	206	handle ErrorMsg.Error => ()
559e89e9	207
c53e82e4 AC	208	fun requestDir dname =
	209	let
	210	val (user, bio) = requestBio (fn () => ignore (checkDir dname))
	211
	212	val b = basis ()
	213
	214	val dir = Posix.FileSys.opendir dname
	215
	216	fun loop files =
	217	case Posix.FileSys.readdir dir of
	218	NONE => (Posix.FileSys.closedir dir;
	219	files)
	220	\| SOME fname =>
	221	if notTmp fname then
	222	loop (OS.Path.joinDirFile {dir = dname,
	223	file = fname}
	224	:: files)
	225	else
	226	loop files
	227
	228	val files = loop []
	229	val (_, files) = Order.order (SOME b) files
	230
	231	val _ = if !ErrorMsg.anyErrors then
	232	raise ErrorMsg.Error
	233	else
	234	()
	235
	236	val codes = map (fn fname =>
	237	let
	238	val inf = TextIO.openIn fname
	239
	240	fun loop lines =
	241	case TextIO.inputLine inf of
	242	NONE => String.concat (rev lines)
	243	\| SOME line => loop (line :: lines)
	244	in
	245	loop []
	246	before TextIO.closeIn inf
	247	end) files
	248	in
	249	Msg.send (bio, MsgMultiConfig codes);
	250	case Msg.recv bio of
	251	NONE => print "Server closed connection unexpectedly.\n"
	252	\| SOME m =>
	253	case m of
	254	MsgOk => print "Configuration succeeded.\n"
	255	\| MsgError s => print ("Configuration failed: " ^ s ^ "\n")
	256	\| _ => print "Unexpected server reply.\n";
	257	OpenSSL.close bio
	258	end
	259	handle ErrorMsg.Error => ()
	260
5ee41dd0 AC	261	fun requestGrant acl =
	262	let
	263	val (user, bio) = requestBio (fn () => ())
	264	in
	265	Msg.send (bio, MsgGrant acl);
	266	case Msg.recv bio of
	267	NONE => print "Server closed connection unexpectedly.\n"
	268	\| SOME m =>
	269	case m of
	270	MsgOk => print "Grant succeeded.\n"
	271	\| MsgError s => print ("Grant failed: " ^ s ^ "\n")
	272	\| _ => print "Unexpected server reply.\n";
	273	OpenSSL.close bio
	274	end
	275
411a85f2 AC	276	fun requestRevoke acl =
	277	let
	278	val (user, bio) = requestBio (fn () => ())
	279	in
	280	Msg.send (bio, MsgRevoke acl);
	281	case Msg.recv bio of
	282	NONE => print "Server closed connection unexpectedly.\n"
	283	\| SOME m =>
	284	case m of
	285	MsgOk => print "Revoke succeeded.\n"
	286	\| MsgError s => print ("Revoke failed: " ^ s ^ "\n")
	287	\| _ => print "Unexpected server reply.\n";
	288	OpenSSL.close bio
	289	end
	290
08a04eb4 AC	291	fun requestListPerms user =
	292	let
	293	val (_, bio) = requestBio (fn () => ())
	294	in
	295	Msg.send (bio, MsgListPerms user);
	296	(case Msg.recv bio of
	297	NONE => (print "Server closed connection unexpectedly.\n";
	298	NONE)
	299	\| SOME m =>
	300	case m of
	301	MsgPerms perms => SOME perms
	302	\| MsgError s => (print ("Listing failed: " ^ s ^ "\n");
	303	NONE)
	304	\| _ => (print "Unexpected server reply.\n";
	305	NONE))
	306	before OpenSSL.close bio
	307	end
	308
094877b1 AC	309	fun requestWhoHas perm =
	310	let
	311	val (_, bio) = requestBio (fn () => ())
	312	in
	313	Msg.send (bio, MsgWhoHas perm);
	314	(case Msg.recv bio of
	315	NONE => (print "Server closed connection unexpectedly.\n";
	316	NONE)
	317	\| SOME m =>
	318	case m of
	319	MsgWhoHasResponse users => SOME users
	320	\| MsgError s => (print ("whohas failed: " ^ s ^ "\n");
	321	NONE)
	322	\| _ => (print "Unexpected server reply.\n";
	323	NONE))
	324	before OpenSSL.close bio
	325	end
	326
3b267643	327	fun service () =
07cc384c	328	let
aa56e112 AC	329	val () = Acl.read Config.aclFile
aa56e112 AC	330
3b267643 AC	331	val context = OpenSSL.context (Config.serverCert,
	332	Config.serverKey,
	333	Config.trustStore)
36e42cb8	334	val _ = Domain.set_context context
3b267643	335
60534712	336	val sock = OpenSSL.listen (context, Config.dispatcherPort)
3b267643 AC	337
3b267643 AC	338	fun loop () =
60534712	339	case OpenSSL.accept sock of
3b267643 AC	340	NONE => ()
	341	\| SOME bio =>
	342	let
aa56e112 AC	343	val user = OpenSSL.peerCN bio
	344	val () = print ("\nConnection from " ^ user ^ "\n")
	345	val () = Domain.setUser user
	346
c53e82e4 AC	347	fun doConfig codes =
	348	let
	349	val _ = print "Configuration:\n"
	350	val _ = app (fn s => (print s; print "\n")) codes
	351	val _ = print "\n"
	352
	353	val outname = OS.FileSys.tmpName ()
	354
	355	fun doOne code =
	356	let
	357	val outf = TextIO.openOut outname
	358	in
	359	TextIO.output (outf, code);
	360	TextIO.closeOut outf;
	361	eval outname
	362	end
	363	in
	364	(app doOne codes;
	365	Msg.send (bio, MsgOk))
	366	handle ErrorMsg.Error =>
	367	(print "Compilation error\n";
	368	Msg.send (bio,
	369	MsgError "Error during configuration evaluation"))
	370	\| OpenSSL.OpenSSL s =>
	371	(print "OpenSSL error\n";
	372	Msg.send (bio,
	373	MsgError
	374	("Error during configuration evaluation: "
	375	^ s)));
	376	OS.FileSys.remove outname;
	377	(ignore (OpenSSL.readChar bio);
	378	OpenSSL.close bio)
	379	handle OpenSSL.OpenSSL _ => ();
	380	loop ()
	381	end
	382
36e42cb8 AC	383	fun cmdLoop () =
	384	case Msg.recv bio of
	385	NONE => (OpenSSL.close bio
	386	handle OpenSSL.OpenSSL _ => ();
	387	loop ())
	388	\| SOME m =>
	389	case m of
c53e82e4 AC	390	MsgConfig code => doConfig [code]
c53e82e4 AC	391	\| MsgMultiConfig codes => doConfig codes
5ee41dd0 AC	392
5ee41dd0 AC	393	\| MsgGrant acl =>
be1bea4c	394	if Acl.query {user = user, class = "priv", value = "all"} then
5ee41dd0 AC	395	((Acl.grant acl;
5ee41dd0 AC	396	Acl.write Config.aclFile;
411a85f2 AC	397	Msg.send (bio, MsgOk);
411a85f2 AC	398	print ("Granted permission " ^ #value acl ^ " to " ^ #user acl ^ " in " ^ #class acl ^ ".\n"))
5ee41dd0 AC	399	handle OpenSSL.OpenSSL s =>
	400	(print "OpenSSL error\n";
	401	Msg.send (bio,
	402	MsgError
	403	("Error during granting: "
	404	^ s)));
	405	(ignore (OpenSSL.readChar bio);
	406	OpenSSL.close bio)
	407	handle OpenSSL.OpenSSL _ => ();
	408	loop ())
	409	else
	410	((Msg.send (bio, MsgError "Not authorized to grant privileges");
411a85f2 AC	411	print "Unauthorized user asked to grant a permission!\n";
	412	ignore (OpenSSL.readChar bio);
	413	OpenSSL.close bio)
	414	handle OpenSSL.OpenSSL _ => ();
	415	loop ())
	416
	417	\| MsgRevoke acl =>
be1bea4c	418	if Acl.query {user = user, class = "priv", value = "all"} then
411a85f2 AC	419	((Acl.revoke acl;
	420	Acl.write Config.aclFile;
	421	Msg.send (bio, MsgOk);
	422	print ("Revoked permission " ^ #value acl ^ " from " ^ #user acl ^ " in " ^ #class acl ^ ".\n"))
	423	handle OpenSSL.OpenSSL s =>
	424	(print "OpenSSL error\n";
	425	Msg.send (bio,
	426	MsgError
	427	("Error during revocation: "
	428	^ s)));
	429	(ignore (OpenSSL.readChar bio);
	430	OpenSSL.close bio)
	431	handle OpenSSL.OpenSSL _ => ();
	432	loop ())
	433	else
	434	((Msg.send (bio, MsgError "Not authorized to revoke privileges");
	435	print "Unauthorized user asked to revoke a permission!\n";
5ee41dd0 AC	436	ignore (OpenSSL.readChar bio);
	437	OpenSSL.close bio)
	438	handle OpenSSL.OpenSSL _ => ();
	439	loop ())
	440
08a04eb4 AC	441	\| MsgListPerms user =>
	442	((Msg.send (bio, MsgPerms (Acl.queryAll user));
	443	print ("Sent permission list for user " ^ user ^ ".\n"))
	444	handle OpenSSL.OpenSSL s =>
	445	(print "OpenSSL error\n";
	446	Msg.send (bio,
	447	MsgError
	448	("Error during permission listing: "
	449	^ s)));
	450	(ignore (OpenSSL.readChar bio);
	451	OpenSSL.close bio)
	452	handle OpenSSL.OpenSSL _ => ();
	453	loop ())
	454
094877b1 AC	455	\| MsgWhoHas perm =>
	456	((Msg.send (bio, MsgWhoHasResponse (Acl.whoHas perm));
	457	print ("Sent whohas response for " ^ #class perm ^ " / " ^ #value perm ^ ".\n"))
	458	handle OpenSSL.OpenSSL s =>
	459	(print "OpenSSL error\n";
	460	Msg.send (bio,
	461	MsgError
	462	("Error during whohas: "
	463	^ s)));
	464	(ignore (OpenSSL.readChar bio);
	465	OpenSSL.close bio)
	466	handle OpenSSL.OpenSSL _ => ();
	467	loop ())
	468
36e42cb8 AC	469	\| _ =>
	470	(Msg.send (bio, MsgError "Unexpected command")
	471	handle OpenSSL.OpenSSL _ => ();
	472	OpenSSL.close bio
	473	handle OpenSSL.OpenSSL _ => ();
	474	loop ())
	475	in
	476	cmdLoop ()
	477	end
97665758 AC	478	handle OpenSSL.OpenSSL s =>
	479	(print ("OpenSSL error: " ^ s ^ "\n");
	480	OpenSSL.close bio
	481	handle OpenSSL.OpenSSL _ => ();
	482	loop ())
	483	\| OS.SysErr (s, _) =>
	484	(print ("System error: " ^ s ^ "\n");
	485	OpenSSL.close bio
	486	handle OpenSSL.OpenSSL _ => ();
	487	loop ())
36e42cb8	488	in
361a1e7f	489	print "Listening for connections....\n";
36e42cb8 AC	490	loop ();
	491	OpenSSL.shutdown sock
	492	end
	493
	494	fun slave () =
	495	let
6e62228d	496	val host = Slave.hostname ()
36e42cb8 AC	497
36e42cb8 AC	498	val context = OpenSSL.context (Config.certDir ^ "/" ^ host ^ ".pem",
a088cea6	499	Config.keyDir ^ "/" ^ host ^ "/key.pem",
36e42cb8 AC	500	Config.trustStore)
	501
	502	val sock = OpenSSL.listen (context, Config.slavePort)
	503
	504	fun loop () =
	505	case OpenSSL.accept sock of
	506	NONE => ()
	507	\| SOME bio =>
	508	let
	509	val peer = OpenSSL.peerCN bio
	510	val () = print ("\nConnection from " ^ peer ^ "\n")
3b267643	511	in
36e42cb8 AC	512	if peer <> Config.dispatcherName then
	513	(print "Not authorized!\n";
	514	OpenSSL.close bio;
	515	loop ())
	516	else let
	517	fun loop' files =
	518	case Msg.recv bio of
	519	NONE => print "Dispatcher closed connection unexpectedly\n"
	520	\| SOME m =>
	521	case m of
	522	MsgFile file => loop' (file :: files)
	523	\| MsgDoFiles => (Slave.handleChanges files;
	524	Msg.send (bio, MsgOk))
	525	\| _ => (print "Dispatcher sent unexpected command\n";
	526	Msg.send (bio, MsgError "Unexpected command"))
	527	in
	528	loop' [];
	529	ignore (OpenSSL.readChar bio);
	530	OpenSSL.close bio;
	531	loop ()
	532	end
3196000d AC	533	end handle OpenSSL.OpenSSL s =>
	534	(print ("OpenSSL error: "^ s ^ "\n");
	535	OpenSSL.close bio
	536	handle OpenSSL.OpenSSL _ => ();
	537	loop ())
7af7d4cb AC	538	\| OS.SysErr (s, _) =>
	539	(print ("System error: "^ s ^ "\n");
	540	OpenSSL.close bio
	541	handle OpenSSL.OpenSSL _ => ();
	542	loop ())
07cc384c	543	in
3b267643 AC	544	loop ();
3b267643 AC	545	OpenSSL.shutdown sock
07cc384c AC	546	end
07cc384c AC	547
3196000d AC	548	fun autodocBasis outdir =
	549	let
	550	val dir = Posix.FileSys.opendir Config.libRoot
	551
	552	fun loop files =
	553	case Posix.FileSys.readdir dir of
	554	NONE => (Posix.FileSys.closedir dir;
	555	files)
	556	\| SOME fname =>
	557	if String.isSuffix ".dtl" fname then
	558	loop (OS.Path.joinDirFile {dir = Config.libRoot,
	559	file = fname}
	560	:: files)
	561	else
	562	loop files
	563
	564	val files = loop []
	565	in
	566	Autodoc.autodoc {outdir = outdir, infiles = files}
	567	end
	568
234b917a	569	end