STL/stl/inc/__msvc_filebuf.hpp

791 строка
28 KiB
C++

// __msvc_filebuf.hpp internal header
// Copyright (c) Microsoft Corporation.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#ifndef __MSVC_FILEBUF_HPP
#define __MSVC_FILEBUF_HPP
#include <yvals.h>
#if _STL_COMPILER_PREPROCESSOR
#include <cstdio>
#include <streambuf>
#pragma pack(push, _CRT_PACKING)
#pragma warning(push, _STL_WARNING_LEVEL)
#pragma warning(disable : _STL_DISABLED_WARNINGS)
_STL_DISABLE_CLANG_WARNINGS
#pragma push_macro("new")
#undef new
// TRANSITION, ABI: The _Path_ish functions accepting filesystem::path or experimental::filesystem::path are templates
// which always use the same types as a workaround for user code deriving from iostreams types and
// __declspec(dllexport)ing the derived types. Adding member functions to iostreams broke the ABI of such DLLs.
// Deriving and __declspec(dllexport)ing standard library types is not supported, but in this particular case
// the workaround was inexpensive. The workaround will be removed in the next ABI breaking release of the
// Visual C++ Libraries.
_STD_BEGIN
#if _HAS_CXX17
namespace filesystem {
_EXPORT_STD class path;
}
#endif // _HAS_CXX17
#ifndef _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
#ifdef _M_CEE
#define _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM 0
#else // ^^^ defined(_M_CEE) / !defined(_M_CEE) vvv
#define _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM 1
#endif // ^^^ !defined(_M_CEE) ^^^
#endif // _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
#if _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
namespace experimental {
namespace filesystem {
inline namespace v1 {
class path;
}
} // namespace filesystem
} // namespace experimental
#endif // _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
// clang-format off
template <class _Ty>
constexpr bool _Is_any_path = _Is_any_of_v<_Ty
#if _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
, experimental::filesystem::path
#endif // _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
#if _HAS_CXX17
, filesystem::path
#endif // _HAS_CXX17
>;
// clang-format on
extern "C++" _CRTIMP2_PURE FILE* __CLRCALL_PURE_OR_CDECL _Fiopen(const char*, ios_base::openmode, int);
extern "C++" _CRTIMP2_PURE FILE* __CLRCALL_PURE_OR_CDECL _Fiopen(const wchar_t*, ios_base::openmode, int);
template <class _Elem>
bool _Fgetc(_Elem& _Ch, FILE* _File) { // get an element from a C stream
return _CSTD fread(&_Ch, sizeof(_Elem), 1, _File) == 1;
}
template <>
inline bool _Fgetc(char& _Byte, FILE* _File) { // get a char element from a C stream
int _Meta;
if ((_Meta = _CSTD fgetc(_File)) == EOF) {
return false;
} else { // got one, convert to char
_Byte = static_cast<char>(_Meta);
return true;
}
}
template <>
inline bool _Fgetc(wchar_t& _Wchar, FILE* _File) { // get a wchar_t element from a C stream
wint_t _Meta;
if ((_Meta = _CSTD fgetwc(_File)) == WEOF) {
return false;
} else { // got one, convert to wchar_t
_Wchar = static_cast<wchar_t>(_Meta);
return true;
}
}
#ifdef _CRTBLD
template <>
inline bool _Fgetc(unsigned short& _Wchar, FILE* _File) { // get an unsigned short element from a C stream
wint_t _Meta;
if ((_Meta = _CSTD fgetwc(_File)) == WEOF) {
return false;
} else { // got one, convert to unsigned short
_Wchar = static_cast<unsigned short>(_Meta);
return true;
}
}
#endif // defined(_CRTBLD)
template <class _Elem>
bool _Fputc(_Elem _Ch, FILE* _File) { // put an element to a C stream
return _CSTD fwrite(&_Ch, 1, sizeof(_Elem), _File) == sizeof(_Elem);
}
template <>
inline bool _Fputc(char _Byte, FILE* _File) { // put a char element to a C stream
return _CSTD fputc(_Byte, _File) != EOF;
}
template <>
inline bool _Fputc(wchar_t _Wchar, FILE* _File) { // put a wchar_t element to a C stream
return _CSTD fputwc(_Wchar, _File) != WEOF;
}
#ifdef _CRTBLD
template <>
inline bool _Fputc(unsigned short _Wchar, FILE* _File) { // put an unsigned short element to a C stream
return _CSTD fputwc(_Wchar, _File) != WEOF;
}
#endif // defined(_CRTBLD)
template <class _Elem>
bool _Ungetc(const _Elem&, FILE*) { // put back an arbitrary element to a C stream (always fail)
return false;
}
template <>
inline bool _Ungetc(const char& _Byte, FILE* _File) { // put back a char element to a C stream
return _CSTD ungetc(static_cast<unsigned char>(_Byte), _File) != EOF;
}
template <>
inline bool _Ungetc(const signed char& _Byte, FILE* _File) { // put back a signed char element to a C stream
return _CSTD ungetc(static_cast<unsigned char>(_Byte), _File) != EOF;
}
template <>
inline bool _Ungetc(const unsigned char& _Byte, FILE* _File) { // put back an unsigned char element to a C stream
return _CSTD ungetc(_Byte, _File) != EOF;
}
template <>
inline bool _Ungetc(const wchar_t& _Wchar, FILE* _File) { // put back a wchar_t element to a C stream
return _CSTD ungetwc(_Wchar, _File) != WEOF;
}
#ifdef _CRTBLD
template <>
inline bool _Ungetc(const unsigned short& _Wchar, FILE* _File) { // put back an unsigned short element to a C stream
return _CSTD ungetwc(_Wchar, _File) != WEOF;
}
#endif // defined(_CRTBLD)
_EXPORT_STD template <class _Elem, class _Traits>
class basic_filebuf : public basic_streambuf<_Elem, _Traits> { // stream buffer associated with a C stream
public:
using _Mysb = basic_streambuf<_Elem, _Traits>;
using _Cvt = codecvt<_Elem, char, typename _Traits::state_type>;
basic_filebuf() : _Mysb() {
_Init(nullptr, _Newfl);
}
explicit basic_filebuf(FILE* const _File) : _Mysb() { // extension
_Init(_File, _Newfl);
}
__CLR_OR_THIS_CALL ~basic_filebuf() noexcept override {
if (_Myfile) {
_Reset_back(); // revert from _Mychar buffer
}
if (_Closef) {
close();
}
}
using int_type = typename _Traits::int_type;
using pos_type = typename _Traits::pos_type;
using off_type = typename _Traits::off_type;
basic_filebuf(_Uninitialized) noexcept : _Mysb(_Noinit) {}
basic_filebuf(basic_filebuf&& _Right) {
_Init(_Right._Myfile, _Newfl); // match buffering styles
_Init(static_cast<FILE*>(nullptr), _Closefl); // then make *this look closed
_Assign_rv(_STD move(_Right));
}
basic_filebuf& operator=(basic_filebuf&& _Right) {
_Assign_rv(_STD move(_Right));
return *this;
}
void _Assign_rv(basic_filebuf&& _Right) {
if (this != _STD addressof(_Right)) {
close();
this->swap(_Right);
}
}
void swap(basic_filebuf& _Right) noexcept /* strengthened */ {
if (this != _STD addressof(_Right)) {
FILE* _Myfile_sav = _Myfile;
const _Cvt* _Pcvt_sav = _Pcvt;
typename _Traits::state_type _State_sav = _State;
bool _Wrotesome_sav = _Wrotesome;
bool _Closef_sav = _Closef;
bool _Set_eback_sav = _Mysb::eback() == &_Mychar;
bool _Set_eback_live = _Mysb::gptr() == &_Mychar;
_Elem* _Pfirst0 = _Mysb::pbase();
_Elem* _Pnext0 = _Mysb::pptr();
_Elem* _Pend = _Mysb::epptr();
_Elem* _Gfirst0 = _Mysb::eback();
_Elem* _Gnext0 = _Mysb::gptr();
_Elem* _Gend = _Mysb::egptr();
// reinitialize *this
_Init(_Right._Myfile, _Right._Myfile ? _Openfl : _Newfl);
_Mysb::setp(_Right.pbase(), _Right.pptr(), _Right.epptr());
if (_Right.eback() != &_Right._Mychar) {
_Mysb::setg(_Right.eback(), _Right.gptr(), _Right.egptr());
} else if (_Right.gptr() != &_Right._Mychar) {
_Mysb::setg(&_Mychar, &_Mychar + 1, &_Mychar + 1);
} else {
_Mysb::setg(&_Mychar, &_Mychar, &_Mychar + 1);
}
_Pcvt = _Right._Pcvt;
_State = _Right._State;
_Wrotesome = _Right._Wrotesome;
_Closef = _Right._Closef;
// reinitialize _Right
_Right._Init(_Myfile_sav, _Myfile_sav ? _Openfl : _Newfl);
_Right.setp(_Pfirst0, _Pnext0, _Pend);
if (!_Set_eback_sav) {
_Right.setg(_Gfirst0, _Gnext0, _Gend);
} else if (!_Set_eback_live) {
_Right.setg(&_Right._Mychar, &_Right._Mychar + 1, &_Right._Mychar + 1);
} else {
_Right.setg(&_Right._Mychar, &_Right._Mychar, &_Right._Mychar + 1);
}
_Right._Pcvt = _Pcvt_sav;
_Right._State = _State_sav;
_Right._Wrotesome = _Wrotesome_sav;
_Right._Closef = _Closef_sav;
// swap ancillary data
_STD swap(_Set_eback, _Right._Set_eback);
_STD swap(_Set_egptr, _Right._Set_egptr);
_STD swap(_Mychar, _Right._Mychar);
_STD swap(_Mysb::_Plocale, _Right._Plocale);
}
}
basic_filebuf(const basic_filebuf&) = delete;
basic_filebuf& operator=(const basic_filebuf&) = delete;
enum _Initfl { // reasons for a call to _Init
_Newfl,
_Openfl,
_Closefl
};
_NODISCARD bool is_open() const noexcept /* strengthened */ {
return static_cast<bool>(_Myfile);
}
basic_filebuf* open(const char* _Filename, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// _Prot is an extension
if (_Myfile) {
return nullptr;
}
const auto _File = _Fiopen(_Filename, _Mode, _Prot);
if (!_File) {
return nullptr; // open failed
}
_Init(_File, _Openfl);
_Initcvt(_STD use_facet<_Cvt>(_Mysb::getloc()));
return this; // open succeeded
}
basic_filebuf* open(const string& _Str, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// _Prot is an extension
return open(_Str.c_str(), _Mode, _Prot);
}
#if _HAS_OLD_IOSTREAMS_MEMBERS
basic_filebuf* open(const char* _Filename, ios_base::open_mode _Mode) {
return open(_Filename, static_cast<ios_base::openmode>(_Mode));
}
#endif // _HAS_OLD_IOSTREAMS_MEMBERS
basic_filebuf* open(const wchar_t* _Filename, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// in standard as const std::filesystem::path::value_type *; _Prot is an extension
if (_Myfile) {
return nullptr;
}
const auto _File = _Fiopen(_Filename, _Mode, _Prot);
if (!_File) {
return nullptr; // open failed
}
_Init(_File, _Openfl);
_Initcvt(_STD use_facet<_Cvt>(_Mysb::getloc()));
return this; // open succeeded
}
basic_filebuf* open(const wstring& _Str, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// extension
return open(_Str.c_str(), _Mode, _Prot);
}
#if _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
template <class _Path_ish = experimental::filesystem::path>
basic_filebuf* open(
const _Identity_t<_Path_ish>& _Path, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// _Prot is an extension
return open(_Path.c_str(), _Mode, _Prot);
}
#endif // _FSTREAM_SUPPORTS_EXPERIMENTAL_FILESYSTEM
#if _HAS_CXX17
template <int = 0, class _Path_ish = filesystem::path>
basic_filebuf* open(
const _Identity_t<_Path_ish>& _Path, ios_base::openmode _Mode, int _Prot = ios_base::_Default_open_prot) {
// _Prot is an extension
return open(_Path.c_str(), _Mode, _Prot);
}
#endif // _HAS_CXX17
#if _HAS_OLD_IOSTREAMS_MEMBERS
basic_filebuf* open(const wchar_t* _Filename, ios_base::open_mode _Mode) {
// in standard as const std::filesystem::path::value_type *
return open(_Filename, static_cast<ios_base::openmode>(_Mode));
}
#endif // _HAS_OLD_IOSTREAMS_MEMBERS
basic_filebuf* close() {
basic_filebuf* _Ans;
if (_Myfile) { // put any homing sequence and close file
_Reset_back(); // revert from _Mychar buffer
_Ans = this;
if (!_Endwrite()) {
_Ans = nullptr;
}
if (_CSTD fclose(_Myfile) != 0) {
_Ans = nullptr;
}
} else {
_Ans = nullptr;
}
_Init(nullptr, _Closefl);
return _Ans;
}
void __CLR_OR_THIS_CALL _Lock() override { // lock file instead of stream buffer
if (_Myfile) {
_CSTD _lock_file(_Myfile);
}
}
void __CLR_OR_THIS_CALL _Unlock() override { // unlock file instead of stream buffer
if (_Myfile) {
_CSTD _unlock_file(_Myfile);
}
}
#if _HAS_CXX23 && defined(_CPPRTTI)
template <class _UnicodeConsoleFn, class _FallbackFn, class _Filebuf_type>
friend ios_base::iostate _Do_on_maybe_unicode_console(ostream&, _UnicodeConsoleFn, _FallbackFn);
#endif // ^^^ _HAS_CXX23 && defined(_CPPRTTI) ^^^
protected:
int_type __CLR_OR_THIS_CALL overflow(int_type _Meta = _Traits::eof()) override { // put an element to stream
if (_Traits::eq_int_type(_Traits::eof(), _Meta)) {
return _Traits::not_eof(_Meta); // EOF, return success code
}
if (_Mysb::pptr() && _Mysb::pptr() < _Mysb::epptr()) { // room in buffer, store it
*_Mysb::_Pninc() = _Traits::to_char_type(_Meta);
return _Meta;
}
if (!_Myfile) {
return _Traits::eof(); // no open C stream, fail
}
_Reset_back(); // revert from _Mychar buffer
if (!_Pcvt) { // no codecvt facet, put as is
return _Fputc(_Traits::to_char_type(_Meta), _Myfile) ? _Meta : _Traits::eof();
}
// put using codecvt facet
constexpr size_t _Codecvt_temp_buf = 32;
char _Str[_Codecvt_temp_buf];
const _Elem _Ch = _Traits::to_char_type(_Meta);
const _Elem* _Src;
char* _Dest;
// test result of converting one element
switch (_Pcvt->out(_State, &_Ch, &_Ch + 1, _Src, _Str, _Str + _Codecvt_temp_buf, _Dest)) {
case codecvt_base::partial:
case codecvt_base::ok:
{ // converted something, try to put it out
const auto _Count = static_cast<size_t>(_Dest - _Str);
if (0 < _Count && _Count != static_cast<size_t>(_CSTD fwrite(_Str, 1, _Count, _Myfile))) {
return _Traits::eof(); // write failed
}
_Wrotesome = true; // write succeeded
if (_Src != &_Ch) {
return _Meta; // converted whole element
}
return _Traits::eof(); // conversion failed
}
case codecvt_base::noconv:
// no conversion, put as is
return _Fputc(_Ch, _Myfile) ? _Meta : _Traits::eof();
default:
return _Traits::eof(); // conversion failed
}
}
int_type __CLR_OR_THIS_CALL pbackfail(int_type _Meta = _Traits::eof()) override {
// put an element back to stream
if (_Mysb::gptr() && _Mysb::eback() < _Mysb::gptr()
&& (_Traits::eq_int_type(_Traits::eof(), _Meta)
|| _Traits::eq_int_type(_Traits::to_int_type(_Mysb::gptr()[-1]),
_Meta))) { // just back up position
_Mysb::_Gndec();
return _Traits::not_eof(_Meta);
} else if (!_Myfile || _Traits::eq_int_type(_Traits::eof(), _Meta)) {
return _Traits::eof(); // no open C stream or EOF, fail
} else if (!_Pcvt && _Ungetc(_Traits::to_char_type(_Meta), _Myfile)) {
return _Meta; // no facet and unget succeeded, return
} else if (_Mysb::gptr() != &_Mychar) { // putback to _Mychar
_Mychar = _Traits::to_char_type(_Meta);
_Set_back(); // switch to _Mychar buffer
return _Meta;
} else {
return _Traits::eof(); // nowhere to put back
}
}
int_type __CLR_OR_THIS_CALL underflow() override { // get an element from stream, but don't point past it
int_type _Meta;
if (_Mysb::gptr() && _Mysb::gptr() < _Mysb::egptr()) {
return _Traits::to_int_type(*_Mysb::gptr()); // return buffered
} else if (_Traits::eq_int_type(_Traits::eof(), _Meta = uflow())) {
return _Meta; // uflow failed, return EOF
} else { // get a char, don't point past it
pbackfail(_Meta);
return _Meta;
}
}
int_type __CLR_OR_THIS_CALL uflow() override { // get an element from stream, point past it
if (_Mysb::gptr() && _Mysb::gptr() < _Mysb::egptr()) {
return _Traits::to_int_type(*_Mysb::_Gninc()); // return buffered
}
if (!_Myfile) {
return _Traits::eof(); // no open C stream, fail
}
_Reset_back(); // revert from _Mychar buffer
if (!_Pcvt) { // no codecvt facet, just get it
_Elem _Ch;
return _Fgetc(_Ch, _Myfile) ? _Traits::to_int_type(_Ch) : _Traits::eof();
}
// build string until codecvt succeeds
string _Str;
for (;;) { // get using codecvt facet
const char* _Src;
int _Meta = _CSTD fgetc(_Myfile);
if (_Meta == EOF) {
return _Traits::eof(); // partial char?
}
_Str.push_back(static_cast<char>(_Meta)); // append byte and convert
_Elem _Ch;
_Elem* _Dest;
// test result of converting one element
switch (_Pcvt->in(_State, _Str.data(), _Str.data() + _Str.size(), _Src, &_Ch, &_Ch + 1, _Dest)) {
case codecvt_base::partial:
case codecvt_base::ok:
if (_Dest != &_Ch) { // got an element, put back excess and deliver it
auto _Nleft = _Str.data() + _Str.size() - _Src;
while (0 < _Nleft) {
_CSTD ungetc(_Src[--_Nleft], _Myfile);
}
return _Traits::to_int_type(_Ch);
}
_Str.erase(0, static_cast<size_t>(_Src - _Str.data())); // partial, discard used input
break;
case codecvt_base::noconv:
// noconv is only possible if _Elem is char, so we can use it directly
return static_cast<int_type>(_Str.front());
default:
return _Traits::eof(); // conversion failed
}
}
}
streamsize __CLR_OR_THIS_CALL xsgetn(_Elem* _Ptr, streamsize _Count) override {
// get _Count characters from stream
if constexpr (sizeof(_Elem) == 1) {
if (_Count <= 0) {
return 0;
}
if (_Pcvt) { // if we need a nontrivial codecvt transform, do the default expensive thing
return _Mysb::xsgetn(_Ptr, _Count);
}
// assuming this is OK because _Ptr + _Count must be valid
auto _Count_s = static_cast<size_t>(_Count);
const auto _Start_count = _Count;
const auto _Available = static_cast<size_t>(_Mysb::_Gnavail());
if (0 < _Available) { // copy from get area
const auto _Read_size = (_STD min)(_Count_s, _Available);
_Traits::copy(_Ptr, _Mysb::gptr(), _Read_size);
_Ptr += _Read_size;
_Count_s -= _Read_size;
_Mysb::gbump(static_cast<int>(_Read_size));
}
if (_Myfile) { // open C stream, attempt read
_Reset_back(); // revert from _Mychar buffer
// process in 4k - 1 chunks to avoid tripping over fread's clobber-the-end behavior when
// doing \r\n -> \n translation
constexpr size_t _Read_size = 4095; // _INTERNAL_BUFSIZ - 1
while (_Read_size < _Count_s) {
const auto _Actual_read = _CSTD fread(_Ptr, sizeof(_Elem), _Read_size, _Myfile);
_Ptr += _Actual_read;
_Count_s -= _Actual_read;
if (_Actual_read != _Read_size) {
return static_cast<streamsize>(_Start_count - _Count_s);
}
}
if (0 < _Count_s) {
_Count_s -= _CSTD fread(_Ptr, sizeof(_Elem), _Count_s, _Myfile);
}
}
return static_cast<streamsize>(_Start_count - _Count_s);
} else { // non-chars always get element-by-element processing
return _Mysb::xsgetn(_Ptr, _Count);
}
}
streamsize __CLR_OR_THIS_CALL xsputn(const _Elem* _Ptr, streamsize _Count) override {
// put _Count characters to stream
if constexpr (sizeof(_Elem) == 1) {
if (_Pcvt) { // if we need a nontrivial codecvt transform, do the default expensive thing
return _Mysb::xsputn(_Ptr, _Count);
}
const streamsize _Start_count = _Count;
streamsize _Size = _Mysb::_Pnavail();
if (0 < _Count && 0 < _Size) { // copy to write buffer
if (_Count < _Size) {
_Size = _Count;
}
_Traits::copy(_Mysb::pptr(), _Ptr, static_cast<size_t>(_Size));
_Ptr += _Size;
_Count -= _Size;
_Mysb::pbump(static_cast<int>(_Size));
}
if (0 < _Count && _Myfile) { // open C stream, attempt write
_Count -= _CSTD fwrite(_Ptr, sizeof(_Elem), static_cast<size_t>(_Count), _Myfile);
}
return _Start_count - _Count;
} else { // non-chars always get element-by-element processing
return _Mysb::xsputn(_Ptr, _Count);
}
}
pos_type __CLR_OR_THIS_CALL seekoff(off_type _Off, ios_base::seekdir _Way,
ios_base::openmode = ios_base::in | ios_base::out) override { // change position by _Off
fpos_t _Fileposition;
if (_Mysb::gptr() == &_Mychar // something putback
&& _Way == ios_base::cur // a relative seek
&& !_Pcvt) { // not converting
_Off -= static_cast<off_type>(sizeof(_Elem)); // back up over _Elem bytes
}
if (!_Myfile || !_Endwrite()
|| ((_Off != 0 || _Way != ios_base::cur) && _CSTD _fseeki64(_Myfile, _Off, _Way) != 0)
|| _CSTD fgetpos(_Myfile, &_Fileposition) != 0) {
return pos_type{off_type{-1}}; // report failure
}
_Reset_back(); // revert from _Mychar buffer, discarding any putback
return pos_type{_State, _Fileposition}; // return new position
}
pos_type __CLR_OR_THIS_CALL seekpos(pos_type _Pos, ios_base::openmode = ios_base::in | ios_base::out) override {
// change position to _Pos
off_type _Off = static_cast<off_type>(_Pos);
if (!_Myfile || !_Endwrite() || _CSTD fsetpos(_Myfile, &_Off) != 0) {
return pos_type{off_type{-1}}; // report failure
}
_State = _Pos.state();
_Reset_back(); // revert from _Mychar buffer, discarding any putback
return pos_type{_State, _Off}; // return new position
}
_Mysb* __CLR_OR_THIS_CALL setbuf(_Elem* _Buffer, streamsize _Count) override { // offer _Buffer to C stream
int _Mode;
if (!_Buffer && _Count == 0) {
_Mode = _IONBF;
} else {
_Mode = _IOFBF;
}
const size_t _Size = static_cast<size_t>(_Count) * sizeof(_Elem);
if (!_Myfile || _CSTD setvbuf(_Myfile, reinterpret_cast<char*>(_Buffer), _Mode, _Size) != 0) {
return nullptr; // failed
}
// new buffer, reinitialize pointers
_Init(_Myfile, _Openfl);
return this;
}
int __CLR_OR_THIS_CALL sync() override { // synchronize C stream with external file
if (!_Myfile || _Traits::eq_int_type(_Traits::eof(), overflow()) || 0 <= _CSTD fflush(_Myfile)) {
return 0;
}
return -1;
}
void __CLR_OR_THIS_CALL imbue(const locale& _Loc) override {
// set locale to argument (capture nontrivial codecvt facet)
_Initcvt(_STD use_facet<_Cvt>(_Loc));
}
void _Init(FILE* _File, _Initfl _Which) noexcept { // initialize to C stream _File after {new, open, close}
using _State_type = typename _Traits::state_type;
__PURE_APPDOMAIN_GLOBAL static _State_type _Stinit; // initial state
_Closef = _Which == _Openfl;
_Wrotesome = false;
_Mysb::_Init(); // initialize stream buffer base object
if (_File && sizeof(_Elem) == 1) { // point inside C stream with [first, first + count) buffer
_Elem** _Pb = nullptr;
_Elem** _Pn = nullptr;
int* _Nr = nullptr;
::_get_stream_buffer_pointers(
_File, reinterpret_cast<char***>(&_Pb), reinterpret_cast<char***>(&_Pn), &_Nr);
int* _Nw = _Nr;
_Mysb::_Init(_Pb, _Pn, _Nr, _Pb, _Pn, _Nw);
}
_Myfile = _File;
_State = _Stinit;
_Pcvt = nullptr; // pointer to codecvt facet
}
bool _Endwrite() { // put shift to initial conversion state, as needed
if (!_Pcvt || !_Wrotesome) {
return true;
}
// may have to put
if (_Traits::eq_int_type(_Traits::eof(), overflow())) {
return false;
}
constexpr size_t _Codecvt_temp_buf = 32;
char _Str[_Codecvt_temp_buf];
char* _Dest;
switch (_Pcvt->unshift(_State, _Str, _Str + _Codecvt_temp_buf, _Dest)) { // test result of homing conversion
case codecvt_base::ok:
_Wrotesome = false; // homed successfully
_FALLTHROUGH;
case codecvt_base::partial:
{ // put any generated bytes
const auto _Count = static_cast<size_t>(_Dest - _Str);
if (0 < _Count && _Count != static_cast<size_t>(_CSTD fwrite(_Str, 1, _Count, _Myfile))) {
return false; // write failed
}
return !_Wrotesome;
}
case codecvt_base::noconv:
_Wrotesome = false; // homed successfully
return true; // nothing else to do
default:
return false; // conversion failed
}
}
void _Initcvt(const _Cvt& _Newcvt) noexcept { // initialize codecvt pointer
if (_Newcvt.always_noconv()) {
_Pcvt = nullptr; // nothing to do
} else { // set up for nontrivial codecvt facet
_Pcvt = _STD addressof(_Newcvt);
_Mysb::_Init(); // reset any buffering
}
}
private:
const _Cvt* _Pcvt; // pointer to codecvt facet (may be null)
_Elem _Mychar; // putback character, when _Ungetc fails
bool _Wrotesome; // true if homing sequence may be needed
typename _Traits::state_type _State; // current conversion state
bool _Closef; // true if C stream must be closed
FILE* _Myfile; // pointer to C stream
void _Reset_back() noexcept { // restore buffer after putback
if (_Mysb::eback() == &_Mychar) {
_Mysb::setg(_Set_eback, _Set_eback, _Set_egptr);
}
}
void _Set_back() noexcept { // set up putback area
if (_Mysb::eback() != &_Mychar) { // save current get buffer
_Set_eback = _Mysb::eback();
_Set_egptr = _Mysb::egptr();
}
_Mysb::setg(&_Mychar, &_Mychar, &_Mychar + 1);
}
_Elem* _Set_eback; // saves eback() during one-element putback
_Elem* _Set_egptr; // saves egptr()
};
_EXPORT_STD template <class _Elem, class _Traits>
void swap(basic_filebuf<_Elem, _Traits>& _Left, basic_filebuf<_Elem, _Traits>& _Right) noexcept /* strengthened */ {
_Left.swap(_Right);
}
_STD_END
#pragma pop_macro("new")
_STL_RESTORE_CLANG_WARNINGS
#pragma warning(pop)
#pragma pack(pop)
#endif // _STL_COMPILER_PREPROCESSOR
#endif // __MSVC_FILEBUF_HPP