0
0
Fork 0
mirror of https://git.do.srb2.org/STJr/SRB2.git synced 2024-12-23 11:10:56 +00:00
SRB2/tools/masterserver/common.cpp
2014-03-15 13:11:35 -04:00

207 lines
4.3 KiB
C++

// Emacs style mode select -*- C++ -*-
//-----------------------------------------------------------------------------
//
// Copyright (C) 2000 by DooM Legacy Team.
//
// 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 2
// of the License, or (at your option) 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.
//
//-----------------------------------------------------------------------------
#include <time.h>
#include <stdio.h>
#include <stdarg.h>
#include "common.h"
// ================================== GLOBALS =================================
static const char *fatal_error_msg[NUM_FATAL_ERROR] =
{
"Error: signal()",
"Error: select()",
"Error: read()",
"Error: write()",
};
// used by xxxPrintf() functions as temporary variable
static char str[1024] ="";
static va_list arglist;
#ifdef _WIN32
static size_t len = 0;
static HANDLE co = INVALID_HANDLE_VALUE;
static DWORD bytesWritten = 0;
static CONSOLE_SCREEN_BUFFER_INFO ConsoleScreenBufferInfo;
#endif
// ================================= FUNCTIONS ================================
/*
* clearScreen():
*/
void clearScreen()
{
#ifdef _WIN32
#else
printf("\033[0m \033[2J \033[0;0H");
#endif
}
/*
* fatalError():
*/
void fatalError(fatal_error_t num)
{
clearScreen();
printf("\n");
perror(fatal_error_msg[num]);
printf("\n");
exit(-1);
}
/*
* dbgPrintf():
*/
#ifdef _WIN32
void dbgPrintf(DWORDLONG col, const char *lpFmt, ...)
#else
void dbgPrintf(const char *col, const char *lpFmt, ...)
#endif
{
#if defined (__DEBUG__)
va_start(arglist, lpFmt);
vsnprintf(str, sizeof str, lpFmt, arglist);
va_end(arglist);
#ifdef _WIN32
len = strlen(str);
co = GetStdHandle(STD_OUTPUT_HANDLE);
if (co == INVALID_HANDLE_VALUE)
return;
if (col == DEFCOL)
{
if (!GetConsoleScreenBufferInfo(co, &ConsoleScreenBufferInfo))
ConsoleScreenBufferInfo.wAttributes =
FOREGROUND_RED|FOREGROUND_GREEN|FOREGROUND_BLUE;
SetConsoleTextAttribute(co, (WORD)col);
}
if (GetFileType(co) == FILE_TYPE_CHAR)
WriteConsoleA(co, str, (DWORD)len, &bytesWritten, NULL);
else
WriteFile(co, str, (DWORD)len, &bytesWritten, NULL);
if (col != DEFCOL)
SetConsoleTextAttribute(co,
ConsoleScreenBufferInfo.wAttributes);
#else
printf("%s%s", col, str);
fflush(stdout);
#endif
#else
(void)col;
(void)lpFmt;
#endif
}
/*
* conPrintf()
*/
#ifdef _WIN32
void conPrintf(DWORDLONG col, const char *lpFmt, ...)
#else
void conPrintf(const char *col, const char *lpFmt, ...)
#endif
{
va_start(arglist, lpFmt);
vsnprintf(str, sizeof str, lpFmt, arglist);
va_end(arglist);
#ifdef _WIN32
len = strlen(str);
co = GetStdHandle(STD_OUTPUT_HANDLE);
if (col == DEFCOL)
{
if (!GetConsoleScreenBufferInfo(co, &ConsoleScreenBufferInfo))
ConsoleScreenBufferInfo.wAttributes =
FOREGROUND_RED|FOREGROUND_GREEN|FOREGROUND_BLUE;
SetConsoleTextAttribute(co, (WORD)col);
}
if (co == INVALID_HANDLE_VALUE)
return;
if (GetFileType(co) == FILE_TYPE_CHAR)
WriteConsoleA(co, str, (DWORD)len, &bytesWritten, NULL);
else
WriteFile(co, str, (DWORD)len, &bytesWritten, NULL);
if (col != DEFCOL)
SetConsoleTextAttribute(co,
ConsoleScreenBufferInfo.wAttributes);
#else
printf("%s%s", col, str);
fflush(stdout);
#endif
}
/*
* logPrintf():
*/
void logPrintf(FILE *f, const char *lpFmt, ...)
{
char *ct;
time_t t;
va_start(arglist, lpFmt);
vsnprintf(str, sizeof str, lpFmt, arglist);
va_end(arglist);
t = time(NULL);
ct = ctime(&t);
ct[strlen(ct)-1] = '\0';
fprintf(f, "%s: %s", ct, str);
fflush(f);
#if defined (__DEBUG__)
#ifdef _WIN32
len = strlen(str);
printf("%s", str);
#else
printf("%s%s", DEFCOL, str);
#endif
fflush(stdout);
#endif
}
/*
* openFile():
*/
FILE *openFile(const char *filename)
{
return fopen(filename, "a+t");
}
void strrand(char *s, const int len) {
static const char alphanum[] =
"0123456789"
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"abcdefghijklmnopqrstuvwxyz";
for (int i = 0; i < len; ++i) {
s[i] = alphanum[rand() % (sizeof(alphanum) - 1)];
}
s[len] = 0;
}