120f6301ed
Fix a crash when gathering value profile data on i386 Darwin. The Darwin linker shrinks sections containing aligned structures when padding is not explicitly added to the end of the structure. When iterating over these structures, be sure to not walk past the end of the section. No tests added, since running `ninja check-profile` on i386 Darwin is enough to reproduce the original crash. llvm-svn: 261683
70 lines
2.1 KiB
C
70 lines
2.1 KiB
C
/*===- InstrProfiling.c - Support library for PGO instrumentation ---------===*\
|
|
|*
|
|
|* The LLVM Compiler Infrastructure
|
|
|*
|
|
|* This file is distributed under the University of Illinois Open Source
|
|
|* License. See LICENSE.TXT for details.
|
|
|*
|
|
\*===----------------------------------------------------------------------===*/
|
|
|
|
#include "InstrProfiling.h"
|
|
#include "InstrProfilingInternal.h"
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#define INSTR_PROF_VALUE_PROF_DATA
|
|
#include "InstrProfData.inc"
|
|
|
|
char *(*GetEnvHook)(const char *) = 0;
|
|
|
|
COMPILER_RT_WEAK uint64_t __llvm_profile_raw_version = INSTR_PROF_RAW_VERSION;
|
|
|
|
COMPILER_RT_VISIBILITY uint64_t __llvm_profile_get_magic(void) {
|
|
return sizeof(void *) == sizeof(uint64_t) ? (INSTR_PROF_RAW_MAGIC_64)
|
|
: (INSTR_PROF_RAW_MAGIC_32);
|
|
}
|
|
|
|
/* Return the number of bytes needed to add to SizeInBytes to make it
|
|
* the result a multiple of 8.
|
|
*/
|
|
COMPILER_RT_VISIBILITY uint8_t
|
|
__llvm_profile_get_num_padding_bytes(uint64_t SizeInBytes) {
|
|
return 7 & (sizeof(uint64_t) - SizeInBytes % sizeof(uint64_t));
|
|
}
|
|
|
|
COMPILER_RT_VISIBILITY uint64_t __llvm_profile_get_version(void) {
|
|
return __llvm_profile_raw_version;
|
|
}
|
|
|
|
COMPILER_RT_VISIBILITY void __llvm_profile_reset_counters(void) {
|
|
uint64_t *I = __llvm_profile_begin_counters();
|
|
uint64_t *E = __llvm_profile_end_counters();
|
|
|
|
memset(I, 0, sizeof(uint64_t) * (E - I));
|
|
|
|
const __llvm_profile_data *DataBegin = __llvm_profile_begin_data();
|
|
const __llvm_profile_data *DataEnd = __llvm_profile_end_data();
|
|
const __llvm_profile_data *DI;
|
|
for (DI = DataBegin; DI < DataEnd; ++DI) {
|
|
uint64_t CurrentVSiteCount = 0;
|
|
uint32_t VKI, i;
|
|
if (!DI->Values)
|
|
continue;
|
|
|
|
ValueProfNode **ValueCounters = (ValueProfNode **)DI->Values;
|
|
|
|
for (VKI = IPVK_First; VKI <= IPVK_Last; ++VKI)
|
|
CurrentVSiteCount += DI->NumValueSites[VKI];
|
|
|
|
for (i = 0; i < CurrentVSiteCount; ++i) {
|
|
ValueProfNode *CurrentVNode = ValueCounters[i];
|
|
|
|
while (CurrentVNode) {
|
|
CurrentVNode->VData.Count = 0;
|
|
CurrentVNode = CurrentVNode->Next;
|
|
}
|
|
}
|
|
}
|
|
}
|