BM算法

来源：互联网发布：2014mac mini能做什么编辑：程序博客网时间：2024/04/29 02:07

由于毕业设计（入侵检测）的需要，这两天仔细研究了BM模式匹配算法，稍有心得，特此记下。

首先，先简单说明一下有关BM算法的一些基本概念。

BM算法是一种精确字符串匹配算法（区别于模糊匹配）。

BM算法采用从右向左比较的方法，同时应用到了两种启发式规则，即坏字符规则和好后缀规则，来决定向右跳跃的距离。

BM算法的基本流程: 设文本串T，模式串为P。首先将T与P进行左对齐，然后进行从右向左比较，如下图所示:

若是某趟比较不匹配时，BM算法就采用两条启发式规则，即坏字符规则和好后缀规则，来计算模式串向右移动的距离，直到整个匹配过程的结束。

下面，来详细介绍一下坏字符规则和好后缀规则。

首先，诠释一下坏字符和好后缀的概念。

请看下图：

图中，第一个不匹配的字符（红色部分）为坏字符，已匹配部分（绿色）为好后缀。

1）坏字符规则（Bad Character）：

在BM算法从右向左扫描的过程中，若发现某个字符x不匹配，则按如下两种情况讨论：

i. 如果字符x在模式P中没有出现，那么从字符x开始的m个文本显然不可能与P匹配成功，直接全部跳过该区域即可。

ii. 如果x在模式P中出现，则以该字符进行对齐。

用数学公式表示，设Skip(x)为P右移的距离，m为模式串P的长度，max(x)为字符x在P中最右位置。

例1：

下图红色部分，发生了一次不匹配。

计算移动距离Skip(c) = 5 - 3 = 2，则P向右移动2位。

移动后如下图：

2）好后缀规则（Good Suffix）：

若发现某个字符不匹配的同时，已有部分字符匹配成功，则按如下两种情况讨论：

i. 如果在P中位置t处已匹配部分P'在P中的某位置t'也出现，且位置t'的前一个字符与位置t的前一个字符不相同，则将P右移使t'对应t方才的所在的位置。

ii. 如果在P中任何位置已匹配部分P'都没有再出现，则找到与P'的后缀P''相同的P的最长前缀x，向右移动P，使x对应方才P''后缀所在的位置。

用数学公式表示，设Shift(j)为P右移的距离，m为模式串P的长度，j 为当前所匹配的字符位置，s为t'与t的距离（以上情况i）或者x与P''的距离（以上情况ii）。

以上过程有点抽象，所以我们继续图解。

例2：

下图中，已匹配部分cab（绿色）在P中再没出现。

再看下图，其后缀T'（蓝色）与P中前缀P'（红色）匹配，则将P'移动到T'的位置。

移动后如下图：

自此，两个规则讲解完毕。

在BM算法匹配的过程中，取SKip(x)与Shift(j)中的较大者作为跳跃的距离。

BM算法预处理时间复杂度为O（m+s），空间复杂度为O(s)，s是与P, T相关的有限字符集长度，搜索阶段时间复杂度为O(m·n)。

最好情况下的时间复杂度为O(n/m)，最坏情况下时间复杂度为O(m·n)。

BM算法的实现很复杂，但是高手程序员就是能写出令人敬佩的代码。

当然，高手程序员不是指我自己，是SNORT2.7.0的设计者/实现者之一。

下面是SNORT2.7.0中提取出的代码。

我将以行注的形式进行讲解。

