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