From c782d11816c30224fa2a39fe5e10a68e1cdfa04a Mon Sep 17 00:00:00 2001
From: xiaojunnuo <xiaojunnuo@qq.com>
Date: Wed, 25 Nov 2020 01:25:47 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=87=8D=E6=9E=84=E5=8A=A0=E9=80=9F?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E4=BC=A0=E5=8F=82=E6=96=B9=E5=BC=8F=EF=BC=8C?=
 =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=94=A8=E6=88=B7=E5=90=8D=E4=B8=AD=E9=97=B4?=
 =?UTF-8?q?=E6=9C=89=E7=A9=BA=E6=A0=BC=E6=97=B6=E5=AE=89=E8=A3=85=E6=A0=B9?=
 =?UTF-8?q?=E8=AF=81=E4=B9=A6=E7=9B=AE=E5=BD=95=E4=BD=8D=E7=BD=AE=E4=B8=8D?=
 =?UTF-8?q?=E5=AF=B9=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 packages/core/src/config.js                   |  5 ++--
 packages/core/src/config/index.js             | 16 ++++++++++++
 packages/core/src/expose.js                   |  6 ++---
 .../core/src/modules/plugin/node/index.js     |  9 ++++---
 packages/core/src/modules/server/index.js     |  8 ++++--
 packages/core/src/shell/scripts/setup-ca.js   |  9 +++----
 packages/core/src/utils/util.log.js           |  7 +++---
 packages/core/src/utils/util.proxy.js         |  3 +++
 packages/core/start/mitmproxy.js              |  5 +++-
 packages/core/start/user_config.json5         |  2 +-
 packages/gui/src/bridge/index.js              |  3 +--
 packages/gui/src/bridge/mitmproxy.js          |  9 ++++---
 packages/gui/src/utils/util.log.js            |  5 ++--
 packages/gui/src/view/components/setup-ca.vue |  2 --
 packages/gui/src/view/pages/index.vue         |  7 +++---
 packages/gui/src/view/pages/server.vue        |  4 +++
 packages/mitmproxy/src/index.js               | 19 ++++++++------
 .../mitmproxy/src/lib/proxy/common/config.js  |  8 ++++++
 .../src/lib/proxy/mitmproxy/index.js          | 25 +++++++++++--------
 .../mitmproxy/src/lib/proxy/tls/tlsUtils.js   |  5 +---
 packages/mitmproxy/src/options.js             | 11 ++++++--
 packages/mitmproxy/src/utils/util.process.js  |  8 ++++++
 22 files changed, 115 insertions(+), 61 deletions(-)
 create mode 100644 packages/core/src/utils/util.proxy.js
 create mode 100644 packages/mitmproxy/src/utils/util.process.js

diff --git a/packages/core/src/config.js b/packages/core/src/config.js
index 1ca2911a..fd922103 100644
--- a/packages/core/src/config.js
+++ b/packages/core/src/config.js
@@ -1,7 +1,7 @@
 const Shell = require('./shell')
 const lodash = require('lodash')
 const defConfig = require('./config/index.js')
-const proxyServer = require('@docmirror/mitmproxy')
+
 let configTarget = lodash.cloneDeep(defConfig)
 function _deleteDisabledItem (target) {
   lodash.forEach(target, (item, key) => {
@@ -13,6 +13,7 @@ function _deleteDisabledItem (target) {
     }
   })
 }
