]> git.sven.stormbind.net Git - sven/exfatprogs.git/blobdiff - tune/tune.c
Add CVE ID to debian changelog
[sven/exfatprogs.git] / tune / tune.c
index d160a372c31a6360a733dfc40824c9bee078171c..f883556c794ed2bcfd82b9837a6c2f7e62298443 100644 (file)
 
 #include "exfat_ondisk.h"
 #include "libexfat.h"
+#include "exfat_fs.h"
 
 static void usage(void)
 {
        fprintf(stderr, "Usage: tune.exfat\n");
        fprintf(stderr, "\t-l | --print-label                    Print volume label\n");
-       fprintf(stderr, "\t-L | --volume-label=label             Set volume label\n");
+       fprintf(stderr, "\t-L | --set-label=label                Set volume label\n");
+       fprintf(stderr, "\t-u | --print-guid                     Print volume GUID\n");
+       fprintf(stderr, "\t-U | --set-guid=guid                  Set volume GUID\n");
+       fprintf(stderr, "\t-i | --print-serial                   Print volume serial\n");
+       fprintf(stderr, "\t-I | --set-serial=value               Set volume serial\n");
        fprintf(stderr, "\t-V | --version                        Show version\n");
        fprintf(stderr, "\t-v | --verbose                        Print debug\n");
        fprintf(stderr, "\t-h | --help                           Show help\n");
@@ -29,6 +34,10 @@ static void usage(void)
 static struct option opts[] = {
        {"print-label",         no_argument,            NULL,   'l' },
        {"set-label",           required_argument,      NULL,   'L' },
+       {"print-guid",          no_argument,            NULL,   'u' },
+       {"set-guid",            required_argument,      NULL,   'U' },
+       {"print-serial",        no_argument,            NULL,   'i' },
+       {"set-serial",          required_argument,      NULL,   'I' },
        {"version",             no_argument,            NULL,   'V' },
        {"verbose",             no_argument,            NULL,   'v' },
        {"help",                no_argument,            NULL,   'h' },
@@ -36,101 +45,6 @@ static struct option opts[] = {
        {NULL,                  0,                      NULL,    0  }
 };
 
-static off_t exfat_get_root_entry_offset(struct exfat_blk_dev *bd)
-{
-       struct pbr *bs;
-       int nbytes;
-       unsigned int cluster_size;
-       off_t root_clu_off;
-
-       bs = (struct pbr *)malloc(sizeof(struct pbr));
-       if (!bs) {
-               exfat_err("failed to allocate memory\n");
-               return -ENOMEM;
-       }
-
-       nbytes = exfat_read(bd->dev_fd, bs, sizeof(struct pbr), 0);
-       if (nbytes != sizeof(struct pbr)) {
-               exfat_err("boot sector read failed: %d\n", errno);
-               return -1;
-       }
-
-       cluster_size = (1 << bs->bsx.sect_per_clus_bits) * bd->sector_size;
-       root_clu_off = le32_to_cpu(bs->bsx.clu_offset) * bd->sector_size +
-               le32_to_cpu(bs->bsx.root_cluster - EXFAT_REVERVED_CLUSTERS)
-               * cluster_size;
-       free(bs);
-
-       return root_clu_off;
-}
-
-static int exfat_get_volume_label(struct exfat_blk_dev *bd, off_t root_clu_off)
-{
-       struct exfat_dentry *vol_entry;
-       char volume_label[VOLUME_LABEL_BUFFER_SIZE];
-       __le16 disk_label[VOLUME_LABEL_MAX_LEN];
-       int nbytes;
-
-       vol_entry = malloc(sizeof(struct exfat_dentry));
-       if (!vol_entry) {
-               exfat_err("failed to allocate memory\n");
-               return -ENOMEM;
-       }
-
-       nbytes = exfat_read(bd->dev_fd, vol_entry,
-               sizeof(struct exfat_dentry), root_clu_off);
-       if (nbytes != sizeof(struct exfat_dentry)) {
-               exfat_err("volume entry read failed: %d\n", errno);
-               return -1;
-       }
-
-       memcpy(disk_label, vol_entry->vol_label, sizeof(disk_label));
-       memset(volume_label, 0, sizeof(volume_label));
-       if (exfat_utf16_dec(disk_label, vol_entry->vol_char_cnt*2,
-               volume_label, sizeof(volume_label)) < 0) {
-               exfat_err("failed to decode volume label\n");
-               return -1;
-       }
-
-       exfat_info("label: %s\n", volume_label);
-       return 0;
-}
-
-static int exfat_set_volume_label(struct exfat_blk_dev *bd,
-               char *label_input, off_t root_clu_off)
-{
-       struct exfat_dentry vol;
-       int nbytes;
-       __u16 volume_label[VOLUME_LABEL_MAX_LEN];
-       int volume_label_len;
-
-       volume_label_len = exfat_utf16_enc(label_input,
-                       volume_label, sizeof(volume_label));
-       if (volume_label_len < 0) {
-               exfat_err("failed to encode volume label\n");
-               return -1;
-       }
-
-       vol.type = EXFAT_VOLUME;
-       memset(vol.vol_label, 0, sizeof(vol.vol_label));
-       memcpy(vol.vol_label, volume_label, volume_label_len);
-       vol.vol_char_cnt = volume_label_len/2;
-
-       nbytes = exfat_write(bd->dev_fd, &vol, sizeof(struct exfat_dentry),
-                       root_clu_off);
-       if (nbytes != sizeof(struct exfat_dentry)) {
-               exfat_err("volume entry write failed: %d\n", errno);
-               return -1;
-       }
-       fsync(bd->dev_fd);
-
-       exfat_info("new label: %s\n", label_input);
-       return 0;
-}
-
-#define EXFAT_GET_LABEL        0x1
-#define EXFAT_SET_LABEL        0x2
-
 int main(int argc, char *argv[])
 {
        int c;
@@ -140,7 +54,8 @@ int main(int argc, char *argv[])
        bool version_only = false;
        int flags = 0;
        char label_input[VOLUME_LABEL_BUFFER_SIZE];
-       off_t root_clu_off;
+       struct exfat *exfat = NULL;
+       struct pbr *bs;
 
        init_user_input(&ui);
 
@@ -148,15 +63,30 @@ int main(int argc, char *argv[])
                exfat_err("failed to init locale/codeset\n");
 
        opterr = 0;
-       while ((c = getopt_long(argc, argv, "L:lVvh", opts, NULL)) != EOF)
+       while ((c = getopt_long(argc, argv, "I:iL:lU:uVvh", opts, NULL)) != EOF)
                switch (c) {
                case 'l':
-                       flags = EXFAT_GET_LABEL;
+                       flags = EXFAT_GET_VOLUME_LABEL;
                        break;
                case 'L':
                        snprintf(label_input, sizeof(label_input), "%s",
                                        optarg);
-                       flags = EXFAT_SET_LABEL;
+                       flags = EXFAT_SET_VOLUME_LABEL;
+                       break;
+               case 'u':
+                       flags = EXFAT_GET_VOLUME_GUID;
+                       break;
+               case 'U':
+                       if (*optarg != '\0' && *optarg != '\r')
+                               ui.guid = optarg;
+                       flags = EXFAT_SET_VOLUME_GUID;
+                       break;
+               case 'i':
+                       flags = EXFAT_GET_VOLUME_SERIAL;
+                       break;
+               case 'I':
+                       ui.volume_serial = strtoul(optarg, NULL, 0);
+                       flags = EXFAT_SET_VOLUME_SERIAL;
                        break;
                case 'V':
                        version_only = true;
@@ -184,15 +114,53 @@ int main(int argc, char *argv[])
        if (ret < 0)
                goto out;
 
-       root_clu_off = exfat_get_root_entry_offset(&bd);
-       if (root_clu_off < 0)
-               goto out;
+       /* Mode to change or display volume serial */
+       if (flags == EXFAT_GET_VOLUME_SERIAL) {
+               ret = exfat_show_volume_serial(bd.dev_fd);
+               goto close_fd_out;
+       } else if (flags == EXFAT_SET_VOLUME_SERIAL) {
+               ret = exfat_set_volume_serial(&bd, &ui);
+               goto close_fd_out;
+       }
+
+       ret = read_boot_sect(&bd, &bs);
+       if (ret)
+               goto close_fd_out;
 
-       if (flags == EXFAT_GET_LABEL)
-               ret = exfat_get_volume_label(&bd, root_clu_off);
-       else if (flags == EXFAT_SET_LABEL)
-               ret = exfat_set_volume_label(&bd, label_input, root_clu_off);
+       exfat = exfat_alloc_exfat(&bd, bs);
+       if (!exfat) {
+               free(bs);
+               ret = -ENOMEM;
+               goto close_fd_out;
+       }
+
+       exfat->root = exfat_alloc_inode(ATTR_SUBDIR);
+       if (!exfat->root) {
+               ret = -ENOMEM;
+               goto close_fd_out;
+       }
+
+       exfat->root->first_clus = le32_to_cpu(exfat->bs->bsx.root_cluster);
+       if (exfat_root_clus_count(exfat)) {
+               exfat_err("failed to follow the cluster chain of root\n");
+               exfat_free_inode(exfat->root);
+               ret = -EINVAL;
+               goto close_fd_out;
+       }
 
+       if (flags == EXFAT_GET_VOLUME_LABEL)
+               ret = exfat_read_volume_label(exfat);
+       else if (flags == EXFAT_SET_VOLUME_LABEL)
+               ret = exfat_set_volume_label(exfat, label_input);
+       else if (flags == EXFAT_GET_VOLUME_GUID)
+               ret = exfat_read_volume_guid(exfat);
+       else if (flags == EXFAT_SET_VOLUME_GUID)
+               ret = exfat_set_volume_guid(exfat, ui.guid);
+
+close_fd_out:
+       close(bd.dev_fd);
+       if (exfat)
+               exfat_free_exfat(exfat);
 out:
        return ret;
 }