99 lines · cpp
1//===-- sanitizer_vector_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 *Sanitizer runtime.10//11//===----------------------------------------------------------------------===//12#include "sanitizer_common/sanitizer_ring_buffer.h"13#include "gtest/gtest.h"14 15namespace __sanitizer {16 17struct LargeStruct {18 int64_t v;19 int64_t extra[3];20 21 explicit LargeStruct(int64_t v) : v(v) {}22 operator int64_t() { return v; }23};24 25struct Struct10Bytes {26 short t[3];27};28 29TEST(RingBuffer, Construct) {30 RingBuffer<int64_t> *RBlong = RingBuffer<int64_t>::New(20);31 EXPECT_EQ(RBlong->size(), 20U);32 RBlong->Delete();33}34 35template <class T> void TestRB() {36 RingBuffer<T> *RB;37 const size_t Sizes[] = {1, 2, 3, 5, 8, 16, 20, 40, 10000};38 for (size_t Size : Sizes) {39 RB = RingBuffer<T>::New(Size);40 EXPECT_EQ(RB->size(), Size);41 RB->Delete();42 }43 44 RB = RingBuffer<T>::New(4);45 EXPECT_EQ(RB->size(), 4U);46#define EXPECT_RING_BUFFER(a0, a1, a2, a3) \47 EXPECT_EQ((int64_t)(*RB)[0], (int64_t)a0); \48 EXPECT_EQ((int64_t)(*RB)[1], (int64_t)a1); \49 EXPECT_EQ((int64_t)(*RB)[2], (int64_t)a2); \50 EXPECT_EQ((int64_t)(*RB)[3], (int64_t)a3);51 52 RB->push(T(1)); EXPECT_RING_BUFFER(1, 0, 0, 0);53 RB->push(T(2)); EXPECT_RING_BUFFER(2, 1, 0, 0);54 RB->push(T(3)); EXPECT_RING_BUFFER(3, 2, 1, 0);55 RB->push(T(4)); EXPECT_RING_BUFFER(4, 3, 2, 1);56 RB->push(T(5)); EXPECT_RING_BUFFER(5, 4, 3, 2);57 RB->push(T(6)); EXPECT_RING_BUFFER(6, 5, 4, 3);58 RB->push(T(7)); EXPECT_RING_BUFFER(7, 6, 5, 4);59 RB->push(T(8)); EXPECT_RING_BUFFER(8, 7, 6, 5);60 RB->push(T(9)); EXPECT_RING_BUFFER(9, 8, 7, 6);61 RB->push(T(10)); EXPECT_RING_BUFFER(10, 9, 8, 7);62 RB->push(T(11)); EXPECT_RING_BUFFER(11, 10, 9, 8);63 RB->push(T(12)); EXPECT_RING_BUFFER(12, 11, 10, 9);64 65#undef EXPECT_RING_BUFFER66}67 68#if SANITIZER_WORDSIZE == 6469TEST(RingBuffer, int64) {70 TestRB<int64_t>();71}72 73TEST(RingBuffer, LargeStruct) {74 TestRB<LargeStruct>();75}76 77template<typename T>78CompactRingBuffer<T> *AllocCompactRingBuffer(size_t count) {79 size_t sz = sizeof(T) * count;80 EXPECT_EQ(0ULL, sz % 4096);81 void *p = MmapAlignedOrDieOnFatalError(sz, sz * 2, "CompactRingBuffer");82 return new CompactRingBuffer<T>(p, sz);83}84 85TEST(CompactRingBuffer, int64) {86 const size_t page_sizes[] = {1, 2, 4, 128};87 88 for (size_t pages : page_sizes) {89 size_t count = 4096 * pages / sizeof(int64_t);90 auto R = AllocCompactRingBuffer<int64_t>(count);91 int64_t top = count * 3 + 13;92 for (int64_t i = 0; i < top; ++i) R->push(i);93 for (int64_t i = 0; i < (int64_t)count; ++i)94 EXPECT_EQ(top - i - 1, (*R)[i]);95 }96}97#endif98} // namespace __sanitizer99