本篇文章聊一下strcpy和memcpy的代码实现,这两个也是c和c++面试中常考的问题点。
1. 手写strcpy
首先看一下,一份标准的strcpy
的实现如下:
char *strcpy(char* strDest, const char* strSrc)
{
assert( (strDest != NULL) && (strSrc != NULL));
char *address = strDest;
while((*strDest++ = *strSrc++) != '\0');
return address;
}
这里有如下几点需要注意:
- 源字符串要使用const类型,避免在程序中被修改;
- 在函数入口处要加上断言,检查源字符串指针和目标字符串指针是否有为空的,否则会产生不可预料的错误;
- 使用while循环要简单而清晰,使用尽量精简的代码;
- 返回值要是
char*
,且返回了目标字符串指针的原值,使得函数可以支持链式表达式,增加了函数的附加值。
以上这些点不只适用于strcpy这里,而是我们任何时候写代码都尽量遵循这些规则,这样才能写出高可用、高健壮性的代码。
对于以上代码,我们可以看出来,它是存在隐患的,当源字符串的长度超出目标字符串时,会导致把数据写入到我们无法控制的地址中去,存在很大的风险,所以就有了strncpy,下面也给一个strncpy的实现,如下:
char *strncpy(char* strDest, const char* strSrc, size_t n)
{
assert( (strDest != NULL) && (strSrc != NULL));
char *address = strDest