Skip to content

灾难性回溯

有些正则表达式看上去很简单,但是执行起来耗时非常非常非常长,甚至会导致 JavaScript 引擎「挂起」。

开发者们很容易一不小心就写出这类正则表达式,所以我们迟早会面对这种意外问题。

典型的症状就是 —— 一个正则表达式有时能正常工作,但对于某些特定的字符串就会消耗 100% 的 CPU 算力,出现“挂起”现象。

在这种情况下,Web 浏览器会建议杀死脚本并重新载入页面。这显然不是我们愿意看到的。

在服务器端 JavaScript 中,在使用这种正则表达式处理用户数据时可能会引发程序漏洞。

例子

假设,我们现在有一个字符串,我们想检查其中是否包含一些单词 pattern:\w+,允许字符后跟着可选的空格符 pattern:\s?

我们使用一个这样的正则 pattern:^(\w+\s?)*$,它指定了 0 个或更多个此类的字符。

我们运行一下:

let regexp = /^(\w+\s?)*$/;

alert( regexp.test("A good string") ); // true
alert( regexp.test("Bad characters: $@#") ); // false
1
2
3
4

这似乎能正常工作。结果是正确的。但是在特定的字符串上,它会消耗很多时间。它耗时太久以至于让 CPU 会跑满 100% 负载,导致 JavaScript 引擎「挂起」。

如果你运行下面这个例子,由于 JavaScript 会进入「挂起」状态,因此你可能什么结果都看不到。此时浏览器会停止对事件的响应,UI 也会停止运作。一段时间之后浏览器会建议重新载入页面。所以请谨慎对待:

let regexp = /^(\w+\s?)*$/;
let str = "An input string that takes a long time or even makes this regexp to hang!";

// 会耗费大量时间
alert( regexp.test(str) );
1
2
3
4
5

有些正则引擎能够处理好这种查询,但大多数引擎对此都无能为力。

简化的例子

问题在哪?为何正则表达式会「挂起」?

为了理解它,我们来简化一下例子:移除空格符 pattern:\s?,使其成为 pattern:^(\w+)*$

同时为了让问题更显著,再用 pattern:\d 替换掉 pattern:\w。这个新的正则表达式执行时仍然会挂起,比如:

let regexp = /^(\d+)*$/;

let str = "012345678901234567890123456789!";

// 会耗费大量时间
alert( regexp.test(str) );
1
2
3
4
5
6

所以正则到底哪里出了问题?

首先,有人可能发现了这个正则 pattern:(\d+)* 有点奇怪,量词 pattern:* 有点画蛇添足。如果我们要匹配数字,那我们可以使用 pattern:\d+

实际上,正则表达式是非常死板、机械化的。造成它运行缓慢的原因和上面我们看到的那样,所以让我们来理解它运作过程,然后问题的原因就会显而易见了。

subject:123456789! 这行中(这里简写了,看得更清晰一些)中查询 pattern:^(\d+)*$ 时到底发生了什么,要耗时这么久呢?

  1. 首先,正则引擎尝试查一个数字 pattern:\d+。加号 pattern:+ 默认为贪婪模式,所以它囊括/消耗(consume)了所有数字:

    \d+.......
    (123456789)z
    
    1
    2

    然后它尝试应用星号量词,但是此时已经没有更多数字了,所以星号匹配不到任何东西。

    模式中接下来的 pattern:$ 匹配字符串的结束,但是我们例子的文字中有 subject:!,所以匹配失败,没有匹配结果:

               X
    \d+........$
    (123456789)!
    
    1
    2
    3
  2. 由于没有匹配结果,贪婪量词 pattern:+ 的重复匹配次数会减一,并往前回溯一个字符。

    现在 pattern:\d+ 会匹配除了最后一个数字之外的所有数字:

    \d+.......
    (12345678)9!
    
    1
    2
  3. 然后引擎尝试从新位置 (9) 继续搜索。

    星号 pattern:(\d+)* 可以成功应用 -- 它匹配到了数字 match:9

    
    \d+.......\d+
    (12345678)(9)!
    
    1
    2
    3

    引擎再次去尝试匹配 pattern:$,但是失败了,因为它遇到了 subject:!

                 X
    \d+.......\d+
    (12345678)(9)z
    
    1
    2
    3
  4. 没有匹配结果,所以引擎继续回溯,减少重复匹配次数。回溯的运行过程基本上是这样的:最后一个贪婪量词逐渐减少重复匹配次数,然后前一个贪婪量词再减少重复匹配次数,以此类推。

    它会尝试所有可能的排列组合,这里是他们的例子:

    第一串数字 pattern:\d+ 有 7 位数,后面跟着一串 2 位数的数字:

                 X
    \d+......\d+
    (1234567)(89)!
    
    1
    2
    3

    第一串数字有 7 位数,后面跟着两串数字,每串数字各有 1 位数:

                   X
    \d+......\d+\d+
    (1234567)(8)(9)!
    
    1
    2
    3

    第一串数字有 6 位数,后面跟着一串 3 位数的数字:

                 X
    \d+.......\d+
    (123456)(789)!
    
    1
    2
    3

    第一串数字有 6 位数,后面跟着两串数字:

                   X
    \d+.....\d+ \d+
    (123456)(78)(9)!
    
    1
    2
    3

    ……以此类推。

