libexfat/mount.c

   1 /*
   2         mount.c (22.10.09)
   3         exFAT file system implementation library.
   4
   5         Copyright (C) 2009, 2010  Andrew Nayenko
   6
   7         This program is free software: you can redistribute it and/or modify
   8         it under the terms of the GNU General Public License as published by
   9         the Free Software Foundation, either version 3 of the License, or
  10         (at your option) any later version.
  11
  12         This program is distributed in the hope that it will be useful,
  13         but WITHOUT ANY WARRANTY; without even the implied warranty of
  14         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15         GNU General Public License for more details.
  16
  17         You should have received a copy of the GNU General Public License
  18         along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "exfat.h"
  22 #include <string.h>
  23 #include <stdlib.h>
  24 #include <errno.h>
  25 #include <fcntl.h>
  26 #include <unistd.h>
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #define _XOPEN_SOURCE /* for tzset() in Linux */
  30 #include <time.h>
  31
  32 static uint64_t rootdir_size(const struct exfat* ef)
  33 {
  34         uint64_t clusters = 0;
  35         cluster_t rootdir_cluster = le32_to_cpu(ef->sb->rootdir_cluster);
  36
  37         while (!CLUSTER_INVALID(rootdir_cluster))
  38         {
  39                 clusters++;
  40                 /* root directory cannot be contiguous because there is no flag
  41                    to indicate this */
  42                 rootdir_cluster = exfat_next_cluster(ef, ef->root, rootdir_cluster);
  43         }
  44         return clusters * CLUSTER_SIZE(*ef->sb);
  45 }
  46
  47 static const char* get_option(const char* options, const char* option_name)
  48 {
  49         const char* p;
  50         size_t length = strlen(option_name);
  51
  52         for (p = strstr(options, option_name); p; p = strstr(p + 1, option_name))
  53                 if ((p == options || p[-1] == ',') && p[length] == '=')
  54                         return p + length + 1;
  55         return NULL;
  56 }
  57
  58 static int get_int_option(const char* options, const char* option_name,
  59                 int base, int default_value)
  60 {
  61         const char* p = get_option(options, option_name);
  62
  63         if (p == NULL)
  64                 return default_value;
  65         return strtol(p, NULL, base);
  66 }
  67
  68 static int match_option(const char* options, const char* option_name)
  69 {
  70         const char* p;
  71         size_t length = strlen(option_name);
  72
  73         for (p = strstr(options, option_name); p; p = strstr(p + 1, option_name))
  74                 if ((p == options || p[-1] == ',') &&
  75                                 (p[length] == ',' || p[length] == '\0'))
  76                         return 1;
  77         return 0;
  78 }
  79
  80 static void parse_options(struct exfat* ef, const char* options)
  81 {
  82         int sys_umask = umask(0);
  83         int opt_umask;
  84
  85         umask(sys_umask); /* restore umask */
  86         opt_umask = get_int_option(options, "umask", 8, sys_umask);
  87         ef->dmask = get_int_option(options, "dmask", 8, opt_umask) & 0777;
  88         ef->fmask = get_int_option(options, "fmask", 8, opt_umask) & 0777;
  89
  90         ef->uid = get_int_option(options, "uid", 10, geteuid());
  91         ef->gid = get_int_option(options, "gid", 10, getegid());
  92
  93         ef->ro = match_option(options, "ro");
  94         ef->noatime = match_option(options, "noatime");
  95 }
  96
  97 static int verify_vbr_checksum(void* block, off_t block_size, int fd)
  98 {
  99         uint32_t vbr_checksum;
 100         int i;
 101
 102         exfat_read_raw(block, block_size, 0, fd);
 103         vbr_checksum = exfat_vbr_start_checksum(block, block_size);
 104         for (i = 1; i < 11; i++)
 105         {
 106                 exfat_read_raw(block, block_size, i * block_size, fd);
 107                 vbr_checksum = exfat_vbr_add_checksum(block, block_size, vbr_checksum);
 108         }
 109         exfat_read_raw(block, block_size, i * block_size, fd);
 110         for (i = 0; i < block_size / sizeof(vbr_checksum); i++)
 111                 if (((const uint32_t*) block)[i] != vbr_checksum)
 112                 {
 113                         exfat_error("invalid VBR checksum 0x%x (expected 0x%x)",
 114                                         ((const uint32_t*) block)[i], vbr_checksum);
 115                         return 1;
 116                 }
 117         return 0;
 118 }
 119
 120 int exfat_mount(struct exfat* ef, const char* spec, const char* options)
 121 {
 122         int rc;
 123
 124         tzset();
 125         memset(ef, 0, sizeof(struct exfat));
 126
 127         ef->sb = malloc(sizeof(struct exfat_super_block));
 128         if (ef->sb == NULL)
 129         {
 130                 exfat_error("memory allocation failed");
 131                 return -ENOMEM;
 132         }
 133         memset(ef->sb, 0, sizeof(struct exfat_super_block));
 134
 135         parse_options(ef, options);
 136
 137         ef->fd = open(spec, ef->ro ? O_RDONLY : O_RDWR);
 138         if (ef->fd < 0)
 139         {
 140                 free(ef->sb);
 141                 exfat_error("failed to open `%s'", spec);
 142                 return -EIO;
 143         }
 144
 145         exfat_read_raw(ef->sb, sizeof(struct exfat_super_block), 0, ef->fd);
 146         if (memcmp(ef->sb->oem_name, "EXFAT   ", 8) != 0)
 147         {
 148                 close(ef->fd);
 149                 free(ef->sb);
 150                 exfat_error("exFAT file system is not found");
 151                 return -EIO;
 152         }
 153         if (ef->sb->version.major != 1 || ef->sb->version.minor != 0)
 154         {
 155                 close(ef->fd);
 156                 exfat_error("unsupported exFAT version: %hhu.%hhu",
 157                                 ef->sb->version.major, ef->sb->version.minor);
 158                 free(ef->sb);
 159                 return -EIO;
 160         }
 161         if (ef->sb->fat_count != 1)
 162         {
 163                 close(ef->fd);
 164                 free(ef->sb);
 165                 exfat_error("unsupported FAT count: %hhu", ef->sb->fat_count);
 166                 return -EIO;
 167         }
 168         /* officially exFAT supports cluster size up to 32 MB */
 169         if ((int) ef->sb->block_bits + (int) ef->sb->bpc_bits > 25)
 170         {
 171                 close(ef->fd);
 172                 free(ef->sb);
 173                 exfat_error("too big cluster size: 2^%d",
 174                                 (int) ef->sb->block_bits + (int) ef->sb->bpc_bits);
 175                 return -EIO;
 176         }
 177
 178         ef->zero_block = malloc(BLOCK_SIZE(*ef->sb));
 179         if (ef->zero_block == NULL)
 180         {
 181                 close(ef->fd);
 182                 free(ef->sb);
 183                 exfat_error("failed to allocate zero block");
 184                 return -ENOMEM;
 185         }
 186         /* use zero_block as a temporary buffer for VBR checksum verification */
 187         if (verify_vbr_checksum(ef->zero_block, BLOCK_SIZE(*ef->sb), ef->fd) != 0)
 188         {
 189                 free(ef->zero_block);
 190                 close(ef->fd);
 191                 free(ef->sb);
 192                 return -EIO;
 193         }
 194         memset(ef->zero_block, 0, BLOCK_SIZE(*ef->sb));
 195
 196         ef->root = malloc(sizeof(struct exfat_node));
 197         if (ef->root == NULL)
 198         {
 199                 free(ef->zero_block);
 200                 close(ef->fd);
 201                 free(ef->sb);
 202                 exfat_error("failed to allocate root node");
 203                 return -ENOMEM;
 204         }
 205         memset(ef->root, 0, sizeof(struct exfat_node));
 206         ef->root->flags = EXFAT_ATTRIB_DIR;
 207         ef->root->start_cluster = le32_to_cpu(ef->sb->rootdir_cluster);
 208         ef->root->fptr_cluster = ef->root->start_cluster;
 209         ef->root->name[0] = cpu_to_le16('\0');
 210         ef->root->size = rootdir_size(ef);
 211         /* exFAT does not have time attributes for the root directory */
 212         ef->root->mtime = 0;
 213         ef->root->atime = 0;
 214         /* always keep at least 1 reference to the root node */
 215         exfat_get_node(ef->root);
 216
 217         rc = exfat_cache_directory(ef, ef->root);
 218         if (rc != 0)
 219                 goto error;
 220         if (ef->upcase == NULL)
 221         {
 222                 exfat_error("upcase table is not found");
 223                 goto error;
 224         }
 225         if (ef->cmap.chunk == NULL)
 226         {
 227                 exfat_error("clusters bitmap is not found");
 228                 goto error;
 229         }
 230
 231         return 0;
 232
 233 error:
 234         exfat_put_node(ef, ef->root);
 235         exfat_reset_cache(ef);
 236         free(ef->root);
 237         free(ef->zero_block);
 238         close(ef->fd);
 239         free(ef->sb);
 240         return -EIO;
 241 }
 242
 243 void exfat_unmount(struct exfat* ef)
 244 {
 245         exfat_put_node(ef, ef->root);
 246         exfat_reset_cache(ef);
 247         free(ef->root);
 248         ef->root = NULL;
 249         free(ef->zero_block);
 250         ef->zero_block = NULL;
 251         free(ef->cmap.chunk);
 252         ef->cmap.chunk = NULL;
 253         if (fsync(ef->fd) < 0)
 254                 exfat_error("fsync failed");
 255         if (close(ef->fd) < 0)
 256                 exfat_error("close failed");
 257         ef->fd = 0;
 258         free(ef->sb);
 259         ef->sb = NULL;
 260         free(ef->upcase);
 261         ef->upcase = NULL;
 262         ef->upcase_chars = 0;
 263 }