1 /*===- InstrProfilingMerge.c - Profile in-process Merging ---------------===*\
2 |*
3 |* The LLVM Compiler Infrastructure
4 |*
5 |* This file is distributed under the University of Illinois Open Source
6 |* License. See LICENSE.TXT for details.
7 |*
8 |*===----------------------------------------------------------------------===*
9 |* This file defines the API needed for in-process merging of profile data
10 |* stored in memory buffer.
11 \*===---------------------------------------------------------------------===*/
12
13 #include "InstrProfiling.h"
14 #include "InstrProfilingInternal.h"
15 #include "InstrProfilingUtil.h"
16
17 #define INSTR_PROF_VALUE_PROF_DATA
18 #include "InstrProfData.inc"
19
20 COMPILER_RT_WEAK void (*VPMergeHook)(ValueProfData *,
21 __llvm_profile_data *) = NULL;
22 COMPILER_RT_VISIBILITY
lprofGetLoadModuleSignature()23 uint64_t lprofGetLoadModuleSignature() {
24 /* A very fast way to compute a module signature. */
25 uint64_t CounterSize = (uint64_t)(__llvm_profile_end_counters() -
26 __llvm_profile_begin_counters());
27 uint64_t DataSize = __llvm_profile_get_data_size(__llvm_profile_begin_data(),
28 __llvm_profile_end_data());
29 uint64_t NamesSize =
30 (uint64_t)(__llvm_profile_end_names() - __llvm_profile_begin_names());
31 uint64_t NumVnodes =
32 (uint64_t)(__llvm_profile_end_vnodes() - __llvm_profile_begin_vnodes());
33 const __llvm_profile_data *FirstD = __llvm_profile_begin_data();
34
35 return (NamesSize << 40) + (CounterSize << 30) + (DataSize << 20) +
36 (NumVnodes << 10) + (DataSize > 0 ? FirstD->NameRef : 0);
37 }
38
39 /* Returns 1 if profile is not structurally compatible. */
40 COMPILER_RT_VISIBILITY
__llvm_profile_check_compatibility(const char * ProfileData,uint64_t ProfileSize)41 int __llvm_profile_check_compatibility(const char *ProfileData,
42 uint64_t ProfileSize) {
43 /* Check profile header only for now */
44 __llvm_profile_header *Header = (__llvm_profile_header *)ProfileData;
45 __llvm_profile_data *SrcDataStart, *SrcDataEnd, *SrcData, *DstData;
46 SrcDataStart =
47 (__llvm_profile_data *)(ProfileData + sizeof(__llvm_profile_header));
48 SrcDataEnd = SrcDataStart + Header->DataSize;
49
50 if (ProfileSize < sizeof(__llvm_profile_header))
51 return 1;
52
53 /* Check the header first. */
54 if (Header->Magic != __llvm_profile_get_magic() ||
55 Header->Version != __llvm_profile_get_version() ||
56 Header->DataSize !=
57 __llvm_profile_get_data_size(__llvm_profile_begin_data(),
58 __llvm_profile_end_data()) ||
59 Header->CountersSize != (uint64_t)(__llvm_profile_end_counters() -
60 __llvm_profile_begin_counters()) ||
61 Header->NamesSize != (uint64_t)(__llvm_profile_end_names() -
62 __llvm_profile_begin_names()) ||
63 Header->ValueKindLast != IPVK_Last)
64 return 1;
65
66 if (ProfileSize < sizeof(__llvm_profile_header) +
67 Header->DataSize * sizeof(__llvm_profile_data) +
68 Header->NamesSize + Header->CountersSize)
69 return 1;
70
71 for (SrcData = SrcDataStart,
72 DstData = (__llvm_profile_data *)__llvm_profile_begin_data();
73 SrcData < SrcDataEnd; ++SrcData, ++DstData) {
74 if (SrcData->NameRef != DstData->NameRef ||
75 SrcData->FuncHash != DstData->FuncHash ||
76 SrcData->NumCounters != DstData->NumCounters)
77 return 1;
78 }
79
80 /* Matched! */
81 return 0;
82 }
83
84 COMPILER_RT_VISIBILITY
__llvm_profile_merge_from_buffer(const char * ProfileData,uint64_t ProfileSize)85 void __llvm_profile_merge_from_buffer(const char *ProfileData,
86 uint64_t ProfileSize) {
87 __llvm_profile_data *SrcDataStart, *SrcDataEnd, *SrcData, *DstData;
88 __llvm_profile_header *Header = (__llvm_profile_header *)ProfileData;
89 uint64_t *SrcCountersStart;
90 const char *SrcNameStart;
91 ValueProfData *SrcValueProfDataStart, *SrcValueProfData;
92
93 SrcDataStart =
94 (__llvm_profile_data *)(ProfileData + sizeof(__llvm_profile_header));
95 SrcDataEnd = SrcDataStart + Header->DataSize;
96 SrcCountersStart = (uint64_t *)SrcDataEnd;
97 SrcNameStart = (const char *)(SrcCountersStart + Header->CountersSize);
98 SrcValueProfDataStart =
99 (ValueProfData *)(SrcNameStart + Header->NamesSize +
100 __llvm_profile_get_num_padding_bytes(
101 Header->NamesSize));
102
103 for (SrcData = SrcDataStart,
104 DstData = (__llvm_profile_data *)__llvm_profile_begin_data(),
105 SrcValueProfData = SrcValueProfDataStart;
106 SrcData < SrcDataEnd; ++SrcData, ++DstData) {
107 uint64_t *SrcCounters;
108 uint64_t *DstCounters = (uint64_t *)DstData->CounterPtr;
109 unsigned I, NC, NVK = 0;
110
111 NC = SrcData->NumCounters;
112 SrcCounters = SrcCountersStart +
113 ((size_t)SrcData->CounterPtr - Header->CountersDelta) /
114 sizeof(uint64_t);
115 for (I = 0; I < NC; I++)
116 DstCounters[I] += SrcCounters[I];
117
118 /* Now merge value profile data. */
119 if (!VPMergeHook)
120 continue;
121
122 for (I = 0; I <= IPVK_Last; I++)
123 NVK += (SrcData->NumValueSites[I] != 0);
124
125 if (!NVK)
126 continue;
127
128 VPMergeHook(SrcValueProfData, DstData);
129 SrcValueProfData = (ValueProfData *)((char *)SrcValueProfData +
130 SrcValueProfData->TotalSize);
131 }
132 }
133