From 263b6809448c12af82609e60b3d170e947dbfbef Mon Sep 17 00:00:00 2001
From: CodeX <ethanseow2002@gmail.com>
Date: Wed, 26 Oct 2022 14:50:39 +0800
Subject: [PATCH] Update screenshotBOF.cna

---
 bin/BOF/screenshotBOF.cna | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/bin/BOF/screenshotBOF.cna b/bin/BOF/screenshotBOF.cna
index 3fcd237..bb9bbbc 100644
--- a/bin/BOF/screenshotBOF.cna
+++ b/bin/BOF/screenshotBOF.cna
@@ -7,7 +7,11 @@ beacon_command_register(
 
 alias screenshot_bof {
     local('$barch $handle $data $args $target_pid');
-    println(@_);
+    if (size(@_) != 2)
+    {
+        berror($1, "Please specify a filename. e.g. screenshot_bof screen.bmp");
+                return;
+    }
     # figure out the arch of this session
     $barch  = barch($1);
     # read in the right BOF file