这道题主要考察一下几点 1、对于参数非法的判断,空指针的判断和size大小的判断 2、源数据不能被改写 3、源数据和目标数据是否存在内存重叠
不考虑内存重叠的情况
char* memcpy(char* dst,const char* src,size_t count) { char* ptr = dst; if(dst == NULL || src == NULL || count <= 0) { return NULL; } while(count--) { *dst++ = *src++; } //不能返回dst的,因为dst指针被++了 return ptr; }考虑内存重叠的情况
char* memcpy(char* dst,const char* src,size_t count) { char* ptr = dst; if(dst == NULL || src == NULL || count <= 0) { return NULL; } //判断是否存在内存重叠 if(dst <= src || dst >= src+count) { //低地址先拷贝 while(count--) { *dst++ = *src++; } } else { //高地址先拷贝 dst = dst + count -1; src = src + count -1; while(count--) { *dst-- = *src--; } } return ptr; }函数参数count为size_t,是无符号类型,判断无符号数<0是没有任何意义的,如果传进来-1的数,会产生一个很大的无符号整型数据。所以要么在调用函数之前判断count是否为负数,要么memcpy参数由size_t改成int类型。