Reorganization of all modules.

bin/lsb-set

@@ -0,0 +1,95 @@
+#!/usr/bin/env python
+#-*- coding: utf-8 -*-
+
+# Stéganô - Stéganô is a basic Python Steganography module.
+# Copyright (C) 2010-2016  Cédric Bonhomme - https://www.cedricbonhomme.org
+#
+# For more information : https://github.com/cedricbonhomme/Stegano
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>
+
+__author__ = "Cedric Bonhomme"
+__version__ = "$Revision: 0.5 $"
+__date__ = "$Date: 2016/03/18 $"
+__license__ = "GPLv3"
+
+try:
+    from stegano import lsbset
+except:
+    print("Install stegano: sudo pip install Stegano")
+
+from stegano import tools
+
+from optparse import OptionParser
+parser = OptionParser(version=__version__)
+parser.add_option('--hide', action='store_true', default=False,
+                help="Hides a message in an image.")
+parser.add_option('--reveal', action='store_true', default=False,
+                help="Reveals the message hided in an image.")
+# Original image
+parser.add_option("-i", "--input", dest="input_image_file",
+                help="Input image file.")
+
+# Generator
+parser.add_option("-g", "--generator", dest="generator_function",
+                help="Generator")
+
+# Image containing the secret
+parser.add_option("-o", "--output", dest="output_image_file",
+                help="Output image containing the secret.")
+
+# Non binary secret message to hide
+parser.add_option("-m", "--secret-message", dest="secret_message",
+                help="Your secret message to hide (non binary).")
+
+# Binary secret to hide (OGG, executable, etc.)
+parser.add_option("-f", "--secret-file", dest="secret_file",
+                help="Your secret to hide (Text or any binary file).")
+# Output for the binary binary secret.
+parser.add_option("-b", "--binary", dest="secret_binary",
+                help="Output for the binary secret (Text or any binary file).")
+
+parser.set_defaults(input_image_file = './pictures/Lenna.png',
+                    generator_function = 'fermat',
+                    output_image_file = './pictures/Lenna_enc.png',
+                    secret_message = '', secret_file = '', secret_binary = "")
+
+(options, args) = parser.parse_args()
+
+
+if options.hide:
+    if options.secret_message != "" and options.secret_file == "":
+        secret = options.secret_message
+    elif options.secret_message == "" and options.secret_file != "":
+        secret = tools.binary2base64(options.secret_file)
+
+    img_encoded = lsbset.hide(options.input_image_file, secret, options.generator_function)
+    try:
+        img_encoded.save(options.output_image_file)
+    except Exception as e:
+        # If hide() returns an error (Too long message).
+        print(e)
+
+elif options.reveal:
+    try:
+        secret = lsbset.reveal(options.input_image_file, options.generator_function)
+    except IndexError:
+        print("Impossible to detect message.")
+        exit(0)
+    if options.secret_binary != "":
+        data = tools.base642binary(secret)
+        with open(options.secret_binary, "w") as f:
+            f.write(data)
+    else:
+        print(secret)