Mise à jour terminée. Pour connaître les apports de la version 13.8.4 par rapport à notre ancienne version vous pouvez lire les "Release Notes" suivantes :
https://about.gitlab.com/releases/2021/02/11/security-release-gitlab-13-8-4-released/
https://about.gitlab.com/releases/2021/02/05/gitlab-13-8-3-released/

pmodule.ml 46.3 KB
Newer Older
1 2 3
(********************************************************************)
(*                                                                  *)
(*  The Why3 Verification Platform   /   The Why3 Development Team  *)
Guillaume Melquiond's avatar
Guillaume Melquiond committed
4
(*  Copyright 2010-2018   --   Inria - CNRS - Paris-Sud University  *)
5 6 7 8 9 10 11
(*                                                                  *)
(*  This software is distributed under the terms of the GNU Lesser  *)
(*  General Public License version 2.1, with the special exception  *)
(*  on linking described in file LICENSE.                           *)
(*                                                                  *)
(********************************************************************)

12
open Wstdlib
13 14
open Ident
open Ty
15
open Term
Andrei Paskevich's avatar
Andrei Paskevich committed
16
open Decl
17 18 19
open Theory
open Ity
open Expr
20
open Pdecl
21 22 23 24 25 26

(** *)

type prog_symbol =
  | PV of pvsymbol
  | RS of rsymbol
27
  | OO of Srs.t
28 29 30 31

type namespace = {
  ns_ts : itysymbol   Mstr.t;  (* type symbols *)
  ns_ps : prog_symbol Mstr.t;  (* program symbols *)
32
  ns_xs : xsymbol     Mstr.t;  (* exception symbols *)
33 34 35 36 37 38
  ns_ns : namespace   Mstr.t;  (* inner namespaces *)
}

let empty_ns = {
  ns_ts = Mstr.empty;
  ns_ps = Mstr.empty;
39
  ns_xs = Mstr.empty;
40 41 42 43 44
  ns_ns = Mstr.empty;
}

let ns_replace eq chk x vo vn =
  if not chk then vn else
45
  if eq vo vn then vo else
46 47
  raise (ClashSymbol x)

48 49
let merge_ts = ns_replace its_equal
let merge_xs = ns_replace xs_equal
50

51
type overload =
52 53 54
  | FixedRes of ity (* t -> t -> ... -> T *)
  | SameType        (* t -> t -> ... -> t *)
  | NoOver          (* neither *)
55 56 57 58

let overload_of_rs {rs_cty = cty} =
  if cty.cty_effect.eff_ghost then NoOver else
  if cty.cty_mask <> MaskVisible then NoOver else
59
  let same_type ity a = not a.pv_ghost && ity_equal a.pv_ity ity in
60
  match cty.cty_args with
61 62 63
  | a::al when not a.pv_ghost && List.for_all (same_type a.pv_ity) al ->
      let res = cty.cty_result in
      if ity_equal res a.pv_ity then SameType else
64
      if ity_closed res && res.ity_pure then FixedRes res else NoOver
65 66
  | _ -> NoOver

