天天看点

【面试题,纸上写程序】实现memcpy函数

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

先看下标准memcpy()的解释:注意下面的注释,对于地址重叠的情况,该函数的行为是未定义的。

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

另外,标准库也提供了地址重叠时的内存拷贝函数:memmove(),那么为什么还要考虑重写memcpy()函数呢?因为memmove()函数的实现效率问题,该函数把源字符串拷贝到临时buf里,然后再从临时buf里写到目的地址,增加了一次不必要的开销。

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

1. void*Memcpy(void *dst, const void *src, size_t size);
2. 
3. int main(intargc, char *argv[])
4. {
5. char buf[100] = "abcdefghijk";
6. memcpy(buf+2, buf, 5);
7. //Memcpy(buf+2, buf, 5);
8. printf("%s\n", buf+2);
9. 
10. return 0;
11. }
12. 
13. void*Memcpy(void *dst, const void *src, size_t size)
14. {
15. char *psrc;
16. char *pdst;
17. 
18. if(NULL == dst || NULL == src)
19.     {
20. return NULL;
21.     }
22. 
23. if((src < dst) && (char *)src +size > (char *)dst) // 自后向前拷贝
24.     {
25.         psrc = (char *)src + size - 1;
26.         pdst = (char *)dst + size - 1;
27. while(size--)
28.         {
29.             *pdst-- = *psrc--;
30.         }
31.     }
32. else
33.     {
34.         psrc = (char *)src;
35.         pdst = (char *)dst;
36. while(size--)
37.         {
38.             *pdst++ = *psrc++;
39.         }
40.     }
41. 
42. return dst;
43. }      

使用Memcpy()的结果: abcdehijk

使用memcpy()的结果: abcdehijk,//在VC6.0中仍然输出abcdehijk,可能的原因就是:有的函数库实现的memcpy函数考虑到了地址重叠,所以不会有错,而另外一些函数库并没有考虑到。

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

继续阅读