From fbf361b6a6142649f526febe50e32cb0d1608a5e Mon Sep 17 00:00:00 2001
From: Zribe <66210834+ZribeDev@users.noreply.github.com>
Date: Thu, 29 Feb 2024 17:43:20 +0200
Subject: [PATCH] Update README.md

---
 README.md | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/README.md b/README.md
index abf7a74..da5d5e8 100644
--- a/README.md
+++ b/README.md
@@ -44,3 +44,44 @@ static void SetProcessCriticalStatus(int pid, bool setStatus, Action<string, Col
         logAction($"Something went wrong: {ex.Message}", Color.Red);
     }
 }
+```
+Since you don't have the same logging setup as me, I've tweaked the code to use Console.WriteLine instead. Check it out:
+```csharp
+using System;
+using System.Diagnostics;
+using System.Runtime.InteropServices;
+
+const int STATUS_SUCCESS = 0;
+const int ProcessBreakOnTermination = 0x1D;
+
+[DllImport("ntdll.dll", SetLastError = true)]
+static extern int NtSetInformationProcess(IntPtr processHandle, int processInformationClass, ref int processInformation, int processInformationLength);
+
+static void SetProcessCriticalStatus(int pid, bool setStatus)
+{
+    try
+    {
+        Process process = Process.GetProcessById(pid);
+        if (process == null)
+        {
+            Console.WriteLine("Couldn't find the process.");
+            return;
+        }
+
+        int isCritical = setStatus ? 1 : 0;
+        int result = NtSetInformationProcess(process.Handle, ProcessBreakOnTermination, ref isCritical, sizeof(int));
+        if (result == STATUS_SUCCESS)
+        {
+            Console.WriteLine($"Process {(setStatus ? "is now vulnerable" : "is back to normal")} successfully.");
+        }
+        else
+        {
+            Console.WriteLine($"Couldn't change the process. Error: {result}");
+        }
+    }
+    catch (Exception ex)
+    {
+        Console.WriteLine($"Something went wrong: {ex.Message}");
+    }
+}
+