Add minidump file writer. Tested on Mac, but should compile on POSIX systems.

git-svn-id: http://google-breakpad.googlecode.com/svn/trunk@65 4c0a9323-5329-0410-9bdc-e9ce6186880e

src/client/minidump_file_writer.cc

@@ -0,0 +1,233 @@
+// Copyright (c) 2006, Google Inc.
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are
+// met:
+//
+//     * Redistributions of source code must retain the above copyright
+// notice, this list of conditions and the following disclaimer.
+//     * Redistributions in binary form must reproduce the above
+// copyright notice, this list of conditions and the following disclaimer
+// in the documentation and/or other materials provided with the
+// distribution.
+//     * Neither the name of Google Inc. nor the names of its
+// contributors may be used to endorse or promote products derived from
+// this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+// minidump_file_writer.cc: Minidump file writer implementation.
+//
+// See minidump_file_writer.h for documentation.
+
+#include <fcntl.h>
+#include <limits.h>
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "client/minidump_file_writer-inl.h"
+
+namespace google_airbag {
+
+MinidumpFileWriter::MinidumpFileWriter() : file_(-1), position_(0), size_(0) {
+}
+
+MinidumpFileWriter::~MinidumpFileWriter() {
+  Close();
+}
+
+bool MinidumpFileWriter::Open(const std::string &path) {
+  assert(file_ == -1);
+  file_ = open(path.c_str(), O_WRONLY | O_CREAT | O_TRUNC, 0666);
+
+  return file_ != -1;
+}
+
+bool MinidumpFileWriter::Close() {
+  bool result = true;
+
+  if (file_ != -1) {
+    ftruncate(file_, position_);
+    result = close(file_) == 0;
+    file_ = -1;
+  }
+
+  return result;
+}
+
+bool MinidumpFileWriter::WriteString(const wchar_t *str,
+                                     unsigned int length,
+                                     MDLocationDescriptor *location) {
+  assert(str);
+  assert(location);
+  // Calculate the mdstring length by either limiting to |length| as passed in
+  // or by finding the location of the NULL character.
+  if (!length)
+    length = INT_MAX;
+
+  unsigned int mdstring_length = 0;
+  for (; mdstring_length < length && str[mdstring_length]; ++mdstring_length) {
+  }
+
+  // Allocate the string buffer
+  TypedMDRVA<MDString> mdstring(this);
+
+  if (!mdstring.AllocateObjectAndArray(mdstring_length + 1, sizeof(u_int16_t)))
+    return false;
+
+  // Set length excluding the NULL
+  mdstring.get()->length = mdstring_length * sizeof(u_int16_t);
+
+  u_int16_t ch;
+  bool result = true;
+
+  if (sizeof(wchar_t) == sizeof(u_int16_t)) {
+    // Shortcut if wchar_t is the same size as MDString's buffer
+    result = mdstring.Copy(str, mdstring.get()->length);
+  } else {
+    // Copy the string character by character
+    for (unsigned int c = 0; c < mdstring_length && result == true; c++) {
+      ch = str[c];
+      // TODO: For the UTF-32->UTF-16 conversion, it's possible that there
+      // are characters that will require more than one UTF-16 character to
+      // represent it.  Fully supporting this will require a more sophisticated
+      // calculation of the size of the resulting string and for converting the
+      // UTF-32 character into the two UTF-16 characters.
+      result = mdstring.CopyIndexAfterObject(c, &ch, sizeof(ch));
+    }
+  }
+
+  // NULL terminate
+  if (result) {
+    ch = 0;
+    result = mdstring.CopyIndexAfterObject(mdstring_length, &ch, sizeof(ch));
+
+    if (result)
+      *location = mdstring.location();
+  }
+
+  return result;
+}
+
+bool MinidumpFileWriter::WriteString(const char *str, unsigned int length,
+                                     MDLocationDescriptor *location) {
+  assert(str);
+  assert(location);
+  // Calculate the mdstring length by either limiting to |length| as passed in
+  // or by finding the location of the NULL character.
+  if (!length)
+    length = INT_MAX;
+
+  unsigned int mdstring_length = 0;
+  for (; mdstring_length < length && str[mdstring_length]; ++mdstring_length) {
+  }
+
+  // Allocate the string buffer
+  TypedMDRVA<MDString> mdstring(this);
+
+  if (!mdstring.AllocateObjectAndArray(mdstring_length + 1, sizeof(u_int16_t)))
+    return false;
+
+  // Set length excluding the NULL
+  mdstring.get()->length = mdstring_length * sizeof(u_int16_t);
+
+  u_int16_t ch;
+  bool result = true;
+
+  // Copy the string character by character
+  for (unsigned int c = 0; c < mdstring_length && result == true; c++) {
+    ch = str[c];
+    // TODO: For the UTF-8->UTF-16 conversion, it's possible that there are
+    // characters that will convert one or more UTF-8 character into a single
+    // UTF-16 character.  Fully supporting this will require a more
+    // sophisticated calculation of the size of the resulting string and for
+    // converting the UTF-8 characters into a UTF-16 character.
+    result = mdstring.CopyIndexAfterObject(c, &ch, sizeof(ch));
+  }
+
+  // NULL terminate
+  if (result) {
+    ch = 0;
+    result = mdstring.CopyIndexAfterObject(mdstring_length, &ch, sizeof(ch));
+
+    if (result)
+      *location = mdstring.location();
+  }
+
+  return result;
+}
+
+bool MinidumpFileWriter::WriteMemory(const void *src, size_t size,
+                                     MDMemoryDescriptor *output) {
+  assert(src);
+  assert(output);
+  UntypedMDRVA mem(this);
+
+  if (!mem.Allocate(size))
+    return false;
+
+  if (!mem.Copy(src, mem.size()))
+    return false;
+
+  output->start_of_memory_range = reinterpret_cast<u_int64_t>(src);
+  output->memory = mem.location();
+
+  return true;
+}
+
+MDRVA MinidumpFileWriter::Allocate(size_t size) {
+  assert(size);
+  assert(file_ != -1);
+
+  size_t aligned_size = (size + 7) & ~7;  // 64-bit alignment
+
+  if (position_ + aligned_size > size_) {
+    size_t growth = aligned_size;
+    size_t minimal_growth = getpagesize();
+
+    // Ensure that the file grows by at least the size of a memory page
+    if (growth < minimal_growth)
+      growth = minimal_growth;
+
+    size_t new_size = size_ + growth;
+
+    if (ftruncate(file_, new_size) != 0)
+      return kInvalidMDRVA;
+
+    size_ = new_size;
+  }
+
+  MDRVA current_position = position_;
+  position_ += static_cast<MDRVA>(aligned_size);
+
+  return current_position;
+}
+
+bool MinidumpFileWriter::Copy(MDRVA position, const void* src, ssize_t size) {
+  assert(file_ != -1);
+
+  // Ensure that the data will fit in the allocated space
+  if (size + position > size_)
+    return false;
+
+  // Seek and write the data
+  if (lseek(file_, position, SEEK_SET) == static_cast<off_t>(position))
+    if (write(file_, src, size) == size)
+      return true;
+
+  return false;
+}
+
+}  // namespace google_airbag