strcpy的需要注意的幾點

C++中的strcpy沒有考慮全面是我沒有想到的:C++的strcpy無法解決內存覆蓋的問題,比如:char str[100] = "hello"; strcpy(str + 1, str); 會出現意外的結果,請自己試一試。

既然C++不能解決內存覆蓋的strcpy(但是memcpy確實能解決,我卻不知道這是問什麼了,難道他們認爲strcpy不會去解決內存覆蓋的情況麼?),那麼我們就自己寫一個:

代碼如下:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <assert.h>

using namespace std;

/*
memcpy的實現方法
*/
void* memCpy(void* memDest, void* memSrc, unsigned int size)
{
    assert(memDest != NULL && memSrc != NULL);
    if (memDest < memSrc || (char*) memDest > (char*)memSrc + size) //判斷內存是否出現重疊
    {
        while (size--)
        {
            *(char*)memDest = *(char*)memSrc;
            memDest = (char*)memDest + 1;
            memSrc = (char*)memSrc + 1;
        }
    }
    else
    {
        memDest = (char*)memDest + size - 1;
        memSrc = (char*)memSrc + size - 1;
        while (size--)
        {
            *(char*)memDest = *(char*)memSrc;
            memDest = (char*)memDest - 1;
            memSrc = (char*)memSrc - 1;
        }
    }
}

unsigned strLen(char* str)
{
    char* oldVal = str;
    while (*str++ != '\0');
    return str - oldVal - 1;
}

/*
這個纔是真正正確的strCpy的實現方法
*/
char* strCpy(char* strDest, const char* strSrc)
{
    assert(strDest != NULL && strSrc != NULL);
    if (strDest == strSrc)
    {
        return strDest;
    }
    memCpy(static_cast<void*>(strDest), static_cast<void*>(const_cast<char*>(strSrc)), strLen(const_cast<char*>(strSrc)) + 1);
    return strDest;
}

/* 這個實現跟系統實現的做法應該差不多,如果是在同一個字符串就執行不正確!
char* strCpy(char* strDest, const char* strSrc)
{
    if (strSrc == NULL || strDest == NULL)
        return NULL;
    if (strDest == strSrc)
        return strDest;
    char* temp = strDest;
    while ((*strDest++ = *strSrc++) != '\0');
    return temp;
}
*/
int main()
{
    char strDest[20]; //亂指的指針。很害人啊!
    char strSrc[] = "hello";
    strCpy(strSrc + 1, strSrc);
    cout << strSrc << endl;
    return 0;
}

好像不用memcpy那麼麻煩,從後往前就可以避免了。

char* StrCpy2(char* strDest, char* strSrc)
{
    assert(strDest != NULL && strSrc != NULL);
    int len = strlen(strSrc);
    cout << "len = " << len << endl;
    char* endOfSrc = strSrc + len;
    char* endOfDest = strDest + len;
    len++;
    while (len--)
    {
        (*endOfDest) = (*endOfSrc);
        endOfDest--;
        endOfSrc--;
    }
    return strDest;
}


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