From 65050fb6190cf0aed52a5ae3794e9e3c85cd8a38 Mon Sep 17 00:00:00 2001
From: ztimson <zaktimson@gmail.com>
Date: Sun, 20 Mar 2022 09:01:52 -0400
Subject: [PATCH] Rootkit will skip rooting if it already has it to save time

---
 README.md          |  2 +-
 scripts/rootkit.js | 57 +++++++++++++++++++++-------------------------
 2 files changed, 27 insertions(+), 32 deletions(-)

diff --git a/README.md b/README.md
index 2f9874c..e7c8e4e 100644
--- a/README.md
+++ b/README.md
@@ -218,7 +218,7 @@ Options:
 ```
 
 ### [rootkit.js](./scripts/rootkit.js)
-**RAM:** 4.75 GB - 4.90 GB <small>(depending on un-commented programs)</small>
+**RAM:** 4.80 GB - 4.95 GB <small>(depending on un-commented programs)</small>
 
 Automatically gain root on a target machine. A file can also be uploaded & executed.
 
diff --git a/scripts/rootkit.js b/scripts/rootkit.js
index e5e04a5..827b127 100644
--- a/scripts/rootkit.js
+++ b/scripts/rootkit.js
@@ -36,44 +36,39 @@ export async function main(ns) {
 			ns.tprint('===================================================');
 		}
 
-		let spacer = false;
-		const sleep = 750;
-		try {
-			// Run exploits
-			ns.brutessh(args['device']);
-			if(!args['silent']) {
-				await slowPrint(ns, `Attacking over SSH (${args['device']}:22)...`, 0.5, 1.5);
-				await ns.sleep(sleep);
-				spacer = true;
-			}
-			ns.ftpcrack(args['device']);
-			if(!args['silent']) {
-				await slowPrint(ns, `Attacking over FTP (${args['device']}:24)...`, 0.5, 1.5);
-				await ns.sleep(sleep);
-			}
-			ns.relaysmtp(args['device']);
-			if(!args['silent']) {
-				await slowPrint(ns, `Attacking over SMTP (${args['device']}:25)...`, 0.5, 1.5);
-				await ns.sleep(sleep);
-			}
-		} catch {
-		} finally {
+		// Check if we already have root
+		if(ns.hasRootAccess(args['device'])) {
+			if(!args['silent']) ns.tprint('Root: Skipped');
+		} else {
+			let spacer = false;
 			try {
-				// Attempt root
-				if(spacer) ns.tprint('');
-				ns.nuke(args['device'])
+				// Run exploits
+				ns.brutessh(args['device']);
 				if(!args['silent']) {
-					ns.tprint(`Root: Success!`);
-					ns.tprint('');
+					await slowPrint(ns, `Attacking over SSH (${args['device']}:22)...`, 0.5, 1.5);
+					spacer = true;
 				}
+				ns.ftpcrack(args['device']);
+				if(!args['silent']) await slowPrint(ns, `Attacking over FTP (${args['device']}:24)...`, 0.5, 1.5);
+				ns.relaysmtp(args['device']);
+				if(!args['silent']) await slowPrint(ns, `Attacking over SMTP (${args['device']}:25)...`, 0.5, 1.5);
 			} catch {
-				if(!args['silent']) {
-					ns.tprint(`Root: Failed`);
-					ns.tprint('');
+			} finally {
+				try {
+					// Attempt root
+					if(spacer) ns.tprint('');
+					ns.nuke(args['device']);
+					if(!args['silent']) ns.tprint(`Root: Success!`);
+				} catch {
+					if(!args['silent']) {
+						ns.tprint(`Root: Failed`);
+						ns.tprint('');
+					}
+					ns.exit();
 				}
-				ns.exit();
 			}
 		}
+		ns.tprint('');
 
 		if(args['script']) {
 			// Detect script dependencies & copy everything to target