git.zerfleddert.de Git - proxmark3-svn/blob

1 //-----------------------------------------------------------------------------

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 //-----------------------------------------------------------------------------

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

22 #include "ui.h"

24 double CursorScaleFactor = 1;

 int PlotGridX=0, PlotGridY=0, PlotGridXdefault= 64, PlotGridYdefault= 64, CursorCPos= 0, CursorDPos= 0;

 bool flushAfterWrite = false;  //buzzy

27 int GridOffset = 0;

28 bool GridLocked = false;

29 bool showDemod = true;

 static char *logfilename = "proxmark3.log";

33 #ifndef EXTERNAL_PRINTANDLOG

34 static pthread_mutex_t print_lock = PTHREAD_MUTEX_INITIALIZER;

 void PrintAndLogEx(logLevel_t level, char *fmt, ...) {

38 // skip debug messages if client debugging is turned off i.e. 'DATA SETDEBUG 0'

39 // if (g_debugMode == 0 && level == DEBUG)

40 // return;

         char buffer[MAX_PRINT_BUFFER] = {0};

         char buffer2[MAX_PRINT_BUFFER] = {0};

         char prefix[20] = {0};

45 char *token = NULL;

46 int size = 0;

47 // {NORMAL, SUCCESS, INFO, FAILED, WARNING, ERR, DEBUG}

         static char *prefixes[7] = { "", "", "INFO: ", "FAILED: ", "WARNING: ", "ERROR: ", "#: "};

50 switch( level ) {

51 case FAILED:

                         strncpy(prefix,_RED_(FAILED: ), sizeof(prefix)-1);

53 break;

54 case DEBUG:

                         strncpy(prefix,_BLUE_(#: ), sizeof(prefix)-1);                  

56 break;

57 case SUCCESS:

                         strncpy(prefix,_GREEN_( ), sizeof(prefix)-1);

59 break;

60 case WARNING:

                         strncpy(prefix,_CYAN_(WARNING: ), sizeof(prefix)-1);

62 break;

63 default:

                         strncpy(prefix, prefixes[level], sizeof(prefix)-1);

65 break;

66 }

68 va_list args;

69 va_start(args, fmt);

         vsnprintf(buffer, sizeof(buffer), fmt, args);

71 va_end(args);

73 // no prefixes for normal

74 if ( level == NORMAL ) {

75 PrintAndLog(buffer);

76 return;

77 }

         if (strchr(buffer, '\n')) {

                 const char delim[2] = "\n";

83 // line starts with newline

                 if (buffer[0] == '\n') 

85 PrintAndLog("");

                 token = strtok(buffer, delim);

89 while (token != NULL) {

91 size = strlen(buffer2);

                         if (strlen(token))

                                 snprintf(buffer2+size, sizeof(buffer2)-size, "%s%s\n", prefix, token);

95 else

                                 snprintf(buffer2+size, sizeof(buffer2)-size, "\n");

                         token = strtok(NULL, delim);

99 }

100 PrintAndLog(buffer2);

101 } else {

                 snprintf(buffer2, sizeof(buffer2), "%s%.*s", prefix, MAX_PRINT_BUFFER - 20, buffer);

103 PrintAndLog(buffer2);

104 }

105 }

106

 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) {

                 logfile=fopen(logfilename, "a");

120 if (!logfile) {

                         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.

         int need_hack = (rl_readline_state & RL_STATE_READCMD) > 0;

132

133 if (need_hack) {

134 saved_point = rl_point;

                 saved_line = rl_copy_text(0, rl_end);

136 rl_save_prompt();

                 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);

         printf("          "); // cleaning prompt

146 va_end(argptr);

         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();

                 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) {

                 vfprintf(logfile, fmt, argptr2);

                 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

 void SetLogFilename(char *fn)

177 {

178 logfilename = fn;

179 }

180

 void SetFlushAfterWrite(bool flush_after_write) {

182 flushAfterWrite = flush_after_write;

183 }

184