锐客网

  1. 首页 > it技术 > >

linux|memcpy 函数实现

linuxc


memcpy函数的作用：
将由src指向地址为起始地址的连续n个字节的数据复制到以dest指向地址为起始地址的空间内，函数返回一个指向dest的指针。

想必大多数人在面试时被要求写 memcpy的实现，很不幸，我也吃过这个亏(这种题要是写的一塌糊涂后面完全没戏)，所以还是得提前准备一下，不然就只能呵呵了。

先来看看一段错误的示范: 找茬


void * memcpy(void *dest, const void *src, unsigned int count);
{
if ((src =https://www.it610.com/article/= NULL) || (dest == NULL))
return;

while (count--) *dest++ = *src++;

return dest;
}

特别说明：
1、与strcpy相比，memcpy遇到'\0'不会结束，而是一定要拷贝完n个字节，所以要指定拷贝的数据长度
2、memcpy可以拷贝任何数据类型的对象，如果dest和src的指针类型不一样，也需要处理，不能直接++使地址自增（例如: int* p和 char*q， p++指针的值是4个4个加(0,4,8)，q++是1个1个加(0,1,2,3,4)）
3、如果dest本身就有数据，执行memcpy()之后会覆盖原有的数据,所以src和dest所指向的内存区域不能有重叠
4、不能改变形参的值，定义新的临时变量来操作
5、参数提供的地址可能为空

完善后的代码：
void *memcpy(void *dest, const void *src, size_t count)
{
if (dest == NULL || src =https://www.it610.com/article/= NULL )
{
return NULL;
}

char *tmp_dest = dest;
const char *tmp_src = https://www.it610.com/article/src;

while (count--) *tmp_dest++ = *tmp_src++ ;

return dest;
}

这样写的代码语法上看起来没什么问题了，但是不要忘记 src和dest所指向的内存区域不能有重叠，所以上面这段代码还是有bug。这个算是不容易看出来的bug，如果这样写了以后代码出现问题就很难排除了，这也就是为什么写不好这个函数指定会被淘汰的原因了。


dest和src所指向的地址有重叠的情况
【linux|memcpy 函数实现】

内存地址重叠的情况分为两种，第一种是dest的地址在src地址的后面，另一种则是dest地址在src地址的前面。

这两种情况在memcpy中都是不允许出现的，需要在代码中处理以避免。
为了处理上面这两种情况，后来又提供了另一个函数memmove，在不需要保留原来内存区域的数据的时候可以使用memmove。

最终的正确的版本：

void *memcpy(void *dest, const void *src, size_t count)
{
if (dest == NULL || src =https://www.it610.com/article/= NULL || dest <= src+count)
{
return NULL;
}

char *tmp_dest = dest;
const char *tmp_src = https://www.it610.com/article/src;

while (count--) *tmp_dest++ = *tmp_src++ ;

return dest;
}

上面就是正确的版本了，我所能考虑到的问题都处理了。

另外，这段代码还可以优化，比如根据CPU的字节长度，把原来一个一个字节拷贝转换为按CPU的长度拷贝，等等。
感兴趣的话自己研究一下吧，这里就不介绍了。

    推荐阅读


    上一篇：编程语言|为什么依赖注入对于程序员来说是件好事（）

    下一篇：rtmpdump传输直播流 -- pause and resume