1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 // Some common file utilities for plugin code. 6 7 #include "ppapi/native_client/src/trusted/plugin/file_utils.h" 8 9 #include <fcntl.h> 10 #include <stdio.h> 11 #include <stdlib.h> 12 13 #include <sys/stat.h> 14 #include <sys/types.h> 15 16 #include "native_client/src/include/nacl_scoped_ptr.h" 17 #include "native_client/src/include/portability_io.h" 18 #include "native_client/src/include/portability_string.h" 19 20 21 namespace plugin { 22 namespace file_utils { 23 24 StatusCode SlurpFile(int32_t fd, 25 nacl::string& out_buf, 26 size_t max_size_to_read) { 27 struct stat stat_buf; 28 if (fstat(fd, &stat_buf) != 0) { 29 CLOSE(fd); 30 return PLUGIN_FILE_ERROR_STAT; 31 } 32 33 // Figure out how large a buffer we need to slurp the whole file (with a 34 // '\0' at the end). 35 size_t bytes_to_read = static_cast<size_t>(stat_buf.st_size); 36 if (bytes_to_read > max_size_to_read - 1) { 37 CLOSE(fd); 38 return PLUGIN_FILE_ERROR_FILE_TOO_LARGE; 39 } 40 41 FILE* input_file = fdopen(fd, "rb"); 42 if (!input_file) { 43 CLOSE(fd); 44 return PLUGIN_FILE_ERROR_OPEN; 45 } 46 // From here on, closing input_file will automatically close fd. 47 48 nacl::scoped_array<char> buffer(new char[bytes_to_read + 1]); 49 if (buffer == NULL) { 50 fclose(input_file); 51 return PLUGIN_FILE_ERROR_MEM_ALLOC; 52 } 53 54 size_t total_bytes_read = 0; 55 while (bytes_to_read > 0) { 56 size_t bytes_this_read = fread(&buffer[total_bytes_read], 57 sizeof(char), 58 bytes_to_read, 59 input_file); 60 if (bytes_this_read < bytes_to_read && (feof(input_file) || 61 ferror(input_file))) { 62 fclose(input_file); 63 return PLUGIN_FILE_ERROR_READ; 64 } 65 total_bytes_read += bytes_this_read; 66 bytes_to_read -= bytes_this_read; 67 } 68 69 fclose(input_file); 70 buffer[total_bytes_read] = '\0'; 71 out_buf = buffer.get(); 72 return PLUGIN_FILE_SUCCESS; 73 } 74 75 } // namespace file_utils 76 } // namespace plugin 77 78
