Merge pull request #3 from ngkaho1234/master
[lwext4.git] / lwext4 / ext4.c
1 /*
2  * Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)
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 lwext4
30  * @{
31  */
32 /**
33  * @file  ext4.h
34  * @brief Ext4 high level operations (file, directory, mountpoints...)
35  */
36
37 #include "ext4_config.h"
38 #include "ext4_blockdev.h"
39 #include "ext4_types.h"
40 #include "ext4_debug.h"
41 #include "ext4_errno.h"
42 #include "ext4_fs.h"
43 #include "ext4_dir.h"
44 #include "ext4_inode.h"
45 #include "ext4_super.h"
46 #include "ext4_dir_idx.h"
47 #include "ext4.h"
48
49 #include <stdlib.h>
50 #include <string.h>
51
52 /**@brief   Mount point OS dependent lock*/
53 #define EXT4_MP_LOCK(_m)                                                       \
54         do {                                                                   \
55                 if ((_m)->os_locks)                                            \
56                         (_m)->os_locks->lock();                                \
57         } while (0)
58
59 /**@brief   Mount point OS dependent unlock*/
60 #define EXT4_MP_UNLOCK(_m)                                                     \
61         do {                                                                   \
62                 if ((_m)->os_locks)                                            \
63                         (_m)->os_locks->unlock();                              \
64         } while (0)
65
66 /**@brief   Mount point descriptor.*/
67 struct ext4_mountpoint {
68
69         /**@brief   Mount done flag.*/
70         bool mounted;
71
72         /**@brief   Mount point name (@ref ext4_mount)*/
73         char name[32];
74
75         /**@brief   OS dependent lock/unlock functions.*/
76         const struct ext4_lock *os_locks;
77
78         /**@brief   Ext4 filesystem internals.*/
79         struct ext4_fs fs;
80
81         /**@brief   Dynamic allocation cache flag.*/
82         bool cache_dynamic;
83 };
84
85 /**@brief   Block devices descriptor.*/
86 struct _ext4_devices {
87
88         /**@brief   Block device name (@ref ext4_device_register)*/
89         char name[32];
90
91         /**@brief   Block device handle.*/
92         struct ext4_blockdev *bd;
93
94         /**@brief   Block cache handle.*/
95         struct ext4_bcache *bc;
96 };
97
98 /**@brief   Block devices.*/
99 struct _ext4_devices _bdevices[CONFIG_EXT4_BLOCKDEVS_COUNT];
100
101 /**@brief   Mountpoints.*/
102 struct ext4_mountpoint _mp[CONFIG_EXT4_MOUNTPOINTS_COUNT];
103
104 int ext4_device_register(struct ext4_blockdev *bd, struct ext4_bcache *bc,
105                          const char *dev_name)
106 {
107         uint32_t i;
108         ext4_assert(bd && dev_name);
109
110         for (i = 0; i < CONFIG_EXT4_BLOCKDEVS_COUNT; ++i) {
111                 if (!_bdevices[i].bd) {
112                         strcpy(_bdevices[i].name, dev_name);
113                         _bdevices[i].bd = bd;
114                         _bdevices[i].bc = bc;
115                         return EOK;
116                 }
117
118                 if (!strcmp(_bdevices[i].name, dev_name))
119                         return EOK;
120         }
121         return ENOSPC;
122 }
123
124 /****************************************************************************/
125
126 static bool ext4_is_dots(const uint8_t *name, size_t name_size)
127 {
128         if ((name_size == 1) && (name[0] == '.'))
129                 return true;
130
131         if ((name_size == 2) && (name[0] == '.') && (name[1] == '.'))
132                 return true;
133
134         return false;
135 }
136
137 static int ext4_has_children(bool *has_children, struct ext4_inode_ref *enode)
138 {
139         struct ext4_fs *fs = enode->fs;
140
141         /* Check if node is directory */
142         if (!ext4_inode_is_type(&fs->sb, enode->inode,
143                                 EXT4_INODE_MODE_DIRECTORY)) {
144                 *has_children = false;
145                 return EOK;
146         }
147
148         struct ext4_directory_iterator it;
149         int rc = ext4_dir_iterator_init(&it, enode, 0);
150         if (rc != EOK)
151                 return rc;
152
153         /* Find a non-empty directory entry */
154         bool found = false;
155         while (it.current != NULL) {
156                 if (ext4_dir_entry_ll_get_inode(it.current) != 0) {
157                         uint16_t name_size = ext4_dir_entry_ll_get_name_length(
158                             &fs->sb, it.current);
159                         if (!ext4_is_dots(it.current->name, name_size)) {
160                                 found = true;
161                                 break;
162                         }
163                 }
164
165                 rc = ext4_dir_iterator_next(&it);
166                 if (rc != EOK) {
167                         ext4_dir_iterator_fini(&it);
168                         return rc;
169                 }
170         }
171
172         rc = ext4_dir_iterator_fini(&it);
173         if (rc != EOK)
174                 return rc;
175
176         *has_children = found;
177
178         return EOK;
179 }
180
181 static int ext4_link(struct ext4_mountpoint *mp, struct ext4_inode_ref *parent,
182                      struct ext4_inode_ref *child, const char *name,
183                      uint32_t name_len)
184 {
185         /* Check maximum name length */
186         if (name_len > EXT4_DIRECTORY_FILENAME_LEN)
187                 return EINVAL;
188
189         /* Add entry to parent directory */
190         int rc = ext4_dir_add_entry(parent, name, name_len, child);
191         if (rc != EOK)
192                 return rc;
193
194         /* Fill new dir -> add '.' and '..' entries.
195          * Also newly allocated inode should have 0 link count.
196          */
197         if (ext4_inode_is_type(&mp->fs.sb, child->inode,
198                                EXT4_INODE_MODE_DIRECTORY) &&
199             ext4_inode_get_links_count(child->inode) == 0) {
200                 rc = ext4_dir_add_entry(child, ".", strlen("."), child);
201                 if (rc != EOK) {
202                         ext4_dir_remove_entry(parent, name, strlen(name));
203                         return rc;
204                 }
205
206                 rc = ext4_dir_add_entry(child, "..", strlen(".."), parent);
207                 if (rc != EOK) {
208                         ext4_dir_remove_entry(parent, name, strlen(name));
209                         ext4_dir_remove_entry(child, ".", strlen("."));
210                         return rc;
211                 }
212
213                 /*New empty directory. Two links (. and ..) */
214                 ext4_inode_set_links_count(child->inode, 2);
215
216 #if CONFIG_DIR_INDEX_ENABLE
217                 /* Initialize directory index if supported */
218                 if (ext4_sb_has_feature_compatible(
219                         &mp->fs.sb, EXT4_FEATURE_COMPAT_DIR_INDEX)) {
220                         rc = ext4_dir_dx_init(child);
221                         if (rc != EOK)
222                                 return rc;
223
224                         ext4_inode_set_flag(child->inode,
225                                             EXT4_INODE_FLAG_INDEX);
226                         child->dirty = true;
227                 }
228 #endif
229
230                 ext4_fs_inode_links_count_inc(parent);
231                 child->dirty = true;
232                 parent->dirty = true;
233         } else {
234                 if (ext4_inode_is_type(&mp->fs.sb, child->inode,
235                                         EXT4_INODE_MODE_DIRECTORY)) {
236                         int has_flag_index =
237                                 ext4_inode_has_flag(child->inode,
238                                                     EXT4_INODE_FLAG_INDEX);
239                         struct ext4_directory_search_result result;
240                         if (!has_flag_index) {
241                                 rc = ext4_dir_find_entry(&result,
242                                                          child, "..",
243                                                          strlen(".."));
244                                 if (rc != EOK)
245                                         return EIO;
246
247                                 ext4_dir_entry_ll_set_inode(result.dentry,
248                                                             parent->index);
249                                 result.block.dirty = true;
250                                 rc = ext4_dir_destroy_result(child, &result);
251                                 if (rc != EOK)
252                                         return rc;
253
254                         } else {
255 #if CONFIG_DIR_INDEX_ENABLE
256                                 rc = ext4_dir_dx_reset_parent_inode(parent,
257                                                 parent->index);
258                                 if (rc != EOK)
259                                         return rc;
260
261 #endif
262                         }
263
264                         ext4_fs_inode_links_count_inc(parent);
265                         parent->dirty = true;
266                 } else {
267                         ext4_fs_inode_links_count_inc(child);
268                         child->dirty = true;
269                 }
270         }
271
272         return rc;
273 }
274
275 static int ext4_unlink(struct ext4_mountpoint *mp,
276                        struct ext4_inode_ref *parent,
277                        struct ext4_inode_ref *child_inode_ref, const char *name,
278                        uint32_t name_len)
279 {
280         bool has_children;
281         int rc = ext4_has_children(&has_children, child_inode_ref);
282         if (rc != EOK)
283                 return rc;
284
285         /* Cannot unlink non-empty node */
286         if (has_children)
287                 return ENOTSUP;
288
289         /* Remove entry from parent directory */
290         rc = ext4_dir_remove_entry(parent, name, name_len);
291         if (rc != EOK)
292                 return rc;
293
294         bool is_dir = ext4_inode_is_type(&mp->fs.sb, child_inode_ref->inode,
295                                          EXT4_INODE_MODE_DIRECTORY);
296
297         /* If directory - handle links from parent */
298         if (is_dir) {
299                 // ext4_assert(ext4_inode_get_links_count(child_inode_ref->inode)
300                 // == 1);
301                 ext4_fs_inode_links_count_dec(parent);
302                 parent->dirty = true;
303         }
304
305         /*
306          * TODO: Update timestamps of the parent
307          * (when we have wall-clock time).
308          *
309          * ext4_inode_set_change_inode_time(parent->inode, (uint32_t) now);
310          * ext4_inode_set_modification_time(parent->inode, (uint32_t) now);
311          * parent->dirty = true;
312          */
313
314         /*
315          * TODO: Update timestamp for inode.
316          *
317          * ext4_inode_set_change_inode_time(child_inode_ref->inode,
318          *     (uint32_t) now);
319          */
320         if (ext4_inode_get_links_count(child_inode_ref->inode)) {
321                 ext4_fs_inode_links_count_dec(child_inode_ref);
322                 child_inode_ref->dirty = true;
323         }
324
325         return EOK;
326 }
327
328 /****************************************************************************/
329
330 int ext4_mount(const char *dev_name, const char *mount_point)
331 {
332         ext4_assert(mount_point && dev_name);
333         int r;
334         int i;
335
336         uint32_t bsize;
337         struct ext4_blockdev *bd = 0;
338         struct ext4_bcache *bc = 0;
339         struct ext4_mountpoint *mp = 0;
340
341         if (mount_point[strlen(mount_point) - 1] != '/')
342                 return ENOTSUP;
343
344         for (i = 0; i < CONFIG_EXT4_BLOCKDEVS_COUNT; ++i) {
345                 if (_bdevices[i].name) {
346                         if (!strcmp(dev_name, _bdevices[i].name)) {
347                                 bd = _bdevices[i].bd;
348                                 bc = _bdevices[i].bc;
349                                 break;
350                         }
351                 }
352         }
353
354         if (!bd)
355                 return ENODEV;
356
357         for (i = 0; i < CONFIG_EXT4_MOUNTPOINTS_COUNT; ++i) {
358                 if (!_mp[i].mounted) {
359                         strcpy(_mp[i].name, mount_point);
360                         _mp[i].mounted = 1;
361                         mp = &_mp[i];
362                         break;
363                 }
364
365                 if (!strcmp(_mp[i].name, mount_point))
366                         return EOK;
367         }
368
369         if (!mp)
370                 return ENOMEM;
371
372         r = ext4_block_init(bd);
373         if (r != EOK)
374                 return r;
375
376         r = ext4_fs_init(&mp->fs, bd);
377         if (r != EOK) {
378                 ext4_block_fini(bd);
379                 return r;
380         }
381
382         bsize = ext4_sb_get_block_size(&mp->fs.sb);
383         ext4_block_set_lb_size(bd, bsize);
384
385         mp->cache_dynamic = 0;
386
387         if (!bc) {
388                 /*Automatic block cache alloc.*/
389                 mp->cache_dynamic = 1;
390                 bc = malloc(sizeof(struct ext4_bcache));
391
392                 r = ext4_bcache_init_dynamic(bc, CONFIG_BLOCK_DEV_CACHE_SIZE,
393                                              bsize);
394                 if (r != EOK) {
395                         free(bc);
396                         ext4_block_fini(bd);
397                         return r;
398                 }
399         }
400
401         if (bsize != bc->itemsize)
402                 return ENOTSUP;
403
404         /*Bind block cache to block device*/
405         r = ext4_block_bind_bcache(bd, bc);
406         if (r != EOK) {
407                 ext4_block_fini(bd);
408                 if (mp->cache_dynamic) {
409                         ext4_bcache_fini_dynamic(bc);
410                         free(bc);
411                 }
412                 return r;
413         }
414
415         return r;
416 }
417
418 int ext4_umount(const char *mount_point)
419 {
420         int i;
421         int r;
422         struct ext4_mountpoint *mp = 0;
423
424         for (i = 0; i < CONFIG_EXT4_MOUNTPOINTS_COUNT; ++i) {
425                 if (!strcmp(_mp[i].name, mount_point)) {
426                         mp = &_mp[i];
427                         break;
428                 }
429         }
430
431         if (!mp)
432                 return ENODEV;
433
434         r = ext4_fs_fini(&mp->fs);
435         if (r != EOK)
436                 return r;
437
438         mp->mounted = 0;
439
440         if (mp->cache_dynamic) {
441                 ext4_bcache_fini_dynamic(mp->fs.bdev->bc);
442                 free(mp->fs.bdev->bc);
443         }
444
445         return ext4_block_fini(mp->fs.bdev);
446 }
447
448 int ext4_mount_point_stats(const char *mount_point,
449                            struct ext4_mount_stats *stats)
450 {
451         uint32_t i;
452         struct ext4_mountpoint *mp = 0;
453
454         for (i = 0; i < CONFIG_EXT4_MOUNTPOINTS_COUNT; ++i) {
455                 if (!strcmp(_mp[i].name, mount_point)) {
456                         mp = &_mp[i];
457                         break;
458                 }
459         }
460         if (!mp)
461                 return ENOENT;
462
463         EXT4_MP_LOCK(mp);
464         stats->inodes_count = ext4_get32(&mp->fs.sb, inodes_count);
465         stats->free_inodes_count = ext4_get32(&mp->fs.sb, free_inodes_count);
466         stats->blocks_count = ext4_sb_get_blocks_cnt(&mp->fs.sb);
467         stats->free_blocks_count = ext4_sb_get_free_blocks_cnt(&mp->fs.sb);
468         stats->block_size = ext4_sb_get_block_size(&mp->fs.sb);
469
470         stats->block_group_count = ext4_block_group_cnt(&mp->fs.sb);
471         stats->blocks_per_group = ext4_get32(&mp->fs.sb, blocks_per_group);
472         stats->inodes_per_group = ext4_get32(&mp->fs.sb, inodes_per_group);
473
474         memcpy(stats->volume_name, mp->fs.sb.volume_name, 16);
475         EXT4_MP_UNLOCK(mp);
476
477         return EOK;
478 }
479
480 int ext4_mount_setup_locks(const char *mount_point,
481                            const struct ext4_lock *locks)
482 {
483         uint32_t i;
484         struct ext4_mountpoint *mp = 0;
485
486         for (i = 0; i < CONFIG_EXT4_MOUNTPOINTS_COUNT; ++i) {
487                 if (!strcmp(_mp[i].name, mount_point)) {
488                         mp = &_mp[i];
489                         break;
490                 }
491         }
492         if (!mp)
493                 return ENOENT;
494
495         mp->os_locks = locks;
496         return EOK;
497 }
498
499 /********************************FILE OPERATIONS*****************************/
500
501 static struct ext4_mountpoint *ext4_get_mount(const char *path)
502 {
503         int i;
504         for (i = 0; i < CONFIG_EXT4_MOUNTPOINTS_COUNT; ++i) {
505
506                 if (!_mp[i].mounted)
507                         continue;
508
509                 if (!strncmp(_mp[i].name, path, strlen(_mp[i].name)))
510                         return &_mp[i];
511         }
512         return 0;
513 }
514
515 static int ext4_path_check(const char *path, bool *is_goal)
516 {
517         int i;
518
519         for (i = 0; i < EXT4_DIRECTORY_FILENAME_LEN; ++i) {
520
521                 if (path[i] == '/') {
522                         *is_goal = false;
523                         return i;
524                 }
525
526                 if (path[i] == 0) {
527                         *is_goal = true;
528                         return i;
529                 }
530         }
531
532         return 0;
533 }
534
535 static bool ext4_parse_flags(const char *flags, uint32_t *file_flags)
536 {
537         if (!flags)
538                 return false;
539
540         if (!strcmp(flags, "r") || !strcmp(flags, "rb")) {
541                 *file_flags = O_RDONLY;
542                 return true;
543         }
544
545         if (!strcmp(flags, "w") || !strcmp(flags, "wb")) {
546                 *file_flags = O_WRONLY | O_CREAT | O_TRUNC;
547                 return true;
548         }
549
550         if (!strcmp(flags, "a") || !strcmp(flags, "ab")) {
551                 *file_flags = O_WRONLY | O_CREAT | O_APPEND;
552                 return true;
553         }
554
555         if (!strcmp(flags, "r+") || !strcmp(flags, "rb+") ||
556             !strcmp(flags, "r+b")) {
557                 *file_flags = O_RDWR;
558                 return true;
559         }
560
561         if (!strcmp(flags, "w+") || !strcmp(flags, "wb+") ||
562             !strcmp(flags, "w+b")) {
563                 *file_flags = O_RDWR | O_CREAT | O_TRUNC;
564                 return true;
565         }
566
567         if (!strcmp(flags, "a+") || !strcmp(flags, "ab+") ||
568             !strcmp(flags, "a+b")) {
569                 *file_flags = O_RDWR | O_CREAT | O_APPEND;
570                 return true;
571         }
572
573         return false;
574 }
575
576 /*
577  * NOTICE: if filetype is equal to EXT4_DIRECTORY_FILETYPE_UNKNOWN,
578  * any filetype of the target dir entry will be accepted.
579  */
580 static int ext4_generic_open2(ext4_file *f, const char *path, int flags,
581                               int filetype, uint32_t *parent_inode,
582                               uint32_t *name_off)
583 {
584         bool is_goal = false;
585         uint8_t inode_type = EXT4_DIRECTORY_FILETYPE_DIR;
586         uint32_t next_inode;
587
588         int r;
589         struct ext4_mountpoint *mp = ext4_get_mount(path);
590         struct ext4_directory_search_result result;
591         struct ext4_inode_ref ref;
592
593         f->mp = 0;
594
595         if (!mp)
596                 return ENOENT;
597
598         f->flags = flags;
599
600         /*Skip mount point*/
601         path += strlen(mp->name);
602
603         if (name_off)
604                 *name_off = strlen(mp->name);
605
606         /*Load root*/
607         r = ext4_fs_get_inode_ref(&mp->fs, EXT4_INODE_ROOT_INDEX, &ref);
608
609         if (r != EOK)
610                 return r;
611
612         if (parent_inode)
613                 *parent_inode = ref.index;
614
615         int len = ext4_path_check(path, &is_goal);
616
617         while (1) {
618
619                 len = ext4_path_check(path, &is_goal);
620
621                 if (!len) {
622                         /*If root open was request.*/
623                         if (is_goal &&
624                             ((filetype == EXT4_DIRECTORY_FILETYPE_DIR) ||
625                              (filetype == EXT4_DIRECTORY_FILETYPE_UNKNOWN)))
626                                 break;
627
628                         r = ENOENT;
629                         break;
630                 }
631
632                 r = ext4_dir_find_entry(&result, &ref, path, len);
633                 if (r != EOK) {
634
635                         if (r != ENOENT)
636                                 break;
637
638                         if (!(f->flags & O_CREAT))
639                                 break;
640
641                         /*O_CREAT allows create new entry*/
642                         struct ext4_inode_ref child_ref;
643                         r = ext4_fs_alloc_inode(
644                             &mp->fs, &child_ref,
645                             is_goal ? (filetype == EXT4_DIRECTORY_FILETYPE_DIR)
646                                     : true);
647                         if (r != EOK)
648                                 break;
649
650                         /*Destroy last result*/
651                         ext4_dir_destroy_result(&ref, &result);
652
653                         /*Link with root dir.*/
654                         r = ext4_link(mp, &ref, &child_ref, path, len);
655                         if (r != EOK) {
656                                 /*Fail. Free new inode.*/
657                                 ext4_fs_free_inode(&child_ref);
658                                 /*We do not want to write new inode.
659                                   But block has to be released.*/
660                                 child_ref.dirty = false;
661                                 ext4_fs_put_inode_ref(&child_ref);
662                                 break;
663                         }
664
665                         ext4_fs_put_inode_ref(&child_ref);
666
667                         continue;
668                 }
669
670                 if (parent_inode)
671                         *parent_inode = ref.index;
672
673                 next_inode = ext4_dir_entry_ll_get_inode(result.dentry);
674                 inode_type =
675                     ext4_dir_entry_ll_get_inode_type(&mp->fs.sb, result.dentry);
676
677                 r = ext4_dir_destroy_result(&ref, &result);
678                 if (r != EOK)
679                         break;
680
681                 /*If expected file error*/
682                 if (inode_type != EXT4_DIRECTORY_FILETYPE_DIR && !is_goal) {
683                         r = ENOENT;
684                         break;
685                 }
686                 if (filetype != EXT4_DIRECTORY_FILETYPE_UNKNOWN) {
687                         if ((inode_type != filetype) && is_goal) {
688                                 r = ENOENT;
689                                 break;
690                         }
691                 }
692
693                 r = ext4_fs_put_inode_ref(&ref);
694                 if (r != EOK)
695                         break;
696
697                 r = ext4_fs_get_inode_ref(&mp->fs, next_inode, &ref);
698                 if (r != EOK)
699                         break;
700
701                 if (is_goal)
702                         break;
703
704                 path += len + 1;
705
706                 if (name_off)
707                         *name_off += len + 1;
708         };
709
710         if (r != EOK) {
711                 ext4_fs_put_inode_ref(&ref);
712                 return r;
713         }
714
715         if (is_goal) {
716
717                 if ((f->flags & O_TRUNC) &&
718                     (inode_type == EXT4_DIRECTORY_FILETYPE_REG_FILE)) {
719
720                         r = ext4_fs_truncate_inode(&ref, 0);
721                         if (r != EOK) {
722                                 ext4_fs_put_inode_ref(&ref);
723                                 return r;
724                         }
725                 }
726
727                 f->mp = mp;
728                 f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
729                 f->inode = ref.index;
730                 f->fpos = 0;
731
732                 if (f->flags & O_APPEND)
733                         f->fpos = f->fsize;
734         }
735
736         r = ext4_fs_put_inode_ref(&ref);
737         return r;
738 }
739
740 /****************************************************************************/
741
742 static int ext4_generic_open(ext4_file *f, const char *path, const char *flags,
743                              bool file_expect, uint32_t *parent_inode,
744                              uint32_t *name_off)
745 {
746         uint32_t iflags;
747         int filetype;
748         if (ext4_parse_flags(flags, &iflags) == false)
749                 return EINVAL;
750
751         if (file_expect == true)
752                 filetype = EXT4_DIRECTORY_FILETYPE_REG_FILE;
753         else
754                 filetype = EXT4_DIRECTORY_FILETYPE_DIR;
755
756         return ext4_generic_open2(f, path, iflags, filetype, parent_inode,
757                                   name_off);
758 }
759
760 static int __ext4_create_hardlink(const char *path,
761                 struct ext4_inode_ref *child_ref)
762 {
763         bool is_goal = false;
764         uint8_t inode_type = EXT4_DIRECTORY_FILETYPE_DIR;
765         uint32_t next_inode;
766
767         int r;
768         struct ext4_mountpoint *mp = ext4_get_mount(path);
769         struct ext4_directory_search_result result;
770         struct ext4_inode_ref ref;
771
772         if (!mp)
773                 return ENOENT;
774
775         /*Skip mount point*/
776         path += strlen(mp->name);
777
778         /*Load root*/
779         r = ext4_fs_get_inode_ref(&mp->fs, EXT4_INODE_ROOT_INDEX, &ref);
780
781         if (r != EOK)
782                 return r;
783
784         int len = ext4_path_check(path, &is_goal);
785
786         while (1) {
787
788                 len = ext4_path_check(path, &is_goal);
789
790                 if (!len) {
791                         /*If root open was request.*/
792                         if (is_goal)
793                                 r = EINVAL;
794                         else
795                                 r = ENOENT;
796                         break;
797                 }
798
799                 r = ext4_dir_find_entry(&result, &ref, path, len);
800                 if (r != EOK) {
801
802                         if (r != ENOENT || !is_goal)
803                                 break;
804
805                         /*Destroy last result*/
806                         ext4_dir_destroy_result(&ref, &result);
807
808                         /*Link with root dir.*/
809                         r = ext4_link(mp, &ref, child_ref, path, len);
810                         break;
811                 }
812
813                 next_inode = result.dentry->inode;
814                 inode_type =
815                         ext4_dir_entry_ll_get_inode_type(&mp->fs.sb, result.dentry);
816
817                 r = ext4_dir_destroy_result(&ref, &result);
818                 if (r != EOK)
819                         break;
820
821                 if (inode_type == EXT4_DIRECTORY_FILETYPE_REG_FILE) {
822                         if (is_goal)
823                                 r = EEXIST;
824                         else
825                                 r = ENOENT;
826
827                         break;
828                 }
829
830                 r = ext4_fs_put_inode_ref(&ref);
831                 if (r != EOK)
832                         break;
833
834                 r = ext4_fs_get_inode_ref(&mp->fs, next_inode, &ref);
835                 if (r != EOK)
836                         break;
837
838                 if (is_goal)
839                         break;
840
841                 path += len + 1;
842         };
843
844         if (r != EOK) {
845                 ext4_fs_put_inode_ref(&ref);
846                 return r;
847         }
848
849         r = ext4_fs_put_inode_ref(&ref);
850         return r;
851 }
852
853 static int __ext4_remove_hardlink(const char *path,
854                                   uint32_t name_off,
855                                   struct ext4_inode_ref *parent_ref,
856                                   struct ext4_inode_ref *child_ref)
857 {
858         bool is_goal;
859         int r;
860         int len;
861         struct ext4_mountpoint *mp = ext4_get_mount(path);
862
863         if (!mp)
864                 return ENOENT;
865
866         /*Set path*/
867         path += name_off;
868
869         len = ext4_path_check(path, &is_goal);
870
871         /*Unlink from parent*/
872         r = ext4_unlink(mp, parent_ref, child_ref, path, len);
873         if (r != EOK)
874                 goto Finish;
875
876 Finish:
877         if (r != EOK)
878                 ext4_fs_put_inode_ref(child_ref);
879
880         ext4_fs_put_inode_ref(parent_ref);
881         return r;
882 }
883
884 int ext4_flink(const char *path, const char *hardlink_path)
885 {
886         int r;
887         ext4_file f;
888         uint32_t name_off;
889         bool child_loaded = false;
890         uint32_t parent_inode, child_inode;
891         struct ext4_mountpoint *mp = ext4_get_mount(path);
892         struct ext4_mountpoint *target_mp = ext4_get_mount(hardlink_path);
893         struct ext4_inode_ref child_ref;
894
895         if (!mp)
896                 return ENOENT;
897
898         /* Will that happen? Anyway return EINVAL for such case. */
899         if (mp != target_mp)
900                 return EINVAL;
901
902         EXT4_MP_LOCK(mp);
903
904         r = ext4_generic_open2(&f, path, O_RDONLY,
905                                EXT4_DIRECTORY_FILETYPE_UNKNOWN,
906                                &parent_inode, &name_off);
907         if (r != EOK)
908                 goto Finish;
909
910         child_inode = f.inode;
911         ext4_fclose(&f);
912
913         /*We have file to unlink. Load it.*/
914         r = ext4_fs_get_inode_ref(&mp->fs, child_inode, &child_ref);
915         if (r != EOK)
916                 goto Finish;
917
918         child_loaded = true;
919
920         r = __ext4_create_hardlink(hardlink_path, &child_ref);
921
922 Finish:
923         if (child_loaded)
924                 ext4_fs_put_inode_ref(&child_ref);
925
926         EXT4_MP_UNLOCK(mp);
927         return r;
928
929 }
930
931 int ext4_frename(const char *path, const char *new_path)
932 {
933         int r;
934         ext4_file f;
935         uint32_t name_off;
936         bool parent_loaded = false, child_loaded = false;
937         uint32_t parent_inode, child_inode;
938         struct ext4_mountpoint *mp = ext4_get_mount(path);
939         struct ext4_inode_ref child_ref, parent_ref;
940
941         if (!mp)
942                 return ENOENT;
943
944         EXT4_MP_LOCK(mp);
945
946         r = ext4_generic_open2(&f, path, O_RDONLY,
947                         EXT4_DIRECTORY_FILETYPE_UNKNOWN,
948                         &parent_inode, &name_off);
949         if (r != EOK)
950                 goto Finish;
951
952         child_inode = f.inode;
953         ext4_fclose(&f);
954
955         /*Load parent*/
956         r = ext4_fs_get_inode_ref(&mp->fs, parent_inode, &parent_ref);
957         if (r != EOK)
958                 goto Finish;
959
960         parent_loaded = true;
961
962         /*We have file to unlink. Load it.*/
963         r = ext4_fs_get_inode_ref(&mp->fs, child_inode, &child_ref);
964         if (r != EOK)
965                 goto Finish;
966
967         child_loaded = true;
968
969         r = __ext4_create_hardlink(new_path, &child_ref);
970         if (r != EOK)
971                 goto Finish;
972
973         r = __ext4_remove_hardlink(path, name_off, &parent_ref, &child_ref);
974         if (r != EOK)
975                 goto Finish;
976
977 Finish:
978         if (parent_loaded)
979                 ext4_fs_put_inode_ref(&parent_ref);
980
981         if (child_loaded)
982                 ext4_fs_put_inode_ref(&child_ref);
983
984         EXT4_MP_UNLOCK(mp);
985         return r;
986
987 }
988
989 /****************************************************************************/
990
991 int ext4_get_sblock(const char *mount_point, struct ext4_sblock **sb)
992 {
993         struct ext4_mountpoint *mp = ext4_get_mount(mount_point);
994
995         if (!mp)
996                 return ENOENT;
997
998         *sb = &mp->fs.sb;
999         return EOK;
1000 }
1001
1002 int ext4_cache_write_back(const char *path, bool on)
1003 {
1004         struct ext4_mountpoint *mp = ext4_get_mount(path);
1005
1006         if (!mp)
1007                 return ENOENT;
1008
1009         EXT4_MP_LOCK(mp);
1010         ext4_block_cache_write_back(mp->fs.bdev, on);
1011         EXT4_MP_UNLOCK(mp);
1012         return EOK;
1013 }
1014
1015 int ext4_fremove(const char *path)
1016 {
1017         ext4_file f;
1018         uint32_t parent_inode;
1019         uint32_t name_off;
1020         bool is_goal;
1021         int r;
1022         int len;
1023         struct ext4_inode_ref child;
1024         struct ext4_inode_ref parent;
1025         struct ext4_mountpoint *mp = ext4_get_mount(path);
1026
1027         if (!mp)
1028                 return ENOENT;
1029
1030         EXT4_MP_LOCK(mp);
1031         r = ext4_generic_open(&f, path, "r", true, &parent_inode, &name_off);
1032         if (r != EOK) {
1033                 EXT4_MP_UNLOCK(mp);
1034                 return r;
1035         }
1036
1037         /*Load parent*/
1038         r = ext4_fs_get_inode_ref(&mp->fs, parent_inode, &parent);
1039         if (r != EOK) {
1040                 EXT4_MP_UNLOCK(mp);
1041                 return r;
1042         }
1043
1044         /*We have file to delete. Load it.*/
1045         r = ext4_fs_get_inode_ref(&mp->fs, f.inode, &child);
1046         if (r != EOK) {
1047                 ext4_fs_put_inode_ref(&parent);
1048                 EXT4_MP_UNLOCK(mp);
1049                 return r;
1050         }
1051
1052         /*Set path*/
1053         path += name_off;
1054
1055         len = ext4_path_check(path, &is_goal);
1056
1057         /*Unlink from parent*/
1058         r = ext4_unlink(mp, &parent, &child, path, len);
1059         if (r != EOK)
1060                 goto Finish;
1061
1062         /*Link count is zero, the inode should be freed. */
1063         if (!ext4_inode_get_links_count(child.inode)) {
1064                 ext4_inode_set_deletion_time(child.inode, 0xFFFFFFFF);
1065                 /*Turncate*/
1066                 ext4_block_cache_write_back(mp->fs.bdev, 1);
1067                 /*Truncate may be IO heavy. Do it writeback cache mode.*/
1068                 r = ext4_fs_truncate_inode(&child, 0);
1069                 ext4_block_cache_write_back(mp->fs.bdev, 0);
1070
1071                 if (r != EOK)
1072                         goto Finish;
1073
1074                 r = ext4_fs_free_inode(&child);
1075                 if (r != EOK)
1076                         goto Finish;
1077         }
1078
1079 Finish:
1080         ext4_fs_put_inode_ref(&child);
1081         ext4_fs_put_inode_ref(&parent);
1082         EXT4_MP_UNLOCK(mp);
1083         return r;
1084 }
1085
1086 int ext4_fill_raw_inode(const char *mount_point, uint32_t ino,
1087                         struct ext4_inode *inode)
1088 {
1089         int r;
1090         struct ext4_inode_ref inode_ref;
1091         struct ext4_mountpoint *mp = ext4_get_mount(mount_point);
1092
1093         if (!mp)
1094                 return ENOENT;
1095
1096         EXT4_MP_LOCK(mp);
1097
1098         /*Load parent*/
1099         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1100         if (r != EOK) {
1101                 EXT4_MP_UNLOCK(mp);
1102                 return r;
1103         }
1104
1105         memcpy(inode, inode_ref.inode, sizeof(struct ext4_inode));
1106
1107         ext4_fs_put_inode_ref(&inode_ref);
1108         EXT4_MP_UNLOCK(mp);
1109         return r;
1110 }
1111
1112 int ext4_fopen(ext4_file *f, const char *path, const char *flags)
1113 {
1114         struct ext4_mountpoint *mp = ext4_get_mount(path);
1115         int r;
1116
1117         if (!mp)
1118                 return ENOENT;
1119
1120         EXT4_MP_LOCK(mp);
1121         ext4_block_cache_write_back(mp->fs.bdev, 1);
1122         r = ext4_generic_open(f, path, flags, true, 0, 0);
1123         ext4_block_cache_write_back(mp->fs.bdev, 0);
1124         EXT4_MP_UNLOCK(mp);
1125         return r;
1126 }
1127
1128 int ext4_fopen2(ext4_file *f, const char *path, int flags, bool file_expect)
1129 {
1130         struct ext4_mountpoint *mp = ext4_get_mount(path);
1131         int r;
1132         int filetype;
1133
1134         if (!mp)
1135                 return ENOENT;
1136
1137         if (file_expect == true)
1138                 filetype = EXT4_DIRECTORY_FILETYPE_REG_FILE;
1139         else
1140                 filetype = EXT4_DIRECTORY_FILETYPE_DIR;
1141
1142         EXT4_MP_LOCK(mp);
1143         ext4_block_cache_write_back(mp->fs.bdev, 1);
1144         r = ext4_generic_open2(f, path, flags, filetype, 0, 0);
1145         ext4_block_cache_write_back(mp->fs.bdev, 0);
1146         EXT4_MP_UNLOCK(mp);
1147         return r;
1148 }
1149
1150 int ext4_fclose(ext4_file *f)
1151 {
1152         ext4_assert(f && f->mp);
1153
1154         f->mp = 0;
1155         f->flags = 0;
1156         f->inode = 0;
1157         f->fpos = f->fsize = 0;
1158
1159         return EOK;
1160 }
1161
1162 static int ext4_ftruncate_no_lock(ext4_file *f, uint64_t size)
1163 {
1164         struct ext4_inode_ref ref;
1165         int r;
1166
1167
1168         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1169         if (r != EOK) {
1170                 EXT4_MP_UNLOCK(f->mp);
1171                 return r;
1172         }
1173
1174         /*Sync file size*/
1175         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1176         if (f->fsize <= size) {
1177                 r = EOK;
1178                 goto Finish;
1179         }
1180
1181         if ((ext4_inode_get_mode(&f->mp->fs.sb, ref.inode) & EXT4_INODE_MODE_SOFTLINK)
1182                         == EXT4_INODE_MODE_SOFTLINK
1183                         && f->fsize < sizeof(ref.inode->blocks)
1184                         && !ext4_inode_get_blocks_count(&f->mp->fs.sb, ref.inode)) {
1185                 char *content = (char *)ref.inode->blocks;
1186                 memset(content + size, 0, sizeof(ref.inode->blocks) - size);
1187                 ext4_inode_set_size(ref.inode, size);
1188                 ref.dirty = true;
1189
1190                 f->fsize = size;
1191                 if (f->fpos > size)
1192                         f->fpos = size;
1193
1194                 r = EOK;
1195                 goto Finish;
1196         }
1197
1198         /*Start write back cache mode.*/
1199         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1200         if (r != EOK)
1201                 goto Finish;
1202
1203         r = ext4_fs_truncate_inode(&ref, size);
1204         if (r != EOK)
1205                 goto Finish;
1206
1207         f->fsize = size;
1208         if (f->fpos > size)
1209                 f->fpos = size;
1210
1211         /*Stop write back cache mode*/
1212         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1213
1214         if (r != EOK)
1215                 goto Finish;
1216
1217 Finish:
1218         ext4_fs_put_inode_ref(&ref);
1219         return r;
1220
1221 }
1222
1223 int ext4_ftruncate(ext4_file *f, uint64_t size)
1224 {
1225         int r;
1226         ext4_assert(f && f->mp);
1227
1228         if (f->flags & O_RDONLY)
1229                 return EPERM;
1230
1231         EXT4_MP_LOCK(f->mp);
1232
1233         r = ext4_ftruncate_no_lock(f, size);
1234
1235         EXT4_MP_UNLOCK(f->mp);
1236         return r;
1237 }
1238
1239 int ext4_fread(ext4_file *f, void *buf, size_t size, size_t *rcnt)
1240 {
1241         uint32_t u;
1242         uint32_t fblock;
1243         uint32_t fblock_start;
1244         uint32_t fblock_cnt;
1245         uint32_t sblock;
1246         uint32_t sblock_end;
1247         uint32_t block_size;
1248         uint8_t *u8_buf = buf;
1249         int r;
1250         struct ext4_block b;
1251         struct ext4_inode_ref ref;
1252
1253         ext4_assert(f && f->mp);
1254
1255         if (f->flags & O_WRONLY)
1256                 return EPERM;
1257
1258         if (!size)
1259                 return EOK;
1260
1261         EXT4_MP_LOCK(f->mp);
1262
1263         if (rcnt)
1264                 *rcnt = 0;
1265
1266         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1267         if (r != EOK) {
1268                 EXT4_MP_UNLOCK(f->mp);
1269                 return r;
1270         }
1271
1272         /*Sync file size*/
1273         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1274
1275         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1276         size = size > (f->fsize - f->fpos) ? (f->fsize - f->fpos) : size;
1277         sblock = (f->fpos) / block_size;
1278         sblock_end = (f->fpos + size) / block_size;
1279         u = (f->fpos) % block_size;
1280
1281         if (u) {
1282
1283                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1284
1285                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1286                 if (r != EOK)
1287                         goto Finish;
1288
1289                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1290                 if (r != EOK)
1291                         goto Finish;
1292
1293                 memcpy(u8_buf, b.data + u, ll);
1294
1295                 r = ext4_block_set(f->mp->fs.bdev, &b);
1296                 if (r != EOK)
1297                         goto Finish;
1298
1299                 u8_buf += ll;
1300                 size -= ll;
1301                 f->fpos += ll;
1302
1303                 if (rcnt)
1304                         *rcnt += ll;
1305
1306                 sblock++;
1307         }
1308
1309         fblock_start = 0;
1310         fblock_cnt = 0;
1311         while (size >= block_size) {
1312                 while (sblock < sblock_end) {
1313                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1314                                                                &fblock);
1315                         if (r != EOK)
1316                                 goto Finish;
1317
1318                         sblock++;
1319
1320                         if (!fblock_start) {
1321                                 fblock_start = fblock;
1322                         }
1323
1324                         if ((fblock_start + fblock_cnt) != fblock)
1325                                 break;
1326
1327                         fblock_cnt++;
1328                 }
1329
1330                 r = ext4_blocks_get_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1331                                            fblock_cnt);
1332                 if (r != EOK)
1333                         goto Finish;
1334
1335                 size -= block_size * fblock_cnt;
1336                 u8_buf += block_size * fblock_cnt;
1337                 f->fpos += block_size * fblock_cnt;
1338
1339                 if (rcnt)
1340                         *rcnt += block_size * fblock_cnt;
1341
1342                 fblock_start = fblock;
1343                 fblock_cnt = 1;
1344         }
1345
1346         if (size) {
1347                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1348                 if (r != EOK)
1349                         goto Finish;
1350
1351                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1352                 if (r != EOK)
1353                         goto Finish;
1354
1355                 memcpy(u8_buf, b.data, size);
1356
1357                 r = ext4_block_set(f->mp->fs.bdev, &b);
1358                 if (r != EOK)
1359                         goto Finish;
1360
1361                 f->fpos += size;
1362
1363                 if (rcnt)
1364                         *rcnt += size;
1365         }
1366
1367 Finish:
1368         ext4_fs_put_inode_ref(&ref);
1369         EXT4_MP_UNLOCK(f->mp);
1370         return r;
1371 }
1372
1373 int ext4_fwrite(ext4_file *f, const void *buf, size_t size, size_t *wcnt)
1374 {
1375         uint32_t u;
1376         uint32_t fblock;
1377
1378         uint32_t sblock;
1379         uint32_t sblock_end;
1380         uint32_t file_blocks;
1381         uint32_t block_size;
1382         uint32_t fblock_start;
1383         uint32_t fblock_cnt;
1384
1385         struct ext4_block b;
1386         struct ext4_inode_ref ref;
1387         const uint8_t *u8_buf = buf;
1388         int r;
1389
1390         ext4_assert(f && f->mp);
1391
1392         if (f->flags & O_RDONLY)
1393                 return EPERM;
1394
1395         if (!size)
1396                 return EOK;
1397
1398         EXT4_MP_LOCK(f->mp);
1399
1400         if (wcnt)
1401                 *wcnt = 0;
1402
1403         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1404         if (r != EOK) {
1405                 EXT4_MP_UNLOCK(f->mp);
1406                 return r;
1407         }
1408
1409         /*Sync file size*/
1410         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1411
1412         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1413
1414         sblock_end = (f->fpos + size) > f->fsize ? (f->fpos + size) : f->fsize;
1415         sblock_end /= block_size;
1416         file_blocks = (f->fsize / block_size);
1417
1418         if (f->fsize % block_size)
1419                 file_blocks++;
1420
1421         sblock = (f->fpos) / block_size;
1422
1423         u = (f->fpos) % block_size;
1424
1425         if (u) {
1426                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1427
1428                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1429                 if (r != EOK)
1430                         goto Finish;
1431
1432                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1433                 if (r != EOK)
1434                         goto Finish;
1435
1436                 memcpy(b.data + u, u8_buf, ll);
1437                 b.dirty = true;
1438
1439                 r = ext4_block_set(f->mp->fs.bdev, &b);
1440                 if (r != EOK)
1441                         goto Finish;
1442
1443                 u8_buf += ll;
1444                 size -= ll;
1445                 f->fpos += ll;
1446
1447                 if (wcnt)
1448                         *wcnt += ll;
1449
1450                 sblock++;
1451         }
1452
1453         /*Start write back cache mode.*/
1454         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1455         if (r != EOK)
1456                 goto Finish;
1457
1458         fblock_start = 0;
1459         fblock_cnt = 0;
1460         while (size >= block_size) {
1461
1462                 while (sblock < sblock_end) {
1463                         if (sblock < file_blocks) {
1464                                 r = ext4_fs_get_inode_data_block_index(
1465                                     &ref, sblock, &fblock);
1466                                 if (r != EOK)
1467                                         break;
1468                         } else {
1469                                 r = ext4_fs_append_inode_block(&ref, &fblock,
1470                                                                &sblock);
1471                                 if (r != EOK)
1472                                         break;
1473                         }
1474
1475                         sblock++;
1476
1477                         if (!fblock_start) {
1478                                 fblock_start = fblock;
1479                         }
1480
1481                         if ((fblock_start + fblock_cnt) != fblock)
1482                                 break;
1483
1484                         fblock_cnt++;
1485                 }
1486
1487                 r = ext4_blocks_set_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1488                                            fblock_cnt);
1489                 if (r != EOK)
1490                         break;
1491
1492                 size -= block_size * fblock_cnt;
1493                 u8_buf += block_size * fblock_cnt;
1494                 f->fpos += block_size * fblock_cnt;
1495
1496                 if (wcnt)
1497                         *wcnt += block_size * fblock_cnt;
1498
1499                 fblock_start = fblock;
1500                 fblock_cnt = 1;
1501         }
1502
1503         /*Stop write back cache mode*/
1504         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1505
1506         if (r != EOK)
1507                 goto Finish;
1508
1509         if (size) {
1510                 if (sblock < file_blocks) {
1511                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1512                                                                &fblock);
1513                         if (r != EOK)
1514                                 goto Finish;
1515                 } else {
1516                         r = ext4_fs_append_inode_block(&ref, &fblock, &sblock);
1517                         if (r != EOK)
1518                                 goto Finish;
1519                 }
1520
1521                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1522                 if (r != EOK)
1523                         goto Finish;
1524
1525                 memcpy(b.data, u8_buf, size);
1526                 b.dirty = true;
1527
1528                 r = ext4_block_set(f->mp->fs.bdev, &b);
1529                 if (r != EOK)
1530                         goto Finish;
1531
1532                 f->fpos += size;
1533
1534                 if (wcnt)
1535                         *wcnt += size;
1536         }
1537
1538         if (f->fpos > f->fsize) {
1539                 f->fsize = f->fpos;
1540                 ext4_inode_set_size(ref.inode, f->fsize);
1541                 ref.dirty = true;
1542         }
1543
1544 Finish:
1545         ext4_fs_put_inode_ref(&ref);
1546         EXT4_MP_UNLOCK(f->mp);
1547         return r;
1548 }
1549
1550 int ext4_fseek(ext4_file *f, uint64_t offset, uint32_t origin)
1551 {
1552         switch (origin) {
1553         case SEEK_SET:
1554                 if (offset > f->fsize)
1555                         return EINVAL;
1556
1557                 f->fpos = offset;
1558                 return EOK;
1559         case SEEK_CUR:
1560                 if ((offset + f->fpos) > f->fsize)
1561                         return EINVAL;
1562
1563                 f->fpos += offset;
1564                 return EOK;
1565         case SEEK_END:
1566                 if (offset > f->fsize)
1567                         return EINVAL;
1568
1569                 f->fpos = f->fsize - offset;
1570                 return EOK;
1571         }
1572         return EINVAL;
1573 }
1574
1575 uint64_t ext4_ftell(ext4_file *f) { return f->fpos; }
1576
1577 uint64_t ext4_fsize(ext4_file *f) { return f->fsize; }
1578
1579 int ext4_fchmod(ext4_file *f, uint32_t mode)
1580 {
1581         int r;
1582         uint32_t ino;
1583         struct ext4_sblock *sb;
1584         struct ext4_inode_ref inode_ref;
1585         struct ext4_mountpoint *mp = f->mp;
1586
1587         if (!mp)
1588                 return ENOENT;
1589
1590         EXT4_MP_LOCK(mp);
1591
1592         ino = f->inode;
1593         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1594         if (r != EOK) {
1595                 EXT4_MP_UNLOCK(mp);
1596                 return r;
1597         }
1598
1599         sb = &f->mp->fs.sb;
1600         ext4_inode_set_mode(sb, inode_ref.inode, mode);
1601         inode_ref.dirty = true;
1602
1603         ext4_fs_put_inode_ref(&inode_ref);
1604         EXT4_MP_UNLOCK(mp);
1605         return r;
1606 }
1607
1608 int ext4_fchown(ext4_file *f, uint32_t uid, uint32_t gid)
1609 {
1610         int r;
1611         uint32_t ino;
1612         struct ext4_inode_ref inode_ref;
1613         struct ext4_mountpoint *mp = f->mp;
1614
1615         if (!mp)
1616                 return ENOENT;
1617
1618         EXT4_MP_LOCK(mp);
1619
1620         ino = f->inode;
1621         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1622         if (r != EOK) {
1623                 EXT4_MP_UNLOCK(mp);
1624                 return r;
1625         }
1626
1627         ext4_inode_set_uid(inode_ref.inode, uid);
1628         ext4_inode_set_gid(inode_ref.inode, gid);
1629         inode_ref.dirty = true;
1630
1631         ext4_fs_put_inode_ref(&inode_ref);
1632         EXT4_MP_UNLOCK(mp);
1633         return r;
1634 }
1635
1636 int ext4_file_set_atime(ext4_file *f, uint32_t atime)
1637 {
1638         int r;
1639         uint32_t ino;
1640         struct ext4_inode_ref inode_ref;
1641         struct ext4_mountpoint *mp = f->mp;
1642
1643         if (!mp)
1644                 return ENOENT;
1645
1646         EXT4_MP_LOCK(mp);
1647
1648         ino = f->inode;
1649         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1650         if (r != EOK) {
1651                 EXT4_MP_UNLOCK(mp);
1652                 return r;
1653         }
1654
1655         ext4_inode_set_access_time(inode_ref.inode, atime);
1656         inode_ref.dirty = true;
1657
1658         ext4_fs_put_inode_ref(&inode_ref);
1659         EXT4_MP_UNLOCK(mp);
1660         return r;
1661 }
1662
1663 int ext4_file_set_mtime(ext4_file *f, uint32_t mtime)
1664 {
1665         int r;
1666         uint32_t ino;
1667         struct ext4_inode_ref inode_ref;
1668         struct ext4_mountpoint *mp = f->mp;
1669
1670         if (!mp)
1671                 return ENOENT;
1672
1673         EXT4_MP_LOCK(mp);
1674
1675         ino = f->inode;
1676         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1677         if (r != EOK) {
1678                 EXT4_MP_UNLOCK(mp);
1679                 return r;
1680         }
1681
1682         ext4_inode_set_modification_time(inode_ref.inode, mtime);
1683         inode_ref.dirty = true;
1684
1685         ext4_fs_put_inode_ref(&inode_ref);
1686         EXT4_MP_UNLOCK(mp);
1687         return r;
1688 }
1689
1690 int ext4_file_set_ctime(ext4_file *f, uint32_t ctime)
1691 {
1692         int r;
1693         uint32_t ino;
1694         struct ext4_inode_ref inode_ref;
1695         struct ext4_mountpoint *mp = f->mp;
1696
1697         if (!mp)
1698                 return ENOENT;
1699
1700         EXT4_MP_LOCK(mp);
1701
1702         ino = f->inode;
1703         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1704         if (r != EOK) {
1705                 EXT4_MP_UNLOCK(mp);
1706                 return r;
1707         }
1708
1709         ext4_inode_set_change_inode_time(inode_ref.inode, ctime);
1710         inode_ref.dirty = true;
1711
1712         ext4_fs_put_inode_ref(&inode_ref);
1713         EXT4_MP_UNLOCK(mp);
1714         return r;
1715 }
1716
1717 /*********************************DIRECTORY OPERATION************************/
1718
1719 int ext4_dir_rm(const char *path)
1720 {
1721         int r;
1722         int len;
1723         ext4_file f;
1724
1725         struct ext4_mountpoint *mp = ext4_get_mount(path);
1726         struct ext4_inode_ref current;
1727         struct ext4_inode_ref child;
1728         struct ext4_directory_iterator it;
1729
1730         uint32_t name_off;
1731         uint32_t inode_up;
1732         uint32_t inode_current;
1733         uint32_t depth = 1;
1734
1735         bool has_children;
1736         bool is_goal;
1737         bool dir_end;
1738
1739         if (!mp)
1740                 return ENOENT;
1741
1742         EXT4_MP_LOCK(mp);
1743
1744         /*Check if exist.*/
1745         r = ext4_generic_open(&f, path, "r", false, &inode_up, &name_off);
1746         if (r != EOK) {
1747                 EXT4_MP_UNLOCK(mp);
1748                 return r;
1749         }
1750
1751         path += name_off;
1752         len = ext4_path_check(path, &is_goal);
1753
1754         inode_current = f.inode;
1755         dir_end = false;
1756
1757         ext4_block_cache_write_back(mp->fs.bdev, 1);
1758
1759         do {
1760                 /*Load directory node.*/
1761                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_current, &current);
1762                 if (r != EOK) {
1763                         break;
1764                 }
1765
1766                 /*Initialize iterator.*/
1767                 r = ext4_dir_iterator_init(&it, &current, 0);
1768                 if (r != EOK) {
1769                         ext4_fs_put_inode_ref(&current);
1770                         break;
1771                 }
1772
1773                 while (r == EOK) {
1774
1775                         if (!it.current) {
1776                                 dir_end = true;
1777                                 break;
1778                         }
1779
1780                         /*Get up directory inode when ".." entry*/
1781                         if ((it.current->name_length == 2) &&
1782                             ext4_is_dots(it.current->name,
1783                                          it.current->name_length)) {
1784                                 inode_up = ext4_dir_entry_ll_get_inode(it.current);
1785                         }
1786
1787                         /*If directory or file entry,  but not "." ".." entry*/
1788                         if (!ext4_is_dots(it.current->name,
1789                                           it.current->name_length)) {
1790
1791                                 /*Get child inode reference do unlink
1792                                  * directory/file.*/
1793                                 r = ext4_fs_get_inode_ref(&f.mp->fs,
1794                                         ext4_dir_entry_ll_get_inode(it.current),
1795                                         &child);
1796                                 if (r != EOK)
1797                                         break;
1798
1799                                 /*If directory with no leaf children*/
1800                                 r = ext4_has_children(&has_children, &child);
1801                                 if (r != EOK) {
1802                                         ext4_fs_put_inode_ref(&child);
1803                                         break;
1804                                 }
1805
1806                                 if (has_children) {
1807                                         /*Has directory children. Go into this
1808                                          * directory.*/
1809                                         inode_up = inode_current;
1810                                         inode_current = ext4_dir_entry_ll_get_inode(it.current);
1811                                         depth++;
1812                                         ext4_fs_put_inode_ref(&child);
1813                                         break;
1814                                 }
1815
1816                                 /*No children in child directory or file. Just
1817                                  * unlink.*/
1818                                 r = ext4_unlink(f.mp, &current, &child,
1819                                                 (char *)it.current->name,
1820                                                 it.current->name_length);
1821                                 if (r != EOK) {
1822                                         ext4_fs_put_inode_ref(&child);
1823                                         break;
1824                                 }
1825
1826                                 ext4_inode_set_deletion_time(child.inode,
1827                                                              0xFFFFFFFF);
1828                                 ext4_inode_set_links_count(child.inode, 0);
1829                                 child.dirty = true;
1830                                 /*Turncate*/
1831                                 r = ext4_fs_truncate_inode(&child, 0);
1832                                 if (r != EOK) {
1833                                         ext4_fs_put_inode_ref(&child);
1834                                         break;
1835                                 }
1836
1837                                 r = ext4_fs_free_inode(&child);
1838                                 if (r != EOK) {
1839                                         ext4_fs_put_inode_ref(&child);
1840                                         break;
1841                                 }
1842
1843                                 r = ext4_fs_put_inode_ref(&child);
1844                                 if (r != EOK)
1845                                         break;
1846                         }
1847
1848                         r = ext4_dir_iterator_next(&it);
1849                 }
1850
1851                 if (dir_end) {
1852                         /*Directory iterator reached last entry*/
1853                         ext4_has_children(&has_children, &current);
1854                         if (!has_children) {
1855                                 inode_current = inode_up;
1856                                 if (depth)
1857                                         depth--;
1858                         }
1859                         /*Last unlink*/
1860                         if (!depth) {
1861                                 /*Load parent.*/
1862                                 struct ext4_inode_ref parent;
1863                                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_up,
1864                                                           &parent);
1865                                 if (r != EOK)
1866                                         goto End;
1867
1868                                 /* In this place all directories should be
1869                                  * unlinked.
1870                                  * Last unlink from root of current directory*/
1871                                 r = ext4_unlink(f.mp, &parent, &current,
1872                                                 (char *)path, len);
1873                                 if (r != EOK) {
1874                                         ext4_fs_put_inode_ref(&parent);
1875                                         goto End;
1876                                 }
1877
1878                                 if (ext4_inode_get_links_count(current.inode) ==
1879                                     2) {
1880                                         ext4_inode_set_deletion_time(
1881                                             current.inode, 0xFFFFFFFF);
1882                                         ext4_inode_set_links_count(
1883                                             current.inode, 0);
1884                                         current.dirty = true;
1885                                         /*Turncate*/
1886                                         r = ext4_fs_truncate_inode(&current, 0);
1887                                         if (r != EOK) {
1888                                                 ext4_fs_put_inode_ref(&parent);
1889                                                 goto End;
1890                                         }
1891
1892                                         r = ext4_fs_free_inode(&current);
1893                                         if (r != EOK) {
1894                                                 ext4_fs_put_inode_ref(&parent);
1895                                                 goto End;
1896                                         }
1897                                 }
1898
1899                                 r = ext4_fs_put_inode_ref(&parent);
1900                                 if (r != EOK)
1901                                         goto End;
1902                         }
1903                 }
1904
1905         End:
1906                 ext4_dir_iterator_fini(&it);
1907                 ext4_fs_put_inode_ref(&current);
1908                 dir_end = false;
1909
1910                 /*When something goes wrong. End loop.*/
1911                 if (r != EOK)
1912                         break;
1913
1914         } while (depth);
1915
1916         ext4_block_cache_write_back(mp->fs.bdev, 0);
1917         EXT4_MP_UNLOCK(mp);
1918         return r;
1919 }
1920
1921 int ext4_dir_mk(const char *path)
1922 {
1923         int r;
1924         ext4_file f;
1925
1926         struct ext4_mountpoint *mp = ext4_get_mount(path);
1927
1928         if (!mp)
1929                 return ENOENT;
1930
1931         EXT4_MP_LOCK(mp);
1932
1933         /*Check if exist.*/
1934         r = ext4_generic_open(&f, path, "r", false, 0, 0);
1935         if (r == EOK) {
1936                 /*Directory already created*/
1937                 EXT4_MP_UNLOCK(mp);
1938                 return r;
1939         }
1940
1941         /*Create new dir*/
1942         r = ext4_generic_open(&f, path, "w", false, 0, 0);
1943         if (r != EOK) {
1944                 EXT4_MP_UNLOCK(mp);
1945                 return r;
1946         }
1947
1948         EXT4_MP_UNLOCK(mp);
1949         return r;
1950 }
1951
1952 int ext4_dir_open(ext4_dir *d, const char *path)
1953 {
1954         struct ext4_mountpoint *mp = ext4_get_mount(path);
1955         int r;
1956
1957         if (!mp)
1958                 return ENOENT;
1959
1960         EXT4_MP_LOCK(mp);
1961         r = ext4_generic_open(&d->f, path, "r", false, 0, 0);
1962         d->next_off = 0;
1963         EXT4_MP_UNLOCK(mp);
1964         return r;
1965 }
1966
1967 int ext4_dir_close(ext4_dir *d) { return ext4_fclose(&d->f); }
1968
1969 const ext4_direntry *ext4_dir_entry_next(ext4_dir *d)
1970 {
1971 #define EXT4_DIR_ENTRY_OFFSET_TERM (uint64_t)(-1)
1972
1973         int r;
1974         ext4_direntry *de = 0;
1975         struct ext4_inode_ref dir;
1976         struct ext4_directory_iterator it;
1977
1978         EXT4_MP_LOCK(d->f.mp);
1979
1980         if (d->next_off == EXT4_DIR_ENTRY_OFFSET_TERM) {
1981                 EXT4_MP_UNLOCK(d->f.mp);
1982                 return 0;
1983         }
1984
1985         r = ext4_fs_get_inode_ref(&d->f.mp->fs, d->f.inode, &dir);
1986         if (r != EOK) {
1987                 goto Finish;
1988         }
1989
1990         r = ext4_dir_iterator_init(&it, &dir, d->next_off);
1991         if (r != EOK) {
1992                 ext4_fs_put_inode_ref(&dir);
1993                 goto Finish;
1994         }
1995
1996         memcpy(&d->de, it.current, sizeof(ext4_direntry));
1997         de = &d->de;
1998
1999         ext4_dir_iterator_next(&it);
2000
2001         d->next_off =
2002             it.current ? it.current_offset : EXT4_DIR_ENTRY_OFFSET_TERM;
2003
2004         ext4_dir_iterator_fini(&it);
2005         ext4_fs_put_inode_ref(&dir);
2006
2007 Finish:
2008         EXT4_MP_UNLOCK(d->f.mp);
2009         return de;
2010 }
2011
2012 /**
2013  * @}
2014  */