123456789 这样一串数字,分割成多个数的话有好几种分割方式。准确的说,如果这数字长度是 n ,则共有 2n-1 种方式去分割它。

假设 n=20,那么就有差不多一百万种排列组合,假设 n=30,那就得再乘上一千倍。正因为要尝试每种排列组合,所以才导致会消耗这么多时间。

那我们该怎么办?

我们应转而使用懒惰模式吗?

不幸的是,这没用:如果我们用 pattern:\d+? 去替代 pattern:\d+,它还是会挂起。排列组合的顺序会变化,但是总数不变。

有些正则表达式引擎应经过严密的测试,并自带一种能够避免遍历所有排列组合的有限自动机来优化速度。但并不是所有引擎能够做到,也不是在所有场合下都有效果。

回到字符和字符串

在我们第一个例子中,当我们用 pattern:^(\w+\s?)*$ 这种模式在字符串 subject:An input that hangs! 中查找字符时,也遇到了相同的问题。

原因是 pattern:\w+ 可以用来表示一个或多个字符:

(input)
(inpu)(t)
(inp)(u)(t)
(in)(p)(ut)
...
1
2
3
4
5

对于我们人类来说,很显然它们无法匹配成功,因为例子中的字符串以感叹号 ! 结尾,然而正则表达式期望在末尾有一个词语式字符 pattern:\w 或者空格 pattern:\s 来结尾。正则引擎理解不了这种状况。

它尝试了所有 pattern:(\w+\s?)* 的排列组合试图去囊括整个字符串,包含了带空格 pattern:(\w+\s)* 的情形和不带空格 pattern:(\w+)* 的情形(因为 pattern:\s? 是可选的)。由于各种排列组合的数量太多了,所以耗费了大量时间去查询。

如何解决问题?

主要有 2 种解决方法。

第一种去试着减少各种排列组合的数量。

我们用把正则重写成 pattern:^(\w+\s)*\w* - 此处我们会查找后面跟着一个空格的、任意数量的单字字符 pattern:(\w+\s)*,然后跟着一个(可选的)单字字符 pattern:\w*

这个正则表达式在查询效果上等同于之前那个(查找的内容是相同的),运行起来也没问题:

let regexp = /^(\w+\s)*\w*$/;
let str = "An input string that takes a long time or even makes this regex to hang!";

alert( regexp.test(str) ); // false
1
2
3
4

为什么问题消失了?

现在星号 pattern:* 跟在 pattern:\w+\s 后面,而不是 pattern:\w+\s? 后面。这意味着它无法匹配一个拥有多个连续单字字符串 pattern:\w+ 的单词,也就省下了原本去尝试这些排列组合的时间。

举个例子,之前那个模式 pattern:(\w+\s?)* 可能以两个 pattern:\w+ 的方式来匹配单词 subject:string

\w+\w+
string
1
2

之前那个模式,由于存在可选的 pattern:\s,它允许 pattern:\w+pattern:\w+\spattern:\w+\w+ 等等的变体形式。

我们重写之后的 pattern:(\w+\s)* 就不存在这些情况:它可能会是 pattern:\w+\s 或者 pattern:\w+\s\w+\s,但不可能是 pattern:\w+\w+。所以总体上,排列组合的可能性大大减少了。

防止回溯

有时候重写正则会比较麻烦,而且要推敲如何重写正则恐怕也并非易事。

另一种思路是禁止量词的回溯。

有些正则表达式我们人眼一看就知道无法匹配成功,但正则引擎还是会硬去尝试很多它的排列组合。

