feat: script hash verify

Author: Esonhugh

craft.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import os
 import json
+from hashlib import sha256
 from base64 import b64encode
 
 def read_file_return_base64ed_content(filename):
@@ -28,7 +29,19 @@ def create_value_map(filepath:str):
 vscode_path = ".vscode"
 
 template_head = """#!/bin/zsh
-createhack () {
+# __        __                        __     ______   ____          _      
+# \ \      / /__  __ _ _ __   ___  _ _\ \   / / ___| / ___|___   __| | ___ 
+#  \ \ /\ / / _ \/ _` | '_ \ / _ \| '_ \ \ / /\___ \| |   / _ \ / _` |/ _ \\
+#   \ V  V /  __/ (_| | |_) | (_) | | | \ V /  ___) | |__| (_) | (_| |  __/
+#    \_/\_/ \___|\__,_| .__/ \___/|_| |_|\_/  |____/ \____\___/ \__,_|\___|
+#                    |_|                                                  
+#
+# Author: Esonhugh <[email protected]>
+# Path: createhackenv.sh
+# Usage: createhackenv <foldername>
+# Description: Create a hacking project with vscode based environment
+
+weapon_vscode () {
 	echo "Creating VSCode Hack Environment"
 	echo "Usage: $0 <foldername>"
 	if [ -z "$1" ]
@@ -48,6 +61,12 @@ def create_value_map(filepath:str):
 	echo "Generate Success" "Launched VSCode"
 	code $1
 }
+
+# Create Alias for weapon_vscode
+alias createhackenv=weapon_vscode
+alias createhack=weapon_vscode
+
+# Script Hash: __HASH__
 """
 
 
@@ -75,10 +94,13 @@ def test():
     print(filename_to_variable_name("./vscode/source.zsh"))
     print(generate_bash_script())
 
+def content_hash(content):
+    return sha256(content.encode('utf-8')).hexdigest()
 
 def main():
     print("reGenerating bash script")
     content =  generate_bash_script()
+    content = content.replace("__HASH__", content_hash(content))
     print("Content: ")
     print(content)
     print("Writing to createhackenv.sh file")

createhackenv.sh

@@ -1,5 +1,17 @@
 #!/bin/zsh
-createhack () {
+# __        __                        __     ______   ____          _      
+# \ \      / /__  __ _ _ __   ___  _ _\ \   / / ___| / ___|___   __| | ___ 
+#  \ \ /\ / / _ \/ _` | '_ \ / _ \| '_ \ \ / /\___ \| |   / _ \ / _` |/ _ \
+#   \ V  V /  __/ (_| | |_) | (_) | | | \ V /  ___) | |__| (_) | (_| |  __/
+#    \_/\_/ \___|\__,_| .__/ \___/|_| |_|\_/  |____/ \____\___/ \__,_|\___|
+#                    |_|                                                  
+#
+# Author: Esonhugh <[email protected]>
+# Path: createhackenv.sh
+# Usage: createhackenv <foldername>
+# Description: Create a hacking project with vscode based environment
+
+weapon_vscode () {
 	echo "Creating VSCode Hack Environment"
 	echo "Usage: $0 <foldername>"
 	if [ -z "$1" ]
@@ -45,3 +57,9 @@ createhack () {
 	echo "Generate Success" "Launched VSCode"
 	code $1
 }
+
+# Create Alias for weapon_vscode
+alias createhackenv=weapon_vscode
+alias createhack=weapon_vscode
+
+# Script Hash: 2ac1b5c9ab4545d52df6fa05c64da92a2e8bfe442be8433c80fad25f3a1331aa

verify.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+
+from hashlib import sha256
+import sys
+
+def content_hash(content:str):
+    return sha256(content.encode('utf-8')).hexdigest()
+
+def read_file(filename):
+    with open(filename, 'r') as f:
+        data = f.readlines()
+    return data
+
+def regenerate_hash(filename):
+    data = read_file("createhackenv.sh")
+    script_hash_line = data[-1]
+    # print(f"Script hash line: {script_hash_line}")
+    raw_script_hash = script_hash_line.split(" ")[3].strip()
+    # print(f"Script hash in Script: {raw_script_hash}")
+    content = "".join(data)
+    content = content.replace(raw_script_hash, "__HASH__")
+    calc_script_hash = content_hash(content)
+    # print(f"Script reCalc hash: {calc_script_hash}")
+    # print(f"Script hash: {raw_script_hash}")
+    return calc_script_hash, raw_script_hash
+
+def verify_hash(calc_script_hash, raw_script_hash):
+    print("Verifying script hash...")
+    print(f"Script reCalc hash: {calc_script_hash}")
+    print(f"Script hash wait for check: {raw_script_hash}")
+    if calc_script_hash == raw_script_hash:
+        print("Script hash is correct")
+    else:
+        print("Script hash is incorrect!")
+        exit(114)
+
+def main():
+    if len(sys.argv) == 2 :
+        regen_hash,raw_hash = regenerate_hash(sys.argv[1])
+        verify_hash(regen_hash, raw_hash)
+    elif len(sys.argv) == 3:
+        user_hash = sys.argv[2]
+        regen_hash,raw_hash = regenerate_hash(sys.argv[1])
+        verify_hash(regen_hash, user_hash)
+    else:
+        print("Usage: python3 verify.py <filename> [<check_hash>]")
+        exit(1)
+
+if __name__ == "__main__":
+    main()