fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-14 01:38:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

chg: [style] Reformated with black.

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

104
bin/lsb.py
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:

162
bin/lsbset.py
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(
action=ValidateGenerator, "-g",
nargs='*', required=True, help="Generator (with optional arguments)") "--generator",
parser_hide.add_argument("-s", "--shift", dest="shift", dest="generator_function",
default=0, help="Shift for the generator") action=ValidateGenerator,
nargs="*",
required=True,
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", )
action=ValidateGenerator, parser_reveal.add_argument(
nargs='*', required=True, help="Generator (with optional arguments)") "-e",
parser_reveal.add_argument("-s", "--shift", dest="shift", "--encoding",
default=0, help="Shift for the generator") dest="encoding",
parser_reveal.add_argument("-o", dest="secret_binary", choices=tools.ENCODINGS.keys(),
help="Output for the binary secret (Text or any binary file).") 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,
nargs="*",
required=True,
help="Generator (with optional arguments)",
)
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__))

21
bin/parity.py
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)

32
bin/red.py
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)

8
bin/statistics.py
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)

83
docs/conf.py
View file

@ -16,158 +16,157 @@ import sys, os
# If extensions (or modules to document with autodoc) are in another directory, # If extensions (or modules to document with autodoc) are in another directory,
# add these directories to sys.path here. If the directory is relative to the # add these directories to sys.path here. If the directory is relative to the
# documentation root, use os.path.abspath to make it absolute, like shown here. # documentation root, use os.path.abspath to make it absolute, like shown here.
#sys.path.insert(0, os.path.abspath('.')) # sys.path.insert(0, os.path.abspath('.'))
# -- General configuration ----------------------------------------------------- # -- General configuration -----------------------------------------------------
# If your documentation needs a minimal Sphinx version, state it here. # If your documentation needs a minimal Sphinx version, state it here.
#needs_sphinx = '1.0' # needs_sphinx = '1.0'
# Add any Sphinx extension module names here, as strings. They can be extensions # Add any Sphinx extension module names here, as strings. They can be extensions
# coming with Sphinx (named 'sphinx.ext.*') or your custom ones. # coming with Sphinx (named 'sphinx.ext.*') or your custom ones.
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.
#language = None # language = None
# There are two options for replacing |today|: either, you set today to some # There are two options for replacing |today|: either, you set today to some
# non-false value, then it is used: # non-false value, then it is used:
#today = '' # today = ''
# Else, today_fmt is used as the format for a strftime call. # Else, today_fmt is used as the format for a strftime call.
#today_fmt = '%B %d, %Y' # today_fmt = '%B %d, %Y'
# 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
# If true, '()' will be appended to :func: etc. cross-reference text. # If true, '()' will be appended to :func: etc. cross-reference text.
#add_function_parentheses = True # add_function_parentheses = True
# If true, the current module name will be prepended to all description # If true, the current module name will be prepended to all description
# unit titles (such as .. function::). # unit titles (such as .. function::).
#add_module_names = True # add_module_names = True
# If true, sectionauthor and moduleauthor directives will be shown in the # If true, sectionauthor and moduleauthor directives will be shown in the
# output. They are ignored by default. # output. They are ignored by default.
#show_authors = False # show_authors = False
# A list of ignored prefixes for module index sorting. # A list of ignored prefixes for module index sorting.
#modindex_common_prefix = [] # modindex_common_prefix = []
# -- Options for HTML output --------------------------------------------------- # -- Options for HTML output ---------------------------------------------------
# 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
# documentation. # documentation.
#html_theme_options = {} # html_theme_options = {}
# Add any paths that contain custom themes here, relative to this directory. # Add any paths that contain custom themes here, relative to this directory.
#html_theme_path = [] # html_theme_path = []
# The name for this set of Sphinx documents. If None, it defaults to # The name for this set of Sphinx documents. If None, it defaults to
# "<project> v<release> documentation". # "<project> v<release> documentation".
#html_title = None # html_title = None
# A shorter title for the navigation bar. Default is the same as html_title. # A shorter title for the navigation bar. Default is the same as html_title.
#html_short_title = None # html_short_title = None
# The name of an image file (relative to this directory) to place at the top # The name of an image file (relative to this directory) to place at the top
# of the sidebar. # of the sidebar.
#html_logo = None # html_logo = None
# The name of an image file (within the static path) to use as favicon of the # The name of an image file (within the static path) to use as favicon of the
# docs. This file should be a Windows icon file (.ico) being 16x16 or 32x32 # docs. This file should be a Windows icon file (.ico) being 16x16 or 32x32
# pixels large. # pixels large.
#html_favicon = None # html_favicon = None
# 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.
#html_last_updated_fmt = '%b %d, %Y' # html_last_updated_fmt = '%b %d, %Y'
# If true, SmartyPants will be used to convert quotes and dashes to # If true, SmartyPants will be used to convert quotes and dashes to
# typographically correct entities. # typographically correct entities.
#html_use_smartypants = True # html_use_smartypants = True
# Custom sidebar templates, maps document names to template names. # Custom sidebar templates, maps document names to template names.
#html_sidebars = {} # html_sidebars = {}
# Additional templates that should be rendered to pages, maps page names to # Additional templates that should be rendered to pages, maps page names to
# template names. # template names.
#html_additional_pages = {} # html_additional_pages = {}
# If false, no module index is generated. # If false, no module index is generated.
#html_domain_indices = True # html_domain_indices = True
# If true, the index is split into individual pages for each letter. # If true, the index is split into individual pages for each letter.
#html_split_index = False # html_split_index = False
# If true, links to the reST sources are added to the pages. # If true, links to the reST sources are added to the pages.
#html_show_sourcelink = True # html_show_sourcelink = True
# If true, "Created using Sphinx" is shown in the HTML footer. Default is True. # If true, "Created using Sphinx" is shown in the HTML footer. Default is True.
#html_show_sphinx = True # html_show_sphinx = True
# If true, "(C) Copyright ..." is shown in the HTML footer. Default is True. # If true, "(C) Copyright ..." is shown in the HTML footer. Default is True.
#html_show_copyright = True # html_show_copyright = True
# If true, an OpenSearch description file will be output, and all pages will # If true, an OpenSearch description file will be output, and all pages will
# contain a <link> tag referring to it. The value of this option must be the # contain a <link> tag referring to it. The value of this option must be the
# base URL from which the finished HTML is served. # base URL from which the finished HTML is served.
#html_use_opensearch = '' # html_use_opensearch = ''
# This is the file name suffix for HTML files (e.g. ".xhtml"). # This is the file name suffix for HTML files (e.g. ".xhtml").
#html_file_suffix = None # html_file_suffix = None
# -- 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

12
stegano/exifHeader/exifHeader.py
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

5
stegano/lsb/lsb.py
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
@ -128,4 +127,4 @@ def reveal(input_image: Union[str, IO[bytes]], encoding: str = "UTF-8", shift: i
if len(bitab) - len(str(limit)) - 1 == limit: if len(bitab) - len(str(limit)) - 1 == limit:
img.close() img.close()
return "".join(bitab)[len(str(limit)) + 1:] return "".join(bitab)[len(str(limit)) + 1 :]

3
stegano/lsbset/lsbset.py
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

9
stegano/tools.py
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)

3
tests/test_exifHeader.py
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=""
) )

8
tests/test_lsbset.py
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)

9
tests/test_steganalysis.py
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")