cvs.zerfleddert.de Git - proxmark3-svn/blob - client/wingui.c

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