cvs.zerfleddert.de Git - proxmark3-svn/blob - client/proxmark3.c

   1 //-----------------------------------------------------------------------------
   2 // Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
   3 // Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
   4 //
   5 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
   6 // at your option, any later version. See the LICENSE.txt file for the text of
   7 // the license.
   8 //-----------------------------------------------------------------------------
   9 // Main binary
  10 //-----------------------------------------------------------------------------
  11
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <string.h>
  15 #include <pthread.h>
  16 #include <unistd.h>
  17 #include <readline/readline.h>
  18 #include <readline/history.h>
  19 #include "proxusb.h"
  20 #include "proxmark3.h"
  21 #include "proxgui.h"
  22 #include "cmdmain.h"
  23
  24 struct usb_receiver_arg
  25 {
  26   int run;
  27 };
  28
  29 struct main_loop_arg
  30 {
  31   int usb_present;
  32 };
  33
  34 static void *usb_receiver(void *targ)
  35 {
  36   struct usb_receiver_arg *arg = (struct usb_receiver_arg*)targ;
  37   UsbCommand cmdbuf;
  38
  39   while (arg->run) {
  40     if (ReceiveCommandPoll(&cmdbuf)) {
  41       for (int i = 0; i < strlen(PROXPROMPT); i++)
  42         putchar(0x08);
  43       UsbCommandReceived(&cmdbuf);
  44                         // there is a big bug )
  45                         if (cmdbuf.cmd >= 0x0100 && cmdbuf.cmd <= 0x0110) { // debug commands
  46                                 printf(">");
  47 //                              rl_on_new_line_with_prompt();
  48 //                              rl_forced_update_display();
  49                         }
  50       fflush(NULL);
  51     }
  52   }
  53
  54   pthread_exit(NULL);
  55   return NULL;
  56 }
  57
  58 static void *main_loop(void *targ)
  59 {
  60   struct main_loop_arg *arg = (struct main_loop_arg*)targ;
  61   struct usb_receiver_arg rarg;
  62   char *cmd = NULL;
  63   pthread_t reader_thread;
  64
  65   if (arg->usb_present == 1) {
  66     rarg.run=1;
  67     pthread_create(&reader_thread, NULL, &usb_receiver, &rarg);
  68   }
  69
  70         read_history(".history");
  71         while(1) {
  72                 cmd = readline(PROXPROMPT);
  73                 if (cmd) {
  74                         while(cmd[strlen(cmd) - 1] == ' ')
  75                         cmd[strlen(cmd) - 1] = 0x00;
  76
  77                         if (cmd[0] != 0x00) {
  78                                 if (strncmp(cmd, "quit", 4) == 0) {
  79                                         write_history(".history");
  80                                         break;
  81                                 }
  82
  83                                 CommandReceived(cmd);
  84                                 add_history(cmd);
  85                         }
  86                         free(cmd);
  87                 } else {
  88                         printf("\n");
  89                         break;
  90                 }
  91         }
  92
  93   if (arg->usb_present == 1) {
  94     rarg.run = 0;
  95     pthread_join(reader_thread, NULL);
  96   }
  97
  98   ExitGraphics();
  99   pthread_exit(NULL);
 100   return NULL;
 101 }
 102
 103 int main(int argc, char **argv)
 104 {
 105   struct main_loop_arg marg;
 106   pthread_t main_loop_t;
 107   usb_init();
 108
 109   if (!OpenProxmark(1)) {
 110     fprintf(stderr,"PROXMARK3: NOT FOUND!\n");
 111     marg.usb_present = 0;
 112     offline = 1;
 113   } else {
 114     marg.usb_present = 1;
 115     offline = 0;
 116   }
 117
 118   pthread_create(&main_loop_t, NULL, &main_loop, &marg);
 119   InitGraphics(argc, argv);
 120
 121   MainGraphics();
 122
 123   pthread_join(main_loop_t, NULL);
 124
 125   if (marg.usb_present == 1) {
 126     CloseProxmark();
 127   }
 128   return 0;
 129 }