make dependency edge handling more robust
[libfirm] / ir / ir / instrument.c
1 /*
2  * Copyright (C) 1995-2008 University of Karlsruhe.  All right reserved.
3  *
4  * This file is part of libFirm.
5  *
6  * This file may be distributed and/or modified under the terms of the
7  * GNU General Public License version 2 as published by the Free Software
8  * Foundation and appearing in the file LICENSE.GPL included in the
9  * packaging of this file.
10  *
11  * Licensees holding valid libFirm Professional Edition licenses may use
12  * this file in accordance with the libFirm Commercial License.
13  * Agreement provided with the Software.
14  *
15  * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
16  * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE.
18  */
19
20 /**
21  * @file
22  * @brief   Instrumentation of graphs.
23  * @date    14.4.2008
24  * @author  Michael Beck
25  */
26 #include "config.h"
27
28 #include "irgraph_t.h"
29 #include "iredges.h"
30 #include "error.h"
31 #include "ircons.h"
32 #include "instrument.h"
33
34 void instrument_initcall(ir_graph *irg, ir_entity *ent)
35 {
36         const ir_edge_t *edge;
37         ir_node         *initial_exec;
38         ir_node         *initial_mem;
39         ir_node         *start_block;
40         ir_node         *adr, *call, *new_mem;
41         ir_node         *first_block = NULL;
42         int             i, idx, need_new_block;
43         symconst_symbol sym;
44
45         assure_edges(irg);
46
47         /* find the first block */
48         initial_exec = get_irg_initial_exec(irg);
49         start_block  = get_irg_start_block(irg);
50
51         foreach_out_edge(initial_exec, edge) {
52                 ir_node *succ = get_edge_src_irn(edge);
53
54                 if (succ != start_block && is_Block(succ)) {
55                         /* found the first block */
56                         first_block = succ;
57                         break;
58                 }
59         }
60         if (first_block == NULL) {
61                 panic("Cannot find first block of irg %+F", irg);
62         }
63
64         /* check if this block has only one predecessor */
65         idx = -1;
66         need_new_block = 0;
67         for (i = get_Block_n_cfgpreds(first_block) - 1; i >= 0; --i) {
68                 ir_node *p = get_Block_cfgpred(first_block, i);
69
70                 if (is_Bad(p))
71                         continue;
72                 if (p == initial_exec)
73                         idx = i;
74                 else
75                         need_new_block = 1;
76         }
77
78         if (need_new_block) {
79                 ir_node *blk = new_r_Block(irg, 1, &initial_exec);
80                 set_Block_cfgpred(first_block, idx, new_r_Jmp(blk));
81                 first_block = blk;
82         }
83
84         /* place the call */
85         sym.entity_p = ent;
86         adr = new_r_SymConst(irg, mode_P_code, sym, symconst_addr_ent);
87
88         call    = new_r_Call(first_block, get_irg_no_mem(irg), adr, 0, NULL, get_entity_type(ent));
89         new_mem = new_r_Proj(call, mode_M, pn_Call_M);
90
91         initial_mem = get_irg_initial_mem(irg);
92         edges_reroute(initial_mem, new_mem);
93         /* beware: reroute routes anchor edges also, revert this */
94         set_irg_initial_mem(irg, initial_mem);
95         set_Call_mem(call, initial_mem);
96 }