michael@0: // Copyright (c) 2011, Google Inc.
michael@0: // All rights reserved.
michael@0: //
michael@0: // Redistribution and use in source and binary forms, with or without
michael@0: // modification, are permitted provided that the following conditions are
michael@0: // met:
michael@0: //
michael@0: //     * Redistributions of source code must retain the above copyright
michael@0: // notice, this list of conditions and the following disclaimer.
michael@0: //     * Redistributions in binary form must reproduce the above
michael@0: // copyright notice, this list of conditions and the following disclaimer
michael@0: // in the documentation and/or other materials provided with the
michael@0: // distribution.
michael@0: //     * Neither the name of Google Inc. nor the names of its
michael@0: // contributors may be used to endorse or promote products derived from
michael@0: // this software without specific prior written permission.
michael@0: //
michael@0: // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
michael@0: // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
michael@0: // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
michael@0: // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
michael@0: // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
michael@0: // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
michael@0: // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
michael@0: // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
michael@0: // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
michael@0: // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
michael@0: // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0: 
michael@0: // file_utils.cc: Implement utility functions for file manipulation.
michael@0: // See file_utils.h for details.
michael@0: 
michael@0: #include <fcntl.h>
michael@0: #include <stdio.h>
michael@0: #include <string.h>
michael@0: #include <sys/stat.h>
michael@0: #include <unistd.h>
michael@0: 
michael@0: #include "common/linux/eintr_wrapper.h"
michael@0: #include "common/tests/file_utils.h"
michael@0: 
michael@0: namespace google_breakpad {
michael@0: 
michael@0: bool CopyFile(const char* from_path, const char* to_path) {
michael@0:   int infile = HANDLE_EINTR(open(from_path, O_RDONLY));
michael@0:   if (infile < 0) {
michael@0:     perror("open");
michael@0:     return false;
michael@0:   }
michael@0: 
michael@0:   int outfile = HANDLE_EINTR(creat(to_path, 0666));
michael@0:   if (outfile < 0) {
michael@0:     perror("creat");
michael@0:     if (HANDLE_EINTR(close(infile)) < 0) {
michael@0:       perror("close");
michael@0:     }
michael@0:     return false;
michael@0:   }
michael@0: 
michael@0:   char buffer[1024];
michael@0:   bool result = true;
michael@0: 
michael@0:   while (result) {
michael@0:     ssize_t bytes_read = HANDLE_EINTR(read(infile, buffer, sizeof(buffer)));
michael@0:     if (bytes_read < 0) {
michael@0:       perror("read");
michael@0:       result = false;
michael@0:       break;
michael@0:     }
michael@0:     if (bytes_read == 0)
michael@0:       break;
michael@0:     ssize_t bytes_written_per_read = 0;
michael@0:     do {
michael@0:       ssize_t bytes_written_partial = HANDLE_EINTR(write(
michael@0:           outfile,
michael@0:           &buffer[bytes_written_per_read],
michael@0:           bytes_read - bytes_written_per_read));
michael@0:       if (bytes_written_partial < 0) {
michael@0:         perror("write");
michael@0:         result = false;
michael@0:         break;
michael@0:       }
michael@0:       bytes_written_per_read += bytes_written_partial;
michael@0:     } while (bytes_written_per_read < bytes_read);
michael@0:   }
michael@0: 
michael@0:   if (HANDLE_EINTR(close(infile)) == -1) {
michael@0:     perror("close");
michael@0:     result = false;
michael@0:   }
michael@0:   if (HANDLE_EINTR(close(outfile)) == -1) {
michael@0:     perror("close");
michael@0:     result = false;
michael@0:   }
michael@0: 
michael@0:   return result;
michael@0: }
michael@0: 
michael@0: bool ReadFile(const char* path, void* buffer, ssize_t* buffer_size) {
michael@0:   int fd = HANDLE_EINTR(open(path, O_RDONLY));
michael@0:   if (fd == -1) {
michael@0:     perror("open");
michael@0:     return false;
michael@0:   }
michael@0: 
michael@0:   bool ok = true;
michael@0:   if (buffer && buffer_size && *buffer_size > 0) {
michael@0:     memset(buffer, 0, sizeof(*buffer_size));
michael@0:     *buffer_size = HANDLE_EINTR(read(fd, buffer, *buffer_size));
michael@0:     if (*buffer_size == -1) {
michael@0:       perror("read");
michael@0:       ok = false;
michael@0:     }
michael@0:   }
michael@0:   if (HANDLE_EINTR(close(fd)) == -1) {
michael@0:     perror("close");
michael@0:     ok = false;
michael@0:   }
michael@0:   return ok;
michael@0: }
michael@0: 
michael@0: bool WriteFile(const char* path, const void* buffer, size_t buffer_size) {
michael@0:   int fd = HANDLE_EINTR(open(path, O_CREAT | O_TRUNC | O_WRONLY, S_IRWXU));
michael@0:   if (fd == -1) {
michael@0:     perror("open");
michael@0:     return false;
michael@0:   }
michael@0: 
michael@0:   bool ok = true;
michael@0:   if (buffer) {
michael@0:     size_t bytes_written_total = 0;
michael@0:     ssize_t bytes_written_partial = 0;
michael@0:     const char* data = reinterpret_cast<const char*>(buffer);
michael@0:     while (bytes_written_total < buffer_size) {
michael@0:       bytes_written_partial =
michael@0:           HANDLE_EINTR(write(fd, data + bytes_written_total,
michael@0:                              buffer_size - bytes_written_total));
michael@0:       if (bytes_written_partial < 0) {
michael@0:         perror("write");
michael@0:         ok = false;
michael@0:         break;
michael@0:       }
michael@0:       bytes_written_total += bytes_written_partial;
michael@0:     }
michael@0:   }
michael@0:   if (HANDLE_EINTR(close(fd)) == -1) {
michael@0:     perror("close");
michael@0:     ok = false;
michael@0:   }
michael@0:   return ok;
michael@0: }
michael@0: 
michael@0: }  // namespace google_breakpad