chg: [style] Reformated with black.

This commit is contained in:
Cédric Bonhomme 2021-11-24 09:16:05 +01:00
parent 891051f77b
commit a1213a9163
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
13 changed files with 275 additions and 184 deletions

View file

@ -34,70 +34,104 @@ except:
from stegano import tools from stegano import tools
def main(): def main():
parser = argparse.ArgumentParser(prog='stegano-lsb') parser = argparse.ArgumentParser(prog="stegano-lsb")
subparsers = parser.add_subparsers(help='sub-command help', dest='command') subparsers = parser.add_subparsers(help="sub-command help", dest="command")
# Subparser: Hide # Subparser: Hide
parser_hide = subparsers.add_parser('hide', help='hide help') parser_hide = subparsers.add_parser("hide", help="hide help")
# Original image # Original image
parser_hide.add_argument("-i", "--input", dest="input_image_file", parser_hide.add_argument(
required=True, help="Input image file.") "-i",
parser_hide.add_argument("-e", "--encoding", dest="encoding", "--input",
choices=tools.ENCODINGS.keys(), default='UTF-8', dest="input_image_file",
help="Specify the encoding of the message to hide." + required=True,
" UTF-8 (default) or UTF-32LE.") help="Input image file.",
)
parser_hide.add_argument(
"-e",
"--encoding",
dest="encoding",
choices=tools.ENCODINGS.keys(),
default="UTF-8",
help="Specify the encoding of the message to hide."
+ " UTF-8 (default) or UTF-32LE.",
)
group_secret = parser_hide.add_mutually_exclusive_group(required=True) group_secret = parser_hide.add_mutually_exclusive_group(required=True)
# Non binary secret message to hide # Non binary secret message to hide
group_secret.add_argument("-m", dest="secret_message", group_secret.add_argument(
help="Your secret message to hide (non binary).") "-m", dest="secret_message", help="Your secret message to hide (non binary)."
)
# Binary secret message to hide # Binary secret message to hide
group_secret.add_argument("-f", dest="secret_file", group_secret.add_argument(
help="Your secret to hide (Text or any binary file).") "-f", dest="secret_file", help="Your secret to hide (Text or any binary file)."
)
# Image containing the secret # Image containing the secret
parser_hide.add_argument("-o", "--output", dest="output_image_file", parser_hide.add_argument(
required=True, help="Output image containing the secret.") "-o",
"--output",
dest="output_image_file",
required=True,
help="Output image containing the secret.",
)
# Shift the message to hide # Shift the message to hide
parser_hide.add_argument("-s", "--shift", dest="shift", default=0, parser_hide.add_argument(
help="Shift for the message to hide") "-s", "--shift", dest="shift", default=0, help="Shift for the message to hide"
)
# Subparser: Reveal # Subparser: Reveal
parser_reveal = subparsers.add_parser('reveal', help='reveal help') parser_reveal = subparsers.add_parser("reveal", help="reveal help")
parser_reveal.add_argument("-i", "--input", dest="input_image_file", parser_reveal.add_argument(
required=True, help="Input image file.") "-i",
parser_reveal.add_argument("-e", "--encoding", dest="encoding", "--input",
choices=tools.ENCODINGS.keys(), default='UTF-8', dest="input_image_file",
help="Specify the encoding of the message to reveal." + required=True,
" UTF-8 (default) or UTF-32LE.") help="Input image file.",
parser_reveal.add_argument("-o", dest="secret_binary", )
help="Output for the binary secret (Text or any binary file).") parser_reveal.add_argument(
"-e",
"--encoding",
dest="encoding",
choices=tools.ENCODINGS.keys(),
default="UTF-8",
help="Specify the encoding of the message to reveal."
+ " UTF-8 (default) or UTF-32LE.",
)
parser_reveal.add_argument(
"-o",
dest="secret_binary",
help="Output for the binary secret (Text or any binary file).",
)
# Shift the message to reveal # Shift the message to reveal
parser_reveal.add_argument("-s", "--shift", dest="shift", default=0, parser_reveal.add_argument(
help="Shift for the reveal") "-s", "--shift", dest="shift", default=0, help="Shift for the reveal"
)
arguments = parser.parse_args() arguments = parser.parse_args()
if arguments.command == "hide":
if arguments.command == 'hide':
if arguments.secret_message != None: if arguments.secret_message != None:
secret = arguments.secret_message secret = arguments.secret_message
elif arguments.secret_file != None: elif arguments.secret_file != None:
secret = tools.binary2base64(arguments.secret_file) secret = tools.binary2base64(arguments.secret_file)
img_encoded = lsb.hide(arguments.input_image_file, secret, img_encoded = lsb.hide(
arguments.encoding, int(arguments.shift)) arguments.input_image_file, secret, arguments.encoding, int(arguments.shift)
)
try: try:
img_encoded.save(arguments.output_image_file) img_encoded.save(arguments.output_image_file)
except Exception as e: except Exception as e:
# If hide() returns an error (Too long message). # If hide() returns an error (Too long message).
print(e) print(e)
elif arguments.command == 'reveal': elif arguments.command == "reveal":
secret = lsb.reveal(arguments.input_image_file, arguments.encoding, secret = lsb.reveal(
int(arguments.shift)) arguments.input_image_file, arguments.encoding, int(arguments.shift)
)
if arguments.secret_binary != None: if arguments.secret_binary != None:
data = tools.base642binary(secret) data = tools.base642binary(secret)
with open(arguments.secret_binary, "wb") as f: with open(arguments.secret_binary, "wb") as f:

