memcpy内存复制代码的安全版本

C代码关于memcpy复制内存函数的安全版本

#include 
#include 
#include 

void *Memcpy(void *dst, const void *src, size_t size);

int main(int argc, char *argv[])
{
    char buf[100] = "vincentabcdefghijk";
    //memcpy(buf+5, buf, 7);
    Memcpy(buf+5, buf, 7);
    printf("%s\n", buf);
}

/**
内存复制函数,将src内存地址开始的size个字节复制到dst内存开始的地方
假如dst地址在src地址到(char *)src + size地址之间,那么地址有重叠,那么将覆盖原来src中dst地址后面
的元素

dst不在src地址到(char *)src + size地址之间,则不会发生覆盖
*/
void *Memcpy(void *dst, const void *src, size_t size)
{
    const char *psrc;
    char *pdst;

    if(NULL == dst || NULL == src)
    {
        return NULL;
    }

    /**
    dst 在src到src+count之间的内存空间重叠区域
    */
    if((src < dst) && ((char *)src + size) > (char *)dst) // 如果有地址重叠时的内存拷贝,自后向前拷贝
    {
        psrc = (char *)src + size - 1;
        pdst = (char *)dst + size - 1;
        while(size--)
        {
            *pdst-- = *psrc--;
        }
    }
    else
    {
        psrc = (char *)src;
        pdst = (char *)dst;
        while(size--)
        {
            *pdst++ = *psrc++;
        }
    }

    return pdst;
}

你可能感兴趣的:(C/C++,面试,C进阶教程)