本文轉自：http://my.oschina.net/renhc/blog/36345
面試中如問到memcpy的實現，那就要小心了，這里有陷阱。
先看下標准memcpy()的解釋：
|
1
2
|
void
*
memcpy
(
void
*dst,
const
void
*src,
size_t
n);
//If copying takes place between objects that overlap, the behavior is undefined.
|
注意下面的注釋，對於地址重疊的情況，該函數的行為是未定義的。
事實上所說的陷阱也在於此，自己動手實現memcpy()時就需要考慮地址重疊的情況。
另外，標准庫也提供了地址重疊時的內存拷貝函數：memmove()，那么為什么還要考慮重寫memcpy()函數呢?
因為memmove()函數的實現效率問題，該函數把源字符串拷貝到臨時buf里，然后再從臨時buf里寫到目的地址，增加了一次不必要的開銷。
下面給出memcpy()的實現，為了與標准庫函數區分，我們實現其包裹函數：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
void
*Memcpy(
void
*dst,
const
void
*src,
size_t
size);
int
main(
int
argc,
char
*argv[])
{
char
buf[100] =
"abcdefghijk"
;
//memcpy(buf+2, buf, 5);
Memcpy(buf+2, buf, 5);
printf
(
"%s\n"
, buf+2);
}
void
*Memcpy(
void
*dst,
const
void
*src,
size_t
size)
{
char
*psrc;
char
*pdst;
if
(NULL == dst || NULL == src)
{
return
NULL;
}
if
((src < dst) && (
char
*)src + size > (
char
*)dst)
// 自后向前拷貝
{
psrc = (
char
*)src + size - 1;
pdst = (
char
*)dst + size - 1;
while
(size--)
{
*pdst-- = *psrc--;
}
}
else
{
psrc = (
char
*)src;
pdst = (
char
*)dst;
while
(size--)
{
*pdst++ = *psrc++;
}
}
return
dst;
}
|