src/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>
#include <string.h>

#ifdef _WIN32
#include <Windows.h>
#include <sysinfoapi.h>
unsigned long cx_system_page_size(void) {
    static unsigned long ps = 0;
    if (ps == 0) {
        SYSTEM_INFO sysinfo;
        GetSystemInfo(&sysinfo);
        ps = (unsigned long) sysinfo.dwPageSize;
    }
    return ps;
}
#else
#include <unistd.h>
unsigned long cx_system_page_size(void) {
    static unsigned long ps = 0;
    if (ps == 0) {
        long sc = sysconf(_SC_PAGESIZE);
        if (sc < 0) {
            // fallback for systems which do not report a value here
            ps = 4096; // LCOV_EXCL_LINE
        } else {
            ps = (unsigned long) sc;
        }
    }
    return ps;
}
#endif

static void *cx_malloc_stdlib(
        CX_UNUSED void *d,
        size_t n
) {
    return malloc(n);
}

static void *cx_realloc_stdlib(
        CX_UNUSED void *d,
        void *mem,
        size_t n
) {
    return realloc(mem, n);
}

static void *cx_calloc_stdlib(
        CX_UNUSED void *d,
        size_t nmemb,
        size_t size
) {
    return calloc(nmemb, size);
}

static void cx_free_stdlib(
        CX_UNUSED void *d,
        void *mem
) {
    free(mem);
}

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

struct cx_allocator_s cx_stdlib_allocator = {
        &cx_stdlib_allocator_class,
        NULL
};
const CxAllocator * const cxStdlibAllocator = &cx_stdlib_allocator;
const CxAllocator * cxDefaultAllocator = &cx_stdlib_allocator;

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

int cx_reallocatearray_(
        void **mem,
        size_t nmemb,
        size_t size
) {
    if (nmemb == 0 || size == 0) {
        free(*mem);
        *mem = NULL;
        return 0;
    }
    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 *cxZalloc(
        const CxAllocator *allocator,
        size_t n
) {
    void *mem = allocator->cl->malloc(allocator->data, n);
    if (mem != NULL) {
        memset(mem, 0, n);
    }
    return mem;
}

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);
    }
}

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

int cxReallocateArray_(
        const CxAllocator *allocator,
        void **mem,
        size_t nmemb,
        size_t size
) {
    if (nmemb == 0 || size == 0) {
        cxFree(allocator, *mem);
        *mem = NULL;
        return 0;
    }
    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 nmemb,
        size_t size
) {
    return allocator->cl->calloc(allocator->data, nmemb, size);
}

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

void cxFreeDefault(void *mem) {
    cxDefaultAllocator->cl->free(cxDefaultAllocator->data, mem);
}