如何检测JavaScript中的死循环示例详解

前言

如果我们需要执行用户写的代码，如和避免死循环？我们最近遇到了这个问题，因为写错代码很常见，所以我们进行了一下尝试。

首先 parse用户的代码为AST

import { parse } from '@babel/parser'
const ast = parse(code)

然后我们准备一下需要插入的代码。

代码有两部分，第一部分是function定义，实际上可以在头部插入，所以字符串就够了。第二部分是function的调用，这部分需要插入到AST中，所以也需要parse一下。

const prefix = `
 let __count = 0
 const __detectInfiniteLoop = () => {
  if (__count > 10000) {
   throw new Error('Infinite Loop detected')
  }
  __count += 1
 }
`

const detector = parse(`__detectInfiniteLoop()`)

接下来就找到 while/for/do..while 的位置，然后插入detector的调用。

import traverse from '@babel/traverse'
traverse(ast, {
 ForStatement: function (path) {
  path.node.body.body.push(...detector.program.body)
 },
 WhileStatement: function (path) {
  path.node.body.body.push(...detector.program.body)
 },
 DoWhileStatement: function (path) {
  path.node.body.body.push(...detector.program.body)
 }
})

AST修改好了，最后一步就是生成最终的代码，然后放到iframe中执行。

import generate from '@babel/generator'
const newCode = prefix + generate(ast).code

这个方法不是完美的，不过满足了我们自己的需求。你可以根据需要进行一下调整。

到此这篇关于如何检测JavaScript中的死循环的文章就介绍到这了,更多相关检测JavaScript死循环内容请搜索来客网以前的文章或继续浏览下面的相关文章希望大家以后多多支持来客网！