Global Metrics

Spaces Data

Minimal test - lines (11, 51)

Code

namespace mozilla {

static void AssignStdHandle(const char* aPath, const char* aMode, FILE* aStream,
                            DWORD aStdHandle) {
  // Visual Studio's _fileno() returns -2 for the standard
  // streams if they aren't associated with an output stream.
  const int fd = _fileno(aStream);
  if (fd == -2) {
    freopen(aPath, aMode, aStream);
    return;
  }
  if (fd < 0) {
    return;
  }

  const HANDLE handle = reinterpret_cast(_get_osfhandle(fd));
  if (handle == INVALID_HANDLE_VALUE) {
    return;
  }

  const HANDLE oldHandle = GetStdHandle(aStdHandle);
  if (handle == oldHandle) {
    return;
  }

  SetStdHandle(aStdHandle, handle);
}

// This code attaches the process to the appropriate console.
void UseParentConsole() {
  if (AttachConsole(ATTACH_PARENT_PROCESS)) {
    // Redirect the standard streams to the existing console, but
    // only if they haven't been redirected to a valid file.
    AssignStdHandle("CONOUT$", "w", stdout, STD_OUTPUT_HANDLE);
    // There is no CONERR$, so use CONOUT$ for stderr as well.
    AssignStdHandle("CONOUT$", "w", stderr, STD_ERROR_HANDLE);
    AssignStdHandle("CONIN$", "r", stdin, STD_INPUT_HANDLE);
  }
}

}  // namespace mozilla