面试中如问到memcpy的实现,那就要小心了,这里有陷阱。
先看下标准memcpy()的解释:
1 |
void * memcpy ( void *dst, const void *src, size_t n); |
2 |
//If
copying takes place between objects that overlap, the behavior is undefined. |
注意下面的注释,对于地址重叠的情况,该函数的行为是未定义的。
事实上所说的陷阱也在于此,自己动手实现memcpy()时就需要考虑地址重叠的情况。
另外,标准库也提供了地址重叠时的内存拷贝函数:memmove(),那么为什么还要考虑重写memcpy()函数呢?
因为memmove()函数的实现效率问题,该函数把源字符串拷贝到临时buf里,然后再从临时buf里写到目的地址,增加了一次不必要的开销。
下面给出memcpy()的实现,为了与标准库函数区分,我们实现其包裹函数:
01 |
#include
<stdio.h> |
02 |
#include
<stdlib.h> |
03 |
#include
<string.h> |
04 |
05 |
void *Memcpy( void *dst, const void *src, size_t size); |
06 |
07 |
int main( int argc, char *argv[]) |
08 |
{ |
09 |
char buf[100]
= "abcdefghijk" ; |
10 |
//memcpy(buf+2,
buf, 5); |
11 |
Memcpy(buf+2,
buf, 5); |
12 |
printf ( "%s\n" ,
buf+2); |
13 |
} |
14 |
15 |
void *Memcpy( void *dst, const void *src, size_t size) |
16 |
{ |
17 |
char *psrc; |
18 |
char *pdst; |
19 |
20 |
if (NULL
== dst || NULL == src) |
21 |
{ |
22 |
return NULL; |
23 |
} |
24 |
25 |
if ((src
< dst) && ( char *)src
+ size > ( char *)dst) //
自后向前拷贝 |
26 |
{ |
27 |
psrc
= ( char *)src
+ size - 1; |
28 |
pdst
= ( char *)dst
+ size - 1; |
29 |
while (size--) |
30 |
{ |
31 |
*pdst--
= *psrc--; |
32 |
} |
33 |
} |
34 |
else |
35 |
{ |
36 |
psrc
= ( char *)src; |
37 |
pdst
= ( char *)dst; |
38 |
while (size--) |
39 |
{ |
40 |
*pdst++
= *psrc++; |
41 |
} |
42 |
} |
43 |
44 |
return dst; |
45 |
} |
使用Memcpy()的结果:
1 |
abcdehijk |
使用memcpy()的结果:
1 |
abadehijk |
可以看到标准库函数的源字符串在拷贝的过程中被污染了。