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

Management of unicode. It would be perfect to manage ASCII and unicode in the same time (so 8 bits to 32 bits caracters).

Browse source
This commit is contained in:
Cédric Bonhomme 2017-03-13 09:34:43 +01:00
parent e803386361
commit 6ad140bdfb
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
4 changed files with 12 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
stegano/lsb/lsb.py
View file

@ -106,9 +106,9 @@ def reveal(input_image_file):
if img.mode == 'RGBA': if img.mode == 'RGBA':
pixel = pixel[:3] # ignore the alpha pixel = pixel[:3] # ignore the alpha
for color in pixel: for color in pixel:
buff += (color&1)<<(7-count) buff += (color&1)<<(31-count)
count += 1 count += 1
if count == 8: if count == 32:
bitab.append(chr(buff)) bitab.append(chr(buff))
buff, count = 0, 0 buff, count = 0, 0
if bitab[-1] == ":" and limit == None: if bitab[-1] == ":" and limit == None:

4
stegano/lsbset/lsbset.py
View file

@ -104,9 +104,9 @@ def reveal(input_image_file, generator):
generated_number = next(generator) generated_number = next(generator)
# color = [r, g, b] # color = [r, g, b]
for color in img_list[generated_number]: for color in img_list[generated_number]:
buff += (color&1)<<(7-count) buff += (color&1)<<(31-count)
count += 1 count += 1
if count == 8: if count == 32:
bitab.append(chr(buff)) bitab.append(chr(buff))
buff, count = 0, 0 buff, count = 0, 0
if bitab[-1] == ":" and limit == None: if bitab[-1] == ":" and limit == None:

3
stegano/tools.py
View file

@ -57,7 +57,8 @@ def a2bits_list(chars: str) -> List[str]:
>>> "".join(a2bits_list("Hello World!")) >>> "".join(a2bits_list("Hello World!"))
'010010000110010101101100011011000110111100100000010101110110111101110010011011000110010000100001' '010010000110010101101100011011000110111100100000010101110110111101110010011011000110010000100001'
""" """
return [bin(ord(x))[2:].rjust(8,"0") for x in chars] #return [bin(ord(x))[2:].rjust(8,"0") for x in chars]
return [bin(ord(x))[2:].rjust(32,"0") for x in chars]
def bs(s: int) -> str: def bs(s: int) -> str:
"""Converts an int to its bits representation as a string of 0's and 1's. """Converts an int to its bits representation as a string of 0's and 1's.

12
tests/test_lsb.py
View file

@ -43,7 +43,7 @@ class TestLSB(unittest.TestCase):
secret = lsb.hide("./tests/sample-files/Lenna.png", "") secret = lsb.hide("./tests/sample-files/Lenna.png", "")
def test_hide_and_reveal(self): def test_hide_and_reveal(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"] messages_to_hide = ['🍕', 'a', 'foo', 'Hello World!', ':Python:']
for message in messages_to_hide: for message in messages_to_hide:
secret = lsb.hide("./tests/sample-files/Lenna.png", message) secret = lsb.hide("./tests/sample-files/Lenna.png", message)
secret.save("./image.png") secret.save("./image.png")
@ -53,7 +53,7 @@ class TestLSB(unittest.TestCase):
self.assertEqual(message, clear_message) self.assertEqual(message, clear_message)
def test_with_transparent_png(self): def test_with_transparent_png(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"] messages_to_hide = ['🍕', 'a', 'foo', 'Hello World!', ':Python:']
for message in messages_to_hide: for message in messages_to_hide:
secret = lsb.hide("./tests/sample-files/transparent.png", message) secret = lsb.hide("./tests/sample-files/transparent.png", message)
secret.save("./image.png") secret.save("./image.png")
@ -64,24 +64,24 @@ class TestLSB(unittest.TestCase):
@patch('builtins.input', return_value='y') @patch('builtins.input', return_value='y')
def test_manual_convert_rgb(self, input): def test_manual_convert_rgb(self, input):
message_to_hide = "I love 🍕 and 🍫!" message_to_hide = 'I love 🍕 and 🍫!'
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png", secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide) message_to_hide)
@patch('builtins.input', return_value='n') @patch('builtins.input', return_value='n')
def test_refuse_convert_rgb(self, input): def test_refuse_convert_rgb(self, input):
message_to_hide = "I love 🍕 and 🍫!" message_to_hide = 'I love 🍕 and 🍫!'
with self.assertRaises(Exception): with self.assertRaises(Exception):
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png", secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide) message_to_hide)
def test_auto_convert_rgb(self): def test_auto_convert_rgb(self):
message_to_hide = "I love 🍕 and 🍫!" message_to_hide = 'I love 🍕 and 🍫!'
secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png", secret = lsb.hide("./tests/sample-files/Lenna-grayscale.png",
message_to_hide, True) message_to_hide, True)
def test_with_text_file(self): def test_with_text_file(self):
text_file_to_hide = "./tests/sample-files/lorem_ipsum.txt" text_file_to_hide = './tests/sample-files/lorem_ipsum.txt'
with open(text_file_to_hide) as f: with open(text_file_to_hide) as f:
message = f.read() message = f.read()
secret = lsb.hide("./tests/sample-files/Lenna.png", message) secret = lsb.hide("./tests/sample-files/Lenna.png", message)