面試中如問到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函數考慮到了位址重疊,是以不會有錯,而另外一些函數庫并沒有考慮到。
可以看到标準庫函數的源字元串在拷貝的過程中被污染了。