Simon Rozman
6aa63f3bd6
We need this for a rare use-case, so maintaining three flavors of diagstream is too expensive. Furthermore, the buffering was removed for simplicity. Signed-off-by: Simon Rozman <simon@rozman.si>
672 lines
24 KiB
C++
672 lines
24 KiB
C++
/*
|
|
SPDX-License-Identifier: MIT
|
|
Copyright © 2023 Amebis
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
#include "endian.hpp"
|
|
#include "internal.hpp"
|
|
#include "string.hpp"
|
|
#include "sal.hpp"
|
|
#include <assert.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <algorithm>
|
|
#include <chrono>
|
|
#include <fstream>
|
|
#include <iterator>
|
|
#include <istream>
|
|
#include <ostream>
|
|
#include <vector>
|
|
|
|
namespace stdex
|
|
{
|
|
///
|
|
/// Binary stream writer
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_ostreamfmt
|
|
{
|
|
public:
|
|
std::basic_ostream<_Elem, _Traits> &sp; // Write stream
|
|
|
|
inline basic_ostreamfmt(_Inout_ std::basic_ostream<_Elem, _Traits> &stream) : sp(stream) {}
|
|
|
|
using pos_type = typename _Traits::pos_type;
|
|
using off_type = typename _Traits::off_type;
|
|
inline pos_type tellp() { return sp.tellp(); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& seekp(pos_type pos) { sp.seekp(pos); return *this; }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& seekp(off_type off, std::ios_base::seekdir dir) { sp.seekp(off, dir); return *this; }
|
|
inline bool good() const noexcept { return sp.good(); }
|
|
inline bool eof() const noexcept { return sp.eof(); }
|
|
inline bool fail() const noexcept { return sp.fail(); }
|
|
inline bool bad() const noexcept { return sp.bad(); }
|
|
|
|
inline basic_ostreamfmt<_Elem, _Traits>& write(_In_reads_bytes_(size) const void* data, _In_ std::streamsize size)
|
|
{
|
|
sp.write(reinterpret_cast<const _Elem*>(data), size/sizeof(_Elem));
|
|
return *this;
|
|
}
|
|
|
|
template <class T>
|
|
inline basic_ostreamfmt<_Elem, _Traits>& write(_In_ T value)
|
|
{
|
|
HE2LE(&value);
|
|
sp.write(reinterpret_cast<const _Elem*>(&value), sizeof(T)/sizeof(_Elem));
|
|
return *this;
|
|
}
|
|
|
|
inline basic_ostreamfmt<_Elem, _Traits>& write(_In_z_ const char* value)
|
|
{
|
|
size_t count = strlen(value);
|
|
if (count > UINT32_MAX)
|
|
throw std::invalid_argument("string too big");
|
|
sp.write(static_cast<uint32_t>(count));
|
|
sp.write(reinterpret_cast<const _Elem*>(value), (std::streamsize)count * sizeof(char)/sizeof(_Elem));
|
|
return *this;
|
|
}
|
|
|
|
inline basic_ostreamfmt<_Elem, _Traits>& write(_In_z_ const wchar_t* value)
|
|
{
|
|
size_t count = strlen(value);
|
|
if (count > UINT32_MAX)
|
|
throw std::invalid_argument("string too big");
|
|
sp.write(static_cast<uint32_t>(count));
|
|
#ifdef BIG_ENDIAN
|
|
for (size_t i = 0; i < count; ++i)
|
|
sp.write(value[i]);
|
|
#else
|
|
sp.write(reinterpret_cast<const _Elem*>(value), (std::streamsize)count * sizeof(wchar_t)/sizeof(_Elem));
|
|
#endif
|
|
return *this;
|
|
}
|
|
|
|
inline basic_ostreamfmt<_Elem, _Traits>& write_byte(_In_ uint8_t value)
|
|
{
|
|
return write(value);
|
|
}
|
|
|
|
///
|
|
/// Formats string using `printf()` and write it to stream.
|
|
///
|
|
/// \param[in] format String template using `printf()` style
|
|
/// \param[in] locale Stdlib locale used to perform formatting. Use `NULL` to use locale globally set by `setlocale()`.
|
|
/// \param[in] arg Arguments to `format`
|
|
///
|
|
template <class _Elem2>
|
|
void vprintf(_In_z_ _Printf_format_string_ const _Elem2 *format, _In_opt_ locale_t locale, _In_ va_list arg)
|
|
{
|
|
std::basic_string<_Elem2> str;
|
|
vappendf(str, format, locale, arg);
|
|
sp.write(reinterpret_cast<const _Elem*>(str.c_str()), str.size() * sizeof(_Elem2)/sizeof(_Elem));
|
|
}
|
|
|
|
///
|
|
/// Formats string using `printf()` and write it to stream.
|
|
///
|
|
/// \param[in] format String template using `printf()` style
|
|
/// \param[in] locale Stdlib locale used to perform formatting. Use `NULL` to use locale globally set by `setlocale()`.
|
|
///
|
|
template <class _Elem2>
|
|
void printf(_In_z_ _Printf_format_string_ const _Elem2 *format, _In_opt_ locale_t locale, ...)
|
|
{
|
|
va_list arg;
|
|
va_start(arg, locale);
|
|
vprintf(format, locale, arg);
|
|
va_end(arg);
|
|
}
|
|
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ int8_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ int16_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ int32_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ int64_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ uint8_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ uint16_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ uint32_t value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ uint64_t value) { return write(value); }
|
|
#if defined(_NATIVE_SIZE_T_DEFINED) && defined(_WIN64)
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ size_t value) { return write(value); }
|
|
#endif
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ float value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ double value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ char value) { return write(value); }
|
|
#ifdef _NATIVE_WCHAR_T_DEFINED
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_ wchar_t value) { return write(value); }
|
|
#endif
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_z_ const char* value) { return write(value); }
|
|
inline basic_ostreamfmt<_Elem, _Traits>& operator <<(_In_z_ const wchar_t* value) { return write(value); }
|
|
};
|
|
|
|
using ostreamfmt = basic_ostreamfmt<char, std::char_traits<char>>;
|
|
using wostreamfmt = basic_ostreamfmt<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
///
|
|
/// Binary stream reader
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_istreamfmt
|
|
{
|
|
public:
|
|
std::basic_istream<_Elem, _Traits> &sg; // Read stream
|
|
|
|
inline basic_istreamfmt(_Inout_ std::basic_istream<_Elem, _Traits> &stream) : sg(stream) {}
|
|
|
|
using pos_type = typename _Traits::pos_type;
|
|
using off_type = typename _Traits::off_type;
|
|
inline pos_type tellg() { return sg.tellg(); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& seekg(pos_type pos) { sg.seekg(pos); return *this; }
|
|
inline basic_istreamfmt<_Elem, _Traits>& seekg(off_type off, std::ios_base::seekdir dir) { sg.seekg(off, dir); return *this; }
|
|
inline bool good() const noexcept { return sg.good(); }
|
|
inline bool eof() const noexcept { return sg.eof(); }
|
|
inline bool fail() const noexcept { return sg.fail(); }
|
|
inline bool bad() const noexcept { return sg.bad(); }
|
|
inline std::streamsize gcount() const noexcept { return sg.gcount(); }
|
|
|
|
inline basic_istreamfmt<_Elem, _Traits>& read(_Out_writes_bytes_(size) void* data, std::streamsize size)
|
|
{
|
|
sg.read(reinterpret_cast<_Elem*>(data), size/sizeof(_Elem));
|
|
return *this;
|
|
}
|
|
|
|
template <class T>
|
|
inline basic_istreamfmt<_Elem, _Traits>& read(_Out_ T& value)
|
|
{
|
|
sg.read(reinterpret_cast<_Elem*>(&value), sizeof(T)/sizeof(_Elem));
|
|
if (sg.good())
|
|
LE2HE(&value);
|
|
return *this;
|
|
}
|
|
|
|
template <class _Traits = std::char_traits<char>, class _Alloc = std::allocator<char>>
|
|
inline basic_istreamfmt<_Elem, _Traits>& read(_Inout_ std::basic_string<char, _Traits, _Alloc>& value)
|
|
{
|
|
uint32_t count;
|
|
sg.read(count);
|
|
if (sg.good()) {
|
|
value.resize(count);
|
|
sg.read(reinterpret_cast<_Elem*>(&value[0]), (std::streamsize)count * sizeof(char)/sizeof(_Elem));
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template <class _Traits = std::char_traits<wchar_t>, class _Alloc = std::allocator<wchar_t>>
|
|
inline basic_istreamfmt<_Elem, _Traits>& read(_Inout_ std::basic_string<wchar_t, _Traits, _Alloc>& value)
|
|
{
|
|
uint32_t count;
|
|
sg.read(count);
|
|
if (sg.good()) {
|
|
value.resize(count);
|
|
#ifdef BIG_ENDIAN
|
|
for (size_t i = 0; i < count; ++i)
|
|
sg.read(value[i]);
|
|
#else
|
|
sg.read(reinterpret_cast<_Elem*>(&value[0]), (std::streamsize)count * sizeof(wchar_t)/sizeof(_Elem));
|
|
#endif
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
inline uint8_t read_byte()
|
|
{
|
|
uint8_t value;
|
|
read(value);
|
|
return value;
|
|
}
|
|
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ int8_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ int16_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ int32_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ int64_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ uint8_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ uint16_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ uint32_t& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ uint64_t& value) { return read(value); }
|
|
#if defined(_NATIVE_SIZE_T_DEFINED) && defined(_WIN64)
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ size_t& value) { return read(value); }
|
|
#endif
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ float& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ double& value) { return read(value); }
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ char& value) { return read(value); }
|
|
#ifdef _NATIVE_WCHAR_T_DEFINED
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Out_ wchar_t& value) { return read(value); }
|
|
#endif
|
|
template <class _Traits = std::char_traits<char>, class _Alloc = std::allocator<char>>
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Inout_ std::basic_string<char, _Traits, _Alloc>& value) { return read(value); }
|
|
template <class _Traits = std::char_traits<wchar_t>, class _Alloc = std::allocator<wchar_t>>
|
|
inline basic_istreamfmt<_Elem, _Traits>& operator >>(_Inout_ std::basic_string<wchar_t, _Traits, _Alloc>& value) { return read(value); }
|
|
};
|
|
|
|
using istreamfmt = basic_istreamfmt<char, std::char_traits<char>>;
|
|
using wistreamfmt = basic_istreamfmt<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
///
|
|
/// Binary stream reader/writer
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_iostreamfmt : public basic_ostreamfmt<_Elem, _Traits>, public basic_istreamfmt<_Elem, _Traits>
|
|
{
|
|
public:
|
|
inline basic_iostreamfmt(_Inout_ std::basic_iostream<_Elem, _Traits> &stream) :
|
|
basic_ostreamfmt<_Elem, _Traits>(stream),
|
|
basic_istreamfmt<_Elem, _Traits>(stream)
|
|
{}
|
|
};
|
|
|
|
using iostreamfmt = basic_iostreamfmt<char, std::char_traits<char>>;
|
|
using wiostreamfmt = basic_iostreamfmt<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
///
|
|
/// Shared-memory string stream buffer
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_sharedstrbuf : public std::basic_streambuf<_Elem, _Traits>
|
|
{
|
|
public:
|
|
basic_sharedstrbuf(_In_reads_(size) const _Elem* data, _In_ size_t size)
|
|
{
|
|
setg(const_cast<_Elem*>(data), const_cast<_Elem*>(data), const_cast<_Elem*>(data + size));
|
|
}
|
|
|
|
basic_sharedstrbuf(_In_ const basic_sharedstrbuf<_Elem, _Traits>& other)
|
|
{
|
|
setg(other.eback(), other.gptr(), other.egptr());
|
|
}
|
|
|
|
basic_sharedstrbuf<_Elem, _Traits>& operator =(_In_ const basic_sharedstrbuf<_Elem, _Traits>& other)
|
|
{
|
|
if (this != std::addressof(other))
|
|
std::basic_streambuf<_Elem, _Traits>::operator =(other);
|
|
return *this;
|
|
}
|
|
|
|
private:
|
|
basic_sharedstrbuf(_Inout_ basic_sharedstrbuf<_Elem, _Traits>&& other) noexcept;
|
|
basic_sharedstrbuf<_Elem, _Traits>& operator =(_Inout_ basic_sharedstrbuf<_Elem, _Traits>&& other) noexcept;
|
|
|
|
protected:
|
|
virtual pos_type seekoff(off_type off, std::ios_base::seekdir way, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out)
|
|
{
|
|
if (which & std::ios_base::in) {
|
|
_Elem* target;
|
|
switch (way) {
|
|
case std::ios_base::beg: target = eback() + static_cast<ptrdiff_t>(off); break;
|
|
case std::ios_base::cur: target = gptr() + static_cast<ptrdiff_t>(off); break;
|
|
case std::ios_base::end: target = egptr() + static_cast<ptrdiff_t>(off); break;
|
|
default: throw std::invalid_argument("invalid seek reference");
|
|
}
|
|
if (eback() <= target && target <= egptr()) {
|
|
gbump(static_cast<int>(target - gptr()));
|
|
return pos_type{ off_type{ target - eback() } };
|
|
}
|
|
}
|
|
return pos_type{ off_type{-1} };
|
|
}
|
|
|
|
virtual pos_type seekpos(pos_type pos, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out)
|
|
{
|
|
if (which & std::ios_base::in) {
|
|
_Elem* target = eback() + static_cast<size_t>(pos);
|
|
if (eback() <= target && target <= egptr()) {
|
|
gbump(static_cast<int>(target - gptr()));
|
|
return pos_type{ off_type{ target - eback() } };
|
|
}
|
|
}
|
|
return pos_type{ off_type{-1} };
|
|
}
|
|
};
|
|
|
|
template <class _Elem, class _Traits>
|
|
class basic_isharedstrstream : public std::basic_istream<_Elem, _Traits>
|
|
{
|
|
public:
|
|
basic_isharedstrstream(_In_reads_(size) const _Elem* data, _In_ size_t size) :
|
|
m_buf(data, size),
|
|
std::basic_istream<_Elem, _Traits>(&m_buf)
|
|
{}
|
|
|
|
protected:
|
|
basic_sharedstrbuf<_Elem, _Traits> m_buf;
|
|
};
|
|
|
|
using isharedstrstream = basic_isharedstrstream<char, std::char_traits<char>>;
|
|
using wisharedstrstream = basic_isharedstrstream<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
///
|
|
/// Diagnostic input stream buffer
|
|
///
|
|
/// Verifies multiple input streams read the same data.
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_diagstreambuf : public std::basic_streambuf<_Elem, _Traits>
|
|
{
|
|
public:
|
|
using guest_stream = std::basic_iostream<_Elem, _Traits>;
|
|
|
|
template<typename _Iter>
|
|
basic_diagstreambuf(_In_ const _Iter first, _In_ const _Iter last) : m_streams(first, last) {}
|
|
basic_diagstreambuf(_In_reads_(count) guest_stream* const* streams, _In_ size_t count) : basic_diagstreambuf(streams, streams + count) {}
|
|
|
|
private:
|
|
basic_diagstreambuf(_In_ const basic_diagstreambuf<_Elem, _Traits>& other);
|
|
basic_diagstreambuf<_Elem, _Traits>& operator =(_In_ const basic_diagstreambuf<_Elem, _Traits>& other);
|
|
basic_diagstreambuf(_Inout_ basic_diagstreambuf<_Elem, _Traits>&& other) noexcept;
|
|
basic_diagstreambuf<_Elem, _Traits>& operator =(_Inout_ basic_diagstreambuf<_Elem, _Traits>&& other) noexcept;
|
|
|
|
protected:
|
|
virtual pos_type seekoff(off_type off, std::ios_base::seekdir way, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out)
|
|
{
|
|
if (m_streams.empty())
|
|
return pos_type{ off_type{-1} };
|
|
auto r = pos_type{ off_type{-1} };
|
|
if ((which & std::ios_base::in)) {
|
|
m_streams[0]->seekg(off, way);
|
|
r = m_streams[0]->bad() ? pos_type{ off_type{-1} } : m_streams[0]->tellg();
|
|
for (size_t i = 1, n = m_streams.size(); i < n; ++i) {
|
|
m_streams[i]->seekg(off, way);
|
|
if (m_streams[i]->bad())
|
|
r = pos_type{ off_type{-1} };
|
|
}
|
|
}
|
|
if ((which & std::ios_base::out)) {
|
|
m_streams[0]->seekp(off, way);
|
|
r = m_streams[0]->bad() ? pos_type{ off_type{-1} } : m_streams[0]->tellp();
|
|
for (size_t i = 1, n = m_streams.size(); i < n; ++i) {
|
|
m_streams[i]->seekp(off, way);
|
|
if (m_streams[i]->bad())
|
|
r = pos_type{ off_type{-1} };
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
virtual pos_type seekpos(pos_type pos, std::ios_base::openmode which = std::ios_base::in | std::ios_base::out)
|
|
{
|
|
return seekoff(pos, std::ios_base::beg, which);
|
|
}
|
|
|
|
virtual int_type underflow()
|
|
{
|
|
int_type eof = _Traits::eof();
|
|
if (m_streams.empty())
|
|
eof;
|
|
_Elem data;
|
|
m_streams[0]->read(&data, 1);
|
|
int_type r = m_streams[0]->gcount() == 1 ? _Traits::to_int_type(data) : eof;
|
|
for (size_t i = 1, n = m_streams.size(); i < n; ++i) {
|
|
m_streams[i]->read(&data, 1);
|
|
int_type temp_r = m_streams[i]->gcount() == 1 ? _Traits::to_int_type(data) : eof;
|
|
if (r != temp_r)
|
|
r = eof;
|
|
}
|
|
return r;
|
|
}
|
|
|
|
virtual int_type overflow(int_type ch = _Traits::eof())
|
|
{
|
|
if (_Traits::not_eof(ch)) {
|
|
_Elem data = _Traits::to_char_type(ch);
|
|
bool good = true;
|
|
for (size_t i = 0, n = m_streams.size(); i < n; ++i) {
|
|
m_streams[i]->write(&data, 1);
|
|
good &= m_streams[i]->good();
|
|
}
|
|
return good ? 0 : _Traits::eof();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
virtual int sync()
|
|
{
|
|
int r = 0;
|
|
for (size_t i = 0, n = m_streams.size(); i < n; ++i)
|
|
if (m_streams[i]->sync() < 0)
|
|
r = -1;
|
|
return r;
|
|
}
|
|
|
|
protected:
|
|
std::vector<guest_stream*> m_streams;
|
|
};
|
|
|
|
///
|
|
/// Diagnostic input/output stream
|
|
///
|
|
/// Verifies multiple input streams read the same data.
|
|
/// Writes to multiple output streams the same data.
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_diagstream : public std::basic_iostream<_Elem, _Traits>
|
|
{
|
|
public:
|
|
using guest_stream = std::basic_iostream<_Elem, _Traits>;
|
|
|
|
template<typename _Iter>
|
|
basic_diagstream(_In_ const _Iter first, _In_ const _Iter last) :
|
|
m_buf(first, last),
|
|
std::basic_iostream<_Elem, _Traits>(&m_buf)
|
|
{}
|
|
|
|
basic_diagstream(_In_reads_(count) guest_stream* const* streams, _In_ size_t count) :
|
|
basic_diagstream(streams, streams + count)
|
|
{}
|
|
|
|
protected:
|
|
basic_diagstreambuf<_Elem, _Traits> m_buf;
|
|
};
|
|
|
|
using diagstream = basic_diagstream<char, std::char_traits<char>>;
|
|
using wdiagstream = basic_diagstream<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
#ifdef _WIN32
|
|
/// \cond internal
|
|
template struct robber<getter<FILE*, std::filebuf>, &std::filebuf::_Myfile>;
|
|
template struct robber<getter<FILE*, std::wfilebuf>, &std::wfilebuf::_Myfile>;
|
|
|
|
inline FILE* filebuf_fhandle(_In_ std::filebuf* rb)
|
|
{
|
|
return (*rb).*get(getter<FILE*, std::filebuf>());
|
|
}
|
|
|
|
inline FILE* filebuf_fhandle(_In_ std::wfilebuf* rb)
|
|
{
|
|
return (*rb).*get(getter<FILE*, std::wfilebuf>());
|
|
}
|
|
/// \endcond
|
|
#endif
|
|
|
|
///
|
|
/// File stream with additional std::filesystem features
|
|
///
|
|
template <class _Elem, class _Traits>
|
|
class basic_fstream : public std::basic_fstream<_Elem, _Traits>
|
|
{
|
|
public:
|
|
using _Mybase = std::basic_fstream<_Elem, _Traits>;
|
|
#if _HAS_CXX20
|
|
using time_point = std::chrono::time_point<std::chrono::file_clock>;
|
|
#else
|
|
using time_point = std::chrono::time_point<std::chrono::system_clock>;
|
|
#endif
|
|
|
|
basic_fstream() {}
|
|
|
|
explicit basic_fstream(
|
|
_In_z_ const char* file_name,
|
|
_In_ ios_base::openmode mode = ios_base::in | ios_base::out,
|
|
_In_ int prot = ios_base::_Default_open_prot) : _Mybase(file_name, mode, prot) {}
|
|
|
|
explicit basic_fstream(
|
|
_In_z_ const wchar_t* file_name,
|
|
_In_ ios_base::openmode mode = ios_base::in | ios_base::out,
|
|
_In_ int prot = ios_base::_Default_open_prot) : _Mybase(file_name, mode, prot) {}
|
|
|
|
template<class _Elem2, class _Traits2, class _Ax>
|
|
explicit basic_fstream(
|
|
_In_ const std::basic_string<_Elem2, _Traits2, _Ax>& str,
|
|
_In_ ios_base::openmode mode = ios_base::in | ios_base::out,
|
|
_In_ int prot = ios_base::_Default_open_prot) : basic_fstream(str.c_str(), mode, prot) {}
|
|
|
|
explicit basic_fstream(_In_ FILE* file) : _Mybase(file) {}
|
|
|
|
basic_fstream(_Inout_ basic_fstream&& other) : _Mybase(std::move(other)) {}
|
|
|
|
///
|
|
/// Sets end of file at current put position
|
|
///
|
|
void truncate()
|
|
{
|
|
flush();
|
|
auto h = os_fhandle();
|
|
#ifdef _WIN32
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
throw std::runtime_error("invalid handle");
|
|
auto pos = tellp();
|
|
LONG
|
|
pos_lo = static_cast<LONG>(pos & 0xffffffff),
|
|
pos_hi = static_cast<LONG>((pos >> 32) & 0xffffffff);
|
|
if (SetFilePointer(h, pos_lo, &pos_hi, FILE_BEGIN) == INVALID_SET_FILE_POINTER)
|
|
throw std::runtime_error("failed to seek");
|
|
if (!SetEndOfFile(h))
|
|
throw std::runtime_error("failed to truncate");
|
|
#else
|
|
#error Implement!
|
|
#endif
|
|
}
|
|
|
|
///
|
|
/// Returns file modification time
|
|
///
|
|
/// \returns File modification time
|
|
///
|
|
time_point mtime() const
|
|
{
|
|
auto h = os_fhandle();
|
|
#ifdef _WIN32
|
|
if (h == INVALID_HANDLE_VALUE)
|
|
throw std::runtime_error("invalid handle");
|
|
FILETIME ft;
|
|
if (!GetFileTime(h, NULL, NULL, &ft))
|
|
throw std::runtime_error("failed to get mtime");
|
|
#if _HAS_CXX20
|
|
return time_point(time_point::duration(((static_cast<int64_t>(ft.dwHighDateTime) << 32) | ft.dwLowDateTime)));
|
|
#else
|
|
// Adjust epoch to std::chrono::time_point<std::chrono::system_clock>/time_t.
|
|
return time_point(time_point::duration(((static_cast<int64_t>(ft.dwHighDateTime) << 32) | ft.dwLowDateTime) - 116444736000000000ll));
|
|
#endif
|
|
#else
|
|
#error Implement!
|
|
#endif
|
|
}
|
|
|
|
protected:
|
|
#ifdef _WIN32
|
|
HANDLE os_fhandle() const
|
|
{
|
|
FILE* f = filebuf_fhandle(rdbuf());
|
|
if (f == NULL)
|
|
return INVALID_HANDLE_VALUE;
|
|
|
|
int fd = _fileno(f);
|
|
if (fd == -1)
|
|
return INVALID_HANDLE_VALUE;
|
|
|
|
return (HANDLE)_get_osfhandle(fd);
|
|
}
|
|
#else
|
|
#error Implement!
|
|
#endif
|
|
};
|
|
|
|
using fstream = basic_fstream<char, std::char_traits<char>>;
|
|
using wfstream = basic_fstream<wchar_t, std::char_traits<wchar_t>>;
|
|
|
|
///
|
|
/// String stream
|
|
///
|
|
template <class _Elem, class _Traits, class _Alloc>
|
|
class basic_stringstream : public std::basic_stringstream<_Elem, _Traits, _Alloc> {
|
|
public:
|
|
using _Mybase = std::basic_stringstream<_Elem, _Traits, _Alloc>;
|
|
using _Mystr = std::basic_string<_Elem, _Traits, _Alloc>;
|
|
|
|
basic_stringstream() {}
|
|
explicit basic_stringstream(_In_ std::ios_base::openmode mode) : _Mybase(mode) {}
|
|
explicit basic_stringstream(_In_ const _Mystr& str, _In_ std::ios_base::openmode mode = std::ios_base::in | std::ios_base::out) : _Mybase(str, mode) {}
|
|
basic_stringstream(_Inout_ basic_stringstream&& other) : _Mybase(std::move(other)) {}
|
|
|
|
///
|
|
/// Initializes stream with content from file.
|
|
///
|
|
/// \param[in] filename File name
|
|
/// \param[in] mode Mode flags to open file. The std::stringstream returned is always opened as in|out.
|
|
/// \param[in] prot Protection flags to open file
|
|
///
|
|
template <class T>
|
|
explicit basic_stringstream(_In_z_ const T* filename, _In_ std::ios_base::openmode mode = std::ios_base::in, _In_ int prot = std::ios_base::_Default_open_prot) :
|
|
_Mybase(std::ios_base::in | std::ios_base::out | (mode & std::ios_base::binary | std::ios_base::app))
|
|
{
|
|
std::basic_ifstream<_Elem, _Traits> input(filename, mode & ~(std::ios_base::ate | std::ios_base::app), prot);
|
|
input.seekg(0, input.end);
|
|
auto size = input.tellg();
|
|
if (size > SIZE_MAX)
|
|
throw std::runtime_error("file too big to fit into memory");
|
|
str().reserve(static_cast<size_t>(size));
|
|
input.seekg(0);
|
|
do {
|
|
_Elem buf[0x1000];
|
|
input.read(buf, _countof(buf));
|
|
write(buf, input.gcount());
|
|
} while (!input.eof());
|
|
if (!(mode & (std::ios_base::ate | std::ios_base::app)))
|
|
seekp(0);
|
|
}
|
|
|
|
///
|
|
/// Initializes stream with content from file.
|
|
///
|
|
/// \param[in] filename File name
|
|
/// \param[in] mode Mode flags to open file. The std::stringstream returned is always opened as in|out.
|
|
/// \param[in] prot Protection flags to open file
|
|
///
|
|
template <class _Elem2, class _Traits2 = std::char_traits<_Elem2>, class _Alloc2 = std::allocator<_Elem2>>
|
|
explicit basic_stringstream(_In_ const std::basic_string<_Elem2, _Traits2, _Alloc2>& filename, _In_ std::ios_base::openmode mode = std::ios_base::in, _In_ int prot = std::ios_base::_Default_open_prot) :
|
|
basic_stringstream(filename.c_str(), mode, prot)
|
|
{}
|
|
|
|
///
|
|
/// Saves stream content to a file.
|
|
///
|
|
/// \param[in] filename File name
|
|
/// \param[in] mode Mode flags to open file
|
|
/// \param[in] prot Protection flags to open file
|
|
///
|
|
template <class T>
|
|
void save(_In_z_ const T* filename, _In_ std::ios_base::openmode mode = std::ios_base::out, _In_ int prot = std::ios_base::_Default_open_prot)
|
|
{
|
|
std::basic_ofstream<_Elem, _Traits> output(filename, mode, prot);
|
|
auto origin = tellg();
|
|
seekg(0, end);
|
|
auto size = tellg();
|
|
do {
|
|
_Elem buf[0x1000];
|
|
read(buf, _countof(buf));
|
|
output.write(buf, gcount());
|
|
} while (!eof());
|
|
seekg(origin);
|
|
}
|
|
|
|
template <class _Elem2, class _Traits2 = std::char_traits<T>, class _Alloc2 = std::allocator<T>>
|
|
void save(_In_ const std::basic_string<_Elem2, _Traits2, _Alloc2>& filename, _In_ std::ios_base::openmode mode = std::ios_base::out, _In_ int prot = std::ios_base::_Default_open_prot)
|
|
{
|
|
save(filename.data(), mode, prot);
|
|
}
|
|
};
|
|
|
|
using stringstream = basic_stringstream<char, std::char_traits<char>, std::allocator<char>>;
|
|
using wstringstream = basic_stringstream<wchar_t, std::char_traits<wchar_t>, std::allocator<char>>;
|
|
}
|