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"
25 #include "cmdparser.h"
30 // a global mutex to prevent interlaced printing from different threads
31 pthread_mutex_t print_lock
;
33 static serial_port sp
;
34 static UsbCommand txcmd
;
35 volatile static bool txcmd_pending
= false;
37 void SendCommand(UsbCommand
*c
) {
39 printf("Sending %d bytes\n", sizeof(UsbCommand
));
43 PrintAndLog("Sending bytes to proxmark failed - offline");
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.
63 static void *uart_receiver(void *targ
) {
64 struct receiver_arg
*arg
= (struct receiver_arg
*)targ
;
69 rxlen
= sizeof(UsbCommand
);
70 if (uart_receive(sp
, prx
, &rxlen
)) {
72 if (((prx
-rx
) % sizeof(UsbCommand
)) != 0) {
75 cmd_count
= (prx
-rx
) / sizeof(UsbCommand
);
77 for (size_t i
= 0; i
< cmd_count
; i
++) {
78 UsbCommandReceived((UsbCommand
*)(rx
+(i
*sizeof(UsbCommand
))));
84 if (!uart_send(sp
, (byte_t
*) &txcmd
, sizeof(UsbCommand
))) {
85 PrintAndLog("Sending bytes to proxmark failed");
87 txcmd_pending
= false;
96 void main_loop(char *script_cmds_file
, bool usb_present
) {
97 struct receiver_arg rarg
;
99 pthread_t reader_thread
;
103 pthread_create(&reader_thread
, NULL
, &uart_receiver
, &rarg
);
104 // cache Version information now:
108 FILE *script_file
= NULL
;
109 char script_cmd_buf
[256]; // iceman, needs lua script the same file_path_buffer as the rest
111 if (script_cmds_file
) {
112 script_file
= fopen(script_cmds_file
, "r");
114 printf("using 'scripting' commands file %s\n", script_cmds_file
);
118 read_history(".history");
122 // If there is a script file
125 if (!fgets(script_cmd_buf
, sizeof(script_cmd_buf
), script_file
)) {
130 nl
= strrchr(script_cmd_buf
, '\r');
133 nl
= strrchr(script_cmd_buf
, '\n');
136 if ((cmd
= (char*) malloc(strlen(script_cmd_buf
) + 1)) != NULL
) {
137 memset(cmd
, 0, strlen(script_cmd_buf
));
138 strcpy(cmd
, script_cmd_buf
);
145 cmd
= readline(PROXPROMPT
);
150 while(cmd
[strlen(cmd
) - 1] == ' ')
151 cmd
[strlen(cmd
) - 1] = 0x00;
153 if (cmd
[0] != 0x00) {
154 int ret
= CommandReceived(cmd
);
156 if (ret
== 99) { // exit or quit
167 write_history(".history");
171 pthread_join(reader_thread
, NULL
);
181 static void dumpAllHelp(int markdown
)
183 printf("\n%sProxmark3 command dump%s\n\n",markdown
?"# ":"",markdown
?"":"\n======================");
184 printf("Some commands are available only if a Proxmark is actually connected.%s\n",markdown
?" ":"");
185 printf("Check column \"offline\" for their availability.\n");
187 command_t
*cmds
= getTopLevelCommandTable();
188 dumpCommandsRecursive(cmds
, markdown
);
191 static char *my_executable_path
= NULL
;
192 static char *my_executable_directory
= NULL
;
194 const char *get_my_executable_path(void)
196 return my_executable_path
;
199 const char *get_my_executable_directory(void)
201 return my_executable_directory
;
204 static void set_my_executable_path(void)
206 int path_length
= wai_getExecutablePath(NULL
, 0, NULL
);
207 if (path_length
!= -1) {
208 my_executable_path
= (char*)malloc(path_length
+ 1);
209 int dirname_length
= 0;
210 if (wai_getExecutablePath(my_executable_path
, path_length
, &dirname_length
) != -1) {
211 my_executable_path
[path_length
] = '\0';
212 my_executable_directory
= (char *)malloc(dirname_length
+ 2);
213 strncpy(my_executable_directory
, my_executable_path
, dirname_length
+1);
214 my_executable_directory
[dirname_length
+1] = '\0';
220 int main(int argc
, char* argv
[]) {
224 printf("syntax: %s <port>\n\n",argv
[0]);
225 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
226 printf("help: %s -h\n\n", argv
[0]);
227 printf("\tDump all interactive help at once\n");
228 printf("markdown: %s -m\n\n", argv
[0]);
229 printf("\tDump all interactive help at once in markdown syntax\n");
232 if (strcmp(argv
[1], "-h") == 0) {
233 printf("syntax: %s <port>\n\n",argv
[0]);
234 printf("\tLinux example:'%s /dev/ttyACM0'\n\n", argv
[0]);
238 if (strcmp(argv
[1], "-m") == 0) {
243 set_my_executable_path();
245 bool usb_present
= false;
246 char *script_cmds_file
= NULL
;
248 sp
= uart_open(argv
[1]);
249 if (sp
== INVALID_SERIAL_PORT
) {
250 printf("ERROR: invalid serial port\n");
253 } else if (sp
== CLAIMED_SERIAL_PORT
) {
254 printf("ERROR: serial port is claimed by another process\n");
262 // If the user passed the filename of the 'script' to execute, get it
263 if (argc
> 2 && argv
[2]) {
264 if (argv
[2][0] == 'f' && //buzzy, if a word 'flush' passed, flush the output after every log entry.
270 printf("Output will be flushed after every print.\n");
274 script_cmds_file
= argv
[2];
277 // create a mutex to avoid interlacing print commands from our different threads
278 pthread_mutex_init(&print_lock
, NULL
);
281 InitGraphics(argc
, argv
, script_cmds_file
, usb_present
);
284 main_loop(script_cmds_file
, usb_present
);
293 pthread_mutex_destroy(&print_lock
);