模拟实现strstr函数功能

strstr函数的功能：有两个字符串str1，str2。如果s2是s1的子集，返回s2在s1中首次出现的地址，否则返回NULL。如str1 = “we are student！”,str2 = “are”。那么返回的就是“are”的第一个字符‘a’在s1中的地址。

#include <stdio.h>
#include <windows.h>
#include <string.h>
#include <assert.h>

char *my_strstr(const char *str1, const char *str2)
{
assert(str1);
assert(str2);

char *dest = (char *)str1;
char *sorc = (char *)str2;
char *flag = NULL;

if (!*str2) //如果第二个字符串为NULL，直接返回NULL
{
return NULL;
}

while (*dest) //当*dest不是'\0'时，进入循环
{
flag = dest;
sorc = (char *)str2;
while (*flag && *sorc && (*flag == *sorc))
{
flag++;
sorc++;
}

if (!*sorc) //当*sorc='\0'，表示str2的所有字符都比较完了，此时可以判定str2是str1的子串
{
return dest;
}
else
{
dest++;
}

}
return NULL;
}

int main()
{
char s1[] = "we are student!";
char s2[] = "are";

printf("%s", my_strstr(s1, s2));

system("pause");
return 0;
}