142 lines
4.2 KiB
C
142 lines
4.2 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 <assert.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#if !defined(_MSC_VER) && !defined(__MINGW32__)
|
|
#include <unistd.h>
|
|
#else
|
|
#include <io.h>
|
|
#endif
|
|
#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((char**)&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((char**)&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;
|
|
}
|
|
|
|
|
|
/*
|
|
* Retrieves the file descriptor for the profile file.
|
|
*/
|
|
int getOutFile() {
|
|
static int OutFile = -1;
|
|
|
|
/* 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, 0666);
|
|
lseek(OutFile, 0, SEEK_END); /* O_APPEND prevents seeking */
|
|
if (OutFile == -1) {
|
|
fprintf(stderr, "LLVM profiling runtime: while opening '%s': ",
|
|
OutputFilename);
|
|
perror("");
|
|
return(OutFile);
|
|
}
|
|
|
|
/* Output the command line arguments to the file. */
|
|
{
|
|
int PTy = ArgumentInfo;
|
|
int Zeros = 0;
|
|
if (write(OutFile, &PTy, sizeof(int)) < 0 ||
|
|
write(OutFile, &SavedArgsLength, sizeof(unsigned)) < 0 ||
|
|
write(OutFile, SavedArgs, SavedArgsLength) < 0 ) {
|
|
fprintf(stderr,"error: unable to write to output file.");
|
|
exit(0);
|
|
}
|
|
/* Pad out to a multiple of four bytes */
|
|
if (SavedArgsLength & 3) {
|
|
if (write(OutFile, &Zeros, 4-(SavedArgsLength&3)) < 0) {
|
|
fprintf(stderr,"error: unable to write to output file.");
|
|
exit(0);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return(OutFile);
|
|
}
|
|
|
|
/* 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) {
|
|
int PTy;
|
|
int outFile = getOutFile();
|
|
|
|
/* Write out this record! */
|
|
PTy = PT;
|
|
if( write(outFile, &PTy, sizeof(int)) < 0 ||
|
|
write(outFile, &NumElements, sizeof(unsigned)) < 0 ||
|
|
write(outFile, Start, NumElements*sizeof(unsigned)) < 0 ) {
|
|
fprintf(stderr,"error: unable to write to output file.");
|
|
exit(0);
|
|
}
|
|
}
|