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 ENOTEMPTY;
 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;
 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
 646                                     : EXT4_DIRECTORY_FILETYPE_DIR);
 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_open2(&f, path, O_RDWR,
1032                                EXT4_DIRECTORY_FILETYPE_UNKNOWN,
1033                                &parent_inode, &name_off);
1034         if (r != EOK) {
1035                 EXT4_MP_UNLOCK(mp);
1036                 return r;
1037         }
1038
1039         /*Load parent*/
1040         r = ext4_fs_get_inode_ref(&mp->fs, parent_inode, &parent);
1041         if (r != EOK) {
1042                 EXT4_MP_UNLOCK(mp);
1043                 return r;
1044         }
1045
1046         /*We have file to delete. Load it.*/
1047         r = ext4_fs_get_inode_ref(&mp->fs, f.inode, &child);
1048         if (r != EOK) {
1049                 ext4_fs_put_inode_ref(&parent);
1050                 EXT4_MP_UNLOCK(mp);
1051                 return r;
1052         }
1053
1054         /*Set path*/
1055         path += name_off;
1056
1057         len = ext4_path_check(path, &is_goal);
1058
1059         /*Unlink from parent*/
1060         r = ext4_unlink(mp, &parent, &child, path, len);
1061         if (r != EOK)
1062                 goto Finish;
1063
1064         /*Link count is zero, the inode should be freed. */
1065         if (!ext4_inode_get_links_count(child.inode)) {
1066                 ext4_inode_set_deletion_time(child.inode, 0xFFFFFFFF);
1067                 /*Turncate*/
1068                 ext4_block_cache_write_back(mp->fs.bdev, 1);
1069                 /*Truncate may be IO heavy. Do it writeback cache mode.*/
1070                 r = ext4_fs_truncate_inode(&child, 0);
1071                 ext4_block_cache_write_back(mp->fs.bdev, 0);
1072
1073                 if (r != EOK)
1074                         goto Finish;
1075
1076                 r = ext4_fs_free_inode(&child);
1077                 if (r != EOK)
1078                         goto Finish;
1079         }
1080
1081 Finish:
1082         ext4_fs_put_inode_ref(&child);
1083         ext4_fs_put_inode_ref(&parent);
1084         EXT4_MP_UNLOCK(mp);
1085         return r;
1086 }
1087
1088 int ext4_fill_raw_inode(const char *mount_point, uint32_t ino,
1089                         struct ext4_inode *inode)
1090 {
1091         int r;
1092         struct ext4_inode_ref inode_ref;
1093         struct ext4_mountpoint *mp = ext4_get_mount(mount_point);
1094
1095         if (!mp)
1096                 return ENOENT;
1097
1098         EXT4_MP_LOCK(mp);
1099
1100         /*Load parent*/
1101         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1102         if (r != EOK) {
1103                 EXT4_MP_UNLOCK(mp);
1104                 return r;
1105         }
1106
1107         memcpy(inode, inode_ref.inode, sizeof(struct ext4_inode));
1108
1109         ext4_fs_put_inode_ref(&inode_ref);
1110         EXT4_MP_UNLOCK(mp);
1111         return r;
1112 }
1113
1114 int ext4_fopen(ext4_file *f, const char *path, const char *flags)
1115 {
1116         struct ext4_mountpoint *mp = ext4_get_mount(path);
1117         int r;
1118
1119         if (!mp)
1120                 return ENOENT;
1121
1122         EXT4_MP_LOCK(mp);
1123         ext4_block_cache_write_back(mp->fs.bdev, 1);
1124         r = ext4_generic_open(f, path, flags, true, 0, 0);
1125         ext4_block_cache_write_back(mp->fs.bdev, 0);
1126         EXT4_MP_UNLOCK(mp);
1127         return r;
1128 }
1129
1130 int ext4_fopen2(ext4_file *f, const char *path, int flags, bool file_expect)
1131 {
1132         struct ext4_mountpoint *mp = ext4_get_mount(path);
1133         int r;
1134         int filetype;
1135
1136         if (!mp)
1137                 return ENOENT;
1138
1139         if (file_expect == true)
1140                 filetype = EXT4_DIRECTORY_FILETYPE_REG_FILE;
1141         else
1142                 filetype = EXT4_DIRECTORY_FILETYPE_DIR;
1143
1144         EXT4_MP_LOCK(mp);
1145         ext4_block_cache_write_back(mp->fs.bdev, 1);
1146         r = ext4_generic_open2(f, path, flags, filetype, 0, 0);
1147         ext4_block_cache_write_back(mp->fs.bdev, 0);
1148         EXT4_MP_UNLOCK(mp);
1149         return r;
1150 }
1151
1152 int ext4_fopen_all(ext4_file *f, const char *path, int flags)
1153 {
1154         struct ext4_mountpoint *mp = ext4_get_mount(path);
1155         int r;
1156         int filetype;
1157
1158         if (!mp)
1159                 return ENOENT;
1160
1161         if (flags & O_CREAT)
1162                 return EINVAL;
1163
1164         filetype = EXT4_DIRECTORY_FILETYPE_UNKNOWN;
1165
1166         EXT4_MP_LOCK(mp);
1167         ext4_block_cache_write_back(mp->fs.bdev, 1);
1168         r = ext4_generic_open2(f, path, flags, filetype, 0, 0);
1169         ext4_block_cache_write_back(mp->fs.bdev, 0);
1170         EXT4_MP_UNLOCK(mp);
1171         return r;
1172 }
1173
1174 int ext4_fclose(ext4_file *f)
1175 {
1176         ext4_assert(f && f->mp);
1177
1178         f->mp = 0;
1179         f->flags = 0;
1180         f->inode = 0;
1181         f->fpos = f->fsize = 0;
1182
1183         return EOK;
1184 }
1185
1186 static int ext4_ftruncate_no_lock(ext4_file *f, uint64_t size)
1187 {
1188         struct ext4_inode_ref ref;
1189         int r;
1190
1191
1192         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1193         if (r != EOK) {
1194                 EXT4_MP_UNLOCK(f->mp);
1195                 return r;
1196         }
1197
1198         /*Sync file size*/
1199         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1200         if (f->fsize <= size) {
1201                 r = EOK;
1202                 goto Finish;
1203         }
1204
1205         /*Start write back cache mode.*/
1206         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1207         if (r != EOK)
1208                 goto Finish;
1209
1210         r = ext4_fs_truncate_inode(&ref, size);
1211         if (r != EOK)
1212                 goto Finish;
1213
1214         f->fsize = size;
1215         if (f->fpos > size)
1216                 f->fpos = size;
1217
1218         /*Stop write back cache mode*/
1219         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1220
1221         if (r != EOK)
1222                 goto Finish;
1223
1224 Finish:
1225         ext4_fs_put_inode_ref(&ref);
1226         return r;
1227
1228 }
1229
1230 int ext4_ftruncate(ext4_file *f, uint64_t size)
1231 {
1232         int r;
1233         ext4_assert(f && f->mp);
1234
1235         if (f->flags & O_RDONLY)
1236                 return EPERM;
1237
1238         EXT4_MP_LOCK(f->mp);
1239
1240         r = ext4_ftruncate_no_lock(f, size);
1241
1242         EXT4_MP_UNLOCK(f->mp);
1243         return r;
1244 }
1245
1246 int ext4_fread(ext4_file *f, void *buf, size_t size, size_t *rcnt)
1247 {
1248         uint32_t u;
1249         uint32_t fblock;
1250         uint32_t fblock_start;
1251         uint32_t fblock_cnt;
1252         uint32_t sblock;
1253         uint32_t sblock_end;
1254         uint32_t block_size;
1255         uint8_t *u8_buf = buf;
1256         int r;
1257         struct ext4_block b;
1258         struct ext4_inode_ref ref;
1259
1260         ext4_assert(f && f->mp);
1261
1262         if (f->flags & O_WRONLY)
1263                 return EPERM;
1264
1265         if (!size)
1266                 return EOK;
1267
1268         EXT4_MP_LOCK(f->mp);
1269
1270         if (rcnt)
1271                 *rcnt = 0;
1272
1273         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1274         if (r != EOK) {
1275                 EXT4_MP_UNLOCK(f->mp);
1276                 return r;
1277         }
1278
1279         /*Sync file size*/
1280         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1281
1282         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1283         size = size > (f->fsize - f->fpos) ? (f->fsize - f->fpos) : size;
1284         sblock = (f->fpos) / block_size;
1285         sblock_end = (f->fpos + size) / block_size;
1286         u = (f->fpos) % block_size;
1287
1288         /*If the size of symlink is smaller than 60 bytes*/
1289         if (ext4_inode_is_type(&f->mp->fs.sb, ref.inode, EXT4_INODE_MODE_SOFTLINK)
1290                         && f->fsize < sizeof(ref.inode->blocks)
1291                         && !ext4_inode_get_blocks_count(&f->mp->fs.sb, ref.inode)) {
1292                 char *content = (char *)ref.inode->blocks;
1293                 if (f->fpos < f->fsize) {
1294                         r = (u + size > f->fsize)
1295                                 ?(f->fsize - u)
1296                                 :(size);
1297                         memcpy(buf, content + u, r);
1298                         if (rcnt)
1299                                 *rcnt = r;
1300
1301                 } else {
1302                         r = 0;
1303                         if (rcnt)
1304                                 *rcnt = 0;
1305
1306                 }
1307
1308                 r = EOK;
1309                 goto Finish;
1310         }
1311
1312         if (u) {
1313
1314                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1315
1316                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1317                 if (r != EOK)
1318                         goto Finish;
1319
1320                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1321                 if (r != EOK)
1322                         goto Finish;
1323
1324                 memcpy(u8_buf, b.data + u, ll);
1325
1326                 r = ext4_block_set(f->mp->fs.bdev, &b);
1327                 if (r != EOK)
1328                         goto Finish;
1329
1330                 u8_buf += ll;
1331                 size -= ll;
1332                 f->fpos += ll;
1333
1334                 if (rcnt)
1335                         *rcnt += ll;
1336
1337                 sblock++;
1338         }
1339
1340         fblock_start = 0;
1341         fblock_cnt = 0;
1342         while (size >= block_size) {
1343                 while (sblock < sblock_end) {
1344                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1345                                                                &fblock);
1346                         if (r != EOK)
1347                                 goto Finish;
1348
1349                         sblock++;
1350
1351                         if (!fblock_start) {
1352                                 fblock_start = fblock;
1353                         }
1354
1355                         if ((fblock_start + fblock_cnt) != fblock)
1356                                 break;
1357
1358                         fblock_cnt++;
1359                 }
1360
1361                 r = ext4_blocks_get_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1362                                            fblock_cnt);
1363                 if (r != EOK)
1364                         goto Finish;
1365
1366                 size -= block_size * fblock_cnt;
1367                 u8_buf += block_size * fblock_cnt;
1368                 f->fpos += block_size * fblock_cnt;
1369
1370                 if (rcnt)
1371                         *rcnt += block_size * fblock_cnt;
1372
1373                 fblock_start = fblock;
1374                 fblock_cnt = 1;
1375         }
1376
1377         if (size) {
1378                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1379                 if (r != EOK)
1380                         goto Finish;
1381
1382                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1383                 if (r != EOK)
1384                         goto Finish;
1385
1386                 memcpy(u8_buf, b.data, size);
1387
1388                 r = ext4_block_set(f->mp->fs.bdev, &b);
1389                 if (r != EOK)
1390                         goto Finish;
1391
1392                 f->fpos += size;
1393
1394                 if (rcnt)
1395                         *rcnt += size;
1396         }
1397
1398 Finish:
1399         ext4_fs_put_inode_ref(&ref);
1400         EXT4_MP_UNLOCK(f->mp);
1401         return r;
1402 }
1403
1404 int ext4_fwrite(ext4_file *f, const void *buf, size_t size, size_t *wcnt)
1405 {
1406         uint32_t u;
1407         uint32_t fblock;
1408
1409         uint32_t sblock;
1410         uint32_t sblock_end;
1411         uint32_t file_blocks;
1412         uint32_t block_size;
1413         uint32_t fblock_start;
1414         uint32_t fblock_cnt;
1415
1416         struct ext4_block b;
1417         struct ext4_inode_ref ref;
1418         const uint8_t *u8_buf = buf;
1419         int r;
1420
1421         ext4_assert(f && f->mp);
1422
1423         if (f->flags & O_RDONLY)
1424                 return EPERM;
1425
1426         if (!size)
1427                 return EOK;
1428
1429         EXT4_MP_LOCK(f->mp);
1430
1431         if (wcnt)
1432                 *wcnt = 0;
1433
1434         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1435         if (r != EOK) {
1436                 EXT4_MP_UNLOCK(f->mp);
1437                 return r;
1438         }
1439
1440         /*Sync file size*/
1441         f->fsize = ext4_inode_get_size(&f->mp->fs.sb, ref.inode);
1442
1443         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1444
1445         sblock_end = (f->fpos + size) > f->fsize ? (f->fpos + size) : f->fsize;
1446         sblock_end /= block_size;
1447         file_blocks = (f->fsize / block_size);
1448
1449         if (f->fsize % block_size)
1450                 file_blocks++;
1451
1452         sblock = (f->fpos) / block_size;
1453
1454         u = (f->fpos) % block_size;
1455
1456         if (u) {
1457                 uint32_t ll = size > (block_size - u) ? (block_size - u) : size;
1458
1459                 r = ext4_fs_get_inode_data_block_index(&ref, sblock, &fblock);
1460                 if (r != EOK)
1461                         goto Finish;
1462
1463                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1464                 if (r != EOK)
1465                         goto Finish;
1466
1467                 memcpy(b.data + u, u8_buf, ll);
1468                 b.dirty = true;
1469
1470                 r = ext4_block_set(f->mp->fs.bdev, &b);
1471                 if (r != EOK)
1472                         goto Finish;
1473
1474                 u8_buf += ll;
1475                 size -= ll;
1476                 f->fpos += ll;
1477
1478                 if (wcnt)
1479                         *wcnt += ll;
1480
1481                 sblock++;
1482         }
1483
1484         /*Start write back cache mode.*/
1485         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1486         if (r != EOK)
1487                 goto Finish;
1488
1489         fblock_start = 0;
1490         fblock_cnt = 0;
1491         while (size >= block_size) {
1492
1493                 while (sblock < sblock_end) {
1494                         if (sblock < file_blocks) {
1495                                 r = ext4_fs_get_inode_data_block_index(
1496                                     &ref, sblock, &fblock);
1497                                 if (r != EOK)
1498                                         break;
1499                         } else {
1500                                 r = ext4_fs_append_inode_block(&ref, &fblock,
1501                                                                &sblock);
1502                                 if (r != EOK)
1503                                         break;
1504                         }
1505
1506                         sblock++;
1507
1508                         if (!fblock_start) {
1509                                 fblock_start = fblock;
1510                         }
1511
1512                         if ((fblock_start + fblock_cnt) != fblock)
1513                                 break;
1514
1515                         fblock_cnt++;
1516                 }
1517
1518                 r = ext4_blocks_set_direct(f->mp->fs.bdev, u8_buf, fblock_start,
1519                                            fblock_cnt);
1520                 if (r != EOK)
1521                         break;
1522
1523                 size -= block_size * fblock_cnt;
1524                 u8_buf += block_size * fblock_cnt;
1525                 f->fpos += block_size * fblock_cnt;
1526
1527                 if (wcnt)
1528                         *wcnt += block_size * fblock_cnt;
1529
1530                 fblock_start = fblock;
1531                 fblock_cnt = 1;
1532         }
1533
1534         /*Stop write back cache mode*/
1535         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1536
1537         if (r != EOK)
1538                 goto Finish;
1539
1540         if (size) {
1541                 if (sblock < file_blocks) {
1542                         r = ext4_fs_get_inode_data_block_index(&ref, sblock,
1543                                                                &fblock);
1544                         if (r != EOK)
1545                                 goto Finish;
1546                 } else {
1547                         r = ext4_fs_append_inode_block(&ref, &fblock, &sblock);
1548                         if (r != EOK)
1549                                 goto Finish;
1550                 }
1551
1552                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1553                 if (r != EOK)
1554                         goto Finish;
1555
1556                 memcpy(b.data, u8_buf, size);
1557                 b.dirty = true;
1558
1559                 r = ext4_block_set(f->mp->fs.bdev, &b);
1560                 if (r != EOK)
1561                         goto Finish;
1562
1563                 f->fpos += size;
1564
1565                 if (wcnt)
1566                         *wcnt += size;
1567         }
1568
1569         if (f->fpos > f->fsize) {
1570                 f->fsize = f->fpos;
1571                 ext4_inode_set_size(ref.inode, f->fsize);
1572                 ref.dirty = true;
1573         }
1574
1575 Finish:
1576         ext4_fs_put_inode_ref(&ref);
1577         EXT4_MP_UNLOCK(f->mp);
1578         return r;
1579 }
1580
1581 int ext4_fseek(ext4_file *f, uint64_t offset, uint32_t origin)
1582 {
1583         switch (origin) {
1584         case SEEK_SET:
1585                 if (offset > f->fsize)
1586                         return EINVAL;
1587
1588                 f->fpos = offset;
1589                 return EOK;
1590         case SEEK_CUR:
1591                 if ((offset + f->fpos) > f->fsize)
1592                         return EINVAL;
1593
1594                 f->fpos += offset;
1595                 return EOK;
1596         case SEEK_END:
1597                 if (offset > f->fsize)
1598                         return EINVAL;
1599
1600                 f->fpos = f->fsize - offset;
1601                 return EOK;
1602         }
1603         return EINVAL;
1604 }
1605
1606 uint64_t ext4_ftell(ext4_file *f) { return f->fpos; }
1607
1608 uint64_t ext4_fsize(ext4_file *f) { return f->fsize; }
1609
1610 int ext4_fchmod(ext4_file *f, uint32_t mode)
1611 {
1612         int r;
1613         uint32_t ino;
1614         struct ext4_sblock *sb;
1615         struct ext4_inode_ref inode_ref;
1616         struct ext4_mountpoint *mp = f->mp;
1617
1618         if (!mp)
1619                 return ENOENT;
1620
1621         EXT4_MP_LOCK(mp);
1622
1623         ino = f->inode;
1624         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1625         if (r != EOK) {
1626                 EXT4_MP_UNLOCK(mp);
1627                 return r;
1628         }
1629
1630         sb = &f->mp->fs.sb;
1631         ext4_inode_set_mode(sb, inode_ref.inode, mode);
1632         inode_ref.dirty = true;
1633
1634         ext4_fs_put_inode_ref(&inode_ref);
1635         EXT4_MP_UNLOCK(mp);
1636         return r;
1637 }
1638
1639 int ext4_fchown(ext4_file *f, uint32_t uid, uint32_t gid)
1640 {
1641         int r;
1642         uint32_t ino;
1643         struct ext4_inode_ref inode_ref;
1644         struct ext4_mountpoint *mp = f->mp;
1645
1646         if (!mp)
1647                 return ENOENT;
1648
1649         EXT4_MP_LOCK(mp);
1650
1651         ino = f->inode;
1652         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1653         if (r != EOK) {
1654                 EXT4_MP_UNLOCK(mp);
1655                 return r;
1656         }
1657
1658         ext4_inode_set_uid(inode_ref.inode, uid);
1659         ext4_inode_set_gid(inode_ref.inode, gid);
1660         inode_ref.dirty = true;
1661
1662         ext4_fs_put_inode_ref(&inode_ref);
1663         EXT4_MP_UNLOCK(mp);
1664         return r;
1665 }
1666
1667 int ext4_file_set_atime(ext4_file *f, uint32_t atime)
1668 {
1669         int r;
1670         uint32_t ino;
1671         struct ext4_inode_ref inode_ref;
1672         struct ext4_mountpoint *mp = f->mp;
1673
1674         if (!mp)
1675                 return ENOENT;
1676
1677         EXT4_MP_LOCK(mp);
1678
1679         ino = f->inode;
1680         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1681         if (r != EOK) {
1682                 EXT4_MP_UNLOCK(mp);
1683                 return r;
1684         }
1685
1686         ext4_inode_set_access_time(inode_ref.inode, atime);
1687         inode_ref.dirty = true;
1688
1689         ext4_fs_put_inode_ref(&inode_ref);
1690         EXT4_MP_UNLOCK(mp);
1691         return r;
1692 }
1693
1694 int ext4_file_set_mtime(ext4_file *f, uint32_t mtime)
1695 {
1696         int r;
1697         uint32_t ino;
1698         struct ext4_inode_ref inode_ref;
1699         struct ext4_mountpoint *mp = f->mp;
1700
1701         if (!mp)
1702                 return ENOENT;
1703
1704         EXT4_MP_LOCK(mp);
1705
1706         ino = f->inode;
1707         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1708         if (r != EOK) {
1709                 EXT4_MP_UNLOCK(mp);
1710                 return r;
1711         }
1712
1713         ext4_inode_set_modification_time(inode_ref.inode, mtime);
1714         inode_ref.dirty = true;
1715
1716         ext4_fs_put_inode_ref(&inode_ref);
1717         EXT4_MP_UNLOCK(mp);
1718         return r;
1719 }
1720
1721 int ext4_file_set_ctime(ext4_file *f, uint32_t ctime)
1722 {
1723         int r;
1724         uint32_t ino;
1725         struct ext4_inode_ref inode_ref;
1726         struct ext4_mountpoint *mp = f->mp;
1727
1728         if (!mp)
1729                 return ENOENT;
1730
1731         EXT4_MP_LOCK(mp);
1732
1733         ino = f->inode;
1734         r = ext4_fs_get_inode_ref(&mp->fs, ino, &inode_ref);
1735         if (r != EOK) {
1736                 EXT4_MP_UNLOCK(mp);
1737                 return r;
1738         }
1739
1740         ext4_inode_set_change_inode_time(inode_ref.inode, ctime);
1741         inode_ref.dirty = true;
1742
1743         ext4_fs_put_inode_ref(&inode_ref);
1744         EXT4_MP_UNLOCK(mp);
1745         return r;
1746 }
1747
1748 static int ext4_fsymlink_set(ext4_file *f, const void *buf, uint32_t size)
1749 {
1750         struct ext4_block b;
1751         struct ext4_inode_ref ref;
1752         uint32_t sblock, fblock;
1753         uint32_t block_size;
1754         int r;
1755
1756         ext4_assert(f && f->mp);
1757
1758         if (!size)
1759                 return EOK;
1760
1761         r = ext4_fs_get_inode_ref(&f->mp->fs, f->inode, &ref);
1762         if (r != EOK) {
1763                 EXT4_MP_UNLOCK(f->mp);
1764                 return r;
1765         }
1766
1767         /*Sync file size*/
1768         block_size = ext4_sb_get_block_size(&f->mp->fs.sb);
1769         if (size > block_size) {
1770                 r = EINVAL;
1771                 goto Finish;
1772         }
1773         r = ext4_ftruncate_no_lock(f, 0);
1774         if (r != EOK)
1775                 goto Finish;
1776
1777         /*Start write back cache mode.*/
1778         r = ext4_block_cache_write_back(f->mp->fs.bdev, 1);
1779         if (r != EOK)
1780                 goto Finish;
1781
1782         /*If the size of symlink is smaller than 60 bytes*/
1783         if (size < sizeof(ref.inode->blocks)) {
1784                 char *content = (char *)ref.inode->blocks;
1785                 memset(content, 0, sizeof(ref.inode->blocks));
1786                 memcpy(content, buf, size);
1787                 ext4_inode_clear_flag(ref.inode, EXT4_INODE_FLAG_EXTENTS);
1788         } else {
1789                 ext4_fs_inode_blocks_init(&f->mp->fs, &ref);
1790                 r = ext4_fs_append_inode_block(&ref, &fblock, &sblock);
1791                 if (r != EOK)
1792                         goto Finish;
1793
1794                 r = ext4_block_get(f->mp->fs.bdev, &b, fblock);
1795                 if (r != EOK)
1796                         goto Finish;
1797
1798                 memcpy(b.data, buf, size);
1799                 b.dirty = true;
1800                 r = ext4_block_set(f->mp->fs.bdev, &b);
1801                 if (r != EOK)
1802                         goto Finish;
1803         }
1804
1805         /*Stop write back cache mode*/
1806         ext4_block_cache_write_back(f->mp->fs.bdev, 0);
1807
1808         if (r != EOK)
1809                 goto Finish;
1810
1811         ext4_inode_set_size(ref.inode, size);
1812         ref.dirty = true;
1813
1814         f->fsize = size;
1815         if (f->fpos > size)
1816                 f->fpos = size;
1817
1818 Finish:
1819         ext4_fs_put_inode_ref(&ref);
1820         return r;
1821 }
1822
1823 int ext4_fsymlink(const char *target, const char *path)
1824 {
1825         struct ext4_mountpoint *mp = ext4_get_mount(path);
1826         int r;
1827         ext4_file f;
1828         int filetype;
1829
1830         if (!mp)
1831                 return ENOENT;
1832
1833         filetype = EXT4_DIRECTORY_FILETYPE_SYMLINK;
1834
1835         EXT4_MP_LOCK(mp);
1836         ext4_block_cache_write_back(mp->fs.bdev, 1);
1837         r = ext4_generic_open2(&f, path, O_RDWR|O_CREAT, filetype, 0, 0);
1838         if (r == EOK)
1839                 r = ext4_fsymlink_set(&f, target, strlen(target));
1840         else
1841                 goto Finish;
1842
1843         ext4_fclose(&f);
1844
1845 Finish:
1846         ext4_block_cache_write_back(mp->fs.bdev, 0);
1847         EXT4_MP_UNLOCK(mp);
1848         return r;
1849 }
1850
1851 /*********************************DIRECTORY OPERATION************************/
1852
1853 int ext4_dir_rm(const char *path)
1854 {
1855         int r;
1856         int len;
1857         ext4_file f;
1858
1859         struct ext4_mountpoint *mp = ext4_get_mount(path);
1860         struct ext4_inode_ref current;
1861         struct ext4_inode_ref child;
1862         struct ext4_directory_iterator it;
1863
1864         uint32_t name_off;
1865         uint32_t inode_up;
1866         uint32_t inode_current;
1867         uint32_t depth = 1;
1868
1869         bool has_children;
1870         bool is_goal;
1871         bool dir_end;
1872
1873         if (!mp)
1874                 return ENOENT;
1875
1876         EXT4_MP_LOCK(mp);
1877
1878         /*Check if exist.*/
1879         r = ext4_generic_open(&f, path, "r", false, &inode_up, &name_off);
1880         if (r != EOK) {
1881                 EXT4_MP_UNLOCK(mp);
1882                 return r;
1883         }
1884
1885         path += name_off;
1886         len = ext4_path_check(path, &is_goal);
1887
1888         inode_current = f.inode;
1889         dir_end = false;
1890
1891         ext4_block_cache_write_back(mp->fs.bdev, 1);
1892
1893         do {
1894                 /*Load directory node.*/
1895                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_current, &current);
1896                 if (r != EOK) {
1897                         break;
1898                 }
1899
1900                 /*Initialize iterator.*/
1901                 r = ext4_dir_iterator_init(&it, &current, 0);
1902                 if (r != EOK) {
1903                         ext4_fs_put_inode_ref(&current);
1904                         break;
1905                 }
1906
1907                 while (r == EOK) {
1908
1909                         if (!it.current) {
1910                                 dir_end = true;
1911                                 break;
1912                         }
1913
1914                         /*Get up directory inode when ".." entry*/
1915                         if ((it.current->name_length == 2) &&
1916                             ext4_is_dots(it.current->name,
1917                                          it.current->name_length)) {
1918                                 inode_up = ext4_dir_entry_ll_get_inode(it.current);
1919                         }
1920
1921                         /*If directory or file entry,  but not "." ".." entry*/
1922                         if (!ext4_is_dots(it.current->name,
1923                                           it.current->name_length)) {
1924
1925                                 /*Get child inode reference do unlink
1926                                  * directory/file.*/
1927                                 r = ext4_fs_get_inode_ref(&f.mp->fs,
1928                                         ext4_dir_entry_ll_get_inode(it.current),
1929                                         &child);
1930                                 if (r != EOK)
1931                                         break;
1932
1933                                 /*If directory with no leaf children*/
1934                                 r = ext4_has_children(&has_children, &child);
1935                                 if (r != EOK) {
1936                                         ext4_fs_put_inode_ref(&child);
1937                                         break;
1938                                 }
1939
1940                                 if (has_children) {
1941                                         /*Has directory children. Go into this
1942                                          * directory.*/
1943                                         inode_up = inode_current;
1944                                         inode_current = ext4_dir_entry_ll_get_inode(it.current);
1945                                         depth++;
1946                                         ext4_fs_put_inode_ref(&child);
1947                                         break;
1948                                 }
1949
1950                                 /*No children in child directory or file. Just
1951                                  * unlink.*/
1952                                 r = ext4_unlink(f.mp, &current, &child,
1953                                                 (char *)it.current->name,
1954                                                 it.current->name_length);
1955                                 if (r != EOK) {
1956                                         ext4_fs_put_inode_ref(&child);
1957                                         break;
1958                                 }
1959
1960                                 ext4_inode_set_deletion_time(child.inode,
1961                                                              0xFFFFFFFF);
1962                                 ext4_inode_set_links_count(child.inode, 0);
1963                                 child.dirty = true;
1964                                 /*Turncate*/
1965                                 r = ext4_fs_truncate_inode(&child, 0);
1966                                 if (r != EOK) {
1967                                         ext4_fs_put_inode_ref(&child);
1968                                         break;
1969                                 }
1970
1971                                 r = ext4_fs_free_inode(&child);
1972                                 if (r != EOK) {
1973                                         ext4_fs_put_inode_ref(&child);
1974                                         break;
1975                                 }
1976
1977                                 r = ext4_fs_put_inode_ref(&child);
1978                                 if (r != EOK)
1979                                         break;
1980                         }
1981
1982                         r = ext4_dir_iterator_next(&it);
1983                 }
1984
1985                 if (dir_end) {
1986                         /*Directory iterator reached last entry*/
1987                         ext4_has_children(&has_children, &current);
1988                         if (!has_children) {
1989                                 inode_current = inode_up;
1990                                 if (depth)
1991                                         depth--;
1992                         }
1993                         /*Last unlink*/
1994                         if (!depth) {
1995                                 /*Load parent.*/
1996                                 struct ext4_inode_ref parent;
1997                                 r = ext4_fs_get_inode_ref(&f.mp->fs, inode_up,
1998                                                           &parent);
1999                                 if (r != EOK)
2000                                         goto End;
2001
2002                                 /* In this place all directories should be
2003                                  * unlinked.
2004                                  * Last unlink from root of current directory*/
2005                                 r = ext4_unlink(f.mp, &parent, &current,
2006                                                 (char *)path, len);
2007                                 if (r != EOK) {
2008                                         ext4_fs_put_inode_ref(&parent);
2009                                         goto End;
2010                                 }
2011
2012                                 if (ext4_inode_get_links_count(current.inode) ==
2013                                     2) {
2014                                         ext4_inode_set_deletion_time(
2015                                             current.inode, 0xFFFFFFFF);
2016                                         ext4_inode_set_links_count(
2017                                             current.inode, 0);
2018                                         current.dirty = true;
2019                                         /*Turncate*/
2020                                         r = ext4_fs_truncate_inode(&current, 0);
2021                                         if (r != EOK) {
2022                                                 ext4_fs_put_inode_ref(&parent);
2023                                                 goto End;
2024                                         }
2025
2026                                         r = ext4_fs_free_inode(&current);
2027                                         if (r != EOK) {
2028                                                 ext4_fs_put_inode_ref(&parent);
2029                                                 goto End;
2030                                         }
2031                                 }
2032
2033                                 r = ext4_fs_put_inode_ref(&parent);
2034                                 if (r != EOK)
2035                                         goto End;
2036                         }
2037                 }
2038
2039         End:
2040                 ext4_dir_iterator_fini(&it);
2041                 ext4_fs_put_inode_ref(&current);
2042                 dir_end = false;
2043
2044                 /*When something goes wrong. End loop.*/
2045                 if (r != EOK)
2046                         break;
2047
2048         } while (depth);
2049
2050         ext4_block_cache_write_back(mp->fs.bdev, 0);
2051         EXT4_MP_UNLOCK(mp);
2052         return r;
2053 }
2054
2055 int ext4_dir_mk(const char *path)
2056 {
2057         int r;
2058         ext4_file f;
2059
2060         struct ext4_mountpoint *mp = ext4_get_mount(path);
2061
2062         if (!mp)
2063                 return ENOENT;
2064
2065         EXT4_MP_LOCK(mp);
2066
2067         /*Check if exist.*/
2068         r = ext4_generic_open(&f, path, "r", false, 0, 0);
2069         if (r == EOK) {
2070                 /*Directory already created*/
2071                 EXT4_MP_UNLOCK(mp);
2072                 return r;
2073         }
2074
2075         /*Create new dir*/
2076         r = ext4_generic_open(&f, path, "w", false, 0, 0);
2077         if (r != EOK) {
2078                 EXT4_MP_UNLOCK(mp);
2079                 return r;
2080         }
2081
2082         EXT4_MP_UNLOCK(mp);
2083         return r;
2084 }
2085
2086 int ext4_dir_open(ext4_dir *d, const char *path)
2087 {
2088         struct ext4_mountpoint *mp = ext4_get_mount(path);
2089         int r;
2090
2091         if (!mp)
2092                 return ENOENT;
2093
2094         EXT4_MP_LOCK(mp);
2095         r = ext4_generic_open(&d->f, path, "r", false, 0, 0);
2096         d->next_off = 0;
2097         EXT4_MP_UNLOCK(mp);
2098         return r;
2099 }
2100
2101 int ext4_dir_close(ext4_dir *d) { return ext4_fclose(&d->f); }
2102
2103 const ext4_direntry *ext4_dir_entry_next(ext4_dir *d)
2104 {
2105 #define EXT4_DIR_ENTRY_OFFSET_TERM (uint64_t)(-1)
2106
2107         int r;
2108         ext4_direntry *de = 0;
2109         struct ext4_inode_ref dir;
2110         struct ext4_directory_iterator it;
2111
2112         EXT4_MP_LOCK(d->f.mp);
2113
2114         if (d->next_off == EXT4_DIR_ENTRY_OFFSET_TERM) {
2115                 EXT4_MP_UNLOCK(d->f.mp);
2116                 return 0;
2117         }
2118
2119         r = ext4_fs_get_inode_ref(&d->f.mp->fs, d->f.inode, &dir);
2120         if (r != EOK) {
2121                 goto Finish;
2122         }
2123
2124         r = ext4_dir_iterator_init(&it, &dir, d->next_off);
2125         if (r != EOK) {
2126                 ext4_fs_put_inode_ref(&dir);
2127                 goto Finish;
2128         }
2129
2130         memcpy(&d->de, it.current, sizeof(ext4_direntry));
2131         de = &d->de;
2132
2133         ext4_dir_iterator_next(&it);
2134
2135         d->next_off =
2136             it.current ? it.current_offset : EXT4_DIR_ENTRY_OFFSET_TERM;
2137
2138         ext4_dir_iterator_fini(&it);
2139         ext4_fs_put_inode_ref(&dir);
2140
2141 Finish:
2142         EXT4_MP_UNLOCK(d->f.mp);
2143         return de;
2144 }
2145
2146 /**
2147  * @}
2148  */