(C语言)memcpy函数原型的实现

时间:2023-03-08 22:01:26
(C语言)memcpy函数原型的实现

在网上看到一道题,实现一个memcpy函数,于是查了一下memcpy的函数原型,如下:

void* memcpy(char *strDest, const char *strSrc, int Count)
{
if(strDest == NULL || strSrc == NULL || Count <= 0) return NULL;
unsigned char *tempDest;
unsigned char *tempSrc;
tempDest = (unsigned char *)tempDest;
tempSrc = (unsigned char *)tempSrc;
while(Count--)
{
*tempDest = *tempSrc;
tempDest++;
tempSrc++;
}
return strDest;
}

该函数原型实现中并没有考虑两个问题:

1、strDest与strSrc内存覆盖的问题;

2、Count的大小问题,正常的话Count的大小应该满足大于0而小于strSrc指向资源的大小,且小于strDest指向资源的大小。

对于问题1,解决办法如下:

void my_memcpy(void *dst,const void *src,size_t num){
//assert((dst!=NULL) && (src!=NULL) && (num>=0))
if(dst==NULL || src==NULL || num<0)
return;
if(dst==src)
return;
// convert to char type
char* pdst=(char*)dst;
const char* psrc=(const char*)src;
// memory overlap
if(pdst>psrc && pdst<psrc+num){
for(int i=num-1;i>=0;i--)
pdst[i]=psrc[i];
}
else{
for(size_t i=0;i<num;i++)
pdst[i]=psrc[i];
}
}