用C语言模拟实现strstr/memcpy/memmove函数

一、strstr函数

1.1 strstr函数的作用

查找str1中的字符串是否包含str2子字符串，若包含，则返回子字符串的首地址；若不包含，则返回NULL。

1.2 C语言模拟实现strstr函数

//1.16 模拟实现strstr函数
#include<stdio.h>
#include<assert.h>
char* my_strstr(const char* arr1, const char* arr2)
{
	assert(arr1 && arr2);
	const char* s1 = arr1;
	const char* s2 = arr2;
	const char* p = arr1;
	while (*p)
	{
		s1 = p;//p起标记作用，每次找不到子串，p就往后挪一位；如果找到子串，则返回p
		s2 = arr2;
		while (*s1 == *s2 && *s1 != '\0' && *s2 != '\0')
		{
			s1++;
			s2++;
		}
		if (*s2 == '\0')
			return (char*)p;
		p++;
	}
	return NULL;
}
int main()
{
	char arr1[20] = "abcdefghi";
	char arr2[] = "cdef";
	char* ret=my_strstr(arr1, arr2);
	if (ret == NULL)
		printf("找不到子串\n");
	else
		printf("%s\n",ret);
	return 0;
}

二、memcpy函数

2.1 memcpy函数的作用

内存复制，将source中num个字节的数据copy到destination中

2.2 C语言模拟实现memcpy函数

#include<stdio.h>
#include<assert.h>
void* my_memcpy(void* dest,const void* src, size_t num)//void*是泛型，任何类型的数据都可复制，num为无符号整数
{
	assert(dest && src);
	void* ret = dest;
	while (num--)
	{
		*(char*)dest= *(char*)src;//将src强制转换为1个字节类型即char型，然后再赋值
		src = (char*)src + 1;
		dest = (char*)dest + 1;
	}
	return ret;

}
int main()
{
	int arr1[] = { 1,5,6,8,4,1,9,8 };
	int arr2[10] = { 0 };
	my_memcpy(arr2, arr1, 20);//将arr1中的20个字节的数据复制到arr2中，即5个整数
	for (int i = 0; i < 10; i++)
		printf("%d ", arr2[i]);
	return 0;
}

三、memmove函数

3.1 memmove函数的作用

memmove用于拷贝字节，如果目标区域和源区域有重叠的话，memmove能够保证源串在被覆盖之前将重叠区域的字节拷贝到目标区域中，但复制后源内容会被更改。但是当目标区域与源区域没有重叠则和memcpy函数功能相同。即源空间和目标空间出现重叠，就使用memmove函数；如果空间不重叠，则memcpy和memmove都可以使用，使用效果是一样的。

3.2 C语言模拟实现memmove函数

#include<stdio.h>
#include<assert.h>
#include<string.h>
void* my_memmove(void* dest,const void* src,size_t num)
{
	assert(src && dest);
	void* ret = dest;
	if (dest < src)//当dest在src左侧时，需要移动的内存数据需从前往后copy才不会被重叠区域覆盖
	{
		while (num--)
		{
			*(char*)dest = *(char*)src;
			dest = (char*)dest + 1;
			src = (char*)src + 1;
		}
	}
	else//当dest在src右侧时，需要移动的内存数据需从后往前copy才不会被重叠区域覆盖
	{
		while (num--)
		{
			*(char*)dest + num = *(char*)src+num;
		}
	}
	return ret;
}
int main()
{
	int arr1[] = { 1,2,3,4,5,6,7,8,9 };
	my_memmove(arr1, arr1 + 3, 20);
	for (int i = 0; i < 9; i++)
		printf("%d ", arr1[i]);
	return 0;
}