Sweden-Number/programs/whoami/main.c

96 lines
2.8 KiB
C

/*
* Copyright 2020 Brendan Shanks for CodeWeavers
*
* 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
*/
#define SECURITY_WIN32
#include <windows.h>
#include <security.h>
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(whoami);
static int output_write(const WCHAR* str, int len)
{
DWORD ret, count;
ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), str, len, &count, NULL);
if (!ret)
{
DWORD lenA;
char* strA;
/* 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.
*/
lenA = WideCharToMultiByte(GetConsoleOutputCP(), 0, str, len,
NULL, 0, NULL, NULL);
strA = HeapAlloc(GetProcessHeap(), 0, lenA);
if (!strA)
return 0;
WideCharToMultiByte(GetConsoleOutputCP(), 0, str, len, strA, lenA,
NULL, NULL);
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), strA, lenA, &count, FALSE);
HeapFree(GetProcessHeap(), 0, strA);
}
return count;
}
int __cdecl wmain(int argc, WCHAR *argv[])
{
WCHAR *buf = NULL;
ULONG size = 0;
BOOL result;
if (argc > 1)
{
int i;
WINE_FIXME("unsupported arguments:");
for (i = 0; i < argc; i++)
WINE_FIXME(" %s", wine_dbgstr_w(argv[i]));
WINE_FIXME("\n");
}
result = GetUserNameExW(NameSamCompatible, NULL, &size);
if (result || GetLastError() != ERROR_MORE_DATA)
{
WINE_ERR("GetUserNameExW failed, result %d, error %d\n", result, GetLastError());
return 1;
}
buf = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
if (!buf)
{
WINE_ERR("Memory allocation failed\n");
return 1;
}
result = GetUserNameExW(NameSamCompatible, buf, &size);
if (result)
{
output_write(buf, size);
output_write(L"\r\n", 2);
}
else
WINE_ERR("GetUserNameExW failed, error %d\n", GetLastError());
HeapFree(GetProcessHeap(), 0, buf);
return 0;
}