X-Git-Url: http://git.zerfleddert.de/cgi-bin/gitweb.cgi/proxmark3-svn/blobdiff_plain/bdd1de1b0258ccfd55459e53212b0ce84e37fe57..9440213d6b05ca9f5c7b1158f91654708448d776:/client/cmdmain.c diff --git a/client/cmdmain.c b/client/cmdmain.c index 30b431f5..2fdaabc1 100644 --- a/client/cmdmain.c +++ b/client/cmdmain.c @@ -1,8 +1,20 @@ +//----------------------------------------------------------------------------- +// Copyright (C) 2010 iZsh +// +// This code is licensed to you under the terms of the GNU GPL, version 2 or, +// at your option, any later version. See the LICENSE.txt file for the text of +// the license. +//----------------------------------------------------------------------------- +// Main command parser entry point +//----------------------------------------------------------------------------- + #include #include #include #include +#include "sleep.h" #include "cmdparser.h" +#include "proxmark3.h" #include "data.h" #include "usb_cmd.h" #include "ui.h" @@ -11,9 +23,12 @@ #include "cmdhw.h" #include "cmdlf.h" #include "cmdmain.h" +#include "util.h" unsigned int current_command = CMD_UNKNOWN; unsigned int received_command = CMD_UNKNOWN; +UsbCommand current_response; +UsbCommand current_response_user; static int CmdHelp(const char *Cmd); static int CmdQuit(const char *Cmd); @@ -21,11 +36,11 @@ static int CmdQuit(const char *Cmd); static command_t CommandTable[] = { {"help", CmdHelp, 1, "This help. Use ' help' for details of the following commands:\n"}, - {"data", CmdData, 1, "Plot window / data buffer manipulation"}, + {"data", CmdData, 1, "{ Plot window / data buffer manipulation... }"}, {"exit", CmdQuit, 1, "Exit program"}, - {"hf", CmdHF, 1, "HF commands"}, - {"hw", CmdHW, 1, "Hardware commands"}, - {"lf", CmdLF, 1, "LF commands"}, + {"hf", CmdHF, 1, "{ HF commands... }"}, + {"hw", CmdHW, 1, "{ Hardware commands... }"}, + {"lf", CmdLF, 1, "{ LF commands... }"}, {"quit", CmdQuit, 1, "Quit program"}, {NULL, NULL, 0, NULL} }; @@ -42,27 +57,40 @@ int CmdQuit(const char *Cmd) return 0; } -void WaitForResponse(uint32_t response_type) -{ - while (received_command != response_type) { -#ifdef WIN32 - UsbCommand c; - if (ReceiveCommandPoll(&c)) - UsbCommandReceived(&c); - Sleep(0); -#else - usleep(10000); // XXX ugh -#endif +bool WaitForResponseTimeout(uint32_t cmd, UsbCommand* response, size_t ms_timeout) { + + // Wait until the command is received + for(size_t i=0; received_command != cmd && i < ms_timeout/10; i++) { + msleep(10); // XXX ugh + if (i == 200) { // Two seconds elapsed + PrintAndLog("Waiting for a response from the proxmark..."); + PrintAndLog("Don't forget to cancel its operation first by pressing on the button"); + } + } + + // Check if timeout occured + if(received_command != cmd) return false; + + // Copy the received response (if supplied) + if (response) { + memcpy(response, ¤t_response, sizeof(UsbCommand)); } + + // Reset the received command received_command = CMD_UNKNOWN; + + return true; +} + +bool WaitForResponse(uint32_t cmd, UsbCommand* response) { + return WaitForResponseTimeout(cmd,response,-1); } //----------------------------------------------------------------------------- // Entry point into our code: called whenever the user types a command and // then presses Enter, which the full command line that they typed. //----------------------------------------------------------------------------- -void CommandReceived(char *Cmd) -{ +void CommandReceived(char *Cmd) { CmdsParse(CommandTable, Cmd); } @@ -72,23 +100,31 @@ void CommandReceived(char *Cmd) //----------------------------------------------------------------------------- void UsbCommandReceived(UsbCommand *UC) { + // Debug + // printf("UsbCommand length[len=%d]\n",sizeof(UsbCommand)); + // printf(" cmd[len=%d]: %x\n",sizeof(UC->cmd),UC->cmd); + // printf(" arg0[len=%d]: %x\n",sizeof(UC->arg[0]),UC->arg[0]); + // printf(" arg1[len=%d]: %x\n",sizeof(UC->arg[1]),UC->arg[1]); + // printf(" arg2[len=%d]: %x\n",sizeof(UC->arg[2]),UC->arg[2]); + // printf(" data[len=%d]: %02x%02x%02x...\n",sizeof(UC->d.asBytes),UC->d.asBytes[0],UC->d.asBytes[1],UC->d.asBytes[2]); + // printf("%s(%x) current cmd = %x\n", __FUNCTION__, c->cmd, current_command); - /* If we recognize a response, return to avoid further processing */ + // If we recognize a response, return to avoid further processing switch(UC->cmd) { + // First check if we are handling a debug message case CMD_DEBUG_PRINT_STRING: { - char s[100]; - if(UC->arg[0] > 70 || UC->arg[0] < 0) { - UC->arg[0] = 0; - } - memcpy(s, UC->d.asBytes, UC->arg[0]); - s[UC->arg[0]] = '\0'; - PrintAndLog("#db# %s", s); + char s[USB_CMD_DATA_SIZE+1]; + size_t len = MIN(UC->arg[0],USB_CMD_DATA_SIZE); + memcpy(s,UC->d.asBytes,len); + s[len] = 0x00; + PrintAndLog("#db# %s ", s); return; - } + } break; - case CMD_DEBUG_PRINT_INTEGERS: - PrintAndLog("#db# %08x, %08x, %08x\r\n", UC->arg[0], UC->arg[1], UC->arg[2]); + case CMD_DEBUG_PRINT_INTEGERS: { + PrintAndLog("#db# %08x, %08x, %08x \r\n", UC->arg[0], UC->arg[1], UC->arg[2]); return; + } break; case CMD_MEASURED_ANTENNA_TUNING: { int peakv, peakf; @@ -99,7 +135,6 @@ void UsbCommandReceived(UsbCommand *UC) peakf = UC->arg[2] & 0xffff; peakv = UC->arg[2] >> 16; PrintAndLog(""); - PrintAndLog(""); PrintAndLog("# LF antenna: %5.2f V @ 125.00 kHz", vLf125/1000.0); PrintAndLog("# LF antenna: %5.2f V @ 134.00 kHz", vLf134/1000.0); PrintAndLog("# LF optimal: %5.2f V @%9.2f kHz", peakv/1000.0, 12000.0/(peakf+1)); @@ -112,12 +147,50 @@ void UsbCommandReceived(UsbCommand *UC) PrintAndLog("# Your HF antenna is unusable."); else if (vHf<5000) PrintAndLog("# Your HF antenna is marginal."); - return; - } - default: - break; + } break; + + case CMD_DOWNLOADED_RAW_ADC_SAMPLES_125K: { +// printf("received samples: "); +// print_hex(UC->d.asBytes,512); + sample_buf_len += UC->arg[1]; +// printf("samples: %zd offset: %d\n",sample_buf_len,UC->arg[0]); + memcpy(sample_buf+(UC->arg[0]),UC->d.asBytes,UC->arg[1]); + } break; + + +// case CMD_ACK: { +// PrintAndLog("Receive ACK\n"); +// } break; + + default: { + // Maybe it's a response + switch(current_command) { + case CMD_DOWNLOAD_RAW_ADC_SAMPLES_125K: { + if (UC->cmd != CMD_DOWNLOADED_RAW_ADC_SAMPLES_125K) { + PrintAndLog("unrecognized command %08x\n", UC->cmd); + break; + } +// int i; + PrintAndLog("received samples %d\n",UC->arg[0]); + memcpy(sample_buf+UC->arg[0],UC->d.asBytes,48); + sample_buf_len += 48; +// for(i=0; i<48; i++) sample_buf[i] = UC->d.asBytes[i]; + received_command = UC->cmd; + } break; + + default: { + } break; + } +// // Store the last received command +// memcpy(¤t_response, UC, sizeof(UsbCommand)); +// received_command = UC->cmd; + } break; } - /* Maybe it's a response: */ + // Store the last received command + memcpy(¤t_response, UC, sizeof(UsbCommand)); + received_command = UC->cmd; +/* + // Maybe it's a response: switch(current_command) { case CMD_DOWNLOAD_RAW_ADC_SAMPLES_125K: if (UC->cmd != CMD_DOWNLOADED_RAW_ADC_SAMPLES_125K) goto unexpected_response; @@ -133,7 +206,12 @@ void UsbCommandReceived(UsbCommand *UC) return; default: unexpected_response: - PrintAndLog("unrecognized command %08x\n", UC->cmd); - break; + + if(UC->cmd != CMD_ACK) + PrintAndLog("unrecognized command %08x \n", UC->cmd); + else + memcpy(¤t_response, UC, sizeof(UsbCommand)); + received_command = UC->cmd; } + */ }