Merge branch 'master' of ssh://git.sven.stormbind.net/home/sven/www/sven_htdocs/git...
[sven/exfat-utils.git] / 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 }