NodeJS中VM2设置console为'redirect'时捕获控制台输出的方法

嘿，作为Node.js新手能想到用VM2来隔离运行用户代码已经很赞啦！你已经把console配置设为'redirect'，这一步走对了——接下来只需要给VM实例监听console事件，就能轻松捕获沙箱里的所有控制台输出了。我给你写个完整的示例，一步步讲清楚：

完整实现代码 #

const { NodeVM } = require('vm2');

// 初始化你的VM实例，保留你原有的配置
const vm = new NodeVM({
  console: 'redirect',
  timeout: 30000,
  sandbox: { request, state, response },
  require: { external: true }
});

// 关键：监听console事件捕获沙箱输出
vm.on('console', (type, ...args) => {
  // type 对应控制台方法类型：'log'、'warn'、'error'、'info'等
  // args 是沙箱内调用console时传入的所有参数（和普通console参数逻辑一致）
  
  // 示例：把沙箱输出打印到主进程的控制台
  console.log(`[沙箱-${type}]`, ...args);

  // 如果只想处理log类型的输出，可以加判断
  if (type === 'log') {
    // 这里可以做自定义处理：比如把消息存到变量、返回给用户等
    // 比如收集所有log消息到一个数组里
    // sandboxLogs.push(args.join(' '));
    console.log('捕获到一条log消息:', args.join(' '));
  }
});

// 执行沙箱内的代码（示例代码，替换成你实际要运行的内容）
try {
  vm.run(`
    // 沙箱内的测试代码
    console.log('Hello 沙箱！');
    console.warn('注意：这是沙箱里的警告');
    console.error('哎呀，沙箱里出错了');
    console.log('支持多参数输出', 123, { key: 'value' });
  `);
} catch (err) {
  console.error('沙箱代码执行出错:', err);
}

核心要点说明 #

• 事件监听时机：一定要在调用vm.run()或vm.runFile()之前添加监听器，不然会漏掉早期的输出。
• 参数解析：
  • type：准确对应沙箱内调用的控制台方法，比如console.log对应'log'，console.error对应'error'。
  • args：和普通console方法的参数完全一致，支持多个值、对象、甚至复杂类型，你可以像处理普通控制台参数一样处理它们。
• 自定义处理：你可以根据type区分不同类型的输出，把log存到数据库、返回给前端，或者过滤掉不需要的内容，完全按需来。

如果是运行外部文件（vm.runFile('user-code.js')），监听逻辑完全一样，只要保证监听器在执行前挂载就行。

内容的提问来源于stack exchange，提问作者Greg