Basic client/server thing going on with unencrypted OpenSSL

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

diff --git a/src/main.sml b/src/main.sml
index 14eaa4a..bcadb63 100644 (file)
--- a/src/main.sml
+++ b/src/main.sml
@@ -139,7 +139,11 @@ fun eval fname =
            Eval.exec (SM.map (fn f => f ()) (!defaultV)) body'
       | NONE => ()
 
-val dispatcher : C.rw ZString.zstring' = ZString.dupML' Config.dispatcher
+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
@@ -157,6 +161,23 @@ fun ssl_err s =
 
 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
@@ -203,4 +224,26 @@ fun request fname =
            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