面試中memcopy函數實現中的陷阱!

先看下標準memcopy。其函數原型爲

 void memcopy(void *dst, const void *src, size_t n);
// if copying takes place between objects that overlap, the behavior is undefined.



通過閱讀上圖中的“Description”會發現我們需要注意的一個點是:注意考慮src 和dest 的內存存在overlap的情況!

overlap分爲4中情況:


其中情況1,2,4對應下述代碼的else部分,情況3對應於if部分!

另外,標準庫提供了地址重疊時的內存拷貝函數:memmove(),那麼爲什麼還要考慮重寫memcpy()函數呢?因爲memmove()函數的實現效率問題,該函數把源字符串拷貝到臨時buf中,然後從臨時buf裏再拷貝到目的地址,增加了一次不必要的開銷。具體的可以使用命令: man memmove函數看詳細解釋!如下:

#include<assert.h>
#include<cstddef>
#include<iostream>
void *memcpy(void* dst,const void *src,size_t num)
{
    assert((dst!=NULL)&&(src!=NULL));
    size_t word_num = num/sizeof(long);
    size_t slice = num%sizeof(long);
    if((src<dst)&&((char*)src+num > (char*)dst))//對應於情況3
    {
        if(word_num>0)                              //要拷貝的空間大於等於一個long的情況!
        {
            long *psrc = (long*)((char*)src+num-sizeof(long));
            long *pdst = (long*)((char*)dst+num-sizeof(long));
            while(word_num--)
                *pdst-- = *psrc--;
            while(slice--)
                *((char*)pdst--) = *((char*)psrc--);
        }
        else                                        //要拷貝的空間小於一個long的情況!
        {
            char *psrc = (char*)src+num-1;
            char *pdst = (char*)dst+num-1;
            while(slice--)
                *((char*)pdst--) = *((char*)psrc--);
        }
    }
    else                                        //對應於情況1,2,4
    {
        long *pdst = (long*)dst;
        long *psrc = (long*)src;
        while(word_num--)
            *pdst++ = *psrc++;
        while(slice--)
            *((char*)pdst++) = *((char*)psrc++);
    }
    return dst;
}

int main()
{
    char src[101]="1234567890";
    char dst[101];
    memcpy(dst,src,10);
    for(int i=0;i<101;i++)
    {
        std::cout<<src[i]<<" ";
    }
}

另外考慮效率的問題:我們需要考慮是將void*轉換爲何種類型的指針!因爲如果是char*型的指針那麼每次只拷貝一個字節,但是如果是long型的,每次就會拷貝8個字節!

這就是需要考慮的兩個點,重新將其列出來:

1.注意考慮src 和dest 的內存存在overlap的情況!

2.考慮效率的問題:我們需要考慮是將void*轉換爲何種類型的指針!

基於這兩點的代碼實現如下:


待添加的內容:

1.單元測試,使用boost的單元測試進行操作!

2.添加性能測試,來驗證一次copy一個char和一次copy一個long在性能上的區別!以及差距有幾倍?


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章