经常看到需要自己实现字符串拷贝函数,网上也有很多博文,但是很多实现都是忽略了一些问题。
具体看代码分析。
#include <iostream>
#include <assert.h>
using namespace std;
//这个写法是比较常见的写法,但是这样处理内存重叠情况会崩溃
char *MyStrcpy(char *des,const char *src)
{
assert(des!=NULL);
assert(src!=NULL);
char *res=des;
while ((*des++=*src++)!='\0')
{
}
return res;
}
char *my_strcpy(char *dst,const char *src)
{
assert(dst != NULL);
assert(src != NULL);
char *ret = dst;
memcpy(dst,src,strlen(src)+1);//把src中的'\0'也复制过去,strlen 只计算了字符串长度
return ret;
}
/*memcpy 从源src所指的内存地址的起始位置开始拷贝n个字节到目标dest所指的内存地址的起始位置中
strcpy和memcpy主要有以下3方面的区别。
1、复制的内容不同。strcpy只能复制字符串,而memcpy可以复制任意内容,例如字符数组、整型、结构体、类等。
2、复制的方法不同。strcpy不需要指定长度,它遇到被复制字符的串结束符"\0"才结束,所以容易溢出。memcpy则是根据其第3个参数决定复制的长度。
3、用途不同。通常在复制字符串时用strcpy,而需要复制其他类型数据时则一般用memcpy
*/
//自己实现memcpy函数
void * my_memcpy(void *dst,const void *src,unsigned int count)
{
assert(dst);
assert(src);
void * ret = dst;
//源地址和目的地址不重叠,低字节向高字节拷贝
if (dst <= src || (char *)dst >= ((char *)src + count))
{
while(count--)
{
*(char *)dst = *(char *)src;
dst = (char *)dst + 1;
src = (char *)src + 1;
}
}
else//源地址和目的地址重叠,高字节向低字节拷贝
{
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;
}
int main()
{
char str[10]="abc";
char *des=new char[10];
//cout<<my_strcpy(str+1,str)<<endl;//处理内存覆盖问题
cout<<str<<endl;
//cout<<MyStrcpy(str+1,str)<<endl;//程序出现崩溃
cout<<strcpy(str+1,str)<<endl;
cout<<str<<endl;
cout<<my_strcpy(str+1,str)<<endl;
cout<<str<<endl;
}