• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 */
get_new_serial(void)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 
exfat_setup_boot_sector(struct pbr * ppbr,struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
exfat_write_boot_sector(struct exfat_blk_dev * bd,struct exfat_user_input * ui,unsigned int * checksum,bool is_backup)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 
exfat_write_extended_boot_sectors(struct exfat_blk_dev * bd,unsigned int * checksum,bool is_backup)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 
exfat_write_oem_sector(struct exfat_blk_dev * bd,unsigned int * checksum,bool is_backup)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 
exfat_create_volume_boot_record(struct exfat_blk_dev * bd,struct exfat_user_input * ui,bool is_backup)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 
write_fat_entry(int fd,__le32 clu,unsigned long long offset)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 
write_fat_entries(struct exfat_user_input * ui,int fd,unsigned int clu,unsigned int length)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 
exfat_create_fat_table(struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
exfat_create_bitmap(struct exfat_blk_dev * bd)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 
exfat_create_root_dir(struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
usage(void)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  */
exfat_pack_bitmap(const struct exfat_user_input * ui)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 
exfat_build_mkfs_info(struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
exfat_zero_out_disk(struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
make_exfat(struct exfat_blk_dev * bd,struct exfat_user_input * ui)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 
parse_size(const char * size)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 
main(int argc,char * argv[])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 }
713