strcpy函數的實現

越來越感覺，如果一個程序員連C都不搞不明白，那還怎麼好意思說自己是個程序員！！！

轉自：點擊打開鏈接

已知strcpy函數的原型是：

char *strcpy(char *dst, const char *src);

1. 實現strcpy函數
2. 解釋爲什麼要返回char *
3. 假如考慮dst和src內存重疊的情況，strcpy該怎麼實現

 

1.strcpy的實現代碼

char * strcpy(char *dst,const char *src)   //[1]
{
    assert(dst != NULL && src != NULL);    //[2]

    char *ret = dst;  //[3]

    while ((*dst++=*src++)!='\0'); //[4]

    return ret;
}

[1]const修飾

源字符串參數用const修飾，防止修改源字符串。

[2]空指針檢查

(A)不檢查指針的有效性，說明答題者不注重代碼的健壯性。

(B)檢查指針的有效性時使用assert(!dst && !src);

char *轉換爲bool即是類型隱式轉換，這種功能雖然靈活，但更多的是導致出錯概率增大和維護成本升高。

(C)檢查指針的有效性時使用assert(dst != 0 && src != 0);

直接使用常量（如本例中的0）會減少程序的可維護性。而使用NULL代替0，如果出現拼寫錯誤，編譯器就會檢查出來。

[3]返回目標地址

(A)忘記保存原始的strdst值。

[4]'\0'

(A)循環寫成while (*dst++=*src++);明顯是錯誤的。

(B)循環寫成while (*src!='\0') *dst++=*src++;

循環體結束後，dst字符串的末尾沒有正確地加上'\0'。

 

2.爲什麼要返回char *？

返回dst的原始值使函數能夠支持鏈式表達式。

鏈式表達式的形式如：

int l=strlen(strcpy(strA,strB));

又如：

char * strA=strcpy(new char[10],strB);

返回strSrc的原始值是錯誤的。

其一，源字符串肯定是已知的，返回它沒有意義。

其二，不能支持形如第二例的表達式。

其三，把const char *作爲char *返回，類型不符，編譯報錯。

 

3.假如考慮dst和src內存重疊的情況，strcpy該怎麼實現

char s[10]="hello";

strcpy(s, s+1); //應返回ello，

//strcpy(s+1, s); //應返回hhello，但實際會報錯，因爲dst與src重疊了，把'\0'覆蓋了

所謂重疊，就是src未處理的部分已經被dst給覆蓋了，只有一種情況：src<=dst<=src+strlen(src)

C函數memcpy自帶內存重疊檢測功能，下面給出memcpy的實現my_memcpy。

char * strcpy(char *dst,const char *src)
{
    assert(dst != NULL && src != NULL);

    char *ret = dst;

    my_memcpy(dst, src, strlen(src)+1);

    return ret;
}

my_memcpy的實現如下

char *my_memcpy(char *dst, const char* src, int cnt)
{
    assert(dst != NULL && src != NULL);

    char *ret = dst; 

    if (dst >= src && dst <= src+cnt-1) //內存重疊，從高地址開始複製
    {
        dst = dst+cnt-1;
        src = src+cnt-1;
        while (cnt--)
            *dst-- = *src--;
    }
    else    //正常情況，從低地址開始複製
    {
        while (cnt--)
            *dst++ = *src++;
    }
    
    return ret;
}