是否有用于 16 和/或 32 位值的 memset() 函数？

Question

memset() 非常快，因为它在内部受益于stosb 操作码。 是否有适用于 16 位和 32 位值的函数，可以有效地受益于 stosb、stosw 和/或 stosd？

wmemset() 不可移植，对 16 位值没有帮助。

Answer 1

标准 C 中没有这样的功能，但根据您要执行的操作，您可能会使用特定于 CPU 的 SIMD“内在功能”来构建一个。您提到 stosb 让我觉得您使用的是 x86，因此请查看各种 *mmintrin.h 标头及其提供的功能的文档。

Answer 2

是的，它有很多变体。

例如

void *memset16(void *m, uint16_t val, size_t count)
{
    uint16_t *buf = m;

    while(count--) *buf++ = val;
    return m;
}

void *memset32(void *m, uint32_t val, size_t count)
{
    uint32_t *buf = m;

    while(count--) *buf++ = val;
    return m;
}

void *memsetXX(void *m, void *val, size_t size, size_t count)
{
    char *buf = m;

    while(count--)
    {
        memcpy(buf, val, size);
        buf += size;
    }
    return m;
}

更安全的版本：

void *memset16safe(void *m, uint16_t val, size_t count)
{
    char *buf = m;
    union 
    {
        uint8_t d8[2];
        uint16_t d16;
    }u16 = {.d16 = val};

    while(count--) 
    {
        *buf++ = u16.d8[0];
        *buf++ = u16.d8[1];
    }
    return m;
}

void *memset32(void *m, uint32_t val, size_t count)
{
    char *buf = m;
    union 
    {
        uint8_t d8[4];
        uint32_t d32;
    }u32 = {.d32 = val};

    while(count--) 
    {
        *buf++ = u32.d8[0];
        *buf++ = u32.d8[1];
        *buf++ = u32.d8[2];
        *buf++ = u32.d8[3];
    }
    return m;
}