/*
 * Copyright 2025, Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT license.
 */


#include <stdint.h>
#include <string.h>
#undef memset


#define MISALIGNMENT(PTR, TYPE) ((addr_t)(PTR) & (sizeof(TYPE) - 1))

void*
memset(void* dest, int _value, size_t count)
{
        uint8_t* d = (uint8_t*)dest;
        uint8_t value = _value;

        if (count > sizeof(size_t)) {
                while (count > 0 && MISALIGNMENT(d, size_t) != 0) {
                        *d = value;
                        d++;
                        count--;
                }

                const size_t valuesz = value * (SIZE_MAX / UINT8_MAX);
                size_t* dsz = (size_t*)d;

                while (count >= sizeof(size_t)) {
                        *dsz = valuesz;
                        dsz++;
                        count -= sizeof(size_t);
                }
                d = (uint8_t*)dsz;
        }

        while (count > 0) {
                *d = value;
                d++;
                count--;
        }

        return dest;
}