data.js 3.37 KB
const { ipcRenderer } = require("electron");
const { BrowserWindow } = require("electron").remote;
const path = require('path')
const fs = require('fs')
var exec = require('child_process').exec, child; //!!!! 2.
var io = require('socket.io').listen(3000);
var mysql = require('mysql');


const connect_btn = document.getElementById("connect_btn");

connect_btn.addEventListener("click", () => {
    win = new BrowserWindow({
        width: 600,
        height: 450,
        frame: false,
        alwaysOnTop: true,
        webPreferences: {
          nodeIntegration: true
        }
      });

      //win.webContents.openDevTools();
      win.on("close", () => {
        win = null;
      });
      win.loadFile("./src/connect.html");
      win.show();
    
});

var connection = mysql.createConnection({
  host:'localhost',
  user:'root',
  password:'1234'
});

connection.connect();

function apkInstall(){
	cmd_install = exec('adb -s '+device+' install -r app-release.apk',function(error, stdout, stderr){
        cc.innerHTML="install apk";
        cmd_exec = exec('adb -s '+device+' shell am start -n com.example.dataextraction/com.example.dataextraction.MainActivity',function(error, stdout, stderr){
          cc.innerHTML="start app";
          cc.innerHTML="extract data";
        })
    });
    cmd_install();
}

var device = new String();
ipcRenderer.on("device", (e, arg) => {
    device = arg;
    connect_btn.innerHTML = "Connected : ".concat(device);
    //device.innerHTML = arg.toString();

    apkInstall();
});


function extractDB () {
  cmd_backup = exec('adb -s '+device+' backup com.example.dataextraction -f backup.dataextraction.ad', function(error, stdout, stderr){
    cc.innerHTML="backup app";
    setTimeout(function(){
      cmd_unpack = exec('java -jar abe.jar unpack backup.dataextraction.ad dataextraction.tar',function(error, stdout, stderr){
        cc.innerHTML="unpack backup";
        cmd_unzip = exec('tar xvf dataextraction.tar', function(error, stdout, stderr){
          cc.innerHTML="unzip tar";
          cmd_moveIdb = fs.rename('apps/com.example.dataextraction/db/InnerDatabase.db', 'InnerDatabase.db', function(){
            cc.innerHTML="move InnerDB File";
            cmd_moveNdb = fs.rename('apps/com.example.dataextraction/db/networkDatabase.db', 'networkDatabase.db',function(){
              cc.innerHTML="move NetworkDB File";
              cmd_sqlIdb = exec('sqlite3 InnerDatabase.db .dump | python SQLiteToMysql.py > InnerDatabase.sql',function(error, stdout, stderr){
                cc.innerHTML="make InnerDB SQL";
                cmd_sqlNdb = exec('sqlite3 networkDatabase.db .dump | python SQLiteToMysql_network.py > networkDatabase.sql', function(error, stdout, stderr){
                  cc.innerHTML="make NetworkDB SQL";

                  connection.query('DROP DATABASE IF EXISTS DATAEXTRACTION; CREATE DATABASE DATAEXTRACTION CHARACTER SET utf8mb4 COLLATE utf8mb4_bin',function(error, result,fields){
                  connection.query('DROP DATABASE IF EXISTS DATAEXTRACTION_NETWORK; CREATE DATABASE DATAEXTRACTION_NETWORK',function(error, result, fields){
                  })
                });
              })
            })
          })
        })
      })
    })
    },7000);
  });
  //cmd_backup();
}

//db insert done
io.on('connection', function(socket){
	cc.innerHTML="connect";
	var instanceId = socket.id;

	socket.on('alert', function(data){
		extractDB();
  })
})