已知strcpy函數的原型是:
char *strcpy(char *dst, const char *src);
要求:
1、實現strcpy函數
2、解釋爲什麼要返回char *
3、假如考慮dst和src內存重疊的情況,strcpy該怎麼實現
一: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); 不能簡單返回NULL,否則用戶不知道是否複製成功!
char *轉換爲bool即是類型隱式轉換,這種功能雖然靈活,但更多的是導致出錯概率增大和維護成本升高。
(C)檢查指針的有效性時使用assert(dst != 0 && src != 0);
直接使用常量(如本例中的0)會減少程序的可維護性。而使用NULL代替0,如果出現拼寫錯誤,編譯器就會檢查出來。
[3]返回目標地址
(A)忘記保存原始的strdstt值。
[4]'\0'
(A)循環寫成while (*dst++=*src++);明顯是不合適的,與[2][B]相同解釋!
(B)循環寫成while (*src!='\0') *dst++=*src++; 循環體結束後,dst字符串的末尾沒有正確地加上'\0'。
總結:
1、參數中儘量用const修飾,增加程序健壯性!
2、對輸入參數進行判斷!當不正確的時候,要用assert函數!不能簡單return!
3、對於判空等條件,儘量用原裝句型,不要省略!該 dest != NULL,就不要寫成!dest 或 dest != 0。增加健壯性!
4、不要遺漏邊界條件!
二、爲什麼要返回char *?
返回dst的原始值使函數能夠支持鏈式表達式。
鏈式表達式的形式如:
int l=strlen(strcpy(strA,strB));
又如:
char * strA=strcpy(new char[10],strB);
返回strSrc的原始值是錯誤的。------- 我們要在新複製的字符串上進行操作,直接返回源串,會導致程序邏輯錯誤。
其一,源字符串肯定是已知的,返回它沒有意義。
其二,不能支持形如第二例的表達式。
其三,把const char *作爲char *返回,類型不符,編譯報錯。
三、假如考慮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。
可以參考具體memcpy詳解:http://blog.csdn.net/a45872055555/article/details/38817355
char * strcpy(char *dst,const char *src)
{
assert(dst != NULL && src != NULL);
char *ret = dst;
my_memcpy(dst, src, strlen(src)+1);
return ret;
}
由於memcpy參數是void類型,注意程序中的類型轉換。
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;
}
擴展:
在上面的基礎上,我們不難寫出strncpy()函數,但要注意提前截止或者拷貝完n個字符後,要添加結束符!
char * strncpy(char *dst,const char *src,int n) //[1]
{
assert(dst != NULL && src != NULL); //[2]
char *tmp = dst; //[3]
if(n){
while ((*tmp++ = *src++) != '\0') {
if (--n == 0) { //如果已經減爲0,就添加結束符
*tmp = '\0';
break;
}
}
}
return dst;
}