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 <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 #include "exfat_fs.h"
17
usage(void)18 static void usage(void)
19 {
20 fprintf(stderr, "Usage: tune.exfat\n");
21 fprintf(stderr, "\t-l | --print-label Print volume label\n");
22 fprintf(stderr, "\t-L | --set-label=label Set volume label\n");
23 fprintf(stderr, "\t-i | --print-serial Print volume serial\n");
24 fprintf(stderr, "\t-I | --set-serial=value Set volume serial\n");
25 fprintf(stderr, "\t-V | --version Show version\n");
26 fprintf(stderr, "\t-v | --verbose Print debug\n");
27 fprintf(stderr, "\t-h | --help Show help\n");
28
29 exit(EXIT_FAILURE);
30 }
31
32 static struct option opts[] = {
33 {"print-label", no_argument, NULL, 'l' },
34 {"set-label", required_argument, NULL, 'L' },
35 {"print-serial", no_argument, NULL, 'i' },
36 {"set-serial", required_argument, NULL, 'I' },
37 {"version", no_argument, NULL, 'V' },
38 {"verbose", no_argument, NULL, 'v' },
39 {"help", no_argument, NULL, 'h' },
40 {"?", no_argument, NULL, '?' },
41 {NULL, 0, NULL, 0 }
42 };
43
main(int argc,char * argv[])44 int main(int argc, char *argv[])
45 {
46 int c;
47 int ret = EXIT_FAILURE;
48 struct exfat_blk_dev bd;
49 struct exfat_user_input ui;
50 bool version_only = false;
51 int flags = 0;
52 char label_input[VOLUME_LABEL_BUFFER_SIZE];
53 struct exfat *exfat = NULL;
54 struct pbr *bs;
55
56 init_user_input(&ui);
57
58 if (!setlocale(LC_CTYPE, ""))
59 exfat_err("failed to init locale/codeset\n");
60
61 opterr = 0;
62 while ((c = getopt_long(argc, argv, "I:iL:lVvh", opts, NULL)) != EOF)
63 switch (c) {
64 case 'l':
65 flags = EXFAT_GET_VOLUME_LABEL;
66 break;
67 case 'L':
68 snprintf(label_input, sizeof(label_input), "%s",
69 optarg);
70 flags = EXFAT_SET_VOLUME_LABEL;
71 break;
72 case 'i':
73 flags = EXFAT_GET_VOLUME_SERIAL;
74 break;
75 case 'I':
76 ui.volume_serial = strtoul(optarg, NULL, 0);
77 flags = EXFAT_SET_VOLUME_SERIAL;
78 break;
79 case 'V':
80 version_only = true;
81 break;
82 case 'v':
83 print_level = EXFAT_DEBUG;
84 break;
85 case '?':
86 case 'h':
87 default:
88 usage();
89 }
90
91 show_version();
92 if (version_only)
93 exit(EXIT_FAILURE);
94
95 if (argc < 3)
96 usage();
97
98 memset(ui.dev_name, 0, sizeof(ui.dev_name));
99 snprintf(ui.dev_name, sizeof(ui.dev_name), "%s", argv[argc - 1]);
100
101 ret = exfat_get_blk_dev_info(&ui, &bd);
102 if (ret < 0)
103 goto out;
104
105 /* Mode to change or display volume serial */
106 if (flags == EXFAT_GET_VOLUME_SERIAL) {
107 ret = exfat_show_volume_serial(bd.dev_fd);
108 goto close_fd_out;
109 } else if (flags == EXFAT_SET_VOLUME_SERIAL) {
110 ret = exfat_set_volume_serial(&bd, &ui);
111 goto close_fd_out;
112 }
113
114 ret = read_boot_sect(&bd, &bs);
115 if (ret)
116 goto close_fd_out;
117
118 exfat = exfat_alloc_exfat(&bd, bs);
119 if (!exfat) {
120 free(bs);
121 ret = -ENOMEM;
122 goto close_fd_out;
123 }
124
125 exfat->root = exfat_alloc_inode(ATTR_SUBDIR);
126 if (!exfat->root) {
127 ret = -ENOMEM;
128 goto close_fd_out;
129 }
130
131 exfat->root->first_clus = le32_to_cpu(exfat->bs->bsx.root_cluster);
132 if (exfat_root_clus_count(exfat)) {
133 exfat_err("failed to follow the cluster chain of root\n");
134 exfat_free_inode(exfat->root);
135 ret = -EINVAL;
136 goto close_fd_out;
137 }
138
139 if (flags == EXFAT_GET_VOLUME_LABEL)
140 ret = exfat_read_volume_label(exfat);
141 else if (flags == EXFAT_SET_VOLUME_LABEL)
142 ret = exfat_set_volume_label(exfat, label_input);
143 close_fd_out:
144 close(bd.dev_fd);
145 if (exfat)
146 exfat_free_exfat(exfat);
147 out:
148 return ret;
149 }
150