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;
}