git.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
  20 #include "proxmark3.h"
  21 #include "proxgui.h"
  22 #include "cmdmain.h"
  23 #include "uart.h"
  24 #include "ui.h"
  25 #include "sleep.h"
  26 #include "cmdparser.h"
  27 #include "cmdhw.h"
  28
  29 // a global mutex to prevent interlaced printing from different threads
  30 pthread_mutex_t print_lock;
  31
  32 static serial_port sp;
  33 static UsbCommand txcmd;
  34 volatile static bool txcmd_pending = false;
  35
  36 void SendCommand(UsbCommand *c) {
  37         #if 0
  38   printf("Sending %d bytes\n", sizeof(UsbCommand));
  39         #endif
  40
  41         if (offline) {
  42       PrintAndLog("Sending bytes to proxmark failed - offline");
  43       return;
  44     }
  45   /**
  46         The while-loop below causes hangups at times, when the pm3 unit is unresponsive
  47         or disconnected. The main console thread is alive, but comm thread just spins here.
  48         Not good.../holiman
  49         **/
  50   while(txcmd_pending);
  51   txcmd = *c;
  52   txcmd_pending = true;
  53 }
  54
  55 struct receiver_arg {
  56   int run;
  57 };
  58
  59 struct main_loop_arg {
  60   int usb_present;
  61   char *script_cmds_file;
  62 };
  63
  64 byte_t rx[0x1000000];
  65 byte_t* prx = rx;
  66
  67 static void *uart_receiver(void *targ) {
  68   struct receiver_arg *arg = (struct receiver_arg*)targ;
  69   size_t rxlen;
  70   size_t cmd_count;
  71
  72   while (arg->run) {
  73     rxlen = sizeof(UsbCommand);
  74                 if (uart_receive(sp, prx, &rxlen)) {
  75       prx += rxlen;
  76       if (((prx-rx) % sizeof(UsbCommand)) != 0) {
  77         continue;
  78       }
  79       cmd_count = (prx-rx) / sizeof(UsbCommand);
  80
  81                         for (size_t i = 0; i < cmd_count; i++) {
  82         UsbCommandReceived((UsbCommand*)(rx+(i*sizeof(UsbCommand))));
  83       }
  84     }
  85     prx = rx;
  86
  87     if(txcmd_pending) {
  88                         if (!uart_send(sp, (byte_t*) &txcmd, sizeof(UsbCommand))) {
  89         PrintAndLog("Sending bytes to proxmark failed");
  90       }
  91       txcmd_pending = false;
  92     }
  93   }
  94
  95   pthread_exit(NULL);
  96   return NULL;
  97 }
  98
  99 static void *main_loop(void *targ) {
 100   struct main_loop_arg *arg = (struct main_loop_arg*)targ;
 101   struct receiver_arg rarg;
 102   char *cmd = NULL;
 103   pthread_t reader_thread;
 104
 105   if (arg->usb_present == 1) {
 106                 rarg.run = 1;
 107     pthread_create(&reader_thread, NULL, &uart_receiver, &rarg);
 108                 // cache Version information now:
 109                 CmdVersion(NULL);
 110   }
 111
 112   FILE *script_file = NULL;
 113         char script_cmd_buf[256];  // iceman, needs lua script the same file_path_buffer as the rest
 114
 115         if (arg->script_cmds_file) {
 116     script_file = fopen(arg->script_cmds_file, "r");
 117                 if (script_file) {
 118       printf("using 'scripting' commands file %s\n", arg->script_cmds_file);
 119     }
 120   }
 121
 122         read_history(".history");
 123
 124         while(1)  {
 125
 126     // If there is a script file
 127     if (script_file)
 128     {
 129                         if (!fgets(script_cmd_buf, sizeof(script_cmd_buf), script_file)) {
 130         fclose(script_file);
 131         script_file = NULL;
 132                         } else {
 133         char *nl;
 134         nl = strrchr(script_cmd_buf, '\r');
 135         if (nl) *nl = '\0';
 136
 137         nl = strrchr(script_cmd_buf, '\n');
 138         if (nl) *nl = '\0';
 139
 140                                 if ((cmd = (char*) malloc(strlen(script_cmd_buf) + 1)) != NULL) {
 141           memset(cmd, 0, strlen(script_cmd_buf));
 142           strcpy(cmd, script_cmd_buf);
 143           printf("%s\n", cmd);
 144         }
 145       }
 146     }
 147
 148                 if (!script_file) {
 149       cmd = readline(PROXPROMPT);
 150                 }
 151
 152                 if (cmd) {
 153
 154                         while(cmd[strlen(cmd) - 1] == ' ')
 155         cmd[strlen(cmd) - 1] = 0x00;
 156
 157                         if (cmd[0] != 0x00) {
 158                                 int ret = CommandReceived(cmd);
 159                                 add_history(cmd);
 160                                 if (ret == 99) {  // exit or quit
 161                                         break;
 162                                 }
 163                         }
 164                         free(cmd);
 165                 } else {
 166                         printf("\n");
 167                         break;
 168                 }
 169         }
 170
 171         write_history(".history");
 172
 173   if (arg->usb_present == 1) {
 174     rarg.run = 0;
 175     pthread_join(reader_thread, NULL);
 176   }
 177
 178         if (script_file) {
 179     fclose(script_file);
 180     script_file = NULL;
 181   }
 182
 183   ExitGraphics();
 184   pthread_exit(NULL);
 185   return NULL;
 186 }
 187
 188 static void dumpAllHelp(int markdown)
 189 {
 190   printf("\n%sProxmark3 command dump%s\n\n",markdown?"# ":"",markdown?"":"\n======================");
 191   printf("Some commands are available only if a Proxmark is actually connected.%s\n",markdown?"  ":"");
 192   printf("Check column \"offline\" for their availability.\n");
 193   printf("\n");
 194   command_t *cmds = getTopLevelCommandTable();
 195   dumpCommandsRecursive(cmds, markdown);
 196 }
 197
 198 int main(int argc, char* argv[]) {
 199         srand(time(0));
 200
 201         if (argc < 2) {
 202                 printf("syntax: %s <port>\n\n",argv[0]);
 203                 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv[0]);
 204                 printf("help:   %s -h\n\n", argv[0]);
 205                 printf("\tDump all interactive help at once\n");
 206                 printf("markdown:   %s -m\n\n", argv[0]);
 207                 printf("\tDump all interactive help at once in markdown syntax\n");
 208                 return 1;
 209         }
 210         if (strcmp(argv[1], "-h") == 0) {
 211                 printf("syntax: %s <port>\n\n",argv[0]);
 212                 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv[0]);
 213                 dumpAllHelp(0);
 214                 return 0;
 215         }
 216         if (strcmp(argv[1], "-m") == 0) {
 217                 dumpAllHelp(1);
 218                 return 0;
 219         }
 220         // Make sure to initialize
 221         struct main_loop_arg marg = {
 222                 .usb_present = 0,
 223                 .script_cmds_file = NULL
 224         };
 225         pthread_t main_loop_threat;
 226
 227
 228         sp = uart_open(argv[1]);
 229         if (sp == INVALID_SERIAL_PORT) {
 230                 printf("ERROR: invalid serial port\n");
 231                 marg.usb_present = 0;
 232                 offline = 1;
 233         } else if (sp == CLAIMED_SERIAL_PORT) {
 234                 printf("ERROR: serial port is claimed by another process\n");
 235                 marg.usb_present = 0;
 236                 offline = 1;
 237         } else {
 238                 marg.usb_present = 1;
 239                 offline = 0;
 240         }
 241
 242         // If the user passed the filename of the 'script' to execute, get it
 243         if (argc > 2 && argv[2]) {
 244                 if (argv[2][0] == 'f' &&  //buzzy, if a word 'flush' passed, flush the output after every log entry.
 245                         argv[2][1] == 'l' &&
 246                         argv[2][2] == 'u' &&
 247                         argv[2][3] == 's' &&
 248                         argv[2][4] == 'h')
 249                 {
 250                         printf("Output will be flushed after every print.\n");
 251                         flushAfterWrite = 1;
 252                 }
 253                 else
 254                 marg.script_cmds_file = argv[2];
 255         }
 256
 257         // create a mutex to avoid interlacing print commands from our different threads
 258         pthread_mutex_init(&print_lock, NULL);
 259
 260         pthread_create(&main_loop_threat, NULL, &main_loop, &marg);
 261         InitGraphics(argc, argv);
 262
 263         MainGraphics();
 264
 265         pthread_join(main_loop_threat, NULL);
 266
 267         // Clean up the port
 268         if (offline == 0) {
 269         uart_close(sp);
 270         }
 271
 272         // clean up mutex
 273         pthread_mutex_destroy(&print_lock);
 274
 275         exit(0);
 276 }