You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

112 lines
3.2 KiB

(*
* Copyright (c) 2009 - 2013 Monoidics ltd.
* Copyright (c) 2013 - present Facebook, Inc.
* All rights reserved.
*
* This source code is licensed under the BSD style license found in the
* LICENSE file in the root directory of this source tree. An additional grant
* of patent rights can be found in the PATENTS file in the same directory.
*)
open! IStd
module Hashtbl = Caml.Hashtbl
open Javalib_pack
open Sawja_pack
module NodeTbl = Procdesc.NodeHash
type jump_kind = Next | Jump of int | Exit
(** Translation data *)
type icfg = {tenv: Tenv.t; cfg: Cfg.t}
type t =
{ icfg: icfg
; procdesc: Procdesc.t
; impl: JBir.t
; mutable var_map: (Pvar.t * Typ.t * Typ.t) JBir.VarMap.t
; if_jumps: int NodeTbl.t
; goto_jumps: (int, jump_kind) Hashtbl.t
; cn: JBasics.class_name
; source_file: SourceFile.t
; program: JClasspath.program }
let create_context icfg procdesc impl cn source_file program =
{ icfg
; procdesc
; impl
; var_map= JBir.VarMap.empty
; if_jumps= NodeTbl.create 10
; goto_jumps= Hashtbl.create 10
; cn
; source_file
; program }
let get_tenv context = context.icfg.tenv
let set_var_map context var_map = context.var_map <- var_map
let get_or_set_pvar_type context var typ =
let var_map = context.var_map in
try
let pvar, otyp, _ = JBir.VarMap.find var var_map in
let tenv = get_tenv context in
if Prover.Subtyping_check.check_subtype tenv typ otyp
|| Prover.Subtyping_check.check_subtype tenv otyp typ
then set_var_map context (JBir.VarMap.add var (pvar, otyp, typ) var_map)
else set_var_map context (JBir.VarMap.add var (pvar, typ, typ) var_map) ;
(pvar, typ)
with Not_found ->
let procname = Procdesc.get_proc_name context.procdesc in
let varname = Mangled.from_string (JBir.var_name_g var) in
let pvar = Pvar.mk varname procname in
set_var_map context (JBir.VarMap.add var (pvar, typ, typ) var_map) ;
(pvar, typ)
let set_pvar context var typ = fst (get_or_set_pvar_type context var typ)
let reset_pvar_type context =
let var_map = context.var_map in
let aux var item =
match item with pvar, otyp, _ ->
set_var_map context (JBir.VarMap.add var (pvar, otyp, otyp) var_map)
in
JBir.VarMap.iter aux var_map
let get_var_type context var =
try
let _, _, otyp = JBir.VarMap.find var context.var_map in
Some otyp
with Not_found -> None
let get_if_jumps context = context.if_jumps
let get_goto_jumps context = context.goto_jumps
let add_if_jump context node pc = NodeTbl.add (get_if_jumps context) node pc
let get_if_jump context node =
try Some (NodeTbl.find (get_if_jumps context) node) with Not_found -> None
let add_goto_jump context pc jump = Hashtbl.add (get_goto_jumps context) pc jump
let get_goto_jump context pc =
try Hashtbl.find (get_goto_jumps context) pc with Not_found -> Next
let is_goto_jump context pc =
try match Hashtbl.find (get_goto_jumps context) pc with Jump _ -> true | _ -> false
with Not_found -> false
let exn_node_table = Typ.Procname.Hash.create 100
let reset_exn_node_table () = Typ.Procname.Hash.clear exn_node_table
let add_exn_node procname (exn_node: Procdesc.Node.t) =
Typ.Procname.Hash.add exn_node_table procname exn_node