(C语言)memcpy函数原型的实现
在网上看到一道题,实现一个memcpy函数,于是查了一下memcpy的函数原型,如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | 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,解决办法如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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]; } } |