c++-strcpy函数的实现

kdyonly

发布于 2023-03-03 19:25:02

3510

发布于 2023-03-03 19:25:02

浏览量 2

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;  
}

阅读原文

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2021-11-27，如有侵权请联系 cloudcommunity@tencent.com 删除

memcpy