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>
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 PrintAndLog("Sending bytes to proxmark failed - offline");
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.
59 struct main_loop_arg
{
61 char *script_cmds_file
;
67 static void *uart_receiver(void *targ
) {
68 struct receiver_arg
*arg
= (struct receiver_arg
*)targ
;
73 rxlen
= sizeof(UsbCommand
);
74 if (uart_receive(sp
, prx
, &rxlen
)) {
76 if (((prx
-rx
) % sizeof(UsbCommand
)) != 0) {
79 cmd_count
= (prx
-rx
) / sizeof(UsbCommand
);
81 for (size_t i
= 0; i
< cmd_count
; i
++) {
82 UsbCommandReceived((UsbCommand
*)(rx
+(i
*sizeof(UsbCommand
))));
88 if (!uart_send(sp
, (byte_t
*) &txcmd
, sizeof(UsbCommand
))) {
89 PrintAndLog("Sending bytes to proxmark failed");
91 txcmd_pending
= false;
99 static void *main_loop(void *targ
) {
100 struct main_loop_arg
*arg
= (struct main_loop_arg
*)targ
;
101 struct receiver_arg rarg
;
103 pthread_t reader_thread
;
105 if (arg
->usb_present
== 1) {
107 pthread_create(&reader_thread
, NULL
, &uart_receiver
, &rarg
);
110 FILE *script_file
= NULL
;
111 char script_cmd_buf
[256]; // iceman, needs lua script the same file_path_buffer as the rest
113 if (arg
->script_cmds_file
) {
114 script_file
= fopen(arg
->script_cmds_file
, "r");
116 printf("using 'scripting' commands file %s\n", arg
->script_cmds_file
);
120 read_history(".history");
124 // If there is a script file
127 if (!fgets(script_cmd_buf
, sizeof(script_cmd_buf
), script_file
)) {
132 nl
= strrchr(script_cmd_buf
, '\r');
135 nl
= strrchr(script_cmd_buf
, '\n');
138 if ((cmd
= (char*) malloc(strlen(script_cmd_buf
) + 1)) != NULL
) {
139 memset(cmd
, 0, strlen(script_cmd_buf
));
140 strcpy(cmd
, script_cmd_buf
);
147 cmd
= readline(PROXPROMPT
);
152 while(cmd
[strlen(cmd
) - 1] == ' ')
153 cmd
[strlen(cmd
) - 1] = 0x00;
155 if (cmd
[0] != 0x00) {
156 if (strncmp(cmd
, "quit", 4) == 0) {
160 CommandReceived(cmd
);
170 write_history(".history");
172 if (arg
->usb_present
== 1) {
174 pthread_join(reader_thread
, NULL
);
187 static void dumpAllHelp(int markdown
)
189 printf("\n%sProxmark3 command dump%s\n\n",markdown
?"# ":"",markdown
?"":"\n======================");
190 printf("Some commands are available only if a Proxmark is actually connected.%s\n",markdown
?" ":"");
191 printf("Check column \"offline\" for their availability.\n");
193 command_t
*cmds
= getTopLevelCommandTable();
194 dumpCommandsRecursive(cmds
, markdown
);
197 int main(int argc
, char* argv
[]) {
201 printf("syntax: %s <port>\n\n",argv
[0]);
202 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
203 printf("help: %s -h\n\n", argv
[0]);
204 printf("\tDump all interactive help at once\n");
205 printf("markdown: %s -m\n\n", argv
[0]);
206 printf("\tDump all interactive help at once in markdown syntax\n");
209 if (strcmp(argv
[1], "-h") == 0) {
210 printf("syntax: %s <port>\n\n",argv
[0]);
211 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
215 if (strcmp(argv
[1], "-m") == 0) {
219 // Make sure to initialize
220 struct main_loop_arg marg
= {
222 .script_cmds_file
= NULL
224 pthread_t main_loop_t
;
227 sp
= uart_open(argv
[1]);
228 if (sp
== INVALID_SERIAL_PORT
) {
229 printf("ERROR: invalid serial port\n");
230 marg
.usb_present
= 0;
232 } else if (sp
== CLAIMED_SERIAL_PORT
) {
233 printf("ERROR: serial port is claimed by another process\n");
234 marg
.usb_present
= 0;
237 marg
.usb_present
= 1;
241 // If the user passed the filename of the 'script' to execute, get it
242 if (argc
> 2 && argv
[2]) {
243 if (argv
[2][0] == 'f' && //buzzy, if a word 'flush' passed, flush the output after every log entry.
249 printf("Output will be flushed after every print.\n");
253 marg
.script_cmds_file
= argv
[2];
256 // create a mutex to avoid interlacing print commands from our different threads
257 pthread_mutex_init(&print_lock
, NULL
);
259 pthread_create(&main_loop_t
, NULL
, &main_loop
, &marg
);
260 InitGraphics(argc
, argv
);
264 pthread_join(main_loop_t
, NULL
);
270 pthread_mutex_destroy(&print_lock
);