[llvm] [Support] mmap when possible in getSTDIN. (PR #162013)
Yanzuo Liu via llvm-commits
llvm-commits at lists.llvm.org
Sun Oct 5 18:53:32 PDT 2025
================
@@ -573,6 +573,27 @@ ErrorOr<std::unique_ptr<MemoryBuffer>> MemoryBuffer::getSTDIN() {
// FIXME: That isn't necessarily true, we should try to mmap stdin and
// fallback if it fails.
sys::ChangeStdinMode(sys::fs::OF_Text);
+ std::error_code EC;
+ sys::fs::file_type Type;
+ sys::fs::file_status Status;
+ EC = sys::fs::status(sys::fs::getStdinHandle(), Status);
+ if (EC)
+ return EC;
+
+ Type = Status.type();
+
+ /*
+ * If the FD is regular file or block file,
+ * we try to create a mmap buffer first.
+ * If failed, rollback to read and copy.
+ */
+ if (Type == sys::fs::file_type::regular_file ||
+ Type == sys::fs::file_type::block_file) {
+ std::unique_ptr<MemoryBuffer> Result(new MemoryBufferMMapFile<MemoryBuffer>(
+ false, sys::fs::getStdinHandle(), Status.getSize(), 0, EC));
----------------
zwuis wrote:
Can we use `std::make_unique`?
https://github.com/llvm/llvm-project/pull/162013
More information about the llvm-commits
mailing list