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