在C中,有一个strcpy函数,将一个字符串的内容烤到另外一个字符串中,他是如何实现的呢,如果普通写法可能是这样
char* my_strcpy(char *dst, char *src)
{
assert(dst != NULL && src != NULL);
char *ret = dst;
while ((*dst++ = *src++) != '\0');
return ret;
}
但是如果my_strcpy(tmp + 1, tmp)这种情况的话,程序就会崩溃,因为tmp最后的一个结束符'\0'会被冲掉,因为他们共享内存,然后dst比tmp的位置相对靠后,那么他是先会走到最后一位的,然而src还没有走到最后一位,说明还是一个正常的字符,那么原内存中存在的结束符'\0'就会被冲掉,所以造成程序崩溃。所以我们要判断一下是否为共享内存。其实在C语言中,提供一个memcpy的函数,这个函数会自动的判断是否有共享内存这个问题,所以,我们可以直接调用一下这个,当然我们在这里是理解原理,所以还是要手动实现一下memcpy。当src < dst < src + strlen(src); src和dst就存在共享内存问题,在memcpy中需要判断一下,从后往前处理,免得结束符'\0'被覆盖
char* my_memcpy(char *dst, const char *src, int cnt)
{
assert(dst != NULL && src != NULL);
char *ret = dst;
if (dst > src && dst < src + cnt)
{
dst = dst + cnt - 1;
src = src + cnt - 1;
while (cnt--)
*dst-- = *src--;
}
else
{
while (cnt--)
*dst++ = *src++;
}
return ret;
}
那么更改后的strcpy得代码如下
char* my_strcpy(char *dst, char *src)
{
assert(dst != NULL && src != NULL);
char *ret = dst;
my_memcpy(dst, src, strlen(src) + 1);
//while ((*dst++ = *src++) != '\0');
return ret;
}