diff --git a/Application/img_steg.py b/Application/img_steg.py new file mode 100644 index 0000000..384224e --- /dev/null +++ b/Application/img_steg.py @@ -0,0 +1,160 @@ +import cv2 +import numpy as np + + +class img_steg: + def __init__(self, image_name: str = "image.png", bit_to_hide: int = 2): + """ + Initialize the class + :param image_name: Name of the image to encode or decode + :type image_name: str + :param bit_to_hide: Bit to hide the data in (1 - LSB to 8 - MSB) + :type bit_to_hide: int + """ + self.image_name = image_name + self.bit_to_hide = 8 - bit_to_hide + self.delimiter = "\n" + + def to_bin(self, data: str) -> str | list[str]: + """ + Convert data to binary format as string + Args: + data: String + + Returns: + Binary Data: String | List[String] + """ + if isinstance(data, str): + return ''.join([format(ord(i), "08b") for i in data]) + elif isinstance(data, bytes) or isinstance(data, np.ndarray): + return [format(i, "08b") for i in data] + elif isinstance(data, int) or isinstance(data, np.unit8): + return format(data, "08b") + else: + raise TypeError("Type not supported") + + def from_bin(self, data: str) -> str: + """ + Convert binary `data` back to the original format + Args: + data: String + + Returns: + Original Data: String + """ + return ''.join([chr(int(data[i:i + 8], 2)) for i in range(0, len(data), 8)]) + + def decode(self) -> str: + """ + Decode the data hidden in the image + Args: + None + :return: Decoded Data: String + """ + print("[+] Decoding...") + image = cv2.imread(self.image_name) # read image + binary_data = "" + bit_to_hide = self.bit_to_hide + for row in image: + for pixel in row: + r, g, b = self.to_bin(pixel) + binary_data += r[bit_to_hide] + binary_data += g[bit_to_hide] + binary_data += b[bit_to_hide] + # Split by 8 bits + all_bytes = [binary_data[i: i + 8] for i in range(0, len(binary_data), 8)] + # Convert from bits to characters + decoded_data = self.from_bin(''.join(all_bytes)) + if decoded_data.endswith(self.delimiter): + decoded_data = decoded_data[:-4] + + decoded_data = decoded_data.split(self.delimiter)[0] + return decoded_data + + def encode(self, secret_data: str = "Hello World") -> np.ndarray: + """ + Encode the secret data in the image + Args: + secret_data: str + + Returns: + Encoded Image: np.ndarray + """ + image = cv2.imread(self.image_name) # read image + n_bytes = image.shape[0] * image.shape[1] * 3 // 8 # Max bytes to encode + print("[*] Maximum bytes to encode:", n_bytes) + secret_data += self.delimiter # Add delimiter at the end of data + if len(secret_data) > n_bytes: + raise ValueError("[!] Insufficient bytes, need a bigger image or less data") + print("[*] Encoding Data...") + + # Convert bit to hide position + bit_to_hide = self.bit_to_hide + data_index = 0 + binary_secret_data = self.to_bin(secret_data) # Convert data to binary + data_len = len(binary_secret_data) # size of data to hide + for row in image: + for pixel in row: + r, g, b = self.to_bin(pixel) # Convert RGB Values to binary format + if data_index < data_len: + r = list(r) + r[bit_to_hide] = binary_secret_data[data_index] # hide data into least significant bit of red pixel + pixel[0] = int(''.join(r), 2) + data_index += 1 + if data_index < data_len: + g = list(g) + g[bit_to_hide] = binary_secret_data[data_index] + pixel[1] = int(''.join(g), 2) + data_index += 1 + if data_index < data_len: + b = list(b) + b[bit_to_hide] = binary_secret_data[data_index] + pixel[2] = int(''.join(b), 2) + data_index += 1 + if data_index >= data_len: + break + return image + + +def main(): + print("Welcome to Image Steganography") + print("1. Encode\n2. Decode\n3. Exit") + choice = int(input("Enter your choice: ")) + if choice == 1: + image_name = input("Enter name of image to encode: ") + secret_data = input("Enter data to encode: ") + bit_to_hide = int(input("Enter bit to hide (1 - LSB to 8 - MSB): ")) + encoded_image = img_steg(image_name, bit_to_hide).encode(secret_data) + extension = image_name.split(".")[-1] + cv2.imwrite("encoded_image." + extension, encoded_image) + print("Image Encoded Successfully") + elif choice == 2: + image_name = input("Enter name of image to decode: ") + bit_to_hide = int(input("Enter bit to hide (1 - LSB to 8 - MSB): ")) + decoded_data = img_steg(image_name, bit_to_hide).decode() + print("Decoded Data:", decoded_data) + + +if __name__ == "__main__": + main() + +############################## +# _ _ +# | | | |___ __ _ __ _ ___ +# | | | / __|/ _` |/ _` |/ _ \ +# | |_| \__ \ (_| | (_| | __/ +# \___/|___/\__,_|\__, |\___| +# |___/ +############################## +# +# Initialise class with image name and bit to hide +# +# image_name = image name with extension (String) +# bit_to_hide = bit to hide the data in (1 - LSB to 8 - MSB) (int) +# +# To encode: +# encoded_image = img_steg(image_name, bit_to_hide).encode(secret_data) +# +# To decode: +# decoded_data = img_steg(image_name, bit_to_hide).decode() +# Both are string types \ No newline at end of file diff --git a/Application/lsb_rep.py b/Application/lsb_rep.py new file mode 100644 index 0000000..b0b63f4 --- /dev/null +++ b/Application/lsb_rep.py @@ -0,0 +1,49 @@ +from flask import Flask, render_template, request, redirect +app = Flask(__name__, template_folder='views') + +@app.route('/') +def modeSelection(): + return render_template('mode_selection.html') + +@app.route('/encode') +def encode(): + return render_template('encode.html') + +@app.route("/encoding", methods=['POST']) +def encoding(): + file = request.files['origin'] + payload = request.form['payload'] + + if file.filename != "": + file.save('upload/' + file.filename) + + # run the encoding function + #return encoded file to the result below + return redirect("/encode_result") + +@app.route('/encode_result') +def encode_result(): + return render_template("encode_result.html") + +@app.route('/decode') +def decode(): + return render_template('decode.html') + +@app.route("/decoding", methods=['POST']) +def decoding(): + file = request.files['encoded_file'] + + if file.filename != "": + file.save('upload/' + file.filename) + + # run the decoding function + #return decoded payload to the result below + return redirect("/decode_result") + +@app.route('/decode_result') +def decode_result(): + return render_template("decode_result.html") + +if __name__ == "__main__": + app.debug = True + app.run(host="localhost", port=8000) \ No newline at end of file diff --git a/Application/views/decode.html b/Application/views/decode.html new file mode 100644 index 0000000..76e4ad3 --- /dev/null +++ b/Application/views/decode.html @@ -0,0 +1,86 @@ + + + + + + + + Decode + + + + + +
+
+
+
⇩
+
Drag and drop an image here or click to browse
+ +
+
+ +
+
+
+ + + + + \ No newline at end of file diff --git a/Application/views/decode_result.html b/Application/views/decode_result.html new file mode 100644 index 0000000..e69de29 diff --git a/Application/views/encode.html b/Application/views/encode.html new file mode 100644 index 0000000..d12a071 --- /dev/null +++ b/Application/views/encode.html @@ -0,0 +1,88 @@ + + + + + + + + Encode + + + + + +
+
+
+
⇩
+
Drag and drop an image here or click to browse
+ +
+
+ + +
+
+
+ + + + + \ No newline at end of file diff --git a/Application/views/encode_result.html b/Application/views/encode_result.html new file mode 100644 index 0000000..e69de29 diff --git a/Application/views/mode_selection.html b/Application/views/mode_selection.html new file mode 100644 index 0000000..e184c9c --- /dev/null +++ b/Application/views/mode_selection.html @@ -0,0 +1,48 @@ + + + + + + + + Mode Selection + + + + + +
+

Mode Selection

+ +
+ Encode + Decode +
+
+ + + + \ No newline at end of file