1 //-----------------------------------------------------------------------------
2 // Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
3 // Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
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
8 //-----------------------------------------------------------------------------
10 //-----------------------------------------------------------------------------
17 #include <readline/readline.h>
18 #include <readline/history.h>
19 //#include "proxusb.h"
20 #include "proxmark3.h"
26 #include "cmdparser.h"
29 // a global mutex to prevent interlaced printing from different threads
30 pthread_mutex_t print_lock
;
32 static serial_port sp
;
33 static UsbCommand txcmd
;
34 volatile static bool txcmd_pending
= false;
36 void SendCommand(UsbCommand
*c
) {
38 printf("Sending %d bytes\n", sizeof(UsbCommand
));
42 ERR("Sending command failed, previous command is still pending");
47 PrintAndLog("Sending bytes to proxmark failed - offline");
51 The while-loop below causes hangups at times, when the pm3 unit is unresponsive
52 or disconnected. The main console thread is alive, but comm thread just spins here.
64 struct main_loop_arg
{
66 char *script_cmds_file
;
69 //static void *usb_receiver(void *targ) {
70 // struct receiver_arg *arg = (struct receiver_arg*)targ;
74 // if (ReceiveCommandPoll(&cmdbuf)) {
75 // UsbCommandReceived(&cmdbuf);
80 // pthread_exit(NULL);
87 static void *uart_receiver(void *targ
) {
88 struct receiver_arg
*arg
= (struct receiver_arg
*)targ
;
93 rxlen
= sizeof(UsbCommand
);
94 if (uart_receive(sp
,prx
,&rxlen
)) {
96 if (((prx
-rx
) % sizeof(UsbCommand
)) != 0) {
99 cmd_count
= (prx
-rx
) / sizeof(UsbCommand
);
100 // printf("received %d bytes, which represents %d commands\n",(prx-rx), cmd_count);
101 for (size_t i
=0; i
<cmd_count
; i
++) {
102 UsbCommandReceived((UsbCommand
*)(rx
+(i
*sizeof(UsbCommand
))));
108 if (!uart_send(sp
,(byte_t
*)&txcmd
,sizeof(UsbCommand
))) {
109 PrintAndLog("Sending bytes to proxmark failed");
111 txcmd_pending
= false;
119 static void *main_loop(void *targ
) {
120 struct main_loop_arg
*arg
= (struct main_loop_arg
*)targ
;
121 struct receiver_arg rarg
;
123 pthread_t reader_thread
;
125 if (arg
->usb_present
== 1) {
127 // pthread_create(&reader_thread, NULL, &usb_receiver, &rarg);
128 pthread_create(&reader_thread
, NULL
, &uart_receiver
, &rarg
);
131 FILE *script_file
= NULL
;
132 char script_cmd_buf
[256];
134 if (arg
->script_cmds_file
)
136 script_file
= fopen(arg
->script_cmds_file
, "r");
139 printf("using 'scripting' commands file %s\n", arg
->script_cmds_file
);
143 read_history(".history");
146 // If there is a script file
149 if (!fgets(script_cmd_buf
, sizeof(script_cmd_buf
), script_file
))
157 nl
= strrchr(script_cmd_buf
, '\r');
159 nl
= strrchr(script_cmd_buf
, '\n');
162 if ((cmd
= (char*) malloc(strlen(script_cmd_buf
) + 1)) != NULL
)
164 memset(cmd
, 0, strlen(script_cmd_buf
));
165 strcpy(cmd
, script_cmd_buf
);
173 cmd
= readline(PROXPROMPT
);
177 while(cmd
[strlen(cmd
) - 1] == ' ')
178 cmd
[strlen(cmd
) - 1] = 0x00;
180 if (cmd
[0] != 0x00) {
181 if (strncmp(cmd
, "quit", 4) == 0) {
186 CommandReceived(cmd
);
196 write_history(".history");
198 if (arg
->usb_present
== 1) {
200 pthread_join(reader_thread
, NULL
);
214 //static void dumpHelp(char *parent, ...)
216 // printf("## %s\n\n", parent);
217 // CommandReceived(parent);
222 static void dumpAllHelp(int markdown
)
224 printf("\n%sProxmark3 command dump%s\n\n",markdown
?"# ":"",markdown
?"":"\n======================");
225 printf("Some commands are available only if a Proxmark is actually connected.%s\n",markdown
?" ":"");
226 printf("Check column \"offline\" for their availability.\n");
228 command_t
*cmds
= getTopLevelCommandTable();
229 dumpCommandsRecursive(cmds
, markdown
);
232 int main(int argc
, char* argv
[]) {
236 printf("syntax: %s <port>\n\n",argv
[0]);
237 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
238 printf("help: %s -h\n\n", argv
[0]);
239 printf("\tDump all interactive help at once\n");
240 printf("markdown: %s -m\n\n", argv
[0]);
241 printf("\tDump all interactive help at once in markdown syntax\n");
244 if (strcmp(argv
[1], "-h") == 0) {
245 printf("syntax: %s <port>\n\n",argv
[0]);
246 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
250 if (strcmp(argv
[1], "-m") == 0) {
254 // Make sure to initialize
255 struct main_loop_arg marg
= {
257 .script_cmds_file
= NULL
259 pthread_t main_loop_t
;
263 if (!OpenProxmark(1)) {
264 fprintf(stderr,"PROXMARK3: NOT FOUND!\n");
265 marg.usb_present = 0;
268 marg.usb_present = 1;
273 sp
= uart_open(argv
[1]);
274 if (sp
== INVALID_SERIAL_PORT
) {
275 printf("ERROR: invalid serial port\n");
276 marg
.usb_present
= 0;
278 } else if (sp
== CLAIMED_SERIAL_PORT
) {
279 printf("ERROR: serial port is claimed by another process\n");
280 marg
.usb_present
= 0;
283 marg
.usb_present
= 1;
287 // If the user passed the filename of the 'script' to execute, get it
288 if (argc
> 2 && argv
[2]) {
289 if (argv
[2][0] == 'f' && //buzzy, if a word 'flush' passed, flush the output after every log entry.
295 printf("Output will be flushed after every print.\n");
299 marg
.script_cmds_file
= argv
[2];
302 // create a mutex to avoid interlacing print commands from our different threads
303 pthread_mutex_init(&print_lock
, NULL
);
305 pthread_create(&main_loop_t
, NULL
, &main_loop
, &marg
);
306 InitGraphics(argc
, argv
);
310 pthread_join(main_loop_t
, NULL
);
312 // if (marg.usb_present == 1) {
320 pthread_mutex_destroy(&print_lock
);