From 7ed89bcb4ceb2bf1b9e8ff2dc5acdab20833cf1c Mon Sep 17 00:00:00 2001
From: EnderIce2 <enderice2@users.noreply.github.com>
Date: Sat, 2 Mar 2024 01:40:04 +0200
Subject: [PATCH] Check USTAR archive validation in KernelVFS()

---
 kernel_vfs.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/kernel_vfs.cpp b/kernel_vfs.cpp
index dd3bd26..4f0b91e 100644
--- a/kernel_vfs.cpp
+++ b/kernel_vfs.cpp
@@ -57,7 +57,13 @@ EXTERNC NIF void KernelVFS()
 				}
 
 				vfs::USTAR *ustar = new vfs::USTAR;
-				ustar->ReadArchive(initrdAddress, fs);
+				if (!ustar->TestArchive(initrdAddress))
+				{
+					KPrint("\eFF0000USTAR archive is invalid!");
+					delete ustar;
+				}
+				else
+					ustar->ReadArchive(initrdAddress, fs);
 			}
 		}
 	}