C++手写memcpy,注意内存重叠问题。

内存区域重叠是指内存区域 [src,src+count)和内存区域 [dst, dst+count)有重叠。memcpy只是简单的从前往后拷贝,没有考虑内存有重叠的情况,如果内存有重叠,其行为是不确定的。
memmove()函数在memcpy()函数的基础上加入了对内存重叠拷贝的处理;引入了倒序拷贝的方式处理内存重叠的某些情况;保证拷贝的正确性;
一般来说,如果自己能确定拷贝内存没有重叠,用 memcpy比用 memmove更高效,如果有重叠,则只能用 memmove

内存重叠的两种情况
C++手写memcpy,注意内存重叠问题。_第1张图片

void *memcpy(void *dst, const void *src, size_t count);
void *memmove(void *dst, const void *src, size_t count);

memcpy()函数从src内存中拷贝n个字节到dest内存区域,但是源和目的的内存区域不能重叠。
memmove() 函数从src内存中拷贝n个字节到dest内存区域,但是源和目的的内存可以重叠。

void* my_memcpy(void* dst, const void* src, size_t n)
{
    char *tmp = (char*)dst;
    char *s_src = (char*)src;
 
    while(n--) {
        *tmp++ = *s_src++;
    }
    return dst;
}

从实现中可以看出memcpy()是从内存左侧一个字节一个字节地将src中的内容拷贝到dest的内存中,这种实现方式导致了对于图中第一种内存重叠情形下,后面数字的拷贝值明显不是原先的值了,新的值是变成了src的最开始的几个数字了。而对于第二种内存覆盖情况,memcpy的这种拷贝方式是可以的。

而memmove就是针对第二种内存覆盖情形,对memcpy进行了改进,改进代码如下:

void* my_memmove(void* dst, const void* src, size_t n)
{
    char* s_dst;
    char* s_src;
    s_dst = (char*)dst;
    s_src = (char*)src;
    if(s_dst > s_src && (s_src+n > s_dst)) {      //第二种内存覆盖的情形。
        s_dst = s_dst+n-1;
        s_src = s_src+n-1;
        while(n--) {
            *s_dst-- = *s_src--;
        }
    }else {
        while(n--) {
            *s_dst++ = *s_src++;
        }
    }
    return dst;
}

你可能感兴趣的:(c++,算法,c语言,memcpy)