strcpy函数的实现(注意内存重叠)

大家一般认为名不见经传strcpy函数实现不是很难,流行的strcpy函数写法是:

[cpp]  view plain copy
  1. char *my_strcpy(char *dst,const char *src)  
  2. {  
  3.     assert(dst != NULL);  
  4.     assert(src != NULL);  
  5.     char *ret = dst;  
  6.     while((* dst++ = * src++) != '\0')   
  7.         ;  
  8.     return ret;  
  9. }  
如果注意到:

1,检查指针有效性;

2,返回目的指针des;

3,源字符串的末尾 '\0' 需要拷贝。

写出上面实现函数就不在话下。

然而这样的实现没有考虑拷贝时内存重叠的情况,下面的测试用例就能使调用my_strcp函数的程序崩溃:

[cpp]  view plain copy
  1. char str[10]="abc";  
  2. my_strcpy(str+1,str);  

然而调用系统的strcpy函数程序正常运行,打印str结果为“aabc”!可见系统strcpy函数的实现不是这样的。

strcpy的正确实现应为:

[cpp]  view plain copy
  1. char *my_strcpy(char *dst,const char *src)  
  2. {  
  3.     assert(dst != NULL);  
  4.     assert(src != NULL);  
  5.     char *ret = dst;  
  6.     memcpy(dst,src,strlen(src)+1);  
  7.     return ret;  
  8. }  
memcpy函数实现时考虑到了内存重叠的情况,可以完成指定大小的内存拷贝,它的实现方式建议查看文章“卓越的教练是如何训练高手的?”,会获益良多,这里仅粘帖函数memcpy函数的实现:

[cpp]  view plain copy
  1. void * my_memcpy(void *dst,const void *src,unsigned int count)  
  2. {  
  3.      assert(dst);  
  4.      assert(src);  
  5.      void * ret = dst;  
  6.      if (dst <= src || (char *)dst >= ((char *)src + count))//源地址和目的地址不重叠,低字节向高字节拷贝  
  7.      {  
  8.          while(count--)  
  9.          {  
  10.              *(char *)dst = *(char *)src;  
  11.              dst = (char *)dst + 1;  
  12.              src = (char *)src + 1;  
  13.          }  
  14.      }  
  15.      else                       //源地址和目的地址重叠,高字节向低字节拷贝  
  16.      {   
  17.          dst = (char *)dst + count - 1;  
  18.          src = (char *)src + count - 1;   
  19.          while(count--)   
  20.          {  
  21.              *(char *)dst = *(char *)src;  
  22.              dst = (char *)dst - 1;  
  23.              src = (char *)src - 1;  
  24.          }  
  25.     }  
  26.     return ret;  
  27. }  
两者结合才是strcpy函数的真正实现吧。
  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值