X-Git-Url: https://git.hcoop.net/hcoop/domtool2.git/blobdiff_plain/aa56e112996b3650e8ac343831322d2a9ab0de54..36e42cb86393a7b9e333ecd7edfbdd16c7d9a1ac:/src/main.sml

diff --git a/src/main.sml b/src/main.sml
index e0beaf2..734b10a 100644
--- a/src/main.sml
+++ b/src/main.sml
@@ -20,7 +20,7 @@
 
 structure Main :> MAIN = struct
 
-open Ast Print
+open Ast MsgTypes Print
 
 structure SM = StringMap
 
@@ -69,19 +69,22 @@ fun check fname =
 	val b = basis ()
     in
 	if !ErrorMsg.anyErrors then
-	    (b, NONE)
+	    raise ErrorMsg.Error
 	else
 	    let
 		val _ = ErrorMsg.reset ()
 		val prog = Parse.parse fname
 	    in
 		if !ErrorMsg.anyErrors then
-		    (Env.empty, NONE)
+		    raise ErrorMsg.Error
 		else
 		    let
 			val G' = Tycheck.checkFile b (Defaults.tInit ()) prog
 		    in
-			(G', #3 prog)
+			if !ErrorMsg.anyErrors then
+			    raise ErrorMsg.Error
+			else
+			    (G', #3 prog)
 		    end
 	    end
     end
@@ -111,14 +114,23 @@ fun eval fname =
     case reduce fname of
 	(SOME body') =>
 	if !ErrorMsg.anyErrors then
-	    ()
+	    raise ErrorMsg.Error
 	else
 	    Eval.exec (Defaults.eInit ()) body'
-      | NONE => ()
+      | NONE => raise ErrorMsg.Error
 
 val dispatcher =
     Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort
 
+fun hostname () =
+    let
+	val inf = TextIO.openIn "/etc/hostname"
+    in
+	case TextIO.inputLine inf of
+	    NONE => (TextIO.closeIn inf; raise Fail "No line in /etc/hostname")
+	  | SOME line => (TextIO.closeIn inf; String.substring (line, 0, size line - 1))
+    end
+
 fun request fname =
     let
 	val uid = Posix.ProcEnv.getuid ()
@@ -136,14 +148,22 @@ fun request fname =
 
 	val inf = TextIO.openIn fname
 
-	fun loop () =
+	fun loop lines =
 	    case TextIO.inputLine inf of
-		NONE => ()
-	      | SOME line => (OpenSSL.writeAll (bio, line);
-			      loop ())
+		NONE => String.concat (List.rev lines)
+	      | SOME line => loop (line :: lines)
+
+	val code = loop []
     in
-	loop ();
 	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 => ()
@@ -155,6 +175,7 @@ fun service () =
 	val context = OpenSSL.context (Config.serverCert,
 				       Config.serverKey,
 				       Config.trustStore)
+	val _ = Domain.set_context context
 
 	val sock = OpenSSL.listen (context, Config.dispatcherPort)
 
@@ -167,23 +188,95 @@ fun service () =
 		    val () = print ("\nConnection from " ^ user ^ "\n")
 		    val () = Domain.setUser user
 
-		    val outname = OS.FileSys.tmpName ()
-		    val outf = TextIO.openOut outname
+		    fun cmdLoop () =
+			case Msg.recv bio of
+			    NONE => (OpenSSL.close bio
+				     handle OpenSSL.OpenSSL _ => ();
+				     loop ())
+			  | SOME m =>
+			    case m of
+				MsgConfig code =>
+				let
+				    val _ = print "Configuration:\n"
+				    val _ = print code
+				    val _ = print "\n"
 
-		    fun loop' () =
-			case OpenSSL.readOne bio of
-			    NONE => ()
-			  | SOME line => (TextIO.output (outf, line);
-					  loop' ())
+				    val outname = OS.FileSys.tmpName ()
+				    val outf = TextIO.openOut outname
+				in
+				    TextIO.output (outf, code);
+				    TextIO.closeOut outf;
+				    (eval outname;
+				     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
+			      | _ =>
+				(Msg.send (bio, MsgError "Unexpected command")
+				 handle OpenSSL.OpenSSL _ => ();
+				 OpenSSL.close bio
+				 handle OpenSSL.OpenSSL _ => ();
+				 loop ())
+		in
+		    cmdLoop ()
+		end
+    in
+	loop ();
+	OpenSSL.shutdown sock
+    end
+
+fun slave () =
+    let
+	val host = hostname ()
+
+	val context = OpenSSL.context (Config.certDir ^ "/" ^ host ^ ".pem",
+				       Config.keyDir ^ "/" ^ host ^ ".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
-		    (loop' ();
-		     TextIO.closeOut outf;
-		     eval outname
-		     handle ErrorMsg.Error => ();
-		     OS.FileSys.remove outname;
-		     OpenSSL.close bio)
-		    handle OpenSSL.OpenSSL _ => ();
-		    loop ()
+		    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
     in
 	loop ();