libexfat/cluster.c

   1 /*
   2         cluster.c (03.09.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 <errno.h>
  23 #include <string.h>
  24
  25 /*
  26  * Sector to absolute offset.
  27  */
  28 static off_t s2o(const struct exfat* ef, off_t sector)
  29 {
  30         return sector << ef->sb->sector_bits;
  31 }
  32
  33 /*
  34  * Cluster to sector.
  35  */
  36 static off_t c2s(const struct exfat* ef, cluster_t cluster)
  37 {
  38         if (cluster < EXFAT_FIRST_DATA_CLUSTER)
  39                 exfat_bug("invalid cluster number %u", cluster);
  40         return le32_to_cpu(ef->sb->cluster_sector_start) +
  41                 ((off_t) (cluster - EXFAT_FIRST_DATA_CLUSTER) << ef->sb->spc_bits);
  42 }
  43
  44 /*
  45  * Cluster to absolute offset.
  46  */
  47 off_t exfat_c2o(const struct exfat* ef, cluster_t cluster)
  48 {
  49         return s2o(ef, c2s(ef, cluster));
  50 }
  51
  52 /*
  53  * Sector to cluster.
  54  */
  55 static cluster_t s2c(const struct exfat* ef, off_t sector)
  56 {
  57         return ((sector - le32_to_cpu(ef->sb->cluster_sector_start)) >>
  58                         ef->sb->spc_bits) + EXFAT_FIRST_DATA_CLUSTER;
  59 }
  60
  61 /*
  62  * Size in bytes to size in clusters (rounded upwards).
  63  */
  64 static uint32_t bytes2clusters(const struct exfat* ef, uint64_t bytes)
  65 {
  66         uint64_t cluster_size = CLUSTER_SIZE(*ef->sb);
  67         return (bytes + cluster_size - 1) / cluster_size;
  68 }
  69
  70 cluster_t exfat_next_cluster(const struct exfat* ef,
  71                 const struct exfat_node* node, cluster_t cluster)
  72 {
  73         le32_t next;
  74         off_t fat_offset;
  75
  76         if (cluster < EXFAT_FIRST_DATA_CLUSTER)
  77                 exfat_bug("bad cluster 0x%x", cluster);
  78
  79         if (IS_CONTIGUOUS(*node))
  80                 return cluster + 1;
  81         fat_offset = s2o(ef, le32_to_cpu(ef->sb->fat_sector_start))
  82                 + cluster * sizeof(cluster_t);
  83         exfat_read_raw(&next, sizeof(next), fat_offset, ef->fd);
  84         return le32_to_cpu(next);
  85 }
  86
  87 cluster_t exfat_advance_cluster(const struct exfat* ef,
  88                 struct exfat_node* node, uint32_t count)
  89 {
  90         uint32_t i;
  91
  92         if (node->fptr_index > count)
  93         {
  94                 node->fptr_index = 0;
  95                 node->fptr_cluster = node->start_cluster;
  96         }
  97
  98         for (i = node->fptr_index; i < count; i++)
  99         {
 100                 node->fptr_cluster = exfat_next_cluster(ef, node, node->fptr_cluster);
 101                 if (CLUSTER_INVALID(node->fptr_cluster))
 102                         break;
 103         }
 104         node->fptr_index = count;
 105         return node->fptr_cluster;
 106 }
 107
 108 static cluster_t find_bit_and_set(uint8_t* bitmap, cluster_t start,
 109                 cluster_t end)
 110 {
 111         const cluster_t mid_start = (start + 7) / 8 * 8;
 112         const cluster_t mid_end = end / 8 * 8;
 113         cluster_t c;
 114         cluster_t byte;
 115
 116         for (c = start; c < mid_start; c++)
 117                 if (BMAP_GET(bitmap, c) == 0)
 118                 {
 119                         BMAP_SET(bitmap, c);
 120                         return c + EXFAT_FIRST_DATA_CLUSTER;
 121                 }
 122
 123         for (byte = mid_start / 8; byte < mid_end / 8; byte++)
 124                 if (bitmap[byte] != 0xff)
 125                 {
 126                         cluster_t bit;
 127
 128                         for (bit = 0; bit < 8; bit++)
 129                                 if (!(bitmap[byte] & (1u << bit)))
 130                                 {
 131                                         bitmap[byte] |= (1u << bit);
 132                                         return byte * 8 + bit + EXFAT_FIRST_DATA_CLUSTER;
 133                                 }
 134                 }
 135
 136         for (c = mid_end; c < end; c++)
 137                 if (BMAP_GET(bitmap, c) == 0)
 138                 {
 139                         BMAP_SET(bitmap, c);
 140                         return c + EXFAT_FIRST_DATA_CLUSTER;
 141                 }
 142
 143         return EXFAT_CLUSTER_END;
 144 }
 145
 146 void exfat_flush_cmap(struct exfat* ef)
 147 {
 148         exfat_write_raw(ef->cmap.chunk, (ef->cmap.chunk_size + 7) / 8,
 149                         exfat_c2o(ef, ef->cmap.start_cluster), ef->fd);
 150         ef->cmap.dirty = 0;
 151 }
 152
 153 static void set_next_cluster(const struct exfat* ef, int contiguous,
 154                 cluster_t current, cluster_t next)
 155 {
 156         off_t fat_offset;
 157         le32_t next_le32;
 158
 159         if (contiguous)
 160                 return;
 161         fat_offset = s2o(ef, le32_to_cpu(ef->sb->fat_sector_start))
 162                 + current * sizeof(cluster_t);
 163         next_le32 = cpu_to_le32(next);
 164         exfat_write_raw(&next_le32, sizeof(next_le32), fat_offset, ef->fd);
 165 }
 166
 167 static cluster_t allocate_cluster(struct exfat* ef, cluster_t hint)
 168 {
 169         cluster_t cluster;
 170
 171         hint -= EXFAT_FIRST_DATA_CLUSTER;
 172         if (hint >= ef->cmap.chunk_size)
 173                 hint = 0;
 174
 175         cluster = find_bit_and_set(ef->cmap.chunk, hint, ef->cmap.chunk_size);
 176         if (cluster == EXFAT_CLUSTER_END)
 177                 cluster = find_bit_and_set(ef->cmap.chunk, 0, hint);
 178         if (cluster == EXFAT_CLUSTER_END)
 179         {
 180                 exfat_error("no free space left");
 181                 return EXFAT_CLUSTER_END;
 182         }
 183
 184         ef->cmap.dirty = 1;
 185         /* FIXME update percentage of used space */
 186         return cluster;
 187 }
 188
 189 static void free_cluster(struct exfat* ef, cluster_t cluster)
 190 {
 191         if (CLUSTER_INVALID(cluster))
 192                 exfat_bug("attempting to free invalid cluster");
 193         if (cluster < EXFAT_FIRST_DATA_CLUSTER ||
 194                 cluster - EXFAT_FIRST_DATA_CLUSTER >= ef->cmap.size)
 195                 exfat_bug("bad cluster 0x%x (0x%x)", cluster, ef->cmap.size);
 196
 197         BMAP_CLR(ef->cmap.chunk, cluster - EXFAT_FIRST_DATA_CLUSTER);
 198         ef->cmap.dirty = 1;
 199         /* FIXME update percentage of used space */
 200 }
 201
 202 static void make_noncontiguous(const struct exfat* ef, cluster_t first,
 203                 cluster_t last)
 204 {
 205         cluster_t c;
 206
 207         for (c = first; c < last; c++)
 208                 set_next_cluster(ef, 0, c, c + 1);
 209 }
 210
 211 static int shrink_file(struct exfat* ef, struct exfat_node* node,
 212                 uint32_t current, uint32_t difference);
 213
 214 static int grow_file(struct exfat* ef, struct exfat_node* node,
 215                 uint32_t current, uint32_t difference)
 216 {
 217         cluster_t previous;
 218         cluster_t next;
 219         uint32_t allocated = 0;
 220
 221         if (difference == 0)
 222                 exfat_bug("zero clusters count passed");
 223
 224         if (node->start_cluster != EXFAT_CLUSTER_FREE)
 225         {
 226                 /* get the last cluster of the file */
 227                 previous = exfat_advance_cluster(ef, node, current - 1);
 228                 if (CLUSTER_INVALID(previous))
 229                 {
 230                         exfat_error("invalid cluster in file");
 231                         return -EIO;
 232                 }
 233         }
 234         else
 235         {
 236                 if (node->fptr_index != 0)
 237                         exfat_bug("non-zero pointer index (%u)", node->fptr_index);
 238                 /* file does not have clusters (i.e. is empty), allocate
 239                    the first one for it */
 240                 previous = allocate_cluster(ef, 0);
 241                 if (CLUSTER_INVALID(previous))
 242                         return -ENOSPC;
 243                 node->fptr_cluster = node->start_cluster = previous;
 244                 allocated = 1;
 245                 /* file consists of only one cluster, so it's contiguous */
 246                 node->flags |= EXFAT_ATTRIB_CONTIGUOUS;
 247         }
 248
 249         while (allocated < difference)
 250         {
 251                 next = allocate_cluster(ef, previous + 1);
 252                 if (CLUSTER_INVALID(next))
 253                 {
 254                         if (allocated != 0)
 255                                 shrink_file(ef, node, current + allocated, allocated);
 256                         return -ENOSPC;
 257                 }
 258                 if (next != previous - 1 && IS_CONTIGUOUS(*node))
 259                 {
 260                         /* it's a pity, but we are not able to keep the file contiguous
 261                            anymore */
 262                         make_noncontiguous(ef, node->start_cluster, previous);
 263                         node->flags &= ~EXFAT_ATTRIB_CONTIGUOUS;
 264                         node->flags |= EXFAT_ATTRIB_DIRTY;
 265                 }
 266                 set_next_cluster(ef, IS_CONTIGUOUS(*node), previous, next);
 267                 previous = next;
 268                 allocated++;
 269         }
 270
 271         set_next_cluster(ef, IS_CONTIGUOUS(*node), previous, EXFAT_CLUSTER_END);
 272         return 0;
 273 }
 274
 275 static int shrink_file(struct exfat* ef, struct exfat_node* node,
 276                 uint32_t current, uint32_t difference)
 277 {
 278         cluster_t previous;
 279         cluster_t next;
 280
 281         if (difference == 0)
 282                 exfat_bug("zero difference passed");
 283         if (node->start_cluster == EXFAT_CLUSTER_FREE)
 284                 exfat_bug("unable to shrink empty file (%u clusters)", current);
 285         if (current < difference)
 286                 exfat_bug("file underflow (%u < %u)", current, difference);
 287
 288         /* crop the file */
 289         if (current > difference)
 290         {
 291                 cluster_t last = exfat_advance_cluster(ef, node,
 292                                 current - difference - 1);
 293                 if (CLUSTER_INVALID(last))
 294                 {
 295                         exfat_error("invalid cluster in file");
 296                         return -EIO;
 297                 }
 298                 previous = exfat_next_cluster(ef, node, last);
 299                 set_next_cluster(ef, IS_CONTIGUOUS(*node), last, EXFAT_CLUSTER_END);
 300         }
 301         else
 302         {
 303                 previous = node->start_cluster;
 304                 node->start_cluster = EXFAT_CLUSTER_FREE;
 305         }
 306         node->fptr_index = 0;
 307         node->fptr_cluster = node->start_cluster;
 308
 309         /* free remaining clusters */
 310         while (difference--)
 311         {
 312                 if (CLUSTER_INVALID(previous))
 313                 {
 314                         exfat_error("invalid cluster in file");
 315                         return -EIO;
 316                 }
 317                 next = exfat_next_cluster(ef, node, previous);
 318                 set_next_cluster(ef, IS_CONTIGUOUS(*node), previous,
 319                                 EXFAT_CLUSTER_FREE);
 320                 free_cluster(ef, previous);
 321                 previous = next;
 322         }
 323         return 0;
 324 }
 325
 326 static void erase_raw(struct exfat* ef, size_t size, off_t offset)
 327 {
 328         exfat_write_raw(ef->zero_cluster, size, offset, ef->fd);
 329 }
 330
 331 static int erase_range(struct exfat* ef, struct exfat_node* node,
 332                 uint64_t begin, uint64_t end)
 333 {
 334         uint64_t cluster_boundary;
 335         cluster_t cluster;
 336
 337         if (begin >= end)
 338                 return 0;
 339
 340         cluster_boundary = (begin | (CLUSTER_SIZE(*ef->sb) - 1)) + 1;
 341         cluster = exfat_advance_cluster(ef, node,
 342                         begin / CLUSTER_SIZE(*ef->sb));
 343         if (CLUSTER_INVALID(cluster))
 344         {
 345                 exfat_error("invalid cluster in file");
 346                 return -EIO;
 347         }
 348         /* erase from the beginning to the closest cluster boundary */
 349         erase_raw(ef, MIN(cluster_boundary, end) - begin,
 350                         exfat_c2o(ef, cluster) + begin % CLUSTER_SIZE(*ef->sb));
 351         /* erase whole clusters */
 352         while (cluster_boundary < end)
 353         {
 354                 cluster = exfat_next_cluster(ef, node, cluster);
 355                 /* the cluster cannot be invalid because we have just allocated it */
 356                 if (CLUSTER_INVALID(cluster))
 357                         exfat_bug("invalid cluster in file");
 358                 erase_raw(ef, CLUSTER_SIZE(*ef->sb), exfat_c2o(ef, cluster));
 359                 cluster_boundary += CLUSTER_SIZE(*ef->sb);
 360         }
 361         return 0;
 362 }
 363
 364 int exfat_truncate(struct exfat* ef, struct exfat_node* node, uint64_t size)
 365 {
 366         uint32_t c1 = bytes2clusters(ef, node->size);
 367         uint32_t c2 = bytes2clusters(ef, size);
 368         int rc = 0;
 369
 370         if (node->references == 0 && node->parent)
 371                 exfat_bug("no references, node changes can be lost");
 372
 373         if (node->size == size)
 374                 return 0;
 375
 376         if (c1 < c2)
 377                 rc = grow_file(ef, node, c1, c2 - c1);
 378         else if (c1 > c2)
 379                 rc = shrink_file(ef, node, c1, c1 - c2);
 380
 381         if (rc != 0)
 382                 return rc;
 383
 384         rc = erase_range(ef, node, node->size, size);
 385         if (rc != 0)
 386                 return rc;
 387
 388         exfat_update_mtime(node);
 389         node->size = size;
 390         node->flags |= EXFAT_ATTRIB_DIRTY;
 391         return 0;
 392 }
 393
 394 uint32_t exfat_count_free_clusters(const struct exfat* ef)
 395 {
 396         uint32_t free_clusters = 0;
 397         uint32_t i;
 398
 399         for (i = 0; i < ef->cmap.size; i++)
 400                 if (BMAP_GET(ef->cmap.chunk, i) == 0)
 401                         free_clusters++;
 402         return free_clusters;
 403 }
 404
 405 static int find_used_clusters(const struct exfat* ef,
 406                 cluster_t* a, cluster_t* b)
 407 {
 408         const cluster_t end = le32_to_cpu(ef->sb->cluster_count);
 409
 410         /* find first used cluster */
 411         for (*a = *b + 1; *a < end; (*a)++)
 412                 if (BMAP_GET(ef->cmap.chunk, *a - EXFAT_FIRST_DATA_CLUSTER))
 413                         break;
 414         if (*a >= end)
 415                 return 1;
 416
 417         /* find last contiguous used cluster */
 418         for (*b = *a; *b < end; (*b)++)
 419                 if (BMAP_GET(ef->cmap.chunk, *b - EXFAT_FIRST_DATA_CLUSTER) == 0)
 420                 {
 421                         (*b)--;
 422                         break;
 423                 }
 424
 425         return 0;
 426 }
 427
 428 int exfat_find_used_sectors(const struct exfat* ef, off_t* a, off_t* b)
 429 {
 430         cluster_t ca, cb;
 431
 432         if (*a == 0 && *b == 0)
 433                 ca = cb = EXFAT_FIRST_DATA_CLUSTER - 1;
 434         else
 435         {
 436                 ca = s2c(ef, *a);
 437                 cb = s2c(ef, *b);
 438         }
 439         if (find_used_clusters(ef, &ca, &cb) != 0)
 440                 return 1;
 441         if (*a != 0 || *b != 0)
 442                 *a = c2s(ef, ca);
 443         *b = c2s(ef, cb) + (CLUSTER_SIZE(*ef->sb) - 1) / SECTOR_SIZE(*ef->sb);
 444         return 0;
 445 }