strcpy实现

常见strcpy实现　　

　　一个常见的strcpy实现如下：

char *my_strcpy(char *dst, const char *src)
{
    if (dst == nullptr || src == nullptr)    // 写成!dst或dst == 0都不推荐
        return nullptr;

    if (dst == src)                          // 判断dst和src是否已经指向同一块内存，若是则直接返回
        return dst;

    char *tmp = dst;
    while ((*dst++ = *src++) != '\0')        // 注意在dst末尾加上'\0'
        ;

    return tmp;　　　　　　　　　　　　　　　　　 // 在这里可能会不假思索就返回dst，但这样是有问题的，因为dst经过累加后实际已指向'\0'的下一块内存

}

　　针对strcpy还有常见的一个问题就是：为什么要返回char *？

　　因为要让strcpy支持链式赋值，例如：

strlen(strcpy(new char[20], 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。

char *my_memcpy(char *dst, const char *src, int cnt)
{
    if (dst == nullptr || src == nullptr)
        return nullptr;

    char *tmp = dst;
    if (dst >= src && dst <= src + cnt - 1)    // 内存重叠，从高地址开始复制
    {
        dst = dst + cnt - 1;
        src = src + cnt - 1;
        while (cnt--)
            *dst-- = *src--;
    }
    else                                       // 正常情况，从低地址开始复制
    {
        while (cnt--)
            *dst++ = *src++;
    }

    return tmp;
}

char *my_strcpy(char *dst, const char *src)
{
    if (dst == nullptr || src == nullptr)    // 写成!dst或dst == 0都不推荐
        return nullptr;

    if (dst == src)                          // 判断dst和src是否已经指向同一块内存，若是则直接返回
        return dst;

    char *tmp = dst;
    my_memcpy(dst, src, strlen(src) + 1);　　// 注意strlen并不会将‘\O’统计在内

    return tmp;
}

参考资料

　　strcpy函数的实现