2017-01-21 12:46:45 +01:00
|
|
|
/*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
2017-02-22 23:49:55 +01:00
|
|
|
*
|
|
|
|
* Additional permission under GNU GPL version 3 section 7
|
|
|
|
*
|
|
|
|
* If you modify this Program, or any covered work, by linking or combining
|
|
|
|
* it with OpenSSL (or a modified version of that library), containing parts
|
|
|
|
* covered by the terms of OpenSSL License and SSLeay License, the licensors
|
|
|
|
* of this Program grant you additional permission to convey the resulting work.
|
|
|
|
*
|
2017-01-21 12:46:45 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include "console.h"
|
|
|
|
#include <time.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <stdarg.h>
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include <windows.h>
|
|
|
|
|
|
|
|
int get_key()
|
|
|
|
{
|
|
|
|
DWORD mode, rd;
|
|
|
|
HANDLE h;
|
|
|
|
|
|
|
|
if ((h = GetStdHandle(STD_INPUT_HANDLE)) == NULL)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
GetConsoleMode( h, &mode );
|
|
|
|
SetConsoleMode( h, mode & ~(ENABLE_LINE_INPUT | ENABLE_ECHO_INPUT) );
|
|
|
|
|
|
|
|
int c = 0;
|
|
|
|
ReadConsole( h, &c, 1, &rd, NULL );
|
|
|
|
SetConsoleMode( h, mode );
|
|
|
|
|
|
|
|
return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_colour(out_colours cl)
|
|
|
|
{
|
|
|
|
WORD attr = 0;
|
|
|
|
|
|
|
|
switch(cl)
|
|
|
|
{
|
|
|
|
case K_RED:
|
|
|
|
attr = FOREGROUND_RED | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_GREEN:
|
|
|
|
attr = FOREGROUND_GREEN | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_BLUE:
|
|
|
|
attr = FOREGROUND_BLUE | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_YELLOW:
|
|
|
|
attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_CYAN:
|
|
|
|
attr = FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_MAGENTA:
|
|
|
|
attr = FOREGROUND_BLUE | FOREGROUND_RED | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
case K_WHITE:
|
|
|
|
attr = FOREGROUND_BLUE | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), attr);
|
|
|
|
}
|
|
|
|
|
|
|
|
void reset_colour()
|
|
|
|
{
|
|
|
|
SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);
|
|
|
|
}
|
|
|
|
|
|
|
|
#else
|
|
|
|
#include <termios.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
int get_key()
|
|
|
|
{
|
|
|
|
struct termios oldattr, newattr;
|
|
|
|
int ch;
|
|
|
|
tcgetattr( STDIN_FILENO, &oldattr );
|
|
|
|
newattr = oldattr;
|
|
|
|
newattr.c_lflag &= ~( ICANON | ECHO );
|
|
|
|
tcsetattr( STDIN_FILENO, TCSANOW, &newattr );
|
|
|
|
ch = getchar();
|
|
|
|
tcsetattr( STDIN_FILENO, TCSANOW, &oldattr );
|
|
|
|
return ch;
|
|
|
|
}
|
|
|
|
|
|
|
|
void set_colour(out_colours cl)
|
|
|
|
{
|
|
|
|
switch(cl)
|
|
|
|
{
|
|
|
|
case K_RED:
|
|
|
|
fputs("\x1B[1;31m", stdout);
|
|
|
|
break;
|
|
|
|
case K_GREEN:
|
|
|
|
fputs("\x1B[1;32m", stdout);
|
|
|
|
break;
|
|
|
|
case K_BLUE:
|
|
|
|
fputs("\x1B[1;34m", stdout);
|
|
|
|
break;
|
|
|
|
case K_YELLOW:
|
|
|
|
fputs("\x1B[1;33m", stdout);
|
|
|
|
break;
|
|
|
|
case K_CYAN:
|
|
|
|
fputs("\x1B[1;36m", stdout);
|
|
|
|
break;
|
|
|
|
case K_MAGENTA:
|
|
|
|
fputs("\x1B[1;35m", stdout);
|
|
|
|
break;
|
|
|
|
case K_WHITE:
|
|
|
|
fputs("\x1B[1;37m", stdout);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void reset_colour()
|
|
|
|
{
|
|
|
|
fputs("\x1B[0m", stdout);
|
|
|
|
}
|
|
|
|
#endif // _WIN32
|
|
|
|
|
|
|
|
inline void comp_localtime(const time_t* ctime, tm* stime)
|
|
|
|
{
|
|
|
|
#ifdef _WIN32
|
|
|
|
localtime_s(stime, ctime);
|
|
|
|
#else
|
|
|
|
localtime_r(ctime, stime);
|
|
|
|
#endif // __WIN32
|
|
|
|
}
|
|
|
|
|
|
|
|
printer* printer::oInst = nullptr;
|
|
|
|
|
|
|
|
printer::printer()
|
|
|
|
{
|
|
|
|
verbose_level = LINF;
|
2017-02-22 23:49:55 +01:00
|
|
|
logfile = nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool printer::open_logfile(const char* file)
|
|
|
|
{
|
|
|
|
logfile = fopen(file, "ab+");
|
|
|
|
return logfile != nullptr;
|
2017-01-21 12:46:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void printer::print_msg(verbosity verbose, const char* fmt, ...)
|
|
|
|
{
|
|
|
|
if(verbose > verbose_level)
|
|
|
|
return;
|
|
|
|
|
|
|
|
char buf[1024];
|
|
|
|
size_t bpos;
|
|
|
|
tm stime;
|
|
|
|
|
|
|
|
time_t now = time(nullptr);
|
|
|
|
comp_localtime(&now, &stime);
|
|
|
|
strftime(buf, sizeof(buf), "[%F %T] : ", &stime);
|
|
|
|
bpos = strlen(buf);
|
|
|
|
|
|
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
|
|
vsnprintf(buf+bpos, sizeof(buf)-bpos, fmt, args);
|
|
|
|
va_end(args);
|
|
|
|
bpos = strlen(buf);
|
|
|
|
|
|
|
|
if(bpos+2 >= sizeof(buf))
|
|
|
|
return;
|
|
|
|
|
|
|
|
buf[bpos] = '\n';
|
|
|
|
buf[bpos+1] = '\0';
|
|
|
|
|
|
|
|
std::unique_lock<std::mutex> lck(print_mutex);
|
|
|
|
fputs(buf, stdout);
|
2017-02-22 23:49:55 +01:00
|
|
|
|
|
|
|
if(logfile != nullptr)
|
|
|
|
{
|
|
|
|
fputs(buf, logfile);
|
|
|
|
fflush(logfile);
|
|
|
|
}
|
2017-01-21 12:46:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void printer::print_str(const char* str)
|
|
|
|
{
|
|
|
|
std::unique_lock<std::mutex> lck(print_mutex);
|
|
|
|
fputs(str, stdout);
|
2017-02-22 23:49:55 +01:00
|
|
|
|
|
|
|
if(logfile != nullptr)
|
|
|
|
{
|
|
|
|
fputs(str, logfile);
|
|
|
|
fflush(logfile);
|
|
|
|
}
|
2017-01-21 12:46:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
extern "C" void printer_print_msg(const char* fmt, ...)
|
|
|
|
{
|
|
|
|
char buf[1024];
|
|
|
|
size_t bpos;
|
|
|
|
tm stime;
|
|
|
|
|
|
|
|
time_t now = time(nullptr);
|
|
|
|
comp_localtime(&now, &stime);
|
|
|
|
strftime(buf, sizeof(buf), "[%F %T] : ", &stime);
|
|
|
|
bpos = strlen(buf);
|
|
|
|
|
|
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
|
|
vsnprintf(buf+bpos, sizeof(buf)-bpos, fmt, args);
|
|
|
|
va_end(args);
|
|
|
|
bpos = strlen(buf);
|
|
|
|
|
|
|
|
if(bpos+2 >= sizeof(buf))
|
|
|
|
return;
|
|
|
|
|
|
|
|
buf[bpos] = '\n';
|
|
|
|
buf[bpos+1] = '\0';
|
|
|
|
|
|
|
|
printer::inst()->print_str(buf);
|
|
|
|
}
|
2017-03-21 19:21:43 +01:00
|
|
|
|
|
|
|
extern "C" void printer_print_str(const char* str)
|
|
|
|
{
|
|
|
|
printer::inst()->print_str(buf);
|
|
|
|
}
|