njn | c953984 | 2002-10-02 13:26:35 +0000 | [diff] [blame] | 1 | |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 2 | /*--------------------------------------------------------------------*/ |
nethercote | 137bc55 | 2003-11-14 17:47:54 +0000 | [diff] [blame] | 3 | /*--- Simple tool for counting UInstrs, using a C helper. ---*/ |
njn25 | cac76cb | 2002-09-23 11:21:57 +0000 | [diff] [blame] | 4 | /*--- lk_main.c ---*/ |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 5 | /*--------------------------------------------------------------------*/ |
| 6 | |
| 7 | /* |
nethercote | 137bc55 | 2003-11-14 17:47:54 +0000 | [diff] [blame] | 8 | This file is part of Lackey, an example Valgrind tool that does |
njn | c953984 | 2002-10-02 13:26:35 +0000 | [diff] [blame] | 9 | some simple program measurement. |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 10 | |
njn | 5361242 | 2005-03-12 16:22:54 +0000 | [diff] [blame] | 11 | Copyright (C) 2002-2005 Nicholas Nethercote |
njn | 2bc1012 | 2005-05-08 02:10:27 +0000 | [diff] [blame] | 12 | njn@valgrind.org |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 13 | |
| 14 | This program is free software; you can redistribute it and/or |
| 15 | modify it under the terms of the GNU General Public License as |
| 16 | published by the Free Software Foundation; either version 2 of the |
| 17 | License, or (at your option) any later version. |
| 18 | |
| 19 | This program is distributed in the hope that it will be useful, but |
| 20 | WITHOUT ANY WARRANTY; without even the implied warranty of |
| 21 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 22 | General Public License for more details. |
| 23 | |
| 24 | You should have received a copy of the GNU General Public License |
| 25 | along with this program; if not, write to the Free Software |
| 26 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA |
| 27 | 02111-1307, USA. |
| 28 | |
| 29 | The GNU General Public License is contained in the file COPYING. |
| 30 | */ |
| 31 | |
nethercote | 4606320 | 2004-09-02 08:51:43 +0000 | [diff] [blame] | 32 | #include "tool.h" |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 33 | |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 34 | /* Nb: use ULongs because the numbers can get very big */ |
| 35 | static ULong n_dlrr_calls = 0; |
| 36 | static ULong n_BBs = 0; |
| 37 | static ULong n_UInstrs = 0; |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 38 | static ULong n_guest_instrs = 0; |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 39 | static ULong n_Jccs = 0; |
| 40 | static ULong n_Jccs_untaken = 0; |
| 41 | |
| 42 | static void add_one_dlrr_call(void) |
| 43 | { |
| 44 | n_dlrr_calls++; |
| 45 | } |
| 46 | |
njn | 26f0251 | 2004-11-22 18:33:15 +0000 | [diff] [blame] | 47 | /* See comment above TL_(instrument) for reason why n_machine_instrs is |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 48 | incremented here. */ |
| 49 | static void add_one_BB(void) |
| 50 | { |
| 51 | n_BBs++; |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 52 | n_guest_instrs++; |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 53 | } |
| 54 | |
| 55 | static void add_one_UInstr(void) |
| 56 | { |
| 57 | n_UInstrs++; |
| 58 | } |
| 59 | |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 60 | static void add_one_guest_instr(void) |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 61 | { |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 62 | n_guest_instrs++; |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 63 | } |
| 64 | |
| 65 | static void add_one_Jcc(void) |
| 66 | { |
| 67 | n_Jccs++; |
| 68 | } |
| 69 | |
| 70 | static void add_one_Jcc_untaken(void) |
| 71 | { |
| 72 | n_Jccs_untaken++; |
| 73 | } |
| 74 | |
njn | 26f0251 | 2004-11-22 18:33:15 +0000 | [diff] [blame] | 75 | void TL_(pre_clo_init)(void) |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 76 | { |
njn | 810086f | 2002-11-14 12:42:47 +0000 | [diff] [blame] | 77 | VG_(details_name) ("Lackey"); |
| 78 | VG_(details_version) (NULL); |
nethercote | 137bc55 | 2003-11-14 17:47:54 +0000 | [diff] [blame] | 79 | VG_(details_description) ("an example Valgrind tool"); |
njn | 810086f | 2002-11-14 12:42:47 +0000 | [diff] [blame] | 80 | VG_(details_copyright_author)( |
njn | 5361242 | 2005-03-12 16:22:54 +0000 | [diff] [blame] | 81 | "Copyright (C) 2002-2005, and GNU GPL'd, by Nicholas Nethercote."); |
nethercote | 421281e | 2003-11-20 16:20:55 +0000 | [diff] [blame] | 82 | VG_(details_bug_reports_to) (VG_BUGS_TO); |
sewardj | c0d8f68 | 2002-11-30 00:49:43 +0000 | [diff] [blame] | 83 | VG_(details_avg_translation_sizeB) ( 175 ); |
njn | 8a97c6d | 2005-03-31 04:37:24 +0000 | [diff] [blame] | 84 | |
| 85 | VG_(basic_tool_funcs) (TL_(post_clo_init), |
| 86 | TL_(instrument), |
| 87 | TL_(fini)); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 88 | } |
| 89 | |
njn | 26f0251 | 2004-11-22 18:33:15 +0000 | [diff] [blame] | 90 | void TL_(post_clo_init)(void) |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 91 | { |
| 92 | } |
| 93 | |
nethercote | 7e08f06 | 2004-10-20 10:37:00 +0000 | [diff] [blame] | 94 | /* Note: machine instructions are marked by an INCEIP at the end of each one, |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 95 | except for the final one in the basic block which ends in an |
| 96 | unconditional JMP. Sometimes the final unconditional JMP is preceded by |
| 97 | a conditional JMP (Jcc), and thus it isn't reached. Eg: |
| 98 | |
| 99 | <code a> |
| 100 | INCEIP ... |
| 101 | |
| 102 | <code b> |
| 103 | Jcc ... |
| 104 | JMP ... (will not be reached if Jcc succeeds) |
| 105 | |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 106 | If we simplemindedly added calls to add_one_guest_instr() before INCEIPs |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 107 | and unconditional JMPs, we'd sometimes miss the final call (when a |
nethercote | 7e08f06 | 2004-10-20 10:37:00 +0000 | [diff] [blame] | 108 | preceding conditional JMP succeeds), underestimating the machine instruction |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 109 | count. |
| 110 | |
| 111 | <code a> |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 112 | call add_one_guest_instr() |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 113 | INCEIP ... |
| 114 | |
| 115 | <code b> |
| 116 | Jcc ... |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 117 | call add_one_guest_instr() |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 118 | JMP ... |
| 119 | |
| 120 | Instead we add a call before each INCEIP, and also one at the start of the |
| 121 | block, but not one at the end, viz: |
| 122 | |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 123 | call add_one_guest_instr() |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 124 | |
| 125 | <code a> |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 126 | call add_one_guest_instr() |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 127 | INCEIP ... |
| 128 | |
| 129 | <code b> |
| 130 | Jcc ... |
| 131 | JMP ... |
| 132 | |
| 133 | Which gives us the right answer. And just to avoid two C calls, we fold |
| 134 | the basic-block-beginning call in with add_one_BB(). Phew. |
| 135 | */ |
sewardj | d54babf | 2005-03-21 00:55:49 +0000 | [diff] [blame] | 136 | IRBB* TL_(instrument)(IRBB* bb_in, VexGuestLayout* layout, |
| 137 | IRType gWordTy, IRType hWordTy ) |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 138 | { |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 139 | IRDirty* di; |
| 140 | Int i; |
sewardj | d54babf | 2005-03-21 00:55:49 +0000 | [diff] [blame] | 141 | IRBB* bb; |
| 142 | |
| 143 | if (gWordTy != hWordTy) { |
| 144 | /* We don't currently support this case. */ |
| 145 | VG_(tool_panic)("host/guest word size mismatch"); |
| 146 | } |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 147 | |
| 148 | /* Set up BB */ |
sewardj | d54babf | 2005-03-21 00:55:49 +0000 | [diff] [blame] | 149 | bb = emptyIRBB(); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 150 | bb->tyenv = dopyIRTypeEnv(bb_in->tyenv); |
| 151 | bb->next = dopyIRExpr(bb_in->next); |
| 152 | bb->jumpkind = bb_in->jumpkind; |
| 153 | |
| 154 | #if 0 |
| 155 | /* We need to know the entry point for this bb to do this. In any |
| 156 | case it's pretty meaningless in the presence of bb chasing since |
| 157 | we may enter this function part way through an IRBB. */ |
| 158 | /* Count call to dlrr(), if this BB is dlrr()'s entry point */ |
| 159 | if (VG_(get_fnname_if_entry)(orig_addr, fnname, 100) && |
| 160 | 0 == VG_(strcmp)(fnname, "_dl_runtime_resolve")) |
| 161 | { |
| 162 | addStmtToIRBB( |
| 163 | bb, |
| 164 | IRStmt_Dirty( |
| 165 | unsafeIRDirty_0_N( "add_one_dlrr_call", mkIRExprVec_0() ) |
| 166 | )); |
| 167 | } |
| 168 | #endif |
| 169 | |
| 170 | /* Count this basic block */ |
| 171 | di = unsafeIRDirty_0_N( 0, "add_one_BB", &add_one_BB, mkIRExprVec_0() ); |
| 172 | addStmtToIRBB( bb, IRStmt_Dirty(di) ); |
| 173 | |
| 174 | for (i = 0; i < bb_in->stmts_used; i++) { |
| 175 | IRStmt* st = bb_in->stmts[i]; |
| 176 | if (!st) continue; |
| 177 | |
| 178 | switch (st->tag) { |
| 179 | case Ist_Exit: |
| 180 | /* Count Jcc */ |
| 181 | addStmtToIRBB( |
| 182 | bb, |
| 183 | IRStmt_Dirty( |
| 184 | unsafeIRDirty_0_N( 0, "add_one_Jcc", &add_one_Jcc, |
| 185 | mkIRExprVec_0() ) |
| 186 | )); |
| 187 | addStmtToIRBB( bb, dopyIRStmt(st) ); |
| 188 | /* Count non-taken Jcc */ |
| 189 | addStmtToIRBB( |
| 190 | bb, |
| 191 | IRStmt_Dirty( |
| 192 | unsafeIRDirty_0_N( 0, "add_one_Jcc_untaken", &add_one_Jcc_untaken, |
| 193 | mkIRExprVec_0() ) |
| 194 | )); |
| 195 | break; |
| 196 | |
| 197 | default: |
sewardj | bd598e1 | 2005-01-07 12:10:21 +0000 | [diff] [blame] | 198 | addStmtToIRBB( bb, st ); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 199 | } |
| 200 | } |
| 201 | |
| 202 | return bb; |
| 203 | |
| 204 | |
| 205 | #if 0 |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 206 | UCodeBlock* cb; |
| 207 | Int i; |
| 208 | UInstr* u; |
| 209 | Char fnname[100]; |
| 210 | |
njn | 810086f | 2002-11-14 12:42:47 +0000 | [diff] [blame] | 211 | cb = VG_(setup_UCodeBlock)(cb_in); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 212 | |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 213 | /* Count basic block */ |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 214 | VG_(call_helper_0_0)(cb, (Addr) & add_one_BB); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 215 | |
njn | 810086f | 2002-11-14 12:42:47 +0000 | [diff] [blame] | 216 | for (i = 0; i < VG_(get_num_instrs)(cb_in); i++) { |
| 217 | u = VG_(get_instr)(cb_in, i); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 218 | |
| 219 | switch (u->opcode) { |
sewardj | 7a5ebcf | 2002-11-13 22:42:13 +0000 | [diff] [blame] | 220 | case NOP: case LOCK: case CALLM_S: case CALLM_E: |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 221 | break; |
| 222 | |
| 223 | case INCEIP: |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 224 | /* Count x86 instr */ |
| 225 | VG_(call_helper_0_0)(cb, (Addr) & add_one_x86_instr); |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 226 | VG_(copy_UInstr)(cb, u); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 227 | break; |
| 228 | |
| 229 | case JMP: |
| 230 | if (u->cond != CondAlways) { |
| 231 | /* Count Jcc */ |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 232 | VG_(call_helper_0_0)(cb, (Addr) & add_one_Jcc); |
| 233 | VG_(copy_UInstr)(cb, u); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 234 | /* Count non-taken Jcc */ |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 235 | VG_(call_helper_0_0)(cb, (Addr) & add_one_Jcc_untaken); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 236 | } else { |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 237 | VG_(copy_UInstr)(cb, u); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 238 | } |
| 239 | break; |
sewardj | 3d7c9c8 | 2003-03-26 21:08:13 +0000 | [diff] [blame] | 240 | |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 241 | default: |
| 242 | /* Count UInstr */ |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 243 | VG_(call_helper_0_0)(cb, (Addr) & add_one_UInstr); |
| 244 | VG_(copy_UInstr)(cb, u); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 245 | break; |
| 246 | } |
| 247 | } |
| 248 | |
njn | 4ba5a79 | 2002-09-30 10:23:54 +0000 | [diff] [blame] | 249 | VG_(free_UCodeBlock)(cb_in); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 250 | return cb; |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 251 | #endif |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 252 | } |
| 253 | |
njn | 26f0251 | 2004-11-22 18:33:15 +0000 | [diff] [blame] | 254 | void TL_(fini)(Int exitcode) |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 255 | { |
| 256 | VG_(message)(Vg_UserMsg, |
| 257 | "Counted %d calls to _dl_runtime_resolve()", n_dlrr_calls); |
| 258 | |
| 259 | VG_(message)(Vg_UserMsg, ""); |
| 260 | VG_(message)(Vg_UserMsg, "Executed:"); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 261 | VG_(message)(Vg_UserMsg, " BBs: %u", n_BBs); |
| 262 | VG_(message)(Vg_UserMsg, " guest instrs: %u", n_guest_instrs); |
| 263 | VG_(message)(Vg_UserMsg, " UInstrs: %u", n_UInstrs); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 264 | |
| 265 | VG_(message)(Vg_UserMsg, ""); |
| 266 | VG_(message)(Vg_UserMsg, "Jccs:"); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 267 | VG_(message)(Vg_UserMsg, " total: %u", n_Jccs); |
| 268 | VG_(message)(Vg_UserMsg, " %% taken: %u%%", |
| 269 | (n_Jccs - n_Jccs_untaken)*100 / |
| 270 | (n_Jccs ? n_Jccs : 1)); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 271 | |
| 272 | VG_(message)(Vg_UserMsg, ""); |
| 273 | VG_(message)(Vg_UserMsg, "Ratios:"); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 274 | VG_(message)(Vg_UserMsg, " guest instrs : BB = %3u : 10", |
| 275 | 10 * n_guest_instrs / n_BBs); |
| 276 | VG_(message)(Vg_UserMsg, " UInstrs : BB = %3u : 10", |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 277 | 10 * n_UInstrs / n_BBs); |
sewardj | 9f649aa | 2004-11-22 20:38:40 +0000 | [diff] [blame] | 278 | VG_(message)(Vg_UserMsg, " UInstrs : x86_instr = %3u : 10", |
| 279 | 10 * n_UInstrs / n_guest_instrs); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 280 | |
njn | 7d9f94d | 2003-04-22 21:41:40 +0000 | [diff] [blame] | 281 | VG_(message)(Vg_UserMsg, ""); |
| 282 | VG_(message)(Vg_UserMsg, "Exit code: %d", exitcode); |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 283 | } |
| 284 | |
njn | 26f0251 | 2004-11-22 18:33:15 +0000 | [diff] [blame] | 285 | VG_DETERMINE_INTERFACE_VERSION(TL_(pre_clo_init), 0) |
fitzhardinge | 98abfc7 | 2003-12-16 02:05:15 +0000 | [diff] [blame] | 286 | |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 287 | /*--------------------------------------------------------------------*/ |
njn25 | cac76cb | 2002-09-23 11:21:57 +0000 | [diff] [blame] | 288 | /*--- end lk_main.c ---*/ |
njn25 | e49d8e7 | 2002-09-23 09:36:25 +0000 | [diff] [blame] | 289 | /*--------------------------------------------------------------------*/ |
| 290 | |