Updated scripts to use ArgParser

This commit is contained in:
Zakary Timson 2022-03-09 19:06:14 +00:00
parent b44617a082
commit e67bef2f76
10 changed files with 488 additions and 673 deletions

122
README.md
View File

@ -2,7 +2,17 @@
These scripts are for playing the [open source](https://github.com/danielyxie/bitburner) game [BitBurner](https://danielyxie.github.io/bitburner/) These scripts are for playing the [open source](https://github.com/danielyxie/bitburner) game [BitBurner](https://danielyxie.github.io/bitburner/)
## Table of Contents ## Table of Contents
[[_TOC_]] - [BitBurner - Scripts](#bitburner-scripts)
- [Table of Contents](#table-of-contents)
- [Quick Start](#quick-start)
- [Scripts](#scripts)
- [bruteforce.js (WIP)](#bruteforcejs-wip)
- [crawler.js](#crawlerjs)
- [miner.js](#minerjs)
- [network-graph.js](#network-graphjs)
- [node-manager.js](#node-managerjs)
- [rootkit.js](#rootkitjs)
- [update.js](#updatejs)
## Quick Start ## Quick Start
```bash ```bash
@ -18,39 +28,17 @@ run scripts/crawler.js /scripts/auto-pwn.js {{TARGET}} /scripts/miner.js
``` ```
## Scripts ## Scripts
### [auto-pwn.js](./scripts/auto-pwn.js)
**RAM:** 4.75 GB
Automatically gain root on a target machine. Optionaly after being rooted, a file can be coppied & executed.
```
[home ~/]> run scripts/auto-pwn.js --help
Running script with 1 thread(s), pid 161 and args: ["--help"].
/scripts/auto-pwn.js:
Automatically gain root on a target machine. Optionaly after being rooted, a file can be coppied & executed.
Usage: run auto-pwn.js [TARGET] [SCRIPT] [ARGS]...
run auto-pwn.js --help
TARGET Target machine to root. Defaults to localhost
SCRIPT Script to copy & execute
ARGS Aditional arguments for SCRIPT. Forward the target with "{{TARGET}}"
Options:
-t --threads=num Set number of threads for script
-h --help Display help message
```
### [bruteforce.js](./scripts/bruteforce.js) (WIP) ### [bruteforce.js](./scripts/bruteforce.js) (WIP)
Attacks target until security falls bellow threshold. Useful for throwing extra compute power & cracking a specific computer. Attacks target until security falls bellow threshold. Useful for throwing extra compute power & cracking a specific computer.
### [crawler.js](./scripts/crawler.js) (WIP) ### [crawler.js](./scripts/crawler.js)
**RAM:** 3.05 GB **RAM:** 4.05 GB
Search the network for targets to execute a script against. Search the network for targets to execute a script against.
``` ```
[home ~/]> run scripts/crawler.js --help [home ~/]> run scripts/crawler.js --help
Running script with 1 thread(s), pid 163 and args: ["--help"]. Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/crawler.js: /scripts/crawler.js:
Search the network for targets to execute a script against. Search the network for targets to execute a script against.
@ -59,34 +47,34 @@ Usage: run crawler.js [OPTIONS] SCRIPT [ARGS]...
run crawler.js --help run crawler.js --help
SCRIPT Script to copy & execute SCRIPT Script to copy & execute
ARGS Aditional arguments for SCRIPT. Forward the target with "{{TARGET}}" ARGS Arguments for script. Forward the current target with: {{TARGET}}
Options: Options:
-d --depth=num Number of network hops. Defaults to 3 -c --cpu Number of CPU threads to use with script
-l --level=num Exclude targets with a high hacking level. Defaults to hack level, 0 to disable -d --depth Depth to scan to, defaults to 3
-p --ports=num Exclute targets with too many closed ports -l --level Exclude targets with higher hack level, defaults to current hack level
-t --threads=num Set number of threads for script -p --ports Exclute targets with too many closed ports
-h --help Display help message -h --help Display this help message
``` ```
### [miner.js](./scripts/miner.js) ### [miner.js](./scripts/miner.js)
**RAM:** 2.35 GB **RAM:** 2.45 GB
Weaken, spoof & hack the target in a loop for money. Weaken, Grow, Hack loop to "mine" target machine for money.
``` ```
[home ~/]> run scripts/miner.js --help [home ~/]> run scripts/miner.js --help
Running script with 1 thread(s), pid 165 and args: ["--help"]. Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/miner.js: /scripts/miner.js:
Weaken, spoof & hack the target in a loop for money. Weaken, Grow, Hack loop to "mine" target machine for money.
Usage: run miner.js [TARGET] Usage: run miner.js [TARGET]
run miner.js --help run miner.js --help
TARGET Target to mine. Defaults to localhost TARGET Device to mine, defaults to current machine
Options: Options:
-h --help Display help message -h --help Display this help message
``` ```
### [network-graph.js](./scripts/network-graph.js) ### [network-graph.js](./scripts/network-graph.js)
@ -95,7 +83,7 @@ Options:
Scan the network for devices and display as an ASCII tree. Scan the network for devices and display as an ASCII tree.
``` ```
[home ~/]> run /scripts/network-graph.js --help [home ~/]> run /scripts/network-graph.js --help
Running script with 1 thread(s), pid 138 and args: ["--help"]. Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/network-graph.js: /scripts/network-graph.js:
Scan the network for devices and display as an ASCII tree: Scan the network for devices and display as an ASCII tree:
@ -106,13 +94,15 @@ Scan the network for devices and display as an ASCII tree:
├─ foodnstuff (ROOTED) ├─ foodnstuff (ROOTED)
└─ sigma-cosmetics (ROOTED) └─ sigma-cosmetics (ROOTED)
Usage: run network-graph.js [OPTIONS] Usage: run network-graph.js [OPTIONS] [TARGET]
run network-graph.js --help run network-graph.js --help
TARGET Point to start scan from, defaults to current machine
Options: Options:
-d --depth Depth to scan to, defaults to 3 -d --depth Depth to scan to, defaults to 3
-f --filter Display path to single device -f --filter Display devices matching name
-s --start Point to start scan from, defaults to current machine -r --regex Display devices matching pattern
-v --verbose Displays the required hack level & ports needed to root: (level|port) -v --verbose Displays the required hack level & ports needed to root: (level|port)
-h --help Display this help message -h --help Display this help message
``` ```
@ -120,39 +110,65 @@ Options:
### [node-manager.js](./scripts/node-manager.js) ### [node-manager.js](./scripts/node-manager.js)
**RAM:** 5.70 GB **RAM:** 5.70 GB
Buy, upgrade & manage Hacknet nodes automatically. Buy, upgrade & manage Hacknet nodes automatically. Tail for live updates.
``` ```
[home ~/]> run scripts/node-manager.js --help [home ~/]> run scripts/node-manager.js --help
Running script with 1 thread(s), pid 166 and args: ["--help"]. Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/node-manager.js: /scripts/node-manager.js:
Buy, upgrade & manage Hacknet nodes automatically. Tail for live updates. Buy, upgrade & manage Hacknet nodes automatically. Tail for live updates.
Usage: run node-manager.js [OPTIONS] LIMIT Usage: run node-manager.js [OPTIONS] [LIMIT]
run node-manager.js --balance 1E6 4
run node-manager.js --help run node-manager.js --help
LIMIT Limit the number of nodes the manager will buy LIMIT Limit the number of nodes the manager will buy, defaults to 8
Options: Options:
-b --balance=num Prevent spending bellow this point -b --balance Prevent spending bellow point
-h --help Display help message -h --help Display this help message
```
### [rootkit.js](./scripts/rootkit.js)
**RAM:** 4.75 GB - 4.90 GB <small>(depending on un-commented programs)</small>
Automatically gain root on a target machine. A file can also be uploaded & executed.
Programs can be commented out to lower the cost of running.
```
[home ~/]> run scripts/rootkit.js --help
Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/rootkit.js:
Automatically gain root on a target machine. A file can also be uploaded & executed.
Usage: run rootkit.js [OPTIONS] [TARGET] [SCRIPT] [ARGS]...
run rootkit.js --help
TARGET Target machine to root, defaults to current machine
SCRIPT Script to copy & execute
ARGS Arguments for script. Forward the current target with: {{TARGET}}
Options:
-c --cpu Number of CPU threads to use with script
-h --help Display this help message
``` ```
### [update.js](./scripts/update.js) ### [update.js](./scripts/update.js)
**RAM:** 2.60 GB **RAM:** 2.95 GB
Automatically download the latest versions of all scripts using wget. Download the latest script updates from the repository using wget.
``` ```
[home ~/]> run scripts/update.js --help [home ~/]> run scripts/update.js --help
Running script with 1 thread(s), pid 167 and args: ["--help"]. Running script with 1 thread(s), pid 1 and args: ["--help"].
/scripts/update.js: /scripts/update.js:
Automatically download the latest versions of all scripts using wget. Download the latest script updates from the repository using wget.
Usage: run update.js Usage: run update.js
run update.js --help run update.js --help
TARGET Target device to update, defaults to current machine
Options: Options:
-h --help Display help message -h --help Display help message
``` ```

View File

@ -1,158 +0,0 @@
class ArgParser {
/**
* Create a unix-like argument parser to extract flags from the argument list. Can also create help messages.
* @param opts - {examples: string[], arguments: {key: string, alias: string, type: string, optional: boolean, desc: string}[], desc: string}
*/
constructor(opts) {
this.examples = opts.examples ?? [];
this.arguments = opts.args ?? [];
this.description = opts.desc;
}
/**
* Parse the list for arguments & create a dictionary.
* @param args {any[]} - Array of arguments
* @returns Dictionary of matched flags + unmatched args under 'extra'
*/
parse(args) {
const req = this.arguments.filter(a => !a.optional && !a.skip);
const queue = [...args], parsed = {}, extra = [];
for(let i = 0; i < queue.length; i++) {
if(queue[i][0] != '-') {
extra.push(queue[i]);
continue;
}
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1);
if(parse.indexOf('=')) {
const split = parse.split('=');
parse = split[0];
value = split[1];
}
let arg = this.arguments.find(a => a.key == parse) ?? this.arguments.find(a => a.alias == parse);
if(!arg) {
extra.push(queue[i]);
continue;
}
if(!value) {
value = arg.type == 'bool' ? true : queue[i + 1];
if(arg.type != 'bool') i++;
}
parsed[arg.key] = value;
}
req.forEach((a, i) => parsed[a.key] = extra[i]);
extra.splice(0, req.length);
return {...parsed, extra};
}
/**
* Create a help message of the expected paramters & usage.
* @param msg {String} - Optional message to display with help
*/
help(msg) {
let message = '\n\n';
message += msg ? msg : this.description;
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.join('\n\t');
const required = this.arguments.filter(a => !a.optional);
if(required.length) message += '\n\n\t' + required.map(a => {
const padding = 3 - ~~(a.key.length / 8);
return `${a.key}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
const optional = this.arguments.filter(a => a.optional);
if(optional.length) message += '\n\nOptions:\n\t' + optional.map(a => {
const flgs = `${a.alias ? `-${a.alias} ` : ''}--${a.key}${a.type && a.type != 'bool' ? `=${a.type}` : ''}`;
const padding = 3 - ~~(flgs.length / 8);
return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
return `${message}\n\n`;
}
}
/**
* Pwn a target server will availible tools. Can also copy & execute a script after pwning.
*/
export async function main(ns) {
ns.disableLog('ALL');
/**
* Prints text and waits a random amount of time to emulate work being complete.
* @param text {String} - message to Display
* @param min {Number} - minimum amount of time to wait in seconds. Defaults to 1 second.
* @param max {Number} - maximum amount of time to wait in seconds. Defaults to 1 second.
*/
async function printWithDelay(text, min=1, max=1) {
ns.tprint(text);
await ns.sleep(~~(Math.random() * (max * 1000 - min * 1000)) + min * 1000);
}
// Initilize script arguments
const argParser = new ArgParser({
desc: 'Automatically gain root on a target machine. Optionaly after being rooted, a file can be coppied & executed.',
examples: [
'run auto-pwn.js [TARGET] [SCRIPT] [ARGS]...',
'run auto-pwn.js --help',
],
args: [
{key: 'TARGET', desc: 'Target machine to root. Defaults to localhost'},
{key: 'SCRIPT', desc: 'Script to copy & execute'},
{key: 'ARGS', skip: true, desc: 'Aditional arguments for SCRIPT. Forward the target with "{{TARGET}}"'},
{key: 'threads', alias: 't', type: 'num', optional: true, desc: 'Set number of threads for script'},
{key: 'help', alias: 'h', type: 'bool', optional: true, desc: 'Display help message'},
]
});
const args = argParser.parse(ns.args);
if(args['help']) return ns.tprint(argParser.help());
// Setup
const target = args['TARGET'] && args['TARGET'] != 'localhost' ? args['TARGET'] : ns.getHostname();
const threads = args['threads'] || !args['SCRIPT'] ? 1 : ~~(ns.getServerMaxRam(target) / ns.getScriptRam(args['SCRIPT'], 'home'));
// Banner
ns.tprint('===================================================');
ns.tprint(`🧑‍💻 Pwning: ${target}`);
await printWithDelay('===================================================');
try {
// Open as many ports as possible
ns.brutessh(target);
await printWithDelay(`Attacking (SSH) ⚔️ ${target}:22`, 3, 5);
ns.ftpcrack(target);
await printWithDelay(`Attacking (FTP) ⚔️ ${target}:24`, 3, 5);
} catch {
} finally {
// Attempt to root
try {
ns.nuke(target)
await printWithDelay(`💀 Root Granted 💀`);
} catch {
await printWithDelay(`⚠️ Failed to Root ⚠️`);
ns.exit();
}
}
if(args['SCRIPT']) {
// Copy scripts
ns.tprint('');
await printWithDelay('Copying script:');
const speed = ~~(Math.random() * 100) / 10
await ns.scp(args['SCRIPT'], target);
await printWithDelay(`${args['SCRIPT']} \t [==================>] 100% \t (${speed} MB/s)`);
// Run scripts
ns.tprint('');
ns.tprint('Executing:');
ns.scriptKill(args['SCRIPT'], target);
await printWithDelay(`ssh -c "run ${args['SCRIPT']} ${args['extra'].join(' ')} -t ${threads}" root@${target}`);
const pid = ns.exec(args['SCRIPT'], target, threads, ...args['extra'].map(a => a == '{{TARGET}}' ? target : a));
ns.tprint('');
ns.tprint(pid != null ? '✅ Done!' : '⚠️ Failed to start ⚠️');
ns.tprint('');
}
}
/**
* Autocomplete script arguments
* @param data - provided by API
*/
export function autocomplete(data) {
return [...data.servers, ...data.scripts];
}

View File

@ -1,122 +1,72 @@
class ArgParser { import {ArgError, ArgParser} from './scripts/lib/arg-parser';
/**
* Create a unix-like argument parser to extract flags from the argument list. Can also create help messages.
* @param opts - {examples: string[], arguments: {key: string, alias: string, type: string, optional: boolean, desc: string}[], desc: string}
*/
constructor(opts) {
this.examples = opts.examples ?? [];
this.arguments = opts.args ?? [];
this.description = opts.desc;
}
/**
* Parse the list for arguments & create a dictionary.
* @param args {any[]} - Array of arguments
* @returns Dictionary of matched flags + unmatched args under 'extra'
*/
parse(args) {
const req = this.arguments.filter(a => !a.optional && !a.skip);
const queue = [...args], parsed = {}, extra = [];
for(let i = 0; i < queue.length; i++) {
if(queue[i][0] != '-') {
extra.push(queue[i]);
continue;
}
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1);
if(parse.indexOf('=')) {
const split = parse.split('=');
parse = split[0];
value = split[1];
}
let arg = this.arguments.find(a => a.key == parse) ?? this.arguments.find(a => a.alias == parse);
if(!arg) {
extra.push(queue[i]);
continue;
}
if(!value) {
value = arg.type == 'bool' ? true : queue[i + 1];
if(arg.type != 'bool') i++;
}
parsed[arg.key] = value;
}
req.forEach((a, i) => parsed[a.key] = extra[i]);
extra.splice(0, req.length);
return {...parsed, extra};
}
/**
* Create a help message of the expected paramters & usage.
* @param msg {String} - Optional message to display with help
*/
help(msg) {
let message = '\n\n';
message += msg ? msg : this.description;
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.join('\n\t');
const required = this.arguments.filter(a => !a.optional);
if(required.length) message += '\n\n\t' + required.map(a => {
const padding = 3 - ~~(a.key.length / 8);
return `${a.key}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
const optional = this.arguments.filter(a => a.optional);
if(optional.length) message += '\n\nOptions:\n\t' + optional.map(a => {
const flgs = `${a.alias ? `-${a.alias} ` : ''}--${a.key}${a.type && a.type != 'bool' ? `=${a.type}` : ''}`;
const padding = 3 - ~~(flgs.length / 8);
return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
return `${message}\n\n`;
}
}
/**
* Search the network for targets to execute a script against.
* @param ns {NS} - BitBurner API
*/
export async function main(ns) { export async function main(ns) {
ns.disableLog('ALL'); // Setup
ns.disableLog('ALL');
// Initilize script arguments const argParser = new ArgParser('crawler.js', 'Search the network for targets to execute a script against.', null, [
const argParser = new ArgParser({ {name: 'script', desc: 'Script to copy & execute', type: 'string'},
desc: 'Search the network for targets to execute a script against.', {name: 'args', desc: 'Arguments for script. Forward the current target with: {{TARGET}}', optional: true, extras: true, type: 'string'},
examples: [ {name: 'cpu', desc: 'Number of CPU threads to use with script', flags: ['-c', '--cpu'], default: 1, type: 'num'},
'run crawler.js [OPTIONS] SCRIPT [ARGS]...', {name: 'depth', desc: 'Depth to scan to, defaults to 3', flags: ['-d', '--depth'], default: Infinity, type: 'num'},
'run crawler.js --help', {name: 'level', desc: 'Exclude targets with higher hack level, defaults to current hack level', flags: ['-l', '--level'], default: ns.getHackingLevel(), type: 'num'},
], {name: 'ports', desc: 'Exclute targets with too many closed ports', flags: ['-p', '--ports'], optional: true, default: Infinity, type: 'num'},
args: [ ]);
{key: 'SCRIPT', desc: 'Script to copy & execute'}, let args;
{key: 'ARGS', skip: true, desc: 'Aditional arguments for SCRIPT. Forward the target with "{{TARGET}}"'}, try {
{key: 'depth', alias: 'd', type: 'num', optional: true, desc: 'Number of network hops. Defaults to 3'}, args = argParser.parse(ns.args);
{key: 'level', alias: 'l', type: 'num', optional: true, desc: 'Exclude targets with a high hacking level. Defaults to hack level, 0 to disable'}, } catch(err) {
{key: 'ports', alias: 'p', type: 'num', optional: true, desc: 'Exclute targets with too many closed ports'}, if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
{key: 'threads', alias: 't', type: 'num', optional: true, desc: 'Set number of threads for script'}, throw err;
{key: 'help', alias: 'h', type: 'bool', optional: true, desc: 'Display help message'},
]
});
const args = argParser.parse(ns.args);
if(args['help']) return ns.tprint(argParser.help());
if(!args['SCRIPT']) return ns.tprint(argParser.help('Missing SCRIPT'));
// Setup
const depth = args['depth'] || 3;
const level = args['level'] || ns.getHackingLevel();
const ports = args['ports'] || Infinity;
const threads = args['threads'] || 1;
// Recursively search for targets
const targets = ns.scan().map(h => [h, 1]);
for(let i = 0; i < targets.length; i++) {
if(targets[i][1] < depth) ns.scan(targets[i][0]).forEach(h => {
if(h != 'home' && !targets.find(t => t[0] == h)) targets.push([h, targets[i][1] + 1]);
});
} }
// Execute script on each target /**
for(const target of targets) { * Recursively search network & build a tree
// Check target * @param host {string} - Point to scan from
if(level && level < ns.getServerRequiredHackingLevel(target[0]) || * @param depth {number} - Current scanning depth
(ports && ports < ns.getServerNumPortsRequired(target[0]))) return; * @param blacklist {String[]} - Devices already discovered
* @returns Dicionary of discovered devices
*/
function scan(target = 'home', depth = 1, found = new Set()) {
if(found.size == 0) found.add(target);
ns.scan(target).filter(t => !found.has(t)).forEach(t => {
found.add(t);
scan(t, depth + 1, found);
});
found.delete('home');
return Array.from(found);
}
// Start script // Run
ns.run(args['SCRIPT'], threads, ...args['extra'].map(a => a == '{{TARGET}}' ? target[0] : a)); const targets = scan();
let complete = 0, failed = 0, skipped = 0;
for(let target of targets) {
if(target == 'home') continue;
if(args['level'] < ns.getServerRequiredHackingLevel(target) || args['ports'] < ns.getServerNumPortsRequired(target)) {
skipped++;
continue;
}
const scriptArgs = args['args'].map(arg => arg == '{{TARGET}}' ? target : arg);
const pid = ns.run(args['script'], args['cpu'], ...scriptArgs);
if(pid == 0) {
failed++;
continue;
}
// Wait for script to finish // Wait for script to finish
// do { await ns.sleep(1000); } do { await ns.sleep(1000); }
// while(ns.scriptRunning(args['SCRIPT'], 'home')); while(ns.scriptRunning(args['script'], 'home'));
await ns.sleep(10000); complete++;
} }
// Output report
ns.tprint('===================================================');
ns.tprint(`Crawler Report: ${targets.length} Device${targets.length > 1 ? 's' : ''}`);
ns.tprint('===================================================');
ns.tprint(`Complete: ${complete}\tFailed: ${failed}\tSkipped: ${skipped}`);
ns.tprint('');
} }

View File

@ -11,15 +11,10 @@ export class ArgParser {
constructor(name, desc, examples, argList) { constructor(name, desc, examples, argList) {
this.name = name ?? 'example.js'; this.name = name ?? 'example.js';
this.description = desc ?? 'Example description'; this.description = desc ?? 'Example description';
this.examples = [ this.examples = examples || [`${argList.find(arg => !!arg.flags) ? '[OPTIONS] ' : ''}${argList.filter(arg => !arg.flags).map(arg => (arg.optional ? `[${arg.name.toUpperCase()}]` : arg.name.toUpperCase()) + (arg.extras ? '...' : '')).join(' ')}`];
...examples, this.examples.push('--help');
`[OPTIONS] ${argList.filter(arg => !arg.flags).map(arg => arg.name.toUpperCase())}`, this.argList = argList || [];
'--help' this.argList.push({name: 'help', desc: 'Display this help message', flags: ['-h', '--help'], type: 'bool'});
];
this.argList = [
...argList,
{name: 'help', desc: 'Display this help message', flags: ['-h', '--help'], type: 'bool'}
];
} }
/** /**
@ -41,9 +36,11 @@ export class ArgParser {
queue = parse.substring(1).split('').map(a => `-${a}`).concat(queue); queue = parse.substring(1).split('').map(a => `-${a}`).concat(queue);
} }
// Find & add flag // Find & add flag
const arg = this.argList.find(arg => arg.flags && arg.flags.includes(parse)); const split = parse.split('=');
if(arg == null) throw new ArgError(`Unknown option: ${parse}`); const arg = this.argList.find(arg => arg.flags && arg.flags.includes(split[0] || parse));
const value = arg.type == 'bool' ? true : parse.split('=')[1] || queue.splice(queue.findIndex(q => q[0] != '-'), 1)[0]; if(arg == null) throw new ArgError(`Option unknown: ${parse}`);
if(arg.name == 'help') throw new ArgError('Help');
const value = arg.type == 'bool' ? true : split[1] || queue.splice(queue.findIndex(q => q[0] != '-'), 1)[0];
if(value == null) throw new ArgError(`Option missing value: ${arg.name}`); if(value == null) throw new ArgError(`Option missing value: ${arg.name}`);
parsed[arg.name] = value; parsed[arg.name] = value;
} else { } else {
@ -52,13 +49,14 @@ export class ArgParser {
} }
} }
// Arguments // Arguments
this.argList.filter(arg => !arg.flags).forEach(arg => { this.argList.filter(arg => !arg.flags && !arg.extras).forEach(arg => {
if(!extra.length) throw new ArgError(`Argument missing: ${arg.name}`); if(!arg.optional && !extra.length) throw new ArgError(`Argument missing: ${arg.name.toUpperCase()}`);
parsed[arg.name] = extra.splice(0, 1)[0]; const value = extra.splice(0, 1)[0];
if(value != null) parsed[arg.name] = value;
}); });
// Extras // Extras
if(extra.length) parsed['extra'] = extra; const extraKey = this.argList.find(arg => arg.extras)?.name || 'extra';
if(parsed['help']) throw new ArgError(); parsed[extraKey] = extra;
return parsed; return parsed;
} }
@ -69,7 +67,7 @@ export class ArgParser {
*/ */
help(msg) { help(msg) {
// Description // Description
let message = '\n\n' + (msg ? msg : this.description); let message = '\n\n' + (msg && msg.toLowerCase() != 'help' ? msg : this.description);
// Usage // Usage
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.map(ex => `run ${this.name} ${ex}`).join('\n\t'); if(this.examples.length) message += '\n\nUsage:\t' + this.examples.map(ex => `run ${this.name} ${ex}`).join('\n\t');
// Arguments // Arguments

23
scripts/lib/utils.js Normal file
View File

@ -0,0 +1,23 @@
/**
* Print a download bar to the terminal.
* @params ns {NS} - Bitburner API
* @params file - Filename to display with progress bar
*/
export async function downloadPrint(ns, file) {
const speed = ~~(Math.random() * 100) / 10;
const spacing = Array(5 - Math.floor((file.length) / 8)).fill('\t').join('');
await slowPrint(ns, `${file}${spacing}[==================>] 100% \t (${speed} MB/s)`);
}
/**
* Print text to the terminal & then delay for a random amount of time to emulate execution time.
* @params ns {NS} - Bitburner API
* @params message {string} - Text to display
* @params min {number} - minimum amount of time to wait after printing text
* @params max {number} - maximum amount of time to wait after printing text
*/
export async function slowPrint(ns, message, min = 0.5, max = 1.5) {
const time = ~~(Math.random() * (max * 1000 - min * 1000)) + min * 1000;
ns.tprint(message);
await ns.sleep(time);
}

View File

@ -1,77 +1,29 @@
class ArgParser { import {ArgError, ArgParser} from './scripts/lib/arg-parser';
/**
* Create a unix-like argument parser to extract flags from the argument list. Can also create help messages.
* @param opts - {examples: string[], arguments: {key: string, alias: string, type: string, optional: boolean, desc: string}[], desc: string}
*/
constructor(opts) {
this.examples = opts.examples ?? [];
this.arguments = opts.args ?? [];
this.description = opts.desc;
}
/**
* Parse the list for arguments & create a dictionary.
* @param args {any[]} - Array of arguments
* @returns Dictionary of matched flags + unmatched args under 'extra'
*/
parse(args) {
const req = this.arguments.filter(a => !a.optional && !a.skip);
const queue = [...args], parsed = {}, extra = [];
for(let i = 0; i < queue.length; i++) {
if(queue[i][0] != '-') {
extra.push(queue[i]);
continue;
}
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1);
if(parse.indexOf('=')) {
const split = parse.split('=');
parse = split[0];
value = split[1];
}
let arg = this.arguments.find(a => a.key == parse) ?? this.arguments.find(a => a.alias == parse);
if(!arg) {
extra.push(queue[i]);
continue;
}
if(!value) {
value = arg.type == 'bool' ? true : queue[i + 1];
if(arg.type != 'bool') i++;
}
parsed[arg.key] = value;
}
req.forEach((a, i) => parsed[a.key] = extra[i]);
extra.splice(0, req.length);
return {...parsed, extra};
}
/**
* Create a help message of the expected paramters & usage.
* @param msg {String} - Optional message to display with help
*/
help(msg) {
let message = '\n\n';
message += msg ? msg : this.description;
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.join('\n\t');
const required = this.arguments.filter(a => !a.optional);
if(required.length) message += '\n\n\t' + required.map(a => {
const padding = 3 - ~~(a.key.length / 8);
return `${a.key}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
const optional = this.arguments.filter(a => a.optional);
if(optional.length) message += '\n\nOptions:\n\t' + optional.map(a => {
const flgs = `${a.alias ? `-${a.alias} ` : ''}--${a.key}${a.type && a.type != 'bool' ? `=${a.type}` : ''}`;
const padding = 3 - ~~(flgs.length / 8);
return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
return `${message}\n\n`;
}
}
/** /**
* Hack a server for it's money. * Weaken, Grow, Hack loop to "mine" target machine for money.
* @params ns {NS} - BitBurner API
*/ */
export async function main(ns) { export async function main(ns) {
// Setup
ns.disableLog('ALL'); ns.disableLog('ALL');
const historyLength = 15;
const messageHistory = Array(historyLength).fill('');
let maxBalance, balance, minSecurity, security;
const argParser = new ArgParser('miner.js', 'Weaken, Grow, Hack loop to "mine" target machine for money.', null, [
{name: 'target', desc: 'Device to mine, defaults to current machine', optional: true, default: ns.getHostname(), type: 'string'}
]);
let args;
try {
args = argParser.parse(ns.args);
maxBalance = await ns.getServerMaxMoney(args['target']);
balance = await ns.getServerMoneyAvailable(args['target']);
minSecurity = await ns.getServerMinSecurityLevel(args['target']) + 2;
security = await ns.getServerSecurityLevel(args['target']);
} catch(err) {
if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
throw err;
}
/** /**
* Print header with logs * Print header with logs
@ -81,7 +33,7 @@ export async function main(ns) {
const sec = `${Math.round(security)}/${minSecurity}`; const sec = `${Math.round(security)}/${minSecurity}`;
ns.clearLog(); ns.clearLog();
ns.print('==================================================='); ns.print('===================================================');
ns.print(`💎⛏️ Mining: ${target}`); ns.print(`Mining: ${args['target']}`);
ns.print('==================================================='); ns.print('===================================================');
ns.print(`Security: ${sec}${sec.length < 6 ? '\t' : ''}\tBalance: $${Math.round(balance * 100) / 100}`); ns.print(`Security: ${sec}${sec.length < 6 ? '\t' : ''}\tBalance: $${Math.round(balance * 100) / 100}`);
ns.print('==================================================='); ns.print('===================================================');
@ -90,50 +42,27 @@ export async function main(ns) {
messageHistory.forEach(m => ns.print(m)); messageHistory.forEach(m => ns.print(m));
} }
// Initilize script arguments
const argParser = new ArgParser({
desc: 'Weaken, spoof & hack the target in a loop for money.',
examples: [
'run miner.js [TARGET]',
'run miner.js --help',
],
args: [
{key: 'TARGET', desc: 'Target to mine. Defaults to localhost'},
{key: 'help', alias: 'h', type: 'bool', optional: true, desc: 'Display help message'},
]
});
const args = argParser.parse(ns.args);
if(args['help']) return ns.tprint(argParser.help());
// Setup
const historyLength = 15;
const messageHistory = Array(historyLength).fill('');
const target = args['TARGET'] && args['TARGET'] != 'localhost' ? args['TARGET'] : ns.getHostname();
const minSecurity = ns.getServerMinSecurityLevel(target) + 2;
let orgBalance, balance, security;
log(); log();
while(true) { do {
// Update information // Update information
security = await ns.getServerSecurityLevel(target); security = await ns.getServerSecurityLevel(args['target']);
balance = await ns.getServerMoneyAvailable(target); balance = await ns.getServerMoneyAvailable(args['target']);
if(orgBalance == null) orgBalance = balance;
// Pick step // Pick step
if(security > minSecurity) { // Weaken if(security > minSecurity) { // Weaken
log('Attacking Security...'); log('Attacking Security...');
const w = await ns.weaken(target); const w = await ns.weaken(args['target']);
log(`Security: -${w}`); log(`Security: -${w}`);
} else if(balance <= orgBalance) { // Grow } else if(balance < maxBalance) { // Grow
log('Spoofing Balance...'); log('Spoofing Balance...');
const g = await ns.grow(target); const g = await ns.grow(args['target']);
log(`Balance: +$${Math.round((g * balance - balance) * 100) / 100}`); log(`Balance: +$${Math.round((g * balance - balance) * 100) / 100}`);
} else { // Hack } else { // Hack
log('Hacking Account...'); log('Hacking Account...');
const h = await ns.hack(target); const h = await ns.hack(args['target']);
log(`Balance: -$${h}`); log(`Balance: -$${h}`);
} }
} } while(true);
} }
export function autocomplete(data) { export function autocomplete(data) {

View File

@ -3,10 +3,11 @@ import {ArgError, ArgParser} from './scripts/lib/arg-parser';
export async function main(ns) { export async function main(ns) {
// Setup // Setup
ns.disableLog('ALL'); ns.disableLog('ALL');
const argParser = new ArgParser('network-graph.js', 'Scan the network for devices and display as an ASCII tree:\n home\n ├─ n00dles (ROOTED)\n | └─ max-hardware (80|1)\n | └─ neo-net (50|1)\n ├─ foodnstuff (ROOTED)\n └─ sigma-cosmetics (ROOTED)', [], [ const argParser = new ArgParser('network-graph.js', 'Scan the network for devices and display as an ASCII tree:\n home\n ├─ n00dles (ROOTED)\n | └─ max-hardware (80|1)\n | └─ neo-net (50|1)\n ├─ foodnstuff (ROOTED)\n └─ sigma-cosmetics (ROOTED)', null, [
{name: 'target', desc: 'Point to start scan from, defaults to current machine', optional: true, default: ns.getHostname(), type: 'string'},
{name: 'depth', desc: 'Depth to scan to, defaults to 3', flags: ['-d', '--depth'], default: Infinity, type: 'num'}, {name: 'depth', desc: 'Depth to scan to, defaults to 3', flags: ['-d', '--depth'], default: Infinity, type: 'num'},
{name: 'filter', desc: 'Display path to single device', flags: ['-f', '--filter'], type: 'string'}, {name: 'filter', desc: 'Display devices matching name', flags: ['-f', '--filter'], type: 'string'},
{name: 'start', desc: 'Point to start scan from, defaults to current machine', flags: ['-s', '--start'], default: ns.getHostname(), type: 'string'}, {name: 'regex', desc: 'Display devices matching pattern', flags: ['-r', '--regex'], type: 'string'},
{name: 'verbose', desc: 'Displays the required hack level & ports needed to root: (level|port)', flags: ['-v', '--verbose'], type: 'bool'}, {name: 'verbose', desc: 'Displays the required hack level & ports needed to root: (level|port)', flags: ['-v', '--verbose'], type: 'bool'},
]); ]);
let args; let args;
@ -17,6 +18,38 @@ export async function main(ns) {
throw err; throw err;
} }
/**
* Prune tree down to devices that match name or pattern.
* @param tree {object} - Tree to search
* @param find {string} - Device name or pattern to search for
* @param regex {boolean} - True to use regex, false for raw check
* @returns {object} - Pruned tree
*/
function filter(tree, find, regex = false) {
const found = new Set();
function buildWhitelist(tree, find, path = []) {
const keys = Object.keys(tree);
if(!keys.length) return;
Object.keys(tree).forEach(n => {
const matches = regex ? new RegExp(find).test(n) : n == find;
if(n == 'n00dles') console.log(n, find, matches);
if(matches) {
found.add(n);
path.forEach(p => found.add(p));
}
buildWhitelist(tree[n], find, [...path, n]);
})
}
function prune(tree, whitelist) {
Object.keys(tree).forEach(n => {
if(Object.keys(tree[n]).length) prune(tree[n], whitelist);
if(!whitelist.includes(n)) delete tree[n];
});
}
buildWhitelist(tree, find);
prune(tree, Array.from(found));
}
/** /**
* Recursively search network & build a tree * Recursively search network & build a tree
* @param host {string} - Point to scan from * @param host {string} - Point to scan from
@ -25,9 +58,9 @@ export async function main(ns) {
* @returns Dicionary of discovered devices * @returns Dicionary of discovered devices
*/ */
function scan(host, depth = 1, blacklist = [host]) { function scan(host, depth = 1, blacklist = [host]) {
if(depth >= args['depth']) return {}; if(depth > args['depth']) return {};
const localTargets = ns.scan(host).filter(target => !blacklist.includes(target)); const localTargets = ns.scan(host).filter(target => !blacklist.includes(target));
blacklist = blacklist.concat(localTargets); blacklist = [...blacklist, ...localTargets];
return localTargets.reduce((acc, target) => { return localTargets.reduce((acc, target) => {
const info = ns.getServer(target); const info = ns.getServer(target);
const verb = args['verbose'] ? ` (${info.hasAdminRights ? 'ROOTED' : `${info.requiredHackingSkill}|${info.numOpenPortsRequired}`})` : ''; const verb = args['verbose'] ? ` (${info.hasAdminRights ? 'ROOTED' : `${info.requiredHackingSkill}|${info.numOpenPortsRequired}`})` : '';
@ -37,29 +70,6 @@ export async function main(ns) {
}, {}); }, {});
} }
/**
* Search tree for path to device.
* @param tree {object} - Tree to search
* @param find {string} - Device to search for
* @returns {object} - Path to device
*/
function filter(tree, find) {
function filter(tree, find, path = []) {
return Object.keys(tree).flatMap(n => {
if(n.indexOf(find) == 0) return [...path, n];
if(Object.keys(n).length) return filter(tree[n], find, [...path, n]);
return null;
}).filter(p => !!p);
}
let whitelist = filter(tree, find), acc = {}, next = acc;
while(whitelist.length) {
const n = whitelist.splice(0, 1);
next[n] = {};
next = next[n];
}
return acc;
}
/** /**
* Iterate tree & print to screen * Iterate tree & print to screen
* @param tree {object} - Tree to parse * @param tree {object} - Tree to parse
@ -75,13 +85,14 @@ export async function main(ns) {
} }
// Run // Run
let found = scan(args['start'], args['verbose']); ns.tprint(args['target']);
if(args['filter']) found = filter(found, args['filter']); const found = scan(args['target']);
ns.tprint(args['start']); if(args['regex']) filter(found, args['regex'], true);
else if(args['filter']) filter(found, args['filter']);
render(found); render(found);
ns.tprint(''); ns.tprint('');
} }
export function autocomplete(data) { export function autocomplete(data) {
return [...data.servers]; return [...data.servers];
} }

View File

@ -1,78 +1,26 @@
class ArgParser { import {ArgError, ArgParser} from './scripts/lib/arg-parser';
/**
* Create a unix-like argument parser to extract flags from the argument list. Can also create help messages.
* @param opts - {examples: string[], arguments: {key: string, alias: string, type: string, optional: boolean, desc: string}[], desc: string}
*/
constructor(opts) {
this.examples = opts.examples ?? [];
this.arguments = opts.args ?? [];
this.description = opts.desc;
}
/**
* Parse the list for arguments & create a dictionary.
* @param args {any[]} - Array of arguments
* @returns Dictionary of matched flags + unmatched args under 'extra'
*/
parse(args) {
const req = this.arguments.filter(a => !a.optional && !a.skip);
const queue = [...args], parsed = {}, extra = [];
for(let i = 0; i < queue.length; i++) {
if(queue[i][0] != '-') {
extra.push(queue[i]);
continue;
}
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1);
if(parse.indexOf('=')) {
const split = parse.split('=');
parse = split[0];
value = split[1];
}
let arg = this.arguments.find(a => a.key == parse) ?? this.arguments.find(a => a.alias == parse);
if(!arg) {
extra.push(queue[i]);
continue;
}
if(!value) {
value = arg.type == 'bool' ? true : queue[i + 1];
if(arg.type != 'bool') i++;
}
parsed[arg.key] = value;
}
req.forEach((a, i) => parsed[a.key] = extra[i]);
extra.splice(0, req.length);
return {...parsed, extra};
}
/**
* Create a help message of the expected paramters & usage.
* @param msg {String} - Optional message to display with help
*/
help(msg) {
let message = '\n\n';
message += msg ? msg : this.description;
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.join('\n\t');
const required = this.arguments.filter(a => !a.optional);
if(required.length) message += '\n\n\t' + required.map(a => {
const padding = 3 - ~~(a.key.length / 8);
return `${a.key}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
const optional = this.arguments.filter(a => a.optional);
if(optional.length) message += '\n\nOptions:\n\t' + optional.map(a => {
const flgs = `${a.alias ? `-${a.alias} ` : ''}--${a.key}${a.type && a.type != 'bool' ? `=${a.type}` : ''}`;
const padding = 3 - ~~(flgs.length / 8);
return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t');
return `${message}\n\n`;
}
}
/** /**
* Manages hacknet nodes, purchasing nodes to reach the desired amount. * Buy, upgrade & manage Hacknet nodes automatically.
* Upgrades (Level, RAM, Cores & Cache) will be automatically purchased. * @params ns {NS} - BitBurner API
*/ */
export async function main(ns) { export async function main(ns) {
// Setup
ns.disableLog('ALL'); ns.disableLog('ALL');
const historyLength = 17;
const messageHistory = Array(historyLength).fill('');
let args, nodeCount = ns.hacknet.numNodes();
const argParser = new ArgParser('node-manager.js', 'Buy, upgrade & manage Hacknet nodes automatically. Tail for live updates.', null, [
{name: 'limit', desc: 'Limit the number of nodes the manager will buy, defaults to 8', optional: true, default: 8, type: 'num'},
{name: 'balance', desc: 'Prevent spending bellow point', flags: ['-b', '--balance'], type: 'num'}
]);
try {
args = argParser.parse(ns.args);
} catch(err) {
if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
throw err;
}
/** /**
* Print header with logs * Print header with logs
@ -81,48 +29,20 @@ export async function main(ns) {
function log(message) { function log(message) {
ns.clearLog(); ns.clearLog();
ns.print('==================================================='); ns.print('===================================================');
ns.print(`🖥️ Node Manager: ${nodeCount}/${limit} Nodes`); ns.print(`Node Manager: ${nodeCount}/${args['limit']} Nodes`);
ns.print('==================================================='); ns.print('===================================================');
if(message != null) messageHistory.push(message); if(message != null) messageHistory.push(message);
messageHistory.splice(0, messageHistory.length - historyLength); messageHistory.splice(0, messageHistory.length - historyLength);
messageHistory.forEach(m => ns.print(m)); messageHistory.forEach(m => ns.print(m));
} }
// Initilize script arguments // Run
const argParser = new ArgParser({
desc: 'Buy, upgrade & manage Hacknet nodes automatically. Tail for live updates.',
examples: [
'run node-manager.js [OPTIONS] LIMIT',
'run node-manager.js --balance 1E6 4',
'run node-manager.js --help',
],
args: [
{key: 'LIMIT', desc: 'Limit the number of nodes the manager will buy'},
{key: 'balance', alias: 'b', type: 'num', optional: true, desc: 'Prevent spending bellow this point'},
{key: 'help', alias: 'h', type: 'bool', optional: true, desc: 'Display help message'},
]
});
const args = argParser.parse(ns.args);
// Check arguments
if(args['help']) return ns.tprint(argParser.help());
if(!args['LIMIT']) return ns.tprint(argParser.help('Missing LIMIT'));
if(isNaN(args['LIMIT'])) return ns.tprint(argParser.help('LIMIT must be a number'));
if(!!args['balance'] && isNaN(args['balance'])) return ns.tprint(argParser.help('LIMIT must be a number'));
// Setup
const historyLength = 17;
const messageHistory = Array(historyLength).fill('');
const limit = args['LIMIT'];
const savings = args['balance'] ?? 0
let nodeCount = ns.hacknet.numNodes();
log(); log();
while(true) { while(true) {
const balance = ns.getServerMoneyAvailable('home'); const balance = ns.getServerMoneyAvailable('home');
// Check if we should buy a new node // Check if we should buy a new node
if(nodeCount < limit && balance - ns.hacknet.getPurchaseNodeCost() >= savings) { if(nodeCount < args['limit'] && balance - ns.hacknet.getPurchaseNodeCost() >= args['balance']) {
nodeCount++; nodeCount++;
ns.hacknet.purchaseNode(); ns.hacknet.purchaseNode();
log(`Buying Node ${nodeCount}`); log(`Buying Node ${nodeCount}`);
@ -170,7 +90,7 @@ export async function main(ns) {
}); });
// Apply the cheapest upgrade // Apply the cheapest upgrade
if(nodes.length && balance - nodes[0].bestUpgrade.cost >= savings) { if(nodes.length && balance - nodes[0].bestUpgrade.cost >= args['balance']) {
const cost = Math.round(nodes[0].bestUpgrade.cost * 100) / 100; const cost = Math.round(nodes[0].bestUpgrade.cost * 100) / 100;
log(`Node ${nodes[0].index} - ${nodes[0].bestUpgrade.name} ${nodes[0][nodes[0].bestUpgrade.name] + 1} - $${cost}`); log(`Node ${nodes[0].index} - ${nodes[0].bestUpgrade.name} ${nodes[0][nodes[0].bestUpgrade.name] + 1} - $${cost}`);
nodes[0].bestUpgrade.purchase(); nodes[0].bestUpgrade.purchase();

97
scripts/rootkit.js Normal file
View File

@ -0,0 +1,97 @@
import {ArgError, ArgParser} from './scripts/lib/arg-parser';
import {downloadPrint, slowPrint} from './scripts/lib/utils';
/**
* Pwn a target server with availible tools. Additionally can copy & execute a script after pwning.
* @param ns {NS} - Bitburner API
*/
export async function main(ns) {
// Setup
ns.disableLog('ALL');
const argParser = new ArgParser('rootkit.js', 'Automatically gain root on a target machine. A file can also be uploaded & executed.', null, [
{name: 'target', desc: 'Target machine to root, defaults to current machine', optional: true, default: ns.getHostname(), type: 'string'},
{name: 'script', desc: 'Script to copy & execute', optional: true, type: 'string'},
{name: 'args', desc: 'Arguments for script. Forward the current target with: {{TARGET}}', optional: true, extras: true, type: 'string'},
{name: 'cpu', desc: 'Number of CPU threads to use with script', flags: ['-c', '--cpu'], type: 'num'},
]);
let args;
try {
args = argParser.parse(ns.args);
if(args['script'] && !args['cpu']) args['cpu'] = ~~(ns.getServerMaxRam(args['target']) / ns.getScriptRam(args['script'], 'home')) || 1;
} catch(err) {
if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
throw err;
}
/**
* Detect import statements inside script & build a dependency tree.
* @params file {string} - Path to file to search
* @returns {string[]} - Array of required files
*/
async function dependencyFinder(file) {
const queue = [file], found = [];
while(queue.length) {
const imports = new RegExp(/from ["']\.(.+)["']/g);
const script = await ns.read(queue.splice(0, 1)[0]);
let match;
while((match = imports.exec(script)) != null) {
const path = `${match[1]}.js`;
queue.push(path);
found.push(path);
}
}
return found;
}
// Banner
ns.tprint('===================================================');
ns.tprint(`Rooting: ${args['target']}`);
await slowPrint(ns, '===================================================');
try {
// Run exploits
await slowPrint(ns, `Attacking over SSH (${args['target']}:22)...`, 1, 2);
ns.brutessh(args['target']);
await slowPrint(ns, `Attacking over FTP (${args['target']}:24)...`, 1, 2);
ns.ftpcrack(args['target']);
await slowPrint(ns, `Attacking over SMTP (${args['target']}:25)...`, 1, 2);
ns.relaysmtp(args['target']);
} catch {
} finally {
try {
// Attempt root
ns.tprint('');
ns.nuke(args['target'])
ns.tprint(`Root: Success!`);
ns.tprint('');
} catch {
ns.tprint(`Root: Failed`);
ns.tprint('');
ns.exit();
}
}
if(args['script']) {
// Detect script dependencies & copy everything to target
await ns.sleep(0.5);
await slowPrint(ns, 'Copying files:');
const deps = [...(await dependencyFinder(args['script'])), args['script']];
for(let dep of deps) {
await ns.scp(dep, args['target']);
await downloadPrint(ns, dep);
}
// Run script
ns.tprint('');
await slowPrint(ns, `Executing with ${args['cpu']} thread${args['cpu'] > 1 ? 's' : ''}...`);
ns.scriptKill(args['script'], args['target']);
const pid = ns.exec(args['script'], args['target'], args['cpu'], ...args['args']
.map(a => a == '{{TARGET}}' ? args['target'] : a));
ns.tprint(!!pid ? 'Done!' : 'Failed to start');
ns.tprint('');
}
}
export function autocomplete(data) {
return [...data.servers, ...data.scripts];
}

View File

@ -1,135 +1,164 @@
class ArgParser { export class ArgError extends Error {}
export class ArgParser {
/** /**
* Create a unix-like argument parser to extract flags from the argument list. Can also create help messages. * Create a unix-like argument parser to extract flags from the argument list. Can also create help messages.
* @param opts - {examples: string[], arguments: {key: string, alias: string, type: string, optional: boolean, desc: string}[], desc: string} * @param name {string} - Script name
* @param desc {string} - Help text desciption
* @param examples {string[]} - Help text examples
* @param argList {name: string, desc: string, flags: string[], type: string, default: any}[] - Array of CLI arguments
*/ */
constructor(opts) { constructor(name, desc, examples, argList) {
this.examples = opts.examples ?? []; this.name = name ?? 'example.js';
this.arguments = opts.args ?? []; this.description = desc ?? 'Example description';
this.description = opts.desc; this.examples = examples || [`${argList.find(arg => !!arg.flags) ? '[OPTIONS] ' : ''}${argList.filter(arg => !arg.flags).map(arg => (arg.optional ? `[${arg.name.toUpperCase()}]` : arg.name.toUpperCase()) + (arg.extras ? '...' : '')).join(' ')}`];
this.examples.push('--help');
this.argList = argList || [];
this.argList.push({name: 'help', desc: 'Display this help message', flags: ['-h', '--help'], type: 'bool'});
} }
/** /**
* Parse the list for arguments & create a dictionary. * Parse an array into an arguments dictionary using the configuration.
* @param args {any[]} - Array of arguments * @param args {string[]} - Array of arguments to be parsed
* @returns Dictionary of matched flags + unmatched args under 'extra' * @returns {object} - Dictionary of arguments with defaults applied
*/ */
parse(args) { parse(args) {
const req = this.arguments.filter(a => !a.optional && !a.skip); // Parse arguments
const queue = [...args], parsed = {}, extra = []; const queue = [...args], extra = [];
for(let i = 0; i < queue.length; i++) { const parsed = this.argList.reduce((acc, arg) => ({...acc, [arg.name]: arg.default ?? (arg.type == 'bool' ? false : null)}), {});
if(queue[i][0] != '-') { // Flags
extra.push(queue[i]); while(queue.length) {
continue; let parse = queue.splice(0, 1)[0];
} if(parse[0] == '-') {
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1); // Check combined flags
if(parse.indexOf('=')) { if(parse[1] != '-' && parse.length > 2) {
parse = `-${parse[1]}`;
queue = parse.substring(1).split('').map(a => `-${a}`).concat(queue);
}
// Find & add flag
const split = parse.split('='); const split = parse.split('=');
parse = split[0]; const arg = this.argList.find(arg => arg.flags && arg.flags.includes(split[0] || parse));
value = split[1]; if(arg == null) throw new ArgError(`Option unknown: ${parse}`);
if(arg.name == 'help') throw new ArgError('Help');
const value = arg.type == 'bool' ? true : split[1] || queue.splice(queue.findIndex(q => q[0] != '-'), 1)[0];
if(value == null) throw new ArgError(`Option missing value: ${arg.name}`);
parsed[arg.name] = value;
} else {
// Save for required parsing
extra.push(parse);
} }
let arg = this.arguments.find(a => a.key == parse) ?? this.arguments.find(a => a.alias == parse);
if(!arg) {
extra.push(queue[i]);
continue;
}
if(!value) {
value = arg.type == 'bool' ? true : queue[i + 1];
if(arg.type != 'bool') i++;
}
parsed[arg.key] = value;
} }
req.forEach((a, i) => parsed[a.key] = extra[i]); // Arguments
extra.splice(0, req.length); this.argList.filter(arg => !arg.flags && !arg.extras).forEach(arg => {
return {...parsed, extra}; if(!arg.optional && !extra.length) throw new ArgError(`Argument missing: ${arg.name.toUpperCase()}`);
const value = extra.splice(0, 1)[0];
if(value != null) parsed[arg.name] = value;
});
// Extras
const extraKey = this.argList.find(arg => arg.extras)?.name || 'extra';
parsed[extraKey] = extra;
return parsed;
} }
/** /**
* Create a help message of the expected paramters & usage. * Create help message from the provided description, examples & argument list.
* @param msg {String} - Optional message to display with help * @param message {string} - Message to display, defaults to the description
* @returns {string} - Help message
*/ */
help(msg) { help(msg) {
let message = '\n\n'; // Description
message += msg ? msg : this.description; let message = '\n\n' + (msg && msg.toLowerCase() != 'help' ? msg : this.description);
if(this.examples.length) message += '\n\nUsage:\t' + this.examples.join('\n\t'); // Usage
const required = this.arguments.filter(a => !a.optional); if(this.examples.length) message += '\n\nUsage:\t' + this.examples.map(ex => `run ${this.name} ${ex}`).join('\n\t');
if(required.length) message += '\n\n\t' + required.map(a => { // Arguments
const padding = 3 - ~~(a.key.length / 8); const req = this.argList.filter(a => !a.flags);
return `${a.key}${Array(padding).fill('\t').join('')} ${a.desc}`; if(req.length) message += '\n\n\t' + req.map(arg => {
const padding = 3 - ~~(arg.name.length / 8);
return `${arg.name.toUpperCase()}${Array(padding).fill('\t').join('')} ${arg.desc}`;
}).join('\n\t'); }).join('\n\t');
const optional = this.arguments.filter(a => a.optional); // Flags
if(optional.length) message += '\n\nOptions:\n\t' + optional.map(a => { const opts = this.argList.filter(a => a.flags);
const flgs = `${a.alias ? `-${a.alias} ` : ''}--${a.key}${a.type && a.type != 'bool' ? `=${a.type}` : ''}`; if(opts.length) message += '\n\nOptions:\n\t' + opts.map(a => {
const flgs = a.flags.join(' ');
const padding = 3 - ~~(flgs.length / 8); const padding = 3 - ~~(flgs.length / 8);
return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`; return `${flgs}${Array(padding).fill('\t').join('')} ${a.desc}`;
}).join('\n\t'); }).join('\n\t');
// Print final message
return `${message}\n\n`; return `${message}\n\n`;
} }
} }
/**
* Print a download bar to the terminal.
* @params ns {NS} - Bitburner API
* @params file - Filename to display with progress bar
*/
export async function downloadPrint(ns, file) {
const speed = ~~(Math.random() * 100) / 10;
const spacing = Array(5 - Math.floor((file.length) / 8)).fill('\t').join('');
await slowPrint(ns, `${file}${spacing}[==================>] 100% \t (${speed} MB/s)`);
}
/**
* Print text to the terminal & then delay for a random amount of time to emulate execution time.
* @params ns {NS} - Bitburner API
* @params message {string} - Text to display
* @params min {number} - minimum amount of time to wait after printing text
* @params max {number} - maximum amount of time to wait after printing text
*/
export async function slowPrint(ns, message, min = 0.5, max = 1.5) {
const time = ~~(Math.random() * (max * 1000 - min * 1000)) + min * 1000;
ns.tprint(message);
await ns.sleep(time);
}
/** /**
* Automatically download all the scripts in the repository. * Automatically download all the scripts in the repository.
* @params ns {NS} - BitBurner API
*/ */
export async function main(ns) { export async function main(ns) {
ns.disableLog('ALL');
/**
* Download a file from the repo with some fancy styling & deplays.
* @param file {String} - file name
*/
async function download(file) {
await ns.wget(`${src}${file}`, `${dest}${file}`);
const speed = ~~((Math.random() * 200) + 100) / 10;
ns.tprint(`${file} ${file.length <= 10 ? '\t' : ''}\t [==================>] 100% \t (${speed} MB/s)`);
}
// Initilize script arguments
const argParser = new ArgParser({
desc: 'Automatically download the latest versions of all scripts using wget.',
examples: [
'run update.js',
'run update.js --help',
],
args: [
{key: 'help', alias: 'h', type: 'bool', optional: true, desc: 'Display help message'},
]
});
const args = argParser.parse(ns.args);
if(args['help']) return ns.tprint(argParser.help());
// Setup // Setup
const src = 'https://gitlab.zakscode.com/ztimson/BitBurner/-/raw/develop/scripts/'; ns.disableLog('ALL');
const dest = '/scripts/'; const updateFile = 'update.js';
const argParser = new ArgParser(updateFile, 'Download the latest script updates from the repository using wget.', null, [
{name: 'target', desc: 'Target device to update, defaults to current machine', optional: true, default: ns.getHostname(), type: 'string'}
]);
const src = 'https://gitlab.zakscode.com/ztimson/BitBurner/-/raw/develop/scripts/';
const dest = '/scripts2/';
const fileList = [ const fileList = [
'lib/arg-parser.js', 'lib/arg-parser.js',
'auto-pwn.js', 'lib/utils.js',
'bruteforce.js', 'bruteforce.js',
'crawler.js', 'crawler.js',
'miner.js', 'miner.js',
'network-graph.js', 'network-graph.js',
'node-manager.js' 'node-manager.js',
'rootkit.js'
]; ];
let args;
// Update self & restart try {
if(!ns.args.length) { args = argParser.parse(ns.args || []);
ns.tprint("Updating self:"); } catch(err) {
await ns.sleep(1000); if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
await download('update.js'); throw err;
await ns.sleep(500); }
if(!ns.args.length) { // Update self & restart
await slowPrint(ns, 'Updating self:');
await ns.wget(`${src}${updateFile}`, `${dest}${updateFile}`, args['target']);
await downloadPrint(ns, `${dest}${updateFile}`);
ns.tprint('');
await slowPrint(ns, 'Restarting...');
return ns.exec(`${dest}${updateFile}`, args['target'], 1, 1);
} else { // Update everything else
ns.tprint('');
await slowPrint(ns, 'Downloading scripts:');
for(let file in fileList) {
await ns.wget(`${src}${file}`, `${dest}${file}`, args['target']);
await downloadPrint(ns, `${dest}${file}`);
}
ns.tprint('');
ns.tprint('Done!');
ns.tprint(''); ns.tprint('');
ns.tprint("Restarting...");
await ns.sleep(2000);
return ns.run(`${dest}update.js`, 1, 1);
} }
// Download each file
ns.tprint("Downloading scripts:");
ns.tprint('');
for(const file of fileList) {
await ns.sleep(500);
await download(file);
}
ns.tprint('');
ns.tprint('✅ Done!');
ns.tprint('');
} }