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

Added command for docx, xlsx, pptx

This commit is contained in:
cleontay 2023-05-30 22:44:04 +08:00
parent 7a0b96e805
commit f38d2c2138
3 changed files with 32 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
Application/lsb_rep.py
View File

@ -38,7 +38,7 @@ def encoding():
file.save(WORKING_PATH + file.filename)
file_extension = os.path.splitext(file.filename)[1][1:]
if file_extension == "png" or file_extension == "bmp":
if file_extension in ["bmp", "png"]:
encode = img_steg.img_steg(WORKING_PATH + file.filename, b2c).encode(payload)
cv2.imwrite(WORKING_PATH + "encoded_" + file.filename, encode)
session['image'] = file.filename
@ -61,10 +61,12 @@ def encoding():
f.write(encoded_data)
session['txt'] = file.filename
session['txt2'] = "encoded_" + file.filename
elif file_extension == "mp3" or file_extension == "mp4":
elif file_extension in ["mp3", "mp4", "docx", "xlsx", "pptx"]:
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)
if file_extension in ["docx", "xlsx", "pptx"]:
file_extension = "document"
session[file_extension] = file.filename
session[file_extension + '_2'] = "encoded_" + file.filename
@ -98,6 +100,8 @@ def decoding():
if class_name_str == "img_steg":
session["image"] = file.filename
elif ext in ["docx", "pptx", "xlsx"]:
session["document"] = file.filename
else:
session[ext] = file.filename
@ -110,19 +114,18 @@ def decoding():
file = request.files['encoded_file']
b2c = [int(x) for x in request.form.getlist("b2c")]
if file.filename != "":
_path = WORKING_PATH + file.filename
file.save(_path)
file_extension = os.path.splitext(file.filename)[1][1:]
if file_extension == "png" or file_extension == "bmp":
if file_extension in ["bmp", "png"]:
decode_files(img_steg, _path, b2c, file_extension)
elif file_extension == "wav":
decode_files(wav_steg, _path, b2c, file_extension)
elif file_extension == "txt":
decode_files(txt_steg, _path, b2c, file_extension)
elif file_extension == "mp3" or file_extension == "mp4":
elif file_extension in ["mp3", "mp4", "docx", "xlsx", "pptx"]:
decode_files(file_steg, _path, b2c, file_extension)
return redirect("/decode_result")

10
Application/views/decode_result.html
View File

@ -90,6 +90,16 @@
console.error('Error:', error);
});
}
else if(response.hasOwnProperty("document")){
var html_content = `<section style="margin: 100px auto; width:50%; height:50%; text-align:center;">
<div style="width:100%; height:100%; text-align:center;">
<a href="upload/` + response["document"] + `" download>Download Document</a>
<h1 style="height:15%; width:90%; padding: 5%">Secret Text:</h1>
<p style="color: red">` + response["payload"] + `</p>
</div>
</section>`
$("body").append(html_content)
}
});
})

14
Application/views/encode_result.html
View File

@ -96,6 +96,20 @@
</section>`
$("body").append(html_content)
}
else if(response.hasOwnProperty("docx")){
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>
<a href="upload/` + response["document"] + `" download>Download Document</a>
</div>
<div style="border: 1px solid black; width:100%; height:100%; text-align:center;">
<h1 style="height:15%; width:90%; padding: 5%">Encoded</h1>
<a href="upload/` + response["document_2"] + `" download>Download Document</a>
</div>
</section>`
$("body").append(html_content)
}
else if(response.hasOwnProperty("txt")){
fetch("upload/" + response["txt"])