| 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 | // UI utilities |
| 10 | //----------------------------------------------------------------------------- |
| 11 | |
| 12 | #include <stdbool.h> |
| 13 | #ifndef EXTERNAL_PRINTANDLOG |
| 14 | #include <stdlib.h> |
| 15 | #include <stdio.h> |
| 16 | #include <stdarg.h> |
| 17 | #include <readline/readline.h> |
| 18 | #include <pthread.h> |
| 19 | #include "util.h" |
| 20 | #endif |
| 21 | |
| 22 | #include "ui.h" |
| 23 | |
| 24 | double CursorScaleFactor = 1; |
| 25 | int PlotGridX=0, PlotGridY=0, PlotGridXdefault= 64, PlotGridYdefault= 64, CursorCPos= 0, CursorDPos= 0; |
| 26 | bool flushAfterWrite = false; //buzzy |
| 27 | int GridOffset = 0; |
| 28 | bool GridLocked = false; |
| 29 | bool showDemod = true; |
| 30 | |
| 31 | static char *logfilename = "proxmark3.log"; |
| 32 | |
| 33 | #ifndef EXTERNAL_PRINTANDLOG |
| 34 | static pthread_mutex_t print_lock = PTHREAD_MUTEX_INITIALIZER; |
| 35 | |
| 36 | void PrintAndLogEx(logLevel_t level, char *fmt, ...) { |
| 37 | |
| 38 | // skip debug messages if client debugging is turned off i.e. 'DATA SETDEBUG 0' |
| 39 | // if (g_debugMode == 0 && level == DEBUG) |
| 40 | // return; |
| 41 | |
| 42 | char buffer[MAX_PRINT_BUFFER] = {0}; |
| 43 | char buffer2[MAX_PRINT_BUFFER] = {0}; |
| 44 | char prefix[20] = {0}; |
| 45 | char *token = NULL; |
| 46 | int size = 0; |
| 47 | // {NORMAL, SUCCESS, INFO, FAILED, WARNING, ERR, DEBUG} |
| 48 | static char *prefixes[7] = { "", "", "INFO: ", "FAILED: ", "WARNING: ", "ERROR: ", "#: "}; |
| 49 | |
| 50 | switch( level ) { |
| 51 | case FAILED: |
| 52 | strncpy(prefix,_RED_(FAILED: ), sizeof(prefix)-1); |
| 53 | break; |
| 54 | case DEBUG: |
| 55 | strncpy(prefix,_BLUE_(#: ), sizeof(prefix)-1); |
| 56 | break; |
| 57 | case SUCCESS: |
| 58 | strncpy(prefix,_GREEN_( ), sizeof(prefix)-1); |
| 59 | break; |
| 60 | case WARNING: |
| 61 | strncpy(prefix,_CYAN_(WARNING: ), sizeof(prefix)-1); |
| 62 | break; |
| 63 | default: |
| 64 | strncpy(prefix, prefixes[level], sizeof(prefix)-1); |
| 65 | break; |
| 66 | } |
| 67 | |
| 68 | va_list args; |
| 69 | va_start(args, fmt); |
| 70 | vsnprintf(buffer, sizeof(buffer), fmt, args); |
| 71 | va_end(args); |
| 72 | |
| 73 | // no prefixes for normal |
| 74 | if ( level == NORMAL ) { |
| 75 | PrintAndLog(buffer); |
| 76 | return; |
| 77 | } |
| 78 | |
| 79 | if (strchr(buffer, '\n')) { |
| 80 | |
| 81 | const char delim[2] = "\n"; |
| 82 | |
| 83 | // line starts with newline |
| 84 | if (buffer[0] == '\n') |
| 85 | PrintAndLog(""); |
| 86 | |
| 87 | token = strtok(buffer, delim); |
| 88 | |
| 89 | while (token != NULL) { |
| 90 | |
| 91 | size = strlen(buffer2); |
| 92 | |
| 93 | if (strlen(token)) |
| 94 | snprintf(buffer2+size, sizeof(buffer2)-size, "%s%s\n", prefix, token); |
| 95 | else |
| 96 | snprintf(buffer2+size, sizeof(buffer2)-size, "\n"); |
| 97 | |
| 98 | token = strtok(NULL, delim); |
| 99 | } |
| 100 | PrintAndLog(buffer2); |
| 101 | } else { |
| 102 | snprintf(buffer2, sizeof(buffer2), "%s%.*s", prefix, MAX_PRINT_BUFFER - 20, buffer); |
| 103 | PrintAndLog(buffer2); |
| 104 | } |
| 105 | } |
| 106 | |
| 107 | void PrintAndLog(char *fmt, ...) |
| 108 | { |
| 109 | char *saved_line; |
| 110 | int saved_point; |
| 111 | va_list argptr, argptr2; |
| 112 | static FILE *logfile = NULL; |
| 113 | static int logging=1; |
| 114 | |
| 115 | // lock this section to avoid interlacing prints from different threads |
| 116 | pthread_mutex_lock(&print_lock); |
| 117 | |
| 118 | if (logging && !logfile) { |
| 119 | logfile=fopen(logfilename, "a"); |
| 120 | if (!logfile) { |
| 121 | fprintf(stderr, "Can't open logfile, logging disabled!\n"); |
| 122 | logging=0; |
| 123 | } |
| 124 | } |
| 125 | |
| 126 | // If there is an incoming message from the hardware (eg: lf hid read) in |
| 127 | // the background (while the prompt is displayed and accepting user input), |
| 128 | // stash the prompt and bring it back later. |
| 129 | #ifdef RL_STATE_READCMD |
| 130 | // We are using GNU readline. libedit (OSX) doesn't support this flag. |
| 131 | int need_hack = (rl_readline_state & RL_STATE_READCMD) > 0; |
| 132 | |
| 133 | if (need_hack) { |
| 134 | saved_point = rl_point; |
| 135 | saved_line = rl_copy_text(0, rl_end); |
| 136 | rl_save_prompt(); |
| 137 | rl_replace_line("", 0); |
| 138 | rl_redisplay(); |
| 139 | } |
| 140 | #endif |
| 141 | |
| 142 | va_start(argptr, fmt); |
| 143 | va_copy(argptr2, argptr); |
| 144 | vprintf(fmt, argptr); |
| 145 | printf(" "); // cleaning prompt |
| 146 | va_end(argptr); |
| 147 | printf("\n"); |
| 148 | |
| 149 | #ifdef RL_STATE_READCMD |
| 150 | // We are using GNU readline. libedit (OSX) doesn't support this flag. |
| 151 | if (need_hack) { |
| 152 | rl_restore_prompt(); |
| 153 | rl_replace_line(saved_line, 0); |
| 154 | rl_point = saved_point; |
| 155 | rl_redisplay(); |
| 156 | free(saved_line); |
| 157 | } |
| 158 | #endif |
| 159 | |
| 160 | if (logging && logfile) { |
| 161 | vfprintf(logfile, fmt, argptr2); |
| 162 | fprintf(logfile,"\n"); |
| 163 | fflush(logfile); |
| 164 | } |
| 165 | va_end(argptr2); |
| 166 | |
| 167 | if (flushAfterWrite) //buzzy |
| 168 | { |
| 169 | fflush(NULL); |
| 170 | } |
| 171 | //release lock |
| 172 | pthread_mutex_unlock(&print_lock); |
| 173 | } |
| 174 | #endif |
| 175 | |
| 176 | void SetLogFilename(char *fn) |
| 177 | { |
| 178 | logfilename = fn; |
| 179 | } |
| 180 | |
| 181 | void SetFlushAfterWrite(bool flush_after_write) { |
| 182 | flushAfterWrite = flush_after_write; |
| 183 | } |
| 184 | |