memcpy的实现,考虑重叠地址空间。

来源:互联网 发布:泰勒 抖森 知乎 编辑:程序博客网 时间:2024/06/06 16:30

软件工程师在面试中,经常被要求在白板上写一些小程序来检查程序员的代码书写风格及对细节的处理能力,memcpy就是经常被问到的一个。下面有两个版本的memcpy,第一个没有考虑目标地址空间和源地址空间重叠的问题,第二个则把这个问题考虑了进去。


//version 1, no memory overlap is considered

void *memcpy(void *dst, const void *src, size_t size)

{
 
   //checkargument
    assert(null!= dst && null != src);
    
    byte *pDst =(byte*)dst;
    const byte*pSrc = (byte*)src;
    
    while(size--> 0)
    {
      *pDst++ = *pSrc++;
    }
    
    returndst;

}


//version 2, memory overlap is considered

void *memcpy(void *dst, const void *src, size_t size)
{
    //checkargument
    assert(null!= dst && null != src);
    
    if (dst< src)
    {
      byte *pDst = (byte*)dst;
      const byte *pSrc = (const byte*)src;
      
      while(size-- > 0)
      {
         *pDst++ = *pSrc++;
      }
    }
    else if (dst> src)
    {
      byte *pDst = (byte*)dst+size-1;
      const byte *pSrc = (const byte*)src+size-1;
      
      while(size-- > 0)
      {
         *pDst-- = *pSrc--;
      }
    }
    
    returndst;
}

0 0
原创粉丝点击