面试中如问到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 |
可以看到标准库函数的源字符串在拷贝的过程中被污染了。