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* sector, off_t sector_size, int fd)
  98 {
  99         uint32_t vbr_checksum;
 100         int i;
 101
 102         exfat_read_raw(sector, sector_size, 0, fd);
 103         vbr_checksum = exfat_vbr_start_checksum(sector, sector_size);
 104         for (i = 1; i < 11; i++)
 105         {
 106                 exfat_read_raw(sector, sector_size, i * sector_size, fd);
 107                 vbr_checksum = exfat_vbr_add_checksum(sector, sector_size,
 108                                 vbr_checksum);
 109         }
 110         exfat_read_raw(sector, sector_size, i * sector_size, fd);
 111         for (i = 0; i < sector_size / sizeof(vbr_checksum); i++)
 112                 if (((const uint32_t*) sector)[i] != vbr_checksum)
 113                 {
 114                         exfat_error("invalid VBR checksum 0x%x (expected 0x%x)",
 115                                         ((const uint32_t*) sector)[i], vbr_checksum);
 116                         return 1;
 117                 }
 118         return 0;
 119 }
 120
 121 int exfat_mount(struct exfat* ef, const char* spec, const char* options)
 122 {
 123         int rc;
 124
 125         tzset();
 126         memset(ef, 0, sizeof(struct exfat));
 127
 128         ef->sb = malloc(sizeof(struct exfat_super_block));
 129         if (ef->sb == NULL)
 130         {
 131                 exfat_error("memory allocation failed");
 132                 return -ENOMEM;
 133         }
 134         memset(ef->sb, 0, sizeof(struct exfat_super_block));
 135
 136         parse_options(ef, options);
 137
 138         ef->fd = open(spec, ef->ro ? O_RDONLY : O_RDWR);
 139         if (ef->fd < 0)
 140         {
 141                 free(ef->sb);
 142                 exfat_error("failed to open `%s'", spec);
 143                 return -EIO;
 144         }
 145
 146         exfat_read_raw(ef->sb, sizeof(struct exfat_super_block), 0, ef->fd);
 147         if (memcmp(ef->sb->oem_name, "EXFAT   ", 8) != 0)
 148         {
 149                 close(ef->fd);
 150                 free(ef->sb);
 151                 exfat_error("exFAT file system is not found");
 152                 return -EIO;
 153         }
 154         if (ef->sb->version.major != 1 || ef->sb->version.minor != 0)
 155         {
 156                 close(ef->fd);
 157                 exfat_error("unsupported exFAT version: %hhu.%hhu",
 158                                 ef->sb->version.major, ef->sb->version.minor);
 159                 free(ef->sb);
 160                 return -EIO;
 161         }
 162         if (ef->sb->fat_count != 1)
 163         {
 164                 close(ef->fd);
 165                 free(ef->sb);
 166                 exfat_error("unsupported FAT count: %hhu", ef->sb->fat_count);
 167                 return -EIO;
 168         }
 169         /* officially exFAT supports cluster size up to 32 MB */
 170         if ((int) ef->sb->sector_bits + (int) ef->sb->spc_bits > 25)
 171         {
 172                 close(ef->fd);
 173                 free(ef->sb);
 174                 exfat_error("too big cluster size: 2^%d",
 175                                 (int) ef->sb->sector_bits + (int) ef->sb->spc_bits);
 176                 return -EIO;
 177         }
 178
 179         ef->zero_sector = malloc(SECTOR_SIZE(*ef->sb));
 180         if (ef->zero_sector == NULL)
 181         {
 182                 close(ef->fd);
 183                 free(ef->sb);
 184                 exfat_error("failed to allocate zero sector");
 185                 return -ENOMEM;
 186         }
 187         /* use zero_sector as a temporary buffer for VBR checksum verification */
 188         if (verify_vbr_checksum(ef->zero_sector, SECTOR_SIZE(*ef->sb), ef->fd) != 0)
 189         {
 190                 free(ef->zero_sector);
 191                 close(ef->fd);
 192                 free(ef->sb);
 193                 return -EIO;
 194         }
 195         memset(ef->zero_sector, 0, SECTOR_SIZE(*ef->sb));
 196
 197         ef->root = malloc(sizeof(struct exfat_node));
 198         if (ef->root == NULL)
 199         {
 200                 free(ef->zero_sector);
 201                 close(ef->fd);
 202                 free(ef->sb);
 203                 exfat_error("failed to allocate root node");
 204                 return -ENOMEM;
 205         }
 206         memset(ef->root, 0, sizeof(struct exfat_node));
 207         ef->root->flags = EXFAT_ATTRIB_DIR;
 208         ef->root->start_cluster = le32_to_cpu(ef->sb->rootdir_cluster);
 209         ef->root->fptr_cluster = ef->root->start_cluster;
 210         ef->root->name[0] = cpu_to_le16('\0');
 211         ef->root->size = rootdir_size(ef);
 212         /* exFAT does not have time attributes for the root directory */
 213         ef->root->mtime = 0;
 214         ef->root->atime = 0;
 215         /* always keep at least 1 reference to the root node */
 216         exfat_get_node(ef->root);
 217
 218         rc = exfat_cache_directory(ef, ef->root);
 219         if (rc != 0)
 220                 goto error;
 221         if (ef->upcase == NULL)
 222         {
 223                 exfat_error("upcase table is not found");
 224                 goto error;
 225         }
 226         if (ef->cmap.chunk == NULL)
 227         {
 228                 exfat_error("clusters bitmap is not found");
 229                 goto error;
 230         }
 231
 232         return 0;
 233
 234 error:
 235         exfat_put_node(ef, ef->root);
 236         exfat_reset_cache(ef);
 237         free(ef->root);
 238         free(ef->zero_sector);
 239         close(ef->fd);
 240         free(ef->sb);
 241         return -EIO;
 242 }
 243
 244 void exfat_unmount(struct exfat* ef)
 245 {
 246         exfat_put_node(ef, ef->root);
 247         exfat_reset_cache(ef);
 248         free(ef->root);
 249         ef->root = NULL;
 250         free(ef->zero_sector);
 251         ef->zero_sector = NULL;
 252         free(ef->cmap.chunk);
 253         ef->cmap.chunk = NULL;
 254         if (fsync(ef->fd) < 0)
 255                 exfat_error("fsync failed");
 256         if (close(ef->fd) < 0)
 257                 exfat_error("close failed");
 258         ef->fd = 0;
 259         free(ef->sb);
 260         ef->sb = NULL;
 261         free(ef->upcase);
 262         ef->upcase = NULL;
 263         ef->upcase_chars = 0;
 264 }