From 4741e3581a76fb6b0e3f8550608388279de61006 Mon Sep 17 00:00:00 2001
From: Ben Vanik <ben.vanik@gmail.com>
Date: Thu, 16 Jan 2014 21:43:25 -0800
Subject: [PATCH] Support redirecting stdout/stderr to the launching console.

---
 src/xenia/platform.cc        | 25 +++++++++++++++++++++++++
 tools/xenia-run/xenia-run.cc |  3 +--
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/src/xenia/platform.cc b/src/xenia/platform.cc
index e4e53d0cc..bc1ca6e63 100644
--- a/src/xenia/platform.cc
+++ b/src/xenia/platform.cc
@@ -24,6 +24,8 @@ typedef int (*user_main_t)(int argc, xechar_t** argv);
 #if XE_LIKE_WIN32 && defined(UNICODE) && UNICODE
 
 #include <windows.h>
+#include <fcntl.h>
+#include <io.h>
 #include <shellapi.h>
 
 int xe_main_thunk(
@@ -56,9 +58,32 @@ int xe_main_thunk(
   return result;
 }
 
+void xe_attach_console() {
+  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;
+    return;
+  }
+
+  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);
+}
+
 int xe_main_window_thunk(
     wchar_t* command_line,
     void* user_main, const wchar_t* name, const char* usage) {
+  xe_attach_console();
   wchar_t buffer[2048];
   xestrcpy(buffer, XECOUNT(buffer), name);
   xestrcat(buffer, XECOUNT(buffer), XETEXT(" "));
diff --git a/tools/xenia-run/xenia-run.cc b/tools/xenia-run/xenia-run.cc
index f5f33d37f..ee6ad6fa8 100644
--- a/tools/xenia-run/xenia-run.cc
+++ b/tools/xenia-run/xenia-run.cc
@@ -87,5 +87,4 @@ XECLEANUP:
   delete emulator;
   return result_code;
 }
-XE_MAIN_THUNK(xenia_run, "xenia-run some.xex");
-//XE_MAIN_WINDOW_THUNK(xenia_run, XETEXT("xenia-run"), "xenia-run some.xex");
+XE_MAIN_WINDOW_THUNK(xenia_run, XETEXT("xenia-run"), "xenia-run some.xex");