runtime/libprofile/BlockProfiling.c

   1 /*===-- BlockProfiling.c - Support library for block profiling ------------===*\
   2 |*
   3 |*                     The LLVM Compiler Infrastructure
   4 |*
   5 |* This file was developed by the LLVM research group and is distributed under
   6 |* the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 |*
   8 |*===----------------------------------------------------------------------===*|
   9 |*
  10 |* This file implements the call back routines for the block profiling
  11 |* instrumentation pass.  This should be used with the -insert-block-profiling
  12 |* LLVM pass.
  13 |*
  14 \*===----------------------------------------------------------------------===*/
  15
  16 #include "Profiling.h"
  17 #include <stdlib.h>
  18
  19 static unsigned *ArrayStart;
  20 static unsigned NumElements;
  21
  22 /* BlockProfAtExitHandler - When the program exits, just write out the profiling
  23  * data.
  24  */
  25 static void BlockProfAtExitHandler() {
  26   /* Note that if this were doing something more intellegent with the
  27      instrumentation, that we could do some computation here to expand what we
  28      collected into simple block profiles.  Since we directly count each block,
  29   */
  30   write_profiling_data(Block, ArrayStart, NumElements);
  31 }
  32
  33
  34 /* llvm_start_block_profiling - This is the main entry point of the block
  35  * profiling library.  It is responsible for setting up the atexit handler.
  36  */
  37 int llvm_start_block_profiling(int argc, const char **argv,
  38                                unsigned *arrayStart, unsigned numElements) {
  39   int Ret = save_arguments(argc, argv);
  40   ArrayStart = arrayStart;
  41   NumElements = numElements;
  42   atexit(BlockProfAtExitHandler);
  43   return Ret;
  44 }