X-Git-Url: https://git.hcoop.net/hcoop/domtool2.git/blobdiff_plain/6ae327f88a6be8efd02cfe4b713444f9f3ac2672..07cc384cf2e6e3589f2892026a9b8f9835c8eb2c:/src/main.sml

diff --git a/src/main.sml b/src/main.sml
index 59a76d5..bcadb63 100644
--- a/src/main.sml
+++ b/src/main.sml
@@ -26,8 +26,14 @@ structure SM = StringMap
 
 val dmy = ErrorMsg.dummyLoc
 
+fun init () = (F_OpenSSL_SML_add_all_algorithms.f' ();
+	       F_OpenSSL_SML_load_error_strings.f' ();
+	       F_OpenSSL_SML_load_BIO_strings.f' ())
+
+val () = init ()
+
 val defaultT : record ref = ref SM.empty
-val defaultV : exp SM.map ref = ref SM.empty
+val defaultV : (unit -> exp) SM.map ref = ref SM.empty
 
 fun registerDefault (name, t, v) =
     case SM.find (!defaultT, name) of
@@ -81,6 +87,7 @@ fun basis () =
 fun check fname =
     let
 	val _ = ErrorMsg.reset ()
+	val _ = Env.preTycheck ()
 
 	val b = basis ()
     in
@@ -88,6 +95,7 @@ fun check fname =
 	    (b, NONE)
 	else
 	    let
+		val _ = ErrorMsg.reset ()
 		val prog = Parse.parse fname
 	    in
 		if !ErrorMsg.anyErrors then
@@ -128,7 +136,114 @@ fun eval fname =
 	if !ErrorMsg.anyErrors then
 	    ()
 	else
-	    Eval.exec (!defaultV) body'
+	    Eval.exec (SM.map (fn f => f ()) (!defaultV)) body'
       | NONE => ()
 
+val dispatcher : C.rw ZString.zstring' =
+    ZString.dupML' (Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort)
+
+val listenOn : C.rw ZString.zstring' =
+    ZString.dupML' ("localhost:" ^ Int.toString Config.dispatcherPort)
+
+fun ssl_err s =
+    let
+	val err = F_OpenSSL_SML_get_error.f ()
+    in
+	print s;
+	print "\nReason: ";
+	print (ZString.toML (F_OpenSSL_SML_lib_error_string.f err));
+	print ":";
+	print (ZString.toML (F_OpenSSL_SML_func_error_string.f err));
+	print ":";
+	print (ZString.toML (F_OpenSSL_SML_reason_error_string.f err));
+	print "\n"
+    end
+
+exception OpenSSL of string
+
+val readBuf : (C.uchar, C.rw) C.obj C.ptr' = C.alloc' C.S.uchar (Word.fromInt Config.bufSize)
+val bufSize = Int32.fromInt Config.bufSize
+
+fun readOne bio =
+    let
+	val r = F_OpenSSL_SML_read.f' (bio, C.Ptr.inject' readBuf, bufSize)
+    in
+	if r = 0 then
+	    NONE
+	else if r < 0 then
+	    raise OpenSSL "BIO_read failed"
+	else
+	    SOME (CharVector.tabulate (Int32.toInt r,
+				    fn i => chr (Word32.toInt (C.Get.uchar'
+								   (C.Ptr.sub' C.S.uchar (readBuf, i))))))
+    end
+
+fun writeAll (bio, s) =
+    let
+	val buf = ZString.dupML' s
+
+	fun loop (buf, len) =
+	    let
+		val r = F_OpenSSL_SML_write.f' (bio, C.Ptr.inject' buf, len)
+	    in
+		if r = len then
+		    ()
+		else if r <= 0 then
+		    (C.free' buf;
+		     raise OpenSSL "BIO_write failed")
+		else
+		    loop (C.Ptr.|+! C.S.uchar (buf, Int32.toInt r), Int32.- (len, r))
+	    end
+    in
+	loop (buf, Int32.fromInt (size s));
+	C.free' buf
+    end
+
+fun request fname =
+    let
+	val bio = F_OpenSSL_SML_new_connect.f' dispatcher
+    in
+	if C.Ptr.isNull' bio then
+	    (ssl_err ("Error initializating connection to dispatcher at " ^ Config.dispatcher);
+	     F_OpenSSL_SML_free_all.f' bio)
+	else if F_OpenSSL_SML_do_connect.f' bio <= 0 then
+	    (ssl_err ("Error connecting to dispatcher at " ^ Config.dispatcher);
+	     F_OpenSSL_SML_free_all.f' bio)
+	else let
+		val inf = TextIO.openIn fname
+
+		fun loop () =
+		    case TextIO.inputLine inf of
+			NONE => ()
+		      | SOME line => (writeAll (bio, line);
+				      loop ())
+	    in
+		loop ();
+		TextIO.closeIn inf;
+		F_OpenSSL_SML_free_all.f' bio
+	    end
+    end
+
+fun serviceOne () =
+    let
+	val bio = F_OpenSSL_SML_new_accept.f' listenOn
+    in
+	if C.Ptr.isNull' bio then
+	    (ssl_err "Error initializating listener";
+	     F_OpenSSL_SML_free_all.f' bio)
+	else if F_OpenSSL_SML_do_accept.f' bio <= 0 then
+	    (ssl_err "Error accepting connection";
+	     F_OpenSSL_SML_free_all.f' bio)
+	else let
+		fun loop () =
+		    case readOne bio of
+			NONE => ()
+		      | SOME line => (print line;
+				      loop ())
+	    in
+		loop ();
+		F_OpenSSL_SML_free_all.f' bio
+	    end
+    end
+
 end