瀏覽代碼

Ported the GroupIPAddresses operation.

Callum Fraser 7 年之前
父節點
當前提交
b7bdd4eed6
共有 1 個文件被更改,包括 134 次插入0 次删除
  1. 134 0
      src/core/operations/GroupIPAddresses.mjs

+ 134 - 0
src/core/operations/GroupIPAddresses.mjs

@@ -0,0 +1,134 @@
+/**
+ * @author n1474335 [n1474335@gmail.com]
+ * @copyright Crown Copyright 2016
+ * @license Apache-2.0
+ */
+
+import Operation from "../Operation";
+import Utils from "../Utils";
+import {_ipv6ToStr, _genIpv6Mask, IPV4_REGEX, _strToIpv6,  _ipv4ToStr, IPV6_REGEX, _strToIpv4} from "../lib/Ip";
+
+/**
+ * Group IP addresses operation
+ */
+class GroupIPAddresses extends Operation {
+
+    /**
+     * GroupIPAddresses constructor
+     */
+    constructor() {
+        super();
+
+        this.name = "Group IP addresses";
+        this.module = "JSBN";
+        this.description = "Groups a list of IP addresses into subnets. Supports both IPv4 and IPv6 addresses.";
+        this.inputType = "string";
+        this.outputType = "string";
+        this.args = [
+            {
+                "name": "Delimiter",
+                "type": "option",
+                "value": ["Line feed", "CRLF", "Space", "Comma", "Semi-colon"]
+            },
+            {
+                "name": "Subnet (CIDR)",
+                "type": "number",
+                "value": 24
+            },
+            {
+                "name": "Only show the subnets",
+                "type": "boolean",
+                "value": false,
+            }
+        ];
+    }
+
+    /**
+     * @param {string} input
+     * @param {Object[]} args
+     * @returns {string}
+     */
+    run(input, args) {
+        const delim = Utils.charRep(args[0]),
+            cidr = args[1],
+            onlySubnets = args[2],
+            ipv4Mask = cidr < 32 ? ~(0xFFFFFFFF >>> cidr) : 0xFFFFFFFF,
+            ipv6Mask = _genIpv6Mask(cidr),
+            ips = input.split(delim),
+            ipv4Networks = {},
+            ipv6Networks = {};
+        let match = null,
+            output = "",
+            ip = null,
+            network = null,
+            networkStr = "",
+            i;
+
+        if (cidr < 0 || cidr > 127) {
+            return "CIDR must be less than 32 for IPv4 or 128 for IPv6";
+        }
+
+        // Parse all IPs and add to network dictionary
+        for (i = 0; i < ips.length; i++) {
+            if ((match = IPV4_REGEX.exec(ips[i]))) {
+                ip = _strToIpv4(match[1]) >>> 0;
+                network = ip & ipv4Mask;
+
+                if (ipv4Networks.hasOwnProperty(network)) {
+                    ipv4Networks[network].push(ip);
+                } else {
+                    ipv4Networks[network] = [ip];
+                }
+            } else if ((match = IPV6_REGEX.exec(ips[i]))) {
+                ip = _strToIpv6(match[1]);
+                network = [];
+                networkStr = "";
+
+                for (let j = 0; j < 8; j++) {
+                    network.push(ip[j] & ipv6Mask[j]);
+                }
+
+                networkStr = _ipv6ToStr(network, true);
+
+                if (ipv6Networks.hasOwnProperty(networkStr)) {
+                    ipv6Networks[networkStr].push(ip);
+                } else {
+                    ipv6Networks[networkStr] = [ip];
+                }
+            }
+        }
+
+        // Sort IPv4 network dictionaries and print
+        for (network in ipv4Networks) {
+            ipv4Networks[network] = ipv4Networks[network].sort();
+
+            output += _ipv4ToStr(network) + "/" + cidr + "\n";
+
+            if (!onlySubnets) {
+                for (i = 0; i < ipv4Networks[network].length; i++) {
+                    output += "  " + _ipv4ToStr(ipv4Networks[network][i]) + "\n";
+                }
+                output += "\n";
+            }
+        }
+
+        // Sort IPv6 network dictionaries and print
+        for (networkStr in ipv6Networks) {
+            //ipv6Networks[networkStr] = ipv6Networks[networkStr].sort();  TODO
+
+            output += networkStr + "/" + cidr + "\n";
+
+            if (!onlySubnets) {
+                for (i = 0; i < ipv6Networks[networkStr].length; i++) {
+                    output += "  " + _ipv6ToStr(ipv6Networks[networkStr][i], true) + "\n";
+                }
+                output += "\n";
+            }
+        }
+
+        return output;
+    }
+
+}
+
+export default GroupIPAddresses;