mkfs/mkfs.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  *   Copyright (C) 2019 Namjae Jeon <linkinjeon@kernel.org>
   4  */
   5
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <stdbool.h>
   9 #include <string.h>
  10 #include <unistd.h>
  11 #include <sys/types.h>
  12 #include <sys/stat.h>
  13 #include <sys/ioctl.h>
  14 #include <fcntl.h>
  15 #include <getopt.h>
  16 #include <inttypes.h>
  17 #include <limits.h>
  18 #include <errno.h>
  19 #include <locale.h>
  20 #include <time.h>
  21
  22 #include "exfat_ondisk.h"
  23 #include "libexfat.h"
  24 #include "mkfs.h"
  25
  26 struct exfat_mkfs_info finfo;
  27
  28 /* random serial generator based on current time */
  29 static unsigned int get_new_serial(void)
  30 {
  31         struct timespec ts;
  32
  33         if (clock_gettime(CLOCK_REALTIME, &ts)) {
  34                 /* set 0000-0000 on error */
  35                 ts.tv_sec = 0;
  36                 ts.tv_nsec = 0;
  37         }
  38
  39         return (unsigned int)(ts.tv_nsec << 12 | ts.tv_sec);
  40 }
  41
  42 static void exfat_setup_boot_sector(struct pbr *ppbr,
  43                 struct exfat_blk_dev *bd, struct exfat_user_input *ui)
  44 {
  45         struct bpb64 *pbpb = &ppbr->bpb;
  46         struct bsx64 *pbsx = &ppbr->bsx;
  47         unsigned int i;
  48
  49         /* Fill exfat BIOS parameter block */
  50         pbpb->jmp_boot[0] = 0xeb;
  51         pbpb->jmp_boot[1] = 0x76;
  52         pbpb->jmp_boot[2] = 0x90;
  53         memcpy(pbpb->oem_name, "EXFAT   ", 8);
  54         memset(pbpb->res_zero, 0, 53);
  55
  56         /* Fill exfat extend BIOS parameter block */
  57         pbsx->vol_offset = cpu_to_le64(bd->offset / bd->sector_size);
  58         pbsx->vol_length = cpu_to_le64(bd->size / bd->sector_size);
  59         pbsx->fat_offset = cpu_to_le32(finfo.fat_byte_off / bd->sector_size);
  60         pbsx->fat_length = cpu_to_le32(finfo.fat_byte_len / bd->sector_size);
  61         pbsx->clu_offset = cpu_to_le32(finfo.clu_byte_off / bd->sector_size);
  62         pbsx->clu_count = cpu_to_le32(finfo.total_clu_cnt);
  63         pbsx->root_cluster = cpu_to_le32(finfo.root_start_clu);
  64         pbsx->vol_serial = cpu_to_le32(finfo.volume_serial);
  65         pbsx->vol_flags = 0;
  66         pbsx->sect_size_bits = bd->sector_size_bits;
  67         pbsx->sect_per_clus_bits = 0;
  68         /* Compute base 2 logarithm of ui->cluster_size / bd->sector_size */
  69         for (i = ui->cluster_size / bd->sector_size; i > 1; i /= 2)
  70                 pbsx->sect_per_clus_bits++;
  71         pbsx->num_fats = 1;
  72         /* fs_version[0] : minor and fs_version[1] : major */
  73         pbsx->fs_version[0] = 0;
  74         pbsx->fs_version[1] = 1;
  75         pbsx->phy_drv_no = 0x80;
  76         memset(pbsx->reserved2, 0, 7);
  77
  78         memset(ppbr->boot_code, 0, 390);
  79         ppbr->signature = cpu_to_le16(PBR_SIGNATURE);
  80
  81         exfat_debug("Volume Offset(sectors) : %" PRIu64 "\n",
  82                 le64_to_cpu(pbsx->vol_offset));
  83         exfat_debug("Volume Length(sectors) : %" PRIu64 "\n",
  84                 le64_to_cpu(pbsx->vol_length));
  85         exfat_debug("FAT Offset(sector offset) : %u\n",
  86                 le32_to_cpu(pbsx->fat_offset));
  87         exfat_debug("FAT Length(sectors) : %u\n",
  88                 le32_to_cpu(pbsx->fat_length));
  89         exfat_debug("Cluster Heap Offset (sector offset) : %u\n",
  90                 le32_to_cpu(pbsx->clu_offset));
  91         exfat_debug("Cluster Count : %u\n",
  92                 le32_to_cpu(pbsx->clu_count));
  93         exfat_debug("Root Cluster (cluster offset) : %u\n",
  94                 le32_to_cpu(pbsx->root_cluster));
  95         exfat_debug("Volume Serial : 0x%x\n", le32_to_cpu(pbsx->vol_serial));
  96         exfat_debug("Sector Size Bits : %u\n",
  97                 pbsx->sect_size_bits);
  98         exfat_debug("Sector per Cluster bits : %u\n",
  99                 pbsx->sect_per_clus_bits);
 100 }
 101
 102 static int exfat_write_boot_sector(struct exfat_blk_dev *bd,
 103                 struct exfat_user_input *ui, unsigned int *checksum,
 104                 bool is_backup)
 105 {
 106         struct pbr *ppbr;
 107         unsigned int sec_idx = BOOT_SEC_IDX;
 108         int ret = 0;
 109
 110         if (is_backup)
 111                 sec_idx += BACKUP_BOOT_SEC_IDX;
 112
 113         ppbr = malloc(bd->sector_size);
 114         if (!ppbr) {
 115                 exfat_err("Cannot allocate pbr: out of memory\n");
 116                 return -1;
 117         }
 118         memset(ppbr, 0, bd->sector_size);
 119
 120         exfat_setup_boot_sector(ppbr, bd, ui);
 121
 122         /* write main boot sector */
 123         ret = exfat_write_sector(bd, ppbr, sec_idx);
 124         if (ret < 0) {
 125                 exfat_err("main boot sector write failed\n");
 126                 ret = -1;
 127                 goto free_ppbr;
 128         }
 129
 130         boot_calc_checksum((unsigned char *)ppbr, bd->sector_size,
 131                 true, checksum);
 132
 133 free_ppbr:
 134         free(ppbr);
 135         return ret;
 136 }
 137
 138 static int exfat_write_extended_boot_sectors(struct exfat_blk_dev *bd,
 139                 unsigned int *checksum, bool is_backup)
 140 {
 141         char *peb;
 142         __le16 *peb_signature;
 143         int ret = 0;
 144         int i;
 145         unsigned int sec_idx = EXBOOT_SEC_IDX;
 146
 147         peb = malloc(bd->sector_size);
 148         if (!peb)
 149                 return -1;
 150
 151         if (is_backup)
 152                 sec_idx += BACKUP_BOOT_SEC_IDX;
 153
 154         memset(peb, 0, bd->sector_size);
 155         peb_signature = (__le16*) (peb + bd->sector_size - 2);
 156         *peb_signature = cpu_to_le16(PBR_SIGNATURE);
 157         for (i = 0; i < EXBOOT_SEC_NUM; i++) {
 158                 if (exfat_write_sector(bd, peb, sec_idx++)) {
 159                         exfat_err("extended boot sector write failed\n");
 160                         ret = -1;
 161                         goto free_peb;
 162                 }
 163
 164                 boot_calc_checksum((unsigned char *) peb, bd->sector_size,
 165                         false, checksum);
 166         }
 167
 168 free_peb:
 169         free(peb);
 170         return ret;
 171 }
 172
 173 static int exfat_write_oem_sector(struct exfat_blk_dev *bd,
 174                 unsigned int *checksum, bool is_backup)
 175 {
 176         char *oem;
 177         int ret = 0;
 178         unsigned int sec_idx = OEM_SEC_IDX;
 179
 180         oem = malloc(bd->sector_size);
 181         if (!oem)
 182                 return -1;
 183
 184         if (is_backup)
 185                 sec_idx += BACKUP_BOOT_SEC_IDX;
 186
 187         memset(oem, 0xFF, bd->sector_size);
 188         ret = exfat_write_sector(bd, oem, sec_idx);
 189         if (ret) {
 190                 exfat_err("oem sector write failed\n");
 191                 ret = -1;
 192                 goto free_oem;
 193         }
 194
 195         boot_calc_checksum((unsigned char *)oem, bd->sector_size, false,
 196                 checksum);
 197
 198         /* Zero out reserved sector */
 199         memset(oem, 0, bd->sector_size);
 200         ret = exfat_write_sector(bd, oem, sec_idx + 1);
 201         if (ret) {
 202                 exfat_err("reserved sector write failed\n");
 203                 ret = -1;
 204                 goto free_oem;
 205         }
 206
 207         boot_calc_checksum((unsigned char *)oem, bd->sector_size, false,
 208                 checksum);
 209
 210 free_oem:
 211         free(oem);
 212         return ret;
 213 }
 214
 215 static int exfat_create_volume_boot_record(struct exfat_blk_dev *bd,
 216                 struct exfat_user_input *ui, bool is_backup)
 217 {
 218         unsigned int checksum = 0;
 219         int ret;
 220
 221         ret = exfat_write_boot_sector(bd, ui, &checksum, is_backup);
 222         if (ret)
 223                 return ret;
 224         ret = exfat_write_extended_boot_sectors(bd, &checksum, is_backup);
 225         if (ret)
 226                 return ret;
 227         ret = exfat_write_oem_sector(bd, &checksum, is_backup);
 228         if (ret)
 229                 return ret;
 230
 231         return exfat_write_checksum_sector(bd, checksum, is_backup);
 232 }
 233
 234 static int write_fat_entry(int fd, __le32 clu,
 235                 unsigned long long offset)
 236 {
 237         int nbyte;
 238         off_t fat_entry_offset = finfo.fat_byte_off + (offset * sizeof(__le32));
 239
 240         nbyte = pwrite(fd, (__u8 *) &clu, sizeof(__le32), fat_entry_offset);
 241         if (nbyte != sizeof(int)) {
 242                 exfat_err("write failed, offset : %llu, clu : %x\n",
 243                         offset, clu);
 244                 return -1;
 245         }
 246
 247         return 0;
 248 }
 249
 250 static int write_fat_entries(struct exfat_user_input *ui, int fd,
 251                 unsigned int clu, unsigned int length)
 252 {
 253         int ret;
 254         unsigned int count;
 255
 256         count = clu + round_up(length, ui->cluster_size) / ui->cluster_size;
 257
 258         for (; clu < count - 1; clu++) {
 259                 ret = write_fat_entry(fd, cpu_to_le32(clu + 1), clu);
 260                 if (ret)
 261                         return ret;
 262         }
 263
 264         ret = write_fat_entry(fd, cpu_to_le32(EXFAT_EOF_CLUSTER), clu);
 265         if (ret)
 266                 return ret;
 267
 268         return clu;
 269 }
 270
 271 static int exfat_create_fat_table(struct exfat_blk_dev *bd,
 272                 struct exfat_user_input *ui)
 273 {
 274         int ret, clu;
 275
 276         /* fat entry 0 should be media type field(0xF8) */
 277         ret = write_fat_entry(bd->dev_fd, cpu_to_le32(0xfffffff8), 0);
 278         if (ret) {
 279                 exfat_err("fat 0 entry write failed\n");
 280                 return ret;
 281         }
 282
 283         /* fat entry 1 is historical precedence(0xFFFFFFFF) */
 284         ret = write_fat_entry(bd->dev_fd, cpu_to_le32(0xffffffff), 1);
 285         if (ret) {
 286                 exfat_err("fat 1 entry write failed\n");
 287                 return ret;
 288         }
 289
 290         /* write bitmap entries */
 291         clu = write_fat_entries(ui, bd->dev_fd, EXFAT_FIRST_CLUSTER,
 292                 finfo.bitmap_byte_len);
 293         if (clu < 0)
 294                 return ret;
 295
 296         /* write upcase table entries */
 297         clu = write_fat_entries(ui, bd->dev_fd, clu + 1, finfo.ut_byte_len);
 298         if (clu < 0)
 299                 return ret;
 300
 301         /* write root directory entries */
 302         clu = write_fat_entries(ui, bd->dev_fd, clu + 1, finfo.root_byte_len);
 303         if (clu < 0)
 304                 return ret;
 305
 306         finfo.used_clu_cnt = clu + 1;
 307         exfat_debug("Total used cluster count : %d\n", finfo.used_clu_cnt);
 308
 309         return ret;
 310 }
 311
 312 static int exfat_create_bitmap(struct exfat_blk_dev *bd)
 313 {
 314         char *bitmap;
 315         unsigned int i, nbytes;
 316
 317         bitmap = calloc(round_up(finfo.bitmap_byte_len, sizeof(bitmap_t)),
 318                         sizeof(*bitmap));
 319         if (!bitmap)
 320                 return -1;
 321
 322         for (i = EXFAT_FIRST_CLUSTER; i < finfo.used_clu_cnt; i++)
 323                 exfat_bitmap_set(bitmap, i);
 324
 325         nbytes = pwrite(bd->dev_fd, bitmap, finfo.bitmap_byte_len, finfo.bitmap_byte_off);
 326         if (nbytes != finfo.bitmap_byte_len) {
 327                 exfat_err("write failed, nbytes : %d, bitmap_len : %d\n",
 328                         nbytes, finfo.bitmap_byte_len);
 329                 free(bitmap);
 330                 return -1;
 331         }
 332
 333         free(bitmap);
 334         return 0;
 335 }
 336
 337 static int exfat_create_root_dir(struct exfat_blk_dev *bd,
 338                 struct exfat_user_input *ui)
 339 {
 340         struct exfat_dentry ed[4] = {0};
 341         int dentries_len = sizeof(ed);
 342         int nbytes;
 343
 344         /* Set volume label entry */
 345         ed[0].type = EXFAT_VOLUME;
 346         memset(ed[0].vol_label, 0, 22);
 347         memcpy(ed[0].vol_label, ui->volume_label, ui->volume_label_len);
 348         ed[0].vol_char_cnt = ui->volume_label_len/2;
 349
 350         /* Set volume GUID entry */
 351         if (ui->guid) {
 352                 if (__exfat_set_volume_guid(&ed[1], ui->guid))
 353                         return -1;
 354         } else {
 355                 /*
 356                  * Since a single empty entry cannot be allocated for a
 357                  * file, this can reserve the entry for volume GUID.
 358                  */
 359                 ed[1].type = EXFAT_GUID & ~EXFAT_INVAL;
 360         }
 361
 362         /* Set bitmap entry */
 363         ed[2].type = EXFAT_BITMAP;
 364         ed[2].bitmap_flags = 0;
 365         ed[2].bitmap_start_clu = cpu_to_le32(EXFAT_FIRST_CLUSTER);
 366         ed[2].bitmap_size = cpu_to_le64(finfo.bitmap_byte_len);
 367
 368         /* Set upcase table entry */
 369         ed[3].type = EXFAT_UPCASE;
 370         ed[3].upcase_checksum = cpu_to_le32(0xe619d30d);
 371         ed[3].upcase_start_clu = cpu_to_le32(finfo.ut_start_clu);
 372         ed[3].upcase_size = cpu_to_le64(EXFAT_UPCASE_TABLE_SIZE);
 373
 374         nbytes = pwrite(bd->dev_fd, ed, dentries_len, finfo.root_byte_off);
 375         if (nbytes != dentries_len) {
 376                 exfat_err("write failed, nbytes : %d, dentries_len : %d\n",
 377                         nbytes, dentries_len);
 378                 return -1;
 379         }
 380
 381         return 0;
 382 }
 383
 384 static void usage(void)
 385 {
 386         fputs("Usage: mkfs.exfat\n"
 387                 "\t-L | --volume-label=label                              Set volume label\n"
 388                 "\t-U | --volume-guid=guid                                Set volume GUID\n"
 389                 "\t-c | --cluster-size=size(or suffixed by 'K' or 'M')    Specify cluster size\n"
 390                 "\t-b | --boundary-align=size(or suffixed by 'K' or 'M')  Specify boundary alignment\n"
 391                 "\t     --pack-bitmap                                     Move bitmap into FAT segment\n"
 392                 "\t-f | --full-format                                     Full format\n"
 393                 "\t-V | --version                                         Show version\n"
 394                 "\t-q | --quiet                                           Print only errors\n"
 395                 "\t-v | --verbose                                         Print debug\n"
 396                 "\t-h | --help                                            Show help\n",
 397                 stderr);
 398
 399         exit(EXIT_FAILURE);
 400 }
 401
 402 #define PACK_BITMAP (CHAR_MAX + 1)
 403
 404 static const struct option opts[] = {
 405         {"volume-label",        required_argument,      NULL,   'L' },
 406         {"volume-guid",         required_argument,      NULL,   'U' },
 407         {"cluster-size",        required_argument,      NULL,   'c' },
 408         {"boundary-align",      required_argument,      NULL,   'b' },
 409         {"pack-bitmap",         no_argument,            NULL,   PACK_BITMAP },
 410         {"full-format",         no_argument,            NULL,   'f' },
 411         {"version",             no_argument,            NULL,   'V' },
 412         {"quiet",               no_argument,            NULL,   'q' },
 413         {"verbose",             no_argument,            NULL,   'v' },
 414         {"help",                no_argument,            NULL,   'h' },
 415         {"?",                   no_argument,            NULL,   '?' },
 416         {NULL,                  0,                      NULL,    0  }
 417 };
 418
 419 /*
 420  * Moves the bitmap to just before the alignment boundary if there is space
 421  * between the boundary and the end of the FAT. This may allow the FAT and the
 422  * bitmap to share the same allocation unit on flash media, thereby improving
 423  * performance and endurance.
 424  */
 425 static int exfat_pack_bitmap(const struct exfat_user_input *ui)
 426 {
 427         unsigned int fat_byte_end = finfo.fat_byte_off + finfo.fat_byte_len,
 428                 bitmap_byte_len = finfo.bitmap_byte_len,
 429                 bitmap_clu_len = round_up(bitmap_byte_len, ui->cluster_size),
 430                 bitmap_clu_cnt, total_clu_cnt, new_bitmap_clu_len;
 431
 432         for (;;) {
 433                 bitmap_clu_cnt = bitmap_clu_len / ui->cluster_size;
 434                 if (finfo.clu_byte_off - bitmap_clu_len < fat_byte_end ||
 435                                 finfo.total_clu_cnt > EXFAT_MAX_NUM_CLUSTER -
 436                                         bitmap_clu_cnt)
 437                         return -1;
 438                 total_clu_cnt = finfo.total_clu_cnt + bitmap_clu_cnt;
 439                 bitmap_byte_len = round_up(total_clu_cnt, 8) / 8;
 440                 new_bitmap_clu_len = round_up(bitmap_byte_len, ui->cluster_size);
 441                 if (new_bitmap_clu_len == bitmap_clu_len) {
 442                         finfo.clu_byte_off -= bitmap_clu_len;
 443                         finfo.total_clu_cnt = total_clu_cnt;
 444                         finfo.bitmap_byte_off -= bitmap_clu_len;
 445                         finfo.bitmap_byte_len = bitmap_byte_len;
 446                         return 0;
 447                 }
 448                 bitmap_clu_len = new_bitmap_clu_len;
 449         }
 450 }
 451
 452 static int exfat_build_mkfs_info(struct exfat_blk_dev *bd,
 453                 struct exfat_user_input *ui)
 454 {
 455         unsigned long long total_clu_cnt;
 456         int clu_len;
 457
 458         if (ui->cluster_size < bd->sector_size) {
 459                 exfat_err("cluster size (%u bytes) is smaller than sector size (%u bytes)\n",
 460                           ui->cluster_size, bd->sector_size);
 461                 return -1;
 462         }
 463         if (ui->boundary_align < bd->sector_size) {
 464                 exfat_err("boundary alignment is too small (min %d)\n",
 465                                 bd->sector_size);
 466                 return -1;
 467         }
 468         finfo.fat_byte_off = round_up(bd->offset + 24 * bd->sector_size,
 469                         ui->boundary_align) - bd->offset;
 470         /* Prevent integer overflow when computing the FAT length */
 471         if (bd->num_clusters > UINT32_MAX / 4) {
 472                 exfat_err("cluster size (%u bytes) is too small\n", ui->cluster_size);
 473                 return -1;
 474         }
 475         finfo.fat_byte_len = round_up((bd->num_clusters * 4), ui->cluster_size);
 476         finfo.clu_byte_off = round_up(bd->offset + finfo.fat_byte_off +
 477                 finfo.fat_byte_len, ui->boundary_align) - bd->offset;
 478         if (bd->size <= finfo.clu_byte_off) {
 479                 exfat_err("boundary alignment is too big\n");
 480                 return -1;
 481         }
 482         total_clu_cnt = (bd->size - finfo.clu_byte_off) / ui->cluster_size;
 483         if (total_clu_cnt > EXFAT_MAX_NUM_CLUSTER) {
 484                 exfat_err("cluster size is too small\n");
 485                 return -1;
 486         }
 487         finfo.total_clu_cnt = (unsigned int) total_clu_cnt;
 488
 489         finfo.bitmap_byte_off = finfo.clu_byte_off;
 490         finfo.bitmap_byte_len = round_up(finfo.total_clu_cnt, 8) / 8;
 491         if (ui->pack_bitmap)
 492                 exfat_pack_bitmap(ui);
 493         clu_len = round_up(finfo.bitmap_byte_len, ui->cluster_size);
 494
 495         finfo.ut_start_clu = EXFAT_FIRST_CLUSTER + clu_len / ui->cluster_size;
 496         finfo.ut_byte_off = finfo.bitmap_byte_off + clu_len;
 497         finfo.ut_byte_len = EXFAT_UPCASE_TABLE_SIZE;
 498         clu_len = round_up(finfo.ut_byte_len, ui->cluster_size);
 499
 500         finfo.root_start_clu = finfo.ut_start_clu + clu_len / ui->cluster_size;
 501         finfo.root_byte_off = finfo.ut_byte_off + clu_len;
 502         finfo.root_byte_len = sizeof(struct exfat_dentry) * 3;
 503         finfo.volume_serial = get_new_serial();
 504
 505         return 0;
 506 }
 507
 508 static int exfat_zero_out_disk(struct exfat_blk_dev *bd,
 509                 struct exfat_user_input *ui)
 510 {
 511         int nbytes;
 512         unsigned long long total_written = 0;
 513         char *buf;
 514         unsigned int chunk_size = ui->cluster_size;
 515         unsigned long long size;
 516
 517         if (ui->quick)
 518                 size = finfo.root_byte_off + chunk_size;
 519         else
 520                 size = bd->size;
 521
 522         buf = malloc(chunk_size);
 523         if (!buf)
 524                 return -1;
 525
 526         memset(buf, 0, chunk_size);
 527         lseek(bd->dev_fd, 0, SEEK_SET);
 528         do {
 529
 530                 nbytes = write(bd->dev_fd, buf, chunk_size);
 531                 if (nbytes <= 0) {
 532                         if (nbytes < 0)
 533                                 exfat_err("write failed(errno : %d)\n", errno);
 534                         break;
 535                 }
 536                 total_written += nbytes;
 537         } while (total_written < size);
 538
 539         free(buf);
 540         exfat_debug("zero out written size : %llu, disk size : %llu\n",
 541                 total_written, bd->size);
 542         return 0;
 543 }
 544
 545 static int make_exfat(struct exfat_blk_dev *bd, struct exfat_user_input *ui)
 546 {
 547         int ret;
 548
 549         exfat_info("Creating exFAT filesystem(%s, cluster size=%u)\n\n",
 550                 ui->dev_name, ui->cluster_size);
 551
 552         exfat_info("Writing volume boot record: ");
 553         ret = exfat_create_volume_boot_record(bd, ui, 0);
 554         exfat_info("%s\n", ret ? "failed" : "done");
 555         if (ret)
 556                 return ret;
 557
 558         exfat_info("Writing backup volume boot record: ");
 559         /* backup sector */
 560         ret = exfat_create_volume_boot_record(bd, ui, 1);
 561         exfat_info("%s\n", ret ? "failed" : "done");
 562         if (ret)
 563                 return ret;
 564
 565         exfat_info("Fat table creation: ");
 566         ret = exfat_create_fat_table(bd, ui);
 567         exfat_info("%s\n", ret ? "failed" : "done");
 568         if (ret)
 569                 return ret;
 570
 571         exfat_info("Allocation bitmap creation: ");
 572         ret = exfat_create_bitmap(bd);
 573         exfat_info("%s\n", ret ? "failed" : "done");
 574         if (ret)
 575                 return ret;
 576
 577         exfat_info("Upcase table creation: ");
 578         ret = exfat_create_upcase_table(bd);
 579         exfat_info("%s\n", ret ? "failed" : "done");
 580         if (ret)
 581                 return ret;
 582
 583         exfat_info("Writing root directory entry: ");
 584         ret = exfat_create_root_dir(bd, ui);
 585         exfat_info("%s\n", ret ? "failed" : "done");
 586         if (ret)
 587                 return ret;
 588
 589         return 0;
 590 }
 591
 592 static long long parse_size(const char *size)
 593 {
 594         char *data_unit;
 595         unsigned long long byte_size = strtoull(size, &data_unit, 0);
 596
 597         switch (*data_unit) {
 598         case 'M':
 599         case 'm':
 600                 byte_size <<= 20;
 601                 break;
 602         case 'K':
 603         case 'k':
 604                 byte_size <<= 10;
 605                 break;
 606         case '\0':
 607                 break;
 608         default:
 609                 exfat_err("Wrong unit input('%c') for size\n",
 610                                 *data_unit);
 611                 return -EINVAL;
 612         }
 613
 614         return byte_size;
 615 }
 616
 617 int main(int argc, char *argv[])
 618 {
 619         int c;
 620         int ret = EXIT_FAILURE;
 621         struct exfat_blk_dev bd;
 622         struct exfat_user_input ui;
 623         bool version_only = false;
 624         bool quiet = false;
 625
 626         init_user_input(&ui);
 627
 628         if (!setlocale(LC_CTYPE, ""))
 629                 exfat_err("failed to init locale/codeset\n");
 630
 631         opterr = 0;
 632         while ((c = getopt_long(argc, argv, "n:L:U:c:b:fVqvh", opts, NULL)) != EOF)
 633                 switch (c) {
 634                 /*
 635                  * Make 'n' option fallthrough to 'L' option for for backward
 636                  * compatibility with old utils.
 637                  */
 638                 case 'n':
 639                 case 'L':
 640                 {
 641                         ret = exfat_utf16_enc(optarg,
 642                                 ui.volume_label, sizeof(ui.volume_label));
 643                         if (ret < 0)
 644                                 goto out;
 645
 646                         ui.volume_label_len = ret;
 647                         break;
 648                 }
 649                 case 'U':
 650                         if (*optarg != '\0' && *optarg != '\r')
 651                                 ui.guid = optarg;
 652                         break;
 653                 case 'c':
 654                         ret = parse_size(optarg);
 655                         if (ret < 0)
 656                                 goto out;
 657                         else if (ret & (ret - 1)) {
 658                                 exfat_err("cluster size(%d) is not a power of 2)\n",
 659                                         ret);
 660                                 goto out;
 661                         } else if (ret > EXFAT_MAX_CLUSTER_SIZE) {
 662                                 exfat_err("cluster size(%d) exceeds max cluster size(%d)\n",
 663                                         ui.cluster_size, EXFAT_MAX_CLUSTER_SIZE);
 664                                 goto out;
 665                         }
 666                         ui.cluster_size = ret;
 667                         break;
 668                 case 'b':
 669                         ret = parse_size(optarg);
 670                         if (ret < 0)
 671                                 goto out;
 672                         else if (ret & (ret - 1)) {
 673                                 exfat_err("boundary align(%d) is not a power of 2)\n",
 674                                         ret);
 675                                 goto out;
 676                         }
 677                         ui.boundary_align = ret;
 678                         break;
 679                 case PACK_BITMAP:
 680                         ui.pack_bitmap = true;
 681                         break;
 682                 case 'f':
 683                         ui.quick = false;
 684                         break;
 685                 case 'V':
 686                         version_only = true;
 687                         break;
 688                 case 'q':
 689                         print_level = EXFAT_ERROR;
 690                         quiet = true;
 691                         break;
 692                 case 'v':
 693                         print_level = EXFAT_DEBUG;
 694                         break;
 695                 case '?':
 696                 case 'h':
 697                 default:
 698                         usage();
 699         }
 700
 701         if (version_only) {
 702                 show_version();
 703                 exit(EXIT_FAILURE);
 704         } else if (!quiet) {
 705                 show_version();
 706         }
 707
 708         if (argc - optind != 1) {
 709                 usage();
 710         }
 711
 712         memset(ui.dev_name, 0, sizeof(ui.dev_name));
 713         snprintf(ui.dev_name, sizeof(ui.dev_name), "%s", argv[optind]);
 714
 715         ret = exfat_get_blk_dev_info(&ui, &bd);
 716         if (ret < 0)
 717                 goto out;
 718
 719         ret = exfat_build_mkfs_info(&bd, &ui);
 720         if (ret)
 721                 goto close;
 722
 723         ret = exfat_zero_out_disk(&bd, &ui);
 724         if (ret)
 725                 goto close;
 726
 727         ret = make_exfat(&bd, &ui);
 728         if (ret)
 729                 goto close;
 730
 731         exfat_info("Synchronizing...\n");
 732         ret = fsync(bd.dev_fd);
 733 close:
 734         close(bd.dev_fd);
 735 out:
 736         if (!ret)
 737                 exfat_info("\nexFAT format complete!\n");
 738         else
 739                 exfat_err("\nexFAT format fail!\n");
 740         return ret;
 741 }