1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright (C) 2020 Namjae Jeon <linkinjeon@kernel.org>
4 */
5
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <unistd.h>
10 #include <getopt.h>
11 #include <errno.h>
12 #include <locale.h>
13
14 #include "exfat_ondisk.h"
15 #include "libexfat.h"
16
usage(void)17 static void usage(void)
18 {
19 fprintf(stderr, "Usage: exfatlabel\n");
20 fprintf(stderr, "\t-i | --volume-serial Switch to volume serial mode\n");
21 fprintf(stderr, "\t-V | --version Show version\n");
22 fprintf(stderr, "\t-h | --help Show help\n");
23
24 exit(EXIT_FAILURE);
25 }
26
27 static struct option opts[] = {
28 {"volume-serial", no_argument, NULL, 'i' },
29 {"version", no_argument, NULL, 'V' },
30 {"help", no_argument, NULL, 'h' },
31 {"?", no_argument, NULL, '?' },
32 {NULL, 0, NULL, 0 }
33 };
34
main(int argc,char * argv[])35 int main(int argc, char *argv[])
36 {
37 int c;
38 int ret = EXIT_FAILURE;
39 struct exfat_blk_dev bd;
40 struct exfat_user_input ui;
41 bool version_only = false;
42 off_t root_clu_off;
43 int serial_mode = 0;
44 int flags = 0;
45
46 init_user_input(&ui);
47
48 if (!setlocale(LC_CTYPE, ""))
49 exfat_err("failed to init locale/codeset\n");
50
51 if (argc == 2)
52 flags = EXFAT_GET_VOLUME_LABEL;
53 else if (argc == 3)
54 flags = EXFAT_SET_VOLUME_LABEL;
55
56 opterr = 0;
57 while ((c = getopt_long(argc, argv, "iVh", opts, NULL)) != EOF)
58 switch (c) {
59 case 'i':
60 serial_mode = true;
61 if (argc == 3)
62 flags = EXFAT_GET_VOLUME_SERIAL;
63 else if (argc == 4)
64 flags = EXFAT_SET_VOLUME_SERIAL;
65
66 break;
67 case 'V':
68 version_only = true;
69 break;
70 case '?':
71 case 'h':
72 default:
73 usage();
74 }
75
76 show_version();
77 if (version_only)
78 exit(EXIT_FAILURE);
79
80 if (argc < 2)
81 usage();
82
83 memset(ui.dev_name, 0, sizeof(ui.dev_name));
84 snprintf(ui.dev_name, sizeof(ui.dev_name), "%s", argv[serial_mode + 1]);
85
86 ret = exfat_get_blk_dev_info(&ui, &bd);
87 if (ret < 0)
88 goto out;
89
90 if (serial_mode) {
91 /* Mode to change or display volume serial */
92 if (flags == EXFAT_GET_VOLUME_SERIAL) {
93 ret = exfat_show_volume_serial(bd.dev_fd);
94 } else if (flags == EXFAT_SET_VOLUME_SERIAL) {
95 ui.volume_serial = strtoul(argv[3], NULL, 0);
96 ret = exfat_set_volume_serial(&bd, &ui);
97 }
98 } else {
99 /* Mode to change or display volume label */
100 root_clu_off = exfat_get_root_entry_offset(&bd);
101 if (root_clu_off < 0)
102 goto close_fd_out;
103
104 if (flags == EXFAT_GET_VOLUME_LABEL)
105 ret = exfat_show_volume_label(&bd, root_clu_off);
106 else if (flags == EXFAT_SET_VOLUME_LABEL)
107 ret = exfat_set_volume_label(&bd, argv[2], root_clu_off);
108 }
109
110 close_fd_out:
111 close(bd.dev_fd);
112 out:
113 return ret;
114 }
115