ensureRoot.js

import { list_servers } from 'opened_servers.js';
import { recursiveScan } from 'find_server.js';
import { boxTailSingleton, copyHackingFiles } from 'utils.js';

function maxHackLevel(hackingMultiples) {
    if (hackingMultiples < 2)
        return 150;
    else if (hackingMultiples < 3)
        return 250
    else if (hackingMultiples < 4)
        return 350
    else if (hackingMultiples < 5)
        return 450
    else if (hackingMultiples < 6)
        return 600;
    else if (hackingMultiples < 8)
        return 800;
    return 25000;
}


/** @param {NS} ns **/
export async function main(ns) {
    ns.disableLog('scan');
    ns.disableLog('getHackingLevel');
    ns.disableLog('installBackdoor');
    ns.disableLog('hasRootAccess');
    ns.disableLog('getServerNumPortsRequired');
    ns.disableLog('fileExists');
    ns.disableLog('sleep');
    ns.clearLog();
    // ns.tail();
    const player = JSON.parse(ns.read('/tmp/player.txt'));
    const bitNode = JSON.parse(ns.read(`/tmp/getBitNodeMultipliers.txt`));
    const mult = bitNode.HackingLevelMultiplier * player.hacking_mult + bitNode.HackExpGain * player.hacking_exp_mult;
    const maxHack = maxHackLevel(mult);
    boxTailSingleton(ns, 'ensureroot', '🗝', '200px');
    ns.print(`mul ${mult}, max hack ${maxHack}`);
    let hosts = list_servers(ns).filter(o => o.indexOf('pserv') === -1 && o !== 'darkweb');
    let oldhostLength = hosts.length;
    do {
        let servers = [];
        for (const host of hosts) {
            const fp = `/tmp/${host.replaceAll(/[\.\s]/g, '_')}.txt`;
            const data = ns.read(fp);
            let server;
            try {
                server = JSON.parse(data);
            } catch (e) {
                ns.print(`${fp} BAD JSON ${e.message} ${data}`);
                continue;
            }
            server.hasAdminRights = server.hasAdminRights || ensureRootAccess(ns, server);
            await copyHackingFiles(ns, server);
            if (server.hasAdminRights) {
                server.backdoorInstalled = server.backdoorInstalled || (await ensureBackdoor(ns, server));
            }
            //ns.print(`${server.hostname} ${server.hasAdminRights} && ${server.backdoorInstalled}`);
            await ns.write(fp, JSON.stringify(server, null, 2), "w");
            servers.push(server);
        }
        hosts = servers
            .filter((s) => !(s.hasAdminRights && s.backdoorInstalled) && s.requiredHackingSkill <= maxHack)
            .map(o => o.hostname);

        if (oldhostLength !== hosts.length)
            ns.print('Hosts to backdoor ' + JSON.stringify(hosts));
        oldhostLength = hosts.length;
        await ns.sleep(5000);
    } while (hosts.length > 0);
}

async function ensureBackdoor(ns, server) {
    if (server.requiredHackingSkill <= ns.getHackingLevel()) {
        if (!server.backdoorInstalled) {
            ns.connect('home');
            let route = [];
            recursiveScan(ns, '', 'home', server.hostname, route);
            for (const r of route) {
                if (!ns.connect(r)) {
                    ns.print(`Failed to connect to ${r}`);
                    return false;
                }
            }
            ns.print("backdooring " + server.hostname);
            await ns.installBackdoor();
            ns.print("backdoor installed " + server.hostname);
            ns.connect('home');
        }
        return true;
    }
    return false;
}

function ensureRootAccess(ns, server) {
    if (server.hostname === "home" || server.purchasedByPlayer || server.hasAdminRights) return true;
    let portsReq = server.numOpenPortsRequired;
    let maxPorts = countPorts(ns);
    if (portsReq <= maxPorts) {
        if (ns.fileExists("BruteSSH.exe", "home")) {
            ns.brutessh(server.hostname);
        }
        if (ns.fileExists("HTTPWorm.exe", "home")) {
            ns.httpworm(server.hostname);
        }
        if (ns.fileExists("FTPCrack.exe", "home")) {
            ns.ftpcrack(server.hostname);
        }
        if (ns.fileExists("relaySMTP.exe", "home")) {
            ns.relaysmtp(server.hostname);
        }
        if (ns.fileExists("sqlInject.exe", "home")) {
            ns.sqlinject(server.hostname);
        }
        ns.nuke(server.hostname);
        ns.print("nuked " + server.hostname);
        return true;
    } else {
        return false;
    }
}

function countPorts(ns) {
    let ports = 0
    if (ns.fileExists("BruteSSH.exe", "home")) {
        ports++;
    }
    if (ns.fileExists("HTTPWorm.exe", "home")) {
        ports++;
    }
    if (ns.fileExists("FTPCrack.exe", "home")) {
        ports++;
    }
    if (ns.fileExists("relaySMTP.exe", "home")) {
        ports++;
    }
    if (ns.fileExists("sqlInject.exe", "home")) {
        ports++;
    }
    return ports
}

export function autocomplete(data, args) {
    return data.servers;
}