From ea472c0e9a2a642c611442dbce98c21a86d5e109 Mon Sep 17 00:00:00 2001
From: cedricbonhomme <devnull@localhost>
Date: Thu, 7 Apr 2011 21:56:06 +0200
Subject: [PATCH] Minor changes in the slsb executable file.

---
 bin/slsb | 90 +++++++++++++++++++++++++++-----------------------------
 1 file changed, 44 insertions(+), 46 deletions(-)

diff --git a/bin/slsb b/bin/slsb
index 964e47a..f7fc6c6 100755
--- a/bin/slsb
+++ b/bin/slsb
@@ -29,57 +29,55 @@ try:
 except:
     print "Install stegano: python setup.py install"
 
-if __name__ == '__main__':
-    # Point of entry in execution mode.
-    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.")
-    # Image containing the secret
-    parser.add_option("-o", "--output", dest="output_image_file",
-                    help="Output image containing the secret.")
+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.")
+# 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).")
+# 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).")
+# 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',
-                        output_image_file = './pictures/Lenna_enc.png',
-                        secret_message = '', secret_file = '', secret_binary = "")
+parser.set_defaults(input_image_file = './pictures/Lenna.png',
+                    output_image_file = './pictures/Lenna_enc.png',
+                    secret_message = '', secret_file = '', secret_binary = "")
 
-    (options, args) = parser.parse_args()
+(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)
+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 = slsb.hide(options.input_image_file, secret)
-        try:
-            img_encoded.save(options.output_image_file)
-        except Exception, e:
-            # If hide() returns an error (Too long message).
-            print e
+    img_encoded = slsb.hide(options.input_image_file, secret)
+    try:
+        img_encoded.save(options.output_image_file)
+    except Exception, e:
+        # If hide() returns an error (Too long message).
+        print e
 
-    elif options.reveal:
-        secret = slsb.reveal(options.input_image_file)
-        if options.secret_binary != "":
-            data = tools.base642binary(secret)
-            with open(options.secret_binary, "w") as f:
-                f.write(data)
-        else:
-            print secret
\ No newline at end of file
+elif options.reveal:
+    secret = slsb.reveal(options.input_image_file)
+    if options.secret_binary != "":
+        data = tools.base642binary(secret)
+        with open(options.secret_binary, "w") as f:
+            f.write(data)
+    else:
+        print secret
\ No newline at end of file