天天看點

【面試題,紙上寫程式】實作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函數考慮到了位址重疊,是以不會有錯,而另外一些函數庫并沒有考慮到。

可以看到标準庫函數的源字元串在拷貝的過程中被污染了。

繼續閱讀