c++-strcpy函数的实现

2021-11-27 193 0

1,检查指针有效性;
2,返回目的指针des;
3,源字符串的末尾 '\0' 需要拷贝。
写出了流行的写法,不过不完美。

char *my_strcpy(char *dst,const char *src)
{
    assert(dst != NULL);
    assert(src != NULL);
    char *ret = dst;
    while((* dst++ = * src++) != '\0')
    ;
    return ret;
}

然而这样的实现没有考虑拷贝时内存重叠的情况.
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;
}

阅读原文

相关文章

【转】C++-ADO编程详解
c-各变量在内存中的位置
c-变态跳台阶
C++-编写类String的构造函数、析构函数和赋值函数
c-计蒜客 排序好的数组删除重复数字
c-归并排序

发布评论