Basic SSL connection going

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

diff --git a/src/main.sml b/src/main.sml
index 86da95f..cc5b5b5 100644 (file)
--- a/src/main.sml
+++ b/src/main.sml
@@ -22,12 +22,23 @@ structure Main :> MAIN = struct
 
 open Ast Print
 
+structure SM = StringMap
+
 val dmy = ErrorMsg.dummyLoc
 
-val tInit = (TAction ((CRoot, dmy),
-                     StringMap.empty,
-                     StringMap.empty),
-            dmy)
+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 +50,7 @@ fun check' G fname =
        if !ErrorMsg.anyErrors then
            G
        else
-           Tycheck.checkFile G tInit prog
+           Tycheck.checkFile G (tInit ()) prog
     end
 
 fun basis () =
@@ -48,10 +59,12 @@ fun basis () =
 
        fun loop files =
            case Posix.FileSys.readdir dir of
-               NONE => files
+               NONE => (Posix.FileSys.closedir dir;
+                        files)
              | SOME fname =>
                if String.isSuffix ".dtl" fname then
-                   loop (String.concatWith "/" [Config.libRoot, fname]
+                   loop (OS.Path.joinDirFile {dir = Config.libRoot,
+                                              file = fname}
                          :: files)
                else
                    loop files
@@ -59,12 +72,16 @@ fun basis () =
        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 ()
+       val _ = Env.preTycheck ()
 
        val b = basis ()
     in
@@ -72,13 +89,14 @@ fun check fname =
            (b, NONE)
        else
            let
+               val _ = ErrorMsg.reset ()
                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
@@ -112,7 +130,91 @@ fun eval fname =
        if !ErrorMsg.anyErrors then
            ()
        else
-           Eval.exec StringMap.empty body'
+           Eval.exec (SM.map (fn f => f ()) (!defaultV)) body'
       | NONE => ()
 
+val dispatcher =
+    Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort
+
+fun ssl_err s =
+    let
+       val err = F_OpenSSL_SML_get_error.f ()
+
+       val lib = F_OpenSSL_SML_lib_error_string.f err
+       val func = F_OpenSSL_SML_func_error_string.f err
+       val reason = F_OpenSSL_SML_reason_error_string.f err
+    in
+       print s;
+       print "\nReason: ";
+       if C.Ptr.isNull lib then
+           ()
+       else
+           (print (ZString.toML lib);
+            print ":");
+       if C.Ptr.isNull func then
+           ()
+       else
+           (print (ZString.toML func);
+            print ":");
+       if C.Ptr.isNull reason then
+           ()
+       else
+           print (ZString.toML reason);
+       print "\n"
+    end
+
+
+fun request fname =
+    let
+       val context = OpenSSL.context ("/home/adamc/fake/clientcert.pem",
+                                      "/home/adamc/fake/clientkey.pem",
+                                      Config.trustStore)
+
+       val bio = OpenSSL.connect (context, dispatcher)
+
+       val _ = print ("Subject: " ^ OpenSSL.peerCN bio ^ "\n")
+
+       val inf = TextIO.openIn fname
+
+       fun loop () =
+           case TextIO.inputLine inf of
+               NONE => ()
+             | SOME line => (OpenSSL.writeAll (bio, line);
+                             loop ())
+    in
+       loop ();
+       TextIO.closeIn inf;
+       OpenSSL.close bio
+    end
+
+fun service () =
+    let
+       val context = OpenSSL.context (Config.serverCert,
+                                      Config.serverKey,
+                                      Config.trustStore)
+
+       val sock = OpenSSL.listen (Config.dispatcherPort, Config.queueSize)
+
+       fun loop () =
+           case OpenSSL.accept (context, sock) of
+               NONE => ()
+             | SOME bio =>
+               let
+                   val _ = print ("Subject: " ^ OpenSSL.peerCN bio ^ "\n")
+
+                   fun loop' () =
+                       case OpenSSL.readOne bio of
+                           NONE => ()
+                         | SOME line => (print line;
+                                         loop' ())
+               in
+                   loop' ();
+                   OpenSSL.close bio;
+                   loop ()
+               end
+    in
+       loop ();
+       OpenSSL.shutdown sock
+    end
+
 end