实现C中的strcpy函数

最新推荐文章于 2023-08-17 08:01:13 发布

原创最新推荐文章于 2023-08-17 08:01:13 发布 · 533 阅读

2 ·

CC 4.0 BY-SA版权

本文详细介绍了C语言中strcpy函数的功能及其实现方式，并针对内存重叠问题提出了改进方案。

1、strcpy库函数介绍

C语言中在头文件<string.h>中，C++中在<cstring>中；

函数原型：char * strcpy(char * destStr, char srcStr);

函数功能：将srcStr中的字符串拷贝到destStr的内存中；通过字符串结束符'\0'来控制结束，如果destStr预留的内存不够的话会“溢出”。

2、实现strcpy函数

这个也是一个经常出的面试题，如果面试者只是根据功能写出如下的代码

char * my_strcpy(char *dest, char * src)
{
     if(dest == NULL || src == NULL)
            return NULL;
     char *p_dest = dest;
     char *p_src = src;
     while(*p_src != '\0')
     {
          *p_dest = *p_src;
          ++p_dest;
          ++p_src;
     }
     *p_dest = '\0';
     return dest;
}

这样一个实现看似是正确的:1)考虑了空指针；

2)返回课目的地址；

3)目的串末尾为'\0';

但是它并没有考虑“内存重叠”问题。这个题目的考察点和memcpy是一样的，都在考察“内存重叠”，如果面试官先问你memcpy的代码，然后再问你这个你可能会意识到这个问题。但是如果面试官直接问你strcpy的实现，可能你就蒙了。

对strcpy的改进：

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

这样就可以有效地处理“内存重叠”问题。