blob: f9dd4c5dbe350cfc8ef3cc5796d5bcd5faba1089 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
/*
* Copyright (c) 2013-2020, ARM Limited and Contributors. All rights reserved.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
#include <stddef.h>
#include <string.h>
#include <stdint.h>
void *memset(void *dst, int val, size_t count)
{
char *ptr = dst;
uint64_t *ptr64;
uint64_t fill = (unsigned char)val;
/* Simplify code below by making sure we write at least one byte. */
if (count == 0) {
return dst;
}
/* Handle the first part, until the pointer becomes 64-bit aligned. */
while (((uintptr_t)ptr & 7)) {
*ptr++ = val;
if (--count == 0) {
return dst;
}
}
/* Duplicate the fill byte to the rest of the 64-bit word. */
fill |= fill << 8;
fill |= fill << 16;
fill |= fill << 32;
/* Use 64-bit writes for as long as possible. */
ptr64 = (void *)ptr;
for (; count >= 8; count -= 8) {
*ptr64++ = fill;
}
/* Handle the remaining part byte-per-byte. */
ptr = (void *)ptr64;
while (count--) {
*ptr++ = val;
}
return dst;
}
|