ext4_utils/make_ext4fs_main.c

   1 /*
   2  * Copyright (C) 2010 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 #include <fcntl.h>
  18 #include <libgen.h>
  19 #include <stdio.h>
  20 #include <unistd.h>
  21
  22 #if defined(__linux__)
  23 #include <linux/fs.h>
  24 #elif defined(__APPLE__) && defined(__MACH__)
  25 #include <sys/disk.h>
  26 #endif
  27
  28 #ifdef ANDROID
  29 #include <private/android_filesystem_config.h>
  30 #include <private/canned_fs_config.h>
  31 #endif
  32
  33 #ifndef _WIN32
  34 #include <selinux/selinux.h>
  35 #include <selinux/label.h>
  36 #if !defined(HOST)
  37 #include <selinux/android.h>
  38 #endif
  39 #else
  40 struct selabel_handle;
  41 #endif
  42
  43 #include "ext4_utils/ext4_utils.h"
  44 #include "ext4_utils/make_ext4fs.h"
  45
  46 #ifndef _WIN32 /* O_BINARY is windows-specific flag */
  47 #define O_BINARY 0
  48 #endif
  49
  50 extern struct fs_info info;
  51
  52
  53 static void usage(char *path)
  54 {
  55         fprintf(stderr, "%s [ -l <len> ] [ -j <journal size> ] [ -b <block_size> ]\n", basename(path));
  56         fprintf(stderr, "    [ -g <blocks per group> ] [ -i <inodes> ] [ -I <inode size> ]\n");
  57         fprintf(stderr, "    [ -e <flash erase block size> ] [ -o <flash logical block size> ]\n");
  58         fprintf(stderr, "    [ -L <label> ] [ -f ] [ -a <android mountpoint> ] [ -u ]\n");
  59         fprintf(stderr, "    [ -S file_contexts ] [ -C fs_config ] [ -T timestamp ]\n");
  60         fprintf(stderr, "    [ -z | -s ] [ -w ] [ -c ] [ -J ] [ -v ] [ -B <block_list_file> ]\n");
  61         fprintf(stderr, "    [ -d <base_alloc_file_in> ] [ -D <base_alloc_file_out> ]\n");
  62         fprintf(stderr, "    <filename> [[<directory>] <target_out_directory>]\n");
  63 }
  64
  65 int main(int argc, char **argv)
  66 {
  67         int opt;
  68         const char *filename = NULL;
  69         const char *directory = NULL;
  70         const char *target_out_directory = NULL;
  71         char *mountpoint = NULL;
  72         fs_config_func_t fs_config_func = NULL;
  73         const char *fs_config_file = NULL;
  74         int gzip = 0;
  75         int sparse = 0;
  76         int crc = 0;
  77         int wipe = 0;
  78         int real_uuid = 0;
  79         int fd;
  80         int exitcode;
  81         int verbose = 0;
  82         time_t fixed_time = -1;
  83         struct selabel_handle *sehnd = NULL;
  84         FILE* block_list_file = NULL;
  85         FILE* base_alloc_file_in = NULL;
  86         FILE* base_alloc_file_out = NULL;
  87 #ifndef _WIN32
  88         struct selinux_opt seopts[] = { { SELABEL_OPT_PATH, "" } };
  89 #endif
  90
  91         while ((opt = getopt(argc, argv, "l:j:b:g:i:I:e:o:L:a:S:T:C:B:d:D:fwzJsctvu")) != -1) {
  92                 switch (opt) {
  93                 case 'l':
  94                         info.len = parse_num(optarg);
  95                         break;
  96                 case 'j':
  97                         info.journal_blocks = parse_num(optarg);
  98                         break;
  99                 case 'b':
 100                         info.block_size = parse_num(optarg);
 101                         break;
 102                 case 'g':
 103                         info.blocks_per_group = parse_num(optarg);
 104                         break;
 105                 case 'i':
 106                         info.inodes = parse_num(optarg);
 107                         break;
 108                 case 'I':
 109                         info.inode_size = parse_num(optarg);
 110                         break;
 111                 case 'e':
 112                         info.flash_erase_block_size = parse_num(optarg);
 113                         break;
 114                 case 'o':
 115                         info.flash_logical_block_size = parse_num(optarg);
 116                         break;
 117                 case 'L':
 118                         info.label = optarg;
 119                         break;
 120                 case 'f':
 121                         force = 1;
 122                         break;
 123                 case 'a':
 124 #ifdef ANDROID
 125                         mountpoint = optarg;
 126 #else
 127                         fprintf(stderr, "can't set android permissions - built without android support\n");
 128                         usage(argv[0]);
 129                         exit(EXIT_FAILURE);
 130 #endif
 131                         break;
 132                 case 'w':
 133                         wipe = 1;
 134                         break;
 135                 case 'u':
 136                         real_uuid = 1;
 137                         break;
 138                 case 'z':
 139                         gzip = 1;
 140                         break;
 141                 case 'J':
 142                         info.no_journal = 1;
 143                         break;
 144                 case 'c':
 145                         crc = 1;
 146                         break;
 147                 case 's':
 148                         sparse = 1;
 149                         break;
 150                 case 't':
 151                         fprintf(stderr, "Warning: -t (initialize inode tables) is deprecated\n");
 152                         break;
 153                 case 'S':
 154 #ifndef _WIN32
 155                         seopts[0].value = optarg;
 156                         sehnd = selabel_open(SELABEL_CTX_FILE, seopts, 1);
 157                         if (!sehnd) {
 158                                 perror(optarg);
 159                                 exit(EXIT_FAILURE);
 160                         }
 161 #endif
 162                         break;
 163                 case 'v':
 164                         verbose = 1;
 165                         break;
 166                 case 'T':
 167                         fixed_time = strtoll(optarg, NULL, 0);
 168                         break;
 169                 case 'C':
 170                         fs_config_file = optarg;
 171                         break;
 172                 case 'B':
 173                         block_list_file = fopen(optarg, "w");
 174                         if (block_list_file == NULL) {
 175                                 fprintf(stderr, "failed to open block_list_file: %s\n", strerror(errno));
 176                                 exit(EXIT_FAILURE);
 177                         }
 178                         break;
 179                 case 'd':
 180                         base_alloc_file_in = fopen(optarg, "r");
 181                         if (base_alloc_file_in == NULL) {
 182                                 fprintf(stderr, "failed to open base_alloc_file_in: %s\n", strerror(errno));
 183                                 exit(EXIT_FAILURE);
 184                         }
 185                         break;
 186                 case 'D':
 187                         base_alloc_file_out = fopen(optarg, "w");
 188                         if (base_alloc_file_out == NULL) {
 189                                 fprintf(stderr, "failed to open base_alloc_file_out: %s\n", strerror(errno));
 190                                 exit(EXIT_FAILURE);
 191                         }
 192                         break;
 193                 default: /* '?' */
 194                         usage(argv[0]);
 195                         exit(EXIT_FAILURE);
 196                 }
 197         }
 198
 199 #if !defined(HOST)
 200         // Use only if -S option not requested
 201         if (!sehnd && mountpoint) {
 202                 sehnd = selinux_android_file_context_handle();
 203
 204                 if (!sehnd) {
 205                         perror(optarg);
 206                         exit(EXIT_FAILURE);
 207                 }
 208         }
 209 #endif
 210
 211         if (fs_config_file) {
 212                 if (load_canned_fs_config(fs_config_file) < 0) {
 213                         fprintf(stderr, "failed to load %s\n", fs_config_file);
 214                         exit(EXIT_FAILURE);
 215                 }
 216                 fs_config_func = canned_fs_config;
 217         } else if (mountpoint) {
 218                 fs_config_func = fs_config;
 219         }
 220
 221         if (wipe && sparse) {
 222                 fprintf(stderr, "Cannot specifiy both wipe and sparse\n");
 223                 usage(argv[0]);
 224                 exit(EXIT_FAILURE);
 225         }
 226
 227         if (wipe && gzip) {
 228                 fprintf(stderr, "Cannot specifiy both wipe and gzip\n");
 229                 usage(argv[0]);
 230                 exit(EXIT_FAILURE);
 231         }
 232
 233         if (optind >= argc) {
 234                 fprintf(stderr, "Expected filename after options\n");
 235                 usage(argv[0]);
 236                 exit(EXIT_FAILURE);
 237         }
 238
 239         filename = argv[optind++];
 240
 241         if (optind < argc)
 242                 directory = argv[optind++];
 243
 244         if (optind < argc)
 245                 target_out_directory = argv[optind++];
 246
 247         if (optind < argc) {
 248                 fprintf(stderr, "Unexpected argument: %s\n", argv[optind]);
 249                 usage(argv[0]);
 250                 exit(EXIT_FAILURE);
 251         }
 252
 253         if (strcmp(filename, "-")) {
 254                 fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0644);
 255                 if (fd < 0) {
 256                         perror("open");
 257                         return EXIT_FAILURE;
 258                 }
 259         } else {
 260                 fd = STDOUT_FILENO;
 261         }
 262
 263         exitcode = make_ext4fs_internal(fd, directory, target_out_directory, mountpoint, fs_config_func, gzip,
 264                 sparse, crc, wipe, real_uuid, sehnd, verbose, fixed_time,
 265                 block_list_file, base_alloc_file_in, base_alloc_file_out);
 266         close(fd);
 267         if (block_list_file)
 268                 fclose(block_list_file);
 269         if (base_alloc_file_out)
 270                 fclose(base_alloc_file_out);
 271         if (base_alloc_file_in)
 272                 fclose(base_alloc_file_in);
 273         if (exitcode && strcmp(filename, "-"))
 274                 unlink(filename);
 275         return exitcode;
 276 }