parsing_c/test_parsing_c.ml

   1 open Common
   2
   3 open Ast_c
   4
   5 let score_path = "/home/pad/c-yacfe/tmp"
   6
   7 let tmpfile = "/tmp/output.c"
   8
   9 module Ast_to_flow = Control_flow_c_build
  10
  11 (*****************************************************************************)
  12 (* Subsystem testing *)
  13 (*****************************************************************************)
  14
  15 let test_tokens_c file =
  16   if not (file =~ ".*\\.c")
  17   then pr2 "warning: seems not a .c file";
  18
  19   Flag_parsing_c.debug_lexer := true;
  20   Flag_parsing_c.verbose_lexing := true;
  21   Flag_parsing_c.verbose_parsing := true;
  22
  23   Parse_c.tokens file +> List.iter (fun x -> pr2_gen x);
  24   ()
  25
  26
  27
  28 (* ---------------------------------------------------------------------- *)
  29 let test_parse_gen xs ext =
  30
  31   Flag_parsing_c.debug_typedef := true;
  32   Flag_parsing_c.debug_cpp := true;
  33   Flag_parsing_c.debug_etdt := false;
  34   Flag_parsing_c.filter_msg := true;
  35
  36   (*let dirname_opt =
  37     match xs with
  38     | [x] when is_directory x -> Some x
  39     | _ -> None
  40   in*)
  41
  42   (* old:
  43      let xs = if !Flag.dir then
  44      process_output_to_list ("find " ^ x ^" -name \"*.c\"") else x::xs in
  45   *)
  46   let fullxs = Common.files_of_dir_or_files_no_vcs ext xs in
  47
  48   let stat_list = ref [] in
  49   let newscore  = Common.empty_score () in
  50
  51   Common.check_stack_nbfiles (List.length fullxs);
  52
  53   fullxs +> List.iter (fun file ->
  54     if not (file =~ (".*\\."^ext))
  55     then pr2 ("warning: seems not a ."^ext^" file");
  56
  57
  58     pr2 "";
  59     pr2 ("PARSING: " ^ file);
  60
  61     let (xs, stat) = Parse_c.parse_c_and_cpp file in
  62     xs +> List.iter (fun (ast, (s, toks)) ->
  63       Parse_c.print_commentized toks
  64     );
  65
  66     Common.push2 stat stat_list;
  67     let s =
  68       sprintf "bad = %d, timeout = %B"
  69         stat.Parsing_stat.bad stat.Parsing_stat.have_timeout
  70     in
  71     if stat.Parsing_stat.bad =|= 0 && not stat.Parsing_stat.have_timeout
  72     then Hashtbl.add newscore file (Common.Ok)
  73     else Hashtbl.add newscore file (Common.Pb s)
  74   );
  75
  76 (* uses an explicit path; to fix
  77   dirname_opt +> Common.do_option (fun dirname ->
  78     pr2_xxxxxxxxxxxxxxxxx();
  79     pr2 "regression testing  information";
  80     pr2_xxxxxxxxxxxxxxxxx();
  81     let str = Str.global_replace (Str.regexp "/") "__" dirname in
  82     let def = if !Flag_parsing_c.filter_define_error then "_def_" else "" in
  83     let ext = if ext =$= "c" then "" else ext in
  84     let filename = "score_parsing__" ^str ^ def ^ ext ^ ".marshalled" in
  85     if Sys.file_exists filename
  86     then
  87       Common.regression_testing newscore
  88         (Filename.concat score_path
  89            ("score_parsing__" ^str ^ def ^ ext ^ ".marshalled"))
  90   );
  91 *)
  92
  93   if not (null !stat_list)
  94   then begin
  95     Parsing_stat.print_recurring_problematic_tokens !stat_list;
  96     Parsing_stat.print_parsing_stat_list !stat_list;
  97   end;
  98   ()
  99
 100
 101 let test_parse_c xs =
 102   test_parse_gen xs "c"
 103 let test_parse_h xs =
 104   test_parse_gen xs "h"
 105 let test_parse_ch xs =
 106   test_parse_gen xs "[ch]"
 107
 108 (* could use a simpler parser than heavy parse_c_and_cpp here as there
 109  * is no more cpp stuff in the .i files
 110  *)
 111 let test_parse_i xs =
 112   test_parse_gen xs "i"
 113
 114
 115
 116
 117
 118
 119
 120 (* ---------------------------------------------------------------------- *)
 121 (* file can be   "foo.c"  or "foo.c:main" *)
 122 (* local function that is parameterized by whether to launch gv *)
 123 let local_test_cfg launchgv file =
 124   let (file, specific_func) =
 125     if file =~ "\\(.*\\.c\\):\\(.*\\)"
 126     then
 127       let (a,b) = matched2 file in
 128       a, Some b
 129     else
 130       file, None
 131   in
 132
 133   if not (file =~ ".*\\.c")
 134   then pr2 "warning: seems not a .c file";
 135
 136   let (program, _stat) = Parse_c.parse_c_and_cpp file in
 137
 138   program +> List.iter (fun (e,_) ->
 139     let toprocess =
 140       match specific_func, e with
 141       | None, _ -> true
 142       | Some s, Ast_c.Definition (defbis,_)  ->
 143           s =$= Ast_c.str_of_name (defbis.Ast_c.f_name)
 144       | _, _ -> false
 145     in
 146
 147     if toprocess
 148     then
 149       (* old: Flow_to_ast.test !Flag.show_flow def *)
 150       (try
 151           let flow = Ast_to_flow.ast_to_control_flow e in
 152           flow +> do_option (fun flow ->
 153             Ast_to_flow.deadcode_detection flow;
 154             let flow = Ast_to_flow.annotate_loop_nodes flow in
 155
 156             let flow' =
 157 (*
 158               if !Flag_cocci.show_before_fixed_flow
 159               then flow
 160               else Ctlcocci_integration.fix_flow_ctl flow
 161 *)
 162               flow
 163             in
 164             let filename = Filename.temp_file "output" ".dot" in
 165             Ograph_extended.print_ograph_mutable flow' (filename) launchgv
 166           )
 167         with Ast_to_flow.Error (x) -> Ast_to_flow.report_error x
 168       )
 169   )
 170
 171 let test_cfg = local_test_cfg true
 172
 173
 174
 175 let test_cfg_ifdef file =
 176   let (ast2, _stat) = Parse_c.parse_c_and_cpp file in
 177   let ast = Parse_c.program_of_program2 ast2 in
 178
 179   let ast = Cpp_ast_c.cpp_ifdef_statementize ast in
 180
 181   ast +> List.iter (fun e ->
 182     (try
 183         let flow = Ast_to_flow.ast_to_control_flow e in
 184         flow +> do_option (fun flow ->
 185           Ast_to_flow.deadcode_detection flow;
 186           let flow = Ast_to_flow.annotate_loop_nodes flow in
 187           Ograph_extended.print_ograph_mutable flow ("/tmp/output.dot") true
 188         )
 189       with Ast_to_flow.Error (x) -> Ast_to_flow.report_error x
 190     )
 191   )
 192
 193 (* ---------------------------------------------------------------------- *)
 194 let test_parse_unparse infile =
 195   if not (infile =~ ".*\\.c")
 196   then pr2 "warning: seems not a .c file";
 197
 198   let (program2, _stat) = Parse_c.parse_c_and_cpp infile in
 199   let program2_with_ppmethod =
 200     program2 +> List.map (fun x -> x, Unparse_c.PPnormal)
 201   in
 202   Unparse_c.pp_program program2_with_ppmethod tmpfile;
 203   Common.command2 ("cat " ^ tmpfile);
 204   (* if want see diff of space => no -b -B *)
 205   Common.command2 (spf "diff -u -p  %s %s" infile tmpfile);
 206   (* +> Transformation.test_simple_trans1;*)
 207   ()
 208
 209
 210 let parse_and_print_sexp file =
 211   let (ast2,_stat) = Parse_c.parse_c_and_cpp file in
 212   let ast = Parse_c.program_of_program2 ast2 in
 213   let _ast =
 214     Type_annoter_c.annotate_program !Type_annoter_c.initial_env ast
 215   in
 216
 217   (*
 218   let sexp = Sexp_ast_c.sexp_of_program ast in
 219   let s = Sexp.to_string_hum sexp in
 220   *)
 221   Sexp_ast_c.show_info := false;
 222   let s = Sexp_ast_c.string_of_program ast in
 223   pr2 s;
 224   ()
 225
 226
 227
 228 let test_type_c infile =
 229   if not (infile =~ ".*\\.c")
 230   then pr2 "warning: seems not a .c file";
 231
 232   Flag_parsing_c.pretty_print_type_info := true;
 233
 234   let (program2, _stat) =  Parse_c.parse_c_and_cpp infile in
 235   let _program2 =
 236     program2
 237     +> Common.unzip
 238     +> (fun (program, infos) ->
 239       Type_annoter_c.annotate_program !Type_annoter_c.initial_env
 240         program +> List.map fst,
 241       infos
 242     )
 243     +> Common.uncurry Common.zip
 244   in
 245   let program2_with_ppmethod =
 246     program2 +> List.map (fun x -> x, Unparse_c.PPnormal)
 247   in
 248   Unparse_c.pp_program program2_with_ppmethod tmpfile;
 249   Common.command2 ("cat " ^ tmpfile);
 250   ();;
 251
 252
 253 (* ---------------------------------------------------------------------- *)
 254 (* ex: demos/platform_ifdef.c *)
 255 let test_comment_annotater infile =
 256   let (program2, _stat) =  Parse_c.parse_c_and_cpp infile in
 257   let asts = program2 +> List.map (fun (ast,_) -> ast) in
 258   let toks = program2 +> List.map (fun (ast, (s, toks)) -> toks) +>
 259     List.flatten in
 260
 261   Flag_parsing_c.pretty_print_comment_info := true;
 262
 263   pr2 "pretty print, before comment annotation: --->";
 264   Common.adjust_pp_with_indent (fun () ->
 265   asts +> List.iter (fun ast ->
 266     Pretty_print_c.pp_toplevel_simple ast;
 267   );
 268   );
 269
 270   let _ = Comment_annotater_c.annotate_program toks asts in
 271
 272   Common.adjust_pp_with_indent (fun () ->
 273   pr2 "pretty print, after comment annotation: --->";
 274   asts +> List.iter (fun ast ->
 275     Pretty_print_c.pp_toplevel_simple ast;
 276   );
 277   );
 278
 279
 280   ()
 281
 282
 283 (* ---------------------------------------------------------------------- *)
 284 (* used by generic_makefile now *)
 285 let test_compare_c file1 file2 =
 286   let (correct, diffxs) = Compare_c.compare_default file1 file2 in
 287   let res = Compare_c.compare_result_to_bool correct in
 288   if res
 289   then raise (Common.UnixExit 0)
 290   else raise (Common.UnixExit (-1))
 291
 292
 293 let test_compare_c_hardcoded () =
 294   Compare_c.compare_default
 295     "tests/compare1.c"
 296     "tests/compare2.c"
 297     (*
 298       "tests/equal_modulo1.c"
 299       "tests/equal_modulo2.c"
 300     *)
 301   +> Compare_c.compare_result_to_string
 302   +> pr2
 303
 304
 305
 306 (* ---------------------------------------------------------------------- *)
 307 let test_attributes file =
 308   let (ast2, _stat) = Parse_c.parse_c_and_cpp file in
 309   let ast = Parse_c.program_of_program2 ast2 in
 310
 311   Visitor_c.vk_program { Visitor_c.default_visitor_c with
 312     Visitor_c.kdef = (fun (k, bigf) (defbis, ii) ->
 313       let sattr  = Ast_c.s_of_attr defbis.f_attr in
 314       pr2 (spf "%-30s: %s" (Ast_c.str_of_name (defbis.f_name)) sattr);
 315     );
 316     Visitor_c.kdecl = (fun (k, bigf) decl ->
 317       match decl with
 318       | DeclList (xs, ii) ->
 319           xs +> List.iter (fun (onedecl, iicomma) ->
 320
 321             let sattr  = Ast_c.s_of_attr onedecl.v_attr in
 322             let idname =
 323               match onedecl.v_namei with
 324               | Some (name, ini) -> Ast_c.str_of_name name
 325               | None -> "novar"
 326             in
 327             pr2 (spf "%-30s: %s" idname sattr);
 328           );
 329       | _ -> ()
 330
 331     );
 332   } ast;
 333   ()
 334
 335
 336 let cpp_options () = [
 337   Cpp_ast_c.I "/home/yyzhou/pad/linux/include";
 338 ] ++
 339   Cpp_ast_c.cpp_option_of_cmdline
 340   (!Flag_parsing_c.cpp_i_opts,!Flag_parsing_c.cpp_d_opts)
 341
 342 let test_cpp file =
 343   let (ast2, _stat) = Parse_c.parse_c_and_cpp file in
 344   let dirname = Filename.dirname file in
 345   let ast = Parse_c.program_of_program2 ast2 in
 346   let ast = Cpp_ast_c.cpp_expand_include (cpp_options()) dirname ast in
 347   let _ast = Cpp_ast_c.cpp_ifdef_statementize ast in
 348
 349
 350   ()
 351
 352
 353
 354 (* CONFIG [ch] ? also do for .c ? maybe less needed now that I
 355  * add local_macros.
 356  *)
 357 let extract_macros ~selection dir =
 358   let ext = "h" in
 359   let fullxs = Common.files_of_dir_or_files_no_vcs ext [dir] in
 360   let macros_and_filename =
 361     fullxs +> List.map (fun file ->
 362       pr2 (spf "processing: %s" file);
 363       let xs = Parse_c.extract_macros file in
 364       file, xs
 365     )
 366   in
 367
 368   let macros =
 369     if selection
 370     then Cpp_analysis_c.extract_dangerous_macros macros_and_filename
 371     else macros_and_filename
 372   in
 373   macros +> List.iter (fun (file, defs) ->
 374     pr ("/* PARSING: " ^ file ^ " */");
 375     defs +> List.iter (fun (s, def) ->
 376       let str = Cpp_token_c.string_of_define_def def in
 377       pr str;
 378     )
 379   );
 380   ()
 381
 382
 383 let test_parse xs =
 384
 385   Flag_parsing_c.filter_msg_define_error := true;
 386   Flag_parsing_c.filter_define_error := true;
 387   Flag_parsing_c.verbose_lexing := false;
 388   Flag_parsing_c.verbose_parsing := false;
 389
 390   let dirname_opt =
 391     match xs with
 392     | [x] when is_directory x -> Some x
 393     | _ -> None
 394   in
 395   dirname_opt +> Common.do_option (fun dir ->
 396
 397     let ext = "h" in
 398     let fullxs = Common.files_of_dir_or_files_no_vcs ext [dir] in
 399
 400     let macros_and_filename =
 401       fullxs +> List.map (fun file ->
 402         pr2 (spf "processing: %s" file);
 403         let xs = Parse_c.extract_macros file in
 404         file, xs
 405       )
 406     in
 407     let macros =
 408       Cpp_analysis_c.extract_dangerous_macros macros_and_filename
 409     in
 410     macros +> List.iter (fun (file, xs) ->
 411       xs +> List.iter (fun (x, def) ->
 412         let (s, params, body) = def in
 413         let str = Cpp_token_c.string_of_define_def def in
 414         pr str;
 415         (* builtins ? *)
 416         Hashtbl.replace !Parse_c._defs_builtins s (s, params, body);
 417       );
 418     );
 419   );
 420
 421   let ext = "[ch]" in
 422
 423   let fullxs = Common.files_of_dir_or_files_no_vcs ext xs in
 424
 425   let stat_list = ref [] in
 426   Common.check_stack_nbfiles (List.length fullxs);
 427
 428   fullxs +> List.iter (fun file ->
 429     if not (file =~ (".*\\."^ext))
 430     then pr2 ("warning: seems not a ."^ext^" file");
 431
 432     pr2 "";
 433     pr2 ("PARSING: " ^ file);
 434
 435     let (xs, stat) = Parse_c.parse_c_and_cpp file in
 436     xs +> List.iter (fun (ast, (s, toks)) ->
 437       Parse_c.print_commentized toks
 438     );
 439
 440     Common.push2 stat stat_list;
 441   );
 442
 443   if not (null !stat_list)
 444   then begin
 445     Parsing_stat.print_recurring_problematic_tokens !stat_list;
 446     Parsing_stat.print_parsing_stat_list !stat_list;
 447   end;
 448   ()
 449
 450
 451
 452
 453
 454
 455
 456
 457 (* ---------------------------------------------------------------------- *)
 458 let test_xxx a  =
 459   ()
 460
 461 (*
 462   ignore(Parse_c.parse_cpp_define_file "standard.h")
 463   pr2 "pr2";
 464   pr  "pr"
 465
 466   Format.print_newline();
 467   Format.printf "@[<v 5>--@,--@,@[<v 5>--@,--@,@]--@,--@,@]";
 468   Format.print_newline();
 469   Format.printf "@[<v>(---@[<v>(---@[<v>(---@,)@]@,)@]@,)@]"
 470 *)
 471
 472
 473
 474 (*****************************************************************************)
 475 (* Main entry for Arg *)
 476 (*****************************************************************************)
 477
 478 let actions () = [
 479   "-tokens_c", "   <file>",
 480   Common.mk_action_1_arg test_tokens_c;
 481   "-parse_c", "   <file or dir>",
 482   Common.mk_action_n_arg test_parse_c;
 483   "-parse_h", "   <file or dir>",
 484   Common.mk_action_n_arg test_parse_h;
 485   "-parse_ch", "   <file or dir>",
 486   Common.mk_action_n_arg test_parse_ch;
 487   "-parse_i", "   <file or dir>",
 488   Common.mk_action_n_arg test_parse_i;
 489
 490   "-parse", "   <file or dir>",
 491   Common.mk_action_n_arg test_parse;
 492
 493   "-show_flow", "   <file or file:function>",
 494   Common.mk_action_1_arg (local_test_cfg true);
 495   "-control_flow", "   <file or file:function>",
 496   Common.mk_action_1_arg (local_test_cfg true);
 497   "-control_flow_to_file", "   <file or file:function>",
 498   Common.mk_action_1_arg (local_test_cfg false);
 499   "-test_cfg_ifdef", " <file>",
 500   Common.mk_action_1_arg test_cfg_ifdef;
 501   "-parse_unparse", "   <file>",
 502   Common.mk_action_1_arg test_parse_unparse;
 503   "-parse_and_print_sexp", "   <file>",
 504     Common.mk_action_1_arg parse_and_print_sexp;
 505   "-type_c", "   <file>",
 506   Common.mk_action_1_arg test_type_c;
 507   "-compare_c", "   <file1> <file2>",
 508   Common.mk_action_2_arg test_compare_c (* result is in unix code *);
 509   "-comment_annotater_c", "   <file>",
 510   Common.mk_action_1_arg test_comment_annotater;
 511
 512   "-compare_c_hardcoded", "  ",
 513   Common.mk_action_0_arg test_compare_c_hardcoded;
 514
 515   "-test_attributes", " <file>",
 516   Common.mk_action_1_arg test_attributes;
 517   "-test_cpp", " <file>",
 518   Common.mk_action_1_arg test_cpp;
 519
 520   "-extract_macros", " <file or dir>",
 521   Common.mk_action_1_arg (extract_macros ~selection:false) ;
 522
 523   "-extract_macros_select", " <file or dir>",
 524   Common.mk_action_1_arg (extract_macros ~selection:true);
 525
 526
 527   "-xxx", "   <file1> <>",
 528   Common.mk_action_n_arg test_xxx;
 529 ]
 530