#include <iostream>
using namespace std;
void* memmove(void *dest, const void *src, size_t n)
{
assert((dest!=NULL) || (src!=NULL));
char *tmp,*s;
cout << dest << " " << src << endl;
if(dest<=src)//判别内存区域是否重叠
{//内存区域没有重叠现象,直接从起始地址复制字符
tmp = (char*) dest;
s = (char*) src;
while(n--)
*tmp++=*s++;
}
else//内存重叠情况
{//内存区域重叠,从最后地址的一个字符开始复制
tmp = (char*) dest + n;
s = (char*) src + n;
while(n--)
*--tmp = *--s;
}
return dest;
}
int main(){
char *p="hello,world";
char dest[6]={0};
char *q = (char*)memmove(dest, p, 5);
cout << q << endl;
return 0;
}
一开始没有考虑很多问题,没有考虑assert,以后大代码记得多放点assert,安全多了,而且看起来爽呀!
第二点是没有考虑内存重叠现象,要考虑上内存重叠现象,所以要判断else情况,对比dest和src,如果dest的内存地址比src的内存地址大,那么可能会出现内存重叠显现,就要从后面往前复制。主要是为了避免第三种情况,所以从后面来。
下面是VC里的memmove
void * __cdecl memmove ( void * dst, const void * src, size_t count )
{
void * ret = dst;
if (dst <= src || (char *)dst >= ((char *)src + count)) {
/*
* Non-Overlapping Buffers
* copy from lower addresses to higher addresses
*/
while (count--) {
*(char *)dst = *(char *)src;
dst = (char *)dst + 1;
src = (char *)src + 1;
}
}
else {
/*
* Overlapping Buffers
* copy from higher addresses to lower addresses
*/
dst = (char *)dst + count - 1;
src = (char *)src + count - 1;
while (count--) {
*(char *)dst = *(char *)src;
dst = (char *)dst - 1;
src = (char *)src - 1;
}
}
return(ret);