Add some util functions
This commit is contained in:
parent
8733556ac2
commit
81717ac746
@ -1,7 +1,15 @@
|
|||||||
|
// noinspection JSUnusedGlobalSymbols
|
||||||
|
|
||||||
import {NS} from "NetscriptDefinitions";
|
import {NS} from "NetscriptDefinitions";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Recursively scans all servers connected to the current server, excluding home
|
||||||
|
* @param ns global NS object
|
||||||
|
* @returns A list of all servers connected to the current server
|
||||||
|
*/
|
||||||
export function recursiveScan(ns: NS) {
|
export function recursiveScan(ns: NS) {
|
||||||
// Starting case
|
// Starting case
|
||||||
let servers = ns.scan(ns.getHostname());
|
let servers = ns.scan("home");
|
||||||
// Add all servers to the list
|
// Add all servers to the list
|
||||||
let allServers: string[] = [];
|
let allServers: string[] = [];
|
||||||
while (servers.length > 0) {
|
while (servers.length > 0) {
|
||||||
@ -17,7 +25,87 @@ export function recursiveScan(ns: NS) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Remove the current server
|
// Remove the current server
|
||||||
allServers.splice(allServers.indexOf(ns.getHostname()), 1);
|
allServers.splice(allServers.indexOf("home"), 1);
|
||||||
// Print all servers
|
// Print all servers
|
||||||
return allServers;
|
return allServers;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Removes files passed in from all servers
|
||||||
|
* @param ns Global NS object
|
||||||
|
* @param files The files to remove
|
||||||
|
* @returns void
|
||||||
|
* @example Removes all files from the folder "no-ports" from all servers.
|
||||||
|
* removeFilesOnAllServers(ns, ns.ls("home", "no-ports"));
|
||||||
|
* @note ns.ls() returns the full path of every file, so if the file is not in the same place in every server,
|
||||||
|
* you will need to modify each file path to match the server you are removing it from.
|
||||||
|
*/
|
||||||
|
export function removeFilesOnAllServers(ns: NS, files: string[]) {
|
||||||
|
let hosts = recursiveScan(ns);
|
||||||
|
for(const host of hosts) {
|
||||||
|
for (const file of files) {
|
||||||
|
ns.rm(file, host)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Tries to gain root access to a server
|
||||||
|
* @param ns Global NS object
|
||||||
|
* @param server The server to gain root access to
|
||||||
|
* @returns The number of programs used to gain root access
|
||||||
|
*/
|
||||||
|
export function rootServer(ns: NS, server: string) {
|
||||||
|
let counter = 0;
|
||||||
|
if (ns.fileExists("BruteSSH.exe", "home")) {
|
||||||
|
ns.brutessh(server);
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
if (ns.fileExists("FTPCrack.exe", "home")) {
|
||||||
|
ns.ftpcrack(server);
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
if (ns.fileExists("SMTPCrack.exe", "home")) {
|
||||||
|
ns.relaysmtp(server);
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
if (ns.fileExists("HTTPWorm.exe", "home")) {
|
||||||
|
ns.httpworm(server);
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
if (ns.fileExists("SQLInject.exe", "home")) {
|
||||||
|
ns.sqlinject(server);
|
||||||
|
counter++;
|
||||||
|
}
|
||||||
|
ns.nuke(server)
|
||||||
|
return counter;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs a function on a server if the player is capable of doing so, otherwise returns false
|
||||||
|
* @param ns Global NS object
|
||||||
|
* @param server The server to perform the function on
|
||||||
|
* @param func The function to perform
|
||||||
|
* @param args The arguments to pass to the function
|
||||||
|
* @returns The result of the function if it is performed or true if the function does not return anything, otherwise false
|
||||||
|
*/
|
||||||
|
export function performFunctionIfCapable(ns: NS, server: string, func: CallableFunction, args: any[]) {
|
||||||
|
if (ns.getHackingLevel() < ns.getServerRequiredHackingLevel(server)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (ns.getServerNumPortsRequired(server) < ns.getServer(server).openPortCount) {
|
||||||
|
if (rootServer(ns, server) < ns.getServerNumPortsRequired(server)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!ns.hasRootAccess(server)) {
|
||||||
|
return false;
|
||||||
|
|
||||||
|
}
|
||||||
|
let result = func(...args);
|
||||||
|
if (result === undefined) {
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user