实现C库函数strcpy

时间:2020-12-21 19:55:30

1 实现c库的函数strcpy

原型声明:char* strcpy(char* dst, const char* src);

普通版本的strcpy

//strcpy 实现没有检查dst和src内存重叠问题
char* strcpy(char *dst, const char *src) //const约束,内容不可变
{
assert((src != NULL) && (dst != NULL)); //参数非0检验
char *pStr = dst; //保存dst
while ((*dst++ = *src++) != '\0')
{
/*do nothing*/
}
return pStr; //返回dst,允许链式表达式
}

实现内存重叠的strcpy

// 检查内存重叠
char* strcpy(char *dst, const char *src) //const约束,内容不可变
{
assert((src != NULL) && (dst != NULL)); //参数非0检验
size_t size = strlen(src);
if (src < dst && src + size > dst) { // 内存重叠
// 从后向前复制
char *pSrc = src + size - 1;
char *pDst = dst + size - 1;
while (*pDst-- = *pSrc--) {
/* do nothing */
}
} else {
// 从前向后复制
char *pSrc = src;
char *pDst = dst;
while (*pDst++ = *pSrc++) {
/* do nothing */
}
}
return dst; //返回dst,允许链式表达式
}

已经实现src指针指向的内容拷贝到dst所指的内存中,为什么要strcpy返回 char *?

使函数支持链式表达式,增加函数灵活性。
如strlen(strcpy(strDest, strSrc))可以直接求出strDest字符串长度。