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(child,
 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_fopen_all(ext4_file *f, const char *path, int flags)
1224 {
1225         struct ext4_mountpoint *mp = ext4_get_mount(path);
1226         int r;
1227         int filetype;
1228
1229         if (!mp)
1230                 return ENOENT;
1231
1232         filetype = EXT4_DIRECTORY_FILETYPE_UNKNOWN;
1233
1234         EXT4_MP_LOCK(mp);
1235         ext4_block_cache_write_back(mp->fs.bdev, 1);
1236         r = ext4_generic_open2(f, path, flags, filetype, 0, 0);
1237         ext4_block_cache_write_back(mp->fs.bdev, 0);
1238         EXT4_MP_UNLOCK(mp);
1239         return r;
1240 }
1241
1242 int ext4_ftruncate(ext4_file *f, uint64_t size)
1243 {
1244         int r;
1245         ext4_assert(f && f->mp);
1246
1247         if (f->flags & O_RDONLY)
1248                 return EPERM;
1249
1250         EXT4_MP_LOCK(f->mp);
1251
1252         r = ext4_ftruncate_no_lock(f, size);
1253
1254         EXT4_MP_UNLOCK(f->mp);
1255         return r;
1256 }
1257
1258 int ext4_fread(ext4_file *f, void *buf, uint32_t size, uint32_t *rcnt)
1259 {
1260         uint32_t u;
1261         uint32_t fblock;
1262         uint32_t fblock_start;
1263         uint32_t fblock_cnt;
1264         uint32_t sblock;
1265         uint32_t sblock_end;
1266         uint32_t block_size;
1267         uint8_t *u8_buf = buf;
1268         int r;
1269         struct ext4_block b;
1270         struct ext4_inode_ref ref;
1271
1272         ext4_assert(f && f->mp);
1273
1274         if (f->flags & O_WRONLY)
1275                 return EPERM;
1276
1277         if (!size)
1278                 return EOK;
1279
1280         EXT4_MP_LOCK(f->mp);
1281
1282         if (rcnt)
1283                 *rcnt = 0;
1284
1285         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1286         if (r != EOK) {
1287                 EXT4_MP_UNLOCK(f->mp);
1288                 return r;
1289         }
1290
1291         /*Sync file size*/
1292         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1293
1294         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1295         size = size > (f->fsize - f->fpos) ? (f->fsize - f->fpos) : size;
1296         sblock = (f->fpos) / block_size;
1297         sblock_end = (f->fpos + size) / block_size;
1298         u = (f->fpos) % block_size;
1299
1300         if (u) {
1301
1302                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1303
1304                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1305                 if (r != EOK)
1306                         goto Finish;
1307
1308                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1309                 if (r != EOK)
1310                         goto Finish;
1311
1312                 memcpy(u8_buf, b.data + u, ll);
1313
1314                 r = ext4_block_set(f->mp->fs.bdev, &b);
1315                 if (r != EOK)
1316                         goto Finish;
1317
1318                 u8_buf += ll;
1319                 size -= ll;
1320                 f->fpos += ll;
1321
1322                 if (rcnt)
1323                         *rcnt += ll;
1324
1325                 sblock++;
1326         }
1327
1328         fblock_start = 0;
1329         fblock_cnt = 0;
1330         while (size >= block_size) {
1331                 while (sblock < sblock_end) {
1332                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1333                                                                &fblock);
1334                         if (r != EOK)
1335                                 goto Finish;
1336
1337                         sblock++;
1338
1339                         if (!fblock_start) {
1340                                 fblock_start = fblock;
1341                         }
1342
1343                         if ((fblock_start + fblock_cnt) != fblock)
1344                                 break;
1345
1346                         fblock_cnt++;
1347                 }
1348
1349                 r = ext4_blocks_get_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1350                                            fblock_cnt);
1351                 if (r != EOK)
1352                         goto Finish;
1353
1354                 size -= block_size * fblock_cnt;
1355                 u8_buf += block_size * fblock_cnt;
1356                 f->fpos += block_size * fblock_cnt;
1357
1358                 if (rcnt)
1359                         *rcnt += block_size * fblock_cnt;
1360
1361                 fblock_start = fblock;
1362                 fblock_cnt = 1;
1363         }
1364
1365         if (size) {
1366                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1367                 if (r != EOK)
1368                         goto Finish;
1369
1370                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1371                 if (r != EOK)
1372                         goto Finish;
1373
1374                 memcpy(u8_buf, b.data, size);
1375
1376                 r = ext4_block_set(f->mp->fs.bdev, &b);
1377                 if (r != EOK)
1378                         goto Finish;
1379
1380                 f->fpos += size;
1381
1382                 if (rcnt)
1383                         *rcnt += size;
1384         }
1385
1386 Finish:
1387         ext4_fs_put_inode_ref(&ref);
1388         EXT4_MP_UNLOCK(f->mp);
1389         return r;
1390 }
1391
1392 int ext4_fwrite(ext4_file *f, const void *buf, uint32_t size, uint32_t *wcnt)
1393 {
1394         uint32_t u;
1395         uint32_t fblock;
1396
1397         uint32_t sblock;
1398         uint32_t sblock_end;
1399         uint32_t file_blocks;
1400         uint32_t block_size;
1401         uint32_t fblock_start;
1402         uint32_t fblock_cnt;
1403
1404         struct ext4_block b;
1405         struct ext4_inode_ref ref;
1406         const uint8_t *u8_buf = buf;
1407         int r;
1408
1409         ext4_assert(f && f->mp);
1410
1411         if (f->flags & O_RDONLY)
1412                 return EPERM;
1413
1414         if (!size)
1415                 return EOK;
1416
1417         EXT4_MP_LOCK(f->mp);
1418
1419         if (wcnt)
1420                 *wcnt = 0;
1421
1422         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1423         if (r != EOK) {
1424                 EXT4_MP_UNLOCK(f->mp);
1425                 return r;
1426         }
1427
1428         /*Sync file size*/
1429         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1430
1431         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1432
1433         sblock_end = (f->fpos + size) > f->fsize ? (f->fpos + size) : f->fsize;
1434         sblock_end /= block_size;
1435         file_blocks = (f->fsize / block_size);
1436
1437         if (f->fsize % block_size)
1438                 file_blocks++;
1439
1440         sblock = (f->fpos) / block_size;
1441
1442         u = (f->fpos) % block_size;
1443
1444         if (u) {
1445                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1446
1447                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1448                 if (r != EOK)
1449                         goto Finish;
1450
1451                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1452                 if (r != EOK)
1453                         goto Finish;
1454
1455                 memcpy(b.data + u, u8_buf, ll);
1456                 b.dirty = true;
1457
1458                 r = ext4_block_set(f->mp->fs.bdev, &b);
1459                 if (r != EOK)
1460                         goto Finish;
1461
1462                 u8_buf += ll;
1463                 size -= ll;
1464                 f->fpos += ll;
1465
1466                 if (wcnt)
1467                         *wcnt += ll;
1468
1469                 sblock++;
1470         }
1471
1472         /*Start write back cache mode.*/
1473         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1474         if (r != EOK)
1475                 goto Finish;
1476
1477         fblock_start = 0;
1478         fblock_cnt = 0;
1479         while (size >= block_size) {
1480
1481                 while (sblock < sblock_end) {
1482                         if (sblock < file_blocks) {
1483                                 r = ext4_fs_get_inode_data_block_index(
1484                                     &ref, sblock, &fblock);
1485                                 if (r != EOK)
1486                                         break;
1487                         } else {
1488                                 r = ext4_fs_append_inode_block(&ref, &fblock,
1489                                                                &sblock);
1490                                 if (r != EOK)
1491                                         break;
1492                         }
1493
1494                         sblock++;
1495
1496                         if (!fblock_start) {
1497                                 fblock_start = fblock;
1498                         }
1499
1500                         if ((fblock_start + fblock_cnt) != fblock)
1501                                 break;
1502
1503                         fblock_cnt++;
1504                 }
1505
1506                 r = ext4_blocks_set_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1507                                            fblock_cnt);
1508                 if (r != EOK)
1509                         break;
1510
1511                 size -= block_size * fblock_cnt;
1512                 u8_buf += block_size * fblock_cnt;
1513                 f->fpos += block_size * fblock_cnt;
1514
1515                 if (wcnt)
1516                         *wcnt += block_size * fblock_cnt;
1517
1518                 fblock_start = fblock;
1519                 fblock_cnt = 1;
1520         }
1521
1522         /*Stop write back cache mode*/
1523         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1524
1525         if (r != EOK)
1526                 goto Finish;
1527
1528         if (size) {
1529                 if (sblock < file_blocks) {
1530                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1531                                                                &fblock);
1532                         if (r != EOK)
1533                                 goto Finish;
1534                 } else {
1535                         r = ext4_fs_append_inode_block(&ref, &fblock, &sblock);
1536                         if (r != EOK)
1537                                 goto Finish;
1538                 }
1539
1540                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1541                 if (r != EOK)
1542                         goto Finish;
1543
1544                 memcpy(b.data, u8_buf, size);
1545                 b.dirty = true;
1546
1547                 r = ext4_block_set(f->mp->fs.bdev, &b);
1548                 if (r != EOK)
1549                         goto Finish;
1550
1551                 f->fpos += size;
1552
1553                 if (wcnt)
1554                         *wcnt += size;
1555         }
1556
1557         if (f->fpos > f->fsize) {
1558                 f->fsize = f->fpos;
1559                 ext4_inode_set_size(ref.inode, f->fsize);
1560                 ref.dirty = true;
1561         }
1562
1563 Finish:
1564         ext4_fs_put_inode_ref(&ref);
1565         EXT4_MP_UNLOCK(f->mp);
1566         return r;
1567 }
1568
1569 int ext4_fseek(ext4_file *f, uint64_t offset, uint32_t origin)
1570 {
1571         switch (origin) {
1572         case SEEK_SET:
1573                 if (offset > f->fsize)
1574                         return EINVAL;
1575
1576                 f->fpos = offset;
1577                 return EOK;
1578         case SEEK_CUR:
1579                 if ((offset + f->fpos) > f->fsize)
1580                         return EINVAL;
1581
1582                 f->fpos += offset;
1583                 return EOK;
1584         case SEEK_END:
1585                 if (offset > f->fsize)
1586                         return EINVAL;
1587
1588                 f->fpos = f->fsize - offset;
1589                 return EOK;
1590         }
1591         return EINVAL;
1592 }
1593
1594 uint64_t ext4_ftell(ext4_file *f) { return f->fpos; }
1595
1596 uint64_t ext4_fsize(ext4_file *f) { return f->fsize; }
1597
1598 int ext4_fchmod(ext4_file *f, uint32_t mode)
1599 {
1600         int r;
1601         uint32_t ino;
1602         struct ext4_sblock *sb;
1603         struct ext4_inode_ref inode_ref;
1604         struct ext4_mountpoint *mp = f->mp;
1605
1606         if (!mp)
1607                 return ENOENT;
1608
1609         EXT4_MP_LOCK(mp);
1610
1611         ino = f->inode;
1612         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1613         if (r != EOK) {
1614                 EXT4_MP_UNLOCK(mp);
1615                 return r;
1616         }
1617
1618         sb = &f->mp->fs.sb;
1619         ext4_inode_set_mode(sb, inode_ref.inode, mode);
1620         inode_ref.dirty = true;
1621
1622         ext4_fs_put_inode_ref(&inode_ref);
1623         EXT4_MP_UNLOCK(mp);
1624         return r;
1625 }
1626
1627 int ext4_fchown(ext4_file *f, uint32_t uid, uint32_t gid)
1628 {
1629         int r;
1630         uint32_t ino;
1631         struct ext4_inode_ref inode_ref;
1632         struct ext4_mountpoint *mp = f->mp;
1633
1634         if (!mp)
1635                 return ENOENT;
1636
1637         EXT4_MP_LOCK(mp);
1638
1639         ino = f->inode;
1640         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1641         if (r != EOK) {
1642                 EXT4_MP_UNLOCK(mp);
1643                 return r;
1644         }
1645
1646         ext4_inode_set_uid(inode_ref.inode, uid);
1647         ext4_inode_set_gid(inode_ref.inode, gid);
1648         inode_ref.dirty = true;
1649
1650         ext4_fs_put_inode_ref(&inode_ref);
1651         EXT4_MP_UNLOCK(mp);
1652         return r;
1653 }
1654
1655 int ext4_file_set_atime(ext4_file *f, uint32_t atime)
1656 {
1657         int r;
1658         uint32_t ino;
1659         struct ext4_inode_ref inode_ref;
1660         struct ext4_mountpoint *mp = f->mp;
1661
1662         if (!mp)
1663                 return ENOENT;
1664
1665         EXT4_MP_LOCK(mp);
1666
1667         ino = f->inode;
1668         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1669         if (r != EOK) {
1670                 EXT4_MP_UNLOCK(mp);
1671                 return r;
1672         }
1673
1674         ext4_inode_set_access_time(inode_ref.inode, atime);
1675         inode_ref.dirty = true;
1676
1677         ext4_fs_put_inode_ref(&inode_ref);
1678         EXT4_MP_UNLOCK(mp);
1679         return r;
1680 }
1681
1682 int ext4_file_set_mtime(ext4_file *f, uint32_t mtime)
1683 {
1684         int r;
1685         uint32_t ino;
1686         struct ext4_inode_ref inode_ref;
1687         struct ext4_mountpoint *mp = f->mp;
1688
1689         if (!mp)
1690                 return ENOENT;
1691
1692         EXT4_MP_LOCK(mp);
1693
1694         ino = f->inode;
1695         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1696         if (r != EOK) {
1697                 EXT4_MP_UNLOCK(mp);
1698                 return r;
1699         }
1700
1701         ext4_inode_set_modification_time(inode_ref.inode, mtime);
1702         inode_ref.dirty = true;
1703
1704         ext4_fs_put_inode_ref(&inode_ref);
1705         EXT4_MP_UNLOCK(mp);
1706         return r;
1707 }
1708
1709 int ext4_file_set_ctime(ext4_file *f, uint32_t ctime)
1710 {
1711         int r;
1712         uint32_t ino;
1713         struct ext4_inode_ref inode_ref;
1714         struct ext4_mountpoint *mp = f->mp;
1715
1716         if (!mp)
1717                 return ENOENT;
1718
1719         EXT4_MP_LOCK(mp);
1720
1721         ino = f->inode;
1722         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1723         if (r != EOK) {
1724                 EXT4_MP_UNLOCK(mp);
1725                 return r;
1726         }
1727
1728         ext4_inode_set_change_inode_time(inode_ref.inode, ctime);
1729         inode_ref.dirty = true;
1730
1731         ext4_fs_put_inode_ref(&inode_ref);
1732         EXT4_MP_UNLOCK(mp);
1733         return r;
1734 }
1735
1736 /*********************************DIRECTORY OPERATION************************/
1737
1738 int ext4_dir_rm(const char *path)
1739 {
1740         int r;
1741         int len;
1742         ext4_file f;
1743
1744         struct ext4_mountpoint *mp = ext4_get_mount(path);
1745         struct ext4_inode_ref current;
1746         struct ext4_inode_ref child;
1747         struct ext4_directory_iterator it;
1748
1749         uint32_t name_off;
1750         uint32_t inode_up;
1751         uint32_t inode_current;
1752         uint32_t depth = 1;
1753
1754         bool has_children;
1755         bool is_goal;
1756         bool dir_end;
1757
1758         if (!mp)
1759                 return ENOENT;
1760
1761         EXT4_MP_LOCK(mp);
1762
1763         /*Check if exist.*/
1764         r = ext4_generic_open(&f, path, "r", false, &inode_up, &name_off);
1765         if (r != EOK) {
1766                 EXT4_MP_UNLOCK(mp);
1767                 return r;
1768         }
1769
1770         path += name_off;
1771         len = ext4_path_check(path, &is_goal);
1772
1773         inode_current = f.inode;
1774         dir_end = false;
1775
1776         ext4_block_cache_write_back(mp->fs.bdev, 1);
1777
1778         do {
1779                 /*Load directory node.*/
1780                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_current, &current);
1781                 if (r != EOK) {
1782                         break;
1783                 }
1784
1785                 /*Initialize iterator.*/
1786                 r = ext4_dir_iterator_init(&it, &current, 0);
1787                 if (r != EOK) {
1788                         ext4_fs_put_inode_ref(&current);
1789                         break;
1790                 }
1791
1792                 while (r == EOK) {
1793
1794                         if (!it.current) {
1795                                 dir_end = true;
1796                                 break;
1797                         }
1798
1799                         /*Get up directory inode when ".." entry*/
1800                         if ((it.current->name_length == 2) &&
1801                             ext4_is_dots(it.current->name,
1802                                          it.current->name_length)) {
1803                                 inode_up = ext4_dir_entry_ll_get_inode(it.current);
1804                         }
1805
1806                         /*If directory or file entry,  but not "." ".." entry*/
1807                         if (!ext4_is_dots(it.current->name,
1808                                           it.current->name_length)) {
1809
1810                                 /*Get child inode reference do unlink
1811                                  * directory/file.*/
1812                                 r = ext4_fs_get_inode_ref(&f.mp->fs,
1813                                         ext4_dir_entry_ll_get_inode(it.current),
1814                                         &child);
1815                                 if (r != EOK)
1816                                         break;
1817
1818                                 /*If directory with no leaf children*/
1819                                 r = ext4_has_children(&has_children, &child);
1820                                 if (r != EOK) {
1821                                         ext4_fs_put_inode_ref(&child);
1822                                         break;
1823                                 }
1824
1825                                 if (has_children) {
1826                                         /*Has directory children. Go into this
1827                                          * directory.*/
1828                                         inode_up = inode_current;
1829                                         inode_current = ext4_dir_entry_ll_get_inode(it.current);
1830                                         depth++;
1831                                         ext4_fs_put_inode_ref(&child);
1832                                         break;
1833                                 }
1834
1835                                 /*No children in child directory or file. Just
1836                                  * unlink.*/
1837                                 r = ext4_unlink(f.mp, &current, &child,
1838                                                 (char *)it.current->name,
1839                                                 it.current->name_length);
1840                                 if (r != EOK) {
1841                                         ext4_fs_put_inode_ref(&child);
1842                                         break;
1843                                 }
1844
1845                                 ext4_inode_set_deletion_time(child.inode,
1846                                                              0xFFFFFFFF);
1847                                 ext4_inode_set_links_count(child.inode, 0);
1848                                 child.dirty = true;
1849                                 /*Turncate*/
1850                                 r = ext4_fs_truncate_inode(&child, 0);
1851                                 if (r != EOK) {
1852                                         ext4_fs_put_inode_ref(&child);
1853                                         break;
1854                                 }
1855
1856                                 r = ext4_fs_free_inode(&child);
1857                                 if (r != EOK) {
1858                                         ext4_fs_put_inode_ref(&child);
1859                                         break;
1860                                 }
1861
1862                                 r = ext4_fs_put_inode_ref(&child);
1863                                 if (r != EOK)
1864                                         break;
1865                         }
1866
1867                         r = ext4_dir_iterator_next(&it);
1868                 }
1869
1870                 if (dir_end) {
1871                         /*Directory iterator reached last entry*/
1872                         ext4_has_children(&has_children, &current);
1873                         if (!has_children) {
1874                                 inode_current = inode_up;
1875                                 if (depth)
1876                                         depth--;
1877                         }
1878                         /*Last unlink*/
1879                         if (!depth) {
1880                                 /*Load parent.*/
1881                                 struct ext4_inode_ref parent;
1882                                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_up,
1883                                                           &parent);
1884                                 if (r != EOK)
1885                                         goto End;
1886
1887                                 /* In this place all directories should be
1888                                  * unlinked.
1889                                  * Last unlink from root of current directory*/
1890                                 r = ext4_unlink(f.mp, &parent, &current,
1891                                                 (char *)path, len);
1892                                 if (r != EOK) {
1893                                         ext4_fs_put_inode_ref(&parent);
1894                                         goto End;
1895                                 }
1896
1897                                 if (ext4_inode_get_links_count(current.inode) ==
1898                                     2) {
1899                                         ext4_inode_set_deletion_time(
1900                                             current.inode, 0xFFFFFFFF);
1901                                         ext4_inode_set_links_count(
1902                                             current.inode, 0);
1903                                         current.dirty = true;
1904                                         /*Turncate*/
1905                                         r = ext4_fs_truncate_inode(&current, 0);
1906                                         if (r != EOK) {
1907                                                 ext4_fs_put_inode_ref(&parent);
1908                                                 goto End;
1909                                         }
1910
1911                                         r = ext4_fs_free_inode(&current);
1912                                         if (r != EOK) {
1913                                                 ext4_fs_put_inode_ref(&parent);
1914                                                 goto End;
1915                                         }
1916                                 }
1917
1918                                 r = ext4_fs_put_inode_ref(&parent);
1919                                 if (r != EOK)
1920                                         goto End;
1921                         }
1922                 }
1923
1924         End:
1925                 ext4_dir_iterator_fini(&it);
1926                 ext4_fs_put_inode_ref(&current);
1927                 dir_end = false;
1928
1929                 /*When something goes wrong. End loop.*/
1930                 if (r != EOK)
1931                         break;
1932
1933         } while (depth);
1934
1935         ext4_block_cache_write_back(mp->fs.bdev, 0);
1936         EXT4_MP_UNLOCK(mp);
1937         return r;
1938 }
1939
1940 int ext4_dir_mk(const char *path)
1941 {
1942         int r;
1943         ext4_file f;
1944
1945         struct ext4_mountpoint *mp = ext4_get_mount(path);
1946
1947         if (!mp)
1948                 return ENOENT;
1949
1950         EXT4_MP_LOCK(mp);
1951
1952         /*Check if exist.*/
1953         r = ext4_generic_open(&f, path, "r", false, 0, 0);
1954         if (r == EOK) {
1955                 /*Directory already created*/
1956                 EXT4_MP_UNLOCK(mp);
1957                 return r;
1958         }
1959
1960         /*Create new dir*/
1961         r = ext4_generic_open(&f, path, "w", false, 0, 0);
1962         if (r != EOK) {
1963                 EXT4_MP_UNLOCK(mp);
1964                 return r;
1965         }
1966
1967         EXT4_MP_UNLOCK(mp);
1968         return r;
1969 }
1970
1971 int ext4_dir_open(ext4_dir *d, const char *path)
1972 {
1973         struct ext4_mountpoint *mp = ext4_get_mount(path);
1974         int r;
1975
1976         if (!mp)
1977                 return ENOENT;
1978
1979         EXT4_MP_LOCK(mp);
1980         r = ext4_generic_open(&d->f, path, "r", false, 0, 0);
1981         d->next_off = 0;
1982         EXT4_MP_UNLOCK(mp);
1983         return r;
1984 }
1985
1986 int ext4_dir_close(ext4_dir *d) { return ext4_fclose(&d->f); }
1987
1988 const ext4_direntry *ext4_dir_entry_next(ext4_dir *d)
1989 {
1990 #define EXT4_DIR_ENTRY_OFFSET_TERM (uint64_t)(-1)
1991
1992         int r;
1993         ext4_direntry *de = 0;
1994         struct ext4_inode_ref dir;
1995         struct ext4_directory_iterator it;
1996
1997         EXT4_MP_LOCK(d->f.mp);
1998
1999         if (d->next_off == EXT4_DIR_ENTRY_OFFSET_TERM) {
2000                 EXT4_MP_UNLOCK(d->f.mp);
2001                 return 0;
2002         }
2003
2004         r = ext4_fs_get_inode_ref(&d->f.mp->fs, d->f.inode, &dir);
2005         if (r != EOK) {
2006                 goto Finish;
2007         }
2008
2009         r = ext4_dir_iterator_init(&it, &dir, d->next_off);
2010         if (r != EOK) {
2011                 ext4_fs_put_inode_ref(&dir);
2012                 goto Finish;
2013         }
2014
2015         memcpy(&d->de, it.current, sizeof(ext4_direntry));
2016         de = &d->de;
2017
2018         ext4_dir_iterator_next(&it);
2019
2020         d->next_off =
2021             it.current ? it.current_offset : EXT4_DIR_ENTRY_OFFSET_TERM;
2022
2023         ext4_dir_iterator_fini(&it);
2024         ext4_fs_put_inode_ref(&dir);
2025
2026 Finish:
2027         EXT4_MP_UNLOCK(d->f.mp);
2028         return de;
2029 }
2030
2031 /**
2032  * @}
2033  */