123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439 |
- #ifndef FMT_OS_H_
- #define FMT_OS_H_
- #include "format.h"
- #ifndef FMT_MODULE
- # include <cerrno>
- # include <cstddef>
- # include <cstdio>
- # include <system_error> // std::system_error
- # if FMT_HAS_INCLUDE(<xlocale.h>)
- # include <xlocale.h> // LC_NUMERIC_MASK on macOS
- # endif
- #endif
- #ifndef FMT_USE_FCNTL
- # if FMT_HAS_INCLUDE("winapifamily.h")
- # include <winapifamily.h>
- # endif
- # if (FMT_HAS_INCLUDE(<fcntl.h>) || defined(__APPLE__) || \
- defined(__linux__)) && \
- (!defined(WINAPI_FAMILY) || \
- (WINAPI_FAMILY == WINAPI_FAMILY_DESKTOP_APP))
- # include <fcntl.h> // for O_RDONLY
- # define FMT_USE_FCNTL 1
- # else
- # define FMT_USE_FCNTL 0
- # endif
- #endif
- #ifndef FMT_POSIX
- # if defined(_WIN32) && !defined(__MINGW32__)
- # define FMT_POSIX(call) _##call
- # else
- # define FMT_POSIX(call) call
- # endif
- #endif
- #ifdef FMT_SYSTEM
- # define FMT_HAS_SYSTEM
- # define FMT_POSIX_CALL(call) FMT_SYSTEM(call)
- #else
- # define FMT_SYSTEM(call) ::call
- # ifdef _WIN32
- # define FMT_POSIX_CALL(call) ::_##call
- # else
- # define FMT_POSIX_CALL(call) ::call
- # endif
- #endif
- #ifndef _WIN32
- # define FMT_RETRY_VAL(result, expression, error_result) \
- do { \
- (result) = (expression); \
- } while ((result) == (error_result) && errno == EINTR)
- #else
- # define FMT_RETRY_VAL(result, expression, error_result) result = (expression)
- #endif
- #define FMT_RETRY(result, expression) FMT_RETRY_VAL(result, expression, -1)
- FMT_BEGIN_NAMESPACE
- FMT_BEGIN_EXPORT
- template <typename Char> class basic_cstring_view {
- private:
- const Char* data_;
- public:
-
- basic_cstring_view(const Char* s) : data_(s) {}
-
- basic_cstring_view(const std::basic_string<Char>& s) : data_(s.c_str()) {}
-
- auto c_str() const -> const Char* { return data_; }
- };
- using cstring_view = basic_cstring_view<char>;
- using wcstring_view = basic_cstring_view<wchar_t>;
- #ifdef _WIN32
- FMT_API const std::error_category& system_category() noexcept;
- namespace detail {
- FMT_API void format_windows_error(buffer<char>& out, int error_code,
- const char* message) noexcept;
- }
- FMT_API std::system_error vwindows_error(int error_code, string_view format_str,
- format_args args);
- template <typename... Args>
- std::system_error windows_error(int error_code, string_view message,
- const Args&... args) {
- return vwindows_error(error_code, message, fmt::make_format_args(args...));
- }
- FMT_API void report_windows_error(int error_code, const char* message) noexcept;
- #else
- inline auto system_category() noexcept -> const std::error_category& {
- return std::system_category();
- }
- #endif
- #ifdef __OSX__
- template <typename S, typename... Args, typename Char = char_t<S>>
- void say(const S& format_str, Args&&... args) {
- std::system(format("say \"{}\"", format(format_str, args...)).c_str());
- }
- #endif
- class buffered_file {
- private:
- FILE* file_;
- friend class file;
- explicit buffered_file(FILE* f) : file_(f) {}
- public:
- buffered_file(const buffered_file&) = delete;
- void operator=(const buffered_file&) = delete;
-
- buffered_file() noexcept : file_(nullptr) {}
-
- FMT_API ~buffered_file() noexcept;
- public:
- buffered_file(buffered_file&& other) noexcept : file_(other.file_) {
- other.file_ = nullptr;
- }
- auto operator=(buffered_file&& other) -> buffered_file& {
- close();
- file_ = other.file_;
- other.file_ = nullptr;
- return *this;
- }
-
- FMT_API buffered_file(cstring_view filename, cstring_view mode);
-
- FMT_API void close();
-
- auto get() const noexcept -> FILE* { return file_; }
- FMT_API auto descriptor() const -> int;
- template <typename... T>
- inline void print(string_view fmt, const T&... args) {
- const auto& vargs = fmt::make_format_args(args...);
- detail::is_locking<T...>() ? fmt::vprint_buffered(file_, fmt, vargs)
- : fmt::vprint(file_, fmt, vargs);
- }
- };
- #if FMT_USE_FCNTL
- class FMT_API file {
- private:
- int fd_;
-
- explicit file(int fd) : fd_(fd) {}
- friend struct pipe;
- public:
-
- enum {
- RDONLY = FMT_POSIX(O_RDONLY),
- WRONLY = FMT_POSIX(O_WRONLY),
- RDWR = FMT_POSIX(O_RDWR),
- CREATE = FMT_POSIX(O_CREAT),
- APPEND = FMT_POSIX(O_APPEND),
- TRUNC = FMT_POSIX(O_TRUNC)
- };
-
- file() noexcept : fd_(-1) {}
-
- file(cstring_view path, int oflag);
- public:
- file(const file&) = delete;
- void operator=(const file&) = delete;
- file(file&& other) noexcept : fd_(other.fd_) { other.fd_ = -1; }
-
- auto operator=(file&& other) -> file& {
- close();
- fd_ = other.fd_;
- other.fd_ = -1;
- return *this;
- }
-
- ~file() noexcept;
-
- auto descriptor() const noexcept -> int { return fd_; }
-
- void close();
-
-
- auto size() const -> long long;
-
- auto read(void* buffer, size_t count) -> size_t;
-
- auto write(const void* buffer, size_t count) -> size_t;
-
-
- static auto dup(int fd) -> file;
-
-
- void dup2(int fd);
-
-
- void dup2(int fd, std::error_code& ec) noexcept;
-
-
- auto fdopen(const char* mode) -> buffered_file;
- # if defined(_WIN32) && !defined(__MINGW32__)
-
-
- static file open_windows_file(wcstring_view path, int oflag);
- # endif
- };
- struct FMT_API pipe {
- file read_end;
- file write_end;
-
-
- pipe();
- };
- auto getpagesize() -> long;
- namespace detail {
- struct buffer_size {
- buffer_size() = default;
- size_t value = 0;
- auto operator=(size_t val) const -> buffer_size {
- auto bs = buffer_size();
- bs.value = val;
- return bs;
- }
- };
- struct ostream_params {
- int oflag = file::WRONLY | file::CREATE | file::TRUNC;
- size_t buffer_size = BUFSIZ > 32768 ? BUFSIZ : 32768;
- ostream_params() {}
- template <typename... T>
- ostream_params(T... params, int new_oflag) : ostream_params(params...) {
- oflag = new_oflag;
- }
- template <typename... T>
- ostream_params(T... params, detail::buffer_size bs)
- : ostream_params(params...) {
- this->buffer_size = bs.value;
- }
- # if defined(__INTEL_COMPILER) && __INTEL_COMPILER < 2000
- ostream_params(int new_oflag) : oflag(new_oflag) {}
- ostream_params(detail::buffer_size bs) : buffer_size(bs.value) {}
- # endif
- };
- class file_buffer final : public buffer<char> {
- private:
- file file_;
- FMT_API static void grow(buffer<char>& buf, size_t);
- public:
- FMT_API file_buffer(cstring_view path, const ostream_params& params);
- FMT_API file_buffer(file_buffer&& other) noexcept;
- FMT_API ~file_buffer();
- void flush() {
- if (size() == 0) return;
- file_.write(data(), size() * sizeof(data()[0]));
- clear();
- }
- void close() {
- flush();
- file_.close();
- }
- };
- }
- constexpr auto buffer_size = detail::buffer_size();
- class FMT_API ostream {
- private:
- FMT_MSC_WARNING(suppress : 4251)
- detail::file_buffer buffer_;
- ostream(cstring_view path, const detail::ostream_params& params)
- : buffer_(path, params) {}
- public:
- ostream(ostream&& other) : buffer_(std::move(other.buffer_)) {}
- ~ostream();
- void flush() { buffer_.flush(); }
- template <typename... T>
- friend auto output_file(cstring_view path, T... params) -> ostream;
- void close() { buffer_.close(); }
-
-
- template <typename... T> void print(format_string<T...> fmt, T&&... args) {
- vformat_to(appender(buffer_), fmt, fmt::make_format_args(args...));
- }
- };
- template <typename... T>
- inline auto output_file(cstring_view path, T... params) -> ostream {
- return {path, detail::ostream_params(params...)};
- }
- #endif
- FMT_END_EXPORT
- FMT_END_NAMESPACE
- #endif
|