实现 strcpy 函数

要考虑拷贝时内存重叠的情况:

#include <iostream>
#include <string.h>
#include <assert.h>
using namespace std;

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

int main()
{
  char src[] = "abcd";
  char* dest = src + 1;
  my_strcpy(dest, src);
  cout<<dest<<endl;
  return 0;
}

输出:

[liboyang@database74 test]$ g++ test.cpp -std=c++11
[liboyang@database74 test]$ ./a.out 
abcd

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

两者结合才是 strcpy 函数的真正实现:

#include <iostream>
#include <string.h>
#include <assert.h>
using namespace std;

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

char *my_strcpy(char *dst, const char *src)
{
  assert(dst != NULL);
  assert(src != NULL);
  char *ret = dst;
  my_memcpy(dst, src, strlen(src) + 1);
  return ret;
}

int main()
{
  char src[] = "abcd";
  char* dest = src + 1;
  my_strcpy(dest, src);
  cout<<dest<<endl;
  return 0;
}

输出:

[liboyang@database74 test]$ g++ test.cpp -std=c++11
[liboyang@database74 test]$ ./a.out 
abcd
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值