面试中如问到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函数考虑到了地址重叠,所以不会有错,而另外一些函数库并没有考虑到。
可以看到标准库函数的源字符串在拷贝的过程中被污染了。