10b57cec5SDimitry Andric /*===- InstrProfilingPort.h- Support library for PGO instrumentation ------===*\
20b57cec5SDimitry Andric |*
30b57cec5SDimitry Andric |* Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric |* See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric |* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric |*
70b57cec5SDimitry Andric \*===----------------------------------------------------------------------===*/
80b57cec5SDimitry Andric
90b57cec5SDimitry Andric /* This header must be included after all others so it can provide fallback
100b57cec5SDimitry Andric definitions for stuff missing in system headers. */
110b57cec5SDimitry Andric
120b57cec5SDimitry Andric #ifndef PROFILE_INSTRPROFILING_PORT_H_
130b57cec5SDimitry Andric #define PROFILE_INSTRPROFILING_PORT_H_
140b57cec5SDimitry Andric
150b57cec5SDimitry Andric #ifdef _MSC_VER
160b57cec5SDimitry Andric #define COMPILER_RT_ALIGNAS(x) __declspec(align(x))
170b57cec5SDimitry Andric #define COMPILER_RT_VISIBILITY
180b57cec5SDimitry Andric /* FIXME: selectany does not have the same semantics as weak. */
190b57cec5SDimitry Andric #define COMPILER_RT_WEAK __declspec(selectany)
200b57cec5SDimitry Andric /* Need to include <windows.h> */
210b57cec5SDimitry Andric #define COMPILER_RT_ALLOCA _alloca
220b57cec5SDimitry Andric /* Need to include <stdio.h> and <io.h> */
230b57cec5SDimitry Andric #define COMPILER_RT_FTRUNCATE(f,l) _chsize(_fileno(f),l)
240b57cec5SDimitry Andric #define COMPILER_RT_ALWAYS_INLINE __forceinline
25480093f4SDimitry Andric #define COMPILER_RT_CLEANUP(x)
26*fe6060f1SDimitry Andric #define COMPILER_RT_USED
270b57cec5SDimitry Andric #elif __GNUC__
28e8d8bef9SDimitry Andric #ifdef _WIN32
29e8d8bef9SDimitry Andric #define COMPILER_RT_FTRUNCATE(f, l) _chsize(fileno(f), l)
30e8d8bef9SDimitry Andric #define COMPILER_RT_VISIBILITY
31e8d8bef9SDimitry Andric #define COMPILER_RT_WEAK __attribute__((selectany))
32e8d8bef9SDimitry Andric #else
33e8d8bef9SDimitry Andric #define COMPILER_RT_FTRUNCATE(f, l) ftruncate(fileno(f), l)
340b57cec5SDimitry Andric #define COMPILER_RT_VISIBILITY __attribute__((visibility("hidden")))
350b57cec5SDimitry Andric #define COMPILER_RT_WEAK __attribute__((weak))
36e8d8bef9SDimitry Andric #endif
37e8d8bef9SDimitry Andric #define COMPILER_RT_ALIGNAS(x) __attribute__((aligned(x)))
380b57cec5SDimitry Andric #define COMPILER_RT_ALLOCA __builtin_alloca
390b57cec5SDimitry Andric #define COMPILER_RT_ALWAYS_INLINE inline __attribute((always_inline))
40480093f4SDimitry Andric #define COMPILER_RT_CLEANUP(x) __attribute__((cleanup(x)))
41*fe6060f1SDimitry Andric #define COMPILER_RT_USED __attribute__((used))
420b57cec5SDimitry Andric #endif
430b57cec5SDimitry Andric
440b57cec5SDimitry Andric #if defined(__APPLE__)
450b57cec5SDimitry Andric #define COMPILER_RT_SEG "__DATA,"
460b57cec5SDimitry Andric #else
470b57cec5SDimitry Andric #define COMPILER_RT_SEG ""
480b57cec5SDimitry Andric #endif
490b57cec5SDimitry Andric
500b57cec5SDimitry Andric #ifdef _MSC_VER
510b57cec5SDimitry Andric #define COMPILER_RT_SECTION(Sect) __declspec(allocate(Sect))
520b57cec5SDimitry Andric #else
530b57cec5SDimitry Andric #define COMPILER_RT_SECTION(Sect) __attribute__((section(Sect)))
540b57cec5SDimitry Andric #endif
550b57cec5SDimitry Andric
560b57cec5SDimitry Andric #define COMPILER_RT_MAX_HOSTLEN 128
570b57cec5SDimitry Andric #ifdef __ORBIS__
580b57cec5SDimitry Andric #define COMPILER_RT_GETHOSTNAME(Name, Len) ((void)(Name), (void)(Len), (-1))
590b57cec5SDimitry Andric #else
600b57cec5SDimitry Andric #define COMPILER_RT_GETHOSTNAME(Name, Len) lprofGetHostName(Name, Len)
610b57cec5SDimitry Andric #endif
620b57cec5SDimitry Andric
630b57cec5SDimitry Andric #if COMPILER_RT_HAS_ATOMICS == 1
641106035dSDimitry Andric #ifdef _WIN32
650b57cec5SDimitry Andric #include <windows.h>
661106035dSDimitry Andric #if defined(_MSC_VER) && _MSC_VER < 1900
670b57cec5SDimitry Andric #define snprintf _snprintf
680b57cec5SDimitry Andric #endif
690b57cec5SDimitry Andric #if defined(_WIN64)
700b57cec5SDimitry Andric #define COMPILER_RT_BOOL_CMPXCHG(Ptr, OldV, NewV) \
710b57cec5SDimitry Andric (InterlockedCompareExchange64((LONGLONG volatile *)Ptr, (LONGLONG)NewV, \
720b57cec5SDimitry Andric (LONGLONG)OldV) == (LONGLONG)OldV)
730b57cec5SDimitry Andric #define COMPILER_RT_PTR_FETCH_ADD(DomType, PtrVar, PtrIncr) \
740b57cec5SDimitry Andric (DomType *)InterlockedExchangeAdd64((LONGLONG volatile *)&PtrVar, \
750b57cec5SDimitry Andric (LONGLONG)sizeof(DomType) * PtrIncr)
760b57cec5SDimitry Andric #else /* !defined(_WIN64) */
770b57cec5SDimitry Andric #define COMPILER_RT_BOOL_CMPXCHG(Ptr, OldV, NewV) \
780b57cec5SDimitry Andric (InterlockedCompareExchange((LONG volatile *)Ptr, (LONG)NewV, (LONG)OldV) == \
790b57cec5SDimitry Andric (LONG)OldV)
800b57cec5SDimitry Andric #define COMPILER_RT_PTR_FETCH_ADD(DomType, PtrVar, PtrIncr) \
810b57cec5SDimitry Andric (DomType *)InterlockedExchangeAdd((LONG volatile *)&PtrVar, \
820b57cec5SDimitry Andric (LONG)sizeof(DomType) * PtrIncr)
830b57cec5SDimitry Andric #endif
841106035dSDimitry Andric #else /* !defined(_WIN32) */
850b57cec5SDimitry Andric #define COMPILER_RT_BOOL_CMPXCHG(Ptr, OldV, NewV) \
860b57cec5SDimitry Andric __sync_bool_compare_and_swap(Ptr, OldV, NewV)
870b57cec5SDimitry Andric #define COMPILER_RT_PTR_FETCH_ADD(DomType, PtrVar, PtrIncr) \
880b57cec5SDimitry Andric (DomType *)__sync_fetch_and_add((long *)&PtrVar, sizeof(DomType) * PtrIncr)
890b57cec5SDimitry Andric #endif
900b57cec5SDimitry Andric #else /* COMPILER_RT_HAS_ATOMICS != 1 */
910b57cec5SDimitry Andric #include "InstrProfilingUtil.h"
920b57cec5SDimitry Andric #define COMPILER_RT_BOOL_CMPXCHG(Ptr, OldV, NewV) \
930b57cec5SDimitry Andric lprofBoolCmpXchg((void **)Ptr, OldV, NewV)
940b57cec5SDimitry Andric #define COMPILER_RT_PTR_FETCH_ADD(DomType, PtrVar, PtrIncr) \
950b57cec5SDimitry Andric (DomType *)lprofPtrFetchAdd((void **)&PtrVar, sizeof(DomType) * PtrIncr)
960b57cec5SDimitry Andric #endif
970b57cec5SDimitry Andric
980b57cec5SDimitry Andric #if defined(_WIN32)
990b57cec5SDimitry Andric #define DIR_SEPARATOR '\\'
1000b57cec5SDimitry Andric #define DIR_SEPARATOR_2 '/'
1010b57cec5SDimitry Andric #else
1020b57cec5SDimitry Andric #define DIR_SEPARATOR '/'
1030b57cec5SDimitry Andric #endif
1040b57cec5SDimitry Andric
1050b57cec5SDimitry Andric #ifndef DIR_SEPARATOR_2
1060b57cec5SDimitry Andric #define IS_DIR_SEPARATOR(ch) ((ch) == DIR_SEPARATOR)
1070b57cec5SDimitry Andric #else /* DIR_SEPARATOR_2 */
1080b57cec5SDimitry Andric #define IS_DIR_SEPARATOR(ch) \
1090b57cec5SDimitry Andric (((ch) == DIR_SEPARATOR) || ((ch) == DIR_SEPARATOR_2))
1100b57cec5SDimitry Andric #endif /* DIR_SEPARATOR_2 */
1110b57cec5SDimitry Andric
112480093f4SDimitry Andric #if defined(_WIN32)
113480093f4SDimitry Andric #include <windows.h>
getpagesize()114480093f4SDimitry Andric static inline size_t getpagesize() {
115480093f4SDimitry Andric SYSTEM_INFO S;
116480093f4SDimitry Andric GetNativeSystemInfo(&S);
117480093f4SDimitry Andric return S.dwPageSize;
118480093f4SDimitry Andric }
119480093f4SDimitry Andric #else /* defined(_WIN32) */
120480093f4SDimitry Andric #include <unistd.h>
121480093f4SDimitry Andric #endif /* defined(_WIN32) */
122480093f4SDimitry Andric
1230b57cec5SDimitry Andric #define PROF_ERR(Format, ...) \
1240b57cec5SDimitry Andric fprintf(stderr, "LLVM Profile Error: " Format, __VA_ARGS__);
1250b57cec5SDimitry Andric
1260b57cec5SDimitry Andric #define PROF_WARN(Format, ...) \
1270b57cec5SDimitry Andric fprintf(stderr, "LLVM Profile Warning: " Format, __VA_ARGS__);
1280b57cec5SDimitry Andric
1290b57cec5SDimitry Andric #define PROF_NOTE(Format, ...) \
1300b57cec5SDimitry Andric fprintf(stderr, "LLVM Profile Note: " Format, __VA_ARGS__);
1310b57cec5SDimitry Andric
1320b57cec5SDimitry Andric #ifndef MAP_FILE
1330b57cec5SDimitry Andric #define MAP_FILE 0
1340b57cec5SDimitry Andric #endif
1350b57cec5SDimitry Andric
1360b57cec5SDimitry Andric #ifndef O_BINARY
1370b57cec5SDimitry Andric #define O_BINARY 0
1380b57cec5SDimitry Andric #endif
1390b57cec5SDimitry Andric
1400b57cec5SDimitry Andric #if defined(__FreeBSD__)
1410b57cec5SDimitry Andric
1420b57cec5SDimitry Andric #include <inttypes.h>
1430b57cec5SDimitry Andric #include <sys/types.h>
1440b57cec5SDimitry Andric
1450b57cec5SDimitry Andric #else /* defined(__FreeBSD__) */
1460b57cec5SDimitry Andric
1470b57cec5SDimitry Andric #include <inttypes.h>
1480b57cec5SDimitry Andric #include <stdint.h>
1490b57cec5SDimitry Andric
1500b57cec5SDimitry Andric #endif /* defined(__FreeBSD__) && defined(__i386__) */
1510b57cec5SDimitry Andric
1520b57cec5SDimitry Andric #endif /* PROFILE_INSTRPROFILING_PORT_H_ */
153