Effectness analysis

1 files changed, 109 insertions, 0 deletions

diff --git a/src/effectize.sml b/src/effectize.sml
new file mode 100644
index 00000000..f33b4eb8
--- /dev/null
+++ b/src/effectize.sml
@@ -0,0 +1,109 @@
+(* Copyright (c) 2009, Adam Chlipala
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * - Redistributions of source code must retain the above copyright notice,
+ *   this list of conditions and the following disclaimer.
+ * - Redistributions in binary form must reproduce the above copyright notice,
+ *   this list of conditions and the following disclaimer in the documentation
+ *   and/or other materials provided with the distribution.
+ * - The names of contributors may not be used to endorse or promote products
+ *   derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *)
+
+structure Effective :> EFFECTIZE = struct
+
+open Core
+
+structure U = CoreUtil
+
+structure IM = IntBinaryMap
+structure SS = BinarySetFn(struct
+                           type ord_key = string
+                           val compare = String.compare
+                           end)
+
+val effectful = ["dml", "nextval", "send"]
+val effectful = SS.addList (SS.empty, effectful)
+
+fun effectize file =
+    let
+        fun exp evs e =
+            case e of
+                EFfi ("Basis", s) => SS.member (effectful, s)
+              | EFfiApp ("Basis", s, _) => SS.member (effectful, s)
+              | ENamed n => IM.inDomain (evs, n)
+              | EServerCall (n, _, _, _) => IM.inDomain (evs, n)
+              | _ => false
+
+        fun couldWrite evs = U.Exp.exists {kind = fn _ => false,
+                                           con = fn _ => false,
+                                           exp = exp evs}
+
+        fun doDecl (d, evs) =
+            case #1 d of
+                DVal (x, n, t, e, s) =>
+                (d, if couldWrite evs e then
+                        IM.insert (evs, n, (#2 d, s))
+                    else
+                        evs)
+              | DValRec vis =>
+                let
+                    fun oneRound evs =
+                        foldl (fn ((_, n, _, e, s), (changed, evs)) =>
+                                if couldWrite evs e andalso not (IM.inDomain (evs, n)) then
+                                    (true, IM.insert (evs, n, (#2 d, s)))
+                                else
+                                    (changed, evs)) (false, evs) vis
+
+                    fun loop evs =
+                        let
+                            val (b, evs) = oneRound evs
+                        in
+                            if b then
+                                loop evs
+                            else
+                                evs
+                        end
+                in
+                    (d, loop evs)
+                end
+              | DExport (Link, n) =>
+                (case IM.find (evs, n) of
+                     NONE => ()
+                   | SOME (loc, s) => ErrorMsg.errorAt loc ("A link (" ^ s ^ ") could cause side effects; try implementing it with a form instead");
+                 (d, evs))
+              | DExport (Action _, n) =>
+                ((DExport (Action (if IM.inDomain (evs, n) then
+                                       ReadWrite
+                                   else
+                                       ReadOnly), n), #2 d),
+                 evs)
+              | DExport (Rpc _, n) =>
+                ((DExport (Rpc (if IM.inDomain (evs, n) then
+                                    ReadWrite
+                                else
+                                    ReadOnly), n), #2 d),
+                 evs)
+              | _ => (d, evs)
+
+        val (file, _) = ListUtil.foldlMap doDecl IM.empty file
+    in
+        file
+    end
+
+end