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.
119 lines
2.6 KiB
119 lines
2.6 KiB
(*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*)
|
|
|
|
open! IStd
|
|
module F = Format
|
|
|
|
let all_formals_untainted pdesc =
|
|
let make_untainted (name, typ) = (name, typ, None) in
|
|
List.map ~f:make_untainted (Procdesc.get_formals pdesc)
|
|
|
|
|
|
module type Kind = sig
|
|
include TraceElem.Kind
|
|
|
|
val get :
|
|
caller_pname:Typ.Procname.t
|
|
-> Typ.Procname.t
|
|
-> HilExp.t list
|
|
-> Tenv.t
|
|
-> (t * int option) list
|
|
|
|
val get_tainted_formals : Procdesc.t -> Tenv.t -> (Mangled.t * Typ.t * t option) list
|
|
end
|
|
|
|
module type S = sig
|
|
include TraceElem.S
|
|
|
|
type spec = {source: t; index: int option}
|
|
|
|
val get : caller_pname:Typ.Procname.t -> CallSite.t -> HilExp.t list -> Tenv.t -> spec list
|
|
|
|
val get_tainted_formals : Procdesc.t -> Tenv.t -> (Mangled.t * Typ.t * t option) list
|
|
end
|
|
|
|
module Make (Kind : Kind) = struct
|
|
module Kind = Kind
|
|
|
|
type t = {kind: Kind.t; site: CallSite.t} [@@deriving compare]
|
|
|
|
type spec = {source: t; index: int option}
|
|
|
|
let call_site t = t.site
|
|
|
|
let kind t = t.kind
|
|
|
|
let make ?indexes:_ kind site = {site; kind}
|
|
|
|
let get ~caller_pname site actuals tenv =
|
|
Kind.get ~caller_pname (CallSite.pname site) actuals tenv
|
|
|> List.rev_map ~f:(fun (kind, index) ->
|
|
let source = make kind site in
|
|
{source; index} )
|
|
|
|
|
|
let get_tainted_formals pdesc tenv =
|
|
let site = CallSite.make (Procdesc.get_proc_name pdesc) (Procdesc.get_loc pdesc) in
|
|
List.map
|
|
~f:(fun (name, typ, kind_opt) ->
|
|
(name, typ, Option.map kind_opt ~f:(fun kind -> make kind site)) )
|
|
(Kind.get_tainted_formals pdesc tenv)
|
|
|
|
|
|
let pp fmt s = F.fprintf fmt "%a(%a)" Kind.pp s.kind CallSite.pp s.site
|
|
|
|
let with_callsite t callee_site = {t with site= callee_site}
|
|
|
|
module Set = PrettyPrintable.MakePPSet (struct
|
|
type nonrec t = t
|
|
|
|
let compare = compare
|
|
|
|
let pp = pp
|
|
end)
|
|
end
|
|
|
|
module Dummy = struct
|
|
type t = unit [@@deriving compare]
|
|
|
|
type spec = {source: t; index: int option}
|
|
|
|
let call_site _ = CallSite.dummy
|
|
|
|
let kind t = t
|
|
|
|
let make ?indexes:_ kind _ = kind
|
|
|
|
let pp _ () = ()
|
|
|
|
let get ~caller_pname:_ _ _ _ = []
|
|
|
|
let get_tainted_formals pdesc _ =
|
|
List.map ~f:(fun (name, typ) -> (name, typ, None)) (Procdesc.get_formals pdesc)
|
|
|
|
|
|
module Kind = struct
|
|
type nonrec t = t
|
|
|
|
let compare = compare
|
|
|
|
let matches ~caller ~callee = Int.equal 0 (compare caller callee)
|
|
|
|
let pp = pp
|
|
end
|
|
|
|
module Set = PrettyPrintable.MakePPSet (struct
|
|
type nonrec t = t
|
|
|
|
let compare = compare
|
|
|
|
let pp = pp
|
|
end)
|
|
|
|
let with_callsite t _ = t
|
|
end
|