+
 const configApi = {
   get () {
     return configTarget
@@ -68,7 +69,7 @@ const configApi = {
     })
     if (list.length > 0) {
       const context = {
-        ca_cert_path: proxyServer.config.getDefaultCACertPath()
+        root_ca_cert_path: configApi.get().server.setting.rootCaFile.certPath
       }
       for (const item of noSetList) {
         if (item.value.indexOf('${') >= 0) {
diff --git a/packages/core/src/config/index.js b/packages/core/src/config/index.js
index aeccb9ab..ad5aab56 100644
--- a/packages/core/src/config/index.js
+++ b/packages/core/src/config/index.js
@@ -1,3 +1,14 @@
+const path = require('path')
+function getUserBasePath () {
+  const userHome = process.env.USERPROFILE
+  return path.resolve(userHome, './.dev-sidecar')
+}
+function getRootCaCertPath () {
+  return getUserBasePath() + '/dev-sidecar.ca.crt'
+}
+function getRootCaKeyPath () {
+  return getUserBasePath() + '/dev-sidecar.ca.key.pem'
+}
 module.exports = {
   server: {
     enabled: true,
@@ -7,6 +18,11 @@ module.exports = {
       script: {
         enabled: true,
         defaultDir: '../../../scripts/'
+      },
+      userBasePath: getUserBasePath(),
+      rootCaFile: {
+        certPath: getRootCaCertPath(),
+        keyPath: getRootCaKeyPath()
       }
     },
     intercepts: {
diff --git a/packages/core/src/expose.js b/packages/core/src/expose.js
index bcf633c6..a69396cc 100644
--- a/packages/core/src/expose.js
+++ b/packages/core/src/expose.js
@@ -4,15 +4,12 @@ const event = require('./event')
 const shell = require('./shell')
 const modules = require('./modules')
 const lodash = require('lodash')
-const proxyServer = require('@docmirror/mitmproxy')
-const proxyConfig = proxyServer.config
 const log = require('./utils/util.log')
 const context = {
   config,
   shell,
   status,
   event,
-  rootCaFile: proxyConfig.getDefaultCACertPath(),
   log
 }
 
@@ -127,6 +124,7 @@ module.exports = {
     shell,
     server,
     proxy,
-    plugin
+    plugin,
+    log
   }
 }
diff --git a/packages/core/src/modules/plugin/node/index.js b/packages/core/src/modules/plugin/node/index.js
index 72b1c2fa..8bd62d82 100644
--- a/packages/core/src/modules/plugin/node/index.js
+++ b/packages/core/src/modules/plugin/node/index.js
@@ -1,6 +1,6 @@
 const nodeConfig = require('./config')
 const NodePlugin = function (context) {
-  const { config, shell, event, rootCaFile, log } = context
+  const { config, shell, event, log } = context
   const nodeApi = {
     async start () {
       try {
@@ -101,16 +101,17 @@ const NodePlugin = function (context) {
        NODE_TLS_REJECT_UNAUTHORIZED: false
        */
       const nodeConfig = config.get().plugin.node
+      const rootCaCertFile = config.get().server.setting.rootCaFile.certPath
       if (nodeConfig.setting['strict-ssl']) {
         cmds.push('npm config set strict-ssl false')
       }
       if (nodeConfig.setting.cafile) {
-        cmds.push(`npm config set cafile "${rootCaFile}"`)
+        cmds.push(`npm config set cafile "${rootCaCertFile}"`)
       }
 
       if (nodeConfig.setting.NODE_EXTRA_CA_CERTS) {
-        cmds.push(`npm config set NODE_EXTRA_CA_CERTS "${rootCaFile}"`)
-        env.push({ key: 'NODE_EXTRA_CA_CERTS', value: rootCaFile })
+        cmds.push(`npm config set NODE_EXTRA_CA_CERTS "${rootCaCertFile}"`)
+        env.push({ key: 'NODE_EXTRA_CA_CERTS', value: rootCaCertFile })
       }
 
       if (nodeConfig.setting.NODE_TLS_REJECT_UNAUTHORIZED) {
diff --git a/packages/core/src/modules/server/index.js b/packages/core/src/modules/server/index.js
index df6bd51b..ae042822 100644
--- a/packages/core/src/modules/server/index.js
+++ b/packages/core/src/modules/server/index.js
@@ -4,6 +4,7 @@ const status = require('../../status')
 const lodash = require('lodash')
 const fork = require('child_process').fork
 const log = require('../../utils/util.log')
+const fs = require('fs')
 let server
 function fireStatus (status) {
   event.fire('status', { key: 'server.enabled', value: status })
@@ -45,7 +46,10 @@ const serverApi = {
       })
     }
     // fireStatus('ing') // 启动中
-    const serverProcess = fork(mitmproxyPath, [JSON.stringify(serverConfig)])
+    const basePath = serverConfig.setting.userBasePath
+    const runningConfig = basePath + '/running.json'
+    fs.writeFileSync(runningConfig, JSON.stringify(serverConfig))
+    const serverProcess = fork(mitmproxyPath, [runningConfig])
     server = {
       id: serverProcess.pid,
       process: serverProcess,
@@ -64,7 +68,7 @@ const serverApi = {
         event.fire('error', { key: 'server', value: 'EADDRINUSE', error: msg.event })
       }
     })
-    return { port: config.port }
+    return { port: runningConfig.port }
   },
   async kill () {
     if (server) {
diff --git a/packages/core/src/shell/scripts/setup-ca.js b/packages/core/src/shell/scripts/setup-ca.js
index b9080856..d9ac5204 100644
--- a/packages/core/src/shell/scripts/setup-ca.js
+++ b/packages/core/src/shell/scripts/setup-ca.js
@@ -1,17 +1,16 @@
 const Shell = require('../shell')
 const execute = Shell.execute
-const proxyServer = require('@docmirror/mitmproxy')
 const executor = {
-  async windows (exec) {
-    const cmds = ['start ' + proxyServer.config.getDefaultCACertPath()]
+  async windows (exec, { certPath }) {
+    const cmds = ['start "" "' + certPath + '"']
     // eslint-disable-next-line no-unused-vars
     const ret = await exec(cmds, { type: 'cmd' })
     return true
   },
-  async linux (exec, { port }) {
+  async linux (exec, { certPath }) {
     throw Error('暂未实现此功能')
   },
-  async mac (exec, { port }) {
+  async mac (exec, { certPath }) {
     throw Error('暂未实现此功能')
   }
 }
diff --git a/packages/core/src/utils/util.log.js b/packages/core/src/utils/util.log.js
index 484bd783..04c8a5ae 100644
--- a/packages/core/src/utils/util.log.js
+++ b/packages/core/src/utils/util.log.js
@@ -1,8 +1,7 @@
 const log4js = require('log4js')
-const path = require('path')
-const getDefaultConfigBasePath = function () {
-  const userHome = process.env.USERPROFILE
-  return path.resolve(userHome, './.dev-sidecar')
+const config = require('../config/index')
+function getDefaultConfigBasePath () {
+  return config.server.setting.userBasePath
 }
 log4js.configure({
   appenders: { std: { type: 'stdout' }, file: { type: 'file', pattern: 'yyyy-MM-dd', daysToKeep: 3, filename: getDefaultConfigBasePath() + '/logs/core.log' } },
diff --git a/packages/core/src/utils/util.proxy.js b/packages/core/src/utils/util.proxy.js
new file mode 100644
index 00000000..8dd2cbd5
--- /dev/null
+++ b/packages/core/src/utils/util.proxy.js
@@ -0,0 +1,3 @@
+const proxyConfig = require('@docmirror/mitmproxy/config.js')
+module.exports = {
+}
diff --git a/packages/core/start/mitmproxy.js b/packages/core/start/mitmproxy.js
index b133da98..f0dc2c6a 100644
--- a/packages/core/start/mitmproxy.js
+++ b/packages/core/start/mitmproxy.js
@@ -1,7 +1,10 @@
 // eslint-disable-next-line no-unused-vars
 const server = require('@docmirror/mitmproxy')
-const config = JSON.parse(process.argv[2])
+const configPath = process.argv[2]
+const fs = require('fs')
 const path = require('path')
+const configJson = fs.readFileSync(configPath)
+const config = JSON.parse(configJson)
 const scriptDir = '../../gui/extra/scripts/'
 config.setting.script.defaultDir = path.join(__dirname, scriptDir)
 server.start(config)
diff --git a/packages/core/start/user_config.json5 b/packages/core/start/user_config.json5
index cbf61b09..4cdefd17 100644
--- a/packages/core/start/user_config.json5
+++ b/packages/core/start/user_config.json5
@@ -6,7 +6,7 @@
           abort: true
         },
       }
-    }
+    },
   },
   plugin: {
     node: {
diff --git a/packages/gui/src/bridge/index.js b/packages/gui/src/bridge/index.js
index 9f19a817..caa6fd4d 100644
--- a/packages/gui/src/bridge/index.js
+++ b/packages/gui/src/bridge/index.js
@@ -8,8 +8,7 @@ const pk = require('../../package.json')
 const mitmproxyPath = path.join(__dirname, 'mitmproxy.js')
 const log = require('../utils/util.log')
 const getDefaultConfigBasePath = function () {
-  const userHome = process.env.USERPROFILE
-  return path.resolve(userHome, './.dev-sidecar')
+  return DevSidecar.api.config.get().server.setting.userBasePath
 }
 const localApi = {
   /**
diff --git a/packages/gui/src/bridge/mitmproxy.js b/packages/gui/src/bridge/mitmproxy.js
index 2d555fcc..acd487ac 100644
--- a/packages/gui/src/bridge/mitmproxy.js
+++ b/packages/gui/src/bridge/mitmproxy.js
@@ -1,12 +1,15 @@
 // eslint-disable-next-line no-unused-vars
 const server = require('@docmirror/mitmproxy')
-const config = JSON.parse(process.argv[2])
+const configPath = process.argv[2]
+
+const fs = require('fs')
 const path = require('path')
-const log = require('../utils/util.log')
+const configJson = fs.readFileSync(configPath)
+const config = JSON.parse(configJson)
+
 let scriptDir = '../extra/scripts/'
 if (process.env.NODE_ENV === 'development') {
   scriptDir = '../extra/scripts/'
 }
 config.setting.script.defaultDir = path.join(__dirname, scriptDir)
-log.debug('scriptDir', config.setting.script.defaultDir)
 server.start(config)
diff --git a/packages/gui/src/utils/util.log.js b/packages/gui/src/utils/util.log.js
index 0a99b1cf..8507a401 100644
--- a/packages/gui/src/utils/util.log.js
+++ b/packages/gui/src/utils/util.log.js
@@ -1,8 +1,7 @@
 const log4js = require('log4js')
-const path = require('path')
+const DevSidecar = require('@docmirror/dev-sidecar')
 const getDefaultConfigBasePath = function () {
-  const userHome = process.env.USERPROFILE
-  return path.resolve(userHome, './.dev-sidecar')
+  return DevSidecar.api.config.get().server.setting.userBasePath
 }
 log4js.configure({
   appenders: { std: { type: 'stdout' }, file: { type: 'file', pattern: 'yyyy-MM-dd', daysToKeep: 3, filename: getDefaultConfigBasePath() + '/logs/gui.log' } },
diff --git a/packages/gui/src/view/components/setup-ca.vue b/packages/gui/src/view/components/setup-ca.vue
index 8fccd76d..66867b22 100644
--- a/packages/gui/src/view/components/setup-ca.vue
+++ b/packages/gui/src/view/components/setup-ca.vue
@@ -24,7 +24,6 @@
 </template>
 
 <script>
-import api from '../api'
 export default {
   name: 'setup-ca',
   components: {
@@ -55,7 +54,6 @@ export default {
       this.$emit('update:visible', false)
     },
     async doSetup () {
-      await api.shell.setupCa()
       this.$emit('setup')
     }
   }
diff --git a/packages/gui/src/view/pages/index.vue b/packages/gui/src/view/pages/index.vue
index 14cda756..806fc098 100644
--- a/packages/gui/src/view/pages/index.vue
+++ b/packages/gui/src/view/pages/index.vue
@@ -151,7 +151,9 @@ export default {
     openSetupCa () {
       this.setupCa.visible = true
     },
-    handleCaSetuped () {
+    async handleCaSetuped () {
+      console.log('this.config.server.setting.rootCaFile.certPath', this.config.server.setting.rootCaFile.certPath)
+      await this.$api.shell.setupCa({ certPath: this.config.server.setting.rootCaFile.certPath })
       this.setting.rootCa = this.setting.rootCa || {}
       const rootCa = this.setting.rootCa
       rootCa.setuped = true
@@ -226,8 +228,7 @@ export default {
       })
     },
     doCheckUpdate (fromUser = true) {
-      this.update.fromUser = fromUser
-      this.$api.update.checkForUpdate(this)
+      this.$api.update.checkForUpdate(fromUser)
     },
     openExternal (url) {
       this.$api.ipc.openExternal(url)
diff --git a/packages/gui/src/view/pages/server.vue b/packages/gui/src/view/pages/server.vue
index 88c32290..e3128c47 100644
--- a/packages/gui/src/view/pages/server.vue
+++ b/packages/gui/src/view/pages/server.vue
@@ -37,6 +37,10 @@
             </a-checkbox>
             <div>开启此项之后，被代理应用关闭SSL校验也问题不大了</div>
           </a-form-item>
+          <a-form-item label="根证书：" :label-col="labelCol" :wrapper-col="wrapperCol">
+             <a-input addon-before="Cert" addon-after="选择" v-model="config.server.setting.rootCaFile.certPath" ></a-input>
+             <a-input addon-before="Key" addon-after="选择" v-model="config.server.setting.rootCaFile.keyPath" ></a-input>
+          </a-form-item>
           <a-form-item label="启用脚本" :label-col="labelCol" :wrapper-col="wrapperCol">
             <a-tooltip title="关闭后，github的clone加速链接复制也将关闭">
               <a-checkbox v-model="config.server.setting.script.enabled" >
diff --git a/packages/mitmproxy/src/index.js b/packages/mitmproxy/src/index.js
index f45fcf90..d3fe6da9 100644
--- a/packages/mitmproxy/src/index.js
+++ b/packages/mitmproxy/src/index.js
@@ -1,13 +1,8 @@
 const mitmproxy = require('./lib/proxy')
 const ProxyOptions = require('./options')
-const config = require('./lib/proxy/common/config')
+const proxyConfig = require('./lib/proxy/common/config')
 const log = require('./utils/util.log')
-function fireError (e) {
-  process.send({ type: 'error', event: e })
-}
-function fireStatus (status) {
-  process.send({ type: 'status', event: status })
-}
+const { fireError, fireStatus } = require('./utils/util.process')
 
 let server
 
@@ -42,6 +37,13 @@ function registerProcessListener () {
 const api = {
   async start (config) {
     const proxyOptions = ProxyOptions(config)
+    const setting = config.setting
+    if (setting) {
+      if (setting.userBasePath) {
+        proxyConfig.setDefaultCABasePath(setting.userBasePath)
+      }
+    }
+
     if (proxyOptions.setting && proxyOptions.setting.NODE_TLS_REJECT_UNAUTHORIZED === false) {
       process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0'
     } else {
@@ -95,5 +97,6 @@ const api = {
 
 module.exports = {
   ...api,
-  config
+  config: proxyConfig,
+  log
 }
diff --git a/packages/mitmproxy/src/lib/proxy/common/config.js b/packages/mitmproxy/src/lib/proxy/common/config.js
index ddd03fd1..168e566a 100644
--- a/packages/mitmproxy/src/lib/proxy/common/config.js
+++ b/packages/mitmproxy/src/lib/proxy/common/config.js
@@ -9,7 +9,15 @@ config.defaultPort = 1181
 
 config.caName = 'This certificate is generated locally'
 
+config.caBasePath = buildDefaultCABasePath()
+
 config.getDefaultCABasePath = function () {
+  return config.caBasePath
+}
+config.setDefaultCABasePath = function (path) {
+  config.caBasePath = path
+}
+function buildDefaultCABasePath () {
   const userHome = process.env.USERPROFILE
   return path.resolve(userHome, './.dev-sidecar')
 }
diff --git a/packages/mitmproxy/src/lib/proxy/mitmproxy/index.js b/packages/mitmproxy/src/lib/proxy/mitmproxy/index.js
index 4ac2da05..9c1f80db 100644
--- a/packages/mitmproxy/src/lib/proxy/mitmproxy/index.js
+++ b/packages/mitmproxy/src/lib/proxy/mitmproxy/index.js
@@ -21,15 +21,18 @@ module.exports = {
   }, callback) {
     // Don't reject unauthorized
     // process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0'
-
-    if (!caCertPath && !caKeyPath) {
-      const rs = this.createCA()
-      caCertPath = rs.caCertPath
-      caKeyPath = rs.caKeyPath
-      if (rs.create) {
-        log.info(`CA Cert saved in: ${caCertPath}`)
-        log.info(`CA private key saved in: ${caKeyPath}`)
-      }
+    log.info(`CA Cert read in: ${caCertPath}`)
+    log.info(`CA private key read in: ${caKeyPath}`)
+    if (!caCertPath) {
+      caCertPath = config.getDefaultCACertPath()
+    }
+    if (!caKeyPath) {
+      caKeyPath = config.getDefaultCAKeyPath()
+    }
+    const rs = this.createCA({ caCertPath, caKeyPath })
+    if (rs.create) {
+      log.info(`CA Cert saved in: ${caCertPath}`)
+      log.info(`CA private key saved in: ${caKeyPath}`)
     }
 
     port = ~~port
@@ -88,7 +91,7 @@ module.exports = {
     })
     return server
   },
-  createCA (caBasePath = config.getDefaultCABasePath()) {
-    return tlsUtils.initCA(caBasePath)
+  createCA (caPaths) {
+    return tlsUtils.initCA(caPaths)
   }
 }
diff --git a/packages/mitmproxy/src/lib/proxy/tls/tlsUtils.js b/packages/mitmproxy/src/lib/proxy/tls/tlsUtils.js
index 93d7928b..8ef7c685 100644
--- a/packages/mitmproxy/src/lib/proxy/tls/tlsUtils.js
+++ b/packages/mitmproxy/src/lib/proxy/tls/tlsUtils.js
@@ -237,10 +237,7 @@ utils.getMappingHostNamesFormCert = function (cert) {
 }
 
 // sync
-utils.initCA = function (basePath = config.getDefaultCABasePath()) {
-  const caCertPath = path.resolve(basePath, config.caCertFileName)
-  const caKeyPath = path.resolve(basePath, config.caKeyFileName)
-
+utils.initCA = function ({ caCertPath, caKeyPath }) {
   try {
     fs.accessSync(caCertPath, fs.F_OK)
     fs.accessSync(caKeyPath, fs.F_OK)
diff --git a/packages/mitmproxy/src/options.js b/packages/mitmproxy/src/options.js
index b85b8d74..10a12621 100644
--- a/packages/mitmproxy/src/options.js
+++ b/packages/mitmproxy/src/options.js
@@ -48,13 +48,14 @@ module.exports = (config) => {
 
   const dnsMapping = config.dns.mapping
   const serverConfig = config
-  return {
+  const setting = serverConfig.setting
+  const options = {
     port: serverConfig.port,
     dnsConfig: {
       providers: dnsUtil.initDNS(serverConfig.dns.providers),
       mapping: dnsMapping
     },
-    setting: serverConfig.setting,
+    setting,
     sslConnectInterceptor: (req, cltSocket, head) => {
       const hostname = req.url.split(':')[0]
       const inWhiteList = matchHostname(whiteList, hostname) != null
@@ -103,4 +104,10 @@ module.exports = (config) => {
       return matchIntercepts
     }
   }
+
+  if (setting.rootCaFile) {
+    options.caCertPath = setting.rootCaFile.certPath
+    options.caKeyPath = setting.rootCaFile.keyPath
+  }
+  return options
 }
diff --git a/packages/mitmproxy/src/utils/util.process.js b/packages/mitmproxy/src/utils/util.process.js
new file mode 100644
index 00000000..21a9bc26
--- /dev/null
+++ b/packages/mitmproxy/src/utils/util.process.js
@@ -0,0 +1,8 @@
+module.exports = {
+  fireError (e) {
+    process.send({ type: 'error', event: e })
+  },
+  fireStatus (status) {
+    process.send({ type: 'status', event: status })
+  }
+}