View file

@ -39,11 +39,12 @@ from stegano import tools
import argparse import argparse
class ValidateGenerator(argparse.Action): class ValidateGenerator(argparse.Action):
def __call__(self, parser, args, values, option_string=None): def __call__(self, parser, args, values, option_string=None):
valid_generators = [generator[0] for generator in inspect.getmembers( valid_generators = [
generators, inspect.isfunction)] generator[0]
for generator in inspect.getmembers(generators, inspect.isfunction)
]
# Verify that the generator is valid # Verify that the generator is valid
generator = values[0] generator = values[0]
if generator not in valid_generators: if generator not in valid_generators:
@ -51,81 +52,126 @@ class ValidateGenerator(argparse.Action):
# Set the generator_function arg of the parser # Set the generator_function arg of the parser
setattr(args, self.dest, values) setattr(args, self.dest, values)
def main(): def main():
parser = argparse.ArgumentParser(prog='stegano-lsb-set') parser = argparse.ArgumentParser(prog="stegano-lsb-set")
subparsers = parser.add_subparsers(help='sub-command help', dest='command') subparsers = parser.add_subparsers(help="sub-command help", dest="command")
# Subparser: Hide # Subparser: Hide
parser_hide = subparsers.add_parser('hide', help='hide help') parser_hide = subparsers.add_parser("hide", help="hide help")
# Original image # Original image
parser_hide.add_argument("-i", "--input", dest="input_image_file", parser_hide.add_argument(
required=True, help="Input image file.") "-i",
parser_hide.add_argument("-e", "--encoding", dest="encoding", "--input",
choices=tools.ENCODINGS.keys(), default='UTF-8', dest="input_image_file",
help="Specify the encoding of the message to hide." + required=True,
" UTF-8 (default) or UTF-32LE.") help="Input image file.",
)
parser_hide.add_argument(
"-e",
"--encoding",
dest="encoding",
choices=tools.ENCODINGS.keys(),
default="UTF-8",
help="Specify the encoding of the message to hide."
+ " UTF-8 (default) or UTF-32LE.",
)
# Generator # Generator
parser_hide.add_argument("-g", "--generator", dest="generator_function", parser_hide.add_argument(
"-g",
"--generator",
dest="generator_function",
action=ValidateGenerator, action=ValidateGenerator,
nargs='*', required=True, help="Generator (with optional arguments)") nargs="*",
parser_hide.add_argument("-s", "--shift", dest="shift", required=True,
default=0, help="Shift for the generator") help="Generator (with optional arguments)",
)
parser_hide.add_argument(
"-s", "--shift", dest="shift", default=0, help="Shift for the generator"
)
group_secret = parser_hide.add_mutually_exclusive_group(required=True) group_secret = parser_hide.add_mutually_exclusive_group(required=True)
# Non binary secret message to hide # Non binary secret message to hide
group_secret.add_argument("-m", dest="secret_message", group_secret.add_argument(
help="Your secret message to hide (non binary).") "-m", dest="secret_message", help="Your secret message to hide (non binary)."
)
# Binary secret message to hide # Binary secret message to hide
group_secret.add_argument("-f", dest="secret_file", group_secret.add_argument(
help="Your secret to hide (Text or any binary file).") "-f", dest="secret_file", help="Your secret to hide (Text or any binary file)."
)
# Image containing the secret # Image containing the secret
parser_hide.add_argument("-o", "--output", dest="output_image_file", parser_hide.add_argument(
required=True, help="Output image containing the secret.") "-o",
"--output",
dest="output_image_file",
required=True,
help="Output image containing the secret.",
)
# Subparser: Reveal # Subparser: Reveal
parser_reveal = subparsers.add_parser('reveal', help='reveal help') parser_reveal = subparsers.add_parser("reveal", help="reveal help")
parser_reveal.add_argument("-i", "--input", dest="input_image_file", parser_reveal.add_argument(
required=True, help="Input image file.") "-i",
parser_reveal.add_argument("-e", "--encoding", dest="encoding", "--input",
choices=tools.ENCODINGS.keys(), default='UTF-8', dest="input_image_file",
help="Specify the encoding of the message to reveal." + required=True,
" UTF-8 (default) or UTF-32LE.") help="Input image file.",
parser_reveal.add_argument("-g", "--generator", dest="generator_function", )
parser_reveal.add_argument(
"-e",
"--encoding",
dest="encoding",
choices=tools.ENCODINGS.keys(),
default="UTF-8",
help="Specify the encoding of the message to reveal."
+ " UTF-8 (default) or UTF-32LE.",
)
parser_reveal.add_argument(
"-g",
"--generator",
dest="generator_function",
action=ValidateGenerator, action=ValidateGenerator,
nargs='*', required=True, help="Generator (with optional arguments)") nargs="*",
parser_reveal.add_argument("-s", "--shift", dest="shift", required=True,
default=0, help="Shift for the generator") help="Generator (with optional arguments)",
parser_reveal.add_argument("-o", dest="secret_binary", )
help="Output for the binary secret (Text or any binary file).") parser_reveal.add_argument(
"-s", "--shift", dest="shift", default=0, help="Shift for the generator"
)
parser_reveal.add_argument(
"-o",
dest="secret_binary",
help="Output for the binary secret (Text or any binary file).",
)
# Subparser: List generators # Subparser: List generators
parser_list_generators = subparsers.add_parser('list-generators', parser_list_generators = subparsers.add_parser(
help='list-generators help') "list-generators", help="list-generators help"
)
arguments = parser.parse_args() arguments = parser.parse_args()
if arguments.command != 'list-generators': if arguments.command != "list-generators":
try: try:
arguments.generator_function[0] arguments.generator_function[0]
except AttributeError: except AttributeError:
print('You must specify the name of a generator.') print("You must specify the name of a generator.")
parser.print_help() parser.print_help()
exit(1) exit(1)
try: try:
if (arguments.generator_function[0] == "LFSR"): if arguments.generator_function[0] == "LFSR":
# Compute the size of the image for use by the LFSR generator if needed # Compute the size of the image for use by the LFSR generator if needed
tmp = tools.open_image(arguments.input_image_file) tmp = tools.open_image(arguments.input_image_file)
size = tmp.width * tmp.height size = tmp.width * tmp.height
tmp.close() tmp.close()
arguments.generator_function.append(size) arguments.generator_function.append(size)
if (len(arguments.generator_function) > 1): if len(arguments.generator_function) > 1:
generator = getattr(generators, arguments.generator_function[0])( generator = getattr(generators, arguments.generator_function[0])(
*[int(e) for e in arguments.generator_function[1:]]) *[int(e) for e in arguments.generator_function[1:]]
)
else: else:
generator = getattr(generators, arguments.generator_function[0])() generator = getattr(generators, arguments.generator_function[0])()
@ -133,24 +179,26 @@ def main():
print("Unknown generator: {}".format(arguments.generator_function)) print("Unknown generator: {}".format(arguments.generator_function))
exit(1) exit(1)
if arguments.command == 'hide': if arguments.command == "hide":
if arguments.secret_message != None: if arguments.secret_message != None:
secret = arguments.secret_message secret = arguments.secret_message
elif arguments.secret_file != "": elif arguments.secret_file != "":
secret = tools.binary2base64(arguments.secret_file) secret = tools.binary2base64(arguments.secret_file)
img_encoded = lsbset.hide(arguments.input_image_file, secret, generator, img_encoded = lsbset.hide(
int(arguments.shift)) arguments.input_image_file, secret, generator, int(arguments.shift)
)
try: try:
img_encoded.save(arguments.output_image_file) img_encoded.save(arguments.output_image_file)
except Exception as e: except Exception as e:
# If hide() returns an error (Too long message). # If hide() returns an error (Too long message).
print(e) print(e)
elif arguments.command == 'reveal': elif arguments.command == "reveal":
try: try:
secret = lsbset.reveal(arguments.input_image_file, generator, secret = lsbset.reveal(
int(arguments.shift)) arguments.input_image_file, generator, int(arguments.shift)
)
except IndexError: except IndexError:
print("Impossible to detect message.") print("Impossible to detect message.")
exit(0) exit(0)
@ -161,10 +209,10 @@ def main():
else: else:
print(secret) print(secret)
elif arguments.command == 'list-generators': elif arguments.command == "list-generators":
all_generators = inspect.getmembers(generators, inspect.isfunction) all_generators = inspect.getmembers(generators, inspect.isfunction)
for generator in all_generators: for generator in all_generators:
print('Generator id:') print("Generator id:")
print(' {}'.format(crayons.green(generator[0], bold=True))) print(" {}".format(crayons.green(generator[0], bold=True)))
print('Desciption:') print("Desciption:")
print(' {}'.format(generator[1].__doc__)) print(" {}".format(generator[1].__doc__))

