memcpy函数

面试中如问到memcpy的实现，那就要小心了，这里有陷阱。

先看下标准memcpy()的解释：

1void *memcpy(void *dst, const void *src, size_t n);

2//If copying takes place between objects that overlap, the behavior is undefined.

注意下面的注释，对于地址重叠的情况，该函数的行为是未定义的。

事实上所说的陷阱也在于此，自己动手实现memcpy()时就需要考虑地址重叠的情况。

另外，标准库也提供了地址重叠时的内存拷贝函数：memmove()，那么为什么还要考虑重写memcpy()函数呢?

因为memmove()函数的实现效率问题，该函数把源字符串拷贝到临时buf里，然后再从临时buf里写到目的地址，增加了一次不必要的开销。

下面给出memcpy()的实现，为了与标准库函数区分，我们实现其包裹函数：

01#include <stdio.h>

02#include <stdlib.h>

03#include <string.h>

04 

05void *Memcpy(void *dst, const void *src, size_t size);

06 

07int main(int argc, char *argv[])

08{

09    char buf[100] = "abcdefghijk";

10    //memcpy(buf+2, buf, 5);

11    Memcpy(buf+2, buf, 5);

12    printf("%s\n", buf+2);

13}

14 

15void *Memcpy(void *dst, const void *src, size_t size)

16{

17    char *psrc;

18    char *pdst;

19 

20    if(NULL == dst || NULL == src)

21    {

22        return NULL;

23    }

24 

25    if((src < dst) && (char *)src + size > (char *)dst) // 自后向前拷贝

26    {

27        psrc = (char *)src + size - 1;

28        pdst = (char *)dst + size - 1;

29        while(size--)

30        {

31            *pdst-- = *psrc--;

32        }

33    }

34    else

35    {

36        psrc = (char *)src;

37        pdst = (char *)dst;

38        while(size--)

39        {

40            *pdst++ = *psrc++;

41        }

42    }

43 

44    return dst;

45}

使用Memcpy()的结果:

1abcdehijk

使用memcpy()的结果:

1abadehijk

可以看到标准库函数的源字符串在拷贝的过程中被污染了。