在Headless Chrome/Puppeteer中禁用file://协议，保障PDF生成安全

一、用Puppeteer原生功能彻底禁用file://协议 #

无需依赖输入清理，直接通过Chromium启动参数+Puppeteer请求拦截实现双重防护：

1. 核心启动参数 #

添加Chromium原生启动参数--disable-local-file-accesses，直接阻止浏览器访问本地文件系统：

const browser = await puppeteer.launch({
  args: [
    '--disable-local-file-accesses', // 禁用本地文件访问的核心参数
    '--no-sandbox', // 仅在容器/隔离环境下使用，生产环境优先保留沙箱
    '--disable-dev-shm-usage',
    '--max-memory-limit=256MB' // 额外限制内存，防止资源耗尽
  ]
});

2. 双重保险：请求拦截 #

即使启动参数失效，通过Puppeteer的请求拦截器直接终止所有file://协议请求：

const page = await browser.newPage();
await page.setRequestInterception(true);

page.on('request', (req) => {
  if (req.url().startsWith('file://')) {
    req.abort(); // 直接拒绝file://请求
  } else if (req.resourceType() === 'script') {
    req.abort(); // 可选：禁用所有外部脚本，进一步降低风险
  } else {
    req.continue();
  }
});

// 加载用户内容时用setContent，而非goto，避免意外加载外部页面
await page.setContent(userSubmittedHtml, {
  waitUntil: 'domcontentloaded',
  timeout: 10000 // 设置超时，防止恶意内容阻塞
});

const pdfBuffer = await page.pdf({ format: 'A4' });
await browser.close();

二、该场景下的其他安全风险及防护 #

• SSRF与外部资源攻击：用户HTML可能包含指向恶意域名的图片/脚本，发起服务器端请求伪造。防护：拦截所有外部资源请求，仅允许必要的内部资源（或完全禁用外部加载）。
• 恶意脚本执行：用户提交的HTML中可能嵌入窃取服务器信息的脚本。防护：启用Chromium沙箱（生产环境避免--no-sandbox），配置严格的Content Security Policy（CSP），比如通过page.setExtraHTTPHeaders设置Content-Security-Policy: default-src 'none'; style-src 'unsafe-inline'，禁止脚本执行。
• 资源耗尽攻击：超大HTML、大量高清图片可能导致Chromium进程崩溃或耗尽服务器资源。防护：设置页面加载超时（如上例的timeout:10000），限制进程内存，每个PDF生成请求使用独立的浏览器实例，用完立即销毁。
• CSS/HTML渲染漏洞：恶意构造的CSS可能触发Chromium的渲染引擎漏洞。防护：保持Puppeteer/Chromium版本最新，及时更新安全补丁。

三、替代Headless Chromium/Puppeteer的PDF生成方案 #

如果不想用Puppeteer，以下是本地部署、无第三方API依赖的方案：

• wkhtmltopdf：基于WebKit的轻量命令行工具，支持HTML转PDF。可通过--disable-local-file-access禁用本地文件访问，--disable-javascript禁用脚本，适合对CSS兼容性要求不高的场景。
• WeasyPrint：Python生态的PDF生成工具，支持现代CSS标准。可通过配置资源加载器禁止本地文件访问，适合Python栈项目，安全性可控。
• Playwright：微软开发的自动化工具，与Puppeteer类似，但支持多浏览器。同样可通过启动参数--disable-local-file-accesses和请求拦截禁用file://协议，API兼容Puppeteer，迁移成本低。
• Flying Saucer：Java栈的HTML/CSS转PDF库，可自定义资源加载策略，禁止本地文件访问，适合Java后端项目。

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