strchr，strstr自实现

#include <stdio.h>
#include <stdlib.h>

int strchrcount(char *src,char ch)
{
    int count = 0;
    while(*src)
    {
        if(*src == ch)
            count++;
        src++;
    }
    return count;
}
char *mystrchr(char *src,char ch)
{

    while(*src)
    {
        if(*src == ch)
            return src;
        src++;
    }
    return NULL;
}
int strstrcount(char *src,char*str)
{
    int len = strlen(str);
    int count = 0;
    if(*str)
    {
        while(src = strstr(src,str))
        {
            count++;
            src += len;
        }
    }
    return count;
}

char *mystrstr(char *src,char *str)
{
    if(*str)
    {
        while(*src)
        {
            for(int n = 0;*(src+n) == *(str+n);n++)
                if(*(str+n+1) == '\0')
                    return src;
            src++;
        }
        return NULL;
    }
    else
        return NULL;
}

int main(int argc, char *argv[])
{
    char *p = "dnsndnaaaaddd";
    char ch = 'a';
    char *str = "dn";
    printf("%d\n",strchrcount(p,ch));
    printf("%d\n",strstrcount(p,str));
    printf("%s\n",mystrstr(p,str));
    printf("%s\n",mystrchr(p,ch));
    return 0;
}