package mopsa

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

Source file stmt.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
(****************************************************************************)
(*                                                                          *)
(* This file is part of MOPSA, a Modular Open Platform for Static Analysis. *)
(*                                                                          *)
(* Copyright (C) 2017-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/>.    *)
(*                                                                          *)
(****************************************************************************)

(** Extensible type of statements *)

open Mopsa_utils
open Program
open Expr
open Format


type stmt_kind = ..

type stmt = {
  skind : stmt_kind;
  srange : Location.range;
}

type stmt_kind +=
  | S_program of program * string list option
  | S_assign of expr (** lhs *) * expr (** rhs *)
  | S_assume of expr (** condition *)
  | S_add of expr
  | S_remove of expr
  | S_invalidate of expr
  | S_rename of expr (** old *) * expr (** new *)
  | S_forget of expr
  | S_project of expr list
  | S_expand of expr * expr list
  | S_fold of expr * expr list
  | S_block of stmt list * Var.var list
  | S_breakpoint of string

type block = stmt list

let skind (stmt: stmt) = stmt.skind
let srange (stmt: stmt) = stmt.srange

let stmt_compare_chain = TypeExt.mk_compare_chain (fun s1 s2 ->
    match skind s1, skind s2 with
    | S_assign(x1, e1), S_assign(x2, e2) ->
      Compare.compose [
        (fun () -> compare_expr x1 x2);
        (fun () -> compare_expr e1 e2);
      ]

    | S_assume(e1), S_assume(e2) -> compare_expr e1 e2

    | S_rename(e1, e1'), S_rename(e2, e2') ->
      Compare.compose [
        (fun () -> compare_expr e1 e2);
        (fun () -> compare_expr e1' e2');
      ]

    | S_remove(e1), S_remove(e2) -> compare_expr e1 e2

    | S_invalidate(e1), S_invalidate(e2) -> compare_expr e1 e2

    | S_add(e1), S_add(e2) -> compare_expr e1 e2

    | S_project(el1), S_project(el2) -> Compare.list compare_expr el1 el2

    | S_expand(e, el), S_expand(e', el') ->
      Compare.compose [
        (fun () -> compare_expr e e');
        (fun () -> Compare.list compare_expr el el')
      ]

    | S_fold(e, el), S_fold(e', el') ->
      Compare.compose [
        (fun () -> compare_expr e e');
        (fun () -> Compare.list compare_expr el el')
      ]

    | S_breakpoint b1, S_breakpoint b2 -> String.compare b1 b2

    | _ -> Stdlib.compare s1 s2
  )

let compare_stmt s1 s2 =
  TypeExt.compare stmt_compare_chain s1 s2

let stmt_pp_chain = TypeExt.mk_print_chain (fun fmt stmt ->
    match skind stmt with
    | S_program (prog,args) -> pp_program fmt prog

    | S_remove(e) -> fprintf fmt "remove(%a)" pp_expr e

    | S_invalidate(e) -> fprintf fmt "invalidate(%a)" pp_expr e

    | S_add(e) -> fprintf fmt "add(%a)" pp_expr e

    | S_forget(e) -> fprintf fmt "forget(%a)" pp_expr e

    | S_project(el) ->
      fprintf fmt "project(%a)"
        (pp_print_list ~pp_sep:(fun fmt () -> pp_print_string fmt ", ") pp_expr) el

    | S_rename(e, e') -> fprintf fmt "rename(%a, %a)" pp_expr e pp_expr e'

    | S_assign(x, e) -> fprintf fmt "%a = %a;" pp_expr x pp_expr e

    | S_assume(e) -> fprintf fmt "assume(%a)" pp_expr e

    | S_expand(e, el) ->
      fprintf fmt "expand(%a,{%a})"
        pp_expr e
        (pp_print_list
           ~pp_sep:(fun fmt () -> fprintf fmt ", ")
           pp_expr) el

    | S_fold(e, el) ->
      fprintf fmt "fold(%a,{%a})"
        pp_expr e
        (pp_print_list
           ~pp_sep:(fun fmt () -> fprintf fmt ", ")
           pp_expr) el

    | S_breakpoint b ->
      fprintf fmt "breakpoint(%s)" b

    | _ -> failwith "Pp: Unknown statement"
  )


let pp_stmt fmt stmt =
  Var.force_print_uniq_with_uid false (fun () -> TypeExt.print stmt_pp_chain fmt stmt)

let pp_stmt_with_range fmt stmt =
  Var.force_print_uniq_with_uid false (fun () -> Format.fprintf fmt "%a@%a" (TypeExt.print stmt_pp_chain) stmt Location.pp_range stmt.srange)

let pp_block fmt (block:block) =
  fprintf fmt "@[<v>";
  pp_print_list ~pp_sep:(fun fmt () -> fprintf fmt "@,") pp_stmt fmt block;
  fprintf fmt "@]"

let register_stmt info =
  TypeExt.register info stmt_compare_chain stmt_pp_chain

let register_stmt_compare cmp = TypeExt.register_compare cmp stmt_compare_chain

let register_stmt_pp pp = TypeExt.register_print pp stmt_pp_chain

let () = register_stmt {
    print = (fun next fmt stmt ->
        match skind stmt with
        | S_block([],_) -> fprintf fmt "pass"
        | S_block([s],_) -> pp_stmt fmt s
        | S_block(l,_) ->
          fprintf fmt "@[<v>";
          pp_print_list
            ~pp_sep:(fun fmt () -> fprintf fmt "@,")
            pp_stmt
            fmt l
          ;
          fprintf fmt "@]"
        | _ -> next fmt stmt);
    compare = (fun next s1 s2 ->
        match skind s1, skind s2 with
        | S_block(sl1,vl1), S_block(sl2,vl2) ->
          Compare.pair (Compare.list compare_stmt) (Compare.list Var.compare_var) (sl1,vl1) (sl2,vl2)
        | _ -> next s1 s2
      )
  }

let mk_stmt skind srange = {skind; srange}

let mk_rename v v' =
  mk_stmt (S_rename (v, v'))

let mk_assign v e =
  mk_stmt (S_assign (v, e))

let mk_assume e =
  mk_stmt (S_assume e)

let mk_remove v = mk_stmt (S_remove v)

let mk_remove_var v range =
  mk_remove (mk_var v range) range

let mk_invalidate v = mk_stmt (S_invalidate v)

let mk_invalidate_var v range =
  mk_invalidate (mk_var v range) range

let mk_add v = mk_stmt (S_add v)

let mk_add_var v range =
  mk_add (mk_var v range) range

let mk_rename e e' range =
  mk_stmt (S_rename (e, e')) range

let mk_rename_var v v' range =
  mk_rename (mk_var v range) (mk_var v' range) range

let mk_project vars = mk_stmt (S_project vars)

let mk_project_vars vars range =
  mk_project (List.map (fun v -> mk_var v range) vars) range

let mk_forget e = mk_stmt (S_forget e)

let mk_forget_var v range = mk_forget (mk_var v range) range

let mk_expand v vl range =
  mk_stmt (S_expand(v, vl)) range

let mk_expand_var v vl range =
  mk_expand
    (mk_var v range)
    (List.map (fun v' -> mk_var v' range) vl)
    range

let mk_fold v vl range =
  mk_stmt (S_fold(v, vl)) range

let mk_fold_var v vl range =
  mk_fold
    (mk_var v range)
    (List.map (fun v' -> mk_var v' range) vl)
    range

let mk_breakpoint b range =
  mk_stmt (S_breakpoint b) range

module StmtSet = SetExt.Make(struct
    type t = stmt
    let compare = compare_stmt
  end)

module StmtMap = MapExt.Make(struct
    type t = stmt
    let compare = compare_stmt
  end)
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