blob: 65e6cc29b3a8069d5bee322dbac8f80e9472b8ba [file] [log] [blame]
Geremy Condrade807f22013-07-08 14:04:02 -07001/*
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
Colin Cross0e3f47e2014-04-25 14:28:54 -070019#define _GNU_SOURCE
Geremy Condrade807f22013-07-08 14:04:02 -070020
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
44void usage()
45{
46 fprintf(stderr, "Usage: append2simg <output> <input>\n");
47}
48
49int 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
Colin Cross0e3f47e2014-04-25 14:28:54 -070060 int tmp_fd;
61 char *tmp_path;
62
63 int ret;
64
Geremy Condrade807f22013-07-08 14:04:02 -070065 if (argc == 3) {
66 output_path = argv[1];
67 input_path = argv[2];
68 } else {
69 usage();
70 exit(-1);
71 }
72
Colin Cross0e3f47e2014-04-25 14:28:54 -070073 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
Geremy Condrade807f22013-07-08 14:04:02 -070079 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, 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
Colin Cross0e3f47e2014-04-25 14:28:54 -0700114 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
Geremy Condrade807f22013-07-08 14:04:02 -0700120 lseek64(output, 0, SEEK_SET);
Colin Cross0e3f47e2014-04-25 14:28:54 -0700121 if (sparse_file_write(sparse_output, tmp_fd, false, true, false) < 0) {
Geremy Condrade807f22013-07-08 14:04:02 -0700122 fprintf(stderr, "Failed to write sparse file\n");
123 exit(-1);
124 }
125
126 sparse_file_destroy(sparse_output);
Colin Cross0e3f47e2014-04-25 14:28:54 -0700127 close(tmp_fd);
Geremy Condrade807f22013-07-08 14:04:02 -0700128 close(output);
129 close(input);
Colin Cross0e3f47e2014-04-25 14:28:54 -0700130
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
Geremy Condrade807f22013-07-08 14:04:02 -0700139 exit(0);
140}