added argument parsing for steganography tutorial

Author: x4nth055

ethical-hacking/steganography/README.md

@@ -1,15 +1,41 @@
 # [How to use Steganography to Hide Secret Data in Images in Python](https://www.thepythoncode.com/article/hide-secret-data-in-images-using-steganography-python)
 To run this:
 - `pip3 install -r requimements.txt`
-- To encode some data to the imag `image.PNG` and decode it right away:
+-
     ```
-    python steganography image.PNG "This is some secret data."
+    python steganography.py --help
     ```
-    This will write another image with data encoded in it and **outputs:**
+    **Output**:
+    ```
+        usage: steganography.py [-h] [-t TEXT] [-e ENCODE] [-d DECODE]
+
+    Steganography encoder/decoder, this Python scripts encode data within images.
+
+    optional arguments:
+    -h, --help            show this help message and exit
+    -t TEXT, --text TEXT  The text data to encode into the image, this only
+                            should be specified for encoding
+    -e ENCODE, --encode ENCODE
+                            Encode the following image
+    -d DECODE, --decode DECODE
+                            Decode the following image
+    ```
+- To encode some data to the image `image.PNG`:
+    ```
+    python steganography.py -e image.PNG -t "This is some secret data."
+    ```
+    This will write another image `imaged_encoded.PNG` with data encoded in it and **outputs:**
     ```
     [*] Maximum bytes to encode: 125028
     [*] Encoding data...
+    [+] Saved encoded image.
+    ```
+- To decode the data containing the image `encoded_image.PNG`, use:
+    ```
+    python steganography.py -d encoded_image.PNG
+    ```
+    **Outputs:**
+    ```
     [+] Decoding...
     [+] Decoded data: This is some secret data.
     ```
-- You can isolate encoding and decoding processes in two different Python files, which makes more sense.

ethical-hacking/steganography/encoded_image.PNG

@@ -1,7 +1,6 @@
 import cv2
 import numpy as np
-import sys
-
+import os
 
 def to_bin(data):
     """Convert `data` to binary format as string"""
@@ -79,14 +78,30 @@ def decode(image_name):
 
 
 if __name__ == "__main__":
-    input_image = sys.argv[1]
-    output_image = f"encoded_{input_image}"
-    secret_data = sys.argv[2]
-
-    # encode the data into the image
-    encoded_image = encode(image_name=input_image, secret_data=secret_data)
-    # save the output image (encoded image)
-    cv2.imwrite(output_image, encoded_image)
-    # decode the secret data from the image
-    decoded_data = decode(output_image)
-    print("[+] Decoded data:", decoded_data)
+    import argparse
+    parser = argparse.ArgumentParser(description="Steganography encoder/decoder, this Python scripts encode data within images.")
+    parser.add_argument("-t", "--text", help="The text data to encode into the image, this only should be specified for encoding")
+    parser.add_argument("-e", "--encode", help="Encode the following image")
+    parser.add_argument("-d", "--decode", help="Decode the following image")
+    
+    args = parser.parse_args()
+    secret_data = args.text
+    if args.encode:
+        # if the encode argument is specified
+        input_image = args.encode
+        print("input_image:", input_image)
+        # split the absolute path and the file
+        path, file = os.path.split(input_image)
+        # split the filename and the image extension
+        filename, ext = file.split(".")
+        output_image = os.path.join(path, f"{filename}_encoded.{ext}")
+        # encode the data into the image
+        encoded_image = encode(image_name=input_image, secret_data=secret_data)
+        # save the output image (encoded image)
+        cv2.imwrite(output_image, encoded_image)
+        print("[+] Saved encoded image.")
+    if args.decode:
+        input_image = args.decode
+        # decode the secret data from the image
+        decoded_data = decode(input_image)
+        print("[+] Decoded data:", decoded_data)