使用 Playwright 和沙箱模式实现安全的浏览器自动化

我的需求，做一个流式测试环境，一个结点测试用例跑完，进入下一个结点运行。

必要具备下面特性：
1、使用沙箱模式隔离执行环境
2、使用 Playwright 进行浏览器自动化
3. 实现了安全的变量共享机制
4、包含了错误处理和资源清理
5、支持动态代码执行
实现代码片段

1. const { chromium } = require('playwright')
3. class Sandbox {
4.   constructor() {
5.     this.sharedVariables = {
6.       browser: null,
7.       page: null,
8.       pageScreenshot: null,
9.     }
11.     // 创建代理对象来自动处理变量存取
12.     this.context = new Proxy(this.sharedVariables, {
13.       get: (target, prop) => target[prop],
14.       set: (target, prop, value) => {
15.         target[prop] = value
16.         return true
17.       },
18.     })
19.   }
21.   // 获取代理对象
22.   getContext() {
23.     return this.context
24.   }
26.   // 保留cleanup方法
27.   async cleanup() {
28.     if (this.context.browser) {
29.       await this.context.browser.close()
30.       this.context.browser = null
31.       this.context.page = null
32.     }
33.   }
34. }
36. // 修改 createSandboxExecutor 使用新的 context
37. const createSandboxExecutor = (code, sandbox) => {
38.   return new Function(
39.     'sandbox',
40.     'chromium',
41.     `
42.     return (async () => {
43.       try {
44.         const context = sandbox.getContext();
45.         
46.         // 添加全局对象到上下文
47.         Object.assign(context, {
48.           console,
49.           setTimeout,
50.           setInterval,
51.           clearTimeout,
52.           clearInterval,
53.           chromium,
54.         });
56.         with (context) {
57.           ${code}
58.         }
60.         return context;
61.       } catch (error) {
62.         throw new Error('Sandbox execution error: ' + error.message);
63.       }
64.     })()
65.   `
66.   )
67. }
69. // 执行代码片段
70. async function executeCodeSnippet(sandbox, code) {
71.   try {
72.     const executor = createSandboxExecutor(code, sandbox)
73.     return await executor(sandbox, chromium)
74.   } catch (error) {
75.     await sandbox.cleanup()
76.     throw error
77.   }
78. }
80. // 优化后的代码片段 1
81. const code1 = `
82.   browser = await chromium.launch({
83.     headless: false,
84.     args: ['--no-sandbox', '--disable-setuid-sandbox'],
85.     timeout: 10000
86.   });
87.   page = await browser.newPage();
88.   await page.setViewportSize({ width: 1280, height: 800 });
89.   console.log('Browser and page launched');
90. `
92. // 优化后的代码片段 2
93. const code2 = `
94.   try {
95.     await page.goto('https://music.163.com/', { waitUntil: 'networkidle' });
96.     pageScreenshot = Buffer.from(await page.screenshot({
97.       fullPage: true,
98.       quality: 80,
99.       type: 'jpeg'
100.     })).toString('base64');
101.     console.log('Page screenshot captured');
102.   } catch (error) {
103.     console.error('Failed to capture screenshot:', error);
104.     throw error;
105.   }
106. `
108. async function run() {
109.   const sandbox = new Sandbox()
110.   try {
111.     await executeCodeSnippet(sandbox, code1)
112.     await executeCodeSnippet(sandbox, code2)
114.     const context = sandbox.getContext()
115.     console.log('截图成功:', !!context.pageScreenshot)
116.   } catch (error) {
117.     console.error('执行错误:', error)
118.   } finally {
119.     await sandbox.cleanup()
120.   }
121. }
123. run()

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。