View file

@ -33,12 +33,23 @@ try:
except: except:
print("Install Stegano: pipx install Stegano") print("Install Stegano: pipx install Stegano")
def main(): def main():
parser = argparse.ArgumentParser(prog='stegano-steganalysis-parity') parser = argparse.ArgumentParser(prog="stegano-steganalysis-parity")
parser.add_argument("-i", "--input", dest="input_image_file", parser.add_argument(
required=True, help="Input image file.") "-i",
parser.add_argument("-o", "--output", dest="output_image_file", "--input",
required=True, help="Output image file.") dest="input_image_file",
required=True,
help="Input image file.",
)
parser.add_argument(
"-o",
"--output",
dest="output_image_file",
required=True,
help="Output image file.",
)
arguments = parser.parse_args() arguments = parser.parse_args()
input_image_file = Image.open(arguments.input_image_file) input_image_file = Image.open(arguments.input_image_file)

View file

@ -33,27 +33,31 @@ except:
def main(): def main():
parser = argparse.ArgumentParser(prog='stegano-red') parser = argparse.ArgumentParser(prog="stegano-red")
subparsers = parser.add_subparsers(help='sub-command help', dest='command') subparsers = parser.add_subparsers(help="sub-command help", dest="command")
parser_hide = subparsers.add_parser('hide', help='hide help') parser_hide = subparsers.add_parser("hide", help="hide help")
parser_hide.add_argument("-i", "--input", dest="input_image_file", parser_hide.add_argument(
help="Image file") "-i", "--input", dest="input_image_file", help="Image file"
parser_hide.add_argument("-m", dest="secret_message", )
help="Your secret message to hide (non binary).") parser_hide.add_argument(
parser_hide.add_argument("-o", "--output", dest="output_image_file", "-m", dest="secret_message", help="Your secret message to hide (non binary)."
help="Image file") )
parser_hide.add_argument(
"-o", "--output", dest="output_image_file", help="Image file"
)
parser_reveal = subparsers.add_parser('reveal', help='reveal help') parser_reveal = subparsers.add_parser("reveal", help="reveal help")
parser_reveal.add_argument("-i", "--input", dest="input_image_file", parser_reveal.add_argument(
help="Image file") "-i", "--input", dest="input_image_file", help="Image file"
)
arguments = parser.parse_args() arguments = parser.parse_args()
if arguments.command == 'hide': if arguments.command == "hide":
secret = red.hide(arguments.input_image_file, arguments.secret_message) secret = red.hide(arguments.input_image_file, arguments.secret_message)
secret.save(arguments.output_image_file) secret.save(arguments.output_image_file)
elif arguments.command == 'reveal': elif arguments.command == "reveal":
secret = red.reveal(arguments.input_image_file) secret = red.reveal(arguments.input_image_file)
print(secret) print(secret)

