X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/05b7549a0f980efa33265a091a5174a78851ce05..0765e7c133eb393d8cd0292af444543c2edf8ccc:/lunaix-os/libs/klibc/string/mem.c

diff --git a/lunaix-os/libs/klibc/string/mem.c b/lunaix-os/libs/klibc/string/mem.c
index 0b93a7d..0856358 100755
--- a/lunaix-os/libs/klibc/string/mem.c
+++ b/lunaix-os/libs/klibc/string/mem.c
@@ -1,28 +1,30 @@
-#include <stdint.h>
 #include <klibc/string.h>
+#include <lunaix/types.h>
 
 void*
-memcpy(void* dest, const void* src, size_t num)
+memcpy(void* dest, const void* src, unsigned long num)
 {
-    uint8_t* dest_ptr = (uint8_t*)dest;
-    const uint8_t* src_ptr = (const uint8_t*)src;
-    for (size_t i = 0; i < num; i++) {
-        *(dest_ptr + i) = *(src_ptr + i);
-    }
+    if (!num)
+        return dest;
+    asm volatile("movl %1, %%edi\n"
+                 "rep movsb\n" ::"S"(src),
+                 "r"(dest),
+                 "c"(num)
+                 : "edi", "memory");
     return dest;
 }
 
 void*
-memmove(void* dest, const void* src, size_t num)
+memmove(void* dest, const void* src, unsigned long num)
 {
-    uint8_t* dest_ptr = (uint8_t*)dest;
-    const uint8_t* src_ptr = (const uint8_t*)src;
+    u8_t* dest_ptr = (u8_t*)dest;
+    const u8_t* src_ptr = (const u8_t*)src;
     if (dest_ptr < src_ptr) {
-        for (size_t i = 0; i < num; i++) {
+        for (unsigned long i = 0; i < num; i++) {
             *(dest_ptr + i) = *(src_ptr + i);
         }
     } else {
-        for (size_t i = num; i != 0; i--) {
+        for (unsigned long i = num; i != 0; i--) {
             *(dest_ptr + i - 1) = *(src_ptr + i - 1);
         }
     }
@@ -30,21 +32,22 @@ memmove(void* dest, const void* src, size_t num)
 }
 
 void*
-memset(void* ptr, int value, size_t num)
+memset(void* ptr, int value, unsigned long num)
 {
-    uint8_t* c_ptr = (uint8_t*)ptr;
-    for (size_t i = 0; i < num; i++) {
-        *(c_ptr + i) = (uint8_t)value;
-    }
+    asm volatile("movl %1, %%edi\n"
+                 "rep stosb\n" ::"c"(num),
+                 "r"(ptr),
+                 "a"(value)
+                 : "edi", "memory");
     return ptr;
 }
 
 int
-memcmp(const void* ptr1, const void* ptr2, size_t num)
+memcmp(const void* ptr1, const void* ptr2, unsigned long num)
 {
-    uint8_t* p1 = (uint8_t*)ptr1;
-    uint8_t* p2 = (uint8_t*)ptr2;
-    for (size_t i = 0; i < num; i++) {
+    u8_t* p1 = (u8_t*)ptr1;
+    u8_t* p2 = (u8_t*)ptr2;
+    for (unsigned long i = 0; i < num; i++) {
         int diff = *(p1 + i) - *(p2 + i);
         if (diff != 0) {
             return diff;