bmp, png, txt, wav, Mp3, Mp4 fully integrated

Application/lsb_rep.py

@@ -1,10 +1,12 @@
 from flask import Flask, render_template, request, redirect, session, send_from_directory
-from lib.steganography import img_steg, wav_steg, txt_steg
+from lib.steganography import img_steg, wav_steg, txt_steg, file_steg
 import cv2
 import os
 import sys
 import wave
+import warnings
 
+MAX_SESSION_SIZE = 4093
 WORKING_PATH = os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])), "upload") + os.sep
 
 app = Flask(__name__, template_folder='views')
@@ -30,39 +32,48 @@ def encoding():
     try:
         file = request.files['origin']
         b2c = [int(x) for x in request.form.getlist("b2c")]
-        # b2c = [int(request.form['b2c'])]
         payload = request.form['payload']
 
         if file.filename != "":
             file.save(WORKING_PATH + file.filename)
 
-            file_extension = os.path.splitext(file.filename)[1]
+            file_extension = os.path.splitext(file.filename)[1][1:]
-            if file_extension == ".png" or file_extension == ".bmp":
+            if file_extension == "png" or file_extension == "bmp":
-                steg = img_steg.img_steg(WORKING_PATH + file.filename, b2c).encode(payload)
+                encode = img_steg.img_steg(WORKING_PATH + file.filename, b2c).encode(payload)
-                cv2.imwrite(WORKING_PATH + "encoded_" + file.filename, steg)
+                cv2.imwrite(WORKING_PATH + "encoded_" + file.filename, encode)
                 session['image'] = file.filename
                 session['image2'] = "encoded_" + file.filename
-            elif file_extension == ".wav":
+            elif file_extension == "wav":
-                steg = wav_steg.wav_steg(WORKING_PATH + file.filename, b2c).encode(payload)
+                encode = wav_steg.wav_steg(WORKING_PATH + file.filename, b2c).encode(payload)
 
                 # Write encoded data to file
                 new_wav_file = wave.open(WORKING_PATH + "encoded_" + file.filename, "wb")
-                new_wav_file.setnchannels(steg["num_channels"])
+                new_wav_file.setnchannels(encode["num_channels"])
-                new_wav_file.setsampwidth(steg["sample_width"])
+                new_wav_file.setsampwidth(encode["sample_width"])
-                new_wav_file.setframerate(steg["frame_rate"])
+                new_wav_file.setframerate(encode["frame_rate"])
-                new_wav_file.writeframes(steg["num_frames"])
+                new_wav_file.writeframes(encode["num_frames"])
                 new_wav_file.close()
                 session['wav'] = file.filename
                 session['wav2'] = "encoded_" + file.filename
-            elif file_extension == ".txt":
+            elif file_extension == "txt":
                 encoded_data = txt_steg.txt_steg(WORKING_PATH + file.filename, b2c).encode(payload)
                 with open(os.path.join(WORKING_PATH, "encoded_" + file.filename), "w") as f:
                     f.write(encoded_data)
                 session['txt'] = file.filename
                 session['txt2'] = "encoded_" + file.filename
+            elif file_extension == "mp3" or file_extension == "mp4":
+                encoded_data = file_steg.file_steg(WORKING_PATH + file.filename, b2c).encode(payload)
+                with open(os.path.join(WORKING_PATH, "encoded_" + file.filename), "wb") as f:
+                    f.write(encoded_data)
+                session[file_extension] = file.filename
+                session[file_extension + '_2'] = "encoded_" + file.filename
 
-            return redirect("/encode_result")
+        return redirect("/encode_result")
-    except:
+    except Exception as exception:
+        print("ENCODE(Exception):", exception)
+        return redirect("/unsupported")
+    except UserWarning as warning:
+        print("ENCODE(warning):", warning)
         return redirect("/unsupported")
 
 @app.route('/encode_result')
@@ -79,28 +90,47 @@ def decode():
 
 @app.route("/decoding", methods=['POST'])
 def decoding():
