ucx: test/util_allocator.c@afd87df80b13

     1 /*

     2  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

     3  *

     4  * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved.

     5  *

     6  * Redistribution and use in source and binary forms, with or without

     7  * modification, are permitted provided that the following conditions are met:

     8  *

     9  *   1. Redistributions of source code must retain the above copyright

    10  *      notice, this list of conditions and the following disclaimer.

    11  *

    12  *   2. Redistributions in binary form must reproduce the above copyright

    13  *      notice, this list of conditions and the following disclaimer in the

    14  *      documentation and/or other materials provided with the distribution.

    15  *

    16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

    17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

    19  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE

    20  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

    21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

    22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

    23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

    24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

    25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

    26  * POSSIBILITY OF SUCH DAMAGE.

    27  */

    29 #include "util_allocator.h"

    30 #include <string.h>

    32 void cx_testing_allocator_add(cx_testing_allocator_s *data, void *ptr) {

    33     data->tracked[data->live] = ptr;

    34     data->live++;

    35 }

    37 int cx_testing_allocator_remove(cx_testing_allocator_s *data, void *ptr) {

    38     for (int i = 0; i < data->live; i++) {

    39         if (data->tracked[i] == ptr) {

    40             data->tracked[i] = data->tracked[data->live - 1];

    41             data->live--;

    42             return 0;

    43         }

    44     }

    45     return 1;

    46 }

    48 void *cx_malloc_testing(void *d, size_t n) {

    49     cx_testing_allocator_s *data = d;

    50     void *ptr = malloc(n);

    51     data->alloc_total++;

    52     if (ptr == NULL) {

    53         data->alloc_failed++;

    54     } else {

    55         cx_testing_allocator_add(data, ptr);

    56     }

    57     return ptr;

    58 }

    60 void *cx_realloc_testing(void *d, void *mem, size_t n) {

    61     cx_testing_allocator_s *data = d;

    62     void *ptr = realloc(mem, n);

    63     if (ptr == mem) {

    64         return ptr;

    65     } else {

    66         data->alloc_total++;

    67         if (ptr == NULL) {

    68             data->alloc_failed++;

    69         } else {

    70             data->free_total++;

    71             if (cx_testing_allocator_remove(data, mem)) {

    72                 data->free_failed++;

    73             }

    74             cx_testing_allocator_add(data, ptr);

    75         }

    76         return ptr;

    77     }

    78 }

    80 void *cx_calloc_testing(void *d, size_t nelem, size_t n) {

    81     cx_testing_allocator_s *data = d;

    82     void *ptr = calloc(nelem, n);

    83     data->alloc_total++;

    84     if (ptr == NULL) {

    85         data->alloc_failed++;

    86     } else {

    87         cx_testing_allocator_add(data, ptr);

    88     }

    89     return ptr;

    90 }

    92 void cx_free_testing(void *d, void *mem) {

    93     cx_testing_allocator_s *data = d;

    94     data->free_total++;

    95     if (cx_testing_allocator_remove(data, mem)) {

    96         data->free_failed++;

    97         // do not even attempt to free mem, because it is likely to segfault

    98     } else {

    99         free(mem);

   100     }

   101 }

   103 cx_allocator_class cx_testing_allocator_class = {

   104         cx_malloc_testing,

   105         cx_realloc_testing,

   106         cx_calloc_testing,

   107         cx_free_testing

   108 };

   110 cx_testing_allocator_s cx_testing_allocator_data;

   112 struct cx_allocator_s cx_testing_allocator = {

   113         &cx_testing_allocator_class,

   114         &cx_testing_allocator_data

   115 };

   116 CxAllocator cxTestingAllocator = &cx_testing_allocator;

   118 void cxTestingAllocatorReset(void) {

   119     memset(&cx_testing_allocator_data, 0, sizeof(cx_testing_allocator_s));

   120 }

   122 int cxTestingAllocatorVerify(void) {

   123     return cx_testing_allocator_data.live == 0

   124            && cx_testing_allocator_data.alloc_failed == 0 && cx_testing_allocator_data.free_failed == 0

   125            && cx_testing_allocator_data.alloc_total == cx_testing_allocator_data.free_total;

   126 }

Mercurial > hg > ucx / file revision