优化

‘liusuyi’

2023-10-21 ae92b6823499cfab77201b4033922fa373fb37a3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
"use strict";
 
const execa = require("execa");
const os = require("os");
const net = require("net");
 
const gwArgs = "path Win32_NetworkAdapterConfiguration where IPEnabled=true get DefaultIPGateway,GatewayCostMetric,IPConnectionMetric,Index /format:table".split(" ");
const ifArgs = index => `path Win32_NetworkAdapter where Index=${index} get NetConnectionID,MACAddress /format:table`.split(" ");
 
const spawnOpts = {
  windowsHide: true,
};
 
// Parsing tables like this. The final metric is GatewayCostMetric + IPConnectionMetric
//
// DefaultIPGateway             GatewayCostMetric  Index  IPConnectionMetric
// {"1.2.3.4", "2001:db8::1"}   {0, 256}           12     25
// {"2.3.4.5"}                  {25}               12     55
function parseGwTable(gwTable, family) {
  let [bestGw, bestMetric, bestId] = [null, null, null];
 
  for (let line of (gwTable || "").trim().split(/\r?\n/).splice(1)) {
    line = line.trim();
    const [_, gwArr, gwCostsArr, id, ipMetric] = /({.+?}) +?({.+?}) +?([0-9]+) +?([0-9]+)/g.exec(line) || [];
    if (!gwArr) continue;
 
    const gateways = (gwArr.match(/"(.+?)"/g) || []).map(match => match.substring(1, match.length - 1));
    const gatewayCosts = (gwCostsArr.match(/[0-9]+/g) || []);
 
    for (const [index, gateway] of Object.entries(gateways)) {
      if (!gateway || `v${net.isIP(gateway)}` !== family) continue;
 
      const metric = parseInt(gatewayCosts[index]) + parseInt(ipMetric);
      if (!bestGw || metric < bestMetric) {
        [bestGw, bestMetric, bestId] = [gateway, metric, id];
      }
    }
  }
 
  if (bestGw) return [bestGw, bestId];
}
 
function parseIfTable(ifTable) {
  const line = (ifTable || "").trim().split("\n")[1];
 
  let [mac, name] = line.trim().split(/\s+/);
  mac = mac.toLowerCase();
 
  // try to get the interface name by matching the mac to os.networkInterfaces to avoid wmic's encoding issues
  // https://github.com/silverwind/default-gateway/issues/14
  for (const [osname, addrs] of Object.entries(os.networkInterfaces())) {
    for (const addr of addrs) {
      if (addr && addr.mac && addr.mac.toLowerCase() === mac) {
        return osname;
      }
    }
  }
  return name;
}
 
const promise = async family => {
  const {stdout} = await execa("wmic", gwArgs, spawnOpts);
  const [gateway, id] = parseGwTable(stdout, family) || [];
 
  if (!gateway) {
    throw new Error("Unable to determine default gateway");
  }
 
  let name;
  if (id) {
    const {stdout} = await execa("wmic", ifArgs(id), spawnOpts);
    name = parseIfTable(stdout);
  }
 
  return {gateway, interface: name ? name : null};
};
 
const sync = family => {
  const {stdout} = execa.sync("wmic", gwArgs, spawnOpts);
  const [gateway, id] = parseGwTable(stdout, family) || [];
 
  if (!gateway) {
    throw new Error("Unable to determine default gateway");
  }
 
  let name;
  if (id) {
    const {stdout} = execa.sync("wmic", ifArgs(id), spawnOpts);
    name = parseIfTable(stdout);
  }
 
  return {gateway, interface: name ? name : null};
};
 
module.exports.v4 = () => promise("v4");
module.exports.v6 = () => promise("v6");
 
module.exports.v4.sync = () => sync("v4");
module.exports.v6.sync = () => sync("v6");