InstrProfilingFile.c [plain text]
#include "InstrProfiling.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/errno.h>
#define UNCONST(ptr) ((void *)(uintptr_t)(ptr))
static int writeFile(FILE *File) {
const __llvm_profile_data *DataBegin = __llvm_profile_begin_data();
const __llvm_profile_data *DataEnd = __llvm_profile_end_data();
const uint64_t *CountersBegin = __llvm_profile_begin_counters();
const uint64_t *CountersEnd = __llvm_profile_end_counters();
const char *NamesBegin = __llvm_profile_begin_names();
const char *NamesEnd = __llvm_profile_end_names();
const uint64_t DataSize = DataEnd - DataBegin;
const uint64_t CountersSize = CountersEnd - CountersBegin;
const uint64_t NamesSize = NamesEnd - NamesBegin;
const uint64_t Padding = sizeof(uint64_t) - NamesSize % sizeof(uint64_t);
const char Zeroes[sizeof(uint64_t)] = {0};
uint64_t Header[PROFILE_HEADER_SIZE];
Header[0] = __llvm_profile_get_magic();
Header[1] = __llvm_profile_get_version();
Header[2] = DataSize;
Header[3] = CountersSize;
Header[4] = NamesSize;
Header[5] = (uintptr_t)CountersBegin;
Header[6] = (uintptr_t)NamesBegin;
#define CHECK_fwrite(Data, Size, Length, File) \
do { if (fwrite(Data, Size, Length, File) != Length) return -1; } while (0)
CHECK_fwrite(Header, sizeof(uint64_t), PROFILE_HEADER_SIZE, File);
CHECK_fwrite(DataBegin, sizeof(__llvm_profile_data), DataSize, File);
CHECK_fwrite(CountersBegin, sizeof(uint64_t), CountersSize, File);
CHECK_fwrite(NamesBegin, sizeof(char), NamesSize, File);
CHECK_fwrite(Zeroes, sizeof(char), Padding, File);
#undef CHECK_fwrite
return 0;
}
static int writeFileWithName(const char *OutputName) {
int RetVal;
FILE *OutputFile;
if (!OutputName || !OutputName[0])
return -1;
OutputFile = fopen(OutputName, "a");
if (!OutputFile)
return -1;
RetVal = writeFile(OutputFile);
fclose(OutputFile);
return RetVal;
}
__attribute__((weak)) int __llvm_profile_OwnsFilename = 0;
__attribute__((weak)) const char *__llvm_profile_CurrentFilename = NULL;
static void setFilename(const char *Filename, int OwnsFilename) {
if (__llvm_profile_OwnsFilename)
free(UNCONST(__llvm_profile_CurrentFilename));
__llvm_profile_CurrentFilename = Filename;
__llvm_profile_OwnsFilename = OwnsFilename;
}
static void truncateCurrentFile(void) {
const char *Filename = __llvm_profile_CurrentFilename;
if (!Filename || !Filename[0])
return;
FILE *File = fopen(Filename, "w");
if (!File)
return;
fclose(File);
}
static void setDefaultFilename(void) { setFilename("default.profraw", 0); }
int getpid(void);
static int setFilenameFromEnvironment(void) {
const char *Filename = getenv("LLVM_PROFILE_FILE");
if (!Filename || !Filename[0])
return -1;
#define MAX_PID_SIZE 16
char PidChars[MAX_PID_SIZE] = {0};
int NumPids = 0;
int PidLength = 0;
int I;
for (I = 0; Filename[I]; ++I)
if (Filename[I] == '%' && Filename[++I] == 'p')
if (!NumPids++) {
PidLength = snprintf(PidChars, MAX_PID_SIZE, "%d", getpid());
if (PidLength <= 0)
return -1;
}
if (!NumPids) {
setFilename(Filename, 0);
return 0;
}
char *Allocated = (char*)malloc(I + NumPids*(PidLength - 2) + 1);
if (!Allocated)
return -1;
int J;
for (I = 0, J = 0; Filename[I]; ++I)
if (Filename[I] == '%') {
if (Filename[++I] == 'p') {
memcpy(Allocated + J, PidChars, PidLength);
J += PidLength;
}
} else
Allocated[J++] = Filename[I];
Allocated[J] = 0;
setFilename(Allocated, 1);
return 0;
}
static void setFilenameAutomatically(void) {
if (!setFilenameFromEnvironment())
return;
setDefaultFilename();
}
__attribute__((visibility("hidden")))
void __llvm_profile_initialize_file(void) {
if (__llvm_profile_CurrentFilename)
return;
setFilenameAutomatically();
truncateCurrentFile();
}
__attribute__((visibility("hidden")))
void __llvm_profile_set_filename(const char *Filename) {
setFilename(Filename, 0);
truncateCurrentFile();
}
__attribute__((visibility("hidden")))
int __llvm_profile_write_file(void) {
if (!__llvm_profile_CurrentFilename)
return -1;
int rc = writeFileWithName(__llvm_profile_CurrentFilename);
if (rc && getenv("LLVM_PROFILE_VERBOSE_ERRORS"))
fprintf(stderr, "LLVM Profile: Failed to write file \"%s\": %s\n",
__llvm_profile_CurrentFilename, strerror(errno));
return rc;
}
static void writeFileWithoutReturn(void) {
__llvm_profile_write_file();
}
__attribute__((visibility("hidden")))
int __llvm_profile_register_write_file_atexit(void) {
static int HasBeenRegistered = 0;
if (HasBeenRegistered)
return 0;
HasBeenRegistered = 1;
return atexit(writeFileWithoutReturn);
}