[转] strcpy函数的实现

转自：strcpy函数的实现

代码实现

#include <iostream>
#include <assert.h>
#include <iostream>
//#include <string.h>
using namespace std;
int strlen(const char *src)
{
    assert(src != NULL);
    int lens = 0;
    while (*src++ != ‘\0‘)
        lens++;
    return lens;
}

char *memcpy(char *dst, const char *src, int cnt)
{
    assert(dst != NULL && src != NULL);
    char *ret = dst;
    if (dst >= src && dst <= src + cnt - 1)
    {
        src += cnt - 1;
        dst += cnt - 1;
        while (cnt--)
            *dst-- = *src--;
    }
    else
    {
        while (cnt--)
            *dst++ = *src++;
    }
    return ret;
}

char *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 a[] = "hello";
    char b[] = "b1";
    //strcpy(a, b);
    strcpy(a, a+1);
    char *f = strcpy(a, a+1);
    cout << a << endl;
    cout << f << endl;
    int m = 0;
    int n = 5;
    int c = (m=n++);
    cout << "c:" << c << endl;
    cout << "n:" << n << endl;
    return 0;
}

已知strcpy函数的原型是：

char *strcpy(char *dst, const char *src);

1. 实现strcpy函数
2. 解释为什么要返回char *
3. 假如考虑dst和src内存重叠的情况，strcpy该怎么实现

1.strcpy的实现代码

char * strcpy(char *dst,const char *src)   //[1]
{
    assert(dst != NULL && src != NULL);    //[2]

    char *ret = dst;  //[3]

    while ((*dst++=*src++)!=‘\0‘); //[4]

    return ret;
}

[1]const修饰

• 源字符串参数用const修饰，防止修改源字符串。

[2]空指针检查

• 不检查指针的有效性，说明答题者不注重代码的健壮性。
• 检查指针的有效性时使用assert(!dst && !src);char *转换为bool即是类型隐式转换，这种功能虽然灵活，但更多的是导致出错概率增大和维护成本升高。
• 检查指针的有效性时使用assert(dst != 0 && src != 0);直接使用常量（如本例中的0）会减少程序的可维护性。而使用NULL代替0，如果出现拼写错误，编译器就会检查出来。

[3]返回目标地址

• 忘记保存原始的strdstt值。

[4]‘\0‘

• 循环写成while (*dst++=*src++);明显是错误的。
• 循环写成while (*src!=‘\0‘) *dst++=*src++;

循环体结束后，dst字符串的末尾没有正确地加上‘\0‘。

2.为什么要返回char *？

返回dst的原始值使函数能够支持链式表达式。

链式表达式的形式如：

int l=strlen(strcpy(strA,strB));

又如：

char * strA=strcpy(new char[10],strB);

返回strSrc的原始值是错误的。

其一，源字符串肯定是已知的，返回它没有意义。

其二，不能支持形如第二例的表达式。

其三，把const char *作为char *返回，类型不符，编译报错。

3.假如考虑dst和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 * strcpy(char *dst,const char *src)
{
    assert(dst != NULL && src != NULL);

    char *ret = dst;

    my_memcpy(dst, src, strlen(src)+1);

    return ret;
}

my_memcpy的实现如下

char *my_memcpy(char *dst, const char* src, int cnt)
{
    assert(dst != NULL && src != NULL);

    char *ret = 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 ret;
}