Simple OpenSSL stuff in SML/NJ

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

diff --git a/src/main.sml b/src/main.sml
index 415d9c2..14eaa4a 100644 (file)
--- a/src/main.sml
+++ b/src/main.sml
@@ -22,12 +22,29 @@ structure Main :> MAIN = struct
 
 open Ast Print
 
 
 open Ast Print
 

+structure SM = StringMap
+

 val dmy = ErrorMsg.dummyLoc
 
 val dmy = ErrorMsg.dummyLoc
 

-val tInit = (TAction ((CRoot, dmy),
-                     StringMap.empty,
-                     StringMap.empty),
-            dmy)
+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 : (unit -> exp) SM.map ref = ref SM.empty
+
+fun registerDefault (name, t, v) =
+    case SM.find (!defaultT, name) of
+       NONE => (defaultT := SM.insert (!defaultT, name, t);
+                defaultV := SM.insert (!defaultV, name, v))
+      | SOME _ => raise Fail "Duplicate default environment variable"
+
+fun tInit () = (TAction ((CRoot, dmy),
+                        !defaultT,
+                        StringMap.empty),
+               dmy)

 
 
            
 
 
            


@@ -39,7 +56,7 @@ fun check' G fname =
        if !ErrorMsg.anyErrors then
            G
        else
        if !ErrorMsg.anyErrors then
            G
        else

-           Tycheck.checkFile G tInit prog
+           Tycheck.checkFile G (tInit ()) prog

     end
 
 fun basis () =
     end
 
 fun basis () =


@@ -61,12 +78,16 @@ fun basis () =
        val files = loop []
        val files = Order.order files
     in
        val files = loop []
        val files = Order.order files
     in

-       foldl (fn (fname, G) => check' G fname) Env.empty files
+       if !ErrorMsg.anyErrors then
+           Env.empty
+       else
+           foldl (fn (fname, G) => check' G fname) Env.empty files

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

+       val _ = Env.preTycheck ()

 
        val b = basis ()
     in
 
        val b = basis ()
     in


@@ -74,13 +95,14 @@ fun check fname =
            (b, NONE)
        else
            let
            (b, NONE)
        else
            let

+               val _ = ErrorMsg.reset ()

                val prog = Parse.parse fname
            in
                if !ErrorMsg.anyErrors then
                    (Env.empty, NONE)
                else
                    let
                val prog = Parse.parse fname
            in
                if !ErrorMsg.anyErrors then
                    (Env.empty, NONE)
                else
                    let

-                       val G' = Tycheck.checkFile b tInit prog
+                       val G' = Tycheck.checkFile b (tInit ()) prog

                    in
                        (G', #3 prog)
                    end
                    in
                        (G', #3 prog)
                    end


@@ -114,7 +136,71 @@ fun eval fname =
        if !ErrorMsg.anyErrors then
            ()
        else
        if !ErrorMsg.anyErrors then
            ()
        else

-           Eval.exec StringMap.empty body'
+           Eval.exec (SM.map (fn f => f ()) (!defaultV)) body'

       | NONE => ()
 
       | NONE => ()
 

+val dispatcher : C.rw ZString.zstring' = ZString.dupML' Config.dispatcher
+
+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
+
+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
+

 end
 end