]> git.zerfleddert.de Git - proxmark3-svn/blame - winsrc/gui.cpp
fixed an intermitent pain in the ass to track down crash in the GUI client due to...
[proxmark3-svn] / winsrc / gui.cpp
CommitLineData
6658905f 1//-----------------------------------------------------------------------------\r
2// Routines for the user interface when doing interactive things with prox\r
3// cards; this is basically a command line thing, in one window, and then\r
4// another window to do the graphs.\r
5// Jonathan Westhues, Sept 2005\r
6//-----------------------------------------------------------------------------\r
7#include <windows.h>\r
8#include <limits.h>\r
9#include <commctrl.h>\r
10#include <stdlib.h>\r
11#include <stdio.h>\r
12#include <math.h>\r
13\r
14#include "prox.h"\r
15\r
16#define oops() do { \\r
17 char line[100]; \\r
18 sprintf(line, "Internal error at line %d file '%s'", __LINE__, \\r
19 __FILE__); \\r
20 MessageBox(NULL, line, "Error", MB_ICONERROR); \\r
21 exit(-1); \\r
22} while(0)\r
23\r
24void dbp(char *str, ...)\r
25{\r
26 va_list f;\r
27 char buf[1024];\r
28 va_start(f, str);\r
29 vsprintf(buf, str, f);\r
30 OutputDebugString(buf);\r
31 OutputDebugString("\n");\r
32}\r
33\r
34int GraphBuffer[MAX_GRAPH_TRACE_LEN];\r
35int GraphTraceLen;\r
36\r
37HPEN GreyPen, GreenPen, WhitePen, YellowPen;\r
38HBRUSH GreenBrush, YellowBrush;\r
39\r
40static int GraphStart = 0;\r
41static double GraphPixelsPerPoint = 1;\r
42\r
43static int CursorAPos;\r
44static int CursorBPos;\r
45double CursorScaleFactor = 1.0;\r
46static HPEN CursorAPen;\r
47static HPEN CursorBPen;\r
48\r
49static HWND CommandWindow;\r
50static HWND GraphWindow;\r
51static HWND ScrollbackEdit;\r
52static HWND CommandEdit;\r
53\r
54#define COMMAND_HISTORY_MAX 16\r
55static char CommandHistory[COMMAND_HISTORY_MAX][256];\r
56static int CommandHistoryPos = -1;\r
57static int CommandHistoryNext;\r
58\r
59static HFONT MyFixedFont;\r
60#define FixedFont(x) SendMessage((x), WM_SETFONT, (WPARAM)MyFixedFont, TRUE)\r
61\r
62void ExecCmd(char *cmd)\r
63{\r
64\r
65}\r
66int CommandFinished;\r
30f2a7d3 67int offset = 64;\r
6658905f 68\r
69static void ResizeCommandWindow(void)\r
70{\r
71 int w, h;\r
72 RECT r;\r
73 GetClientRect(CommandWindow, &r);\r
74 w = r.right - r.left;\r
75 h = r.bottom - r.top;\r
76 MoveWindow(ScrollbackEdit, 10, 10, w - 20, h - 50, TRUE);\r
77 MoveWindow(CommandEdit, 10, h - 29, w - 20, 22, TRUE);\r
78}\r
79\r
80void RepaintGraphWindow(void)\r
81{\r
82 InvalidateRect(GraphWindow, NULL, TRUE);\r
83}\r
84\r
85static LRESULT CALLBACK\r
86 CommandWindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)\r
87{\r
88 switch (msg) {\r
89 case WM_DESTROY:\r
90 case WM_QUIT:\r
91 exit(0);\r
92 return 0;\r
93\r
94 case WM_SIZE:\r
95 ResizeCommandWindow();\r
96 return 0;\r
97\r
98 case WM_SETFOCUS:\r
99 SetFocus(CommandEdit);\r
100 break;\r
101\r
102 default:\r
103 return DefWindowProc(hwnd, msg, wParam, lParam);\r
104 }\r
105\r
106 return 1;\r
107}\r
108\r
109static void PaintGraph(HDC hdc)\r
110{\r
111 HBRUSH brush;\r
112 HPEN pen;\r
113\r
114 brush = GreenBrush;\r
115 pen = GreenPen;\r
116\r
117 if(GraphStart < 0) {\r
118 GraphStart = 0;\r
119 }\r
120\r
121 RECT r;\r
122 GetClientRect(GraphWindow, &r);\r
123\r
124 SelectObject(hdc, WhitePen);\r
125\r
30f2a7d3 126 MoveToEx(hdc, r.left + offset, r.top, NULL);\r
127 LineTo(hdc, r.left + offset, r.bottom);\r
6658905f 128\r
129 int zeroHeight = r.top + (r.bottom - r.top) / 2;\r
130 SelectObject(hdc, GreyPen);\r
131 MoveToEx(hdc, r.left, zeroHeight, NULL);\r
132 LineTo(hdc, r.right, zeroHeight);\r
133\r
134 int startMax =\r
30f2a7d3 135 (GraphTraceLen - (int)((r.right - r.left - offset) / GraphPixelsPerPoint));\r
6658905f 136 if(startMax < 0) {\r
137 startMax = 0;\r
138 }\r
139 if(GraphStart > startMax) {\r
140 GraphStart = startMax;\r
141 }\r
142\r
143 int absYMax = 1;\r
144\r
145 SelectObject(hdc, pen);\r
146\r
147 int i;\r
148 for(i = GraphStart; ; i++) {\r
149 if(i >= GraphTraceLen) {\r
150 break;\r
151 }\r
152 if(fabs((double)GraphBuffer[i]) > absYMax) {\r
153 absYMax = (int)fabs((double)GraphBuffer[i]);\r
154 }\r
30f2a7d3 155 int x = offset + (int)((i - GraphStart)*GraphPixelsPerPoint);\r
6658905f 156 if(x > r.right) {\r
157 break;\r
158 }\r
159 }\r
160\r
161 absYMax = (int)(absYMax*1.2 + 1);\r
162 SelectObject(hdc, MyFixedFont);\r
163 SetTextColor(hdc, RGB(255, 255, 255));\r
164 SetBkColor(hdc, RGB(0, 0, 0));\r
165\r
166 // number of points that will be plotted\r
30f2a7d3 167 double span = (int)((r.right - r.left) / GraphPixelsPerPoint);\r
168\r
169 // one label every offset pixels, let us say\r
170 int labels = (r.right - r.left - offset) / offset;\r
6658905f 171 if(labels <= 0) labels = 1;\r
30f2a7d3 172 // round to nearest power of 2\r
173 int pointsPerLabel = (int)(log(span / labels)/log(2.0));\r
6658905f 174 if(pointsPerLabel <= 0) pointsPerLabel = 1;\r
30f2a7d3 175 pointsPerLabel = (int)pow(2.0,pointsPerLabel);\r
6658905f 176\r
177 int yMin = INT_MAX;\r
178 int yMax = INT_MIN;\r
179 int yMean = 0;\r
180 int n = 0;\r
181\r
182 for(i = GraphStart; ; i++) {\r
183 if(i >= GraphTraceLen) {\r
184 break;\r
185 }\r
30f2a7d3 186 int x = offset + (int)((i - GraphStart)*GraphPixelsPerPoint);\r
6658905f 187 if(x > r.right + GraphPixelsPerPoint) {\r
188 break;\r
189 }\r
190\r
191 int y = GraphBuffer[i];\r
192 if(y < yMin) {\r
193 yMin = y;\r
194 }\r
195 if(y > yMax) {\r
196 yMax = y;\r
197 }\r
198 yMean += y;\r
199 n++;\r
200\r
201 y = (y * (r.top - r.bottom) / (2*absYMax)) + zeroHeight;\r
202 if(i == GraphStart) {\r
203 MoveToEx(hdc, x, y, NULL);\r
204 } else {\r
205 LineTo(hdc, x, y);\r
206 }\r
207\r
208 if(GraphPixelsPerPoint > 10) {\r
209 RECT f;\r
210 f.left = x - 3;\r
211 f.top = y - 3;\r
212 f.right = x + 3;\r
213 f.bottom = y + 3;\r
214 FillRect(hdc, &f, brush);\r
215 }\r
216\r
217 if(((i - GraphStart) % pointsPerLabel == 0) && i != GraphStart) {\r
218 SelectObject(hdc, WhitePen);\r
30f2a7d3 219 MoveToEx(hdc, x, zeroHeight - 8, NULL);\r
220 LineTo(hdc, x, zeroHeight + 8);\r
6658905f 221\r
222 char str[100];\r
223 sprintf(str, "+%d", (i - GraphStart));\r
224 SIZE size;\r
225 GetTextExtentPoint32(hdc, str, strlen(str), &size);\r
226 TextOut(hdc, x - size.cx, zeroHeight + 8, str, strlen(str));\r
227\r
228 SelectObject(hdc, pen);\r
229 MoveToEx(hdc, x, y, NULL);\r
230 }\r
231\r
232 if(i == CursorAPos || i == CursorBPos) {\r
233 if(i == CursorAPos) {\r
234 SelectObject(hdc, CursorAPen);\r
235 } else {\r
236 SelectObject(hdc, CursorBPen);\r
237 }\r
238 MoveToEx(hdc, x, r.top, NULL);\r
239 LineTo(hdc, x, r.bottom);\r
240\r
241 SelectObject(hdc, pen);\r
242 MoveToEx(hdc, x, y, NULL);\r
243 }\r
244 }\r
245\r
246 if(n != 0) {\r
247 yMean /= n;\r
248 }\r
249\r
48f8a3d6 250 char str[200];\r
251\r
67b9b62a 252 sprintf(str, "@%d max=%d min=%d mean=%d n=%d/%d dt=%d [%.3f] zoom=%.3f CursorA=%d [%d] CursorB=%d [%d]",\r
6658905f 253 GraphStart, yMax, yMin, yMean, n, GraphTraceLen,\r
48f8a3d6 254 CursorBPos - CursorAPos, (CursorBPos - CursorAPos)/CursorScaleFactor, GraphPixelsPerPoint,\r
255 CursorAPos, GraphBuffer[CursorAPos], CursorBPos, GraphBuffer[CursorBPos]);\r
6658905f 256 TextOut(hdc, 50, r.bottom - 20, str, strlen(str));\r
257}\r
258\r
259static LRESULT CALLBACK\r
260 GraphWindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)\r
261{\r
262 switch (msg) {\r
263 case WM_DESTROY:\r
264 case WM_QUIT:\r
265 GraphWindow = NULL;\r
266 return DefWindowProc(hwnd, msg, wParam, lParam);\r
267\r
268 case WM_SIZE:\r
269 RepaintGraphWindow();\r
270 return 0;\r
271\r
272 case WM_PAINT: {\r
273 PAINTSTRUCT ps;\r
274 HDC hdc = BeginPaint(hwnd, &ps);\r
275 if(GraphStart < 0) {\r
276 GraphStart = 0;\r
277 }\r
278 // This draws the trace.\r
279 PaintGraph(hdc);\r
280 EndPaint(hwnd, &ps);\r
281 break;\r
282 }\r
283 case WM_KEYDOWN:\r
284 switch(wParam) {\r
285 case VK_DOWN:\r
30f2a7d3 286 if(GraphPixelsPerPoint <= 8) {\r
6658905f 287 GraphPixelsPerPoint *= 2;\r
288 }\r
289 break;\r
290\r
291 case VK_UP:\r
30f2a7d3 292 if(GraphPixelsPerPoint >= 0.01) {\r
6658905f 293 GraphPixelsPerPoint /= 2;\r
294 }\r
295 break;\r
296\r
297 case VK_RIGHT:\r
30f2a7d3 298 if(GraphPixelsPerPoint < 16) {\r
299 GraphStart += (int)(16 / GraphPixelsPerPoint);\r
6658905f 300 } else {\r
301 GraphStart++;\r
302 }\r
303 break;\r
304\r
305 case VK_LEFT:\r
30f2a7d3 306 if(GraphPixelsPerPoint < 16) {\r
307 GraphStart -= (int)(16 / GraphPixelsPerPoint);\r
6658905f 308 } else {\r
309 GraphStart--;\r
310 }\r
311 break;\r
312\r
313 default:\r
314 goto nopaint;\r
315 }\r
316 RepaintGraphWindow();\r
317nopaint:\r
318 break;\r
319\r
320 case WM_LBUTTONDOWN:\r
321 case WM_RBUTTONDOWN: {\r
322 int x = LOWORD(lParam);\r
30f2a7d3 323 x -= offset;\r
6658905f 324 x = (int)(x / GraphPixelsPerPoint);\r
325 x += GraphStart;\r
48f8a3d6 326\r
6658905f 327 if(msg == WM_LBUTTONDOWN) {\r
328 CursorAPos = x;\r
329 } else {\r
330 CursorBPos = x;\r
331 }\r
332 RepaintGraphWindow();\r
333 break;\r
334 }\r
335 default:\r
336 return DefWindowProc(hwnd, msg, wParam, lParam);\r
337 }\r
338\r
339 return 1;\r
340}\r
341\r
342void PrintToScrollback(char *fmt, ...)\r
343{\r
344 va_list f;\r
345 char str[1024];\r
346 strcpy(str, "\r\n");\r
347 va_start(f, fmt);\r
348 vsprintf(str+2, fmt, f);\r
349\r
350 static char TextBuf[1024*32];\r
351 SendMessage(ScrollbackEdit, WM_GETTEXT, (WPARAM)sizeof(TextBuf),\r
352 (LPARAM)TextBuf);\r
353\r
354 if(strlen(TextBuf) + strlen(str) + 1 <= sizeof(TextBuf)) {\r
355 strcat(TextBuf, str);\r
356 } else {\r
357 lstrcpyn(TextBuf, str, sizeof(TextBuf));\r
358 }\r
359\r
360 SendMessage(ScrollbackEdit, WM_SETTEXT, 0, (LPARAM)TextBuf);\r
361 SendMessage(ScrollbackEdit, EM_LINESCROLL, 0, (LPARAM)INT_MAX);\r
362}\r
363\r
364void ShowGraphWindow(void)\r
365{\r
366 if(GraphWindow) return;\r
367\r
368 GraphWindow = CreateWindowEx(0, "Graph", "graphed",\r
369 WS_OVERLAPPED | WS_BORDER | WS_MINIMIZEBOX | WS_SYSMENU |\r
370 WS_SIZEBOX | WS_VISIBLE, 200, 150, 600, 500, NULL, NULL, NULL,\r
371 NULL);\r
372 if(!GraphWindow) oops();\r
373}\r
374\r
375void HideGraphWindow(void)\r
376{\r
377 if(GraphWindow) {\r
378 DestroyWindow(GraphWindow);\r
379 GraphWindow = NULL;\r
380 }\r
381}\r
382\r
383static void SetCommandEditTo(char *str)\r
384{\r
385 SendMessage(CommandEdit, WM_SETTEXT, 0, (LPARAM)str);\r
386 SendMessage(CommandEdit, EM_SETSEL, strlen(str), strlen(str));\r
387}\r
388\r
9760414b 389void ShowGui()\r
6658905f 390{\r
391 WNDCLASSEX wc;\r
392 memset(&wc, 0, sizeof(wc));\r
393 wc.cbSize = sizeof(wc);\r
394\r
395 wc.style = CS_BYTEALIGNCLIENT | CS_BYTEALIGNWINDOW | CS_OWNDC;\r
396 wc.lpfnWndProc = (WNDPROC)CommandWindowProc;\r
397 wc.hInstance = NULL;\r
398 wc.hbrBackground = (HBRUSH)(COLOR_BTNSHADOW);\r
399 wc.lpszClassName = "Command";\r
400 wc.lpszMenuName = NULL;\r
401 wc.hCursor = LoadCursor(NULL, IDC_ARROW);\r
402\r
403 if(!RegisterClassEx(&wc)) oops();\r
404\r
405 wc.lpszClassName = "Graph";\r
406 wc.lpfnWndProc = (WNDPROC)GraphWindowProc;\r
407 wc.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);\r
408\r
409 if(!RegisterClassEx(&wc)) oops();\r
410\r
411 CommandWindow = CreateWindowEx(0, "Command", "prox",\r
412 WS_OVERLAPPED | WS_BORDER | WS_MINIMIZEBOX | WS_SYSMENU |\r
413 WS_SIZEBOX | WS_VISIBLE, 20, 20, 500, 400, NULL, NULL, NULL,\r
414 NULL);\r
415 if(!CommandWindow) oops();\r
416\r
417 ScrollbackEdit = CreateWindowEx(WS_EX_CLIENTEDGE, "edit", "",\r
418 WS_CHILD | WS_CLIPSIBLINGS | WS_VISIBLE | ES_MULTILINE |\r
419 ES_AUTOVSCROLL | WS_VSCROLL, 0, 0, 0, 0, CommandWindow, NULL,\r
420 NULL, NULL);\r
421\r
422 CommandEdit = CreateWindowEx(WS_EX_CLIENTEDGE, "edit", "",\r
423 WS_CHILD | WS_CLIPSIBLINGS | WS_TABSTOP | WS_VISIBLE |\r
424 ES_AUTOHSCROLL, 0, 0, 0, 0, CommandWindow, NULL, NULL, NULL);\r
425\r
426 MyFixedFont = CreateFont(14, 0, 0, 0, FW_REGULAR, FALSE, FALSE, FALSE,\r
427 ANSI_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY,\r
428 FF_DONTCARE, "Lucida Console");\r
429 if(!MyFixedFont)\r
430 MyFixedFont = (HFONT)GetStockObject(SYSTEM_FONT);\r
431\r
432 FixedFont(ScrollbackEdit);\r
433 FixedFont(CommandEdit);\r
434\r
435 ResizeCommandWindow();\r
436 SetFocus(CommandEdit);\r
437\r
438 PrintToScrollback(">> Started prox, built " __DATE__ " " __TIME__);\r
439 PrintToScrollback(">> Connected to device");\r
440\r
441 GreyPen = CreatePen(PS_SOLID, 1, RGB(100, 100, 100));\r
442 GreenPen = CreatePen(PS_SOLID, 1, RGB(100, 255, 100));\r
443 YellowPen = CreatePen(PS_SOLID, 1, RGB(255, 255, 0));\r
444 GreenBrush = CreateSolidBrush(RGB(100, 255, 100));\r
445 YellowBrush = CreateSolidBrush(RGB(255, 255, 0));\r
446 WhitePen = CreatePen(PS_SOLID, 1, RGB(255, 255, 255));\r
447\r
448 CursorAPen = CreatePen(PS_DASH, 1, RGB(255, 255, 0));\r
449 CursorBPen = CreatePen(PS_DASH, 1, RGB(255, 0, 255));\r
450\r
451 MSG msg;\r
452 for(;;) {\r
453 if(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {\r
454 if(msg.message == WM_KEYDOWN && msg.wParam == VK_RETURN) {\r
455 char got[1024];\r
456 SendMessage(CommandEdit, WM_GETTEXT, (WPARAM)sizeof(got),\r
457 (LPARAM)got);\r
458\r
459 if(strcmp(got, "cls")==0) {\r
460 SendMessage(ScrollbackEdit, WM_SETTEXT, 0, (LPARAM)"");\r
461 } else {\r
462 CommandReceived(got);\r
463 }\r
464 SendMessage(CommandEdit, WM_SETTEXT, 0, (LPARAM)"");\r
465\r
466 // Insert it into the command history, unless it is\r
467 // identical to the previous command in the history.\r
468 int prev = CommandHistoryNext - 1;\r
469 if(prev < 0) prev += COMMAND_HISTORY_MAX;\r
470 if(strcmp(CommandHistory[prev], got) != 0) {\r
471 strcpy(CommandHistory[CommandHistoryNext], got);\r
472 CommandHistoryNext++;\r
473 if(CommandHistoryNext == COMMAND_HISTORY_MAX) {\r
474 CommandHistoryNext = 0;\r
475 }\r
476 }\r
477 CommandHistoryPos = -1;\r
478 } else if(msg.message == WM_KEYDOWN && msg.wParam == VK_UP &&\r
479 msg.hwnd == CommandEdit)\r
480 {\r
481 if(CommandHistoryPos == -1) {\r
482 CommandHistoryPos = CommandHistoryNext;\r
483 }\r
484 CommandHistoryPos--;\r
485 if(CommandHistoryPos < 0) {\r
486 CommandHistoryPos = COMMAND_HISTORY_MAX-1;\r
487 }\r
488 SetCommandEditTo(CommandHistory[CommandHistoryPos]);\r
489 } else if(msg.message == WM_KEYDOWN && msg.wParam == VK_DOWN &&\r
490 msg.hwnd == CommandEdit)\r
491 {\r
492 CommandHistoryPos++;\r
493 if(CommandHistoryPos >= COMMAND_HISTORY_MAX) {\r
494 CommandHistoryPos = 0;\r
495 }\r
496 SetCommandEditTo(CommandHistory[CommandHistoryPos]);\r
497 } else if(msg.message == WM_KEYDOWN && msg.wParam == VK_ESCAPE &&\r
498 msg.hwnd == CommandEdit)\r
499 {\r
500 SendMessage(CommandEdit, WM_SETTEXT, 0, (LPARAM)"");\r
501 } else {\r
502 if(msg.message == WM_KEYDOWN) {\r
503 CommandHistoryPos = -1;\r
504 }\r
505 TranslateMessage(&msg);\r
506 DispatchMessage(&msg);\r
507 }\r
508 }\r
509\r
9760414b 510 if (!offline)\r
511 {\r
512 UsbCommand c;\r
513 if(ReceiveCommandPoll(&c))\r
514 UsbCommandReceived(&c);\r
6658905f 515 }\r
516\r
517 Sleep(10);\r
518 }\r
519}\r
Impressum, Datenschutz