View file

@ -36,11 +36,9 @@ except:
def main(): def main():
parser = argparse.ArgumentParser(prog='stegano-steganalysis-parity') parser = argparse.ArgumentParser(prog="stegano-steganalysis-parity")
parser.add_argument("-i", "--input", dest="input_image_file", parser.add_argument("-i", "--input", dest="input_image_file", help="Image file")
help="Image file") parser.add_argument("-o", "--output", dest="output_image_file", help="Image file")
parser.add_argument("-o", "--output", dest="output_image_file",
help="Image file")
arguments = parser.parse_args() arguments = parser.parse_args()
input_image_file = Image.open(arguments.input_image_file) input_image_file = Image.open(arguments.input_image_file)

View file

@ -28,30 +28,30 @@ import sys, os
extensions = [] extensions = []
# Add any paths that contain templates here, relative to this directory. # Add any paths that contain templates here, relative to this directory.
templates_path = ['_templates'] templates_path = ["_templates"]
# The suffix of source filenames. # The suffix of source filenames.
source_suffix = '.rst' source_suffix = ".rst"
# The encoding of source files. # The encoding of source files.
# source_encoding = 'utf-8-sig' # source_encoding = 'utf-8-sig'
# The master toctree document. # The master toctree document.
master_doc = 'index' master_doc = "index"
# General information about the project. # General information about the project.
project = u'Stegano' project = u"Stegano"
copyright = u'2010-2021, Cédric Bonhomme' copyright = u"2010-2021, Cédric Bonhomme"
author = 'Cédric Bonhomme <cedric@cedricbonhomme.org>' author = "Cédric Bonhomme <cedric@cedricbonhomme.org>"
# The version info for the project you're documenting, acts as replacement for # The version info for the project you're documenting, acts as replacement for
# |version| and |release|, also used in various other places throughout the # |version| and |release|, also used in various other places throughout the
# built documents. # built documents.
# #
# The short X.Y version. # The short X.Y version.
version = '0.9' version = "0.9"
# The full version, including alpha/beta/rc tags. # The full version, including alpha/beta/rc tags.
release = '0.9.4' release = "0.9.4"
# The language for content autogenerated by Sphinx. Refer to documentation # The language for content autogenerated by Sphinx. Refer to documentation
# for a list of supported languages. # for a list of supported languages.
@ -65,7 +65,7 @@ release = '0.9.4'
# List of patterns, relative to source directory, that match files and # List of patterns, relative to source directory, that match files and
# directories to ignore when looking for source files. # directories to ignore when looking for source files.
exclude_patterns = ['_build'] exclude_patterns = ["_build"]
# The reST default role (used for this markup: `text`) to use for all documents. # The reST default role (used for this markup: `text`) to use for all documents.
# default_role = None # default_role = None
@ -89,7 +89,7 @@ exclude_patterns = ['_build']
# The theme to use for HTML and HTML Help pages. See the documentation for # The theme to use for HTML and HTML Help pages. See the documentation for
# a list of builtin themes. # a list of builtin themes.
html_theme = 'sphinx_rtd_theme' html_theme = "sphinx_rtd_theme"
# Theme options are theme-specific and customize the look and feel of a theme # Theme options are theme-specific and customize the look and feel of a theme
# further. For a list of options available for each theme, see the # further. For a list of options available for each theme, see the
@ -118,7 +118,7 @@ html_theme = 'sphinx_rtd_theme'
# Add any paths that contain custom static files (such as style sheets) here, # Add any paths that contain custom static files (such as style sheets) here,
# relative to this directory. They are copied after the builtin static files, # relative to this directory. They are copied after the builtin static files,
# so a file named "default.css" will overwrite the builtin "default.css". # so a file named "default.css" will overwrite the builtin "default.css".
html_static_path = ['_static'] html_static_path = ["_static"]
# If not '', a 'Last updated on:' timestamp is inserted at every page bottom, # If not '', a 'Last updated on:' timestamp is inserted at every page bottom,
# using the given strftime format. # using the given strftime format.
@ -161,13 +161,12 @@ html_static_path = ['_static']
# -- Options for LaTeX output -------------------------------------------------- # -- Options for LaTeX output --------------------------------------------------
latex_engine = 'pdflatex' latex_engine = "pdflatex"
# Grouping the document tree into LaTeX files. List of tuples # Grouping the document tree into LaTeX files. List of tuples
# (source start file, target name, title, author, documentclass [howto/manual]). # (source start file, target name, title, author, documentclass [howto/manual]).
latex_documents = [ latex_documents = [
('index', 'Stgan.tex', u'Stegano Documentation', ("index", "Stgan.tex", u"Stegano Documentation", u"Cédric Bonhomme", "howto"),
u'Cédric Bonhomme', 'howto'),
] ]
latex_show_urls = True latex_show_urls = True

