67 lines
1.8 KiB
JavaScript
67 lines
1.8 KiB
JavaScript
|
import fs from "fs";
|
||
|
|
import path from "path";
|
||
|
|
import { fileURLToPath } from "url";
|
||
|
|
|
||
|
|
import * as csv from "csv-parse";
|
||
|
|
|
||
|
|
import { exec } from "child_process";
|
||
|
|
import { getCloudASN } from "./getCloudASN.js";
|
||
|
|
import { generateWellKnownIPs } from "./wellKnown";
|
||
|
|
|
||
|
|
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
||
|
|
|
||
|
|
const getCloudASN = async () =>
|
||
|
|
new Promise(async (resolve, reject) => {
|
||
|
|
const results = [];
|
||
|
|
const cloudList = (
|
||
|
|
await fs.promises.readFile(path.join(__dirname, "./cloud_list.md"))
|
||
|
|
)
|
||
|
|
.toString()
|
||
|
|
.split("\n");
|
||
|
|
|
||
|
|
fs.createReadStream("./as.csv")
|
||
|
|
.pipe(csv.parse({ delimiter: ",", from_line: 2 }))
|
||
|
|
.on("data", function ([asn, handle, description]) {
|
||
|
|
if (cloudList.includes(handle) || cloudList.includes(description)) {
|
||
|
|
results.push({ asn, handle, description });
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.on("end", function () {
|
||
|
|
resolve(results);
|
||
|
|
})
|
||
|
|
.on("error", function (error) {
|
||
|
|
console.log(error.message);
|
||
|
|
reject(error.message);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
const buildIpRange = async () => {
|
||
|
|
const dir = path.join(__dirname, "ip_ranges");
|
||
|
|
if (!fs.existsSync(path.join(__dirname, "ip_ranges"))) {
|
||
|
|
fs.mkdirSync(path.join(__dirname, "ip_ranges"));
|
||
|
|
}
|
||
|
|
const cloudASM = await getCloudASN();
|
||
|
|
for (const { asn, handle } of cloudASM) {
|
||
|
|
try {
|
||
|
|
await new Promise((resolve, reject) => {
|
||
|
|
exec(
|
||
|
|
`bgpq3 -h whois.radb.net -j AS${asn} > ip_ranges/${handle}.json`,
|
||
|
|
(error, stdout, stderr) => {
|
||
|
|
if (error) {
|
||
|
|
console.error(`exec error: ${error}`);
|
||
|
|
reject(error);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
resolve();
|
||
|
|
}
|
||
|
|
);
|
||
|
|
});
|
||
|
|
} catch (error) {
|
||
|
|
console.error(`${asn}, ${handle} failed`);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
generateWellKnownIPs();
|
||
|
|
};
|
||
|
|
|
||
|
|
buildIpRange();
|