C代码  
/* 
    函数：int* MakeSkip(char *, int) 
    目的：根据坏字符规则做预处理，建立一张坏字符表 
    参数： 
        ptrn => 模式串P 
        PLen => 模式串P长度 
    返回： 
        int* - 坏字符表 
*/  
int* MakeSkip(char *ptrn, int pLen)  
{     
    int i;  
    //为建立坏字符表，申请256个int的空间  
    /*PS:之所以要申请256个，是因为一个字符是8位， 
      所以字符可能有2的8次方即256种不同情况*/  
    int *skip = (int*)malloc(256*sizeof(int));  
  
    if(skip == NULL)  
    {  
        fprintf(stderr, "malloc failed!");  
        return 0;  
    }     
  
    //初始化坏字符表，256个单元全部初始化为pLen  
    for(i = 0; i < 256; i++)  
    {  
        *(skip+i) = pLen;  
    }  
  
    //给表中需要赋值的单元赋值，不在模式串中出现的字符就不用再赋值了  
    while(pLen != 0)  
    {  
        *(skip+(unsigned char)*ptrn++) = pLen--;  
    }  
  
    return skip;  
}  
  
  
/* 
    函数：int* MakeShift(char *, int) 
    目的：根据好后缀规则做预处理，建立一张好后缀表 
    参数： 
        ptrn => 模式串P 
        PLen => 模式串P长度 
    返回： 
        int* - 好后缀表 
*/  
int* MakeShift(char* ptrn,int pLen)  
{  
    //为好后缀表申请pLen个int的空间  
    int *shift = (int*)malloc(pLen*sizeof(int));  
    int *sptr = shift + pLen - 1;//方便给好后缀表进行赋值的指标  
    char *pptr = ptrn + pLen - 1;//记录好后缀表边界位置的指标  
    char c;  
  
    if(shift == NULL)  
    {  
        fprintf(stderr,"malloc failed!");  
        return 0;  
    }  
  
    c = *(ptrn + pLen - 1);//保存模式串中最后一个字符，因为要反复用到它  
  
    *sptr = 1;//以最后一个字符为边界时，确定移动1的距离  
  
    pptr--;//边界移动到倒数第二个字符（这句是我自己加上去的，因为我总觉得不加上去会有BUG，大家试试“abcdd”的情况，即末尾两位重复的情况）  
  
    while(sptr-- != shift)//该最外层循环完成给好后缀表中每一个单元进行赋值的工作  
    {  
        char *p1 = ptrn + pLen - 2, *p2,*p3;  
          
        //该do...while循环完成以当前pptr所指的字符为边界时，要移动的距离  
        do{  
            while(p1 >= ptrn && *p1-- != c);//该空循环，寻找与最后一个字符c匹配的字符所指向的位置  
              
            p2 = ptrn + pLen - 2;  
            p3 = p1;  
              
            while(p3 >= ptrn && *p3-- == *p2-- && p2 >= pptr);//该空循环，判断在边界内字符匹配到了什么位置  
  
        }while(p3 >= ptrn && p2 >= pptr);  
  
        *sptr = shift + pLen - sptr + p2 - p3;//保存好后缀表中，以pptr所在字符为边界时，要移动的位置  
        /* 
          PS:在这里我要声明一句，*sptr = （shift + pLen - sptr） + p2 - p3; 
             大家看被我用括号括起来的部分，如果只需要计算字符串移动的距离，那么括号中的那部分是不需要的。 
             因为在字符串自左向右做匹配的时候，指标是一直向左移的，这里*sptr保存的内容，实际是指标要移动 
             距离，而不是字符串移动的距离。我想SNORT是出于性能上的考虑，才这么做的。           
        */  
  
        pptr--;//边界继续向前移动  
    }  
  
    return shift;  
}  
  
  
/* 
    函数：int* BMSearch(char *, int , char *, int, int *, int *) 
    目的：判断文本串T中是否包含模式串P 
    参数： 
        buf => 文本串T 
        blen => 文本串T长度 
        ptrn => 模式串P 
        PLen => 模式串P长度 
        skip => 坏字符表 
        shift => 好后缀表 
    返回： 
        int - 1表示成功（文本串包含模式串），0表示失败（文本串不包含模式串）。 
*/  
int BMSearch(char *buf, int blen, char *ptrn, int plen, int *skip, int *shift)  
{  
    int b_idx = plen;    
    if (plen == 0)  
        return 1;  
    while (b_idx <= blen)//计算字符串是否匹配到了尽头  
    {  
        int p_idx = plen, skip_stride, shift_stride;  
        while (buf[--b_idx] == ptrn[--p_idx])//开始匹配  
        {  
            if (b_idx < 0)  
                return 0;  
            if (p_idx == 0)  
            {       
                return 1;  
            }  
        }  
        skip_stride = skip[(unsigned char)buf[b_idx]];//根据坏字符规则计算跳跃的距离  
        shift_stride = shift[p_idx];//根据好后缀规则计算跳跃的距离  
        b_idx += (skip_stride > shift_stride) ? skip_stride : shift_stride;//取大者  
    }  
    return 0;  
}  