128 lines
3.4 KiB
C
128 lines
3.4 KiB
C
/*
|
|
* Copyright 2019 Erich E. Hoover
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <stdlib.h>
|
|
#include "resource.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(chcp);
|
|
|
|
static void output_writeconsole(const WCHAR *str, DWORD wlen)
|
|
{
|
|
DWORD count, ret;
|
|
|
|
ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), str, wlen, &count, NULL);
|
|
if (!ret)
|
|
{
|
|
DWORD len;
|
|
char *msgA;
|
|
|
|
/* On Windows WriteConsoleW() fails if the output is redirected. So fall
|
|
* back to WriteFile(), assuming the console encoding is still the right
|
|
* one in that case.
|
|
*/
|
|
len = WideCharToMultiByte(GetConsoleOutputCP(), 0, str, wlen, NULL, 0, NULL, NULL);
|
|
msgA = malloc(len);
|
|
|
|
WideCharToMultiByte(GetConsoleOutputCP(), 0, str, wlen, msgA, len, NULL, NULL);
|
|
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE);
|
|
free(msgA);
|
|
}
|
|
}
|
|
|
|
static void output_formatstring(const WCHAR *fmt, va_list va_args)
|
|
{
|
|
WCHAR *str;
|
|
DWORD len;
|
|
|
|
len = FormatMessageW(FORMAT_MESSAGE_FROM_STRING|FORMAT_MESSAGE_ALLOCATE_BUFFER,
|
|
fmt, 0, 0, (WCHAR *)&str, 0, &va_args);
|
|
if (!len && GetLastError() != ERROR_NO_WORK_DONE)
|
|
{
|
|
WINE_FIXME("Could not format string: le=%lu, fmt=%s\n", GetLastError(), wine_dbgstr_w(fmt));
|
|
return;
|
|
}
|
|
output_writeconsole(str, len);
|
|
LocalFree(str);
|
|
}
|
|
|
|
static void WINAPIV output_message(unsigned int id, ...)
|
|
{
|
|
WCHAR *fmt = NULL;
|
|
int len;
|
|
va_list va_args;
|
|
|
|
if (!(len = LoadStringW(GetModuleHandleW(NULL), id, (WCHAR *)&fmt, 0)))
|
|
{
|
|
WINE_FIXME("LoadString failed with %ld\n", GetLastError());
|
|
return;
|
|
}
|
|
|
|
len++;
|
|
fmt = malloc(len * sizeof(WCHAR));
|
|
if (!fmt) return;
|
|
|
|
LoadStringW(GetModuleHandleW(NULL), id, fmt, len);
|
|
|
|
va_start(va_args, id);
|
|
output_formatstring(fmt, va_args);
|
|
va_end(va_args);
|
|
|
|
free(fmt);
|
|
}
|
|
|
|
int __cdecl wmain(int argc, WCHAR *argv[])
|
|
{
|
|
int i;
|
|
|
|
if (argc == 1)
|
|
{
|
|
output_message(STRING_ACTIVE_CODE_PAGE, GetConsoleCP());
|
|
return 0;
|
|
}
|
|
else if (argc == 2)
|
|
{
|
|
int codepage, success;
|
|
|
|
if (!lstrcmpW(argv[1], L"/?"))
|
|
{
|
|
output_message(STRING_USAGE);
|
|
return 0;
|
|
}
|
|
|
|
codepage = _wtoi(argv[1]);
|
|
success = SetConsoleCP(codepage) && SetConsoleOutputCP(codepage);
|
|
|
|
if (success)
|
|
output_message(STRING_ACTIVE_CODE_PAGE, codepage);
|
|
else
|
|
output_message(STRING_INVALID_CODE_PAGE);
|
|
|
|
return !success;
|
|
}
|
|
|
|
WINE_FIXME("unexpected arguments:");
|
|
for (i = 0; i < argc; i++)
|
|
WINE_FIXME(" %s", wine_dbgstr_w(argv[i]));
|
|
WINE_FIXME("\n");
|
|
|
|
return 0;
|
|
}
|