libsparse/simg2simg.c

   1 /*
   2  * Copyright (C) 2012 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 <fcntl.h>
  22 #include <stdbool.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28 #include <sys/types.h>
  29 #include <unistd.h>
  30
  31 #include <sparse/sparse.h>
  32
  33 #ifndef O_BINARY
  34 #define O_BINARY 0
  35 #endif
  36
  37 void usage()
  38 {
  39   fprintf(stderr, "Usage: simg2simg <sparse image file> <sparse_image_file> <max_size>\n");
  40 }
  41
  42 int main(int argc, char *argv[])
  43 {
  44         int in;
  45         int out;
  46         int i;
  47         int ret;
  48         struct sparse_file *s;
  49         int64_t max_size;
  50         struct sparse_file **out_s;
  51         int files;
  52         char filename[4096];
  53
  54         if (argc != 4) {
  55                 usage();
  56                 exit(-1);
  57         }
  58
  59         max_size = atoll(argv[3]);
  60
  61         in = open(argv[1], O_RDONLY | O_BINARY);
  62         if (in < 0) {
  63                 fprintf(stderr, "Cannot open input file %s\n", argv[1]);
  64                 exit(-1);
  65         }
  66
  67         s = sparse_file_import(in, true, false);
  68         if (!s) {
  69                 fprintf(stderr, "Failed to import sparse file\n");
  70                 exit(-1);
  71         }
  72
  73         files = sparse_file_resparse(s, max_size, NULL, 0);
  74         if (files < 0) {
  75                 fprintf(stderr, "Failed to resparse\n");
  76                 exit(-1);
  77         }
  78
  79         out_s = calloc(sizeof(struct sparse_file *), files);
  80         if (!out_s) {
  81                 fprintf(stderr, "Failed to allocate sparse file array\n");
  82                 exit(-1);
  83         }
  84
  85         files = sparse_file_resparse(s, max_size, out_s, files);
  86         if (files < 0) {
  87                 fprintf(stderr, "Failed to resparse\n");
  88                 exit(-1);
  89         }
  90
  91         for (i = 0; i < files; i++) {
  92                 ret = snprintf(filename, sizeof(filename), "%s.%d", argv[2], i);
  93                 if (ret >= (int)sizeof(filename)) {
  94                         fprintf(stderr, "Filename too long\n");
  95                         exit(-1);
  96                 }
  97
  98                 out = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0664);
  99                 if (out < 0) {
 100                         fprintf(stderr, "Cannot open output file %s\n", argv[2]);
 101                         exit(-1);
 102                 }
 103
 104                 ret = sparse_file_write(out_s[i], out, false, true, false);
 105                 if (ret) {
 106                         fprintf(stderr, "Failed to write sparse file\n");
 107                         exit(-1);
 108                 }
 109                 close(out);
 110         }
 111
 112         close(in);
 113
 114         exit(0);
 115 }