比如,正则 pattern:(\d+)*$pattern:+ 对于我们人类来说很明显不应去回溯,就算我们用两个独立的 pattern:\d+\d+ 去替换一个 pattern:\d+,也是根本没作用的:

\d+........
(123456789)!

\d+...\d+....
(1234)(56789)!
1
2
3
4
5

原先的那个例子 pattern:^(\w+\s?)*$ 中我们可能希望禁止在 pattern:\w+ 这里去进行回溯。逻辑是: pattern:\w+ 应当尽可能多地去匹配一个完整的单词。在 pattern:\w+ 这里减少重复次数,然后将之进行分割,形成 pattern:\w+\w+,这类的做法没有任何意义。

为此,现代正则表达式引擎支持占有型量词(Possessive Quantifiers)。它们就像贪婪量词一样,但是不会进行回溯(所以比一般的正则量词更简单)。

它们也被成为“原子捕获分组(atomic capturing groups)” - 能够在括号内禁止回溯。

不幸的是,JavaScript 并不支持它,但是仍有其他办法。

用前瞻视角解决问题

我们可以使用前瞻断言来防止回溯。

在不进行回溯的前提下,我们用 pattern:(?=(\w+))\1 这个模式就可以尽可能多地重复匹配 pattern:\w

来解读一下:

  • 前瞻断言 pattern:?= 从当前位置开始,向前查找最长的单词 pattern:\w+
  • 引擎默认不会去记录 pattern:?=... 括号中的内容。为了记录它们,所以我们把 pattern:\w+ 放入括号中,这样引擎会记录括号中的内容了。
  • ……然后用 pattern:\1 来引用括号中的内容。

它的逻辑是:我们先进行前瞻查找 - 如果有符合 pattern:\w+ 的单词,我们就可以用 pattern:\1 来匹配。

为什么?因为前瞻断言查找到一个单词 pattern:\w+,将其作为一个整体,然后进行捕获形成 pattern:\1 。所以我们最终实现了一种占有型加号 pattern:+ 量词。它只将 pattern:\w+ 作为一个整体来捕获,而不会只捕获它的某一部分。

例如,在单词 subject:JavaScript 中不仅可以匹配 match:Java,而且可以忽略 match:Script ,匹配模式的其余部分。

下面是 2 个模式的对比:

alert( "JavaScript".match(/\w+Script/)); // JavaScript
alert( "JavaScript".match(/(?=(\w+))\1Script/)); // null
1
2
  1. 第一个变体 pattern:\w+ 首先捕获整个 subject:JavaScript 单词,然而接下来 pattern:+ 会一个字一个字地进行回溯,试图匹配整个模式的其余部分,直到 pattern:\w+ 匹配到了 match:Java 时,它最终才匹配成功。
  2. 第二个变体 pattern:(?=(\w+)) 前瞻查找并匹配整个单词 subject:JavaScript,然后把整个单词作为一个整体包含进 pattern:\1 中,所以在它后面就无法查找到 subject:Script 了。

当我们需要禁止 pattern:+ 进行回溯的话,我们只要把 pattern:(?=(\w+))\1 中的 pattern:\w 替换成更复杂的正则表达式就能实现了。

注意

这些文章中有更多关于占有型量词和前瞻断言的的内容:Regex: Emulate Atomic Grouping (and Possessive Quantifiers) with LookAheadMimicking Atomic Groups

我们现在用前瞻断言重写第一个例子中的正则来防止回溯吧:

let regexp = /^((?=(\w+))\2\s?)*$/;

alert( regexp.test("A good string") ); // true

let str = "An input string that takes a long time or even makes this regex to hang!";

alert( regexp.test(str) ); // false,执行得很快!
1
2
3
4
5
6
7

这里我们用 pattern:\2 代替 pattern:\1,因为这里附加了额外的外部括号。为了防止数字产生混淆,我们可以给括号命名,例如 pattern:(?<word>\w+)

// 括号被命名为 ?<word>,使用 \k<word> 来引用
let regexp = /^((?=(?<word>\w+))\k<word>\s?)*$/;

let str = "An input string that takes a long time or even makes this regex to hang!";

alert( regexp.test(str) ); // false

alert( regexp.test("A correct string") ); // true
1
2
3
4
5
6
7
8

本文所描述的问题称作“灾难性回溯(catastrophic backtracking)”,又译作“回溯陷阱”。

我们有 2 种处理它的思路:

  • 重写正则表达式,尽可能减少其中排列组合的数量。
  • 防止回溯。
灾难性回溯 has loaded