85 lines · cpp
1//===-- asan_benchmarks_test.cpp ---------------------===//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-exception6//7//===----------------------------------------------------------------------===//8//9// This file is a part of AddressSanitizer, an address sanity checker.10//11// Some benchmarks for the instrumented code.12//===----------------------------------------------------------------------===//13 14#include "asan_test_utils.h"15 16template<class T>17__attribute__((noinline))18static void ManyAccessFunc(T *x, size_t n_elements, size_t n_iter) {19 for (size_t iter = 0; iter < n_iter; iter++) {20 break_optimization(0);21 // hand unroll the loop to stress the reg alloc.22 for (size_t i = 0; i <= n_elements - 16; i += 16) {23 x[i + 0] = i;24 x[i + 1] = i;25 x[i + 2] = i;26 x[i + 3] = i;27 x[i + 4] = i;28 x[i + 5] = i;29 x[i + 6] = i;30 x[i + 7] = i;31 x[i + 8] = i;32 x[i + 9] = i;33 x[i + 10] = i;34 x[i + 11] = i;35 x[i + 12] = i;36 x[i + 13] = i;37 x[i + 14] = i;38 x[i + 15] = i;39 }40 }41}42 43TEST(AddressSanitizer, ManyAccessBenchmark) {44 size_t kLen = 1024;45 int *int_array = new int[kLen];46 ManyAccessFunc(int_array, kLen, 1 << 24);47 delete [] int_array;48}49 50// access 7 char elements in a 7 byte array (i.e. on the border).51__attribute__((noinline))52static void BorderAccessFunc(char *x, size_t n_iter) {53 for (size_t iter = 0; iter < n_iter; iter++) {54 break_optimization(x);55 x[0] = 0;56 x[1] = 0;57 x[2] = 0;58 x[3] = 0;59 x[4] = 0;60 x[5] = 0;61 x[6] = 0;62 }63}64 65TEST(AddressSanitizer, BorderAccessBenchmark) {66 char *char_7_array = new char[7];67 BorderAccessFunc(char_7_array, 1 << 30);68 delete [] char_7_array;69}70 71static void FunctionWithLargeStack() {72 int stack[1000];73 Ident(stack);74}75 76TEST(AddressSanitizer, FakeStackBenchmark) {77 for (int i = 0; i < 10000000; i++)78 Ident(&FunctionWithLargeStack)();79}80 81int main(int argc, char **argv) {82 testing::InitGoogleTest(&argc, argv);83 return RUN_ALL_TESTS();84}85