67 68 69 70 71 72 73
let same_overload r1 r2 =
  List.length r1.rs_cty.cty_args = List.length r2.rs_cty.cty_args &&
  match overload_of_rs r1, overload_of_rs r2 with
  | SameType, SameType -> true
  | FixedRes t1, FixedRes t2 -> ity_equal t1 t2
  | _ -> false (* two NoOver's are not the same *)

74 75
exception IncompatibleNotation of string

Andrei Paskevich's avatar
Andrei Paskevich committed
76 77 78 79 80 81 82 83 84 85 86 87
let merge_ps chk x vo vn =
  let fsty rs = (List.hd rs.rs_cty.cty_args).pv_ity in
  if chk then match vo, vn with (* export namespace *)
    (* currently, we have no way to export notation *)
    | _, OO _ | OO _, _ -> assert false
    | PV v1, PV v2 when pv_equal v1 v2 -> vo
    | RS r1, RS r2 when rs_equal r1 r2 -> vo
    | _ -> raise (ClashSymbol x)
  else match vo, vn with (* import namespace *)
    (* once again, no way to export notation *)
    | _, OO _ -> assert false
    (* but we can merge two compatible symbols *)
88 89
    | RS r1, RS r2 ->
        if rs_equal r1 r2 then vo else
90 91
        if not (same_overload r1 r2) then vn else
        if ity_equal (fsty r1) (fsty r2) then vn else
Andrei Paskevich's avatar
Andrei Paskevich committed
92 93 94
        OO (Srs.add r2 (Srs.singleton r1))
    (* or add a compatible symbol to notation *)
    | OO s1, RS r2 ->
95 96
        if Srs.mem r2 s1 then vo else
        let r1 = Srs.choose s1 in
97
        if not (same_overload r1 r2) then vn else
98
        let ty = fsty r2 in
99
        let confl r = not (ity_equal (fsty r) ty) in
Andrei Paskevich's avatar
Andrei Paskevich committed
100 101 102 103
        let s1 = Srs.filter confl s1 in
        if Srs.is_empty s1 then vn else
        OO (Srs.add r2 s1)
    | _ -> vn
104

105 106 107 108 109 110 111 112 113
let rec merge_ns chk _ no nn =
  if no == nn then no else
  let union merge o n =
    let merge x vo vn = Some (merge chk x vo vn) in
    if o == n then o else Mstr.union merge o n in
  { ns_ts = union merge_ts no.ns_ts nn.ns_ts;
    ns_ps = union merge_ps no.ns_ps nn.ns_ps;
    ns_xs = union merge_xs no.ns_xs nn.ns_xs;
    ns_ns = union merge_ns no.ns_ns nn.ns_ns }
114

115 116
let ns_add merge chk x vn m = Mstr.change (function
  | Some vo -> Some (merge chk x vo vn)
117 118
  | None    -> Some vn) x m

119 120 121 122 123 124
let add_ts chk x ts ns = { ns with ns_ts = ns_add merge_ts chk x ts ns.ns_ts }
let add_ps chk x ps ns = { ns with ns_ps = ns_add merge_ps chk x ps ns.ns_ps }
let add_xs chk x xs ns = { ns with ns_xs = ns_add merge_xs chk x xs ns.ns_xs }
let add_ns chk x nn ns = { ns with ns_ns = ns_add merge_ns chk x nn ns.ns_ns }

let merge_ns chk nn no = merge_ns chk "" no nn (* swap arguments *)
125 126 127 128 129 130

let rec ns_find get_map ns = function
  | []   -> assert false
  | [a]  -> Mstr.find a (get_map ns)
  | a::l -> ns_find get_map (Mstr.find a ns.ns_ns) l

131 132 133 134 135 136 137 138 139 140 141 142
let ns_find_its = ns_find (fun ns -> ns.ns_ts)
let ns_find_xs  = ns_find (fun ns -> ns.ns_xs)
let ns_find_ns  = ns_find (fun ns -> ns.ns_ns)

let ns_find_prog_symbol ns s =
  let ps = ns_find (fun ns -> ns.ns_ps) ns s in
  match ps with
  | RS _ | PV _ -> ps
  | OO ss ->
      let rs1 = Expr.Srs.min_elt ss in
      let rs2 = Expr.Srs.max_elt ss in
      if Expr.rs_equal rs1 rs2 then RS rs1 else ps
143 144 145 146 147 148 149 150 151

let ns_find_pv ns s = match ns_find_prog_symbol ns s with
  | PV pv -> pv | _ -> raise Not_found

let ns_find_rs ns s = match ns_find_prog_symbol ns s with
  | RS rs -> rs | _ -> raise Not_found

(** {2 Module} *)

152
type pmodule = {
Andrei Paskevich's avatar
Andrei Paskevich committed
153 154 155 156 157 158 159 160 161 162 163
  mod_theory : theory;        (* pure theory *)
  mod_units  : mod_unit list; (* module declarations *)
  mod_export : namespace;     (* exported namespace *)
  mod_known  : known_map;     (* known identifiers *)
  mod_local  : Sid.t;         (* locally declared idents *)
  mod_used   : Sid.t;         (* used modules *)
}

and mod_unit =
  | Udecl  of pdecl
  | Uuse   of pmodule
Andrei Paskevich's avatar
Andrei Paskevich committed
164
  | Uclone of mod_inst
Andrei Paskevich's avatar
Andrei Paskevich committed
165
  | Umeta  of meta * meta_arg list
166
  | Uscope of string * mod_unit list
Andrei Paskevich's avatar
Andrei Paskevich committed
167

Andrei Paskevich's avatar
Andrei Paskevich committed
168
and mod_inst = {
Andrei Paskevich's avatar
Andrei Paskevich committed
169
  mi_mod : pmodule;
170 171
  mi_ty  : ity Mts.t;
  mi_ts  : itysymbol Mts.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
172 173
  mi_ls  : lsymbol Mls.t;
  mi_pr  : prsymbol Mpr.t;
174
  mi_pk  : prop_kind Mpr.t;
175
  mi_pv  : pvsymbol Mvs.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
176
  mi_rs  : rsymbol Mrs.t;
177
  mi_xs  : xsymbol Mxs.t;
178
  mi_df  : prop_kind;
179 180
}

181 182 183 184 185 186 187 188 189
let empty_mod_inst m = {
  mi_mod = m;
  mi_ty  = Mts.empty;
  mi_ts  = Mts.empty;
  mi_ls  = Mls.empty;
  mi_pr  = Mpr.empty;
  mi_pk  = Mpr.empty;
  mi_pv  = Mvs.empty;
  mi_rs  = Mrs.empty;
190
  mi_xs  = Mxs.empty;
191
  mi_df  = Plemma;
192 193
}

194 195
(** {2 Module under construction} *)

196
type pmodule_uc = {
197
  muc_theory : theory_uc;
Andrei Paskevich's avatar
Andrei Paskevich committed
198
  muc_units  : mod_unit list;
199 200 201 202 203
  muc_import : namespace list;
  muc_export : namespace list;
  muc_known  : known_map;
  muc_local  : Sid.t;
  muc_used   : Sid.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
204
  muc_env    : Env.env;
205 206 207 208
}

let empty_module env n p = {
  muc_theory = create_theory ~path:p n;
Andrei Paskevich's avatar
Andrei Paskevich committed
209
  muc_units  = [];
210 211 212 213 214 215 216 217 218 219 220 221 222
  muc_import = [empty_ns];
  muc_export = [empty_ns];
  muc_known  = Mid.empty;
  muc_local  = Sid.empty;
  muc_used   = Sid.empty;
  muc_env    = env;
}

let close_module, restore_module =
  let h = Hid.create 17 in
  (fun uc ->
     let th = close_theory uc.muc_theory in (* catches errors *)
     let m = { mod_theory = th;
Andrei Paskevich's avatar
Andrei Paskevich committed
223
               mod_units  = List.rev uc.muc_units;
224 225 226 227 228 229 230 231
               mod_export = List.hd uc.muc_export;
               mod_known  = uc.muc_known;
               mod_local  = uc.muc_local;
               mod_used   = uc.muc_used; } in
     Hid.add h th.th_name m;
     m),
  (fun th -> Hid.find h th.th_name)

232
let open_scope uc s = match uc.muc_import with
233
  | ns :: _ -> { uc with
234
      muc_theory = Theory.open_scope uc.muc_theory s;
235
      muc_units  = [Uscope (s, uc.muc_units)];
236 237 238 239
      muc_import =       ns :: uc.muc_import;
      muc_export = empty_ns :: uc.muc_export; }
  | [] -> assert false

240 241
let close_scope uc ~import =
  let th = Theory.close_scope uc.muc_theory ~import in
Andrei Paskevich's avatar
Andrei Paskevich committed
242
  match List.rev uc.muc_units, uc.muc_import, uc.muc_export with
243
  | [Uscope (_,ul1)], _ :: sti, _ :: ste -> (* empty scope *)
Andrei Paskevich's avatar
Andrei Paskevich committed
244 245
      { uc with muc_theory = th;  muc_units  = ul1;
                muc_import = sti; muc_export = ste; }
246
  | Uscope (s,ul1) :: ul0, _ :: i1 :: sti, e0 :: e1 :: ste ->
247 248 249 250
      let i1 = if import then merge_ns false e0 i1 else i1 in
      let i1 = add_ns false s e0 i1 in
      let e1 = add_ns true  s e0 e1 in
      { uc with
251
          muc_theory = th;
252
          muc_units  = Uscope (s,ul0) :: ul1;
253 254
          muc_import = i1 :: sti;
          muc_export = e1 :: ste; }
Andrei Paskevich's avatar
Andrei Paskevich committed
255
  | _ -> assert false
256

257 258 259 260 261 262 263
let import_scope uc ql = match uc.muc_import with
  | i1 :: sti ->
      let th = Theory.import_scope uc.muc_theory ql in
      let i1 = merge_ns false (ns_find_ns i1 ql) i1 in
      { uc with muc_theory = th; muc_import = i1::sti }
  | _ -> assert false

Andrei Paskevich's avatar
Andrei Paskevich committed
264 265 266 267 268 269
let use_export uc ({mod_theory = mth} as m) =
  let th = Theory.use_export uc.muc_theory mth in
  let uc = if Sid.mem mth.th_name uc.muc_used then uc
    else { uc with
      muc_known = merge_known uc.muc_known m.mod_known;
      muc_used  = Sid.add mth.th_name uc.muc_used } in
270 271 272
  match uc.muc_import, uc.muc_export with
  | i0 :: sti, e0 :: ste -> { uc with
      muc_theory = th;
Andrei Paskevich's avatar
Andrei Paskevich committed
273 274 275
      muc_units  = Uuse m :: uc.muc_units;
      muc_import = merge_ns false m.mod_export i0 :: sti;
      muc_export = merge_ns true  m.mod_export e0 :: ste; }
276 277
  | _ -> assert false

Andrei Paskevich's avatar
Andrei Paskevich committed
278 279 280
let add_meta uc m al = { uc with
  muc_theory = Theory.add_meta uc.muc_theory m al;
  muc_units  = Umeta (m,al) :: uc.muc_units; }
281

282 283
let store_path, store_module, restore_path =
  let id_to_path = Wid.create 17 in
Andrei Paskevich's avatar
Andrei Paskevich committed
284
  let store_path {muc_theory = uc} path id =
285 286
    (* this symbol already belongs to some theory *)
    if Wid.mem id_to_path id then () else
Andrei Paskevich's avatar
Andrei Paskevich committed
287 288 289
    let prefix = List.rev (id.id_string :: path @ uc.uc_prefix) in
    Wid.set id_to_path id (uc.uc_path, uc.uc_name.id_string, prefix) in
  let store_module {mod_theory = {th_name = id} as th} =
290 291
    (* this symbol is already a module *)
    if Wid.mem id_to_path id then () else
Andrei Paskevich's avatar
Andrei Paskevich committed
292
    Wid.set id_to_path id (th.th_path, id.id_string, []) in
293 294 295 296 297 298 299 300
  let restore_path id = Wid.find id_to_path id in
  store_path, store_module, restore_path

let close_module uc =
  let m = close_module uc in
  store_module m;
  m

301
(*
302 303 304 305
let count_regions {muc_known = kn} {pv_ity = ity} mr =
  let add_reg r mr = Mreg.change (fun n -> Some (n <> None)) r mr in
  let meet mr1 mr2 = Mreg.union (fun _ x y -> Some (x || y)) mr1 mr2 in
  let join mr1 mr2 = Mreg.union (fun _ _ _ -> Some true) mr1 mr2 in
306
  let rec down mr ity = if ity.ity_imm then mr else
307 308 309
    match ity.ity_node with
    | Ityreg r -> fields (add_reg r mr) r.reg_its r.reg_args r.reg_regs
    | Ityapp (s,tl,rl) -> fields mr s tl rl
310
    | Ityvar _ -> assert false
311 312 313
  and fields mr s tl rl =
    if s.its_private && rl = [] then mr else
    let isb = its_match_regs s tl rl in
314 315 316 317 318 319 320 321 322 323 324
    let add_ity mr ity = down mr (ity_full_inst isb ity) in
    let add_proj mr f = add_ity mr f.rs_cty.cty_result in
    let add_field mr v = add_ity mr v.pv_ity in
    let d = find_its_defn kn s in
    if d.itd_constructors = [] then
      List.fold_left add_proj mr d.itd_fields
    else
      join mr (List.fold_left (fun mr c ->
        meet mr (List.fold_left add_field Mreg.empty c.rs_cty.cty_args))
        Mreg.empty d.itd_constructors) in
  down mr ity
325
*)
326

327 328 329 330 331 332 333 334
let add_symbol add id v uc =
  store_path uc [] id;
  match uc.muc_import, uc.muc_export with
  | i0 :: sti, e0 :: ste -> { uc with
      muc_import = add false id.id_string v i0 :: sti;
      muc_export = add true  id.id_string v e0 :: ste }
  | _ -> assert false

Andrei Paskevich's avatar
Andrei Paskevich committed
335
let add_pdecl_no_logic uc d =
336
  let uc = { uc with
Andrei Paskevich's avatar
Andrei Paskevich committed
337
    muc_units = Udecl d :: uc.muc_units;
338
    muc_known = known_add_decl uc.muc_known d;
339
    muc_local = Sid.union uc.muc_local d.pd_news } in
340
  let add_rs uc s = add_symbol add_ps s.rs_name (RS s) uc in
341
  let add_rd uc d = add_rs uc d.rec_sym in
342 343 344 345 346 347 348 349
  match d.pd_node with
  | PDtype tdl ->
      let add uc d =
        let uc = List.fold_left add_rs uc d.itd_fields in
        let uc = List.fold_left add_rs uc d.itd_constructors in
        add_symbol add_ts d.itd_its.its_ts.ts_name d.itd_its uc in
      List.fold_left add uc tdl
  | PDlet (LDvar (v,_)) -> add_symbol add_ps v.pv_vs.vs_name (PV v) uc
350
  | PDlet (LDsym (s,_)) -> add_rs uc s
351
  | PDlet (LDrec l) -> List.fold_left add_rd uc l
352
  | PDexn xs -> add_symbol add_xs xs.xs_name xs uc
353
  | PDpure -> uc
354

355
let add_pdecl_raw ?(warn=true) uc d =
Andrei Paskevich's avatar
Andrei Paskevich committed
356
  let uc = add_pdecl_no_logic uc d in
357
  let th = List.fold_left (add_decl ~warn) uc.muc_theory d.pd_pure in
358 359
  let add_meta th (m,l) = Theory.add_meta th m l in
  let th = List.fold_left add_meta th d.pd_meta in
Andrei Paskevich's avatar
Andrei Paskevich committed
360 361
  { uc with muc_theory = th }

362 363
(** {2 Builtin symbols} *)

Andrei Paskevich's avatar
Andrei Paskevich committed
364 365
let dummy_env = Env.create_env []

366
let builtin_module =
Andrei Paskevich's avatar
Andrei Paskevich committed
367 368 369 370
  let uc = empty_module dummy_env (id_fresh "BuiltIn") ["why3";"BuiltIn"] in
  let uc = add_pdecl_no_logic uc pd_int in
  let uc = add_pdecl_no_logic uc pd_real in
  let uc = add_pdecl_no_logic uc pd_equ in
371 372
  let m = close_module uc in
  { m with mod_theory = builtin_theory }
373 374

let bool_module =
Andrei Paskevich's avatar
Andrei Paskevich committed
375 376
  let uc = empty_module dummy_env (id_fresh "Bool") ["why3";"Bool"] in
  let uc = add_pdecl_no_logic uc pd_bool in
377 378
  let m = close_module uc in
  { m with mod_theory = bool_theory }
379 380

let highord_module =
Andrei Paskevich's avatar
Andrei Paskevich committed
381 382 383
  let uc = empty_module dummy_env (id_fresh "HighOrd") ["why3";"HighOrd"] in
  let uc = add_pdecl_no_logic uc pd_func in
  let uc = add_pdecl_no_logic uc pd_func_app in
384 385
  let m = close_module uc in
  { m with mod_theory = highord_theory }
386

387 388
let tuple_module = Hint.memo 17 (fun n ->
  let nm = "Tuple" ^ string_of_int n in
Andrei Paskevich's avatar
Andrei Paskevich committed
389 390
  let uc = empty_module dummy_env (id_fresh nm) ["why3";nm] in
  let uc = add_pdecl_no_logic uc (pd_tuple n) in
391 392
  let m = close_module uc in
  { m with mod_theory = tuple_theory n })
393 394

let unit_module =
Andrei Paskevich's avatar
Andrei Paskevich committed
395
  let uc = empty_module dummy_env (id_fresh "Unit") ["why3";"Unit"] in
396
  let uc = use_export uc (tuple_module 0) in
397
  let add uc d = add_pdecl_raw ~warn:false uc d in
398
  let td = create_alias_decl (id_fresh "unit") [] ity_unit in
399
  close_module (List.fold_left add uc (create_type_decl [td]))
400 401

let create_module env ?(path=[]) n =
Andrei Paskevich's avatar
Andrei Paskevich committed
402
  let m = empty_module env n path in
403 404 405 406 407
  let m = use_export m builtin_module in
  let m = use_export m bool_module in
  let m = use_export m unit_module in
  m

Andrei Paskevich's avatar
Andrei Paskevich committed
408 409 410 411 412 413 414
let add_use uc d = Sid.fold (fun id uc ->
  if id_equal id ts_func.ts_name then
    use_export uc highord_module
  else match is_ts_tuple_id id with
  | Some n -> use_export uc (tuple_module n)
  | None -> uc) (Mid.set_diff d.pd_syms uc.muc_known) uc

415 416
let mk_vc uc d = Vc.vc uc.muc_env uc.muc_known uc.muc_theory d

417
let add_pdecl ?(warn=true) ~vc uc d =
Andrei Paskevich's avatar
Andrei Paskevich committed
418
  let uc = add_use uc d in
419
  let dl = if vc then mk_vc uc d else [] in
420 421 422 423
  (* verification conditions must not add additional dependencies
     on built-in theories like TupleN or HighOrd. Also, we expect
     int.Int or any other library theory to be in the context:
     importing them automatically seems to be too invasive. *)
424
  add_pdecl_raw ~warn (List.fold_left (add_pdecl_raw ~warn) uc dl) d
425

Andrei Paskevich's avatar
Andrei Paskevich committed
426 427 428 429
(** {2 Cloning} *)

type clones = {
  cl_local : Sid.t;
430 431
  mutable ty_table : ity Mts.t;
  mutable ts_table : itysymbol Mts.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
432 433 434
  mutable ls_table : lsymbol Mls.t;
  mutable pr_table : prsymbol Mpr.t;
  mutable rn_table : region Mreg.t;
435
  mutable fd_table : pvsymbol Mpv.t;
436
  mutable pv_table : pvsymbol Mvs.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
437
  mutable rs_table : rsymbol Mrs.t;
438
  mutable xs_table : xsymbol Mxs.t;
Andrei Paskevich's avatar
Andrei Paskevich committed
439 440
}

441 442
let empty_clones m = {
  cl_local = m.mod_local;
443
  ty_table = Mts.empty;
Andrei Paskevich's avatar
Andrei Paskevich committed
444 445 446 447
  ts_table = Mts.empty;
  ls_table = Mls.empty;
  pr_table = Mpr.empty;
  rn_table = Mreg.empty;
448
  fd_table = Mpv.empty;
449
  pv_table = Mvs.empty;
Andrei Paskevich's avatar
Andrei Paskevich committed
450
  rs_table = Mrs.empty;
451
  xs_table = Mxs.empty;
Andrei Paskevich's avatar
Andrei Paskevich committed
452 453 454 455
}

(* populate the clone structure *)

456
let rec sm_trans_ty tym tsm ty = match ty.ty_node with
457
  | Tyapp (s, tl) ->
458
      let tl = List.map (sm_trans_ty tym tsm) tl in
459
      begin match Mts.find_opt s tsm with
460 461
      | Some its -> ty_app its.its_ts tl
      | None -> begin match Mts.find_opt s tym with
462 463
      | Some ity -> ty_inst (ts_match_args s tl) (ty_of_ity ity)
      | None -> ty_app s tl
464
      end end
465 466
  | Tyvar _ -> ty

Andrei Paskevich's avatar
Andrei Paskevich committed
467
let clone_ty cl ty = sm_trans_ty cl.ty_table cl.ts_table ty
468

469 470 471
let cl_find_its cl its =
  if not (Sid.mem its.its_ts.ts_name cl.cl_local) then its
  else Mts.find its.its_ts cl.ts_table
472

473 474 475
let cl_find_ts cl ts =
  if not (Sid.mem ts.ts_name cl.cl_local) then ts
  else (Mts.find ts cl.ts_table).its_ts
476

Andrei Paskevich's avatar
Andrei Paskevich committed
477
let rec clone_ity cl ity = match ity.ity_node with
478
  | Ityreg r ->
Andrei Paskevich's avatar
Andrei Paskevich committed
479
      ity_reg (clone_reg cl r)
480
  | Ityapp (s, tl, rl) ->
Andrei Paskevich's avatar
Andrei Paskevich committed
481
      let tl = List.map (clone_ity cl) tl in
482
      let rl = List.map (clone_ity cl) rl in
483
      begin match Mts.find_opt s.its_ts cl.ts_table with
484 485 486
      | Some its -> ity_app_pure its tl rl
      | None -> (* creative indentation *)
      begin match Mts.find_opt s.its_ts cl.ty_table with
487
      | Some ity -> ity_full_inst (its_match_regs s tl rl) ity
488
      | None -> ity_app_pure s tl rl
489
      end end
Andrei Paskevich's avatar
Andrei Paskevich committed
490 491
  | Ityvar _ -> ity

Andrei Paskevich's avatar
Andrei Paskevich committed
492
and clone_reg cl reg =
493 494 495 496
  (* FIXME: what about top-level non-instantiated regions?
     We cannot check in cl.cl_local to see if they are there.
     Instead, we should prefill cl.pv_table and cl.rn_table
     with all top-level pvsymbols (local or external) before
497 498 499
     descending into a let_defn.
     TODO: add to module/uc a list of locally-defined toplevel
     variables, as well as a set of imported toplevel variables. *)
Andrei Paskevich's avatar
Andrei Paskevich committed
500
  try Mreg.find reg cl.rn_table with Not_found ->
Andrei Paskevich's avatar
Andrei Paskevich committed
501
  let tl = List.map (clone_ity cl) reg.reg_args in
502
  let rl = List.map (clone_ity cl) reg.reg_regs in
503
  let r = match Mts.find_opt reg.reg_its.its_ts cl.ts_table with
504 505
    | Some its ->
        create_region (id_clone reg.reg_name) its tl rl
506 507
    | None -> (* creative indentation *)
    begin match Mts.find_opt reg.reg_its.its_ts cl.ty_table with
508 509 510
    | Some {ity_node = Ityreg r} ->
        let sbs = its_match_regs reg.reg_its tl rl in
        let tl = List.map (ity_full_inst sbs) r.reg_args in
511
        let rl = List.map (ity_full_inst sbs) r.reg_regs in
512
        create_region (id_clone reg.reg_name) r.reg_its tl rl
513
    | Some _ -> assert false
514 515
    | None ->
        create_region (id_clone reg.reg_name) reg.reg_its tl rl
516
    end in
517 518
  cl.rn_table <- Mreg.add reg r cl.rn_table;
  r
Andrei Paskevich's avatar
Andrei Paskevich committed
519 520 521

let cl_find_ls cl ls =
  if not (Sid.mem ls.ls_name cl.cl_local) then ls
522
  else Mls.find ls cl.ls_table
Andrei Paskevich's avatar
Andrei Paskevich committed
523

Andrei Paskevich's avatar
Andrei Paskevich committed
524
let clone_term cl mv t = t_gen_map (clone_ty cl) (cl_find_ls cl) mv t
525

Andrei Paskevich's avatar
Andrei Paskevich committed
526
let clone_fmla cl f = clone_term cl Mvs.empty f (* for closed terms *)
Andrei Paskevich's avatar
Andrei Paskevich committed
527

Andrei Paskevich's avatar
Andrei Paskevich committed
528
let cl_find_pr cl pr =
Andrei Paskevich's avatar
Andrei Paskevich committed
529
  if not (Sid.mem pr.pr_name cl.cl_local) then pr
530
  else Mpr.find pr cl.pr_table
Andrei Paskevich's avatar
Andrei Paskevich committed
531

Andrei Paskevich's avatar
Andrei Paskevich committed
532
let cl_find_pv cl pv =
Andrei Paskevich's avatar
Andrei Paskevich committed
533
  if not (Sid.mem pv.pv_vs.vs_name cl.cl_local) then pv
534
  else Mvs.find pv.pv_vs cl.pv_table
Andrei Paskevich's avatar
Andrei Paskevich committed
535

Andrei Paskevich's avatar
Andrei Paskevich committed
536
let cl_find_rs cl rs =
Andrei Paskevich's avatar
Andrei Paskevich committed
537 538 539
  if not (Sid.mem rs.rs_name cl.cl_local) then rs
  else Mrs.find rs cl.rs_table

Andrei Paskevich's avatar
Andrei Paskevich committed
540
let cl_find_xs cl xs =
Andrei Paskevich's avatar
Andrei Paskevich committed
541
  if not (Sid.mem xs.xs_name cl.cl_local) then xs
542
  else Mxs.find xs cl.xs_table
Andrei Paskevich's avatar
Andrei Paskevich committed
543

544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574
let cl_init m inst =
  let cl = empty_clones m in
  let non_local id =
    if not (Sid.mem id cl.cl_local) then raise (NonLocal id) in
  Mts.iter (fun ts _ -> non_local ts.ts_name) inst.mi_ts;
  Mts.iter (fun ts _ -> non_local ts.ts_name) inst.mi_ty;
  let check_ls ls _ =
    non_local ls.ls_name;
    try  ignore (restore_rs ls); raise (BadInstance ls.ls_name)
    with Not_found -> () in
  Mls.iter check_ls inst.mi_ls;
  let check_rs rs _ =
    non_local rs.rs_name;
    match (Mid.find rs.rs_name m.mod_known).pd_node with
    | PDtype _ -> raise (BadInstance rs.rs_name)
    | PDlet  _ | PDexn  _ | PDpure -> () in
  Mrs.iter check_rs inst.mi_rs;
  Mvs.iter (fun vs _ -> non_local vs.vs_name) inst.mi_pv;
  Mxs.iter (fun xs _ -> non_local xs.xs_name) inst.mi_xs;
  let check_pk pr _ =
    non_local pr.pr_name;
    match (Mid.find pr.pr_name m.mod_known).pd_node with
    | PDtype _ | PDlet  _ | PDexn  _ -> raise (BadInstance pr.pr_name)
    | PDpure -> () in
  Mpr.iter check_pk inst.mi_pk;
  Mpr.iter (fun pr _ -> raise (BadInstance pr.pr_name)) inst.mi_pr;
  cl

(* clone declarations *)

let clone_ls cl ls =
Andrei Paskevich's avatar
Andrei Paskevich committed
575
  let constr = ls.ls_constr in
576
  let id = id_clone ls.ls_name in
Andrei Paskevich's avatar
Andrei Paskevich committed
577 578
  let at = List.map (clone_ty cl) ls.ls_args in
  let vt = Opt.map (clone_ty cl) ls.ls_value in
Andrei Paskevich's avatar
Andrei Paskevich committed
579 580 581 582 583
  let ls' = create_lsymbol ~constr id at vt in
  cl.ls_table <- Mls.add ls ls' cl.ls_table;
  ls'

let clone_decl inst cl uc d = match d.d_node with
584
  | Dtype _ | Ddata _ -> assert false (* impossible *)
585 586 587 588 589 590 591 592 593 594 595 596
  | Dparam ({ls_name = id} as ls) when Mls.mem ls inst.mi_ls ->
      let ls' = Mls.find ls inst.mi_ls in
      let mtch sb ty ty' = try ty_match sb ty' (clone_ty cl ty)
        with TypeMismatch _ -> raise (BadInstance id) in
      let sbs = match ls.ls_value,ls'.ls_value with
        | Some ty, Some ty' -> mtch Mtv.empty ty ty'
        | None, None -> Mtv.empty
        | _ -> raise (BadInstance id) in
      ignore (try List.fold_left2 mtch sbs ls.ls_args ls'.ls_args
        with Invalid_argument _ -> raise (BadInstance id));
      cl.ls_table <- Mls.add ls ls' cl.ls_table;
      uc
Andrei Paskevich's avatar
Andrei Paskevich committed
597
  | Dparam ls ->
598
      let d = create_param_decl (clone_ls cl ls) in
599
      add_pdecl ~warn:false ~vc:false uc (create_pure_decl d)
Andrei Paskevich's avatar
Andrei Paskevich committed
600
  | Dlogic ldl ->
601 602 603
      List.iter (fun (ls,_) ->
        if Mls.mem ls inst.mi_ls then raise (CannotInstantiate ls.ls_name);
        ignore (clone_ls cl ls)) ldl;
Andrei Paskevich's avatar
Andrei Paskevich committed
604
      let get_logic (_,ld) =
Andrei Paskevich's avatar
Andrei Paskevich committed
605
        Opt.get (ls_defn_of_axiom (clone_fmla cl (ls_defn_axiom ld))) in
Andrei Paskevich's avatar
Andrei Paskevich committed
606
      let d = create_logic_decl (List.map get_logic ldl) in
607
      add_pdecl ~warn:false ~vc:false uc (create_pure_decl d)
Andrei Paskevich's avatar
Andrei Paskevich committed
608
  | Dind (s, idl) ->
609 610 611
      let lls = List.map (fun (ls,_) ->
        if Mls.mem ls inst.mi_ls then raise (CannotInstantiate ls.ls_name);
        clone_ls cl ls) idl in
Andrei Paskevich's avatar
Andrei Paskevich committed
612
      let get_case (pr,f) =
613
        if Mpr.mem pr inst.mi_pk then raise (CannotInstantiate pr.pr_name);
Andrei Paskevich's avatar
Andrei Paskevich committed
614 615
        let pr' = create_prsymbol (id_clone pr.pr_name) in
        cl.pr_table <- Mpr.add pr pr' cl.pr_table;
Andrei Paskevich's avatar
Andrei Paskevich committed
616
        pr', clone_fmla cl f in
Andrei Paskevich's avatar
Andrei Paskevich committed
617 618
      let get_ind ls (_,la) = ls, List.map get_case la in
      let d = create_ind_decl s (List.map2 get_ind lls idl) in
619
      add_pdecl ~warn:false ~vc:false uc (create_pure_decl d)
Andrei Paskevich's avatar
Andrei Paskevich committed
620
  | Dprop (k,pr,f) ->
621
      let skip, k' = match k, Mpr.find_opt pr inst.mi_pk with
622 623 624 625
        | Pgoal, _ -> true, Pgoal
        | Plemma, Some Pgoal -> true, Pgoal
        | Plemma, _ -> false, Plemma
        | Paxiom, Some k -> false, k
626
        | Paxiom, None -> false, inst.mi_df in
Andrei Paskevich's avatar
Andrei Paskevich committed
627
      if skip then uc else
Andrei Paskevich's avatar
Andrei Paskevich committed
628 629
      let pr' = create_prsymbol (id_clone pr.pr_name) in
      cl.pr_table <- Mpr.add pr pr' cl.pr_table;
Andrei Paskevich's avatar
Andrei Paskevich committed
630
      let d = create_prop_decl k' pr' (clone_fmla cl f) in
631
      add_pdecl ~warn:false ~vc:false uc (create_pure_decl d)
Andrei Paskevich's avatar
Andrei Paskevich committed
632

633 634 635
let cl_save_ls cl s s' =
  cl.ls_table <- Mls.add_new (CannotInstantiate s.ls_name) s s' cl.ls_table

636 637 638
let cl_save_rs cl s s' =
  cl.rs_table <- Mrs.add s s' cl.rs_table;
  begin match s.rs_field, s'.rs_field with
639 640 641
  | Some v, Some v' -> cl.fd_table <- Mpv.add v v' cl.fd_table
  | None, _ -> () (* can instantiate a non-field with a field *)
  | _ -> assert false (* but not vice versa *)
642 643
  end;
  match s.rs_logic, s'.rs_logic with
644
  | RLls s, RLls s' -> cl_save_ls cl s s'
645 646 647 648
  | RLlemma, RLlemma -> () (* TODO: update cl.pr_table? *)
  | RLnone, RLnone -> ()
  | _ -> assert false

Mário Pereira's avatar
Mário Pereira committed
649 650 651
type smap = {
  sm_vs : vsymbol Mvs.t;
  sm_pv : pvsymbol Mvs.t;
652
  sm_xs : xsymbol Mxs.t;
Mário Pereira's avatar
Mário Pereira committed
653 654 655 656 657 658
  sm_rs : rsymbol Mrs.t;
}

let sm_of_cl cl = {
  sm_vs = Mvs.map (fun v -> v.pv_vs) cl.pv_table;
  sm_pv = cl.pv_table;
659
  sm_xs = cl.xs_table;
Mário Pereira's avatar
Mário Pereira committed
660 661 662 663 664
  sm_rs = cl.rs_table }

let sm_save_vs sm v v' = {
  sm_vs = Mvs.add v v'.pv_vs sm.sm_vs;
  sm_pv = Mvs.add v v' sm.sm_pv;
665
  sm_xs = sm.sm_xs;
Mário Pereira's avatar
Mário Pereira committed
666 667 668 669 670
  sm_rs = sm.sm_rs }

