windows-nt/Source/XPSP1/NT/ds/security/services/w32time/lib/localizedwprintf.cpp

198 lines
5 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
#include "pch.h"
#include <stdio.h>
#define ARRAYSIZE(a) (sizeof(a)/sizeof(a[0]))
extern HINSTANCE g_hThisModule;
HANDLE g_hStdout = NULL;
HRESULT InitializeConsoleOutput() {
g_hStdout = GetStdHandle(STD_OUTPUT_HANDLE);
if (INVALID_HANDLE_VALUE == g_hStdout) {
return HRESULT_FROM_WIN32(GetLastError());
}
return S_OK;
}
BOOL
FileIsConsole(
HANDLE fp
)
{
DWORD htype;
htype = GetFileType(fp);
htype &= ~FILE_TYPE_REMOTE;
return htype == FILE_TYPE_CHAR;
}
HRESULT
MyWriteConsole(
HANDLE fp,
LPWSTR lpBuffer,
DWORD cchBuffer
)
{
HRESULT hr;
LPSTR lpAnsiBuffer = NULL;
//
// Jump through hoops for output because:
//
// 1. printf() family chokes on international output (stops
// printing when it hits an unrecognized character)
//
// 2. WriteConsole() works great on international output but
// fails if the handle has been redirected (i.e., when the
// output is piped to a file)
//
// 3. WriteFile() works great when output is piped to a file
// but only knows about bytes, so Unicode characters are
// printed as two Ansi characters.
//
if (FileIsConsole(fp))
{
hr = WriteConsole(fp, lpBuffer, cchBuffer, &cchBuffer, NULL);
_JumpIfError(hr, error, "WriteConsole");
}
else
{
lpAnsiBuffer = (LPSTR) LocalAlloc(LPTR, cchBuffer * sizeof(WCHAR));
if (lpAnsiBuffer != NULL)
{
cchBuffer = WideCharToMultiByte(CP_OEMCP,
0,
lpBuffer,
cchBuffer,
lpAnsiBuffer,
cchBuffer * sizeof(WCHAR),
NULL,
NULL);
if (cchBuffer != 0)
{
if (!WriteFile(fp, lpAnsiBuffer, cchBuffer, &cchBuffer, NULL))
{
hr = GetLastError();
_JumpError(hr, error, "WriteFile");
}
}
else
{
hr = GetLastError();
_JumpError(hr, error, "WideCharToMultiByte");
}
}
}
error:
if (NULL != lpAnsiBuffer)
LocalFree(lpAnsiBuffer);
return hr;
}
HRESULT LocalizedWPrintf(UINT nResourceID) {
DWORD ccWritten;
DWORD dwRetval;
HRESULT hr;
WCHAR rgwszString[512];
dwRetval = LoadStringW(g_hThisModule, nResourceID, rgwszString, ARRAYSIZE(rgwszString));
if (0 == dwRetval) {
_JumpLastError(hr, error, "LoadStringW");
}
_Verify(512 > dwRetval, hr, error); // Shouldn't fill our buffer...
hr = MyWriteConsole(g_hStdout, rgwszString, dwRetval);
_JumpIfError(hr, error, "MyWriteConsole");
hr = S_OK; // All done!
error:
return hr;
}
HRESULT LocalizedWPrintf2(UINT nResourceID, LPWSTR pwszFormat, ...) {
DWORD ccWritten;
va_list args;
WCHAR pwszBuffer[1024];
HRESULT hr = LocalizedWPrintf(nResourceID);
_JumpIfError(hr, error, "LocalizedWPrintf");
va_start(args, pwszFormat);
_vsnwprintf(pwszBuffer, ARRAYSIZE(pwszBuffer), pwszFormat, args);
va_end(args);
hr = MyWriteConsole(g_hStdout, pwszBuffer, wcslen(pwszBuffer));
_JumpIfError(hr, error, "MyWriteConsole");
hr = S_OK;
error:
return hr;
}
// Same as LocalizedWPrintf, but adds a carriage return.
HRESULT LocalizedWPrintfCR(UINT nResourceID) {
HRESULT hr = LocalizedWPrintf(nResourceID);
wprintf(L"\n");
return hr;
}
VOID DisplayMsg(DWORD dwSource, DWORD dwMsgId, ... )
{
DWORD dwBytesWritten;
DWORD dwLen;
LPWSTR pwszDisplayBuffer = NULL;
va_list ap;
va_start(ap, dwMsgId);
dwLen = FormatMessageW(dwSource | FORMAT_MESSAGE_ALLOCATE_BUFFER,
NULL,
dwMsgId,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPWSTR)&pwszDisplayBuffer,
0,
&ap);
if (dwLen && pwszDisplayBuffer) {
MyWriteConsole(g_hStdout, pwszDisplayBuffer, dwLen);
}
if (NULL != pwszDisplayBuffer) { LocalFree(pwszDisplayBuffer); }
va_end(ap);
}
BOOL WriteMsg(DWORD dwSource, DWORD dwMsgId, LPWSTR *ppMsg, ...)
{
DWORD dwBytesWritten;
DWORD dwLen;
LPWSTR pwszDisplayBuffer = NULL;
va_list ap;
va_start(ap, ppMsg);
dwLen = FormatMessageW(dwSource | FORMAT_MESSAGE_ALLOCATE_BUFFER,
NULL,
dwMsgId,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPWSTR)ppMsg,
0,
&ap);
va_end(ap);
// 0 is the error return value of FormatMessage.
return (0 != dwLen);
}