teshsuite/xbt/mmalloc/mmalloc_test.cpp

   1 /* Copyright (c) 2012-2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "simgrid/Exception.hpp"
   7 #include "xbt.h"
   8 #include "xbt/mmalloc.h"
   9
  10 #include <array>
  11 #include <cassert>
  12 #include <cstdio>
  13 #include <cstdlib>
  14 #include <cstring>
  15 #include <fcntl.h>
  16 #include <sys/stat.h>
  17 #include <unistd.h>
  18 #include <vector>
  19
  20 XBT_LOG_NEW_DEFAULT_CATEGORY(test,"this test");
  21
  22 constexpr int BUFFSIZE = 204800;
  23 constexpr int TESTSIZE = 100;
  24
  25 #define size_of_block(i) ((((i) % 50) + 1) * 100)
  26
  27 static void check_block(const unsigned char* p, unsigned char b, int n)
  28 {
  29   for (int i = 0; i < n; i++)
  30     xbt_assert(p[i] == b, "value mismatch: %p[%d] = %#hhx, expected %#hhx", p, i, p[i], b);
  31 }
  32
  33 int main(int argc, char**argv)
  34 {
  35   xbt_mheap_t heapA = nullptr;
  36   std::array<void*, TESTSIZE> pointers;
  37   xbt_init(&argc,argv);
  38
  39   XBT_INFO("Allocating a new heap");
  40   unsigned long mask = ~((unsigned long)xbt_pagesize - 1);
  41   auto* addr         = reinterpret_cast<void*>(((unsigned long)sbrk(0) + BUFFSIZE) & mask);
  42   heapA              = xbt_mheap_new(addr, 0);
  43   if (heapA == nullptr) {
  44     perror("attach 1 failed");
  45     fprintf(stderr, "bye\n");
  46     exit(1);
  47   }
  48
  49   XBT_INFO("HeapA allocated");
  50
  51   int i;
  52   int size;
  53   for (i = 0; i < TESTSIZE; i++) {
  54     size = size_of_block(i);
  55     pointers[i] = mmalloc(heapA, size);
  56     XBT_INFO("%d bytes allocated with offset %zx", size, (size_t)((char*)pointers[i] - (char*)heapA));
  57   }
  58   XBT_INFO("All blocks were correctly allocated. Free every second block");
  59   for (i = 0; i < TESTSIZE; i+=2) {
  60     mfree(heapA, pointers[i]);
  61   }
  62   XBT_INFO("Memset every second block to zero (yeah, they are not currently allocated :)");
  63   for (i = 0; i < TESTSIZE; i+=2) {
  64     size = size_of_block(i);
  65     memset(pointers[i],0, size);
  66   }
  67   XBT_INFO("Re-allocate every second block");
  68   for (i = 0; i < TESTSIZE; i+=2) {
  69     size = size_of_block(i);
  70     pointers[i] = mmalloc(heapA, size);
  71   }
  72
  73   XBT_INFO("free all blocks");
  74   for (i = 0; i < TESTSIZE; i++)
  75     mfree(heapA, pointers[i]);
  76
  77   XBT_INFO("Let's try different codepaths for mrealloc");
  78   for (i = 0; i < TESTSIZE; i++) {
  79     const std::vector<std::pair<int, unsigned char>> requests = {
  80         {size_of_block(i) / 2, 0x77}, {size_of_block(i) * 2, 0xaa}, {1, 0xc0}, {0, 0}};
  81     pointers[i] = nullptr;
  82     for (unsigned k = 0; k < requests.size(); ++k) {
  83       size        = requests[k].first;
  84       pointers[i] = mrealloc(heapA, pointers[i], size);
  85       if (k > 0)
  86         check_block(static_cast<unsigned char*>(pointers[i]), requests[k - 1].second,
  87                     std::min(size, requests[k - 1].first));
  88       if (size > 0)
  89         memset(pointers[i], requests[k].second, size);
  90     }
  91   }
  92
  93   XBT_INFO("Damnit, I cannot break mmalloc this time. That's SO disappointing.");
  94   return 0;
  95 }