Index: libcxx/include/fstream =================================================================== --- libcxx/include/fstream +++ libcxx/include/fstream @@ -419,25 +419,38 @@ basic_streambuf::swap(__rhs); if (__extbuf_ != __extbuf_min_ && __rhs.__extbuf_ != __rhs.__extbuf_min_) { - _VSTD::swap(__extbuf_, __rhs.__extbuf_); - _VSTD::swap(__extbufnext_, __rhs.__extbufnext_); - _VSTD::swap(__extbufend_, __rhs.__extbufend_); + // Neither *this nor __rhs uses the small buffer, so we can simply swap the pointers. + std::swap(__extbuf_, __rhs.__extbuf_); + std::swap(__extbufnext_, __rhs.__extbufnext_); + std::swap(__extbufend_, __rhs.__extbufend_); } else { - ptrdiff_t __ln = __extbufnext_ - __extbuf_; - ptrdiff_t __le = __extbufend_ - __extbuf_; - ptrdiff_t __rn = __rhs.__extbufnext_ - __rhs.__extbuf_; - ptrdiff_t __re = __rhs.__extbufend_ - __rhs.__extbuf_; + ptrdiff_t __ln = __extbufnext_ ? __extbufnext_ - __extbuf_ : 0; + ptrdiff_t __le = __extbufend_ ? __extbufend_ - __extbuf_ : 0; + ptrdiff_t __rn = __rhs.__extbufnext_ ? __rhs.__extbufnext_ - __rhs.__extbuf_ : 0; + ptrdiff_t __re = __rhs.__extbufend_ ? __rhs.__extbufend_ - __rhs.__extbuf_ : 0; if (__extbuf_ == __extbuf_min_ && __rhs.__extbuf_ != __rhs.__extbuf_min_) { + // *this uses the small buffer, but __rhs doesn't. __extbuf_ = __rhs.__extbuf_; __rhs.__extbuf_ = __rhs.__extbuf_min_; + std::memmove(__rhs.__extbuf_min_, __extbuf_min_, sizeof(__extbuf_min_)); } else if (__extbuf_ != __extbuf_min_ && __rhs.__extbuf_ == __rhs.__extbuf_min_) { + // *this doesn't use the small buffer, but __rhs does. __rhs.__extbuf_ = __extbuf_; __extbuf_ = __extbuf_min_; + std::memmove(__extbuf_min_, __rhs.__extbuf_min_, sizeof(__extbuf_min_)); + } + else + { + // Both *this and __rhs use the small buffer. + char __tmp[sizeof(__extbuf_min_)]; + std::memmove(__tmp, __extbuf_min_, sizeof(__extbuf_min_)); + std::memmove(__extbuf_min_, __rhs.__extbuf_min_, sizeof(__extbuf_min_)); + std::memmove(__rhs.__extbuf_min_, __tmp, sizeof(__extbuf_min_)); } __extbufnext_ = __extbuf_ + __rn; __extbufend_ = __extbuf_ + __re; Index: libcxx/test/std/input.output/file.streams/fstreams/filebuf.assign/nonmember_swap_min.pass.cpp =================================================================== --- /dev/null +++ libcxx/test/std/input.output/file.streams/fstreams/filebuf.assign/nonmember_swap_min.pass.cpp @@ -0,0 +1,68 @@ +//===----------------------------------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +// This tests that swapping filebufs works correctly even when the small buffer +// optimization is in use (https://github.com/llvm/llvm-project/issues/49282). + +// FIXME: This test fails on Windows for some reason, but even before the swap, +// which is what we want to test here. +// UNSUPPORTED: target={{.+}}-windows-{{.+}} + +// + +// template > +// class basic_filebuf + +// template +// void +// swap(basic_filebuf& x, basic_filebuf& y); + +#include +#include +#include "test_macros.h" +#include "platform_support.h" + +int main(int, char**) +{ + std::string tmp = get_temp_file_name(); + std::string tmpA = get_temp_file_name(); + std::string tmpB = get_temp_file_name(); + + { + std::ofstream sa(tmpA), sb(tmpB); + sa << "AAAA"; + sb << "BBBB"; + } + + std::filebuf f1; + assert(f1.open(tmpA, std::ios_base::in) != 0); + assert(f1.is_open()); + f1.pubsetbuf(0, 0); + + std::filebuf f2; + assert(f2.open(tmpB, std::ios_base::in) != 0); + assert(f2.is_open()); + f2.pubsetbuf(0, 0); + + assert(f1.sgetc() == 'A'); + assert(f2.sgetc() == 'B'); + + swap(f1, f2); + + assert(f1.is_open()); + assert(f2.is_open()); + + assert(f1.sgetc() == 'B'); + assert(f2.sgetc() == 'A'); + + std::remove(tmp.c_str()); + std::remove(tmpA.c_str()); + std::remove(tmpB.c_str()); + + return 0; +}