ucx/allocator.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include "cx/allocator.h"

#include <errno.h>

__attribute__((__malloc__, __alloc_size__(2)))
static void *cx_malloc_stdlib(
        cx_attr_unused void *d,
        size_t n
) {
    return malloc(n);
}

__attribute__((__warn_unused_result__, __alloc_size__(3)))
static void *cx_realloc_stdlib(
        cx_attr_unused void *d,
        void *mem,
        size_t n
) {
    return realloc(mem, n);
}

__attribute__((__malloc__, __alloc_size__(2, 3)))
static void *cx_calloc_stdlib(
        cx_attr_unused void *d,
        size_t nelem,
        size_t n
) {
    return calloc(nelem, n);
}

__attribute__((__nonnull__))
static void cx_free_stdlib(
        cx_attr_unused void *d,
        void *mem
) {
    free(mem);
}

static cx_allocator_class cx_default_allocator_class = {
        cx_malloc_stdlib,
        cx_realloc_stdlib,
        cx_calloc_stdlib,
        cx_free_stdlib
};

struct cx_allocator_s cx_default_allocator = {
        &cx_default_allocator_class,
        NULL
};
CxAllocator *cxDefaultAllocator = &cx_default_allocator;

#undef cx_reallocate
int cx_reallocate(
        void **mem,
        size_t n
) {
    void *nmem = realloc(*mem, n);
    if (nmem == NULL) {
        return 1; // LCOV_EXCL_LINE
    } else {
        *mem = nmem;
        return 0;
    }
}

#undef cx_reallocatearray
int cx_reallocatearray(
        void **mem,
        size_t nmemb,
        size_t size
) {
    size_t n;
    if (cx_szmul(nmemb, size, &n)) {
        errno = EOVERFLOW;
        return 1;
    } else {
        void *nmem = realloc(*mem, n);
        if (nmem == NULL) {
            return 1; // LCOV_EXCL_LINE
        } else {
            *mem = nmem;
            return 0;
        }
    }
}

// IMPLEMENTATION OF HIGH LEVEL API

void *cxMalloc(
        const CxAllocator *allocator,
        size_t n
) {
    return allocator->cl->malloc(allocator->data, n);
}

void *cxRealloc(
        const CxAllocator *allocator,
        void *mem,
        size_t n
) {
    return allocator->cl->realloc(allocator->data, mem, n);
}

void *cxReallocArray(
        const CxAllocator *allocator,
        void *mem,
        size_t nmemb,
        size_t size
) {
    size_t n;
    if (cx_szmul(nmemb, size, &n)) {
        errno = EOVERFLOW;
        return NULL;
    } else {
        return allocator->cl->realloc(allocator->data, mem, n);
    }
}

#undef cxReallocate
int cxReallocate(
        const CxAllocator *allocator,
        void **mem,
        size_t n
) {
    void *nmem = allocator->cl->realloc(allocator->data, *mem, n);
    if (nmem == NULL) {
        return 1; // LCOV_EXCL_LINE
    } else {
        *mem = nmem;
        return 0;
    }
}

#undef cxReallocateArray
int cxReallocateArray(
        const CxAllocator *allocator,
        void **mem,
        size_t nmemb,
        size_t size
) {
    void *nmem = cxReallocArray(allocator, *mem, nmemb, size);
    if (nmem == NULL) {
        return 1; // LCOV_EXCL_LINE
    } else {
        *mem = nmem;
        return 0;
    }
}

void *cxCalloc(
        const CxAllocator *allocator,
        size_t nelem,
        size_t n
) {
    return allocator->cl->calloc(allocator->data, nelem, n);
}

void cxFree(
        const CxAllocator *allocator,
        void *mem
) {
    allocator->cl->free(allocator->data, mem);
}