Changeset 392 for pwntcha/trunk/src/main.cTweet

Timestamp:

Jan 4, 2005, 12:26:29 AM (18 years ago)

Author:

Sam Hocevar

Message:

• debug message system

File:

• pwntcha/trunk/src/main.c (modified) (9 diffs)

pwntcha/trunk/src/main.c

@@ -14 +14 @@
 #include <string.h>
 #include <getopt.h>
+#include <stdarg.h>
 
 #include "config.h"
@@ -24 +25 @@
 #endif
 
+/* Used for the debug messages */
+char *argv0 = NULL;
+int debug = 1;
+
 int main(int argc, char *argv[])
 {
     char *mode = "auto";
+
+    argv0 = argv[0];
 
     int c;
@@ -40 +47 @@
             { "mode", 1, 0, 'm' },
             { "help", 0, 0, 'h' },
+            { "quiet", 0, 0, 'q' },
             { "version", 0, 0, 'v' },
             { 0, 0, 0, 0 }
         };
 
-        c = getopt_long(argc, argv, "hm:v", long_options, &option_index);
+        c = getopt_long(argc, argv, "hm:qv", long_options, &option_index);
 #else
-        c = getopt(argc, argv, "hm:v");
+        c = getopt(argc, argv, "hm:qv");
 #endif
         if(c == -1)
@@ -57 +65 @@
 #ifdef HAVE_GETOPT_LONG
             printf("  -m, --mode      force operating mode\n");
+            printf("  -q, --quiet     do not print information messages\n");
             printf("  -h, --help      display this help and exit\n");
             printf("  -v, --version   output version information and exit\n");
 #else
             printf("  -m     force operating mode\n");
+            printf("  -q     do not print information messages\n");
             printf("  -h     display this help and exit\n");
             printf("  -v     output version information and exit\n");
@@ -67 +77 @@
         case 'm': /* --mode */
             mode = optarg;
+            break;
+        case 'q': /* --quiet */
+            debug = 0;
             break;
         case 'v': /* --version */
@@ -101 +114 @@
         if(!img)
         {
-            fprintf(stderr, "%s: cannot load %s\n", argv[0], input);
+            dprintf("cannot load %s\n", input);
             printf("\n");
             continue;
@@ -115 +128 @@
         {
             if(img->width == 320 && img->height == 50)
+            {
+                dprintf("autodetecting phpBB captcha\n");
                 result = decode_phpbb(img);
+            }
             else if(img->height == 69)
+            {
+                dprintf("autodetecting slashdot captcha\n");
                 result = decode_slashdot(img);
+            }
             else
             {
-                fprintf(stderr, "%s: could not guess captcha type\n", argv[0]);
+                dprintf("could not guess captcha type\n");
                 printf("\n");
                 image_free(img);
@@ -131 +150 @@
         if(!result)
         {
-            fprintf(stderr, "%s: sorry, decoding failed\n", argv[0]);
+            dprintf("sorry, decoding failed\n");
             printf("\n");
             continue;
@@ -143 +162 @@
 }
 
+void dprintf(const char *fmt, ...)
+{
+    va_list args;
+
+    if(!debug)
+        return;
+
+    va_start(args, fmt);
+    fprintf(stderr, "%s: ", argv0);
+    vfprintf(stderr, fmt, args);
+    va_end(args);
+}
+