1/*===- InstrProfilingPlatformWindows.c - Profile data on Windows ----------===*\
2|*
3|* Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4|* See https://llvm.org/LICENSE.txt for license information.
5|* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6|*
7\*===----------------------------------------------------------------------===*/
8
9#include <stddef.h>
10
11#include "InstrProfiling.h"
12#include "InstrProfilingInternal.h"
13
14#if defined(_WIN32)
15
16#if defined(_MSC_VER)
17/* Merge read-write sections into .data. */
18#pragma comment(linker, "/MERGE:.lprfb=.data")
19#pragma comment(linker, "/MERGE:.lprfd=.data")
20#pragma comment(linker, "/MERGE:.lprfv=.data")
21#pragma comment(linker, "/MERGE:.lprfnd=.data")
22/* Do *NOT* merge .lprfn and .lcovmap into .rdata. llvm-cov must be able to find
23 * after the fact.
24 * Do *NOT* merge .lprfc .rdata. When binary profile correlation is enabled,
25 * llvm-cov must be able to find after the fact.
26 */
27
28/* Allocate read-only section bounds. */
29#pragma section(".lprfn$A", read)
30#pragma section(".lprfn$Z", read)
31
32/* Allocate read-write section bounds. */
33#pragma section(".lprfd$A", read, write)
34#pragma section(".lprfd$Z", read, write)
35#pragma section(".lprfc$A", read, write)
36#pragma section(".lprfc$Z", read, write)
37#pragma section(".lprfb$A", read, write)
38#pragma section(".lprfb$Z", read, write)
39#pragma section(".lprfnd$A", read, write)
40#pragma section(".lprfnd$Z", read, write)
41#endif
42
43__llvm_profile_data COMPILER_RT_SECTION(".lprfd$A") DataStart = {0};
44__llvm_profile_data COMPILER_RT_SECTION(".lprfd$Z") DataEnd = {0};
45
46const char COMPILER_RT_SECTION(".lprfn$A") NamesStart = '\0';
47const char COMPILER_RT_SECTION(".lprfn$Z") NamesEnd = '\0';
48
49char COMPILER_RT_SECTION(".lprfc$A") CountersStart;
50char COMPILER_RT_SECTION(".lprfc$Z") CountersEnd;
51char COMPILER_RT_SECTION(".lprfb$A") BitmapStart;
52char COMPILER_RT_SECTION(".lprfb$Z") BitmapEnd;
53
54ValueProfNode COMPILER_RT_SECTION(".lprfnd$A") VNodesStart;
55ValueProfNode COMPILER_RT_SECTION(".lprfnd$Z") VNodesEnd;
56
57const __llvm_profile_data *__llvm_profile_begin_data(void) {
58 return &DataStart + 1;
59}
60const __llvm_profile_data *__llvm_profile_end_data(void) { return &DataEnd; }
61
62// Type profiling isn't implemented under MSVC ABI, so return NULL (rather than
63// implementing linker magic on Windows) to make it more explicit. To elaborate,
64// the current type profiling implementation maps a profiled vtable address to a
65// vtable variable through vtables mangled name. Under MSVC ABI, the variable
66// name for vtables might not be the mangled name (see
67// MicrosoftCXXABI::getAddrOfVTable in MicrosoftCXXABI.cpp for more details on
68// how a vtable name is computed). Note the mangled name is still in the vtable
69// IR (just not variable name) for mapping purpose, but more implementation work
70// is required.
71const VTableProfData *__llvm_profile_begin_vtables(void) { return NULL; }
72const VTableProfData *__llvm_profile_end_vtables(void) { return NULL; }
73
74const char *__llvm_profile_begin_names(void) { return &NamesStart + 1; }
75const char *__llvm_profile_end_names(void) { return &NamesEnd; }
76
77// Type profiling isn't supported on Windows, so return NULl to make it more
78// explicit.
79const char *__llvm_profile_begin_vtabnames(void) { return NULL; }
80const char *__llvm_profile_end_vtabnames(void) { return NULL; }
81
82char *__llvm_profile_begin_counters(void) { return &CountersStart + 1; }
83char *__llvm_profile_end_counters(void) { return &CountersEnd; }
84char *__llvm_profile_begin_bitmap(void) { return &BitmapStart + 1; }
85char *__llvm_profile_end_bitmap(void) { return &BitmapEnd; }
86
87ValueProfNode *__llvm_profile_begin_vnodes(void) { return &VNodesStart + 1; }
88ValueProfNode *__llvm_profile_end_vnodes(void) { return &VNodesEnd; }
89
90ValueProfNode *CurrentVNode = &VNodesStart + 1;
91ValueProfNode *EndVNode = &VNodesEnd;
92
93/* lld-link provides __buildid symbol which points to the 16 bytes build id when
94 * using /build-id flag. https://lld.llvm.org/windows_support.html#lld-flags */
95#define BUILD_ID_LEN 16
96COMPILER_RT_WEAK uint8_t __buildid[BUILD_ID_LEN] = {0};
97COMPILER_RT_VISIBILITY int __llvm_write_binary_ids(ProfDataWriter *Writer) {
98 static const uint8_t zeros[BUILD_ID_LEN] = {0};
99 if (memcmp(__buildid, zeros, BUILD_ID_LEN) != 0) {
100 if (Writer &&
101 lprofWriteOneBinaryId(Writer, BUILD_ID_LEN, __buildid, 0) == -1)
102 return -1;
103 return sizeof(uint64_t) + BUILD_ID_LEN;
104 }
105 return 0;
106}
107
108#endif
109