source: mainline/uspace/lib/ext4/libext4_extent.c@ 728d771

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 728d771 was 728d771, checked in by Frantisek Princ <frantisek.princ@…>, 13 years ago

next part of comments on extent code

  • Property mode set to 100644
File size: 26.0 KB
Line 
1/*
2 * Copyright (c) 2012 Frantisek Princ
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup libext4
30 * @{
31 */
32
33/**
34 * @file libext4_extent.c
35 * @brief Ext4 extent structures operations.
36 */
37
38#include <byteorder.h>
39#include <errno.h>
40#include <malloc.h>
41#include "libext4.h"
42
43/** Get logical number of the block covered by extent.
44 *
45 * @param extent extent to load number from
46 * @return logical number of the first block covered by extent
47 */
48uint32_t ext4_extent_get_first_block(ext4_extent_t *extent)
49{
50 return uint32_t_le2host(extent->first_block);
51}
52
53/** Set logical number of the first block covered by extent.
54 *
55 * @param extent extent to set number to
56 * @param iblock logical number of the first block covered by extent
57 */
58void ext4_extent_set_first_block(ext4_extent_t *extent, uint32_t iblock)
59{
60 extent->first_block = host2uint32_t_le(iblock);
61}
62
63/** Get number of blocks covered by extent.
64 *
65 * @param extent extent to load count from
66 * @return number of blocks covered by extent
67 */
68uint16_t ext4_extent_get_block_count(ext4_extent_t *extent)
69{
70 return uint16_t_le2host(extent->block_count);
71}
72
73/** Set number of blocks covered by extent.
74 *
75 * @param extent extent to load count from
76 * @param count number of blocks covered by extent
77 */
78void ext4_extent_set_block_count(ext4_extent_t *extent, uint16_t count)
79{
80 extent->block_count = host2uint16_t_le(count);
81}
82
83/** Get physical number of the first block covered by extent.
84 *
85 * @param extent extent to load number
86 * @return physical number of the first block covered by extent
87 */
88uint64_t ext4_extent_get_start(ext4_extent_t *extent)
89{
90 return ((uint64_t)uint16_t_le2host(extent->start_hi)) << 32 |
91 ((uint64_t)uint32_t_le2host(extent->start_lo));
92}
93
94/** Set physical number of the first block covered by extent.
95 *
96 * @param extent extent to load number
97 * @param fblock physical number of the first block covered by extent
98 */
99void ext4_extent_set_start(ext4_extent_t *extent, uint64_t fblock)
100{
101 extent->start_lo = host2uint32_t_le((fblock << 32) >> 32);
102 extent->start_hi = host2uint16_t_le((uint16_t)(fblock >> 32));
103}
104
105/** Get logical number of the block covered by extent index.
106 *
107 * @param index extent index to load number from
108 * @return logical number of the first block covered by extent index
109 */
110uint32_t ext4_extent_index_get_first_block(ext4_extent_index_t *index)
111{
112 return uint32_t_le2host(index->first_block);
113}
114
115/** Set logical number of the block covered by extent index.
116 *
117 * @param index extent index to set number to
118 * @param iblock logical number of the first block covered by extent index
119 */
120void ext4_extent_index_set_first_block(ext4_extent_index_t *index,
121 uint32_t iblock)
122{
123 index->first_block = host2uint32_t_le(iblock);
124}
125
126/** Get physical number of block where the child node is located.
127 *
128 * @param index extent index to load number from
129 * @return physical number of the block with child node
130 */
131uint64_t ext4_extent_index_get_leaf(ext4_extent_index_t *index)
132{
133 return ((uint64_t)uint16_t_le2host(index->leaf_hi)) << 32 |
134 ((uint64_t)uint32_t_le2host(index->leaf_lo));
135}
136
137/** Set physical number of block where the child node is located.
138 *
139 * @param index extent index to set number to
140 * @param fblock physical number of the block with child node
141 */
142void ext4_extent_index_set_leaf(ext4_extent_index_t *index, uint64_t fblock)
143{
144 index->leaf_lo = host2uint32_t_le((fblock << 32) >> 32);
145 index->leaf_hi = host2uint16_t_le((uint16_t)(fblock >> 32));
146}
147
148/** Get magic value from extent header.
149 *
150 * @param header extent header to load value from
151 * @return magic value of extent header
152 */
153uint16_t ext4_extent_header_get_magic(ext4_extent_header_t *header)
154{
155 return uint16_t_le2host(header->magic);
156}
157
158/** Set magic value to extent header.
159 *
160 * @param header extent header to set value to
161 * @param magic magic value of extent header
162 */
163void ext4_extent_header_set_magic(ext4_extent_header_t *header, uint16_t magic)
164{
165 header->magic = host2uint16_t_le(magic);
166}
167
168/** Get number of entries from extent header
169 *
170 * @param header extent header to get value from
171 * @return number of entries covered by extent header
172 */
173uint16_t ext4_extent_header_get_entries_count(ext4_extent_header_t *header)
174{
175 return uint16_t_le2host(header->entries_count);
176}
177
178/** Set number of entries to extent header
179 *
180 * @param header extent header to set value to
181 * @param count number of entries covered by extent header
182 */
183void ext4_extent_header_set_entries_count(ext4_extent_header_t *header,
184 uint16_t count)
185{
186 header->entries_count = host2uint16_t_le(count);
187}
188
189/** Get maximum number of entries from extent header
190 *
191 * @param header extent header to get value from
192 * @return maximum number of entries covered by extent header
193 */
194uint16_t ext4_extent_header_get_max_entries_count(ext4_extent_header_t *header)
195{
196 return uint16_t_le2host(header->max_entries_count);
197}
198
199/** Set maximum number of entries to extent header
200 *
201 * @param header extent header to set value to
202 * @param max_count maximum number of entries covered by extent header
203 */
204void ext4_extent_header_set_max_entries_count(ext4_extent_header_t *header,
205 uint16_t max_count)
206{
207 header->max_entries_count = host2uint16_t_le(max_count);
208}
209
210/** Get depth of extent subtree.
211 *
212 * @param header extent header to get value from
213 * @return depth of extent subtree
214 */
215uint16_t ext4_extent_header_get_depth(ext4_extent_header_t *header)
216{
217 return uint16_t_le2host(header->depth);
218}
219
220/** Set depth of extent subtree.
221 *
222 * @param header extent header to set value to
223 * @param depth depth of extent subtree
224 */
225void ext4_extent_header_set_depth(ext4_extent_header_t *header, uint16_t depth)
226{
227 header->depth = host2uint16_t_le(depth);
228}
229
230/** Get generation from extent header
231 *
232 * @param header extent header to get value from
233 * @return generation
234 */
235uint32_t ext4_extent_header_get_generation(ext4_extent_header_t *header)
236{
237 return uint32_t_le2host(header->generation);
238}
239
240/** Set generation to extent header
241 *
242 * @param header extent header to set value to
243 * @param generation generation
244 */
245void ext4_extent_header_set_generation(ext4_extent_header_t *header,
246 uint32_t generation)
247{
248 header->generation = host2uint32_t_le(generation);
249}
250
251/** Binary search in extent index node.
252 *
253 * @param header extent header of index node
254 * @param index output value - found index will be set here
255 * @param iblock logical block number to find in index node
256 */
257static void ext4_extent_binsearch_idx(ext4_extent_header_t *header,
258 ext4_extent_index_t **index, uint32_t iblock)
259{
260 ext4_extent_index_t *r, *l, *m;
261
262 uint16_t entries_count = ext4_extent_header_get_entries_count(header);
263
264 // Check trivial situation
265 if (entries_count == 1) {
266 *index = EXT4_EXTENT_FIRST_INDEX(header);
267 return;
268 }
269
270 // Initialize bounds
271 l = EXT4_EXTENT_FIRST_INDEX(header) + 1;
272 r = l + entries_count - 1;
273
274 // Do binary search
275 while (l <= r) {
276 m = l + (r - l) / 2;
277 uint32_t first_block = ext4_extent_index_get_first_block(m);
278 if (iblock < first_block) {
279 r = m - 1;
280 } else {
281 l = m + 1;
282 }
283 }
284
285 // Set output value
286 *index = l - 1;
287}
288
289/** Binary search in extent leaf node.
290 * @param header extent header of leaf node
291 * @param extent output value - found extent will be set here,
292 * or NULL if node is empty
293 * @param iblock logical block number to find in leaf node
294 *
295 */
296static void ext4_extent_binsearch(ext4_extent_header_t *header,
297 ext4_extent_t **extent, uint32_t iblock)
298{
299 ext4_extent_t *r, *l, *m;
300
301 uint16_t entries_count = ext4_extent_header_get_entries_count(header);
302
303 if (entries_count == 0) {
304 // this leaf is empty
305 *extent = NULL;
306 return;
307 }
308
309 // Check trivial situation
310 if (entries_count == 1) {
311 *extent = EXT4_EXTENT_FIRST(header);
312 return;
313 }
314
315 // Initialize bounds
316 l = EXT4_EXTENT_FIRST(header) + 1;
317 r = l + entries_count - 1;
318
319 // Do binary search
320 while (l < r) {
321 m = l + (r - l) / 2;
322 uint32_t first_block = ext4_extent_get_first_block(m);
323 if (iblock < first_block) {
324 r = m - 1;
325 } else {
326 l = m + 1;
327 }
328 }
329
330 // Set output value
331 *extent = l - 1;
332}
333
334/** Find physical block in the extent tree by logical block number.
335 *
336 * There is no need to save path in the tree during this algorithm.
337 *
338 * @param inode_ref i-node to load block from
339 * @param iblock logical block number to find
340 * @param fblock output value for physical block number
341 * @return error code
342 */
343int ext4_extent_find_block(ext4_inode_ref_t *inode_ref,
344 uint32_t iblock, uint32_t *fblock)
345{
346 int rc;
347
348 // Compute bound defined by i-node size
349 uint64_t inode_size = ext4_inode_get_size(
350 inode_ref->fs->superblock, inode_ref->inode);
351
352 uint32_t block_size = ext4_superblock_get_block_size(
353 inode_ref->fs->superblock);
354
355 uint32_t last_idx = (inode_size - 1) / block_size;
356
357 // Check if requested iblock is not over size of i-node
358 if (iblock > last_idx) {
359 *fblock = 0;
360 return EOK;
361 }
362
363 block_t* block = NULL;
364
365 // Walk through extent tree
366 ext4_extent_header_t *header = ext4_inode_get_extent_header(inode_ref->inode);
367 while (ext4_extent_header_get_depth(header) != 0) {
368
369 // Search index in node
370 ext4_extent_index_t *index;
371 ext4_extent_binsearch_idx(header, &index, iblock);
372
373 // Load child node and set values for the next iteration
374 uint64_t child = ext4_extent_index_get_leaf(index);
375
376 if (block != NULL) {
377 block_put(block);
378 }
379
380 rc = block_get(&block, inode_ref->fs->device, child, BLOCK_FLAGS_NONE);
381 if (rc != EOK) {
382 return rc;
383 }
384
385 header = (ext4_extent_header_t *)block->data;
386 }
387
388 // Search extent in the leaf block
389 ext4_extent_t* extent = NULL;
390 ext4_extent_binsearch(header, &extent, iblock);
391
392 // Prevent empty leaf
393 if (extent == NULL) {
394 *fblock = 0;
395 } else {
396
397 // Compute requested physical block address
398 uint32_t phys_block;
399 phys_block = ext4_extent_get_start(extent) + iblock;
400 phys_block -= ext4_extent_get_first_block(extent);
401
402 *fblock = phys_block;
403 }
404
405 // Cleanup
406 if (block != NULL) {
407 block_put(block);
408 }
409
410 return EOK;
411}
412
413
414/** Find extent for specified iblock.
415 *
416 * This function is used for finding block in the extent tree with
417 * saving the path through the tree for possible future modifications.
418 *
419 * @param inode_ref i-node to read extent tree from
420 * @param iblock iblock to find extent for
421 * @param ret_path output value for loaded path from extent tree
422 * @return error code
423 */
424static int ext4_extent_find_extent(ext4_inode_ref_t *inode_ref,
425 uint32_t iblock, ext4_extent_path_t **ret_path)
426{
427 int rc;
428
429 ext4_extent_header_t *eh =
430 ext4_inode_get_extent_header(inode_ref->inode);
431
432 uint16_t depth = ext4_extent_header_get_depth(eh);
433
434 ext4_extent_path_t *tmp_path;
435
436 // Added 2 for possible tree growing
437 tmp_path = malloc(sizeof(ext4_extent_path_t) * (depth + 2));
438 if (tmp_path == NULL) {
439 return ENOMEM;
440 }
441
442 // Initialize structure for algorithm start
443 tmp_path[0].block = inode_ref->block;
444 tmp_path[0].header = eh;
445
446 // Walk through the extent tree
447 uint16_t pos = 0;
448 while (ext4_extent_header_get_depth(eh) != 0) {
449
450 // Search index in index node by iblock
451 ext4_extent_binsearch_idx(tmp_path[pos].header, &tmp_path[pos].index, iblock);
452
453 tmp_path[pos].depth = depth;
454 tmp_path[pos].extent = NULL;
455
456 assert(tmp_path[pos].index != NULL);
457
458 // Load information for the next iteration
459 uint64_t fblock = ext4_extent_index_get_leaf(tmp_path[pos].index);
460
461 block_t *block;
462 rc = block_get(&block, inode_ref->fs->device, fblock, BLOCK_FLAGS_NONE);
463 if (rc != EOK) {
464 goto cleanup;
465 }
466
467 pos++;
468
469 eh = (ext4_extent_header_t *)block->data;
470 tmp_path[pos].block = block;
471 tmp_path[pos].header = eh;
472
473 }
474
475 tmp_path[pos].depth = 0;
476 tmp_path[pos].extent = NULL;
477 tmp_path[pos].index = NULL;
478
479 // Find extent in the leaf node
480 ext4_extent_binsearch(tmp_path[pos].header, &tmp_path[pos].extent, iblock);
481
482 *ret_path = tmp_path;
483
484 return EOK;
485
486cleanup:
487 // Put loaded blocks
488 // From 1 -> 0 is a block with inode data
489 for (uint16_t i = 1; i < tmp_path->depth; ++i) {
490 if (tmp_path[i].block) {
491 block_put(tmp_path[i].block);
492 }
493 }
494
495 // Destroy temporary data structure
496 free(tmp_path);
497
498 return rc;
499}
500
501/** Release extent and all data blocks covered by the extent.
502 *
503 * @param inode_ref i-node to release extent and block from
504 * @param extent extent to release
505 * @return error code
506 */
507static int ext4_extent_release(
508 ext4_inode_ref_t *inode_ref, ext4_extent_t *extent)
509{
510 int rc;
511
512 // Compute number of the first physical block to release
513 uint64_t start = ext4_extent_get_start(extent);
514 uint16_t block_count = ext4_extent_get_block_count(extent);
515
516 rc = ext4_balloc_free_blocks(inode_ref, start, block_count);
517 if (rc != EOK) {
518 EXT4FS_DBG("Error in releasing data blocks");
519 return rc;
520 }
521
522 return EOK;
523}
524
525// TODO comments
526
527// Recursive release
528static int ext4_extent_release_branch(ext4_inode_ref_t *inode_ref,
529 ext4_extent_index_t *index)
530{
531 int rc;
532
533 block_t* block;
534
535 uint32_t fblock = ext4_extent_index_get_leaf(index);
536
537// EXT4FS_DBG("fblock = \%u", fblock);
538
539 rc = block_get(&block, inode_ref->fs->device, fblock, BLOCK_FLAGS_NOREAD);
540 if (rc != EOK) {
541 EXT4FS_DBG("ERROR get_block");
542 return rc;
543 }
544
545 ext4_extent_header_t *header = block->data;
546
547 if (ext4_extent_header_get_depth(header)) {
548
549 ext4_extent_index_t *idx = EXT4_EXTENT_FIRST_INDEX(header);
550
551 for (uint32_t i = 0; i < ext4_extent_header_get_entries_count(header); ++i, ++idx) {
552 rc = ext4_extent_release_branch(inode_ref, idx);
553 if (rc != EOK) {
554 EXT4FS_DBG("error recursion");
555 return rc;
556 }
557 }
558 } else {
559 ext4_extent_t *ext = EXT4_EXTENT_FIRST(header);
560
561 for (uint32_t i = 0; i < ext4_extent_header_get_entries_count(header); ++i, ++ext) {
562 rc = ext4_extent_release(inode_ref, ext);
563 if (rc != EOK) {
564 EXT4FS_DBG("error recursion");
565 return rc;
566 }
567 }
568 }
569
570 rc = block_put(block);
571 if (rc != EOK) {
572 EXT4FS_DBG("ERROR put_block");
573 return rc;
574 }
575
576 ext4_balloc_free_block(inode_ref, fblock);
577
578 return EOK;
579}
580
581int ext4_extent_release_blocks_from(ext4_inode_ref_t *inode_ref,
582 uint32_t iblock_from)
583{
584 int rc = EOK;
585
586 // Find the first extent to modify
587 ext4_extent_path_t *path;
588 rc = ext4_extent_find_extent(inode_ref, iblock_from, &path);
589 if (rc != EOK) {
590 return rc;
591 }
592
593 // Jump to last item of the path (extent)
594 ext4_extent_path_t *path_ptr = path;
595 while (path_ptr->depth != 0) {
596 path_ptr++;
597 }
598
599 assert(path_ptr->extent != NULL);
600
601 // First extent maybe released partially
602 uint32_t first_fblock;
603 first_fblock = ext4_extent_get_start(path_ptr->extent) + iblock_from;
604 first_fblock -= ext4_extent_get_first_block(path_ptr->extent);
605
606 uint16_t block_count = ext4_extent_get_block_count(path_ptr->extent);
607
608 uint16_t delete_count = block_count - first_fblock +
609 ext4_extent_get_start(path_ptr->extent);
610
611 rc = ext4_balloc_free_blocks(inode_ref, first_fblock, delete_count);
612 if (rc != EOK) {
613 goto cleanup;
614 }
615
616 block_count -= delete_count;
617 ext4_extent_set_block_count(path_ptr->extent, block_count);
618
619 uint16_t entries = ext4_extent_header_get_entries_count(path_ptr->header);
620 ext4_extent_t *tmp_ext = path_ptr->extent + 1;
621 ext4_extent_t *stop_ext = EXT4_EXTENT_FIRST(path_ptr->header) + entries;
622
623 // If first extent empty, release it
624 if (block_count == 0) {
625 entries--;
626 ext4_extent_header_set_entries_count(path_ptr->header, entries);
627 }
628
629 // Release all successors of the first extent in the same node
630 while (tmp_ext < stop_ext) {
631 first_fblock = ext4_extent_get_start(tmp_ext);
632 delete_count = ext4_extent_get_block_count(tmp_ext);
633
634 rc = ext4_balloc_free_blocks(inode_ref, first_fblock, delete_count);
635 if (rc != EOK) {
636 goto cleanup;
637 }
638
639 entries--;
640 ext4_extent_header_set_entries_count(path_ptr->header, entries);
641
642 tmp_ext++;
643 }
644
645 // If leaf node is empty, the whole tree must be checked and the node will be released
646 bool check_tree = false;
647
648 // Don't release root block (including inode data) !!!
649 if ((path_ptr != path) && (entries == 0)) {
650 rc = ext4_balloc_free_block(inode_ref, path_ptr->block->lba);
651 if (rc != EOK) {
652 goto cleanup;
653 }
654 check_tree = true;
655 }
656
657 // Jump to the parent
658 --path_ptr;
659
660 // release all successors in all levels
661 while(path_ptr >= path) {
662 entries = ext4_extent_header_get_entries_count(path_ptr->header);
663 ext4_extent_index_t *index = path_ptr->index + 1;
664 ext4_extent_index_t *stop =
665 EXT4_EXTENT_FIRST_INDEX(path_ptr->header) + entries;
666
667 if (check_tree) {
668 entries--;
669 ext4_extent_header_set_entries_count(path_ptr->header, entries);
670 }
671
672 while (index < stop) {
673 rc = ext4_extent_release_branch(inode_ref, index);
674 if (rc != EOK) {
675 goto cleanup;
676 }
677 ++index;
678 --entries;
679 ext4_extent_header_set_entries_count(path_ptr->header, entries);
680 }
681
682 path_ptr->block->dirty = true;
683
684 if ((entries == 0) && (path_ptr != path)) {
685 rc = ext4_balloc_free_block(inode_ref, path_ptr->block->lba);
686 if (rc != EOK) {
687 goto cleanup;
688 }
689 check_tree = true;
690 } else {
691 check_tree = false;
692 }
693
694 --path_ptr;
695 }
696
697
698cleanup:
699 // Put loaded blocks
700 // From 1 -> 0 is a block with inode data
701 for (uint16_t i = 1; i <= path->depth; ++i) {
702 if (path[i].block) {
703 block_put(path[i].block);
704 }
705 }
706
707 // Destroy temporary data structure
708 free(path);
709
710 return rc;
711}
712
713static int ext4_extent_append_extent(ext4_inode_ref_t *inode_ref,
714 ext4_extent_path_t *path, ext4_extent_path_t **last_path_item,
715 uint32_t iblock)
716{
717 EXT4FS_DBG("");
718 int rc;
719
720 ext4_extent_path_t *path_ptr = *last_path_item;
721
722 uint16_t entries = ext4_extent_header_get_entries_count(path_ptr->header);
723 uint16_t limit = ext4_extent_header_get_max_entries_count(path_ptr->header);
724
725 // Trivial way - no splitting
726 if (entries < limit) {
727 EXT4FS_DBG("adding extent entry");
728
729 ext4_extent_header_set_entries_count(path_ptr->header, entries + 1);
730 path_ptr->extent = EXT4_EXTENT_FIRST(path_ptr->header) + entries;
731 ext4_extent_set_block_count(path_ptr->extent, 0);
732 ext4_extent_set_first_block(path_ptr->extent, iblock);
733 ext4_extent_set_start(path_ptr->extent, 0);
734 path_ptr->block->dirty = true;
735
736 return EOK;
737 }
738
739 uint32_t block_size =
740 ext4_superblock_get_block_size(inode_ref->fs->superblock);
741
742 // Trivial tree - grow (extents were in root node)
743 if (path_ptr == path) {
744
745 uint32_t new_fblock;
746 rc = ext4_balloc_alloc_block(inode_ref, &new_fblock);
747 if (rc != EOK) {
748 EXT4FS_DBG("error in block allocation");
749 return rc;
750 }
751
752 block_t *block;
753 rc = block_get(&block, inode_ref->fs->device,
754 new_fblock, BLOCK_FLAGS_NOREAD);
755 if (rc != EOK) {
756 EXT4FS_DBG("error in block_get");
757 return rc;
758 }
759
760 memset(block->data, 0, block_size);
761
762 // Move data from root to the new block
763 memcpy(block->data, inode_ref->inode->blocks,
764 EXT4_INODE_BLOCKS * sizeof(uint32_t));
765
766 path_ptr++;
767 path_ptr->block = block;
768 path_ptr->header = (ext4_extent_header_t *)block->data;
769 path_ptr->depth = ext4_extent_header_get_depth(path_ptr->header);
770 path_ptr->index = NULL;
771
772 uint16_t entries = ext4_extent_header_get_entries_count(path_ptr->header);
773 path_ptr->extent = EXT4_EXTENT_FIRST(path_ptr->header) + entries;
774 ext4_extent_header_set_entries_count(path_ptr->header, entries + 1);
775 uint16_t limit = (block_size - sizeof(ext4_extent_header_t)) /
776 sizeof(ext4_extent_t);
777 ext4_extent_header_set_max_entries_count(path_ptr->header, limit);
778
779 // Modify root (in inode)
780 path->depth = 1;
781 path->extent = NULL;
782 path->index = EXT4_EXTENT_FIRST_INDEX(path->header);
783
784 ext4_extent_header_set_depth(path->header, path_ptr->depth + 1);
785 ext4_extent_header_set_entries_count(path->header, 1);
786
787 ext4_extent_index_set_first_block(path->index, 0);
788 ext4_extent_index_set_leaf(path->index, new_fblock);
789
790 path_ptr->block->dirty = true;
791 path->block->dirty = true;
792
793 *last_path_item = path_ptr;
794
795 return EOK;
796 }
797
798 assert(false);
799
800 // start splitting
801 uint32_t fblock = 0;
802 while (path_ptr > path) {
803
804 // TODO
805
806 rc = ext4_balloc_alloc_block(inode_ref, &fblock);
807 if (rc != EOK) {
808 return rc;
809 }
810
811 block_t *block;
812 rc = block_get(&block, inode_ref->fs->device, fblock, BLOCK_FLAGS_NOREAD);
813 if (rc != EOK) {
814 ext4_balloc_free_block(inode_ref, fblock);
815 return rc;
816 }
817
818 // Init block
819 memset(block->data, 0, block_size);
820
821 // Not modified
822 block_put(path_ptr->block);
823 path_ptr->block = block;
824
825 path_ptr->header = block->data;
826
827 if (path_ptr->depth) {
828 path_ptr->index = EXT4_EXTENT_FIRST_INDEX(path_ptr->header);
829 } else {
830 path_ptr->extent = EXT4_EXTENT_FIRST(path_ptr->header);
831 }
832
833 path_ptr--;
834 }
835
836 // If splitting reached root node
837 if (path_ptr == path) {
838
839 uint32_t new_fblock;
840 rc = ext4_balloc_alloc_block(inode_ref, &new_fblock);
841 if (rc != EOK) {
842 EXT4FS_DBG("error in block allocation");
843 return rc;
844 }
845
846 block_t *block;
847 rc = block_get(&block, inode_ref->fs->device,
848 new_fblock, BLOCK_FLAGS_NOREAD);
849 if (rc != EOK) {
850 EXT4FS_DBG("error in block_get");
851 return rc;
852 }
853
854 memset(block->data, 0, block_size);
855
856 // Move data from root to the new block
857 memcpy(block->data, inode_ref->inode->blocks,
858 EXT4_INODE_BLOCKS * sizeof(uint32_t));
859
860 path_ptr++;
861 path_ptr->block = block;
862 path_ptr->header = (ext4_extent_header_t *)block->data;
863 path_ptr->depth = ext4_extent_header_get_depth(path_ptr->header);
864 path_ptr->index = NULL;
865
866 uint16_t entries = ext4_extent_header_get_entries_count(path_ptr->header);
867 path_ptr->extent = EXT4_EXTENT_FIRST(path_ptr->header) + entries;
868 ext4_extent_header_set_entries_count(path_ptr->header, entries + 1);
869 uint16_t limit = (block_size - sizeof(ext4_extent_header_t)) /
870 sizeof(ext4_extent_t);
871 ext4_extent_header_set_max_entries_count(path_ptr->header, limit);
872
873 // Modify root (in inode)
874 path->depth = 1;
875 path->extent = NULL;
876 path->index = EXT4_EXTENT_FIRST_INDEX(path->header);
877
878 ext4_extent_header_set_depth(path->header, path_ptr->depth + 1);
879 ext4_extent_header_set_entries_count(path->header, 1);
880
881 ext4_extent_index_set_first_block(path->index, 0);
882 ext4_extent_index_set_leaf(path->index, new_fblock);
883
884 path_ptr->block->dirty = true;
885 path->block->dirty = true;
886
887 *last_path_item = path_ptr;
888
889 return EOK;
890 }
891
892 return EOK;
893}
894
895int ext4_extent_append_block(ext4_inode_ref_t *inode_ref,
896 uint32_t *iblock, uint32_t *fblock)
897{
898 EXT4FS_DBG("");
899 int rc = EOK;
900
901 ext4_superblock_t *sb = inode_ref->fs->superblock;
902 uint64_t inode_size = ext4_inode_get_size(sb, inode_ref->inode);
903 uint32_t block_size = ext4_superblock_get_block_size(sb);
904
905 // Calculate number of new logical block
906 uint32_t new_block_idx = 0;
907 if (inode_size > 0) {
908 if ((inode_size % block_size) != 0) {
909 inode_size += block_size - (inode_size % block_size);
910 }
911 new_block_idx = inode_size / block_size;
912 }
913
914 // Load the nearest leaf (with extent)
915 ext4_extent_path_t *path;
916 rc = ext4_extent_find_extent(inode_ref, new_block_idx, &path);
917 if (rc != EOK) {
918 return rc;
919 }
920
921 // Jump to last item of the path (extent)
922 ext4_extent_path_t *path_ptr = path;
923 while (path_ptr->depth != 0) {
924 path_ptr++;
925 }
926
927 // Add new extent to the node
928 if (path_ptr->extent == NULL) {
929 goto append_extent;
930 }
931
932 uint16_t block_count = ext4_extent_get_block_count(path_ptr->extent);
933 uint16_t block_limit = (1 << 15);
934
935 uint32_t phys_block = 0;
936 if (block_count < block_limit) {
937
938 if (block_count == 0) {
939
940 rc = ext4_balloc_alloc_block(inode_ref, &phys_block);
941 if (rc != EOK) {
942 goto finish;
943 }
944
945 ext4_extent_set_first_block(path_ptr->extent, new_block_idx);
946 ext4_extent_set_start(path_ptr->extent, phys_block);
947 ext4_extent_set_block_count(path_ptr->extent, 1);
948
949 ext4_inode_set_size(inode_ref->inode, inode_size + block_size);
950 inode_ref->dirty = true;
951
952 path_ptr->block->dirty = true;
953
954 goto finish;
955 } else {
956
957 phys_block = ext4_extent_get_start(path_ptr->extent);
958 phys_block += ext4_extent_get_block_count(path_ptr->extent);
959
960 bool free;
961 rc = ext4_balloc_try_alloc_block(inode_ref, phys_block, &free);
962 if (rc != EOK) {
963 goto finish;
964 }
965
966 if (! free) {
967 // target is not free
968 goto append_extent;
969 }
970
971
972 ext4_extent_set_block_count(path_ptr->extent, block_count + 1);
973
974 ext4_inode_set_size(inode_ref->inode, inode_size + block_size);
975 inode_ref->dirty = true;
976
977 path_ptr->block->dirty = true;
978
979 goto finish;
980 }
981 }
982
983append_extent:
984
985 phys_block = 0;
986 // Allocate and insert insert new block
987 rc = ext4_balloc_alloc_block(inode_ref, &phys_block);
988 if (rc != EOK) {
989 EXT4FS_DBG("error in block allocation, rc = \%d", rc);
990 goto finish;
991 }
992
993 rc = ext4_extent_append_extent(inode_ref, path, &path_ptr, new_block_idx);
994 if (rc != EOK) {
995 ext4_balloc_free_block(inode_ref, phys_block);
996 goto finish;
997 }
998
999 ext4_extent_set_block_count(path_ptr->extent, 1);
1000 ext4_extent_set_first_block(path_ptr->extent, new_block_idx);
1001 ext4_extent_set_start(path_ptr->extent, phys_block);
1002
1003 ext4_inode_set_size(inode_ref->inode, inode_size + block_size);
1004 inode_ref->dirty = true;
1005
1006 path_ptr->block->dirty = true;
1007
1008finish:
1009
1010 *iblock = new_block_idx;
1011 *fblock = phys_block;
1012
1013 // Put loaded blocks
1014 // From 1 -> 0 is a block with inode data
1015 for (uint16_t i = 1; i <= path->depth; ++i) {
1016 if (path[i].block) {
1017 block_put(path[i].block);
1018 }
1019 }
1020
1021 // Destroy temporary data structure
1022 free(path);
1023
1024 return rc;
1025}
1026
1027/**
1028 * @}
1029 */
Note: See TracBrowser for help on using the repository browser.