devoalda
/
CSF_ACW_1
mirror of https://github.com/Devoalda/CSF_ACW_1.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Txt integration Done

This commit is contained in:
cleontay 2023-05-30 20:07:35 +08:00
parent c438350b1b
commit 3195c2c689
3 changed files with 57 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Application/lsb_rep.py
View File

@ -1,5 +1,5 @@
from flask import Flask, render_template, request, redirect, session, send_from_directory
from lib.steganography import img_steg, wav_steg
from lib.steganography import img_steg, wav_steg, txt_steg
import cv2
import os
import sys
@ -53,6 +53,12 @@ def encoding():
new_wav_file.close()
session['wav'] = file.filename
session['wav2'] = "encoded_" + file.filename
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
return redirect("/encode_result")
@ -84,8 +90,12 @@ def decoding():
payload = wav_steg.wav_steg(WORKING_PATH + file.filename, b2c).decode()
session["payload"] = payload
session["wav"] = file.filename
elif file_extension == ".txt":
payload = txt_steg.txt_steg(WORKING_PATH + file.filename, b2c).decode()
session["payload"] = payload
session["txt"] = file.filename
return redirect("/decode_result")
return redirect("/decode_result")
@app.route('/decode_result')
def decode_result():

19
Application/views/decode_result.html
View File

@ -43,7 +43,24 @@
$("body").append(html_content)
}
else if(response.hasOwnProperty("txt")){
console.log(response)
fetch("upload/" + response["txt"])
.then(res => res.text())
.then(data => {
var html_content = `<section>
<div>
<h1>Decoded</h1>
<textarea>` + data + `</textarea>
</div>
<div>
<h1>Secret Text:</h1>
<p>` + response["payload"] + `</p>
</div>
</section>`
$("body").append(html_content)
})
.catch(error => {
console.error('Error:', error);
});
}
});
})

28
Application/views/encode_result.html
View File

@ -49,7 +49,33 @@
$("body").append(html_content)
}
else if(response.hasOwnProperty("txt")){
console.log(response)
fetch("upload/" + response["txt"])
.then(res => res.text())
.then(data => {
fetch("upload/" + response["txt2"])
.then(res2 => res2.text())
.then(data2 => {
var html_content = `<section style="display: flex; justify-content: flex-start;">
<div>
<h1>Original</h1>
<textarea>` + data + `</textarea>
</div>
<div>
<h1>Encoded</h1>
<textarea>` + data2 + `</textarea>
</div>
</section>`
$("body").append(html_content)
})
.catch(error2 => {
console.error('Error:', error2);
});
})
.catch(error => {
console.error('Error:', error);
});
}
});
})