feat: support encrypt preload.js

dusionlike · Aug 10, 2022 · 8207267 · 8207267
1 parent 177865b
commit 8207267
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 2 deletions.
diff --git a/playground/electron-builder.json b/playground/electron-builder.json
@@ -4,7 +4,7 @@
   "directories": {
     "output": "release/${version}"
   },
-  "files": ["main.js", "renderer"],
+  "files": ["main.js", "preload.js", "renderer"],
   "mac": {
     "artifactName": "${productName}_${version}.${ext}",
     "target": ["dmg"]

diff --git a/playground/main.js b/playground/main.js
@@ -1,9 +1,14 @@
+const path = require('path')
 const { app, BrowserWindow } = require('electron')
 
 function createWindow() {
   const mainWindow = new BrowserWindow({
     width: 800,
     height: 600,
+    webPreferences: {
+      sandbox: false,
+      preload: path.join(__dirname, 'preload.js'),
+    },
   })
 
   if (!app.isPackaged) {

diff --git a/playground/preload.js b/playground/preload.js
@@ -0,0 +1,11 @@
+window.addEventListener('DOMContentLoaded', () => {
+  const el = document.createElement('p')
+  el.innerHTML = 'preload.js is encrypted'
+  document.body.append(el)
+
+  for (const type of ['chrome', 'node', 'electron']) {
+    const el = document.createElement('p')
+    el.innerHTML = `${type}-version: ${process.versions[type]}`
+    document.body.append(el)
+  }
+})
diff --git a/src/index.ts b/src/index.ts
@@ -46,6 +46,18 @@ exports.default = async function (context: AfterPackContext) {
     'utf-8'
   )
 
+  // 将renderer preload.js加密
+  const rendererPreloadJsPath = path.join(mainDir, 'preload.js')
+  if (fs.existsSync(rendererPreloadJsPath)) {
+    const rendererPreloadJsCPath = path.join(mainDir, 'preload-c.jsc')
+    await compileToBytenode(rendererPreloadJsPath, rendererPreloadJsCPath)
+    await fs.promises.writeFile(
+      rendererPreloadJsPath,
+      `"use strict";require('bytenode');require('v8').setFlagsFromString('--no-lazy');require('./preload-c.jsc');`,
+      'utf-8'
+    )
+  }
+
   const rendererDir = path.join(mainDir, 'renderer')
 
   // 加密渲染进程
@@ -64,7 +76,7 @@ exports.default = async function (context: AfterPackContext) {
     'utf-8'
   )
 
-  await fs.promises.rm(tempAppDir, { recursive: true })
+  // await fs.promises.rm(tempAppDir, { recursive: true })
 }
 
 /**