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