]> cvs.zerfleddert.de Git - proxmark3-svn/blob - client/proxmark3.c
88cb5fa76db22a705172e9318fa566f9205b9a15
[proxmark3-svn] / 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 "util_posix.h"
22 #include "proxgui.h"
23 #include "cmdmain.h"
24 #include "ui.h"
25 #include "util.h"
26 #include "cmdparser.h"
27 #include "cmdhw.h"
28 #include "whereami.h"
29
30
31 void
32 #ifdef __has_attribute
33 #if __has_attribute(force_align_arg_pointer)
34 __attribute__((force_align_arg_pointer))
35 #endif
36 #endif
37 main_loop(char *script_cmds_file, char *script_cmd, bool usb_present) {
38 receiver_arg conn;
39 char *cmd = NULL;
40 pthread_t reader_thread;
41 bool execCommand = (script_cmd != NULL);
42 bool stdinOnPipe = !isatty(STDIN_FILENO);
43
44 memset(&conn, 0, sizeof(receiver_arg));
45
46 if (usb_present) {
47 conn.run = true;
48 SetOffline(false);
49 pthread_create(&reader_thread, NULL, &uart_receiver, &conn);
50 // cache Version information now:
51 CmdVersion(NULL);
52 } else {
53 SetOffline(true);
54 }
55
56 // file with script
57 FILE *script_file = NULL;
58 char script_cmd_buf[256] = {0}; // iceman, needs lua script the same file_path_buffer as the rest
59
60 if (script_cmds_file) {
61 script_file = fopen(script_cmds_file, "r");
62 if (script_file) {
63 printf("executing commands from file: %s\n", script_cmds_file);
64 }
65 }
66
67 read_history(".history");
68
69 while (1) {
70 // If there is a script file
71 if (script_file)
72 {
73 memset(script_cmd_buf, 0, sizeof(script_cmd_buf));
74 if (!fgets(script_cmd_buf, sizeof(script_cmd_buf), script_file)) {
75 fclose(script_file);
76 script_file = NULL;
77 } else {
78 strcleanrn(script_cmd_buf, sizeof(script_cmd_buf));
79
80 if ((cmd = strmcopy(script_cmd_buf)) != NULL) {
81 printf(PROXPROMPT"%s\n", cmd);
82 }
83 }
84 } else {
85 // If there is a script command
86 if (execCommand){
87 if ((cmd = strmcopy(script_cmd)) != NULL) {
88 printf(PROXPROMPT"%s\n", cmd);
89 }
90
91 execCommand = false;
92 } else {
93 // exit after exec command
94 if (script_cmd)
95 break;
96
97 // if there is a pipe from stdin
98 if (stdinOnPipe) {
99 memset(script_cmd_buf, 0, sizeof(script_cmd_buf));
100 if (!fgets(script_cmd_buf, sizeof(script_cmd_buf), stdin)) {
101 printf("\nStdin end. Exit...\n");
102 break;
103 }
104 strcleanrn(script_cmd_buf, sizeof(script_cmd_buf));
105
106 if ((cmd = strmcopy(script_cmd_buf)) != NULL) {
107 printf(PROXPROMPT"%s\n", cmd);
108 }
109
110 } else {
111 // read command from command prompt
112 cmd = readline(PROXPROMPT);
113 }
114 }
115 }
116
117 // execute command
118 if (cmd) {
119
120 while(cmd[strlen(cmd) - 1] == ' ')
121 cmd[strlen(cmd) - 1] = 0x00;
122
123 if (cmd[0] != 0x00) {
124 int ret = CommandReceived(cmd);
125 add_history(cmd);
126 if (ret == 99) { // exit or quit
127 break;
128 }
129 }
130 free(cmd);
131 cmd = NULL;
132 } else {
133 printf("\n");
134 break;
135 }
136 }
137
138 write_history(".history");
139
140 if (usb_present) {
141 conn.run = false;
142 pthread_join(reader_thread, NULL);
143 }
144
145 if (script_file) {
146 fclose(script_file);
147 script_file = NULL;
148 }
149 }
150
151 static void dumpAllHelp(int markdown)
152 {
153 printf("\n%sProxmark3 command dump%s\n\n",markdown?"# ":"",markdown?"":"\n======================");
154 printf("Some commands are available only if a Proxmark is actually connected.%s\n",markdown?" ":"");
155 printf("Check column \"offline\" for their availability.\n");
156 printf("\n");
157 command_t *cmds = getTopLevelCommandTable();
158 dumpCommandsRecursive(cmds, markdown);
159 }
160
161 static char *my_executable_path = NULL;
162 static char *my_executable_directory = NULL;
163
164 const char *get_my_executable_path(void)
165 {
166 return my_executable_path;
167 }
168
169 const char *get_my_executable_directory(void)
170 {
171 return my_executable_directory;
172 }
173
174 static void set_my_executable_path(void)
175 {
176 int path_length = wai_getExecutablePath(NULL, 0, NULL);
177 if (path_length != -1) {
178 my_executable_path = (char*)malloc(path_length + 1);
179 int dirname_length = 0;
180 if (wai_getExecutablePath(my_executable_path, path_length, &dirname_length) != -1) {
181 my_executable_path[path_length] = '\0';
182 my_executable_directory = (char *)malloc(dirname_length + 2);
183 strncpy(my_executable_directory, my_executable_path, dirname_length+1);
184 my_executable_directory[dirname_length+1] = '\0';
185 }
186 }
187 }
188
189 static void show_help(bool showFullHelp, char *command_line){
190 printf("syntax: %s <port> [-h|-help|-m|-f|-flush|-w|-wait|-c|-command|-l|-lua] [cmd_script_file_name] [command][lua_script_name]\n", command_line);
191 printf("\texample: %s "SERIAL_PORT_H"\n\n", command_line);
192
193 if (showFullHelp){
194 printf("help: <-h|-help> Dump all interactive command's help at once.\n");
195 printf("\t%s -h\n\n", command_line);
196 printf("markdown: <-m> Dump all interactive help at once in markdown syntax\n");
197 printf("\t%s -m\n\n", command_line);
198 printf("flush: <-f|-flush> Output will be flushed after every print.\n");
199 printf("\t%s -f\n\n", command_line);
200 printf("wait: <-w|-wait> 20sec waiting the serial port to appear in the OS\n");
201 printf("\t%s "SERIAL_PORT_H" -w\n\n", command_line);
202 printf("script: A script file with one proxmark3 command per line.\n\n");
203 printf("command: <-c|-command> Execute one proxmark3 command.\n");
204 printf("\t%s "SERIAL_PORT_H" -c \"hf mf chk 1* ?\"\n", command_line);
205 printf("\t%s "SERIAL_PORT_H" -command \"hf mf nested 1 *\"\n\n", command_line);
206 printf("lua: <-l|-lua> Execute lua script.\n");
207 printf("\t%s "SERIAL_PORT_H" -l hf_read\n\n", command_line);
208 }
209 }
210
211 int main(int argc, char* argv[]) {
212 srand(time(0));
213
214 bool usb_present = false;
215 bool waitCOMPort = false;
216 bool executeCommand = false;
217 bool addLuaExec = false;
218 char *script_cmds_file = NULL;
219 char *script_cmd = NULL;
220
221 if (argc < 2) {
222 show_help(true, argv[0]);
223 return 1;
224 }
225
226 for (int i = 1; i < argc; i++) {
227 if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i],"-help") == 0) {
228 show_help(false, argv[0]);
229 dumpAllHelp(0);
230 return 0;
231 }
232
233 if (strcmp(argv[i], "-m") == 0) {
234 dumpAllHelp(1);
235 return 0;
236 }
237
238 if(strcmp(argv[i],"-f") == 0 || strcmp(argv[i],"-flush") == 0){
239 printf("Output will be flushed after every print.\n");
240 SetFlushAfterWrite(true);
241 }
242
243 if(strcmp(argv[i],"-w") == 0 || strcmp(argv[i],"-wait") == 0){
244 waitCOMPort = true;
245 }
246
247 if(strcmp(argv[i],"-c") == 0 || strcmp(argv[i],"-command") == 0){
248 executeCommand = true;
249 }
250
251 if(strcmp(argv[i],"-l") == 0 || strcmp(argv[i],"-lua") == 0){
252 executeCommand = true;
253 addLuaExec = true;
254 }
255 }
256
257 // If the user passed the filename of the 'script' to execute, get it from last parameter
258 if (argc > 2 && argv[argc - 1] && argv[argc - 1][0] != '-') {
259 if (executeCommand){
260 script_cmd = argv[argc - 1];
261
262 while(script_cmd[strlen(script_cmd) - 1] == ' ')
263 script_cmd[strlen(script_cmd) - 1] = 0x00;
264
265 if (strlen(script_cmd) == 0) {
266 script_cmd = NULL;
267 } else {
268 if (addLuaExec){
269 // add "script run " to command
270 char *ctmp = NULL;
271 int len = strlen(script_cmd) + 11 + 1;
272 if ((ctmp = (char*) malloc(len)) != NULL) {
273 memset(ctmp, 0, len);
274 strcpy(ctmp, "script run ");
275 strcpy(&ctmp[11], script_cmd);
276 script_cmd = ctmp;
277 }
278 }
279
280 printf("Execute command from commandline: %s\n", script_cmd);
281 }
282 } else {
283 script_cmds_file = argv[argc - 1];
284 }
285 }
286
287 // check command
288 if (executeCommand && (!script_cmd || strlen(script_cmd) == 0)){
289 printf("ERROR: execute command: command not found.\n");
290 return 2;
291 }
292
293 // set global variables
294 set_my_executable_path();
295
296 // try to open USB connection to Proxmark
297 usb_present = OpenProxmark(argv[1], waitCOMPort, 20);
298
299 #ifdef HAVE_GUI
300 #ifdef _WIN32
301 InitGraphics(argc, argv, script_cmds_file, script_cmd, usb_present);
302 MainGraphics();
303 #else
304 char* display = getenv("DISPLAY");
305
306 if (display && strlen(display) > 1)
307 {
308 InitGraphics(argc, argv, script_cmds_file, script_cmd, usb_present);
309 MainGraphics();
310 }
311 else
312 {
313 main_loop(script_cmds_file, script_cmd, usb_present);
314 }
315 #endif
316 #else
317 main_loop(script_cmds_file, script_cmd, usb_present);
318 #endif
319
320 // Clean up the port
321 if (usb_present) {
322 CloseProxmark();
323 }
324
325 exit(0);
326 }
Impressum, Datenschutz