StdLibExtras.cpp 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. #include <AK/StdLibExtras.h>
  2. #include <AK/Assertions.h>
  3. #include <AK/Types.h>
  4. #include <AK/kstdio.h>
  5. void* mmx_memcpy(void* dest, const void* src, size_t len)
  6. {
  7. ASSERT(len >= 1024);
  8. auto* dest_ptr = (byte*)dest;
  9. auto* src_ptr = (const byte*)src;
  10. if ((dword)dest_ptr & 7) {
  11. dword prologue = 8 - ((dword)dest_ptr & 7);
  12. len -= prologue;
  13. asm volatile(
  14. "rep movsb\n"
  15. : "=S"(src_ptr), "=D"(dest_ptr), "=c"(prologue)
  16. : "0"(src_ptr), "1"(dest_ptr), "2"(prologue)
  17. : "memory"
  18. );
  19. }
  20. for (dword i = len / 64; i; --i) {
  21. asm volatile(
  22. "movq (%0), %%mm0\n"
  23. "movq 8(%0), %%mm1\n"
  24. "movq 16(%0), %%mm2\n"
  25. "movq 24(%0), %%mm3\n"
  26. "movq 32(%0), %%mm4\n"
  27. "movq 40(%0), %%mm5\n"
  28. "movq 48(%0), %%mm6\n"
  29. "movq 56(%0), %%mm7\n"
  30. "movq %%mm0, (%1)\n"
  31. "movq %%mm1, 8(%1)\n"
  32. "movq %%mm2, 16(%1)\n"
  33. "movq %%mm3, 24(%1)\n"
  34. "movq %%mm4, 32(%1)\n"
  35. "movq %%mm5, 40(%1)\n"
  36. "movq %%mm6, 48(%1)\n"
  37. "movq %%mm7, 56(%1)\n"
  38. :: "r" (src_ptr), "r" (dest_ptr) : "memory");
  39. src_ptr += 64;
  40. dest_ptr += 64;
  41. }
  42. asm volatile("emms":::"memory");
  43. // Whatever remains we'll have to memcpy.
  44. len %= 64;
  45. if (len)
  46. memcpy(dest_ptr, src_ptr, len);
  47. return dest;
  48. }