]> cvs.zerfleddert.de Git - proxmark3-svn/blob - client/proxmark3.c
Properly exit usb thread (e.g. on Ctrl-D)
[proxmark3-svn] / client / proxmark3.c
1 #include <usb.h>
2 #include <stdio.h>
3 #include <unistd.h>
4 #include <stdlib.h>
5 #include <strings.h>
6 #include <string.h>
7 #include <errno.h>
8 #include <readline/readline.h>
9 #include <readline/history.h>
10 #include <pthread.h>
11
12 #include "prox.h"
13 #include "proxmark3.h"
14 #include "proxgui.h"
15
16 struct usb_receiver_arg {
17 int run;
18 };
19
20 struct main_loop_arg {
21 int usb_present;
22 };
23
24 static void *usb_receiver(void *targ) {
25 struct usb_receiver_arg *arg = (struct usb_receiver_arg*)targ;
26 UsbCommand cmdbuf;
27
28 while(arg->run) {
29 if (ReceiveCommandPoll(&cmdbuf)) {
30 int i;
31
32 for (i=0; i<strlen(PROXPROMPT); i++)
33 putchar(0x08);
34
35 UsbCommandReceived(&cmdbuf);
36 printf(PROXPROMPT);
37 fflush(NULL);
38 }
39 }
40
41 pthread_exit(NULL);
42 }
43
44 static void *main_loop(void *targ)
45 {
46 struct main_loop_arg *arg = (struct main_loop_arg*)targ;
47 struct usb_receiver_arg rarg;
48 char *cmd = NULL;
49 pthread_t reader_thread;
50
51 if (arg->usb_present == 1) {
52 rarg.run=1;
53 pthread_create(&reader_thread, NULL, &usb_receiver, &rarg);
54 }
55
56 while(1) {
57
58 cmd = readline(PROXPROMPT);
59 if (cmd) {
60 if (cmd[0] != 0x00) {
61 CommandReceived(cmd);
62 add_history(cmd);
63 }
64 free(cmd);
65 } else {
66 printf("\n");
67 break;
68 }
69 }
70
71 if (arg->usb_present == 1) {
72 rarg.run = 0;
73 pthread_join(reader_thread, NULL);
74 }
75
76 ExitGraphics();
77 pthread_exit(NULL);
78 }
79
80 int main(int argc, char **argv)
81 {
82 struct main_loop_arg marg;
83 pthread_t main_loop_t;
84 usb_init();
85
86 if (!(devh = OpenProxmark(1))) {
87 fprintf(stderr,"PROXMARK3: NOT FOUND!\n");
88 marg.usb_present = 0;
89 offline = 1;
90 } else {
91 marg.usb_present = 1;
92 offline = 0;
93 }
94
95 pthread_create(&main_loop_t, NULL, &main_loop, &marg);
96 InitGraphics(argc, argv);
97
98 MainGraphics();
99
100 pthread_join(main_loop_t, NULL);
101
102 if (marg.usb_present == 1) {
103 CloseProxmark();
104 }
105 return 0;
106 }
Impressum, Datenschutz