54 lines
1.5 KiB
TypeScript
54 lines
1.5 KiB
TypeScript
import { AttackMethod, Proxy, ProxyProtocol } from "./lib";
|
|
|
|
const DEFAULT_HTTP_PORT = 8080;
|
|
const DEFAULT_PROTOCOL: ProxyProtocol = "http";
|
|
|
|
const COMMON_PORTS: { [port: number]: ProxyProtocol } = {
|
|
80: "http",
|
|
443: "https",
|
|
1080: "socks5",
|
|
1081: "socks4",
|
|
8080: "http",
|
|
8443: "https",
|
|
};
|
|
|
|
const METHODS: { [key in AttackMethod]: ProxyProtocol[] } = {
|
|
http_flood: ["http", "https", "socks4", "socks5"],
|
|
http_bypass: ["http", "https", "socks4", "socks5"],
|
|
http_slowloris: ["socks4", "socks5"],
|
|
tcp_flood: ["socks4", "socks5"],
|
|
minecraft_ping: ["socks4", "socks5"],
|
|
};
|
|
|
|
/**
|
|
* Attempts to infer the protocol based on the port.
|
|
*/
|
|
function inferProtocol(port: number | undefined): ProxyProtocol {
|
|
if (port !== undefined && COMMON_PORTS[port]) {
|
|
return COMMON_PORTS[port];
|
|
}
|
|
return DEFAULT_PROTOCOL;
|
|
}
|
|
|
|
/**
|
|
* Ensures a proxy object is safe and normalized by adding default values if missing.
|
|
*/
|
|
function normalizeProxy(proxy: Proxy): Proxy {
|
|
const normalizedPort = proxy.port || DEFAULT_HTTP_PORT;
|
|
const normalizedProtocol = proxy.protocol || inferProtocol(normalizedPort);
|
|
|
|
return {
|
|
...proxy,
|
|
port: normalizedPort,
|
|
protocol: normalizedProtocol,
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Filters proxies based on the attack method and ensures safe parsing of proxies.
|
|
*/
|
|
export function filterProxies(proxies: Proxy[], method: AttackMethod): Proxy[] {
|
|
return proxies
|
|
.map(normalizeProxy)
|
|
.filter((proxy) => METHODS[method].includes(proxy.protocol));
|
|
}
|