Home | History | Annotate | Download | only in tests
      1 //===-- sanitizer_test_utils.h ----------------------------------*- C++ -*-===//
      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 //
     10 // This file is a part of *Sanitizer runtime.
     11 // Common unit tests utilities.
     12 //
     13 //===----------------------------------------------------------------------===//
     14 
     15 #ifndef SANITIZER_TEST_UTILS_H
     16 #define SANITIZER_TEST_UTILS_H
     17 
     18 #if defined(_WIN32)
     19 // <windows.h> should always be the first include on Windows.
     20 # include <windows.h>
     21 // MSVS headers define max/min as macros, so std::max/min gets crazy.
     22 # undef max
     23 # undef min
     24 #endif
     25 
     26 #if !defined(SANITIZER_EXTERNAL_TEST_CONFIG)
     27 # define INCLUDED_FROM_SANITIZER_TEST_UTILS_H
     28 # include "sanitizer_test_config.h"
     29 # undef INCLUDED_FROM_SANITIZER_TEST_UTILS_H
     30 #endif
     31 
     32 #include <stdint.h>
     33 
     34 #if defined(_MSC_VER)
     35 # define NOINLINE __declspec(noinline)
     36 #else  // defined(_MSC_VER)
     37 # define NOINLINE __attribute__((noinline))
     38 #endif  // defined(_MSC_VER)
     39 
     40 #if !defined(_MSC_VER) || defined(__clang__)
     41 # define UNUSED __attribute__((unused))
     42 # define USED __attribute__((used))
     43 #else
     44 # define UNUSED
     45 # define USED
     46 #endif
     47 
     48 #if !defined(__has_feature)
     49 #define __has_feature(x) 0
     50 #endif
     51 
     52 #ifndef ATTRIBUTE_NO_SANITIZE_ADDRESS
     53 # if __has_feature(address_sanitizer) || defined(__SANITIZE_ADDRESS__)
     54 #  define ATTRIBUTE_NO_SANITIZE_ADDRESS \
     55     __attribute__((no_sanitize_address))
     56 # else
     57 #  define ATTRIBUTE_NO_SANITIZE_ADDRESS
     58 # endif
     59 #endif  // ATTRIBUTE_NO_SANITIZE_ADDRESS
     60 
     61 #if __LP64__ || defined(_WIN64)
     62 #  define SANITIZER_WORDSIZE 64
     63 #else
     64 #  define SANITIZER_WORDSIZE 32
     65 #endif
     66 
     67 // Make the compiler thinks that something is going on there.
     68 inline void break_optimization(void *arg) {
     69 #if !defined(_WIN32) || defined(__clang__)
     70   __asm__ __volatile__("" : : "r" (arg) : "memory");
     71 #endif
     72 }
     73 
     74 // This function returns its parameter but in such a way that compiler
     75 // can not prove it.
     76 template<class T>
     77 NOINLINE
     78 static T Ident(T t) {
     79   T ret = t;
     80   break_optimization(&ret);
     81   return ret;
     82 }
     83 
     84 // Simple stand-alone pseudorandom number generator.
     85 // Current algorithm is ANSI C linear congruential PRNG.
     86 static inline uint32_t my_rand_r(uint32_t* state) {
     87   return (*state = *state * 1103515245 + 12345) >> 16;
     88 }
     89 
     90 static uint32_t global_seed = 0;
     91 
     92 static inline uint32_t my_rand() {
     93   return my_rand_r(&global_seed);
     94 }
     95 
     96 // Set availability of platform-specific functions.
     97 
     98 #if !defined(__APPLE__) && !defined(__ANDROID__) && !defined(_WIN32)
     99 # define SANITIZER_TEST_HAS_POSIX_MEMALIGN 1
    100 #else
    101 # define SANITIZER_TEST_HAS_POSIX_MEMALIGN 0
    102 #endif
    103 
    104 #if !defined(__APPLE__) && !defined(__FreeBSD__) && \
    105     !defined(__ANDROID__) && !defined(_WIN32)
    106 # define SANITIZER_TEST_HAS_MEMALIGN 1
    107 # define SANITIZER_TEST_HAS_PVALLOC 1
    108 # define SANITIZER_TEST_HAS_MALLOC_USABLE_SIZE 1
    109 #else
    110 # define SANITIZER_TEST_HAS_MEMALIGN 0
    111 # define SANITIZER_TEST_HAS_PVALLOC 0
    112 # define SANITIZER_TEST_HAS_MALLOC_USABLE_SIZE 0
    113 #endif
    114 
    115 #if !defined(__APPLE__)
    116 # define SANITIZER_TEST_HAS_STRNLEN 1
    117 #else
    118 # define SANITIZER_TEST_HAS_STRNLEN 0
    119 #endif
    120 
    121 #if defined(__FreeBSD__)
    122 # define SANITIZER_TEST_HAS_PRINTF_L 1
    123 #else
    124 # define SANITIZER_TEST_HAS_PRINTF_L 0
    125 #endif
    126 
    127 #endif  // SANITIZER_TEST_UTILS_H
    128