more cleanup

dvnc0 · Mar 23, 2024 · a47fff3 · a47fff3
1 parent 3ff660e
commit a47fff3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/src/Gremlins/Black_Hole_Gremlin.php b/src/Gremlins/Black_Hole_Gremlin.php
@@ -11,11 +11,20 @@ class Black_Hole_Gremlin extends Gremlin {
 	 * @return void
 	 */
 	public function attack(): void {
-		$black_hole = fopen('/dev/null', 'w');
-		fwrite($black_hole, 'Black Hole Gremlin');
-		$size = rand(1000000, 9999999999);
-		$random_string = str_repeat('1', $size);
-		fwrite($black_hole, $random_string);
-		fclose($black_hole);
+		$pid = pcntl_fork();
+		if ($pid === -1) {
+			$this->writeToLog('Black Hole Gremlin failed to fork');
+			exit(1);
+		} elseif ($pid) {
+			$this->writeToLog('Black Hole Gremlin is attacking the system');
+			return;
+		} else {
+			$black_hole = fopen('/dev/null', 'w');
+			fwrite($black_hole, 'Black Hole Gremlin');
+			$size = rand(1000000, 9999999999);
+			$random_string = str_repeat('1', $size);
+			fwrite($black_hole, $random_string);
+			fclose($black_hole);
+		}
 	}
 }
diff --git a/src/Gremlins/Disk_Gremlin.php b/src/Gremlins/Disk_Gremlin.php
@@ -6,7 +6,7 @@
 class Disk_Gremlin extends Gremlin {
 
 	/**
-	 * Attack the system by writing a random amount of data to /dev/null
+	 * Attack the system by writing a random amount of data to disk
 	 *
 	 * @return void
 	 */