bertos/struct/heap_test.c

   1 /**
   2  * \file
   3  * <!--
   4  * This file is part of BeRTOS.
   5  *
   6  * Bertos is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * As a special exception, you may use this file as part of a free software
  21  * library without restriction.  Specifically, if other files instantiate
  22  * templates or use macros or inline functions from this file, or you compile
  23  * this file and link it with other files to produce an executable, this
  24  * file does not by itself cause the resulting executable to be covered by
  25  * the GNU General Public License.  This exception does not however
  26  * invalidate any other reasons why the executable file might be covered by
  27  * the GNU General Public License.
  28  *
  29  * Copyright 2009 Develer S.r.l. (http://www.develer.com/)
  30  *
  31  * -->
  32  *
  33  * \brief Heap test.
  34  *
  35  * \author Francesco Sacchi <batt@codewiz.org>
  36  */
  37
  38
  39 #include <struct/fifobuf.h>
  40 #include <struct/heap.h>
  41
  42 #include <cfg/compiler.h>
  43 #include <cfg/test.h>
  44 #include <cfg/debug.h>
  45
  46 #define TEST_LEN 31
  47 #define ALLOC_SIZE 113
  48
  49 #define TEST_LEN2 32
  50 #define ALLOC_SIZE2 128
  51
  52 #define HEAP_SIZE 4096
  53
  54 HEAP_DEFINE_BUF(heap_buf, HEAP_SIZE);
  55 STATIC_ASSERT(sizeof(heap_buf) % sizeof(heap_buf_t) == 0);
  56
  57 Heap h;
  58
  59 int heap_testSetup(void)
  60 {
  61         kdbg_init();
  62         heap_init(&h, heap_buf, sizeof(heap_buf));
  63         return 0;
  64 }
  65
  66 static void alloc_test(size_t size, size_t test_len)
  67 {
  68         //Simple test
  69         uint8_t *a[test_len];
  70
  71         for (size_t i = 0; i < test_len; i++)
  72         {
  73                 a[i] = heap_allocmem(&h, size);
  74                 ASSERT(a[i]);
  75                 for (size_t j = 0; j < size; j++)
  76                         a[i][j] = i;
  77         }
  78
  79         ASSERT(heap_freeSpace(&h) == HEAP_SIZE - test_len * ROUND_UP2(size, sizeof(MemChunk)));
  80
  81         for (size_t i = 0; i < test_len; i++)
  82         {
  83                 for (size_t j = 0; j < size; j++)
  84                 {
  85                         kprintf("a[%d][%d] = %d\n", i, j, a[i][j]);
  86                         ASSERT(a[i][j] == i);
  87                 }
  88                 heap_freemem(&h, a[i], size);
  89         }
  90         ASSERT(heap_freeSpace(&h) == HEAP_SIZE);
  91 }
  92
  93 int heap_testRun(void)
  94 {
  95         alloc_test(ALLOC_SIZE, TEST_LEN);
  96         alloc_test(ALLOC_SIZE2, TEST_LEN2);
  97         /* Try to allocate the whole heap */
  98         uint8_t *b = heap_allocmem(&h, HEAP_SIZE);
  99         ASSERT(b);
 100         ASSERT(heap_freeSpace(&h) == 0);
 101
 102         ASSERT(!heap_allocmem(&h, HEAP_SIZE));
 103
 104         for (int j = 0; j < HEAP_SIZE; j++)
 105                 b[j] = j;
 106
 107         for (int j = 0; j < HEAP_SIZE; j++)
 108         {
 109                 kprintf("b[%d] = %d\n", j, j);
 110                 ASSERT(b[j] == (j & 0xff));
 111         }
 112         heap_freemem(&h, b, HEAP_SIZE);
 113         ASSERT(heap_freeSpace(&h) == HEAP_SIZE);
 114
 115         return 0;
 116 }
 117
 118 int heap_testTearDown(void)
 119 {
 120         return 0;
 121 }
 122
 123 TEST_MAIN(heap);