mirror of
https://github.com/Fennix-Project/Kernel.git
synced 2025-05-25 22:14:37 +00:00
115 lines
2.4 KiB
C++
115 lines
2.4 KiB
C++
/*
|
|
This file is part of Fennix Kernel.
|
|
|
|
Fennix Kernel is free software: you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License as
|
|
published by the Free Software Foundation, either version 3 of
|
|
the License, or (at your option) any later version.
|
|
|
|
Fennix Kernel is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with Fennix Kernel. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <convert.h>
|
|
|
|
#include <memory.hpp>
|
|
#include <limits.h>
|
|
#include <debug.h>
|
|
#include <cpu.hpp>
|
|
|
|
/*
|
|
TODO: Replace these functions with even more optimized versions.
|
|
The current versions are fast but not as fast as they could be and also we need implementation for avx, not only sse.
|
|
*/
|
|
|
|
// TODO: Implement these functions
|
|
|
|
EXTERNC void *memset_sse(void *dest, int c, size_t n)
|
|
{
|
|
return memset_unsafe(dest, c, n);
|
|
}
|
|
|
|
EXTERNC void *memset_sse2(void *dest, int c, size_t n)
|
|
{
|
|
size_t i;
|
|
if ((size_t)dest & (16 - 1))
|
|
{
|
|
i = 0;
|
|
while (((size_t)dest + i) & (16 - 1) && i < n)
|
|
{
|
|
asmv("stosb\n"
|
|
:
|
|
: "D"((size_t)dest + i),
|
|
"a"(c));
|
|
i++;
|
|
}
|
|
}
|
|
else
|
|
i = 0;
|
|
|
|
for (; i + 64 <= n; i += 64)
|
|
{
|
|
asmv("movd %0, %%xmm0\n"
|
|
"movdqa %%xmm0, (%1)\n"
|
|
"movdqa %%xmm0, 16(%1)\n"
|
|
"movdqa %%xmm0, 32(%1)\n"
|
|
"movdqa %%xmm0, 48(%1)\n"
|
|
:
|
|
: "r"(c), "r"(((size_t)dest) + i)
|
|
: "xmm0");
|
|
}
|
|
|
|
asmv("rep stosb\n" ::"a"((size_t)(c)),
|
|
"D"(((size_t)dest) + i),
|
|
"c"(n - i));
|
|
|
|
i += n - i;
|
|
return (void *)(((size_t)dest) + i);
|
|
}
|
|
|
|
EXTERNC void *memset_sse3(void *dest, int c, size_t n)
|
|
{
|
|
return memset_sse2(dest, c, n);
|
|
}
|
|
|
|
EXTERNC void *memset_ssse3(void *dest, int c, size_t n)
|
|
{
|
|
return memset_sse2(dest, c, n);
|
|
}
|
|
|
|
EXTERNC void *memset_sse4_1(void *dest, int c, size_t n)
|
|
{
|
|
return memset_sse2(dest, c, n);
|
|
}
|
|
|
|
EXTERNC void *memset_sse4_2(void *dest, int c, size_t n)
|
|
{
|
|
#if defined(a64)
|
|
char *d = (char *)dest;
|
|
|
|
if (((uintptr_t)d & 0xF) == 0)
|
|
{
|
|
size_t num_vectors = n / 16;
|
|
for (size_t i = 0; i < num_vectors; i++)
|
|
{
|
|
asmv("movd %0, %%xmm0\n"
|
|
"pshufd $0, %%xmm0, %%xmm0\n"
|
|
"movdqa %%xmm0, (%1)\n"
|
|
:
|
|
: "r"(c), "r"(d)
|
|
: "xmm0");
|
|
d += 16;
|
|
}
|
|
n -= num_vectors * 16;
|
|
}
|
|
|
|
memset_unsafe(d, c, n);
|
|
#endif
|
|
return dest;
|
|
}
|