Release coccinelle-0.2.3rc2
[bpt/coccinelle.git] / parsing_cocci / unitary_ast0.ml
CommitLineData
5636bb2c 1(*
90aeb998
C
2 * Copyright 2010, INRIA, University of Copenhagen
3 * Julia Lawall, Rene Rydhof Hansen, Gilles Muller, Nicolas Palix
4 * Copyright 2005-2009, Ecole des Mines de Nantes, University of Copenhagen
9f8e26f4
C
5 * Yoann Padioleau, Julia Lawall, Rene Rydhof Hansen, Henrik Stuart, Gilles Muller, Nicolas Palix
6 * This file is part of Coccinelle.
7 *
8 * Coccinelle is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, according to version 2 of the License.
11 *
12 * Coccinelle is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with Coccinelle. If not, see <http://www.gnu.org/licenses/>.
19 *
20 * The authors reserve the right to distribute this or future versions of
21 * Coccinelle under other licenses.
22 *)
23
24
34e49164
C
25(* find unitary metavariables *)
26module Ast0 = Ast0_cocci
27module Ast = Ast_cocci
28module V0 = Visitor_ast0
b1b2de81 29module VT0 = Visitor_ast0_types
34e49164
C
30
31let set_minus s minus = List.filter (function n -> not (List.mem n minus)) s
32
33let rec nub = function
34 [] -> []
35 | (x::xs) when (List.mem x xs) -> nub xs
36 | (x::xs) -> x::(nub xs)
37
38(* ----------------------------------------------------------------------- *)
39(* Find the variables that occur free and occur free in a unitary way *)
40
41(* take everything *)
42let minus_checker name = let id = Ast0.unwrap_mcode name in [id]
43
44(* take only what is in the plus code *)
708f4980 45let plus_checker (nm,_,_,mc,_,_) =
951c7801 46 match mc with Ast0.PLUS _ -> [nm] | _ -> []
faf9a90c 47
34e49164
C
48let get_free checker t =
49 let bind x y = x @ y in
50 let option_default = [] in
faf9a90c 51
34e49164
C
52 (* considers a single list *)
53 let collect_unitary_nonunitary free_usage =
54 let free_usage = List.sort compare free_usage in
55 let rec loop1 todrop = function
56 [] -> []
57 | (x::xs) as all -> if x = todrop then loop1 todrop xs else all in
58 let rec loop2 = function
59 [] -> ([],[])
60 | [x] -> ([x],[])
61 | x::y::xs ->
62 if x = y
63 then
64 let (unitary,non_unitary) = loop2(loop1 x xs) in
65 (unitary,x::non_unitary)
66 else
67 let (unitary,non_unitary) = loop2 (y::xs) in
68 (x::unitary,non_unitary) in
69 loop2 free_usage in
faf9a90c 70
34e49164
C
71 (* considers a list of lists *)
72 let detect_unitary_frees l =
73 let (unitary,nonunitary) =
74 List.split (List.map collect_unitary_nonunitary l) in
75 let unitary = nub (List.concat unitary) in
76 let nonunitary = nub (List.concat nonunitary) in
77 let unitary =
78 List.filter (function x -> not (List.mem x nonunitary)) unitary in
79 unitary@nonunitary@nonunitary in
80
1be43e12 81 let whencode afn bfn expression = function
34e49164
C
82 Ast0.WhenNot(a) -> afn a
83 | Ast0.WhenAlways(b) -> bfn b
1be43e12
C
84 | Ast0.WhenModifier(_) -> option_default
85 | Ast0.WhenNotTrue(a) -> expression a
86 | Ast0.WhenNotFalse(a) -> expression a in
faf9a90c 87
34e49164
C
88 let ident r k i =
89 match Ast0.unwrap i with
90 Ast0.MetaId(name,_,_) | Ast0.MetaFunc(name,_,_)
91 | Ast0.MetaLocalFunc(name,_,_) -> checker name
92 | _ -> k i in
faf9a90c 93
34e49164
C
94 let expression r k e =
95 match Ast0.unwrap e with
96 Ast0.MetaErr(name,_,_) | Ast0.MetaExpr(name,_,_,_,_)
97 | Ast0.MetaExprList(name,_,_) -> checker name
98 | Ast0.DisjExpr(starter,expr_list,mids,ender) ->
b1b2de81 99 detect_unitary_frees(List.map r.VT0.combiner_rec_expression expr_list)
34e49164 100 | _ -> k e in
faf9a90c 101
34e49164
C
102 let typeC r k t =
103 match Ast0.unwrap t with
104 Ast0.MetaType(name,_) -> checker name
105 | Ast0.DisjType(starter,types,mids,ender) ->
b1b2de81 106 detect_unitary_frees(List.map r.VT0.combiner_rec_typeC types)
34e49164 107 | _ -> k t in
faf9a90c 108
34e49164
C
109 let parameter r k p =
110 match Ast0.unwrap p with
111 Ast0.MetaParam(name,_) | Ast0.MetaParamList(name,_,_) -> checker name
112 | _ -> k p in
faf9a90c 113
34e49164
C
114 let declaration r k d =
115 match Ast0.unwrap d with
116 Ast0.DisjDecl(starter,decls,mids,ender) ->
b1b2de81 117 detect_unitary_frees(List.map r.VT0.combiner_rec_declaration decls)
34e49164
C
118 | _ -> k d in
119
fc1ad971
C
120 let case_line r k c =
121 match Ast0.unwrap c with
122 Ast0.DisjCase(starter,case_lines,mids,ender) ->
123 detect_unitary_frees(List.map r.VT0.combiner_rec_case_line case_lines)
124 | _ -> k c in
125
34e49164
C
126 let statement r k s =
127 match Ast0.unwrap s with
128 Ast0.MetaStmt(name,_) | Ast0.MetaStmtList(name,_) -> checker name
129 | Ast0.Disj(starter,stmt_list,mids,ender) ->
b1b2de81
C
130 detect_unitary_frees
131 (List.map r.VT0.combiner_rec_statement_dots stmt_list)
34e49164 132 | Ast0.Nest(starter,stmt_dots,ender,whn,multi) ->
b1b2de81 133 bind (r.VT0.combiner_rec_statement_dots stmt_dots)
faf9a90c 134 (detect_unitary_frees
34e49164 135 (List.map
b1b2de81 136 (whencode
ae4735db 137 r.VT0.combiner_rec_statement_dots
b1b2de81
C
138 r.VT0.combiner_rec_statement
139 r.VT0.combiner_rec_expression)
34e49164
C
140 whn))
141 | Ast0.Dots(d,whn) | Ast0.Circles(d,whn) | Ast0.Stars(d,whn) ->
142 detect_unitary_frees
143 (List.map
b1b2de81
C
144 (whencode
145 r.VT0.combiner_rec_statement_dots r.VT0.combiner_rec_statement
146 r.VT0.combiner_rec_expression)
34e49164
C
147 whn)
148 | _ -> k s in
faf9a90c
C
149
150 let res = V0.combiner bind option_default
b1b2de81
C
151 {V0.combiner_functions with
152 VT0.combiner_identfn = ident;
153 VT0.combiner_exprfn = expression;
154 VT0.combiner_tyfn = typeC;
155 VT0.combiner_paramfn = parameter;
156 VT0.combiner_declfn = declaration;
fc1ad971
C
157 VT0.combiner_stmtfn = statement;
158 VT0.combiner_casefn = case_line} in
faf9a90c 159
34e49164 160 collect_unitary_nonunitary
b1b2de81 161 (List.concat (List.map res.VT0.combiner_rec_top_level t))
faf9a90c 162
34e49164
C
163(* ----------------------------------------------------------------------- *)
164(* update the variables that are unitary *)
faf9a90c 165
34e49164 166let update_unitary unitary =
34e49164
C
167 let is_unitary name =
168 match (List.mem (Ast0.unwrap_mcode name) unitary,
485bce71
C
169 !Flag.sgrep_mode2, Ast0.get_mcode_mcodekind name) with
170 (true,true,_) | (true,_,Ast0.CONTEXT(_)) -> Ast0.PureContext
171 | (true,_,_) -> Ast0.Pure
172 | (false,true,_) | (false,_,Ast0.CONTEXT(_)) -> Ast0.Context
173 | (false,_,_) -> Ast0.Impure in
34e49164
C
174
175 let ident r k i =
176 match Ast0.unwrap i with
177 Ast0.MetaId(name,constraints,_) ->
178 Ast0.rewrap i (Ast0.MetaId(name,constraints,is_unitary name))
179 | Ast0.MetaFunc(name,constraints,_) ->
180 Ast0.rewrap i (Ast0.MetaFunc(name,constraints,is_unitary name))
181 | Ast0.MetaLocalFunc(name,constraints,_) ->
182 Ast0.rewrap i (Ast0.MetaLocalFunc(name,constraints,is_unitary name))
183 | _ -> k i in
184
185 let expression r k e =
186 match Ast0.unwrap e with
187 Ast0.MetaErr(name,constraints,_) ->
188 Ast0.rewrap e (Ast0.MetaErr(name,constraints,is_unitary name))
189 | Ast0.MetaExpr(name,constraints,ty,form,_) ->
190 Ast0.rewrap e (Ast0.MetaExpr(name,constraints,ty,form,is_unitary name))
191 | Ast0.MetaExprList(name,lenname,_) ->
192 Ast0.rewrap e (Ast0.MetaExprList(name,lenname,is_unitary name))
193 | _ -> k e in
faf9a90c 194
34e49164
C
195 let typeC r k t =
196 match Ast0.unwrap t with
197 Ast0.MetaType(name,_) ->
198 Ast0.rewrap t (Ast0.MetaType(name,is_unitary name))
199 | _ -> k t in
faf9a90c 200
34e49164
C
201 let parameter r k p =
202 match Ast0.unwrap p with
203 Ast0.MetaParam(name,_) ->
204 Ast0.rewrap p (Ast0.MetaParam(name,is_unitary name))
205 | Ast0.MetaParamList(name,lenname,_) ->
206 Ast0.rewrap p (Ast0.MetaParamList(name,lenname,is_unitary name))
207 | _ -> k p in
faf9a90c 208
34e49164
C
209 let statement r k s =
210 match Ast0.unwrap s with
211 Ast0.MetaStmt(name,_) ->
212 Ast0.rewrap s (Ast0.MetaStmt(name,is_unitary name))
213 | Ast0.MetaStmtList(name,_) ->
214 Ast0.rewrap s (Ast0.MetaStmtList(name,is_unitary name))
215 | _ -> k s in
faf9a90c 216
34e49164 217 let res = V0.rebuilder
b1b2de81
C
218 {V0.rebuilder_functions with
219 VT0.rebuilder_identfn = ident;
220 VT0.rebuilder_exprfn = expression;
221 VT0.rebuilder_tyfn = typeC;
222 VT0.rebuilder_paramfn = parameter;
223 VT0.rebuilder_stmtfn = statement} in
34e49164 224
b1b2de81 225 List.map res.VT0.rebuilder_rec_top_level
34e49164
C
226
227(* ----------------------------------------------------------------------- *)
228
229let rec split3 = function
230 [] -> ([],[],[])
231 | (a,b,c)::xs -> let (l1,l2,l3) = split3 xs in (a::l1,b::l2,c::l3)
232
233let rec combine3 = function
234 ([],[],[]) -> []
235 | (a::l1,b::l2,c::l3) -> (a,b,c) :: combine3 (l1,l2,l3)
236 | _ -> failwith "not possible"
237
238(* ----------------------------------------------------------------------- *)
239(* process all rules *)
240
241let do_unitary rules =
242 let rec loop = function
243 [] -> ([],[])
244 | (r::rules) ->
245 match r with
b1b2de81 246 Ast0.ScriptRule (_,_,_,_)
c3e37e97 247 | Ast0.InitialScriptRule (_,_,_) | Ast0.FinalScriptRule (_,_,_) ->
34e49164
C
248 let (x,rules) = loop rules in
249 (x, r::rules)
faf9a90c 250 | Ast0.CocciRule ((minus,metavars,chosen_isos),((plus,_) as plusz),rt) ->
34e49164
C
251 let mm1 = List.map Ast.get_meta_name metavars in
252 let (used_after, rest) = loop rules in
253 let (m_unitary, m_nonunitary) = get_free minus_checker minus in
254 let (p_unitary, p_nonunitary) = get_free plus_checker plus in
faf9a90c 255 let p_free =
34e49164
C
256 if !Flag.sgrep_mode2 then []
257 else p_unitary @ p_nonunitary in
258 let (in_p, m_unitary) =
259 List.partition (function x -> List.mem x p_free) m_unitary in
260 let m_nonunitary = in_p @ m_nonunitary in
261 let (m_unitary, not_local) =
262 List.partition (function x -> List.mem x mm1) m_unitary in
263 let m_unitary =
264 List.filter (function x -> not (List.mem x used_after))
265 m_unitary in
266 let rebuilt = update_unitary m_unitary minus in
267 (set_minus (m_nonunitary @ used_after) mm1,
268 (Ast0.CocciRule
faf9a90c 269 ((rebuilt, metavars, chosen_isos),plusz,rt))::rest) in
34e49164
C
270 let (_,rules) = loop rules in
271 rules
272
273(*
274let do_unitary minus plus =
275 let (minus,metavars,chosen_isos) = split3 minus in
276 let (plus,_) = List.split plus in
277 let rec loop = function
278 ([],[],[]) -> ([],[])
279 | (mm1::metavars,m1::minus,p1::plus) ->
280 let mm1 = List.map Ast.get_meta_name mm1 in
281 let (used_after,rest) = loop (metavars,minus,plus) in
282 let (m_unitary,m_nonunitary) = get_free minus_checker m1 in
283 let (p_unitary,p_nonunitary) = get_free plus_checker p1 in
284 let p_free =
285 if !Flag.sgrep_mode2
286 then []
287 else p_unitary @ p_nonunitary in
288 let (in_p,m_unitary) =
289 List.partition (function x -> List.mem x p_free) m_unitary in
290 let m_nonunitary = in_p@m_nonunitary in
291 let (m_unitary,not_local) =
292 List.partition (function x -> List.mem x mm1) m_unitary in
293 let m_unitary =
294 List.filter (function x -> not(List.mem x used_after)) m_unitary in
295 let rebuilt = update_unitary m_unitary m1 in
296 (set_minus (m_nonunitary @ used_after) mm1,
297 rebuilt::rest)
298 | _ -> failwith "not possible" in
299 let (_,rules) = loop (metavars,minus,plus) in
300 combine3 (rules,metavars,chosen_isos)
301*)