fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 17:18:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Various improvements in lsb and lsbset modules.

Browse source
This commit is contained in:
Cédric Bonhomme 2017-02-28 07:52:44 +01:00
parent 119aecf3a6
commit b9fad5f044
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
2 changed files with 5 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
stegano/lsb/lsb.py
View file

@ -89,12 +89,9 @@ def hide(input_image_file, message, auto_convert_rgb=False):
img.close()
return encoded
img.close()
return encoded
def reveal(input_image_file):
"""Find a message in an image
(with the LSB technique).
"""Find a message in an image (with the LSB technique).
"""
img = Image.open(input_image_file)
width, height = img.size
@ -123,5 +120,3 @@ def reveal(input_image_file):
if len(bitab)-len(str(limit))-1 == limit :
img.close()
return "".join(bitab)[len(str(limit))+1:]
img.close()
return ""

12
stegano/lsbset/lsbset.py
View file

@ -54,16 +54,15 @@ def hide(input_image_file, message, generator, auto_convert_rgb=False):
width, height = img.size
index = 0
message = str(len(message)) + ":" + str(message)
#message_bits = tools.a2bits(message)
message = str(message_length) + ":" + str(message)
message_bits = "".join(tools.a2bits_list(message))
message_bits += '0' * ((3 - (len(message_bits) % 3)) % 3)
npixels = width * height
len_message_bits = len(message_bits)
if len_message_bits > npixels * 3:
raise Exception("The message you want to hide is too long: {}".\
format(message_length))
raise Exception("The message you want to hide is too long: {}". \
format(message_length))
while index + 3 <= len_message_bits :
generated_number = next(generator)
@ -92,8 +91,7 @@ def hide(input_image_file, message, generator, auto_convert_rgb=False):
def reveal(input_image_file, generator):
"""Find a message in an image
(with the LSB technique).
"""Find a message in an image (with the LSB technique).
"""
img = Image.open(input_image_file)
img_list = list(img.getdata())
@ -118,5 +116,3 @@ def reveal(input_image_file, generator):
pass
if len(bitab)-len(str(limit))-1 == limit :
return "".join(bitab)[len(str(limit))+1:]
return ""