ztimson/bitburner
ztimson/bitburner
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Updated scripts to use ArgParser

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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
``` ```

158
scripts/auto-pwn.js
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];
}

172
scripts/crawler.js
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. * Search the network for targets to execute a script against.
* @param args {any[]} - Array of arguments * @param ns {NS} - BitBurner API
* @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`;
}
}
export async function main(ns) { export async function main(ns) {
ns.disableLog('ALL');
// Initilize script arguments
const argParser = new ArgParser({
desc: 'Search the network for targets to execute a script against.',
examples: [
'run crawler.js [OPTIONS] SCRIPT [ARGS]...',
'run crawler.js --help',
],
args: [
{key: 'SCRIPT', desc: 'Script to copy & execute'},
{key: 'ARGS', skip: true, desc: 'Aditional arguments for SCRIPT. Forward the target with "{{TARGET}}"'},
{key: 'depth', alias: 'd', type: 'num', optional: true, desc: 'Number of network hops. Defaults to 3'},
{key: 'level', alias: 'l', type: 'num', optional: true, desc: 'Exclude targets with a high hacking level. Defaults to hack level, 0 to disable'},
{key: 'ports', alias: 'p', type: 'num', optional: true, desc: 'Exclute targets with too many closed ports'},
{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());
if(!args['SCRIPT']) return ns.tprint(argParser.help('Missing SCRIPT'));
// Setup // Setup
const depth = args['depth'] || 3; ns.disableLog('ALL');
const level = args['level'] || ns.getHackingLevel(); const argParser = new ArgParser('crawler.js', 'Search the network for targets to execute a script against.', null, [
const ports = args['ports'] || Infinity; {name: 'script', desc: 'Script to copy & execute', type: 'string'},
const threads = args['threads'] || 1; {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'], default: 1, type: 'num'},
// Recursively search for targets {name: 'depth', desc: 'Depth to scan to, defaults to 3', flags: ['-d', '--depth'], default: Infinity, type: 'num'},
const targets = ns.scan().map(h => [h, 1]); {name: 'level', desc: 'Exclude targets with higher hack level, defaults to current hack level', flags: ['-l', '--level'], default: ns.getHackingLevel(), type: 'num'},
for(let i = 0; i < targets.length; i++) { {name: 'ports', desc: 'Exclute targets with too many closed ports', flags: ['-p', '--ports'], optional: true, default: Infinity, type: 'num'},
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]); let args;
}); try {
args = argParser.parse(ns.args);
} catch(err) {
if(err instanceof ArgError) return ns.tprint(argParser.help(err.message));
throw err;
} }
// 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('');
} }

34
scripts/lib/arg-parser.js
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);
}

131
scripts/miner.js
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) {

73
scripts/network-graph.js
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,9 +85,10 @@ 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('');
} }

124
scripts/node-manager.js
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];
}

209
scripts/update.js
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] == '-') {
// Check combined flags
if(parse[1] != '-' && parse.length > 2) {
parse = `-${parse[1]}`;
queue = parse.substring(1).split('').map(a => `-${a}`).concat(queue);
} }
let value = null, parse = queue[i].slice(queue[i][1] == '-' ? 2 : 1); // Find & add flag
if(parse.indexOf('=')) {
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) { // Arguments
value = arg.type == 'bool' ? true : queue[i + 1]; this.argList.filter(arg => !arg.flags && !arg.extras).forEach(arg => {
if(arg.type != 'bool') i++; if(!arg.optional && !extra.length) throw new ArgError(`Argument missing: ${arg.name.toUpperCase()}`);
} const value = extra.splice(0, 1)[0];
parsed[arg.key] = value; if(value != null) parsed[arg.name] = value;
} });
req.forEach((a, i) => parsed[a.key] = extra[i]); // Extras
extra.splice(0, req.length); const extraKey = this.argList.find(arg => arg.extras)?.name || 'extra';
return {...parsed, 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
ns.disableLog('ALL');
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 src = 'https://gitlab.zakscode.com/ztimson/BitBurner/-/raw/develop/scripts/';
const dest = '/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);
ns.tprint('');
ns.tprint("Restarting...");
await ns.sleep(2000);
return ns.run(`${dest}update.js`, 1, 1);
} }
// Download each file if(!ns.args.length) { // Update self & restart
ns.tprint("Downloading scripts:"); await slowPrint(ns, 'Updating self:');
await ns.wget(`${src}${updateFile}`, `${dest}${updateFile}`, args['target']);
await downloadPrint(ns, `${dest}${updateFile}`);
ns.tprint(''); ns.tprint('');
for(const file of fileList) { await slowPrint(ns, 'Restarting...');
await ns.sleep(500); return ns.exec(`${dest}${updateFile}`, args['target'], 1, 1);
await download(file); } 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('');
ns.tprint('✅ Done!'); ns.tprint('Done!');
ns.tprint(''); ns.tprint('');
}
} }