File indexing completed on 2024-09-15 13:07:08

 /*
     SPDX-FileCopyrightText: 2018 Milian Wolff <mail@milianw.de>
 
     SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
 #define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
 #include "3rdparty/doctest.h"
 
 #include "track/libheaptrack.h"
 #include "util/linewriter.h"
 
 #include <cmath>
 #include <cstdio>
 
 #include <future>
 #include <iostream>
 #include <thread>
 #include <vector>
 
 #include "tempfile.h"
 
 bool initBeforeCalled = false;
 bool initAfterCalled = false;
 bool stopCalled = false;
 
 using namespace std;
 
 TEST_CASE ("api") {
     TempFile tmp; // opened/closed by heaptrack_init
 
     SUBCASE("init")
     {
         heaptrack_init(
             tmp.fileName.c_str(),
             []() {
                 REQUIRE(!initBeforeCalled);
                 REQUIRE(!initAfterCalled);
                 REQUIRE(!stopCalled);
                 initBeforeCalled = true;
             },
             [](LineWriter& /*out*/) {
                 REQUIRE(initBeforeCalled);
                 REQUIRE(!initAfterCalled);
                 REQUIRE(!stopCalled);
                 initAfterCalled = true;
             },
             []() {
                 REQUIRE(initBeforeCalled);
                 REQUIRE(initAfterCalled);
                 REQUIRE(!stopCalled);
                 stopCalled = true;
             });
 
         REQUIRE(initBeforeCalled);
         REQUIRE(initAfterCalled);
         REQUIRE(!stopCalled);
 
         int data[2] = {0};
 
         SUBCASE("no-op-malloc")
         {
             heaptrack_malloc(0, 0);
         }
         SUBCASE("no-op-malloc-free")
         {
             heaptrack_free(0);
         }
         SUBCASE("no-op-malloc-realloc")
         {
             heaptrack_realloc(data, 1, 0);
         }
 
         SUBCASE("malloc-free")
         {
             heaptrack_malloc(data, 4);
             heaptrack_free(data);
         }
 
         SUBCASE("realloc")
         {
             heaptrack_malloc(data, 4);
             heaptrack_realloc(data, 8, data);
             heaptrack_realloc(data, 16, data + 1);
             heaptrack_free(data + 1);
         }
 
         SUBCASE("invalidate-cache")
         {
             heaptrack_invalidate_module_cache();
         }
 
         SUBCASE("multi-threaded")
         {
             const auto numThreads = min(4u, thread::hardware_concurrency());
 
             cout << "start threads" << endl;
             {
                 vector<future<void>> futures;
                 for (unsigned i = 0; i < numThreads; ++i) {
                     futures.emplace_back(async(launch::async, []() {
                         for (int i = 0; i < 10000; ++i) {
                             heaptrack_malloc(&i, i);
                             heaptrack_realloc(&i, i + 1, &i);
                             heaptrack_free(&i);
                             if (i % 100 == 0) {
                                 heaptrack_invalidate_module_cache();
                             }
                         }
                     }));
                 }
             }
             cout << "threads finished" << endl;
         }
 
         SUBCASE("stop")
         {
             heaptrack_stop();
             REQUIRE(stopCalled);
         }
     }
 }