仰视源码，实现memmove

小咪咪 2022-08-03 00:27 222阅读 0赞

在我之前博客中写过一个内存复制函数memcopy（[http://blog.csdn.net/z702143700/article/details/46628149][http_blog.csdn.net_z702143700_article_details_46628149]），是根据源码修改而来，但是这个函数包括源码中都没有考虑一个问题，就是当memTo和memFrom有重叠的情况。  
今天就根据源码实现memmove函数，感受下底层源码的极致实现。

void* memmove(void* dst,const void* src,size_t count)
    {
        void* ret = dst;
        //dst <= src表示，如果dst在src的前面，从前往后复制不会覆盖src中还没有复制的内容
    
        if (dst <= src || (char*)dst >= ((char*)src + count))
        {
            //从前往后复制，则不会出现覆盖src中没有复制的内容
            while(count--)
            {
                *(char*)dst = *(char*)src; //char类型指针，表示一个字节一个字节的复制
                dst = (char*)dst + 1; //移动一个字节
                src = (char*)src + 1;
            }
        }
        else
        {
            //从后往前复制，则不会出现覆盖src中没有复制的内容
            dst = (char*)dst + count - 1;//移动到末尾
            src = (char*)src + count - 1;
            while(count--)
            {
                *(char*)dst = *(char*)src;
                dst = (char*)dst - 1; //移动一个字节
                src = (char*)src - 1;
            }
        }
        //返回dst的头指针，还方便左值操作。
        //如：ptstr = memmove(ptstr,src,count); cout << memmove(ptstr,src,count);
        return ret; 
    }

有人认为`dst <= src || (char*)dst >= ((char*)src + count)`是表示dst和src区域没有重叠的情况，其实是错误的想法。因为这种情况即使重叠也不会影响复制。如下图中上图所示，当dst复制到第4第5字节时已经覆盖了src的开始第1第2字节，但是并不影响复制。  
下图则表示的是`dst > src || (char*)dst < ((char*)src + count)`的情况，从后往前复制也不会影响复制结果。  
![这里写图片描述][20150728163849779]

很显然，上面类似源代码的实现存在多次类型转换，所以，它还是可以进行优化的，优化后的代码如下：

void* memmove(void* dst,const void* src,size_t count)
    {
        assert(NULL !=src && NULL !=dst);
        char* tmpdst = (char*)dst;
        char* tmpsrc = (char*)src;
    
        if (tmpdst <= tmpsrc || tmpdst >= tmpsrc + count)
        {
            while(count--)
            {
                *tmpdst++ = *tmpsrc++; 
            }
        }
        else
        {
            tmpdst = tmpdst + count - 1;
            tmpsrc = tmpsrc + count - 1;
            while(count--)
            {
                *tmpdst-- = *tmpsrc--;
            }
        }
    
        return dst; 
    }

最后，memcpy在内存没有重复的情况下能够正确复制，若有重叠情况则复制结果错误，但是它的效率比memmove高。所以，在确定没有重复的情况下用memcpy，在不确定是否有内存重复的情况用memmove。

[http_blog.csdn.net_z702143700_article_details_46628149]: http://blog.csdn.net/z702143700/article/details/46628149
[20150728163849779]: /images/20220731/ae8409a1050c4a1ab373e249e58d84ac.png