首页 > 其他分享> > memcpy()函数实现

memcpy()函数实现

2020-09-14 19:34:51 作者：互联网

1.为什么会写memcpy

笔试中遇到过一道笔试题，题目要求实现一个my_memcpy函数。函数原型：void * my_memcpy(void *dst, const void *src, int n);

之前使用的内存拷贝函数是标准库memcpy函数，拿来就用。当你自己在实现过程中，了解的越多，实现起来就越麻烦。

2.按字节(Byte)拷贝实现的memcpy

不考虑写覆盖，简单粗暴：

void* my_memcpy_byte(void* dst, const void* src, size_t n) {
    // Copies n bytes from src to dst
     
    // Since we cannot dereference a void* ptr,
    // we first typecast it to a char* ptr
    // and then do the copying byte by byte,
    // since a char* ptr references a single byte
    char* char_dst = (char*) dst;
    char* char_src = (char*) src;
 
    for (int i=0; i<n; i++) {
        *char_dst++ = *char_src++;
    }
 
    return dst;
}

考虑写覆盖：

void *my_memcpy_byte(void *dst, const void *src, int n)
{
    if (dst == NULL || src == NULL || n <= 0)
        return NULL;

    char * pdst = (char *)dst;
    char * psrc = (char *)src;

    if (pdst > psrc && pdst < psrc + n)
    {
        pdst = pdst + n - 1;
        psrc = psrc + n - 1;
        while (n--)
            *pdst-- = *psrc--;
    }
    else
    {
        while (n--)
            *pdst++ = *psrc++;
    }
    return dst;
}　　　　
//*pdst-- = *psrc--;  查了下运算符优先级（*,--）优先级相同，从右向左结合，psrc--是先使用，后减减　　　　
//等价于*pdst = *psrc;psrc--;pdst--;

3.按4字节拷贝实现的memcpy

void *my_memcpy(void *dst, const void *src, int n)
{
if (dst == NULL || src == NULL || n <= 0)
return NULL;

int * pdst = (int *)dst;
int * psrc = (int *)src;
char *tmp1 = NULL;
char *tmp2 = NULL;
int c1 = n / 4;
int c2 = n % 4;

/*if (pdst > psrc && pdst < psrc + n) 这样判断有问题*/
if (pdst > psrc && pdst < (char *)psrc + n)
{
tmp1 = (char *)pdst + n - 1;
tmp2 = (char *)psrc + n - 1;
while(c2--)
*tmp1-- = *tmp2--;
/*这样有问题，忘记字节偏移
pdst = (int *)tmp1;
psrc = (int *)tmp2;
*/
tmp1++;tmp2++;
pdst = (int *)tmp1;
psrc = (int *)tmp2;
pdst--;psrc--;
while (c1--)
*pdst-- = *psrc--;
}
else
{
while (c1--)
*pdst++ = *psrc++;3536 tmp1 = (char *)pdst;
tmp2 = (char *)psrc;
while (c2--)
*tmp1++ = *tmp2++;
}
return dst;
}　　　　　　