+
+    def decode_files(class_name, _path, b2c, ext):
+        class_name_str = class_name.__name__.split(".")[-1]
+        method = getattr(class_name, class_name_str)
+        payload = method(_path, b2c).decode()
+
+        if class_name_str == "img_steg":
+            session["image"] = file.filename
+        else:
+            session[ext] = file.filename
+
+        if sys.getsizeof(payload) <= MAX_SESSION_SIZE:
+            session["payload"] = payload
+        else:
+            raise Exception("Payload too big")
+
     try:
         file = request.files['encoded_file']
         b2c = [int(x) for x in request.form.getlist("b2c")]
         if file.filename != "":
-            file.save(WORKING_PATH + file.filename)
 
-            file_extension = os.path.splitext(file.filename)[1]
+            _path = WORKING_PATH + file.filename
-            if file_extension == ".png" or file_extension == ".bmp":
+
-                payload = img_steg.img_steg(WORKING_PATH + file.filename, b2c).decode()
+            file.save(_path)
-                session["payload"] = payload
+
-                session["image"] = file.filename
+            file_extension = os.path.splitext(file.filename)[1][1:]
-            elif file_extension == ".wav":
+            if file_extension == "png" or file_extension == "bmp":
-                payload = wav_steg.wav_steg(WORKING_PATH + file.filename, b2c).decode()
+                decode_files(img_steg, _path, b2c, file_extension)
-                session["payload"] = payload
+            elif file_extension == "wav":
-                session["wav"] = file.filename
+                decode_files(wav_steg, _path, b2c, file_extension)
-            elif file_extension == ".txt":
+            elif file_extension == "txt":
-                payload = txt_steg.txt_steg(WORKING_PATH + file.filename, b2c).decode()
+                decode_files(txt_steg, _path, b2c, file_extension)
-                session["payload"] = payload
+            elif file_extension == "mp3" or file_extension == "mp4":
-                session["txt"] = file.filename
+                decode_files(file_steg, _path, b2c, file_extension)
 
         return redirect("/decode_result")
-    except:
+    except Exception as exception:
+        print("DECODE(Exception):", exception)
+        return redirect("/unsupported")
+    except UserWarning as warning:
+        print("DECODE(warning):", warning)
         return redirect("/unsupported")
 
 @app.route('/decode_result')

Application/upload/payload.txt

@@ -1,76 +0,0 @@
-Secret Data
-
-We're no strangers to love
-You know the rules and so do I
-A full commitment's what I'm thinking of
-You wouldn't get this from any other guy
-
-I just wanna tell you how I'm feeling
-Gotta make you understand
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-We've known each other for so long
-Your heart's been aching, but
-You're too shy to say it
-Inside, we both know what's been going on
-We know the game and we're gonna play it
-
-And if you ask me how I'm feeling
-Don't tell me you're too blind to see
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-(Ooh, give you up)
-(Ooh, give you up)
-Never gonna give, never gonna give
-(Give you up)
-Never gonna give, never gonna give
-(Give you up)
-
-We've known each other for so long
-Your heart's been aching, but
-You're too shy to say it
-Inside, we both know what's been going on
-We know the game and we're gonna play it
-
-I just wanna tell you how I'm feeling
-Gotta make you understand
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you

Application/upload/secret_data.txt

@@ -1,76 +0,0 @@
-Secret Data
-
-We're no strangers to love
-You know the rules and so do I
-A full commitment's what I'm thinking of
-You wouldn't get this from any other guy
-
-I just wanna tell you how I'm feeling
-Gotta make you understand
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-We've known each other for so long
-Your heart's been aching, but
-You're too shy to say it
-Inside, we both know what's been going on
-We know the game and we're gonna play it
-
-And if you ask me how I'm feeling
-Don't tell me you're too blind to see
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-(Ooh, give you up)
-(Ooh, give you up)
-Never gonna give, never gonna give
-(Give you up)
-Never gonna give, never gonna give
-(Give you up)
-
-We've known each other for so long
-Your heart's been aching, but
-You're too shy to say it
-Inside, we both know what's been going on
-We know the game and we're gonna play it
-
-I just wanna tell you how I'm feeling
-Gotta make you understand
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you
-
-Never gonna give you up
-Never gonna let you down
-Never gonna run around and desert you
-Never gonna make you cry
-Never gonna say goodbye
-Never gonna tell a lie and hurt you

Application/views/decode_result.html

