package mopsa

  1. Overview
  2. Docs
package mopsa
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Source file visitor.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
(****************************************************************************)
(*                                                                          *)
(* This file is part of MOPSA, a Modular Open Platform for Static Analysis. *)
(*                                                                          *)
(* Copyright (C) 2018-2019 The MOPSA Project.                               *)
(*                                                                          *)
(* This program is free software: you can redistribute it and/or modify     *)
(* it under the terms of the GNU Lesser General Public License as published *)
(* by the Free Software Foundation, either version 3 of the License, or     *)
(* (at your option) any later version.                                      *)
(*                                                                          *)
(* This program is distributed in the hope that it will be useful,          *)
(* but WITHOUT ANY WARRANTY; without even the implied warranty of           *)
(* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            *)
(* GNU Lesser General Public License for more details.                      *)
(*                                                                          *)
(* You should have received a copy of the GNU Lesser General Public License *)
(* along with this program.  If not, see <http://www.gnu.org/licenses/>.    *)
(*                                                                          *)
(****************************************************************************)

(** Visitors for CFG. *)

open Mopsa
open Universal.Ast
open Ast
   

let () =
    register_stmt_visitor (fun default stmt ->
      match skind stmt with
      | S_skip ->
         leaf stmt

      | S_test e ->
         { exprs = [e]; stmts = []; },
         (function { exprs = [e] } -> { stmt with skind = S_test e }
         | _ -> failwith "invalid S_test visitor")

      | S_cfg c ->
         let cfg = c.cfg_graph in
         (* get edges *)
         let l = CFG.edge_list cfg in

         (* construct map from edge id to index in l *)
         let idxmap = Hashtbl.create 16 in
         let rec mkmap i l = match l with
           | [] -> ()
           | a::b -> Hashtbl.add idxmap (CFG.edge_id a) i; mkmap (i+1) b
         in
         mkmap 0 l;
         
         { exprs = []; stmts = List.map CFG.edge_data l; },
         (function { stmts } ->
            (* copy *)
            let cfg = CFG.clone cfg in
            (* update statements *)
            CFG.iter_edges
              (fun id e ->
                let d = List.nth stmts (Hashtbl.find idxmap id) in
                CFG.edge_set_data e d
              )
              cfg;
            mk_cfg { c with cfg_graph = cfg; } (srange stmt)
         )
         
      | _ -> default stmt
      )
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct