106 lines
3.2 KiB
C++
106 lines
3.2 KiB
C++
|
/**
|
||
|
|
******************************************************************************
|
||
|
|
* Xenia : Xbox 360 Emulator Research Project *
|
||
|
|
******************************************************************************
|
||
|
|
* Copyright 2014 Ben Vanik. All rights reserved. *
|
||
|
|
* Released under the BSD license - see LICENSE in the root for more details. *
|
||
|
|
******************************************************************************
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <poly/main.h>
|
||
|
|
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <io.h>
|
||
|
|
#include <shellapi.h>
|
||
|
|
|
||
|
|
#include <gflags/gflags.h>
|
||
|
|
#include <poly/string.h>
|
||
|
|
|
||
|
|
namespace poly {
|
||
|
|
|
||
|
|
bool has_console_attached_ = true;
|
||
|
|
|
||
|
|
bool has_console_attached() { return has_console_attached_; }
|
||
|
|
|
||
|
|
void AttachConsole() {
|
||
|
|
bool has_console = ::AttachConsole(ATTACH_PARENT_PROCESS) == TRUE;
|
||
|
|
if (!has_console) {
|
||
|
|
// We weren't launched from a console, so just return.
|
||
|
|
// We could alloc our own console, but meh:
|
||
|
|
// has_console = AllocConsole() == TRUE;
|
||
|
|
has_console_attached_ = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
has_console_attached_ = true;
|
||
|
|
|
||
|
|
auto std_handle = (intptr_t)GetStdHandle(STD_OUTPUT_HANDLE);
|
||
|
|
auto con_handle = _open_osfhandle(std_handle, _O_TEXT);
|
||
|
|
auto fp = _fdopen(con_handle, "w");
|
||
|
|
*stdout = *fp;
|
||
|
|
setvbuf(stdout, NULL, _IONBF, 0);
|
||
|
|
|
||
|
|
std_handle = (intptr_t)GetStdHandle(STD_ERROR_HANDLE);
|
||
|
|
con_handle = _open_osfhandle(std_handle, _O_TEXT);
|
||
|
|
fp = _fdopen(con_handle, "w");
|
||
|
|
*stderr = *fp;
|
||
|
|
setvbuf(stderr, NULL, _IONBF, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace poly
|
||
|
|
|
||
|
|
// Used in console mode apps; automatically picked based on subsystem.
|
||
|
|
int wmain(int argc, wchar_t* argv[]) {
|
||
|
|
auto entry_info = poly::GetEntryInfo();
|
||
|
|
|
||
|
|
google::SetUsageMessage(std::string("usage: ") +
|
||
|
|
poly::to_string(entry_info.usage));
|
||
|
|
google::SetVersionString("1.0");
|
||
|
|
|
||
|
|
// Convert all args to narrow, as gflags doesn't support wchar.
|
||
|
|
int argca = argc;
|
||
|
|
char** argva = (char**)alloca(sizeof(char*) * argca);
|
||
|
|
for (int n = 0; n < argca; n++) {
|
||
|
|
size_t len = wcslen(argv[n]);
|
||
|
|
argva[n] = (char*)alloca(len + 1);
|
||
|
|
wcstombs_s(NULL, argva[n], len + 1, argv[n], _TRUNCATE);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Parse flags; this may delete some of them.
|
||
|
|
google::ParseCommandLineFlags(&argc, &argva, true);
|
||
|
|
|
||
|
|
// Widen all remaining flags and convert to usable strings.
|
||
|
|
std::vector<std::wstring> args;
|
||
|
|
for (int n = 0; n < argc; n++) {
|
||
|
|
args.push_back(poly::to_wstring(argva[n]));
|
||
|
|
}
|
||
|
|
|
||
|
|
// Call app-provided entry point.
|
||
|
|
int result = entry_info.entry_point(args);
|
||
|
|
|
||
|
|
google::ShutDownCommandLineFlags();
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Used in windowed apps; automatically picked based on subsystem.
|
||
|
|
int WINAPI wWinMain(HINSTANCE, HINSTANCE, LPWSTR command_line, int) {
|
||
|
|
// Attach a console so we can write output to stdout. If the user hasn't
|
||
|
|
// redirected output themselves it'll pop up a window.
|
||
|
|
poly::AttachConsole();
|
||
|
|
|
||
|
|
auto entry_info = poly::GetEntryInfo();
|
||
|
|
|
||
|
|
// Convert to an argv-like format so we can share code/use gflags.
|
||
|
|
std::wstring buffer = entry_info.name + L" " + command_line;
|
||
|
|
int argc;
|
||
|
|
wchar_t** argv = CommandLineToArgvW(buffer.c_str(), &argc);
|
||
|
|
if (!argv) {
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run normal entry point.
|
||
|
|
int result = wmain(argc, argv);
|
||
|
|
|
||
|
|
LocalFree(argv);
|
||
|
|
return result;
|
||
|
|
}
|