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

   1 #include <usb.h>
   2 #include <stdio.h>
   3 #include <unistd.h>
   4 #include <stdlib.h>
   5 #include <strings.h>
   6 #include <string.h>
   7 #include <errno.h>
   8 #include <readline/readline.h>
   9 #include <readline/history.h>
  10 #include <pthread.h>
  11
  12 #include "prox.h"
  13 #include "proxmark3.h"
  14 #include "proxgui.h"
  15
  16 struct usb_receiver_arg {
  17         int run;
  18 };
  19
  20 struct main_loop_arg {
  21         int usb_present;
  22 };
  23
  24 static void *usb_receiver(void *targ) {
  25         struct usb_receiver_arg *arg = (struct usb_receiver_arg*)targ;
  26         UsbCommand cmdbuf;
  27
  28         while(arg->run) {
  29                 if (ReceiveCommandPoll(&cmdbuf)) {
  30                         int i;
  31
  32                         for (i=0; i<strlen(PROXPROMPT); i++)
  33                                 putchar(0x08);
  34
  35                         UsbCommandReceived(&cmdbuf);
  36                         printf(PROXPROMPT);
  37                         fflush(NULL);
  38                 }
  39         }
  40
  41         pthread_exit(NULL);
  42 }
  43
  44 static void *main_loop(void *targ)
  45 {
  46         struct main_loop_arg *arg = (struct main_loop_arg*)targ;
  47         struct usb_receiver_arg rarg;
  48         char *cmd = NULL;
  49         pthread_t reader_thread;
  50
  51         if (arg->usb_present == 1) {
  52                 rarg.run=1;
  53                 pthread_create(&reader_thread, NULL, &usb_receiver, &rarg);
  54         }
  55
  56         while(1) {
  57
  58                 cmd = readline(PROXPROMPT);
  59                 if (cmd) {
  60                         if (cmd[0] != 0x00) {
  61                                 CommandReceived(cmd);
  62                                 add_history(cmd);
  63                         }
  64                         free(cmd);
  65                 } else {
  66                         printf("\n");
  67                         break;
  68                 }
  69         }
  70
  71         if (arg->usb_present == 1) {
  72                 rarg.run = 0;
  73                 pthread_join(reader_thread, NULL);
  74         }
  75
  76         ExitGraphics();
  77         pthread_exit(NULL);
  78 }
  79
  80 int main(int argc, char **argv)
  81 {
  82         struct main_loop_arg marg;
  83         pthread_t main_loop_t;
  84         usb_init();
  85
  86         if (!(devh = OpenProxmark(1))) {
  87                 fprintf(stderr,"PROXMARK3: NOT FOUND!\n");
  88                 marg.usb_present = 0;
  89                 offline = 1;
  90         } else {
  91                 marg.usb_present = 1;
  92                 offline = 0;
  93         }
  94
  95         pthread_create(&main_loop_t, NULL, &main_loop, &marg);
  96         InitGraphics(argc, argv);
  97
  98         MainGraphics();
  99
 100         pthread_join(main_loop_t, NULL);
 101
 102         if (marg.usb_present == 1) {
 103                 CloseProxmark();
 104         }
 105         return 0;
 106 }