llvm/runtime/libprofile/CommonProfiling.c
2007-12-29 22:59:10 +00:00

118 lines
3.6 KiB
C

/*===-- CommonProfiling.c - Profiling support library support -------------===*\
|*
|* The LLVM Compiler Infrastructure
|*
|* This file is distributed under the University of Illinois Open Source
|* License. See LICENSE.TXT for details.
|*
|*===----------------------------------------------------------------------===*|
|*
|* This file implements functions used by the various different types of
|* profiling implementations.
|*
\*===----------------------------------------------------------------------===*/
#include "Profiling.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
static char *SavedArgs = 0;
static unsigned SavedArgsLength = 0;
static const char *OutputFilename = "llvmprof.out";
/* save_arguments - Save argc and argv as passed into the program for the file
* we output.
*/
int save_arguments(int argc, const char **argv) {
unsigned Length, i;
if (SavedArgs || !argv) return argc; /* This can be called multiple times */
/* Check to see if there are any arguments passed into the program for the
* profiler. If there are, strip them off and remember their settings.
*/
while (argc > 1 && !strncmp(argv[1], "-llvmprof-", 10)) {
/* Ok, we have an llvmprof argument. Remove it from the arg list and decide
* what to do with it.
*/
const char *Arg = argv[1];
memmove(&argv[1], &argv[2], (argc-1)*sizeof(char*));
--argc;
if (!strcmp(Arg, "-llvmprof-output")) {
if (argc == 1)
puts("-llvmprof-output requires a filename argument!");
else {
OutputFilename = strdup(argv[1]);
memmove(&argv[1], &argv[2], (argc-1)*sizeof(char*));
--argc;
}
} else {
printf("Unknown option to the profiler runtime: '%s' - ignored.\n", Arg);
}
}
for (Length = 0, i = 0; i != (unsigned)argc; ++i)
Length += strlen(argv[i])+1;
SavedArgs = (char*)malloc(Length);
for (Length = 0, i = 0; i != (unsigned)argc; ++i) {
unsigned Len = strlen(argv[i]);
memcpy(SavedArgs+Length, argv[i], Len);
Length += Len;
SavedArgs[Length++] = ' ';
}
SavedArgsLength = Length;
return argc;
}
/* write_profiling_data - Write a raw block of profiling counters out to the
* llvmprof.out file. Note that we allow programs to be instrumented with
* multiple different kinds of instrumentation. For this reason, this function
* may be called more than once.
*/
void write_profiling_data(enum ProfilingType PT, unsigned *Start,
unsigned NumElements) {
static int OutFile = -1;
int PTy;
/* If this is the first time this function is called, open the output file for
* appending, creating it if it does not already exist.
*/
if (OutFile == -1) {
OutFile = open(OutputFilename, O_CREAT | O_WRONLY | O_APPEND, 0666);
if (OutFile == -1) {
fprintf(stderr, "LLVM profiling runtime: while opening '%s': ",
OutputFilename);
perror("");
return;
}
/* Output the command line arguments to the file. */
{
int PTy = ArgumentInfo;
int Zeros = 0;
write(OutFile, &PTy, sizeof(int));
write(OutFile, &SavedArgsLength, sizeof(unsigned));
write(OutFile, SavedArgs, SavedArgsLength);
/* Pad out to a multiple of four bytes */
if (SavedArgsLength & 3)
write(OutFile, &Zeros, 4-(SavedArgsLength&3));
}
}
/* Write out this record! */
PTy = PT;
write(OutFile, &PTy, sizeof(int));
write(OutFile, &NumElements, sizeof(unsigned));
write(OutFile, Start, NumElements*sizeof(unsigned));
}