1 files changed, 69 insertions, 0 deletions
diff --git a/toolkit/xre/nsNativeAppSupportWin.cpp b/toolkit/xre/nsNativeAppSupportWin.cpp
new file mode 100644
index 0000000000..02c2d39c38
--- /dev/null
+++ b/toolkit/xre/nsNativeAppSupportWin.cpp
@@ -0,0 +1,69 @@
+/* -*- 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/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() {
+  for (int i = 1; i < gArgc; ++i) {
+    if (strcmp("-console", gArgv[i]) == 0 ||
+        strcmp("--console", gArgv[i]) == 0 ||
+        strcmp("/console", gArgv[i]) == 0) {
+      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);
+        }
+      }
+    } else if (strcmp("-attach-console", gArgv[i]) == 0 ||
+               strcmp("--attach-console", gArgv[i]) == 0 ||
+               strcmp("/attach-console", gArgv[i]) == 0) {
+      UseParentConsole();
+    }
+  }
+}
+
+// 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;
+}