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
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
|
(*
* This file is part of Coccinelle, licensed under the terms of the GPL v2.
* See copyright.txt in the Coccinelle source code for more information.
* The Coccinelle source code can be obtained at http://coccinelle.lip6.fr
*)
(* Find a directive or comment at the end of a statement. Things with aft
given None, because they can accommodate their own directives or comments *)
module Ast0 = Ast0_cocci
module Ast = Ast_cocci
module V0 = Visitor_ast0
module VT0 = Visitor_ast0_types
let call_right processor data s cont =
match processor data with
None -> None
| Some(pragmas,data) -> Some (pragmas,Ast0.rewrap s (cont data))
let left_mcode (a,b,info,mcodekind,d,e) =
match (info.Ast0.strings_before,mcodekind) with
([],_) | (_,Ast0.PLUS _) -> None
| (l,_) -> Some(l,(a,b,{info with Ast0.strings_before = []},mcodekind,d,e))
let right_mcode (a,b,info,mcodekind,d,e) =
match (info.Ast0.strings_after,mcodekind) with
([],_) | (_,Ast0.PLUS _) -> None
| (l,_) -> Some(l,(a,b,{info with Ast0.strings_after = []},mcodekind,d,e))
let update_before pragmas (info,x,adj) =
({info with Ast0.strings_before = pragmas @ info.Ast0.strings_before},
Ast0.PLUS Ast.ONE,adj)(*not sure what the arg should be... one seems safe*)
let update_before2 pragmas (info,x) =
({info with Ast0.strings_before = pragmas @ info.Ast0.strings_before},
Ast0.PLUS Ast.ONE)(*not sure what the arg should be... one seems safe*)
let update_after pragmas (info,x) =
({info with Ast0.strings_after = info.Ast0.strings_after @ pragmas},
Ast0.PLUS Ast.ONE) (*not sure what the arg should be... one seems safe*)
let rec right_decl d =
match Ast0.unwrap d with
Ast0.MetaDecl(name,cstr,pure) ->
call_right right_mcode name d
(function name -> Ast0.MetaDecl(name,cstr,pure))
| Ast0.AsDecl(decl,asdecl) -> failwith "not possible"
| Ast0.Init(stg,ty,id,attr,eq,ini,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.Init(stg,ty,id,attr,eq,ini,sem))
| Ast0.UnInit(stg,ty,id,attr,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.UnInit(stg,ty,id,attr,sem))
| Ast0.FunProto(fninfo,id,lp1,params,va,rp1,sem) ->
call_right right_mcode sem d
(function ty -> Ast0.FunProto(fninfo,id,lp1,params,va,rp1,sem))
| Ast0.MacroDecl(stg,name,lp,args,rp,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.MacroDecl(stg,name,lp,args,rp,sem))
| Ast0.MacroDeclInit(stg,name,lp,args,rp,eq,ini,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.MacroDeclInit(stg,name,lp,args,rp,eq,ini,sem))
| Ast0.TyDecl(ty,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.TyDecl(ty,sem))
| Ast0.Typedef(stg,ty,id,sem) ->
call_right right_mcode sem d
(function sem -> Ast0.Typedef(stg,ty,id,sem))
| Ast0.DisjDecl(starter,decls,mids,ender)
| Ast0.ConjDecl(starter,decls,mids,ender) -> None
| Ast0.OptDecl(decl) ->
call_right right_decl decl d (function decl -> Ast0.OptDecl(decl))
let rec right_statement s =
match Ast0.unwrap s with
Ast0.FunDecl(bef,fi,name,lp,params,va,rp,lbrace,body,rbrace,aft) -> None
| Ast0.Decl(bef,decl) ->
call_right right_decl decl s
(function decl -> Ast0.Decl(bef,decl))
| Ast0.Seq(lbrace,body,rbrace) ->
call_right right_mcode rbrace s
(function rbrace -> Ast0.Seq(lbrace,body,rbrace))
| Ast0.ExprStatement(exp,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.ExprStatement(exp,sem))
| Ast0.IfThen(iff,lp,exp,rp,branch1,aft) -> None
| Ast0.IfThenElse(iff,lp,exp,rp,branch1,els,branch2,aft) -> None
| Ast0.While(whl,lp,exp,rp,body,aft) -> None
| Ast0.Do(d,body,whl,lp,exp,rp,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Do(d,body,whl,lp,exp,rp,sem))
| Ast0.For(fr,lp,first,e2,sem2,e3,rp,body,aft) -> None
| Ast0.Iterator(nm,lp,args,rp,body,aft) -> None
| Ast0.Switch(switch,lp,exp,rp,lb,decls,cases,rb) ->
call_right right_mcode rb s
(function rb -> Ast0.Switch(switch,lp,exp,rp,lb,decls,cases,rb))
| Ast0.Break(br,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Break(br,sem))
| Ast0.Continue(cont,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Continue(cont,sem))
| Ast0.Label(l,dd) ->
call_right right_mcode dd s
(function dd -> Ast0.Label(l,dd))
| Ast0.Goto(goto,l,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Goto(goto,l,sem))
| Ast0.Return(ret,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Return(ret,sem))
| Ast0.ReturnExpr(ret,exp,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.ReturnExpr(ret,exp,sem))
| Ast0.Exec(exec,lang,exp,sem) ->
call_right right_mcode sem s
(function sem -> Ast0.Exec(exec,lang,exp,sem))
| Ast0.MetaStmt(name,cstr,pure) ->
call_right right_mcode name s
(function name -> Ast0.MetaStmt(name,cstr,pure))
| Ast0.MetaStmtList(name,lenname,cstr,pure) ->
call_right right_mcode name s
(function name -> Ast0.MetaStmtList(name,lenname,cstr,pure))
| Ast0.AsStmt(stm,asstm) -> failwith "not possible"
| Ast0.Disj(starter,statement_dots_list,mids,ender)
| Ast0.Conj(starter,statement_dots_list,mids,ender) -> None
| Ast0.Nest(starter,stmt_dots,ender,whn,multi) -> None
(* the following are None, because they can't be adjacent to an aft node *)
| Ast0.Exp(exp) -> None
| Ast0.TopExp(exp) -> None
| Ast0.Ty(ty) -> None
| Ast0.TopId(id) -> None
| Ast0.TopInit(init) -> None
| Ast0.Dots(d,whn) -> None
| Ast0.Include(inc,name) ->
call_right right_mcode name s
(function name -> Ast0.Include(inc,name))
| Ast0.MetaInclude(inc,name) ->
None (* likely not correct, wait for pb in practice *)
| Ast0.Undef(def,id) ->
(* nothing available for ident, and not sure code can appear
here anyway *)
None
| Ast0.Define(def,id,params,body) ->
call_right right_statement_dots body s
(function body -> Ast0.Define(def,id,params,body))
| Ast0.Pragma(prg,id,body) ->
(* seems safe; let the code go with the enclosing statement, if any *)
None
| Ast0.OptStm(re) ->
call_right right_statement re s (function re -> Ast0.OptStm(re))
and right_statement_dots sd =
match Ast0.unwrap sd with
[] -> failwith "empty statement dots"
| s::r -> call_right right_statement s sd (function s -> List.rev(s::r))
let rec left_ty t =
match Ast0.unwrap t with
Ast0.ConstVol(cv,ty) ->
call_right left_mcode cv t (function cv -> Ast0.ConstVol(cv,ty))
| Ast0.BaseType(ty,strings) ->
(match strings with
[] -> failwith "empty strings in type"
| s::r ->
call_right left_mcode s t (function s -> Ast0.BaseType(ty,s::r)))
| Ast0.Signed(sign,ty) ->
call_right left_mcode sign t (function sign -> Ast0.Signed(sign,ty))
| Ast0.Pointer(ty,star) ->
call_right left_ty ty t (function ty -> Ast0.Pointer(ty,star))
| Ast0.FunctionPointer(ty,lp1,star,rp1,lp2,params,rp2) ->
call_right left_ty ty t
(function ty -> Ast0.FunctionPointer(ty,lp1,star,rp1,lp2,params,rp2))
| Ast0.Array(ty,lb,size,rb) ->
call_right left_ty ty t (function ty -> Ast0.Array(ty,lb,size,rb))
| Ast0.Decimal(dec,lp,length,comma,precision_opt,rp) ->
call_right left_mcode dec t
(function dec -> Ast0.Decimal(dec,lp,length,comma,precision_opt,rp))
| Ast0.EnumName(kind,name) ->
call_right left_mcode kind t (function kind -> Ast0.EnumName(kind,name))
| Ast0.EnumDef(ty,lb,ids,rb) ->
call_right left_ty ty t
(function ty -> Ast0.EnumDef(ty,lb,ids,rb))
| Ast0.StructUnionName(kind,name) ->
call_right left_mcode kind t
(function kind -> Ast0.StructUnionName(kind,name))
| Ast0.StructUnionDef(ty,lb,decls,rb) ->
call_right left_ty ty t
(function ty -> Ast0.StructUnionDef(ty,lb,decls,rb))
| Ast0.TypeOfExpr(tf,lb,e,rb) ->
call_right left_mcode tf t (function tf -> Ast0.TypeOfExpr(tf,lb,e,rb))
| Ast0.TypeOfType(tf,lb,ty,rb) ->
call_right left_mcode tf t (function tf -> Ast0.TypeOfType(tf,lb,ty,rb))
| Ast0.TypeName(name) ->
call_right left_mcode name t (function name -> Ast0.TypeName(name))
| Ast0.MetaType(name,cstr,x) ->
call_right left_mcode name t (function name -> Ast0.MetaType(name,cstr,x))
| Ast0.AsType(ty,asty) -> failwith "not possible"
| Ast0.DisjType(starter,types,mids,ender)
| Ast0.ConjType(starter,types,mids,ender) -> None
| Ast0.OptType(ty) ->
call_right left_ty ty t (function ty -> Ast0.OptType(ty))
let rec left_ident i =
match Ast0.unwrap i with
Ast0.Id(name) ->
call_right left_mcode name i (function name -> Ast0.Id(name))
| Ast0.MetaId(name,a,b,c) ->
call_right left_mcode name i (function name -> Ast0.MetaId(name,a,b,c))
| Ast0.MetaFunc(name,a,b) ->
call_right left_mcode name i (function name -> Ast0.MetaFunc(name,a,b))
| Ast0.MetaLocalFunc(name,a,b) ->
call_right left_mcode name i
(function name -> Ast0.MetaLocalFunc(name,a,b))
| Ast0.DisjId(starter,ids,mids,ender)
| Ast0.ConjId(starter,ids,mids,ender) -> None
| Ast0.OptIdent(id) ->
call_right left_ident id i (function id -> Ast0.OptIdent(id))
| Ast0.AsIdent(id,asid) -> failwith "not possible"
let left_fundecl name fninfo =
let fncall_right processor data cont =
match processor data with
None -> None
| Some(pragmas,data) -> Some (pragmas,cont data,name) in
match fninfo with
[] ->
(match left_ident name with
None -> None
| Some(pragmas,name) -> Some(pragmas,fninfo,name))
| (Ast0.FStorage sto)::x ->
fncall_right left_mcode sto (function sto -> (Ast0.FStorage sto)::x)
| (Ast0.FType ty)::x ->
fncall_right left_ty ty (function ty -> (Ast0.FType ty)::x)
| (Ast0.FInline inl)::x ->
fncall_right left_mcode inl (function inl -> (Ast0.FInline inl)::x)
| (Ast0.FAttr atr)::x ->
fncall_right left_mcode atr (function atr -> (Ast0.FAttr atr)::x)
let rec left_decl decl =
match Ast0.unwrap decl with
Ast0.MetaDecl(name,cstr,pure) ->
call_right right_mcode name decl
(function name -> Ast0.MetaDecl(name,cstr,pure))
| Ast0.AsDecl(decl,asdecl) -> failwith "not possible"
| Ast0.Init(Some stg,ty,id,attr,eq,ini,sem) ->
call_right left_mcode stg decl
(function stg -> Ast0.Init(Some stg,ty,id,attr,eq,ini,sem))
| Ast0.Init(None,ty,id,attr,eq,ini,sem) ->
call_right left_ty ty decl
(function ty -> Ast0.Init(None,ty,id,attr,eq,ini,sem))
| Ast0.UnInit(Some stg,ty,id,attr,sem) ->
call_right left_mcode stg decl
(function stg -> Ast0.UnInit(Some stg,ty,id,attr,sem))
| Ast0.UnInit(None,ty,id,attr,sem) ->
call_right left_ty ty decl
(function ty -> Ast0.UnInit(None,ty,id,attr,sem))
| Ast0.FunProto(fi,name,lp1,params,va,rp1,sem) ->
(match fi with
[] ->
call_right left_ident name decl
(function name -> Ast0.FunProto(fi,name,lp1,params,va,rp1,sem))
| (Ast0.FStorage sto)::x ->
call_right left_mcode sto decl
(function sto ->
Ast0.FunProto((Ast0.FStorage sto)::x,name,lp1,params,va,rp1,sem))
| (Ast0.FType ty)::x ->
call_right left_ty ty decl
(function ty ->
Ast0.FunProto((Ast0.FType ty)::x,name,lp1,params,va,rp1,sem))
| (Ast0.FInline inl)::x ->
call_right left_mcode inl decl
(function inl ->
Ast0.FunProto((Ast0.FInline inl)::x,name,lp1,params,va,rp1,sem))
| (Ast0.FAttr attr)::x ->
call_right left_mcode attr decl
(function attr ->
Ast0.FunProto((Ast0.FAttr attr)::x,name,lp1,params,va,rp1,sem)))
| Ast0.MacroDecl(Some stg,name,lp,args,rp,sem) ->
call_right left_mcode stg decl
(function stg -> Ast0.MacroDecl(Some stg,name,lp,args,rp,sem))
| Ast0.MacroDecl(None,name,lp,args,rp,sem) ->
call_right left_ident name decl
(function name -> Ast0.MacroDecl(None,name,lp,args,rp,sem))
| Ast0.MacroDeclInit(Some stg,name,lp,args,rp,eq,ini,sem) ->
call_right left_mcode stg decl
(function stg ->
Ast0.MacroDeclInit(Some stg,name,lp,args,rp,eq,ini,sem))
| Ast0.MacroDeclInit(None,name,lp,args,rp,eq,ini,sem) ->
call_right left_ident name decl
(function name -> Ast0.MacroDeclInit(None,name,lp,args,rp,eq,ini,sem))
| Ast0.TyDecl(ty,sem) ->
call_right left_ty ty decl (function ty -> Ast0.TyDecl(ty,sem))
| Ast0.Typedef(stg,ty,id,sem) ->
call_right left_mcode stg decl
(function stg -> Ast0.Typedef(stg,ty,id,sem))
| Ast0.DisjDecl(starter,decls,mids,ender)
| Ast0.ConjDecl(starter,decls,mids,ender) -> None
| Ast0.OptDecl(d) ->
call_right left_decl d decl (function decl -> Ast0.OptDecl(decl))
let process =
let statement r k s =
let s = k s in
Ast0.rewrap s
(match Ast0.unwrap s with
Ast0.FunDecl(bef,fi,name,lp,params,va,rp,lbrace,body,rbrace,aft) ->
let (rbrace,aft) =
match right_mcode rbrace with
None -> (rbrace,aft)
| Some (pragmas,rbrace) -> (rbrace,update_after pragmas aft) in
(match left_fundecl name fi with
None ->
Ast0.FunDecl(bef,fi,name,lp,params,va,rp,lbrace,body,rbrace,aft)
| Some (pragmas,fi,name) ->
Ast0.FunDecl
(update_before2 pragmas bef,
fi,name,lp,params,va,rp,lbrace,body,rbrace,aft))
| Ast0.Decl(bef,decl) ->
(match left_decl decl with
None -> Ast0.unwrap s
| Some (pragmas,decl) ->
Ast0.Decl(update_before2 pragmas bef,decl))
| Ast0.IfThen(iff,lp,exp,rp,branch1,aft) ->
(match right_statement branch1 with
None -> Ast0.unwrap s
| Some (pragmas,branch1) ->
Ast0.IfThen
(iff,lp,exp,rp,branch1,update_before pragmas aft))
| Ast0.IfThenElse(iff,lp,exp,rp,branch1,els,branch2,aft) ->
(match right_statement branch2 with
None -> Ast0.unwrap s
| Some (pragmas,branch2) ->
Ast0.IfThenElse
(iff,lp,exp,rp,branch1,els,branch2,
update_before pragmas aft))
| Ast0.While(whl,lp,exp,rp,body,aft) ->
(match right_statement body with
None -> Ast0.unwrap s
| Some (pragmas,body) ->
Ast0.While(whl,lp,exp,rp,body,update_before pragmas aft))
| Ast0.For(fr,lp,first,e2,sem2,e3,rp,body,aft) ->
(match right_statement body with
None -> Ast0.unwrap s
| Some (pragmas,body) ->
Ast0.For
(fr,lp,first,e2,sem2,e3,rp,body,
update_before pragmas aft))
| Ast0.Iterator(nm,lp,args,rp,body,aft) ->
(match right_statement body with
None -> Ast0.unwrap s
| Some (pragmas,body) ->
Ast0.Iterator(nm,lp,args,rp,body,update_before pragmas aft))
| _ -> Ast0.unwrap s) in
let res = V0.rebuilder
{V0.rebuilder_functions with VT0.rebuilder_stmtfn = statement} in
List.map res.VT0.rebuilder_rec_top_level
|