@@ -15,8 +15,8 @@
     $(document).ready(function(){
 
         $.get('/get_session', function(response) {
+            console.log(response)
             if(response.hasOwnProperty("image")){
-                console.log(response)
                 var html_content = `<section style="margin: 100px auto; width:50%; height:50%; text-align:center;">
                     <div style="width:100%; height:100%; text-align:center;">
                         <h1 style="height:15%; width:90%; padding: 5%">Decoded</h1>
@@ -38,7 +38,34 @@
                     </audio>
                     <div>
                         <h1 style="height:15%; width:90%; padding: 5%">Secret Text:</h1>
-                        <p>` + response["payload"] + `</p>
+                        <p style="color: red">` + response["payload"] + `</p>
+                    </div>
+                </section>`
+                $("body").append(html_content)
+            }
+            else if(response.hasOwnProperty("mp3")){
+                var html_content = `<section style="margin: 100px auto; width:50%; height:50%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Decoded</h1>
+                    <audio controls>
+                        <source src="upload/` + response["mp3"] + `" type="audio/wav">
+                        Your browser does not support the audio element.
+                    </audio>
+                    <div>
+                        <h1 style="height:15%; width:90%; padding: 5%">Secret Text:</h1>
+                        <p style="color: red">` + response["payload"] + `</p>
+                    </div>
+                </section>`
+                $("body").append(html_content)
+            }
+            else if(response.hasOwnProperty("mp4")){
+                var html_content = `<section style="margin: 100px auto; width:50%; height:50%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Decoded</h1>
+                    <video style="width: 100%" src="upload/` + response["mp4"] + `" controls autoplay>
+                        Sorry, your browser doesn't support embedded videos.
+                    </video>
+                    <div>
+                        <h1 style="height:15%; width:90%; padding: 5%">Secret Text:</h1>
+                        <p style="color: red">` + response["payload"] + `</p>
                     </div>
                 </section>`
                 $("body").append(html_content)
@@ -54,7 +81,7 @@
                             </div>
                             <div style="width:100%; height:100%; text-align:center;">
                                 <h1 style="height:15%; width:90%; padding: 5%">Secret Text:</h1>
-                                <p>` + response["payload"] + `</p>
+                                <p style="color: red">` + response["payload"] + `</p>
                             </div>
                         </section>`
                         $("body").append(html_content)

Application/views/encode_result.html

@@ -26,8 +26,8 @@
     $(document).ready(function(){
 
         $.get('/get_session', function(response) {
+            console.log(response)
             if(response.hasOwnProperty("image")){
-                console.log(response)
                 var html_content = `<section style="margin: 100px auto; width:50%; height:50%; display: flex; justify-content: space-between;">
                     <div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
                         <h1 style="height:15%; width:90%; padding: 5%">Original</h1>
@@ -60,6 +60,42 @@
                 </section>`
                 $("body").append(html_content)
             }
+            else if(response.hasOwnProperty("mp3")){
+                var html_content = `<section style="margin: 100px auto; width:50%; height:50%; display: flex; justify-content: space-between;">
+                    <div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Original</h1>
+                    <audio controls>
+                        <source src="upload/` + response["mp3"] + `" type="audio/wav">
+                        Your browser does not support the audio element.
+                    </audio>
+                </div>
+                <div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Encoded</h1>
+                    <audio controls>
+                        <source src="upload/` + response["mp3_2"] + `" type="audio/wav">
+                        Your browser does not support the audio element.
+                    </audio>
+                </div>
+                </section>`
+                $("body").append(html_content)
+            }
+            else if(response.hasOwnProperty("mp4")){
+                var html_content = `<section style="margin: 100px auto; width:50%; height:50%; display: flex; justify-content: space-between;">
+                    <div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Original</h1>
+                    <video style="width: 100%" src="upload/` + response["mp4"] + `" controls autoplay>
+                        Sorry, your browser doesn't support embedded videos.
+                    </video>
+                </div>
+                <div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
+                    <h1 style="height:15%; width:90%; padding: 5%">Encoded</h1>
+                    <video style="width: 100%" src="upload/` + response["mp4_2"] + `" controls autoplay>
+                        Sorry, your browser doesn't support embedded videos.
+                    </video>
+                </div>
+                </section>`
+                $("body").append(html_content)
+            }
             else if(response.hasOwnProperty("txt")){
                 
                 fetch("upload/" + response["txt"])