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