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

Added support for transparent PNGs in the lsbset module.

Browse source
This commit is contained in:
Cédric Bonhomme 2017-01-27 23:47:34 +01:00
parent 556a24341e
commit 3e31b38294
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
2 changed files with 22 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
stegano/lsbset/lsbset.py
View file

@ -46,13 +46,14 @@ def hide(input_image_file, message, generator, auto_convert_rgb=False):
assert message_length != 0, "message length is zero" assert message_length != 0, "message length is zero"
img = Image.open(input_image_file) img = Image.open(input_image_file)
if img.mode != 'RGB':
print('The mode of the image is not RGB. Mode is {}'.format(img.mode)) if img.mode not in ['RGB', 'RGBA']:
if not auto_convert_rgb: if not auto_convert_rgb:
print('The mode of the image is not RGB. Mode is {}'.\
format(img.mode))
answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y' answer = input('Convert the image to RGB ? [Y / n]\n') or 'Y'
if answer.lower() == 'n': if answer.lower() == 'n':
raise Exception('Not a RGB image.') raise Exception('Not a RGB image.')
img = img.convert('RGB') img = img.convert('RGB')
img_list = list(img.getdata()) img_list = list(img.getdata())
@ -72,7 +73,7 @@ def hide(input_image_file, message, generator, auto_convert_rgb=False):
while index + 3 <= len_message_bits : while index + 3 <= len_message_bits :
generated_number = next(generator) generated_number = next(generator)
(r, g, b) = img_list[generated_number] r, g, b, *a = img_list[generated_number]
# Change the Least Significant Bit of each colour component. # Change the Least Significant Bit of each colour component.
r = tools.setlsb(r, message_bits[index]) r = tools.setlsb(r, message_bits[index])
@ -80,7 +81,10 @@ def hide(input_image_file, message, generator, auto_convert_rgb=False):
b = tools.setlsb(b, message_bits[index+2]) b = tools.setlsb(b, message_bits[index+2])
# Save the new pixel # Save the new pixel
img_list[generated_number] = (r, g , b) if img.mode == 'RGBA':
img_list[generated_number] = (r, g , b, a[0])
else:
img_list[generated_number] = (r, g , b)
index += 3 index += 3

13
tests/test_lsbset.py
View file

@ -53,6 +53,19 @@ class TestLSBSet(unittest.TestCase):
self.assertEqual(message, clear_message) self.assertEqual(message, clear_message)
def test_with_transparent_png(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = lsbset.hide("./tests/sample-files/transparent.png",
message, generators.eratosthenes())
secret.save("./image.png")
clear_message = lsbset.reveal("./image.png",
generators.eratosthenes())
self.assertEqual(message, clear_message)
def test_with_too_long_message(self): def test_with_too_long_message(self):
with open("./tests/sample-files/lorem_ipsum.txt") as f: with open("./tests/sample-files/lorem_ipsum.txt") as f:
message = f.read() message = f.read()