1. 程式人生 > >strcpy、memcpy函式的實現

strcpy、memcpy函式的實現

本文轉載自:strcpy函式實現
大家一般認為名不見經傳strcpy函式實現不是很難,流行的strcpy函式寫法是:

char *my_strcpy(char *dst,const char *src)
{
    assert(dst != NULL);
    assert(src != NULL);
    char *ret = dst;
    while((* dst++ = * src++) != '\0') 
        ;
    return ret;
}

如果注意到:
1,檢查指標有效性;

2,返回目的指標des;

3,源字串的末尾 ‘\0’ 需要拷貝。

寫出上面實現函式就不在話下。

然而這樣的實現沒有考慮拷貝時記憶體重疊的情況,下面的測試用例就能使呼叫my_strcp函式的程式崩潰:

char str[10]="abc";
my_strcpy(str+1,str);

然而呼叫系統的strcpy函式程式正常執行,列印str結果為“aabc”!可見系統strcpy函式的實現不是這樣的。
strcpy的正確實現應為:

char *my_strcpy(char *dst,const char *src)
{
    assert(dst != NULL);
    assert(src != NULL);
    char *ret = dst;
    memcpy(dst,src,strlen(src)+1
); return ret; }

memcpy函式實現時考慮到了記憶體重疊的情況,可以完成指定大小的記憶體拷貝,這裡僅粘帖函式memcpy函式的實現:

void * my_memcpy(void *dst,const void *src,unsigned int count)
{
     assert(dst);
     assert(src);
     void * ret = dst;
     if (dst <= src || (char *)dst >= ((char *)src + count))//源地址和目的地址不重疊,低位元組向高位元組拷貝
     {
         while
(count--) { *(char *)dst = *(char *)src; dst = (char *)dst + 1; src = (char *)src + 1; } } else //源地址和目的地址重疊,高位元組向低位元組拷貝 { dst = (char *)dst + count - 1; src = (char *)src + count - 1; while(count--) { *(char *)dst = *(char *)src; dst = (char *)dst - 1; src = (char *)src - 1; } } return ret; }

兩者結合才是strcpy函式的真正實現吧。