109 lines
4.2 KiB
TypeScript
109 lines
4.2 KiB
TypeScript
import {ipcMain} from "electron";
|
|
import { logDebug, logError, logInfo, LogModule, logWarn } from "electron/utils/log";
|
|
|
|
const {exec, spawn} = require("child_process");
|
|
|
|
type LocalPort = {
|
|
protocol: string;
|
|
ip: string;
|
|
port: number;
|
|
}
|
|
|
|
export const initLocalApi = () => {
|
|
const command = process.platform === 'win32'
|
|
? 'netstat -a -n'
|
|
: 'netstat -an | grep LISTEN';
|
|
|
|
ipcMain.on("local.getLocalPorts", async (event, args) => {
|
|
logInfo(LogModule.APP, "Starting to retrieve local ports");
|
|
// 执行命令
|
|
exec(command, (error, stdout, stderr) => {
|
|
if (error) {
|
|
logError(LogModule.APP, `getLocalPorts - error: ${error.message}`);
|
|
return;
|
|
}
|
|
if (stderr) {
|
|
logWarn(LogModule.APP, `getLocalPorts - stderr: ${stderr}`);
|
|
return;
|
|
}
|
|
|
|
logDebug(LogModule.APP, `Command output: ${stdout}`);
|
|
let ports = [];
|
|
if (stdout) {
|
|
if (process.platform === 'win32') {
|
|
// window
|
|
ports = stdout.split('\r\n')
|
|
.filter(f => f.indexOf('TCP') !== -1 || f.indexOf('UDP') !== -1)
|
|
.map(m => {
|
|
const cols = m.split(' ')
|
|
.filter(f => f != '')
|
|
const local = cols[1]
|
|
const s = local.lastIndexOf(":")
|
|
let localIP = local.slice(0, s);
|
|
let localPort = local.slice(s - local.length + 1);
|
|
const singe: LocalPort = {
|
|
protocol: cols[0],
|
|
ip: localIP,
|
|
port: localPort
|
|
}
|
|
|
|
return singe;
|
|
})
|
|
} else if (process.platform === 'darwin') {
|
|
// mac
|
|
ports = stdout.split('\n')
|
|
.filter(m => {
|
|
const cols = m.split(' ')
|
|
.filter(f => f != '')
|
|
const local = cols[3]
|
|
return local
|
|
})
|
|
.map(m => {
|
|
const cols = m.split(' ')
|
|
.filter(f => f != '')
|
|
const local = cols[3]
|
|
const s = local.lastIndexOf(".")
|
|
let localIP = local.slice(0, s);
|
|
let localPort = local.slice(s - local.length + 1);
|
|
const singe: LocalPort = {
|
|
protocol: cols[0],
|
|
ip: localIP,
|
|
port: localPort
|
|
}
|
|
return singe;
|
|
})
|
|
|
|
} else if (process.platform === 'linux') {
|
|
ports = stdout.split('\n')
|
|
.filter(f =>
|
|
f.indexOf('tcp') !== -1||
|
|
f.indexOf('tcp6') !== -1||
|
|
f.indexOf('udp') !== -1 ||
|
|
f.indexOf('udp6') !== -1
|
|
).map(m => {
|
|
const cols = m.split(' ')
|
|
.filter(f => f != '')
|
|
const local = cols[3]
|
|
const s = local.lastIndexOf(":")
|
|
let localIP = local.slice(0, s);
|
|
let localPort = local.slice(s - local.length + 1);
|
|
const singe: LocalPort = {
|
|
protocol: cols[0],
|
|
ip: localIP,
|
|
port: localPort
|
|
}
|
|
return singe;
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
ports.sort((a, b) => a.port - b.port);
|
|
|
|
event.reply("local.getLocalPorts.hook", {
|
|
data: ports
|
|
});
|
|
});
|
|
});
|
|
}
|