Index: util/file/file_writer.cc |
diff --git a/util/file/file_writer.cc b/util/file/file_writer.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..592f0f6195ded89d7f91f922f75dbe2c70f01d20 |
--- /dev/null |
+++ b/util/file/file_writer.cc |
@@ -0,0 +1,149 @@ |
+// Copyright 2014 The Crashpad Authors. All rights reserved. |
+// |
+// Licensed under the Apache License, Version 2.0 (the "License"); |
+// you may not use this file except in compliance with the License. |
+// You may obtain a copy of the License at |
+// |
+// http://www.apache.org/licenses/LICENSE-2.0 |
+// |
+// Unless required by applicable law or agreed to in writing, software |
+// distributed under the License is distributed on an "AS IS" BASIS, |
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
+// See the License for the specific language governing permissions and |
+// limitations under the License. |
+ |
+#include "util/file/file_writer.h" |
+ |
+#include <algorithm> |
+ |
+#include "base/logging.h" |
+#include "base/posix/eintr_wrapper.h" |
+#include "util/file/fd_io.h" |
+ |
+namespace crashpad { |
+ |
+// Ensure type compatibility between WritableIoVec and iovec. |
+COMPILE_ASSERT(sizeof(WritableIoVec) == sizeof(iovec), WritableIoVec_size); |
+COMPILE_ASSERT(offsetof(WritableIoVec, iov_base) == offsetof(iovec, iov_base), |
+ WritableIoVec_base_offset); |
+COMPILE_ASSERT(offsetof(WritableIoVec, iov_len) == offsetof(iovec, iov_len), |
+ WritableIoVec_len_offset); |
+ |
+FileWriter::FileWriter() : fd_() { |
+} |
+ |
+FileWriter::~FileWriter() { |
+} |
+ |
+bool FileWriter::Open(const base::FilePath& path, int oflag, mode_t mode) { |
+ CHECK(!fd_.is_valid()); |
+ |
+ DCHECK((oflag & O_WRONLY) || (oflag & O_RDWR)); |
+ |
+ fd_.reset(HANDLE_EINTR(open(path.value().c_str(), oflag, mode))); |
+ if (!fd_.is_valid()) { |
+ PLOG(ERROR) << "open " << path.value(); |
+ return false; |
+ } |
+ |
+ return true; |
+} |
+ |
+void FileWriter::Close() { |
+ CHECK(fd_.is_valid()); |
+ |
+ fd_.reset(); |
+} |
+ |
+bool FileWriter::Write(const void* data, size_t size) { |
+ DCHECK(fd_.is_valid()); |
+ |
+ // TODO(mark): Write no more than SSIZE_MAX bytes in a single call. |
+ ssize_t written = WriteFD(fd_.get(), data, size); |
+ if (written < 0) { |
+ PLOG(ERROR) << "write"; |
+ return false; |
+ } else if (written == 0) { |
+ LOG(ERROR) << "write: returned 0"; |
+ return false; |
+ } |
+ |
+ return true; |
+} |
+ |
+bool FileWriter::WriteIoVec(std::vector<WritableIoVec>* iovecs) { |
+ DCHECK(fd_.is_valid()); |
+ |
+ ssize_t size = 0; |
+ for (const WritableIoVec& iov : *iovecs) { |
+ // TODO(mark): Check to avoid overflow of ssize_t, and fail with EINVAL. |
+ size += iov.iov_len; |
+ } |
+ |
+ // Get an iovec*, because that’s what writev wants. The only difference |
+ // between WritableIoVec and iovec is that WritableIoVec’s iov_base is a |
+ // pointer to a const buffer, where iovec’s iov_base isn’t. writev doesn’t |
+ // actually write to the data, so this cast is safe here. iovec’s iov_base is |
+ // non-const because the same structure is used for readv and writev, and |
+ // readv needs to write to the buffer that iov_base points to. |
+ iovec* iov = reinterpret_cast<iovec*>(&(*iovecs)[0]); |
+ size_t remaining_iovecs = iovecs->size(); |
+ |
+ while (size > 0) { |
+ size_t writev_iovec_count = |
+ std::min(remaining_iovecs, static_cast<size_t>(IOV_MAX)); |
+ ssize_t written = HANDLE_EINTR(writev(fd_.get(), iov, writev_iovec_count)); |
+ if (written < 0) { |
+ PLOG(ERROR) << "writev"; |
+ return false; |
+ } else if (written == 0) { |
+ LOG(ERROR) << "writev: returned 0"; |
+ return false; |
+ } |
+ |
+ size -= written; |
+ DCHECK_GE(size, 0); |
+ |
+ if (size == 0) { |
+ remaining_iovecs = 0; |
+ break; |
+ } |
+ |
+ while (written > 0) { |
+ size_t wrote_this_iovec = |
+ std::min(static_cast<size_t>(written), iov->iov_len); |
+ written -= wrote_this_iovec; |
+ if (wrote_this_iovec < iov->iov_len) { |
+ iov->iov_base = |
+ reinterpret_cast<char*>(iov->iov_base) + wrote_this_iovec; |
+ iov->iov_len -= wrote_this_iovec; |
+ } else { |
+ ++iov; |
+ --remaining_iovecs; |
+ } |
+ } |
+ } |
+ |
+ DCHECK_EQ(remaining_iovecs, 0u); |
+ |
+#ifndef NDEBUG |
+ // The interface says that |iovecs| is not sacred, so scramble it to make sure |
+ // that nobody depends on it. |
+ memset(&(*iovecs)[0], 0xa5, sizeof((*iovecs)[0]) * iovecs->size()); |
+#endif |
+ |
+ return true; |
+} |
+ |
+off_t FileWriter::Seek(off_t offset, int whence) { |
+ DCHECK(fd_.is_valid()); |
+ |
+ off_t rv = lseek(fd_.get(), offset, whence); |
+ if (rv < 0) { |
+ PLOG(ERROR) << "lseek"; |
+ } |
+ |
+ return rv; |
+} |
+ |
+} // namespace crashpad |