View file

@ -30,10 +30,13 @@ from stegano import tools
def hide( def hide(
input_image_file, img_enc, secret_message=None, secret_file=None, img_format=None, input_image_file,
img_enc,
secret_message=None,
secret_file=None,
img_format=None,
): ):
"""Hide a message (string) in an image. """Hide a message (string) in an image."""
"""
from zlib import compress from zlib import compress
from base64 import b64encode from base64 import b64encode
@ -64,8 +67,7 @@ def hide(
def reveal(input_image_file): def reveal(input_image_file):
"""Find a message in an image. """Find a message in an image."""
"""
from base64 import b64decode from base64 import b64decode
from zlib import decompress from zlib import decompress

View file

@ -98,8 +98,7 @@ def hide(
def reveal(input_image: Union[str, IO[bytes]], encoding: str = "UTF-8", shift: int = 0): def reveal(input_image: Union[str, IO[bytes]], encoding: str = "UTF-8", shift: int = 0):
"""Find a message in an image (with the LSB technique). """Find a message in an image (with the LSB technique)."""
"""
img = tools.open_image(input_image) img = tools.open_image(input_image)
width, height = img.size width, height = img.size
buff, count = 0, 0 buff, count = 0, 0

View file

@ -106,8 +106,7 @@ def reveal(
shift: int = 0, shift: int = 0,
encoding: str = "UTF-8", encoding: str = "UTF-8",
): ):
"""Find a message in an image (with the LSB technique). """Find a message in an image (with the LSB technique)."""
"""
img = tools.open_image(input_image) img = tools.open_image(input_image)
img_list = list(img.getdata()) img_list = list(img.getdata())
width, height = img.size width, height = img.size

View file

@ -67,14 +67,12 @@ def a2bits_list(chars: str, encoding: str = "UTF-8") -> List[str]:
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."""
"""
return str(s) if s <= 1 else bs(s >> 1) + str(s & 1) return str(s) if s <= 1 else bs(s >> 1) + str(s & 1)
def setlsb(component: int, bit: str) -> int: def setlsb(component: int, bit: str) -> int:
"""Set Least Significant Bit of a colour component. """Set Least Significant Bit of a colour component."""
"""
return component & ~1 | int(bit) return component & ~1 | int(bit)
@ -102,8 +100,7 @@ def binary2base64(binary_file: str) -> str:
def base642binary(b64_fname: str) -> bytes: def base642binary(b64_fname: str) -> bytes:
"""Convert a printable string to a binary file. """Convert a printable string to a binary file."""
"""
b64_fname += "===" b64_fname += "==="
return base64.b64decode(b64_fname) return base64.b64decode(b64_fname)

View file

@ -34,8 +34,7 @@ from stegano import exifHeader
class TestEXIFHeader(unittest.TestCase): class TestEXIFHeader(unittest.TestCase):
def test_hide_empty_message(self): def test_hide_empty_message(self):
"""Test hiding the empty string. """Test hiding the empty string."""
"""
secret = exifHeader.hide( secret = exifHeader.hide(
"./tests/sample-files/20160505T130442.jpg", "./image.jpg", secret_message="" "./tests/sample-files/20160505T130442.jpg", "./image.jpg", secret_message=""
) )

View file

@ -72,11 +72,15 @@ class TestLSBSet(unittest.TestCase):
messages_to_hide = ["foo bar"] messages_to_hide = ["foo bar"]
for message in messages_to_hide: for message in messages_to_hide:
secret = lsbset.hide( secret = lsbset.hide(
"./tests/sample-files/Lenna.png", message, generators.shi_tomashi("./tests/sample-files/Lenna.png") "./tests/sample-files/Lenna.png",
message,
generators.shi_tomashi("./tests/sample-files/Lenna.png"),
) )
secret.save("./image.png") secret.save("./image.png")
clear_message = lsbset.reveal("./image.png", generators.shi_tomashi("./tests/sample-files/Lenna.png")) clear_message = lsbset.reveal(
"./image.png", generators.shi_tomashi("./tests/sample-files/Lenna.png")
)
self.assertEqual(message, clear_message) self.assertEqual(message, clear_message)

View file

@ -38,8 +38,7 @@ from PIL import Image, ImageChops
class TestSteganalysis(unittest.TestCase): class TestSteganalysis(unittest.TestCase):
def test_parity(self): def test_parity(self):
"""Test stegano.steganalysis.parity """Test stegano.steganalysis.parity"""
"""
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()
@ -50,8 +49,7 @@ class TestSteganalysis(unittest.TestCase):
self.assertTrue(diff is None) self.assertTrue(diff is None)
def test_parity_rgba(self): def test_parity_rgba(self):
""" Test that stegano.steganalysis.parity works with RGBA images """Test that stegano.steganalysis.parity works with RGBA images"""
"""
img = Image.open("./tests/sample-files/transparent.png") img = Image.open("./tests/sample-files/transparent.png")
analysis = parity.steganalyse(img) analysis = parity.steganalyse(img)
target = Image.open("./tests/expected-results/parity_rgba.png") target = Image.open("./tests/expected-results/parity_rgba.png")
@ -59,8 +57,7 @@ class TestSteganalysis(unittest.TestCase):
self.assertTrue(diff is None) self.assertTrue(diff is None)
def test_statistics(self): def test_statistics(self):
""" Test stegano.steganalysis.statistics """Test stegano.steganalysis.statistics"""
"""
image = Image.open("./tests/sample-files/Lenna.png") image = Image.open("./tests/sample-files/Lenna.png")
stats = str(statistics.steganalyse(image)) + "\n" stats = str(statistics.steganalyse(image)) + "\n"
file = open("./tests/expected-results/statistics") file = open("./tests/expected-results/statistics")