#include <catch.hpp>

#include "pls/internal/base/spin_lock.h"
#include "pls/internal/base/system_details.h"
#include "pls/internal/base/stack_allocator.h"

#include "test_helpers.h"

#include <mutex>
#include <atomic>
#include <thread>

using namespace pls::internal::base;

int base_tests_shared_counter;

TEST_CASE("mmap stack allocator", "[internal/base/stack_allocator.h") {
  mmap_stack_allocator stack_allocator;

  char *stack = stack_allocator.allocate_stack(4096);

  SECTION("stack passes in valid range") {
    stack[0] = 'a';
    stack[4096 - 1] = 'a';

    REQUIRE(true);
  }

  SECTION("guard page sigsev on overflow") {
    REQUIRE(CHECK_ABORT([&]() { stack[-10] = 'a'; }));
  }

  stack_allocator.free_stack(4096, stack);

  SECTION("stack unmaps after free") {
    REQUIRE(CHECK_ABORT([&]() { stack[0] = 'a'; }));
  }
}

TEST_CASE("spinlock protects concurrent counter", "[internal/data_structures/spinlock.h]") {
  constexpr int num_iterations = 10000;
  base_tests_shared_counter = 0;
  std::atomic<int> barrier{2};

  spin_lock lock{};

  SECTION("lock can be used by itself") {
    std::thread t1{[&]() {
      barrier--;
      while (barrier != 0);

      for (int i = 0; i < num_iterations; i++) {
        lock.lock();
        base_tests_shared_counter++;
        lock.unlock();
      }
    }};
    std::thread t2{[&]() {
      barrier--;
      while(barrier != 0);

      for (int i = 0; i < num_iterations; i++) {
        lock.lock();
        base_tests_shared_counter--;
        lock.unlock();
      }
    }};

    t1.join();
    t2.join();

    REQUIRE(base_tests_shared_counter == 0);
  }

  SECTION("lock can be used with std::lock_guard") {
    std::thread t1{[&]() {
      for (int i = 0; i < num_iterations; i++) {
        std::lock_guard<spin_lock> my_lock{lock};
        base_tests_shared_counter++;
      }
    }};
    std::thread t2{[&]() {
      for (int i = 0; i < num_iterations; i++) {
        std::lock_guard<spin_lock> my_lock{lock};
        base_tests_shared_counter--;
      }
    }};

    t1.join();
    t2.join();

    REQUIRE(base_tests_shared_counter == 0);
  }
}