# HG changeset patch # User Adam Chlipala # Date 1271005885 14400 # Node ID 62af4cacd1910d3aba1aa5b71dddb044111b4ab3 # Parent 117f13bdc1fd4b2155a7b51a71e6977130ccf1c4 Update policies diff -r 117f13bdc1fd -r 62af4cacd191 lib/ur/basis.urs --- a/lib/ur/basis.urs Sun Apr 11 12:45:15 2010 -0400 +++ b/lib/ur/basis.urs Sun Apr 11 13:11:25 2010 -0400 @@ -812,5 +812,9 @@ => sql_query [] ([Old = fs] ++ tables) [] -> sql_policy +val mayUpdate : fs ::: {Type} -> tables ::: {{Type}} -> [[Old, New] ~ tables] + => sql_query [] ([Old = fs, New = fs] ++ tables) [] + -> sql_policy + val debug : string -> transaction unit diff -r 117f13bdc1fd -r 62af4cacd191 src/iflow.sml --- a/src/iflow.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/iflow.sml Sun Apr 11 13:11:25 2010 -0400 @@ -958,7 +958,7 @@ in reset (); (*Print.prefaces "Big go" [("hyps", Print.p_list p_atom hyps), - ("goals", Print.p_list p_atom goals)];*) + ("goals", Print.p_list p_atom goals)];*) gls goals (fn () => false) [] end handle Cc.Contradiction => true @@ -1257,6 +1257,7 @@ datatype dml = Insert of string * (string * sqexp) list | Delete of string * sqexp + | Update of string * (string * sqexp) list * sqexp val insert = log "insert" (wrapP (follow (const "INSERT INTO ") @@ -1277,9 +1278,24 @@ sqexp))) (fn ((), (tab, ((), es))) => (tab, es))) +val setting = log "setting" + (wrap (follow uw_ident (follow (const " = ") sqexp)) + (fn (f, ((), e)) => (f, e))) + +val update = log "update" + (wrap (follow (const "UPDATE ") + (follow uw_ident + (follow (const " AS T_T SET ") + (follow (list setting) + (follow (ws (const "WHERE ")) + sqexp))))) + (fn ((), (tab, ((), (fs, ((), e))))) => + (tab, fs, e))) + val dml = log "dml" (altL [wrap insert Insert, - wrap delete Delete]) + wrap delete Delete, + wrap update Update]) fun removeDups (ls : (string * string) list) = case ls of @@ -1576,6 +1592,51 @@ end end +fun updateProp rvN rv e = + let + fun default () = (print ("Warning: Information flow checker can't parse SQL query at " + ^ ErrorMsg.spanToString (#2 e) ^ "\n"); + Unknown) + in + case parse query e of + NONE => default () + | SOME r => + let + val (rvs, rvN) = ListUtil.foldlMap (fn ((_, v), rvN) => + let + val (rvN, e) = rv rvN + in + ((v, e), rvN) + end) rvN (#From r) + + fun rvOf v = + case List.find (fn (v', _) => v' = v) rvs of + NONE => raise Fail "Iflow.insertProp: Bad table variable" + | SOME (_, e) => e + + val p = + foldl (fn ((t, v), p) => + let + val t = + case v of + "New" => "$New" + | _ => t + in + And (p, Reln (Sql t, [rvOf v])) + end) True (#From r) + + val expIn = expIn rv [] rvOf + in + And (Reln (Sql "$Old", [rvOf "Old"]), + case #Where r of + NONE => p + | SOME e => + case expIn (e, rvN) of + (inr p', _) => And (p, p') + | _ => p) + end + end + fun evalPat env e (pt, _) = case pt of PWild => (env, True) @@ -1659,6 +1720,9 @@ val deleted : t -> dml list val addDelete : t * dml -> t + + val updated : t -> dml list + val addUpdate : t * dml -> t end = struct type t = {Var : int, @@ -1666,14 +1730,16 @@ Path : (check * cflow) list, Sent : (check * flow) list, Insert : dml list, - Delete : dml list} + Delete : dml list, + Update : dml list} fun create {Var = v, Ambient = p} = {Var = v, Ambient = p, Path = [], Sent = [], Insert = [], - Delete = []} + Delete = [], + Update = []} fun curVar (t : t) = #Var t fun nextVar (t : t) = ({Var = #Var t + 1, @@ -1681,7 +1747,8 @@ Path = #Path t, Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t}, #Var t) + Delete = #Delete t, + Update = #Update t}, #Var t) fun ambient (t : t) = #Ambient t fun setAmbient (t : t, p) = {Var = #Var t, @@ -1689,7 +1756,8 @@ Path = #Path t, Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun paths (t : t) = #Path t fun addPath (t : t, c) = {Var = #Var t, @@ -1697,25 +1765,29 @@ Path = c :: #Path t, Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun addPaths (t : t, cs) = {Var = #Var t, Ambient = #Ambient t, Path = cs @ #Path t, Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun clearPaths (t : t) = {Var = #Var t, Ambient = #Ambient t, Path = [], Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun setPaths (t : t, cs) = {Var = #Var t, Ambient = #Ambient t, Path = cs, Sent = #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun sent (t : t) = #Sent t fun addSent (t : t, c) = {Var = #Var t, @@ -1723,13 +1795,15 @@ Path = #Path t, Sent = c :: #Sent t, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun setSent (t : t, cs) = {Var = #Var t, Ambient = #Ambient t, Path = #Path t, Sent = cs, Insert = #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun inserted (t : t) = #Insert t fun addInsert (t : t, c) = {Var = #Var t, @@ -1737,7 +1811,8 @@ Path = #Path t, Sent = #Sent t, Insert = c :: #Insert t, - Delete = #Delete t} + Delete = #Delete t, + Update = #Update t} fun deleted (t : t) = #Delete t fun addDelete (t : t, c) = {Var = #Var t, @@ -1745,7 +1820,17 @@ Path = #Path t, Sent = #Sent t, Insert = #Insert t, - Delete = c :: #Delete t} + Delete = c :: #Delete t, + Update = #Update t} + +fun updated (t : t) = #Update t +fun addUpdate (t : t, c) = {Var = #Var t, + Ambient = #Ambient t, + Path = #Path t, + Sent = #Sent t, + Insert = #Insert t, + Delete = #Delete t, + Update = c :: #Update t} end @@ -1984,8 +2069,7 @@ (st, Var n) end - val expIn = expIn rv env (fn "New" => Var new - | _ => raise Fail "Iflow.evalExp: Bad field expression in UPDATE") + val expIn = expIn rv env (fn _ => raise Fail "Iflow.evalExp: Bad field expression in INSERT") val (es, st) = ListUtil.foldlMap (fn ((x, e), st) => @@ -2026,6 +2110,45 @@ Reln (Sql tab, [Var old]))) in (Recd [], St.addDelete (st, (loc, And (St.ambient st, p)))) + end + | Update (tab, fs, e) => + let + val (st, new) = St.nextVar st + val (st, old) = St.nextVar st + + fun rv st = + let + val (st, n) = St.nextVar st + in + (st, Var n) + end + + val expIn = expIn rv env (fn "T" => Var old + | _ => raise Fail "Iflow.evalExp: Bad field expression in UPDATE") + + val (fs, st) = ListUtil.foldlMap + (fn ((x, e), st) => + let + val (e, st) = case expIn (e, st) of + (inl e, st) => (e, st) + | (inr _, _) => raise Fail + ("Iflow.evalExp: Selecting " + ^ "boolean expression") + in + ((x, e), st) + end) + st fs + + val (p, st) = case expIn (e, st) of + (inl e, _) => raise Fail "Iflow.evalExp: UPDATE with non-boolean" + | (inr p, st) => (p, st) + + val p = And (p, + And (Reln (Sql "$New", [Recd fs]), + And (Reln (Sql "$Old", [Var old]), + Reln (Sql tab, [Var old])))) + in + (Recd [], St.addUpdate (st, (loc, And (St.ambient st, p)))) end) | ENextval _ => default () @@ -2063,7 +2186,7 @@ DExport (_, _, n, _, _, _) => IS.add (exptd, n) | _ => exptd) IS.empty file - fun decl ((d, _), (vals, inserts, deletes, client, insert, delete)) = + fun decl ((d, _), (vals, inserts, deletes, updates, client, insert, delete, update)) = case d of DVal (_, n, _, e, _) => let @@ -2083,7 +2206,8 @@ val (_, st) = evalExp env (e, St.create {Var = nv, Ambient = p}) in - (St.sent st @ vals, St.inserted st @ inserts, St.deleted st @ deletes, client, insert, delete) + (St.sent st @ vals, St.inserted st @ inserts, St.deleted st @ deletes, St.updated st @ updates, + client, insert, delete, update) end | DPolicy pol => @@ -2095,27 +2219,34 @@ let val (_, p, _, _, outs) = queryProp [] 0 rv SomeCol e in - (vals, inserts, deletes, (p, outs) :: client, insert, delete) + (vals, inserts, deletes, updates, (p, outs) :: client, insert, delete, update) end | PolInsert e => let val p = insertProp 0 rv e in - (vals, inserts, deletes, client, p :: insert, delete) + (vals, inserts, deletes, updates, client, p :: insert, delete, update) end | PolDelete e => let val p = deleteProp 0 rv e in - (vals, inserts, deletes, client, insert, p :: delete) + (vals, inserts, deletes, updates, client, insert, p :: delete, update) + end + | PolUpdate e => + let + val p = updateProp 0 rv e + in + (vals, inserts, deletes, updates, client, insert, delete, p :: update) end end - | _ => (vals, inserts, deletes, client, insert, delete) + | _ => (vals, inserts, deletes, updates, client, insert, delete, update) val () = reset () - val (vals, inserts, deletes, client, insert, delete) = foldl decl ([], [], [], [], [], []) file + val (vals, inserts, deletes, updates, client, insert, delete, update) = + foldl decl ([], [], [], [], [], [], [], []) file val decompH = decomp true (fn (e1, e2) => e1 andalso e2 ()) val decompG = decomp false (fn (e1, e2) => e1 orelse e2 ()) @@ -2168,7 +2299,8 @@ end) vals; doDml (inserts, insert); - doDml (deletes, delete) + doDml (deletes, delete); + doDml (updates, update) end val check = fn file => diff -r 117f13bdc1fd -r 62af4cacd191 src/mono.sml --- a/src/mono.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/mono.sml Sun Apr 11 13:11:25 2010 -0400 @@ -127,6 +127,7 @@ PolClient of exp | PolInsert of exp | PolDelete of exp + | PolUpdate of exp datatype decl' = DDatatype of (string * int * (string * int * typ option) list) list diff -r 117f13bdc1fd -r 62af4cacd191 src/mono_print.sml --- a/src/mono_print.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/mono_print.sml Sun Apr 11 13:11:25 2010 -0400 @@ -423,6 +423,9 @@ | PolDelete e => box [string "mayDelete", space, p_exp env e] + | PolUpdate e => box [string "mayUpdate", + space, + p_exp env e] fun p_decl env (dAll as (d, _) : decl) = case d of diff -r 117f13bdc1fd -r 62af4cacd191 src/mono_shake.sml --- a/src/mono_shake.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/mono_shake.sml Sun Apr 11 13:11:25 2010 -0400 @@ -64,6 +64,7 @@ PolClient e1 => e1 | PolInsert e1 => e1 | PolDelete e1 => e1 + | PolUpdate e1 => e1 in usedVars st e1 end diff -r 117f13bdc1fd -r 62af4cacd191 src/mono_util.sml --- a/src/mono_util.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/mono_util.sml Sun Apr 11 13:11:25 2010 -0400 @@ -550,6 +550,9 @@ | PolDelete e => S.map2 (mfe ctx e, PolDelete) + | PolUpdate e => + S.map2 (mfe ctx e, + PolUpdate) and mfvi ctx (x, n, t, e, s) = S.bind2 (mft t, diff -r 117f13bdc1fd -r 62af4cacd191 src/monoize.sml --- a/src/monoize.sml Sun Apr 11 12:45:15 2010 -0400 +++ b/src/monoize.sml Sun Apr 11 13:11:25 2010 -0400 @@ -3750,6 +3750,8 @@ (e, L'.PolInsert) | L.EApp ((L.ECApp ((L.ECApp ((L.EFfi ("Basis", "mayDelete"), _), _), _), _), _), e) => (e, L'.PolDelete) + | L.EApp ((L.ECApp ((L.ECApp ((L.EFfi ("Basis", "mayUpdate"), _), _), _), _), _), e) => + (e, L'.PolUpdate) | _ => (poly (); (e, L'.PolClient)) val (e, fm) = monoExp (env, St.empty, fm) e