實現memcpy()函數及過程總結

本文轉載自查看原文 2018-02-13 21:53 12292 C

1.為什么會寫memcpy

在之前的應聘筆試上遇到一道筆試題，題目要求實現一個my_memcpy函數。函數原型：void * my_memcpy(void *dst, const void *src, int n);

之前使用的內存拷貝函數是標准庫memcpy函數，拿來就用，真沒有對這個函數做過多了解。在網上查了一下，有好多關於memcpy函數優化的文章。

在實現過程中了解的越多，往往實現起來越麻煩。還是先實現簡單的memcpy函數。

2.按字節(Byte)拷貝實現的memcpy

 1 void *my_memcpy_byte(void *dst, const void *src, int n)
 2 {
 3     if (dst == NULL || src == NULL || n <= 0)
 4         return NULL;
 5 
 6     char * pdst = (char *)dst;
 7     char * psrc = (char *)src;
 8 
 9     if (pdst > psrc && pdst < psrc + n)
10     {
11         pdst = pdst + n - 1;
12         psrc = psrc + n - 1;
13         while (n--)
14             *pdst-- = *psrc--;
15     }
16     else
17     {
18         while (n--)
19             *pdst++ = *psrc++;
20     }
21     return dst;
22 }
　　　　//20200104  看到評論，又看了下之前寫的memcpy實現
　　　　//按字節拷貝實現的memcpy沒有問題
　　　　//*pdst-- = *psrc--;  查了下運算符優先級（*,--）優先級相同，從右向左結合，psrc--是先使用，后減減
　　　　//等價於*pdst = *psrc;psrc--;pdst--;

這里要考慮寫覆蓋的情況

3.按4字節拷貝實現的memcpy

 1 void *my_memcpy(void *dst, const void *src, int n)
 2 {
 3     if (dst == NULL || src == NULL || n <= 0)
 4         return NULL;
 5 
 6     int * pdst = (int *)dst;
 7     int * psrc = (int *)src;
 8     char *tmp1 = NULL;
 9     char *tmp2 = NULL;
10     int c1 = n / 4;
11     int c2 = n % 4;
12 
13     /*if (pdst > psrc && pdst < psrc + n) 這樣判斷有問題*/
14     if (pdst > psrc && pdst < (char *)psrc + n)
15     {
16         tmp1 = (char *)pdst + n - 1;
17         tmp2 = (char *)psrc + n - 1;
18         while(c2--)
19             *tmp1-- = *tmp2--;
20         /*這樣有問題，忘記字節偏移
21         pdst = (int *)tmp1;
22         psrc = (int *)tmp2;
23         */
24         tmp1++;tmp2++;
25         pdst = (int *)tmp1;
26         psrc = (int *)tmp2;
27         pdst--;psrc--;
28         while (c1--)
29             *pdst-- = *psrc--;
30     }
31     else
32     {
33         while (c1--)
34             *pdst++ = *psrc++;
35
36         tmp1 = (char *)pdst;
37         tmp2 = (char *)psrc;
38         while (c2--)
39             *tmp1++ = *tmp2++;
40     }
41     return dst;
42 }
　　　　　　//20200104 查看評論說四字節寫覆蓋拷貝問題，現在已修改
　　　　　　//備注：void *dst, const void *src這兩個參數是需要按4字節對齊的，如果本身不是4字節對齊，按4字節拷貝效率也會變低。