let sm_save_pv sm v v' = {
  sm_vs = Mvs.add v.pv_vs v'.pv_vs sm.sm_vs;
  sm_pv = Mvs.add v.pv_vs v' sm.sm_pv;
671
  sm_xs = sm.sm_xs;
Mário Pereira's avatar
Mário Pereira committed
672 673
  sm_rs = sm.sm_rs }

674 675 676
let sm_save_xs sm s s' =
  { sm with sm_xs = Mxs.add s s' sm.sm_xs }

Mário Pereira's avatar
Mário Pereira committed
677 678 679 680 681 682 683 684 685 686
let sm_save_rs cl sm s s' =
  let sm = { sm with sm_rs = Mrs.add s s' sm.sm_rs } in
  match s.rs_logic, s'.rs_logic with
  | RLls s, RLls s' -> cl_save_ls cl s s'; sm
  | RLpv v, RLpv v' -> sm_save_pv sm v v'
  | _ -> sm

let sm_find_pv sm v = Mvs.find_def v v.pv_vs sm.sm_pv
  (* non-instantiated global variables are not in sm *)

687 688 689 690
let sm_find_xs sm xs = Mxs.find_def xs xs sm.sm_xs

let sm_find_rs sm rs = Mrs.find_def rs rs sm.sm_rs

Mário Pereira's avatar
Mário Pereira committed
691 692 693 694 695 696
let clone_pv cl {pv_vs = vs; pv_ity = ity; pv_ghost = ghost} =
  create_pvsymbol (id_clone vs.vs_name) ~ghost (clone_ity cl ity)

let clone_invl cl sm invl =
  List.map (fun t -> clone_term cl sm.sm_vs t) invl

697
let clone_varl cl sm varl = List.map (fun (t,r) ->
Andrei Paskevich's avatar
Andrei Paskevich committed
698
  clone_term cl sm.sm_vs t, Opt.map (cl_find_ls cl) r) varl
699

700
let clone_cty cl sm ?(drop_decr=false) cty =
701
  let res = clone_ity cl cty.cty_result in
702
  let args = List.map (clone_pv cl) cty.cty_args in
703
  let sm_args = List.fold_left2 sm_save_pv sm cty.cty_args args in
704
  let add_old o n (sm, olds) = let o' = clone_pv cl o in
705
    sm_save_pv sm o o', Mpv.add o' (sm_find_pv sm_args n) olds in
706
  let sm_olds, olds = Mpv.fold add_old cty.cty_oldies (sm_args, Mpv.empty) in
707 708
  let pre = if drop_decr then List.tl cty.cty_pre else cty.cty_pre in
  let pre = clone_invl cl sm_args pre in
709
  let post = clone_invl cl sm_olds cty.cty_post in
710
  let xpost = Mxs.fold (fun xs fl q ->
711
    let xs = sm_find_xs sm xs in
712
    let fl = clone_invl cl sm_olds fl in
713
    Mxs.add xs fl q) cty.cty_xpost Mxs.empty in
714
  let add_read v s = Spv.add (sm_find_pv sm_args v) s in
715 716 717
  let reads = Spv.fold add_read (cty_reads cty) Spv.empty in
  let reads = List.fold_right add_read cty.cty_args reads in
  let reads = Spv.union reads (Mpv.domain olds) in
Mário Pereira's avatar
Mário Pereira committed
718
  let add_write reg fs m = (* add new mutable fields to functions effect *)
719
    let add_fd fd s = Spv.add (Mpv.find_def fd fd cl.fd_table) s in
720
    let reg' = clone_reg cl reg in
Mário Pereira's avatar
Mário Pereira committed
721 722
    let smf_reg' = Spv.of_list reg'.reg_its.its_mfields in
    let smf_reg = Spv.of_list reg.reg_its.its_mfields in
723 724
    let smf_ref = Spv.fold add_fd smf_reg Spv.empty in
    let smf_new = Spv.diff smf_reg' smf_ref in
Mário Pereira's avatar
Mário Pereira committed
725
    let fs = Spv.fold add_fd fs Spv.empty in
726
    Mreg.add reg' (Spv.union fs smf_new) m in
727
  let writes = Mreg.fold add_write cty.cty_effect.eff_writes Mreg.empty in
Andrei Paskevich's avatar
Andrei Paskevich committed
728
  let add_reset reg s = Sreg.add (clone_reg cl reg) s in
729 730
  let resets = Sreg.fold add_reset cty.cty_effect.eff_resets Sreg.empty in
  let eff = eff_reset (eff_write reads writes) resets in
731
  let add_raise xs eff = eff_raise eff (sm_find_xs sm xs) in
732
  let eff = Sxs.fold add_raise cty.cty_effect.eff_raises eff in
733 734
  let eff = if partial cty.cty_effect.eff_oneway then eff_partial eff else eff in
  let eff = if diverges cty.cty_effect.eff_oneway then eff_diverge eff else eff in
735 736
  let cty = create_cty ~mask:cty.cty_mask args pre post xpost olds eff res in
  cty_ghostify (cty_ghost cty) cty
737 738 739 740 741

let sm_save_args sm c c' =
  List.fold_left2 sm_save_pv sm c.cty_args c'.cty_args

let sm_save_olds sm c c' =
742 743
  let revs = Mpv.fold (fun o n m -> Mpv.add n o m) c'.cty_oldies Mpv.empty in
  let add_old o n sm = sm_save_pv sm o (Mpv.find (sm_find_pv sm n) revs) in
744 745
  Mpv.fold add_old c.cty_oldies sm

746
let clone_ppat cl sm pp mask =
747 748
  let rec conv_pat p = match p.pat_node with
    | Term.Pwild -> PPwild
749 750 751
    | Term.Pvar v -> PPvar (id_clone v.vs_name, (restore_pv v).pv_ghost)
    | Term.Pas (p,v) ->
        PPas (conv_pat p, id_clone v.vs_name, (restore_pv v).pv_ghost)
752 753 754 755
    | Term.Por (p1,p2) -> PPor (conv_pat p1, conv_pat p2)
    | Term.Papp (s,pl) ->
        PPapp (restore_rs (cl_find_ls cl s), List.map conv_pat pl) in
  let pre = conv_pat pp.pp_pat in
756
  let vm, pp' = create_prog_pattern pre (clone_ity cl pp.pp_ity) mask in
757 758 759
  let save v sm = sm_save_vs sm v (Mstr.find v.vs_name.id_string vm) in
  Svs.fold save pp.pp_pat.pat_vars sm, pp'

Andrei Paskevich's avatar
Andrei Paskevich committed
760
let rec clone_expr cl sm e = e_attr_copy e (match e.e_node with
761
  | Evar v -> e_var (sm_find_pv sm v)
762
  | Econst c -> e_const c (clone_ity cl e.e_ity)
763
  | Eexec (c,_) -> e_exec (clone_cexp cl sm c)
764 765 766 767
  | Eassign asl ->
      let conv (r,f,v) =
        e_var (sm_find_pv sm r), cl_find_rs cl f, e_var (sm_find_pv sm v) in
      e_assign (List.map conv asl)
768 769 770 771 772
  | Elet (ld, e) ->
      let sm, ld = clone_let_defn cl sm ld in
      e_let ld (clone_expr cl sm e)
  | Eif (e1, e2, e3) ->
      e_if (clone_expr cl sm e1) (clone_expr cl sm e2) (clone_expr cl sm e3)
773
  | Ematch (d, bl, xl) ->
774
      let d = clone_expr cl sm d in
775
      let conv_rbr (pp, e) =
776
        let sm, pp = clone_ppat cl sm pp d.e_mask in
777
        pp, clone_expr cl sm e in
778 779 780 781
      let conv_xbr xs (vl, e) m =
        let vl' = List.map (clone_pv cl) vl in
        let sm = List.fold_left2 sm_save_pv sm vl vl' in
        Mxs.add (sm_find_xs sm xs) (vl', clone_expr cl sm e) m in
782
      e_match d (List.map conv_rbr bl) (Mxs.fold conv_xbr xl Mxs.empty)
783 784 785
  | Ewhile (c,invl,varl,e) ->
      e_while (clone_expr cl sm c) (clone_invl cl sm invl)
              (clone_varl cl sm varl) (clone_expr cl sm e)
786 787 788 789 790 791
  | Efor (v, (f,dir,t), i, invl, e) ->
      let v' = clone_pv cl v in
      let ism = sm_save_pv sm v v' in
      let i' = if pv_equal v i then v' else clone_pv cl i in
      let ism = if pv_equal v i then ism else sm_save_pv ism i i' in
      e_for v'
792
        (e_var (sm_find_pv sm f)) dir (e_var (sm_find_pv sm t))
793
        i' (clone_invl cl ism invl) (clone_expr cl ism e)
794
  | Eraise (xs, e) ->
795 796 797 798
      e_raise (sm_find_xs sm xs) (clone_expr cl sm e) (clone_ity cl e.e_ity)
  | Eexn ({xs_name = id; xs_mask = mask; xs_ity = ity} as xs, e) ->
      let xs' = create_xsymbol (id_clone id) ~mask (clone_ity cl ity) in
      e_exn xs' (clone_expr cl (sm_save_xs sm xs xs') e)
799
  | Eassert (k, f) ->
Andrei Paskevich's avatar
Andrei Paskevich committed
800
      e_assert k (clone_term cl sm.sm_vs f)
801 802
  | Eghost e ->
      e_ghostify true (clone_expr cl sm e)
803
  | Epure t ->
Andrei Paskevich's avatar
Andrei Paskevich committed
804
      e_pure (clone_term cl sm.sm_vs t)
805
  | Eabsurd -> e_absurd (clone_ity cl e.e_ity))
806

807
and clone_cexp cl sm c = match c.c_node with
808
  | Capp (s,vl) ->
809
      let vl = List.map (fun v -> sm_find_pv sm v) vl in
Andrei Paskevich's avatar
Andrei Paskevich committed
810 811
      let al = List.map (fun v -> clone_ity cl v.pv_ity) c.c_cty.cty_args in
      let res = clone_ity cl c.c_cty.cty_result in
812
      c_app (sm_find_rs sm s) vl al res
813 814 815 816 817
  | Cpur (s,vl) ->
      let vl = List.map (fun v -> sm_find_pv sm v) vl in
      let al = List.map (fun v -> clone_ity cl v.pv_ity) c.c_cty.cty_args in
      let res = clone_ity cl c.c_cty.cty_result in
      c_pur (cl_find_ls cl s) vl al res
818 819 820 821
  | Cfun e ->
      let cty = clone_cty cl sm c.c_cty in
      let sm = sm_save_args sm c.c_cty cty in
      let sm = sm_save_olds sm c.c_cty cty in
822 823
      c_fun ~mask:cty.cty_mask cty.cty_args cty.cty_pre
        cty.cty_post cty.cty_xpost cty.cty_oldies (clone_expr cl sm e)
824
  | Cany ->
825
      c_any (clone_cty cl sm c.c_cty)
826 827 828 829

and clone_let_defn cl sm ld = match ld with
  | LDvar (v,e) ->
      let e' = clone_expr cl sm e in
830 831
      let id = id_clone v.pv_vs.vs_name in
      let ld, v' = let_var id ~ghost:v.pv_ghost e' in
832 833 834 835 836 837 838
      sm_save_pv sm v v', ld
  | LDsym (s,c) ->
      let c' = clone_cexp cl sm c in
      let ld, s' = let_sym (id_clone s.rs_name)
        ~ghost:(rs_ghost s) ~kind:(rs_kind s) c' in
      sm_save_rs cl sm s s', ld
  | LDrec rdl ->
839 840
      let conv_rs mrs {rec_rsym = {rs_name = id} as rs} =
        let cty = clone_cty cl sm ~drop_decr:true rs.rs_cty in
841
        let rs' = create_rsymbol (id_clone id) ~ghost:(rs_