1 /* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #define _FILE_OFFSET_BITS 64 18 #define _LARGEFILE64_SOURCE 1 19 #define _GNU_SOURCE 20 21 #include <errno.h> 22 #include <fcntl.h> 23 #include <stdio.h> 24 #include <stdlib.h> 25 #include <string.h> 26 #include <unistd.h> 27 28 #include <sparse/sparse.h> 29 #include "sparse_file.h" 30 #include "backed_block.h" 31 32 #ifndef O_BINARY 33 #define O_BINARY 0 34 #endif 35 36 #if defined(__APPLE__) && defined(__MACH__) 37 #define lseek64 lseek 38 #endif 39 #if defined(__APPLE__) && defined(__MACH__) 40 #define lseek64 lseek 41 #define off64_t off_t 42 #endif 43 44 void usage() 45 { 46 fprintf(stderr, "Usage: append2simg <output> <input>\n"); 47 } 48 49 int main(int argc, char *argv[]) 50 { 51 int output; 52 int output_block; 53 char *output_path; 54 struct sparse_file *sparse_output; 55 56 int input; 57 char *input_path; 58 off64_t input_len; 59 60 int tmp_fd; 61 char *tmp_path; 62 63 int ret; 64 65 if (argc == 3) { 66 output_path = argv[1]; 67 input_path = argv[2]; 68 } else { 69 usage(); 70 exit(-1); 71 } 72 73 ret = asprintf(&tmp_path, "%s.append2simg", output_path); 74 if (ret < 0) { 75 fprintf(stderr, "Couldn't allocate filename\n"); 76 exit(-1); 77 } 78 79 output = open(output_path, O_RDWR | O_BINARY); 80 if (output < 0) { 81 fprintf(stderr, "Couldn't open output file (%s)\n", strerror(errno)); 82 exit(-1); 83 } 84 85 sparse_output = sparse_file_import_auto(output, false, true); 86 if (!sparse_output) { 87 fprintf(stderr, "Couldn't import output file\n"); 88 exit(-1); 89 } 90 91 input = open(input_path, O_RDONLY | O_BINARY); 92 if (input < 0) { 93 fprintf(stderr, "Couldn't open input file (%s)\n", strerror(errno)); 94 exit(-1); 95 } 96 97 input_len = lseek64(input, 0, SEEK_END); 98 if (input_len < 0) { 99 fprintf(stderr, "Couldn't get input file length (%s)\n", strerror(errno)); 100 exit(-1); 101 } else if (input_len % sparse_output->block_size) { 102 fprintf(stderr, "Input file is not a multiple of the output file's block size"); 103 exit(-1); 104 } 105 lseek64(input, 0, SEEK_SET); 106 107 output_block = sparse_output->len / sparse_output->block_size; 108 if (sparse_file_add_fd(sparse_output, input, 0, input_len, output_block) < 0) { 109 fprintf(stderr, "Couldn't add input file\n"); 110 exit(-1); 111 } 112 sparse_output->len += input_len; 113 114 tmp_fd = open(tmp_path, O_WRONLY | O_CREAT | O_BINARY, 0664); 115 if (tmp_fd < 0) { 116 fprintf(stderr, "Couldn't open temporary file (%s)\n", strerror(errno)); 117 exit(-1); 118 } 119 120 lseek64(output, 0, SEEK_SET); 121 if (sparse_file_write(sparse_output, tmp_fd, false, true, false) < 0) { 122 fprintf(stderr, "Failed to write sparse file\n"); 123 exit(-1); 124 } 125 126 sparse_file_destroy(sparse_output); 127 close(tmp_fd); 128 close(output); 129 close(input); 130 131 ret = rename(tmp_path, output_path); 132 if (ret < 0) { 133 fprintf(stderr, "Failed to rename temporary file (%s)\n", strerror(errno)); 134 exit(-1); 135 } 136 137 free(tmp_path); 138 139 exit(0); 140 } 141