strcpy、memcpy函式的實現
阿新 • • 發佈:2019-02-14
本文轉載自: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函式的真正實現吧。