|*
|* The LLVM Compiler Infrastructure
|*
-|* This file was developed by the LLVM research group and is distributed under
-|* the University of Illinois Open Source License. See LICENSE.TXT for details.
-|*
+|* 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 int SavedArgc = 0;
-static const char **SavedArgv = 0;
+static const char *OutputFilename = "llvmprof.out";
/* save_arguments - Save argc and argv as passed into the program for the file
* we output.
*/
-void save_arguments(int argc, const char **argv) {
- if (SavedArgv) return; /* This can be called multiple times */
+int save_arguments(int argc, const char **argv) {
+ unsigned Length, i;
+ if (SavedArgs || !argv) return argc; /* This can be called multiple times */
- /* FIXME: this should copy the arguments out of argv into a string of our own,
- * because the program might modify the arguments!
+ /* 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.
*/
- SavedArgc = argc;
- SavedArgv = argv;
-}
+ 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;
+
+ /* Defensively check for a zero length, even though this is unlikely
+ * to happen in practice. This avoids calling malloc() below with a
+ * size of 0.
+ */
+ if (Length == 0) {
+ SavedArgs = 0;
+ SavedArgsLength = 0;
+ return argc;
+ }
+
+ 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.
+/*
+ * Retrieves the file descriptor for the profile file.
*/
-void write_profiling_data(enum ProfilingType PT, unsigned *Start,
- unsigned NumElements) {
+int getOutFile() {
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 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) {
- off_t Offset;
- OutFile = open("llvmprof.out", O_CREAT | O_WRONLY | O_APPEND, 0666);
+ OutFile = open(OutputFilename, O_CREAT | O_WRONLY, 0666);
+ lseek(OutFile, 0, SEEK_END); /* O_APPEND prevents seeking */
if (OutFile == -1) {
- perror("LLVM profiling: while opening 'llvmprof.out'");
- return;
+ fprintf(stderr, "LLVM profiling runtime: while opening '%s': ",
+ OutputFilename);
+ perror("");
+ return(OutFile);
}
/* Output the command line arguments to the file. */
{
- const char *Args = "";
- int PTy = Arguments;
- int ArgLength = strlen(Args);
+ int PTy = ArgumentInfo;
int Zeros = 0;
- write(OutFile, &PTy, sizeof(int));
- write(OutFile, &ArgLength, sizeof(int));
- write(OutFile, Args, ArgLength);
+ 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 (ArgLength & 3)
- write(OutFile, &Zeros, 4-(ArgLength&3));
+ 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;
- write(OutFile, &PTy, sizeof(int));
- write(OutFile, Start, NumElements*sizeof(unsigned));
+ 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);
+ }
}