1 files changed, 47 insertions, 13 deletions

diff --git a/src/irecovery.c b/src/irecovery.c
index 90f975b..51320a5 100644
--- a/src/irecovery.c
+++ b/src/irecovery.c
@@ -17,9 +17,50 @@
  **/
 
 #include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
 #include <libirecovery.h>
 
+enum {
+        kResetDevice, kSendCommand
+};
+
+void print_usage() {
+        printf("iRecovery - iDevice Recovery Utility\n");
+        printf("Usage: ./irecovery [args]\n");
+        exit(1);
+}
+
 int main(int argc, char** argv) {
+        int opt = 0;
+        int action = 0;
+        char* argument = NULL;
+        if(argc == 1) print_usage();
+        while ((opt = getopt(argc, argv, "dhrc:f:")) > 0) {
+                switch (opt) {
+                case 'd':
+                        irecv_set_debug(1);
+                        break;
+
+                case 'h':
+                        print_usage();
+                        break;
+
+                case 'r':
+                        action = kResetDevice;
+                        break;
+
+                case 'c':
+                        argument = optarg;
+                        action = kSendCommand;
+                        break;
+
+                default:
+                        fprintf(stderr, "Unknown argument\n");
+                        break;
+                }
+        }
+
         irecv_device* device = NULL;
         if(irecv_init(&device) < 0) {
                 fprintf(stderr, "Unable to initialize libirecovery\n");
@@ -31,27 +72,20 @@ int main(int argc, char** argv) {
                 return -1;
         }
 
-        switch (device->mode) {
-        case kRecoveryMode:
-                printf("Found device in recovery mode\n");
+        switch(action) {
+        case kResetDevice:
+                irecv_reset(device);
                 break;
 
-        case kDfuMode:
-                printf("Found device in DFU mode\n");
-                break;
-
-        case kKernelMode:
-                printf("Found device in kernel mode\n");
+        case kSendCommand:
+                irecv_command(device, argument);
                 break;
 
         default:
-                printf("No device found\n");
+                fprintf(stderr, "Unknown action\n");
                 break;
         }
 
-        printf("Sending USB reset...\n");
-        irecv_reset(device);
-
         irecv_exit(device);
         return 0;
 }