forked from OSchip/llvm-project
InstrProf: If libc is available, use it; no functionality change
It was misguided to plan to rely on __llvm_profile_write_buffer() in __llvm_profile_write_file(). It's less complex to duplicate the writing logic than to mmap the file. Since it's here to stay, move `FILE*`-based writing logic into InstrProfilingFile.c. <rdar://problem/15943240> llvm-svn: 204498
This commit is contained in:
parent
be0a5e176b
commit
cf4bb960bd
|
@ -10,8 +10,6 @@
|
|||
#ifndef PROFILE_INSTRPROFILING_H__
|
||||
#define PROFILE_INSTRPROFILING_H__
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#define I386_FREEBSD (defined(__FreeBSD__) && defined(__i386__))
|
||||
|
||||
#if !I386_FREEBSD
|
||||
|
@ -42,16 +40,18 @@ typedef struct __llvm_profile_data {
|
|||
uint64_t *const Counters;
|
||||
} __llvm_profile_data;
|
||||
|
||||
/* TODO: void __llvm_profile_get_size_for_buffer(void); */
|
||||
/*!
|
||||
* \brief Get required size for profile buffer.
|
||||
*/
|
||||
uint64_t __llvm_profile_get_size_for_buffer(void);
|
||||
|
||||
/*!
|
||||
* \brief Write instrumentation data to the given buffer.
|
||||
*
|
||||
* This function is currently broken: it shouldn't rely on libc, but it does.
|
||||
* It should be changed to take a char* buffer, and write binary data directly
|
||||
* to it.
|
||||
* \pre \c Buffer is the start of a buffer at least as big as \a
|
||||
* __llvm_profile_get_size_for_buffer().
|
||||
*/
|
||||
int __llvm_profile_write_buffer(FILE *OutputFile);
|
||||
int __llvm_profile_write_buffer(char *Buffer);
|
||||
|
||||
const __llvm_profile_data *__llvm_profile_data_begin(void);
|
||||
const __llvm_profile_data *__llvm_profile_data_end(void);
|
||||
|
|
|
@ -11,44 +11,4 @@
|
|||
#include <string.h>
|
||||
|
||||
/* TODO: uint64_t __llvm_profile_get_size_for_buffer(void) */
|
||||
|
||||
int __llvm_profile_write_buffer(FILE *OutputFile) {
|
||||
const __llvm_profile_data *DataBegin = __llvm_profile_data_begin();
|
||||
const __llvm_profile_data *DataEnd = __llvm_profile_data_end();
|
||||
const uint64_t *CountersBegin = __llvm_profile_counters_begin();
|
||||
const uint64_t *CountersEnd = __llvm_profile_counters_end();
|
||||
const char *NamesBegin = __llvm_profile_names_begin();
|
||||
const char *NamesEnd = __llvm_profile_names_end();
|
||||
|
||||
/* Calculate size of sections. */
|
||||
const uint64_t DataSize = DataEnd - DataBegin;
|
||||
const uint64_t CountersSize = CountersEnd - CountersBegin;
|
||||
const uint64_t NamesSize = NamesEnd - NamesBegin;
|
||||
|
||||
/* Get rest of header data. */
|
||||
const uint64_t Magic = __llvm_profile_get_magic();
|
||||
const uint64_t Version = __llvm_profile_get_version();
|
||||
const uint64_t CountersDelta = (uint64_t)CountersBegin;
|
||||
const uint64_t NamesDelta = (uint64_t)NamesBegin;
|
||||
|
||||
#define CHECK_fwrite(Data, Size, Length, File) \
|
||||
do { if (fwrite(Data, Size, Length, File) != Length) return -1; } while (0)
|
||||
|
||||
/* Write the header. */
|
||||
CHECK_fwrite(&Magic, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&Version, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&DataSize, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&CountersSize, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&NamesSize, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&CountersDelta, sizeof(uint64_t), 1, OutputFile);
|
||||
CHECK_fwrite(&NamesDelta, sizeof(uint64_t), 1, OutputFile);
|
||||
|
||||
/* Write the data. */
|
||||
CHECK_fwrite(DataBegin, sizeof(__llvm_profile_data), DataSize, OutputFile);
|
||||
CHECK_fwrite(CountersBegin, sizeof(uint64_t), CountersSize, OutputFile);
|
||||
CHECK_fwrite(NamesBegin, sizeof(char), NamesSize, OutputFile);
|
||||
|
||||
#undef CHECK_fwrite
|
||||
|
||||
return 0;
|
||||
}
|
||||
/* TODO: int __llvm_profile_write_buffer(char *Buffer) */
|
||||
|
|
|
@ -12,6 +12,47 @@
|
|||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
static int writeFile(FILE *File) {
|
||||
const __llvm_profile_data *DataBegin = __llvm_profile_data_begin();
|
||||
const __llvm_profile_data *DataEnd = __llvm_profile_data_end();
|
||||
const uint64_t *CountersBegin = __llvm_profile_counters_begin();
|
||||
const uint64_t *CountersEnd = __llvm_profile_counters_end();
|
||||
const char *NamesBegin = __llvm_profile_names_begin();
|
||||
const char *NamesEnd = __llvm_profile_names_end();
|
||||
|
||||
/* Calculate size of sections. */
|
||||
const uint64_t DataSize = DataEnd - DataBegin;
|
||||
const uint64_t CountersSize = CountersEnd - CountersBegin;
|
||||
const uint64_t NamesSize = NamesEnd - NamesBegin;
|
||||
|
||||
/* Get rest of header data. */
|
||||
const uint64_t Magic = __llvm_profile_get_magic();
|
||||
const uint64_t Version = __llvm_profile_get_version();
|
||||
const uint64_t CountersDelta = (uint64_t)CountersBegin;
|
||||
const uint64_t NamesDelta = (uint64_t)NamesBegin;
|
||||
|
||||
#define CHECK_fwrite(Data, Size, Length, File) \
|
||||
do { if (fwrite(Data, Size, Length, File) != Length) return -1; } while (0)
|
||||
|
||||
/* Write the header. */
|
||||
CHECK_fwrite(&Magic, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&Version, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&DataSize, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&CountersSize, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&NamesSize, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&CountersDelta, sizeof(uint64_t), 1, File);
|
||||
CHECK_fwrite(&NamesDelta, sizeof(uint64_t), 1, File);
|
||||
|
||||
/* Write the data. */
|
||||
CHECK_fwrite(DataBegin, sizeof(__llvm_profile_data), DataSize, File);
|
||||
CHECK_fwrite(CountersBegin, sizeof(uint64_t), CountersSize, File);
|
||||
CHECK_fwrite(NamesBegin, sizeof(char), NamesSize, File);
|
||||
|
||||
#undef CHECK_fwrite
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int writeFileWithName(const char *OutputName) {
|
||||
int RetVal;
|
||||
FILE *OutputFile;
|
||||
|
@ -21,7 +62,7 @@ static int writeFileWithName(const char *OutputName) {
|
|||
if (!OutputFile)
|
||||
return -1;
|
||||
|
||||
RetVal = __llvm_profile_write_buffer(OutputFile);
|
||||
RetVal = writeFile(OutputFile);
|
||||
|
||||
fclose(OutputFile);
|
||||
return RetVal;
|
||||
|
|
Loading…
Reference in New Issue