Added interface field (local IP addresses)

Fixed ProfileExport plugin
Xzandro · Mar 11, 2017 · b9cf9d1 · b9cf9d1
1 parent 9c014a4
commit b9cf9d1
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 3 deletions.
diff --git a/app/components/Head.js b/app/components/Head.js
@@ -2,7 +2,7 @@ const { ipcRenderer, remote } = require('electron');
 
 import React from 'react';
 
-import { Menu, Button, Input } from 'semantic-ui-react';
+import { Menu, Button, Input, Select } from 'semantic-ui-react';
 
 class Head extends React.Component {
   constructor() {
@@ -28,8 +28,14 @@ class Head extends React.Component {
   }
 
   render () {
+    const interfaces = ipcRenderer.sendSync('proxyGetInterfaces').map((interfaceEntry, i) => {
+      return { "key": i, "text": interfaceEntry, "value": i }
+    }) 
     return (
       <Menu className="main-menu">
+        <Menu.Item>
+          <Select label='Interfaces' options={interfaces} defaultValue={0} />
+        </Menu.Item>
         <Menu.Item>
           <Input label='Port' defaultValue={this.config.Proxy.port} onChange={this.changePort.bind(this)} />
         </Menu.Item>

diff --git a/app/main.js b/app/main.js
@@ -60,6 +60,10 @@ ipcMain.on('proxyIsRunning', (event, arg) => {
   event.returnValue = proxy.isRunning();
 })
 
+ipcMain.on('proxyGetInterfaces', (event, arg) => {
+  event.returnValue = proxy.getInterfaces();
+})
+
 ipcMain.on('proxyStart', (event, arg) => {
   proxy.start(config.Proxy.port);
 })

diff --git a/app/plugins/profile-export.js b/app/plugins/profile-export.js
@@ -6,7 +6,7 @@ module.exports = {
     enabled: true
   },
   pluginName: 'ProfileExport',
-  init(proxy) {
+  init(proxy, config) {
     proxy.on('HubUserLogin', (req, resp) => {
       if (config.enabled)
         this.writeProfileToFile(proxy, req, resp);

diff --git a/app/proxy/SWProxy.js b/app/proxy/SWProxy.js
@@ -1,6 +1,7 @@
 const EventEmitter = require('events');
 const http = require('http')
 const httpProxy = require('http-proxy');
+const os = require('os');
 
 const {decrypt_request, decrypt_response} = require('./smon_decryptor');
 
@@ -10,6 +11,7 @@ class SWProxy extends EventEmitter {
     this.httpServer = null;
     this.proxy = null;
     this.logEntries = [];
+    this.addresses = [];
   }
   start(port) {
     const self = this; // so event callbacks can access this SWProxy class
@@ -62,7 +64,6 @@ class SWProxy extends EventEmitter {
 
       this.proxy.web(req, resp, { target: req.url, prependPath: false });
     }).listen(port);
-
     this.log({ type: 'info', source: 'proxy', message: `Now listening on port ${port}` })
   }
 
@@ -72,6 +73,19 @@ class SWProxy extends EventEmitter {
     this.log({ type: 'info', source: 'proxy', message: `Proxy stopped` })
   }
 
+  getInterfaces() {
+    let interfaces = os.networkInterfaces();
+    for (let k in interfaces) {
+        for (let k2 in interfaces[k]) {
+            let address = interfaces[k][k2];
+            if (address.family === 'IPv4' && !address.internal) {
+                this.addresses.push(address.address);
+            }
+        }
+    }
+    return this.addresses;
+  }
+
   isRunning() {
     if (this.httpServer && this.httpServer.address()) {
       return true;