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(finfo.bitmap_byte_len, sizeof(*bitmap));
 318         if (!bitmap)
 319                 return -1;
 320
 321         for (i = 0; i < finfo.used_clu_cnt - EXFAT_FIRST_CLUSTER; i++)
 322                 exfat_set_bit(bd, bitmap, i);
 323
 324         nbytes = pwrite(bd->dev_fd, bitmap, finfo.bitmap_byte_len, finfo.bitmap_byte_off);
 325         if (nbytes != finfo.bitmap_byte_len) {
 326                 exfat_err("write failed, nbytes : %d, bitmap_len : %d\n",
 327                         nbytes, finfo.bitmap_byte_len);
 328                 free(bitmap);
 329                 return -1;
 330         }
 331
 332         free(bitmap);
 333         return 0;
 334 }
 335
 336 static int exfat_create_root_dir(struct exfat_blk_dev *bd,
 337                 struct exfat_user_input *ui)
 338 {
 339         struct exfat_dentry ed[3];
 340         int dentries_len = sizeof(struct exfat_dentry) * 3;
 341         int nbytes;
 342
 343         /* Set volume label entry */
 344         ed[0].type = EXFAT_VOLUME;
 345         memset(ed[0].vol_label, 0, 22);
 346         memcpy(ed[0].vol_label, ui->volume_label, ui->volume_label_len);
 347         ed[0].vol_char_cnt = ui->volume_label_len/2;
 348
 349         /* Set bitmap entry */
 350         ed[1].type = EXFAT_BITMAP;
 351         ed[1].bitmap_flags = 0;
 352         ed[1].bitmap_start_clu = cpu_to_le32(EXFAT_FIRST_CLUSTER);
 353         ed[1].bitmap_size = cpu_to_le64(finfo.bitmap_byte_len);
 354
 355         /* Set upcase table entry */
 356         ed[2].type = EXFAT_UPCASE;
 357         ed[2].upcase_checksum = cpu_to_le32(0xe619d30d);
 358         ed[2].upcase_start_clu = cpu_to_le32(finfo.ut_start_clu);
 359         ed[2].upcase_size = cpu_to_le64(EXFAT_UPCASE_TABLE_SIZE);
 360
 361         nbytes = pwrite(bd->dev_fd, ed, dentries_len, finfo.root_byte_off);
 362         if (nbytes != dentries_len) {
 363                 exfat_err("write failed, nbytes : %d, dentries_len : %d\n",
 364                         nbytes, dentries_len);
 365                 return -1;
 366         }
 367
 368         return 0;
 369 }
 370
 371 static void usage(void)
 372 {
 373         fputs("Usage: mkfs.exfat\n"
 374                 "\t-L | --volume-label=label                              Set volume label\n"
 375                 "\t-c | --cluster-size=size(or suffixed by 'K' or 'M')    Specify cluster size\n"
 376                 "\t-b | --boundary-align=size(or suffixed by 'K' or 'M')  Specify boundary alignment\n"
 377                 "\t     --pack-bitmap                                     Move bitmap into FAT segment\n"
 378                 "\t-f | --full-format                                     Full format\n"
 379                 "\t-V | --version                                         Show version\n"
 380                 "\t-v | --verbose                                         Print debug\n"
 381                 "\t-h | --help                                            Show help\n",
 382                 stderr);
 383
 384         exit(EXIT_FAILURE);
 385 }
 386
 387 #define PACK_BITMAP (CHAR_MAX + 1)
 388
 389 static const struct option opts[] = {
 390         {"volume-label",        required_argument,      NULL,   'L' },
 391         {"cluster-size",        required_argument,      NULL,   'c' },
 392         {"boundary-align",      required_argument,      NULL,   'b' },
 393         {"pack-bitmap",         no_argument,            NULL,   PACK_BITMAP },
 394         {"full-format",         no_argument,            NULL,   'f' },
 395         {"version",             no_argument,            NULL,   'V' },
 396         {"verbose",             no_argument,            NULL,   'v' },
 397         {"help",                no_argument,            NULL,   'h' },
 398         {"?",                   no_argument,            NULL,   '?' },
 399         {NULL,                  0,                      NULL,    0  }
 400 };
 401
 402 /*
 403  * Moves the bitmap to just before the alignment boundary if there is space
 404  * between the boundary and the end of the FAT. This may allow the FAT and the
 405  * bitmap to share the same allocation unit on flash media, thereby improving
 406  * performance and endurance.
 407  */
 408 static int exfat_pack_bitmap(const struct exfat_user_input *ui)
 409 {
 410         unsigned int fat_byte_end = finfo.fat_byte_off + finfo.fat_byte_len,
 411                 bitmap_byte_len = finfo.bitmap_byte_len,
 412                 bitmap_clu_len = round_up(bitmap_byte_len, ui->cluster_size),
 413                 bitmap_clu_cnt, total_clu_cnt, new_bitmap_clu_len;
 414
 415         for (;;) {
 416                 bitmap_clu_cnt = bitmap_clu_len / ui->cluster_size;
 417                 if (finfo.clu_byte_off - bitmap_clu_len < fat_byte_end ||
 418                                 finfo.total_clu_cnt > EXFAT_MAX_NUM_CLUSTER -
 419                                         bitmap_clu_cnt)
 420                         return -1;
 421                 total_clu_cnt = finfo.total_clu_cnt + bitmap_clu_cnt;
 422                 bitmap_byte_len = round_up(total_clu_cnt, 8) / 8;
 423                 new_bitmap_clu_len = round_up(bitmap_byte_len, ui->cluster_size);
 424                 if (new_bitmap_clu_len == bitmap_clu_len) {
 425                         finfo.clu_byte_off -= bitmap_clu_len;
 426                         finfo.total_clu_cnt = total_clu_cnt;
 427                         finfo.bitmap_byte_off -= bitmap_clu_len;
 428                         finfo.bitmap_byte_len = bitmap_byte_len;
 429                         return 0;
 430                 }
 431                 bitmap_clu_len = new_bitmap_clu_len;
 432         }
 433 }
 434
 435 static int exfat_build_mkfs_info(struct exfat_blk_dev *bd,
 436                 struct exfat_user_input *ui)
 437 {
 438         unsigned long long total_clu_cnt;
 439         int clu_len;
 440
 441         if (ui->cluster_size < bd->sector_size) {
 442                 exfat_err("cluster size (%u bytes) is smaller than sector size (%u bytes)\n",
 443                           ui->cluster_size, bd->sector_size);
 444                 return -1;
 445         }
 446         if (ui->boundary_align < bd->sector_size) {
 447                 exfat_err("boundary alignment is too small (min %d)\n",
 448                                 bd->sector_size);
 449                 return -1;
 450         }
 451         finfo.fat_byte_off = round_up(bd->offset + 24 * bd->sector_size,
 452                         ui->boundary_align) - bd->offset;
 453         /* Prevent integer overflow when computing the FAT length */
 454         if (bd->num_clusters > UINT32_MAX / 4) {
 455                 exfat_err("cluster size (%u bytes) is too small\n", ui->cluster_size);
 456                 return -1;
 457         }
 458         finfo.fat_byte_len = round_up((bd->num_clusters * 4), ui->cluster_size);
 459         finfo.clu_byte_off = round_up(bd->offset + finfo.fat_byte_off +
 460                 finfo.fat_byte_len, ui->boundary_align) - bd->offset;
 461         if (bd->size <= finfo.clu_byte_off) {
 462                 exfat_err("boundary alignment is too big\n");
 463                 return -1;
 464         }
 465         total_clu_cnt = (bd->size - finfo.clu_byte_off) / ui->cluster_size;
 466         if (total_clu_cnt > EXFAT_MAX_NUM_CLUSTER) {
 467                 exfat_err("cluster size is too small\n");
 468                 return -1;
 469         }
 470         finfo.total_clu_cnt = (unsigned int) total_clu_cnt;
 471
 472         finfo.bitmap_byte_off = finfo.clu_byte_off;
 473         finfo.bitmap_byte_len = round_up(finfo.total_clu_cnt, 8) / 8;
 474         if (ui->pack_bitmap)
 475                 exfat_pack_bitmap(ui);
 476         clu_len = round_up(finfo.bitmap_byte_len, ui->cluster_size);
 477
 478         finfo.ut_start_clu = EXFAT_FIRST_CLUSTER + clu_len / ui->cluster_size;
 479         finfo.ut_byte_off = finfo.bitmap_byte_off + clu_len;
 480         finfo.ut_byte_len = EXFAT_UPCASE_TABLE_SIZE;
 481         clu_len = round_up(finfo.ut_byte_len, ui->cluster_size);
 482
 483         finfo.root_start_clu = finfo.ut_start_clu + clu_len / ui->cluster_size;
 484         finfo.root_byte_off = finfo.ut_byte_off + clu_len;
 485         finfo.root_byte_len = sizeof(struct exfat_dentry) * 3;
 486         finfo.volume_serial = get_new_serial();
 487
 488         return 0;
 489 }
 490
 491 static int exfat_zero_out_disk(struct exfat_blk_dev *bd,
 492                 struct exfat_user_input *ui)
 493 {
 494         int nbytes;
 495         unsigned long long total_written = 0;
 496         char *buf;
 497         unsigned int chunk_size = ui->cluster_size;
 498         unsigned long long size;
 499
 500         if (ui->quick)
 501                 size = finfo.root_byte_off + chunk_size;
 502         else
 503                 size = bd->size;
 504
 505         buf = malloc(chunk_size);
 506         if (!buf)
 507                 return -1;
 508
 509         memset(buf, 0, chunk_size);
 510         lseek(bd->dev_fd, 0, SEEK_SET);
 511         do {
 512
 513                 nbytes = write(bd->dev_fd, buf, chunk_size);
 514                 if (nbytes <= 0) {
 515                         if (nbytes < 0)
 516                                 exfat_err("write failed(errno : %d)\n", errno);
 517                         break;
 518                 }
 519                 total_written += nbytes;
 520         } while (total_written < size);
 521
 522         free(buf);
 523         exfat_debug("zero out written size : %llu, disk size : %llu\n",
 524                 total_written, bd->size);
 525         return 0;
 526 }
 527
 528 static int make_exfat(struct exfat_blk_dev *bd, struct exfat_user_input *ui)
 529 {
 530         int ret;
 531
 532         exfat_info("Creating exFAT filesystem(%s, cluster size=%u)\n\n",
 533                 ui->dev_name, ui->cluster_size);
 534
 535         exfat_info("Writing volume boot record: ");
 536         ret = exfat_create_volume_boot_record(bd, ui, 0);
 537         exfat_info("%s\n", ret ? "failed" : "done");
 538         if (ret)
 539                 return ret;
 540
 541         exfat_info("Writing backup volume boot record: ");
 542         /* backup sector */
 543         ret = exfat_create_volume_boot_record(bd, ui, 1);
 544         exfat_info("%s\n", ret ? "failed" : "done");
 545         if (ret)
 546                 return ret;
 547
 548         exfat_info("Fat table creation: ");
 549         ret = exfat_create_fat_table(bd, ui);
 550         exfat_info("%s\n", ret ? "failed" : "done");
 551         if (ret)
 552                 return ret;
 553
 554         exfat_info("Allocation bitmap creation: ");
 555         ret = exfat_create_bitmap(bd);
 556         exfat_info("%s\n", ret ? "failed" : "done");
 557         if (ret)
 558                 return ret;
 559
 560         exfat_info("Upcase table creation: ");
 561         ret = exfat_create_upcase_table(bd);
 562         exfat_info("%s\n", ret ? "failed" : "done");
 563         if (ret)
 564                 return ret;
 565
 566         exfat_info("Writing root directory entry: ");
 567         ret = exfat_create_root_dir(bd, ui);
 568         exfat_info("%s\n", ret ? "failed" : "done");
 569         if (ret)
 570                 return ret;
 571
 572         return 0;
 573 }
 574
 575 static long long parse_size(const char *size)
 576 {
 577         char *data_unit;
 578         unsigned long long byte_size = strtoull(size, &data_unit, 0);
 579
 580         switch (*data_unit) {
 581         case 'M':
 582         case 'm':
 583                 byte_size <<= 20;
 584                 break;
 585         case 'K':
 586         case 'k':
 587                 byte_size <<= 10;
 588                 break;
 589         case '\0':
 590                 break;
 591         default:
 592                 exfat_err("Wrong unit input('%c') for size\n",
 593                                 *data_unit);
 594                 return -EINVAL;
 595         }
 596
 597         return byte_size;
 598 }
 599
 600 int main(int argc, char *argv[])
 601 {
 602         int c;
 603         int ret = EXIT_FAILURE;
 604         struct exfat_blk_dev bd;
 605         struct exfat_user_input ui;
 606         bool version_only = false;
 607
 608         init_user_input(&ui);
 609
 610         if (!setlocale(LC_CTYPE, ""))
 611                 exfat_err("failed to init locale/codeset\n");
 612
 613         opterr = 0;
 614         while ((c = getopt_long(argc, argv, "n:L:c:b:fVvh", opts, NULL)) != EOF)
 615                 switch (c) {
 616                 /*
 617                  * Make 'n' option fallthrough to 'L' option for for backward
 618                  * compatibility with old utils.
 619                  */
 620                 case 'n':
 621                 case 'L':
 622                 {
 623                         ret = exfat_utf16_enc(optarg,
 624                                 ui.volume_label, sizeof(ui.volume_label));
 625                         if (ret < 0)
 626                                 goto out;
 627
 628                         ui.volume_label_len = ret;
 629                         break;
 630                 }
 631                 case 'c':
 632                         ret = parse_size(optarg);
 633                         if (ret < 0)
 634                                 goto out;
 635                         else if (ret & (ret - 1)) {
 636                                 exfat_err("cluster size(%d) is not a power of 2)\n",
 637                                         ret);
 638                                 goto out;
 639                         } else if (ret > EXFAT_MAX_CLUSTER_SIZE) {
 640                                 exfat_err("cluster size(%d) exceeds max cluster size(%d)\n",
 641                                         ui.cluster_size, EXFAT_MAX_CLUSTER_SIZE);
 642                                 goto out;
 643                         }
 644                         ui.cluster_size = ret;
 645                         break;
 646                 case 'b':
 647                         ret = parse_size(optarg);
 648                         if (ret < 0)
 649                                 goto out;
 650                         else if (ret & (ret - 1)) {
 651                                 exfat_err("boundary align(%d) is not a power of 2)\n",
 652                                         ret);
 653                                 goto out;
 654                         }
 655                         ui.boundary_align = ret;
 656                         break;
 657                 case PACK_BITMAP:
 658                         ui.pack_bitmap = true;
 659                         break;
 660                 case 'f':
 661                         ui.quick = false;
 662                         break;
 663                 case 'V':
 664                         version_only = true;
 665                         break;
 666                 case 'v':
 667                         print_level = EXFAT_DEBUG;
 668                         break;
 669                 case '?':
 670                 case 'h':
 671                 default:
 672                         usage();
 673         }
 674
 675         show_version();
 676         if (version_only)
 677                 exit(EXIT_FAILURE);
 678
 679         if (argc - optind != 1) {
 680                 usage();
 681         }
 682
 683         memset(ui.dev_name, 0, sizeof(ui.dev_name));
 684         snprintf(ui.dev_name, sizeof(ui.dev_name), "%s", argv[optind]);
 685
 686         ret = exfat_get_blk_dev_info(&ui, &bd);
 687         if (ret < 0)
 688                 goto out;
 689
 690         ret = exfat_build_mkfs_info(&bd, &ui);
 691         if (ret)
 692                 goto close;
 693
 694         ret = exfat_zero_out_disk(&bd, &ui);
 695         if (ret)
 696                 goto close;
 697
 698         ret = make_exfat(&bd, &ui);
 699         if (ret)
 700                 goto close;
 701
 702         exfat_info("Synchronizing...\n");
 703         ret = fsync(bd.dev_fd);
 704 close:
 705         close(bd.dev_fd);
 706 out:
 707         if (!ret)
 708                 exfat_info("\nexFAT format complete!\n");
 709         else
 710                 exfat_info("\nexFAT format fail!\n");
 711         return ret;
 712 }