More string swapping and cleaning up main().

src/poly/main_win.cc

@@ -0,0 +1,105 @@
+/**
+ ******************************************************************************
+ * 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;
+}