(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]; } }