#define OUTPUT_BUFFER_SIZE 4096
-
+/* Cache codepage for text streams */
UINT InputCodePage;
UINT OutputCodePage;
return GetConsoleMode(hHandle, &dwMode);
}
+
+
+/********************* Console STREAM IN utility functions ********************/
+
VOID ConInDisable(VOID)
{
HANDLE hInput = GetStdHandle(STD_INPUT_HANDLE);
SetConsoleMode(hInput, dwMode);
}
-
VOID ConInEnable(VOID)
{
HANDLE hInput = GetStdHandle(STD_INPUT_HANDLE);
SetConsoleMode(hInput, dwMode);
}
-
-VOID ConInFlush (VOID)
+VOID ConInFlush(VOID)
{
FlushConsoleInputBuffer(GetStdHandle(STD_INPUT_HANDLE));
}
-
VOID ConInKey(PINPUT_RECORD lpBuffer)
{
HANDLE hInput = GetStdHandle(STD_INPUT_HANDLE);
while (TRUE);
}
-
VOID ConInString(LPTSTR lpInput, DWORD dwLength)
{
DWORD dwOldMode;
SetConsoleMode(hFile, dwOldMode);
}
-static VOID ConWrite(TCHAR *str, DWORD len, DWORD nStdHandle)
+
+
+/******************** Console STREAM OUT utility functions ********************/
+
+static VOID ConWrite(DWORD nStdHandle, TCHAR *str, DWORD len)
{
DWORD dwNumBytes = 0;
HANDLE hOutput = GetStdHandle(nStdHandle);
}
}
-VOID ConOutChar(TCHAR c)
-{
- ConWrite(&c, 1, STD_OUTPUT_HANDLE);
-}
-
-VOID ConPuts(LPTSTR szText, DWORD nStdHandle)
+VOID ConPuts(DWORD nStdHandle, LPTSTR szText)
{
- ConWrite(szText, (DWORD)_tcslen(szText), nStdHandle);
+ ConWrite(nStdHandle, szText, (DWORD)_tcslen(szText));
}
-VOID ConOutResPaging(BOOL NewPage, UINT resID)
+VOID ConResPuts(DWORD nStdHandle, UINT resID)
{
TCHAR szMsg[RC_STRING_MAX_SIZE];
LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
- ConOutPrintfPaging(NewPage, szMsg);
+ ConPuts(nStdHandle, szMsg);
}
-VOID ConOutResPuts(UINT resID)
+VOID ConOutChar(TCHAR c)
{
- TCHAR szMsg[RC_STRING_MAX_SIZE];
- LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
- ConPuts(szMsg, STD_OUTPUT_HANDLE);
+ ConWrite(STD_OUTPUT_HANDLE, &c, 1);
}
-VOID ConOutPuts(LPTSTR szText)
+VOID ConErrChar(TCHAR c)
{
- ConPuts(szText, STD_OUTPUT_HANDLE);
+ ConWrite(STD_ERROR_HANDLE, &c, 1);
}
-
-VOID ConPrintf(LPTSTR szFormat, va_list arg_ptr, DWORD nStdHandle)
+VOID ConPrintfV(DWORD nStdHandle, LPTSTR szFormat, va_list arg_ptr)
{
TCHAR szOut[OUTPUT_BUFFER_SIZE];
DWORD len;
len = (DWORD)_vstprintf(szOut, szFormat, arg_ptr);
- ConWrite(szOut, len, nStdHandle);
+ ConWrite(nStdHandle, szOut, len);
+}
+
+VOID ConPrintf(DWORD nStdHandle, LPTSTR szFormat, ...)
+{
+ va_list arg_ptr;
+
+ va_start(arg_ptr, szFormat);
+ ConPrintfV(nStdHandle, szFormat, arg_ptr);
+ va_end(arg_ptr);
+}
+
+VOID ConResPrintf(DWORD nStdHandle, UINT resID, ...)
+{
+ TCHAR szMsg[RC_STRING_MAX_SIZE];
+ va_list arg_ptr;
+
+ va_start(arg_ptr, resID);
+ LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
+ ConPrintfV(nStdHandle, szMsg, arg_ptr);
+ va_end(arg_ptr);
+}
+
+VOID ConFormatMessage(DWORD nStdHandle, DWORD MessageId, ...)
+{
+ DWORD ret;
+ LPTSTR text;
+ va_list arg_ptr;
+
+ va_start(arg_ptr, MessageId);
+ ret = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
+ NULL,
+ MessageId,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ (LPTSTR)&text,
+ 0,
+ &arg_ptr);
+ va_end(arg_ptr);
+
+ if (ret > 0)
+ {
+ ConPuts(nStdHandle, text);
+ LocalFree(text);
+ }
+ else
+ {
+ ConResPrintf(nStdHandle, STRING_CONSOLE_ERROR, MessageId);
+ }
}
-INT ConPrintfPaging(BOOL NewPage, LPTSTR szFormat, va_list arg_ptr, DWORD nStdHandle)
+
+
+/************************** Console PAGER functions ***************************/
+
+BOOL ConPrintfVPaging(DWORD nStdHandle, BOOL NewPage, LPTSTR szFormat, va_list arg_ptr)
{
INT len;
CONSOLE_SCREEN_BUFFER_INFO csbi;
/* Reset LineCount and return if no string has been given */
if (szFormat == NULL)
- return 0;
+ return TRUE;
/* Get the size of the visual screen that can be printed to */
if (!IsConsoleHandle(hOutput) || !GetConsoleScreenBufferInfo(hOutput, &csbi))
{
/* We assume it's a file handle */
- ConPrintf(szFormat, arg_ptr, nStdHandle);
- return 0;
+ ConPrintfV(nStdHandle, szFormat, arg_ptr);
+ return TRUE;
}
/*
/* Make sure the user doesn't have the screen too small */
if (ScreenLines < 4)
{
- ConPrintf(szFormat, arg_ptr, nStdHandle);
- return 0;
+ ConPrintfV(nStdHandle, szFormat, arg_ptr);
+ return TRUE;
}
len = _vstprintf(szOut, szFormat, arg_ptr);
/* Prompt the user */
if (PagePrompt() != PROMPT_YES)
- {
- return 1;
- }
+ return FALSE;
// TODO: Recalculate 'ScreenLines' in case the user redimensions
// the window during the prompt.
WriteConsole(hOutput, &szOut[from], i-from, &dwWritten, NULL);
- return 0;
-}
-
-VOID ConErrFormatMessage(DWORD MessageId, ...)
-{
- TCHAR szMsg[RC_STRING_MAX_SIZE];
- DWORD ret;
- LPTSTR text;
- va_list arg_ptr;
-
- va_start(arg_ptr, MessageId);
- ret = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
- NULL,
- MessageId,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- (LPTSTR) &text,
- 0,
- &arg_ptr);
-
- va_end(arg_ptr);
- if (ret > 0)
- {
- ConErrPuts(text);
- LocalFree(text);
- }
- else
- {
- LoadString(CMD_ModuleHandle, STRING_CONSOLE_ERROR, szMsg, ARRAYSIZE(szMsg));
- ConErrPrintf(szMsg);
- }
-}
-
-VOID ConOutFormatMessage(DWORD MessageId, ...)
-{
- TCHAR szMsg[RC_STRING_MAX_SIZE];
- DWORD ret;
- LPTSTR text;
- va_list arg_ptr;
-
- va_start(arg_ptr, MessageId);
- ret = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
- NULL,
- MessageId,
- MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- (LPTSTR) &text,
- 0,
- &arg_ptr);
-
- va_end(arg_ptr);
- if (ret > 0)
- {
- ConErrPuts(text);
- LocalFree(text);
- }
- else
- {
- LoadString(CMD_ModuleHandle, STRING_CONSOLE_ERROR, szMsg, ARRAYSIZE(szMsg));
- ConErrPrintf(szMsg);
- }
-}
-
-VOID ConOutResPrintf(UINT resID, ...)
-{
- TCHAR szMsg[RC_STRING_MAX_SIZE];
- va_list arg_ptr;
-
- va_start(arg_ptr, resID);
- LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
- ConPrintf(szMsg, arg_ptr, STD_OUTPUT_HANDLE);
- va_end(arg_ptr);
-}
-
-VOID ConOutPrintf(LPTSTR szFormat, ...)
-{
- va_list arg_ptr;
-
- va_start(arg_ptr, szFormat);
- ConPrintf(szFormat, arg_ptr, STD_OUTPUT_HANDLE);
- va_end(arg_ptr);
+ return TRUE;
}
-INT ConOutPrintfPaging(BOOL NewPage, LPTSTR szFormat, ...)
+BOOL ConOutPrintfPaging(BOOL NewPage, LPTSTR szFormat, ...)
{
- INT iReturn;
+ BOOL bRet;
va_list arg_ptr;
va_start(arg_ptr, szFormat);
- iReturn = ConPrintfPaging(NewPage, szFormat, arg_ptr, STD_OUTPUT_HANDLE);
+ bRet = ConPrintfVPaging(STD_OUTPUT_HANDLE, NewPage, szFormat, arg_ptr);
va_end(arg_ptr);
- return iReturn;
-}
-
-VOID ConErrChar(TCHAR c)
-{
- ConWrite(&c, 1, STD_ERROR_HANDLE);
-}
-
-
-VOID ConErrResPuts(UINT resID)
-{
- TCHAR szMsg[RC_STRING_MAX_SIZE];
- LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
- ConPuts(szMsg, STD_ERROR_HANDLE);
+ return bRet;
}
-VOID ConErrPuts(LPTSTR szText)
-{
- ConPuts(szText, STD_ERROR_HANDLE);
-}
-
-
-VOID ConErrResPrintf(UINT resID, ...)
+VOID ConOutResPaging(BOOL NewPage, UINT resID)
{
TCHAR szMsg[RC_STRING_MAX_SIZE];
- va_list arg_ptr;
-
- va_start(arg_ptr, resID);
LoadString(CMD_ModuleHandle, resID, szMsg, ARRAYSIZE(szMsg));
- ConPrintf(szMsg, arg_ptr, STD_ERROR_HANDLE);
- va_end(arg_ptr);
+ ConOutPrintfPaging(NewPage, szMsg);
}
-VOID ConErrPrintf(LPTSTR szFormat, ...)
-{
- va_list arg_ptr;
- va_start(arg_ptr, szFormat);
- ConPrintf(szFormat, arg_ptr, STD_ERROR_HANDLE);
- va_end(arg_ptr);
-}
+/************************** Console SCREEN functions **************************/
VOID SetCursorXY(SHORT x, SHORT y)
{
+
+BOOL ConSetTitle(IN LPCTSTR lpConsoleTitle)
+{
+ /* Now really set the console title */
+ return SetConsoleTitle(lpConsoleTitle);
+}
+
#ifdef INCLUDE_CMD_BEEP
VOID ConRingBell(HANDLE hOutput)
{
#endif
#ifdef INCLUDE_CMD_COLOR
-BOOL ConSetScreenColor(WORD wColor, BOOL bFill)
+BOOL ConSetScreenColor(HANDLE hOutput, WORD wColor, BOOL bFill)
{
- HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
DWORD dwWritten;
CONSOLE_SCREEN_BUFFER_INFO csbi;
COORD coPos;
/* Fill the whole background if needed */
if (bFill)
{
- GetConsoleScreenBufferInfo(hConsole, &csbi);
+ GetConsoleScreenBufferInfo(hOutput, &csbi);
coPos.X = 0;
coPos.Y = 0;
- FillConsoleOutputAttribute(hConsole,
+ FillConsoleOutputAttribute(hOutput,
wColor & 0x00FF,
csbi.dwSize.X * csbi.dwSize.Y,
coPos,
}
/* Set the text attribute */
- SetConsoleTextAttribute(hConsole, wColor & 0x00FF);
+ SetConsoleTextAttribute(hOutput, wColor & 0x00FF);
return TRUE;
}
#endif