DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (409f3966645a)

VCS Links

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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/. */

// Similar to std::ifstream/ofstream, but takes char16ptr_t on Windows.
// Until C++17, std functions can only take char* filenames. So Unicode
// filenames were lost on Windows. To address this limitations, this wrapper
// uses proprietary wchar_t* overloads on MSVC, and __gnu_cxx::stdio_filebuf
// extension on MinGW. Once we can use C++17 filesystem API everywhere,
// we will be able to avoid this wrapper.

#ifndef mozilla_FStream_h
#define mozilla_FStream_h

#include "mozilla/Char16.h"
#include <istream>
#include <ostream>
#include <fstream>
#if defined(__MINGW32__) && defined(__GLIBCXX__)
#include "mozilla/UniquePtr.h"
#include <fcntl.h>
#include <ext/stdio_filebuf.h>
#endif

namespace mozilla {

#if defined(__MINGW32__) && defined(__GLIBCXX__)
// MinGW does not support wchar_t* overloads that are MSVC extension until
// C++17, so we have to implement widechar wrappers using a GNU extension.
class IFStream : public std::istream
{
public:
  explicit IFStream(char16ptr_t filename, openmode mode = in);

  std::filebuf* rdbuf() const { return mFileBuf.get(); }

  bool is_open() const { return mFileBuf && mFileBuf->is_open(); }
  void open(char16ptr_t filename, openmode mode = in);
  void close() { mFileBuf && mFileBuf->close(); }

private:
  UniquePtr<std::filebuf> mFileBuf;
};

inline
IFStream::IFStream(char16ptr_t filename, openmode mode)
  : std::istream(nullptr)
{
  open(filename, mode);
}

inline void
IFStream::open(char16ptr_t filename, openmode mode)
{
  int fmode = _O_RDONLY;
  if (mode & binary) {
    fmode |= _O_BINARY;
  } else {
    fmode |= _O_TEXT;
  }
  int fd = _wopen(filename, fmode);
  mFileBuf = MakeUnique<__gnu_cxx::stdio_filebuf<char>>(fd, mode);
  std::istream::rdbuf(mFileBuf.get());
}

class OFStream : public std::ostream
{
public:
  explicit OFStream(char16ptr_t filename, openmode mode = out);

  std::filebuf* rdbuf() const { return mFileBuf.get(); }

  bool is_open() const { return mFileBuf && mFileBuf->is_open(); }
  void open(char16ptr_t filename, openmode mode = out);
  void close() { mFileBuf && mFileBuf->close(); }

private:
  UniquePtr<std::filebuf> mFileBuf;
};

inline
OFStream::OFStream(char16ptr_t filename, openmode mode)
  : std::ostream(nullptr)
{
  open(filename, mode);
}

inline void
OFStream::open(char16ptr_t filename, openmode mode)
{
  int fmode = _O_WRONLY;
  if (mode & binary) {
    fmode |= _O_BINARY;
  } else {
    fmode |= _O_TEXT;
  }
  if (mode & trunc) {
    fmode |= _O_CREAT | _O_TRUNC;
  }
  int fd = _wopen(filename, fmode);
  mFileBuf = MakeUnique<__gnu_cxx::stdio_filebuf<char>>(fd, mode);
  std::ostream::rdbuf(mFileBuf.get());
}

#elif defined(XP_WIN)
class IFStream : public std::ifstream
{
public:
  explicit IFStream(char16ptr_t filename, openmode mode = in)
    : std::ifstream(filename, mode) {}

  void open(char16ptr_t filename, openmode mode = in)
  {
    std::ifstream::open(filename, mode);
  }
};

class OFStream : public std::ofstream
{
public:
  explicit OFStream(char16ptr_t filename, openmode mode = out)
    : std::ofstream(filename, mode) {}

  void open(char16ptr_t filename, openmode mode = out)
  {
    std::ofstream::open(filename, mode);
  }
};
#else
using IFStream = std::ifstream;
using OFStream = std::ofstream;
#endif

} // namespace mozilla

#endif /* mozilla_FStream_h */