1 files changed, 66 insertions, 0 deletions
diff --git a/toolkit/xre/nsNativeAppSupportWin.cpp b/toolkit/xre/nsNativeAppSupportWin.cpp
new file mode 100644
index 0000000000..8c4b43b3f6
--- /dev/null
+++ b/toolkit/xre/nsNativeAppSupportWin.cpp
@@ -0,0 +1,66 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "nsNativeAppSupportBase.h"
+#include "nsNativeAppSupportWin.h"
+
+#include "mozilla/CmdLineAndEnvUtils.h"
+#include "mozilla/WindowsConsole.h"
+
+using namespace mozilla;
+
+/*
+ * This code attaches the process to the appropriate console.
+ */
+
+class nsNativeAppSupportWin : public nsNativeAppSupportBase {
+ public:
+  // Utility function to handle a Win32-specific command line
+  // option: "--console", which dynamically creates a Windows
+  // console.
+  void CheckConsole();
+
+ private:
+  ~nsNativeAppSupportWin() {}
+};  // nsNativeAppSupportWin
+
+void nsNativeAppSupportWin::CheckConsole() {
+  if (CheckArg(gArgc, gArgv, "attach-console") == ARG_FOUND) {
+    UseParentConsole();
+  }
+
+  if (CheckArg(gArgc, gArgv, "console") == ARG_FOUND) {
+    if (AllocConsole()) {
+      // Redirect the standard streams to the new console, but
+      // only if they haven't been redirected to a valid file.
+      // Visual Studio's _fileno() returns -2 for the standard
+      // streams if they aren't associated with an output stream.
+      if (_fileno(stdout) == -2) {
+        freopen("CONOUT$", "w", stdout);
+      }
+      // There is no CONERR$, so use CONOUT$ for stderr as well.
+      if (_fileno(stderr) == -2) {
+        freopen("CONOUT$", "w", stderr);
+      }
+      if (_fileno(stdin) == -2) {
+        freopen("CONIN$", "r", stdin);
+      }
+    }
+  }
+}
+
+// Create and return an instance of class nsNativeAppSupportWin.
+nsresult NS_CreateNativeAppSupport(nsINativeAppSupport** aResult) {
+  nsNativeAppSupportWin* pNative = new nsNativeAppSupportWin;
+  if (!pNative) return NS_ERROR_OUT_OF_MEMORY;
+
+  // Check for dynamic console creation request.
+  pNative->CheckConsole();
+
+  *aResult = pNative;
+  NS_ADDREF(*aResult);
+
+  return NS_OK;
+}