內存搜索函數 參考 C 語言庫函數 strstr 函數

 

// C標準庫中文版  strstr 庫函數實現
char *(strstr)(const char *s1, const char *s2)
{
    /* find first occurrence of s2[] in s1[]*/
    if(*s2 == '\0') {
        return ((char*)s1);
    }
    /* match rest of prefix */
    for(; (s1 = mystrchr(s1, *s2)) != NULL; ++s1) {
        const char *sc1, *sc2;

        for(sc1 = s1, sc2 = s2; ; ) {
            if(*++sc2 == '\0') {
                return ((char *)s1);
            } else if(*++sc1 != *sc2) {
                break;
            }
        }
    }
    return (NULL);
}

// 自己寫的內存搜索代碼
/*
MemSrch:
  說明: 並返回子串在主串中第一次出現的指針,  
src: 被搜索的主串
sub: 要匹配的子串 
scrSize: 主串字節數
subSize: 子串字節數 
返回值: 返回子串在主串中第一次出現的指針, 如果失敗返回NULL 
*/ 
u8* MemSrch (u8* src, u8* sub, u16 scrSize, u16 subSize)
{
    u8 *bp;
    u8 *sp;
    s16 restScrSz = (s16)scrSize; // 剩餘主串的個數
    s16 restSubSz = (s16)subSize; // 剩餘子串的個數

    if(!scrSize || !subSize) {
        return NULL;
    }

    /* 遍歷主字符串 */
    while(restScrSz > 0) {
        /* 用來遍歷子串 */
        bp = src;
        sp = sub;

        while(restSubSz > 0) {
            if(*bp == *sp) {  // 如果相等兩個串指針同時加, 剩餘子串個數減一 
                bp++;
                sp++;
                restSubSz--;
                if(!restSubSz) {  // 如果爲0代表子串已經搜索完了
                    return src;
                }
            } else {  // 兩個串內容不相等, 子串計數值重新開始
                restSubSz = subSize;
                break;
            }
        }

        src ++;
        restScrSz--;
    }
    return NULL;
}

 本人菜鳥高手勿噴，在不對的地方請大家指出！！！不吝賜教！！