InstrProfilingWriter.c revision 303975
1234353Sdim/*===- InstrProfilingWriter.c - Write instrumentation to a file or buffer -===*\ 2198090Srdivacky|* 3198090Srdivacky|* The LLVM Compiler Infrastructure 4198090Srdivacky|* 5198090Srdivacky|* This file is distributed under the University of Illinois Open Source 6198090Srdivacky|* License. See LICENSE.TXT for details. 7198090Srdivacky|* 8198090Srdivacky\*===----------------------------------------------------------------------===*/ 9198090Srdivacky 10198090Srdivacky#include "InstrProfiling.h" 11199511Srdivacky#include "InstrProfilingInternal.h" 12198090Srdivacky#include <string.h> 13198090Srdivacky 14207618Srdivacky#define INSTR_PROF_VALUE_PROF_DATA 15198090Srdivacky#include "InstrProfData.inc" 16243830Sdimvoid (*FreeHook)(void *) = NULL; 17198090Srdivackyvoid* (*CallocHook)(size_t, size_t) = NULL; 18198090Srdivackyuint32_t VPBufferSize = 0; 19218893Sdim 20198090Srdivacky/* The buffer writer is reponsponsible in keeping writer state 21198090Srdivacky * across the call. 22198090Srdivacky */ 23198090SrdivackyCOMPILER_RT_VISIBILITY uint32_t llvmBufferWriter(ProfDataIOVec *IOVecs, 24198090Srdivacky uint32_t NumIOVecs, 25198090Srdivacky void **WriterCtx) { 26198090Srdivacky uint32_t I; 27198090Srdivacky char **Buffer = (char **)WriterCtx; 28198090Srdivacky for (I = 0; I < NumIOVecs; I++) { 29243830Sdim size_t Length = IOVecs[I].ElmSize * IOVecs[I].NumElm; 30218893Sdim memcpy(*Buffer, IOVecs[I].Data, Length); 31198090Srdivacky *Buffer += Length; 32218893Sdim } 33218893Sdim return 0; 34207618Srdivacky} 35218893Sdim 36198090Srdivackystatic void llvmInitBufferIO(ProfBufferIO *BufferIO, WriterCallback FileWriter, 37218893Sdim void *File, uint8_t *Buffer, uint32_t BufferSz) { 38218893Sdim BufferIO->File = File; 39207618Srdivacky BufferIO->FileWriter = FileWriter; 40218893Sdim BufferIO->BufferStart = Buffer; 41198090Srdivacky BufferIO->BufferSz = BufferSz; 42198090Srdivacky BufferIO->CurOffset = 0; 43218893Sdim} 44218893Sdim 45198090SrdivackyCOMPILER_RT_VISIBILITY ProfBufferIO * 46198090SrdivackyllvmCreateBufferIO(WriterCallback FileWriter, void *File, uint32_t BufferSz) { 47198090Srdivacky ProfBufferIO *BufferIO = (ProfBufferIO *)CallocHook(1, sizeof(ProfBufferIO)); 48198090Srdivacky uint8_t *Buffer = (uint8_t *)CallocHook(1, BufferSz); 49198090Srdivacky if (!Buffer) { 50198090Srdivacky FreeHook(BufferIO); 51198090Srdivacky return 0; 52198090Srdivacky } 53198090Srdivacky llvmInitBufferIO(BufferIO, FileWriter, File, Buffer, BufferSz); 54218893Sdim return BufferIO; 55198090Srdivacky} 56198090Srdivacky 57198090SrdivackyCOMPILER_RT_VISIBILITY void llvmDeleteBufferIO(ProfBufferIO *BufferIO) { 58198090Srdivacky FreeHook(BufferIO->BufferStart); 59198090Srdivacky FreeHook(BufferIO); 60198090Srdivacky} 61198090Srdivacky 62198090SrdivackyCOMPILER_RT_VISIBILITY int 63199511SrdivackyllvmBufferIOWrite(ProfBufferIO *BufferIO, const uint8_t *Data, uint32_t Size) { 64199511Srdivacky /* Buffer is not large enough, it is time to flush. */ 65243830Sdim if (Size + BufferIO->CurOffset > BufferIO->BufferSz) { 66243830Sdim if (llvmBufferIOFlush(BufferIO) != 0) 67243830Sdim return -1; 68199511Srdivacky } 69199511Srdivacky /* Special case, bypass the buffer completely. */ 70198090Srdivacky ProfDataIOVec IO[] = {{Data, sizeof(uint8_t), Size}}; 71198090Srdivacky if (Size > BufferIO->BufferSz) { 72198090Srdivacky if (BufferIO->FileWriter(IO, 1, &BufferIO->File)) 73198090Srdivacky return -1; 74218893Sdim } else { 75198090Srdivacky /* Write the data to buffer */ 76198090Srdivacky uint8_t *Buffer = BufferIO->BufferStart + BufferIO->CurOffset; 77198090Srdivacky llvmBufferWriter(IO, 1, (void **)&Buffer); 78218893Sdim BufferIO->CurOffset = Buffer - BufferIO->BufferStart; 79198090Srdivacky } 80198090Srdivacky return 0; 81198090Srdivacky} 82198090Srdivacky 83198090SrdivackyCOMPILER_RT_VISIBILITY int llvmBufferIOFlush(ProfBufferIO *BufferIO) { 84226633Sdim if (BufferIO->CurOffset) { 85243830Sdim ProfDataIOVec IO[] = { 86198090Srdivacky {BufferIO->BufferStart, sizeof(uint8_t), BufferIO->CurOffset}}; 87198090Srdivacky if (BufferIO->FileWriter(IO, 1, &BufferIO->File)) 88198090Srdivacky return -1; 89198090Srdivacky BufferIO->CurOffset = 0; 90198090Srdivacky } 91198090Srdivacky return 0; 92198090Srdivacky} 93198090Srdivacky 94198090SrdivackyCOMPILER_RT_VISIBILITY int llvmWriteProfData(WriterCallback Writer, 95218893Sdim void *WriterCtx, 96198090Srdivacky ValueProfData **ValueDataArray, 97198090Srdivacky const uint64_t ValueDataSize) { 98198090Srdivacky /* Match logic in __llvm_profile_write_buffer(). */ 99198090Srdivacky const __llvm_profile_data *DataBegin = __llvm_profile_begin_data(); 100198090Srdivacky const __llvm_profile_data *DataEnd = __llvm_profile_end_data(); 101218893Sdim const uint64_t *CountersBegin = __llvm_profile_begin_counters(); 102198090Srdivacky const uint64_t *CountersEnd = __llvm_profile_end_counters(); 103198090Srdivacky const char *NamesBegin = __llvm_profile_begin_names(); 104198090Srdivacky const char *NamesEnd = __llvm_profile_end_names(); 105 return llvmWriteProfDataImpl(Writer, WriterCtx, DataBegin, DataEnd, 106 CountersBegin, CountersEnd, ValueDataArray, 107 ValueDataSize, NamesBegin, NamesEnd); 108} 109 110#define VP_BUFFER_SIZE 8 * 1024 111static int writeValueProfData(WriterCallback Writer, void *WriterCtx, 112 ValueProfData **ValueDataBegin, 113 uint64_t NumVData) { 114 ProfBufferIO *BufferIO; 115 uint32_t I = 0, BufferSz; 116 117 if (!ValueDataBegin) 118 return 0; 119 120 BufferSz = VPBufferSize ? VPBufferSize : VP_BUFFER_SIZE; 121 BufferIO = llvmCreateBufferIO(Writer, WriterCtx, BufferSz); 122 123 for (I = 0; I < NumVData; I++) { 124 ValueProfData *CurVData = ValueDataBegin[I]; 125 if (!CurVData) 126 continue; 127 if (llvmBufferIOWrite(BufferIO, (const uint8_t *)CurVData, 128 CurVData->TotalSize) != 0) 129 return -1; 130 } 131 132 if (llvmBufferIOFlush(BufferIO) != 0) 133 return -1; 134 llvmDeleteBufferIO(BufferIO); 135 136 return 0; 137} 138 139COMPILER_RT_VISIBILITY int llvmWriteProfDataImpl( 140 WriterCallback Writer, void *WriterCtx, 141 const __llvm_profile_data *DataBegin, const __llvm_profile_data *DataEnd, 142 const uint64_t *CountersBegin, const uint64_t *CountersEnd, 143 ValueProfData **ValueDataBegin, const uint64_t ValueDataSize, 144 const char *NamesBegin, const char *NamesEnd) { 145 146 /* Calculate size of sections. */ 147 const uint64_t DataSize = DataEnd - DataBegin; 148 const uint64_t CountersSize = CountersEnd - CountersBegin; 149 const uint64_t NamesSize = NamesEnd - NamesBegin; 150 const uint64_t Padding = __llvm_profile_get_num_padding_bytes(NamesSize); 151 152 /* Enough zeroes for padding. */ 153 const char Zeroes[sizeof(uint64_t)] = {0}; 154 155 /* Create the header. */ 156 __llvm_profile_header Header; 157 158 if (!DataSize) 159 return 0; 160 161 /* Initialize header struture. */ 162#define INSTR_PROF_RAW_HEADER(Type, Name, Init) Header.Name = Init; 163#include "InstrProfData.inc" 164 165 /* Write the data. */ 166 ProfDataIOVec IOVec[] = {{&Header, sizeof(__llvm_profile_header), 1}, 167 {DataBegin, sizeof(__llvm_profile_data), DataSize}, 168 {CountersBegin, sizeof(uint64_t), CountersSize}, 169 {NamesBegin, sizeof(uint8_t), NamesSize}, 170 {Zeroes, sizeof(uint8_t), Padding}}; 171 if (Writer(IOVec, sizeof(IOVec) / sizeof(*IOVec), &WriterCtx)) 172 return -1; 173 174 return writeValueProfData(Writer, WriterCtx, ValueDataBegin, DataSize); 175} 176