实现C库函数strcpy

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字符串长度。

猜你喜欢

转载自blog.csdn.net/renwotao2009/article/details/52853563