fs/ubifs/dir.c

   1 /* * This file is part of UBIFS.
   2  *
   3  * Copyright (C) 2006-2008 Nokia Corporation.
   4  * Copyright (C) 2006, 2007 University of Szeged, Hungary
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License version 2 as published by
   8  * the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along with
  16  * this program; if not, write to the Free Software Foundation, Inc., 51
  17  * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  18  *
  19  * Authors: Artem Bityutskiy (Битюцкий Артём)
  20  *          Adrian Hunter
  21  *          Zoltan Sogor
  22  */
  23
  24 /*
  25  * This file implements directory operations.
  26  *
  27  * All FS operations in this file allocate budget before writing anything to the
  28  * media. If they fail to allocate it, the error is returned. The only
  29  * exceptions are 'ubifs_unlink()' and 'ubifs_rmdir()' which keep working even
  30  * if they unable to allocate the budget, because deletion %-ENOSPC failure is
  31  * not what users are usually ready to get. UBIFS budgeting subsystem has some
  32  * space reserved for these purposes.
  33  *
  34  * All operations in this file write all inodes which they change straight
  35  * away, instead of marking them dirty. For example, 'ubifs_link()' changes
  36  * @i_size of the parent inode and writes the parent inode together with the
  37  * target inode. This was done to simplify file-system recovery which would
  38  * otherwise be very difficult to do. The only exception is rename which marks
  39  * the re-named inode dirty (because its @i_ctime is updated) but does not
  40  * write it, but just marks it as dirty.
  41  */
  42
  43 #include "ubifs.h"
  44
  45 /**
  46  * inherit_flags - inherit flags of the parent inode.
  47  * @dir: parent inode
  48  * @mode: new inode mode flags
  49  *
  50  * This is a helper function for 'ubifs_new_inode()' which inherits flag of the
  51  * parent directory inode @dir. UBIFS inodes inherit the following flags:
  52  * o %UBIFS_COMPR_FL, which is useful to switch compression on/of on
  53  *   sub-directory basis;
  54  * o %UBIFS_SYNC_FL - useful for the same reasons;
  55  * o %UBIFS_DIRSYNC_FL - similar, but relevant only to directories.
  56  *
  57  * This function returns the inherited flags.
  58  */
  59 static int inherit_flags(const struct inode *dir, umode_t mode)
  60 {
  61         int flags;
  62         const struct ubifs_inode *ui = ubifs_inode(dir);
  63
  64         if (!S_ISDIR(dir->i_mode))
  65                 /*
  66                  * The parent is not a directory, which means that an extended
  67                  * attribute inode is being created. No flags.
  68                  */
  69                 return 0;
  70
  71         flags = ui->flags & (UBIFS_COMPR_FL | UBIFS_SYNC_FL | UBIFS_DIRSYNC_FL);
  72         if (!S_ISDIR(mode))
  73                 /* The "DIRSYNC" flag only applies to directories */
  74                 flags &= ~UBIFS_DIRSYNC_FL;
  75         return flags;
  76 }
  77
  78 /**
  79  * ubifs_new_inode - allocate new UBIFS inode object.
  80  * @c: UBIFS file-system description object
  81  * @dir: parent directory inode
  82  * @mode: inode mode flags
  83  *
  84  * This function finds an unused inode number, allocates new inode and
  85  * initializes it. Returns new inode in case of success and an error code in
  86  * case of failure.
  87  */
  88 struct inode *ubifs_new_inode(struct ubifs_info *c, const struct inode *dir,
  89                               umode_t mode)
  90 {
  91         struct inode *inode;
  92         struct ubifs_inode *ui;
  93
  94         inode = new_inode(c->vfs_sb);
  95         ui = ubifs_inode(inode);
  96         if (!inode)
  97                 return ERR_PTR(-ENOMEM);
  98
  99         /*
 100          * Set 'S_NOCMTIME' to prevent VFS form updating [mc]time of inodes and
 101          * marking them dirty in file write path (see 'file_update_time()').
 102          * UBIFS has to fully control "clean <-> dirty" transitions of inodes
 103          * to make budgeting work.
 104          */
 105         inode->i_flags |= S_NOCMTIME;
 106
 107         inode_init_owner(inode, dir, mode);
 108         inode->i_mtime = inode->i_atime = inode->i_ctime =
 109                          ubifs_current_time(inode);
 110         inode->i_mapping->nrpages = 0;
 111
 112         switch (mode & S_IFMT) {
 113         case S_IFREG:
 114                 inode->i_mapping->a_ops = &ubifs_file_address_operations;
 115                 inode->i_op = &ubifs_file_inode_operations;
 116                 inode->i_fop = &ubifs_file_operations;
 117                 break;
 118         case S_IFDIR:
 119                 inode->i_op  = &ubifs_dir_inode_operations;
 120                 inode->i_fop = &ubifs_dir_operations;
 121                 inode->i_size = ui->ui_size = UBIFS_INO_NODE_SZ;
 122                 break;
 123         case S_IFLNK:
 124                 inode->i_op = &ubifs_symlink_inode_operations;
 125                 break;
 126         case S_IFSOCK:
 127         case S_IFIFO:
 128         case S_IFBLK:
 129         case S_IFCHR:
 130                 inode->i_op  = &ubifs_file_inode_operations;
 131                 break;
 132         default:
 133                 BUG();
 134         }
 135
 136         ui->flags = inherit_flags(dir, mode);
 137         ubifs_set_inode_flags(inode);
 138         if (S_ISREG(mode))
 139                 ui->compr_type = c->default_compr;
 140         else
 141                 ui->compr_type = UBIFS_COMPR_NONE;
 142         ui->synced_i_size = 0;
 143
 144         spin_lock(&c->cnt_lock);
 145         /* Inode number overflow is currently not supported */
 146         if (c->highest_inum >= INUM_WARN_WATERMARK) {
 147                 if (c->highest_inum >= INUM_WATERMARK) {
 148                         spin_unlock(&c->cnt_lock);
 149                         ubifs_err(c, "out of inode numbers");
 150                         make_bad_inode(inode);
 151                         iput(inode);
 152                         return ERR_PTR(-EINVAL);
 153                 }
 154                 ubifs_warn(c, "running out of inode numbers (current %lu, max %u)",
 155                            (unsigned long)c->highest_inum, INUM_WATERMARK);
 156         }
 157
 158         inode->i_ino = ++c->highest_inum;
 159         /*
 160          * The creation sequence number remains with this inode for its
 161          * lifetime. All nodes for this inode have a greater sequence number,
 162          * and so it is possible to distinguish obsolete nodes belonging to a
 163          * previous incarnation of the same inode number - for example, for the
 164          * purpose of rebuilding the index.
 165          */
 166         ui->creat_sqnum = ++c->max_sqnum;
 167         spin_unlock(&c->cnt_lock);
 168         return inode;
 169 }
 170
 171 static int dbg_check_name(const struct ubifs_info *c,
 172                           const struct ubifs_dent_node *dent,
 173                           const struct qstr *nm)
 174 {
 175         if (!dbg_is_chk_gen(c))
 176                 return 0;
 177         if (le16_to_cpu(dent->nlen) != nm->len)
 178                 return -EINVAL;
 179         if (memcmp(dent->name, nm->name, nm->len))
 180                 return -EINVAL;
 181         return 0;
 182 }
 183
 184 static struct dentry *ubifs_lookup(struct inode *dir, struct dentry *dentry,
 185                                    unsigned int flags)
 186 {
 187         int err;
 188         union ubifs_key key;
 189         struct inode *inode = NULL;
 190         struct ubifs_dent_node *dent;
 191         struct ubifs_info *c = dir->i_sb->s_fs_info;
 192
 193         dbg_gen("'%pd' in dir ino %lu", dentry, dir->i_ino);
 194
 195         if (dentry->d_name.len > UBIFS_MAX_NLEN)
 196                 return ERR_PTR(-ENAMETOOLONG);
 197
 198         dent = kmalloc(UBIFS_MAX_DENT_NODE_SZ, GFP_NOFS);
 199         if (!dent)
 200                 return ERR_PTR(-ENOMEM);
 201
 202         dent_key_init(c, &key, dir->i_ino, &dentry->d_name);
 203
 204         err = ubifs_tnc_lookup_nm(c, &key, dent, &dentry->d_name);
 205         if (err) {
 206                 if (err == -ENOENT) {
 207                         dbg_gen("not found");
 208                         goto done;
 209                 }
 210                 goto out;
 211         }
 212
 213         if (dbg_check_name(c, dent, &dentry->d_name)) {
 214                 err = -EINVAL;
 215                 goto out;
 216         }
 217
 218         inode = ubifs_iget(dir->i_sb, le64_to_cpu(dent->inum));
 219         if (IS_ERR(inode)) {
 220                 /*
 221                  * This should not happen. Probably the file-system needs
 222                  * checking.
 223                  */
 224                 err = PTR_ERR(inode);
 225                 ubifs_err(c, "dead directory entry '%pd', error %d",
 226                           dentry, err);
 227                 ubifs_ro_mode(c, err);
 228                 goto out;
 229         }
 230
 231 done:
 232         kfree(dent);
 233         /*
 234          * Note, d_splice_alias() would be required instead if we supported
 235          * NFS.
 236          */
 237         d_add(dentry, inode);
 238         return NULL;
 239
 240 out:
 241         kfree(dent);
 242         return ERR_PTR(err);
 243 }
 244
 245 static int ubifs_create(struct inode *dir, struct dentry *dentry, umode_t mode,
 246                         bool excl)
 247 {
 248         struct inode *inode;
 249         struct ubifs_info *c = dir->i_sb->s_fs_info;
 250         int err, sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 251         struct ubifs_budget_req req = { .new_ino = 1, .new_dent = 1,
 252                                         .dirtied_ino = 1 };
 253         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 254
 255         /*
 256          * Budget request settings: new inode, new direntry, changing the
 257          * parent directory inode.
 258          */
 259
 260         dbg_gen("dent '%pd', mode %#hx in dir ino %lu",
 261                 dentry, mode, dir->i_ino);
 262
 263         err = ubifs_budget_space(c, &req);
 264         if (err)
 265                 return err;
 266
 267         inode = ubifs_new_inode(c, dir, mode);
 268         if (IS_ERR(inode)) {
 269                 err = PTR_ERR(inode);
 270                 goto out_budg;
 271         }
 272
 273         err = ubifs_init_security(dir, inode, &dentry->d_name);
 274         if (err)
 275                 goto out_inode;
 276
 277         mutex_lock(&dir_ui->ui_mutex);
 278         dir->i_size += sz_change;
 279         dir_ui->ui_size = dir->i_size;
 280         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 281         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 0, 0);
 282         if (err)
 283                 goto out_cancel;
 284         mutex_unlock(&dir_ui->ui_mutex);
 285
 286         ubifs_release_budget(c, &req);
 287         insert_inode_hash(inode);
 288         d_instantiate(dentry, inode);
 289         return 0;
 290
 291 out_cancel:
 292         dir->i_size -= sz_change;
 293         dir_ui->ui_size = dir->i_size;
 294         mutex_unlock(&dir_ui->ui_mutex);
 295 out_inode:
 296         make_bad_inode(inode);
 297         iput(inode);
 298 out_budg:
 299         ubifs_release_budget(c, &req);
 300         ubifs_err(c, "cannot create regular file, error %d", err);
 301         return err;
 302 }
 303
 304 static int do_tmpfile(struct inode *dir, struct dentry *dentry,
 305                       umode_t mode, struct inode **whiteout)
 306 {
 307         struct inode *inode;
 308         struct ubifs_info *c = dir->i_sb->s_fs_info;
 309         struct ubifs_budget_req req = { .new_ino = 1, .new_dent = 1};
 310         struct ubifs_budget_req ino_req = { .dirtied_ino = 1 };
 311         struct ubifs_inode *ui, *dir_ui = ubifs_inode(dir);
 312         int err, instantiated = 0;
 313
 314         /*
 315          * Budget request settings: new dirty inode, new direntry,
 316          * budget for dirtied inode will be released via writeback.
 317          */
 318
 319         dbg_gen("dent '%pd', mode %#hx in dir ino %lu",
 320                 dentry, mode, dir->i_ino);
 321
 322         err = ubifs_budget_space(c, &req);
 323         if (err)
 324                 return err;
 325
 326         err = ubifs_budget_space(c, &ino_req);
 327         if (err) {
 328                 ubifs_release_budget(c, &req);
 329                 return err;
 330         }
 331
 332         inode = ubifs_new_inode(c, dir, mode);
 333         if (IS_ERR(inode)) {
 334                 err = PTR_ERR(inode);
 335                 goto out_budg;
 336         }
 337         ui = ubifs_inode(inode);
 338
 339         if (whiteout) {
 340                 init_special_inode(inode, inode->i_mode, WHITEOUT_DEV);
 341                 ubifs_assert(inode->i_op == &ubifs_file_inode_operations);
 342         }
 343
 344         err = ubifs_init_security(dir, inode, &dentry->d_name);
 345         if (err)
 346                 goto out_inode;
 347
 348         mutex_lock(&ui->ui_mutex);
 349         insert_inode_hash(inode);
 350
 351         if (whiteout) {
 352                 mark_inode_dirty(inode);
 353                 drop_nlink(inode);
 354                 *whiteout = inode;
 355         } else {
 356                 d_tmpfile(dentry, inode);
 357         }
 358         ubifs_assert(ui->dirty);
 359
 360         instantiated = 1;
 361         mutex_unlock(&ui->ui_mutex);
 362
 363         mutex_lock(&dir_ui->ui_mutex);
 364         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 1, 0);
 365         if (err)
 366                 goto out_cancel;
 367         mutex_unlock(&dir_ui->ui_mutex);
 368
 369         ubifs_release_budget(c, &req);
 370
 371         return 0;
 372
 373 out_cancel:
 374         mutex_unlock(&dir_ui->ui_mutex);
 375 out_inode:
 376         make_bad_inode(inode);
 377         if (!instantiated)
 378                 iput(inode);
 379 out_budg:
 380         ubifs_release_budget(c, &req);
 381         if (!instantiated)
 382                 ubifs_release_budget(c, &ino_req);
 383         ubifs_err(c, "cannot create temporary file, error %d", err);
 384         return err;
 385 }
 386
 387 static int ubifs_tmpfile(struct inode *dir, struct dentry *dentry,
 388                          umode_t mode)
 389 {
 390         return do_tmpfile(dir, dentry, mode, NULL);
 391 }
 392
 393 /**
 394  * vfs_dent_type - get VFS directory entry type.
 395  * @type: UBIFS directory entry type
 396  *
 397  * This function converts UBIFS directory entry type into VFS directory entry
 398  * type.
 399  */
 400 static unsigned int vfs_dent_type(uint8_t type)
 401 {
 402         switch (type) {
 403         case UBIFS_ITYPE_REG:
 404                 return DT_REG;
 405         case UBIFS_ITYPE_DIR:
 406                 return DT_DIR;
 407         case UBIFS_ITYPE_LNK:
 408                 return DT_LNK;
 409         case UBIFS_ITYPE_BLK:
 410                 return DT_BLK;
 411         case UBIFS_ITYPE_CHR:
 412                 return DT_CHR;
 413         case UBIFS_ITYPE_FIFO:
 414                 return DT_FIFO;
 415         case UBIFS_ITYPE_SOCK:
 416                 return DT_SOCK;
 417         default:
 418                 BUG();
 419         }
 420         return 0;
 421 }
 422
 423 /*
 424  * The classical Unix view for directory is that it is a linear array of
 425  * (name, inode number) entries. Linux/VFS assumes this model as well.
 426  * Particularly, 'readdir()' call wants us to return a directory entry offset
 427  * which later may be used to continue 'readdir()'ing the directory or to
 428  * 'seek()' to that specific direntry. Obviously UBIFS does not really fit this
 429  * model because directory entries are identified by keys, which may collide.
 430  *
 431  * UBIFS uses directory entry hash value for directory offsets, so
 432  * 'seekdir()'/'telldir()' may not always work because of possible key
 433  * collisions. But UBIFS guarantees that consecutive 'readdir()' calls work
 434  * properly by means of saving full directory entry name in the private field
 435  * of the file description object.
 436  *
 437  * This means that UBIFS cannot support NFS which requires full
 438  * 'seekdir()'/'telldir()' support.
 439  */
 440 static int ubifs_readdir(struct file *file, struct dir_context *ctx)
 441 {
 442         int err = 0;
 443         struct qstr nm;
 444         union ubifs_key key;
 445         struct ubifs_dent_node *dent;
 446         struct inode *dir = file_inode(file);
 447         struct ubifs_info *c = dir->i_sb->s_fs_info;
 448
 449         dbg_gen("dir ino %lu, f_pos %#llx", dir->i_ino, ctx->pos);
 450
 451         if (ctx->pos > UBIFS_S_KEY_HASH_MASK || ctx->pos == 2)
 452                 /*
 453                  * The directory was seek'ed to a senseless position or there
 454                  * are no more entries.
 455                  */
 456                 return 0;
 457
 458         if (file->f_version == 0) {
 459                 /*
 460                  * The file was seek'ed, which means that @file->private_data
 461                  * is now invalid. This may also be just the first
 462                  * 'ubifs_readdir()' invocation, in which case
 463                  * @file->private_data is NULL, and the below code is
 464                  * basically a no-op.
 465                  */
 466                 kfree(file->private_data);
 467                 file->private_data = NULL;
 468         }
 469
 470         /*
 471          * 'generic_file_llseek()' unconditionally sets @file->f_version to
 472          * zero, and we use this for detecting whether the file was seek'ed.
 473          */
 474         file->f_version = 1;
 475
 476         /* File positions 0 and 1 correspond to "." and ".." */
 477         if (ctx->pos < 2) {
 478                 ubifs_assert(!file->private_data);
 479                 if (!dir_emit_dots(file, ctx))
 480                         return 0;
 481
 482                 /* Find the first entry in TNC and save it */
 483                 lowest_dent_key(c, &key, dir->i_ino);
 484                 nm.name = NULL;
 485                 dent = ubifs_tnc_next_ent(c, &key, &nm);
 486                 if (IS_ERR(dent)) {
 487                         err = PTR_ERR(dent);
 488                         goto out;
 489                 }
 490
 491                 ctx->pos = key_hash_flash(c, &dent->key);
 492                 file->private_data = dent;
 493         }
 494
 495         dent = file->private_data;
 496         if (!dent) {
 497                 /*
 498                  * The directory was seek'ed to and is now readdir'ed.
 499                  * Find the entry corresponding to @ctx->pos or the closest one.
 500                  */
 501                 dent_key_init_hash(c, &key, dir->i_ino, ctx->pos);
 502                 nm.name = NULL;
 503                 dent = ubifs_tnc_next_ent(c, &key, &nm);
 504                 if (IS_ERR(dent)) {
 505                         err = PTR_ERR(dent);
 506                         goto out;
 507                 }
 508                 ctx->pos = key_hash_flash(c, &dent->key);
 509                 file->private_data = dent;
 510         }
 511
 512         while (1) {
 513                 dbg_gen("feed '%s', ino %llu, new f_pos %#x",
 514                         dent->name, (unsigned long long)le64_to_cpu(dent->inum),
 515                         key_hash_flash(c, &dent->key));
 516                 ubifs_assert(le64_to_cpu(dent->ch.sqnum) >
 517                              ubifs_inode(dir)->creat_sqnum);
 518
 519                 nm.len = le16_to_cpu(dent->nlen);
 520                 if (!dir_emit(ctx, dent->name, nm.len,
 521                                le64_to_cpu(dent->inum),
 522                                vfs_dent_type(dent->type)))
 523                         return 0;
 524
 525                 /* Switch to the next entry */
 526                 key_read(c, &dent->key, &key);
 527                 nm.name = dent->name;
 528                 dent = ubifs_tnc_next_ent(c, &key, &nm);
 529                 if (IS_ERR(dent)) {
 530                         err = PTR_ERR(dent);
 531                         goto out;
 532                 }
 533
 534                 kfree(file->private_data);
 535                 ctx->pos = key_hash_flash(c, &dent->key);
 536                 file->private_data = dent;
 537                 cond_resched();
 538         }
 539
 540 out:
 541         kfree(file->private_data);
 542         file->private_data = NULL;
 543
 544         if (err != -ENOENT)
 545                 ubifs_err(c, "cannot find next direntry, error %d", err);
 546         else
 547                 /*
 548                  * -ENOENT is a non-fatal error in this context, the TNC uses
 549                  * it to indicate that the cursor moved past the current directory
 550                  * and readdir() has to stop.
 551                  */
 552                 err = 0;
 553
 554
 555         /* 2 is a special value indicating that there are no more direntries */
 556         ctx->pos = 2;
 557         return err;
 558 }
 559
 560 /* Free saved readdir() state when the directory is closed */
 561 static int ubifs_dir_release(struct inode *dir, struct file *file)
 562 {
 563         kfree(file->private_data);
 564         file->private_data = NULL;
 565         return 0;
 566 }
 567
 568 /**
 569  * lock_2_inodes - a wrapper for locking two UBIFS inodes.
 570  * @inode1: first inode
 571  * @inode2: second inode
 572  *
 573  * We do not implement any tricks to guarantee strict lock ordering, because
 574  * VFS has already done it for us on the @i_mutex. So this is just a simple
 575  * wrapper function.
 576  */
 577 static void lock_2_inodes(struct inode *inode1, struct inode *inode2)
 578 {
 579         mutex_lock_nested(&ubifs_inode(inode1)->ui_mutex, WB_MUTEX_1);
 580         mutex_lock_nested(&ubifs_inode(inode2)->ui_mutex, WB_MUTEX_2);
 581 }
 582
 583 /**
 584  * unlock_2_inodes - a wrapper for unlocking two UBIFS inodes.
 585  * @inode1: first inode
 586  * @inode2: second inode
 587  */
 588 static void unlock_2_inodes(struct inode *inode1, struct inode *inode2)
 589 {
 590         mutex_unlock(&ubifs_inode(inode2)->ui_mutex);
 591         mutex_unlock(&ubifs_inode(inode1)->ui_mutex);
 592 }
 593
 594 static int ubifs_link(struct dentry *old_dentry, struct inode *dir,
 595                       struct dentry *dentry)
 596 {
 597         struct ubifs_info *c = dir->i_sb->s_fs_info;
 598         struct inode *inode = d_inode(old_dentry);
 599         struct ubifs_inode *ui = ubifs_inode(inode);
 600         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 601         int err, sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 602         struct ubifs_budget_req req = { .new_dent = 1, .dirtied_ino = 2,
 603                                 .dirtied_ino_d = ALIGN(ui->data_len, 8) };
 604
 605         /*
 606          * Budget request settings: new direntry, changing the target inode,
 607          * changing the parent inode.
 608          */
 609
 610         dbg_gen("dent '%pd' to ino %lu (nlink %d) in dir ino %lu",
 611                 dentry, inode->i_ino,
 612                 inode->i_nlink, dir->i_ino);
 613         ubifs_assert(inode_is_locked(dir));
 614         ubifs_assert(inode_is_locked(inode));
 615
 616         err = dbg_check_synced_i_size(c, inode);
 617         if (err)
 618                 return err;
 619
 620         err = ubifs_budget_space(c, &req);
 621         if (err)
 622                 return err;
 623
 624         lock_2_inodes(dir, inode);
 625
 626         /* Handle O_TMPFILE corner case, it is allowed to link a O_TMPFILE. */
 627         if (inode->i_nlink == 0)
 628                 ubifs_delete_orphan(c, inode->i_ino);
 629
 630         inc_nlink(inode);
 631         ihold(inode);
 632         inode->i_ctime = ubifs_current_time(inode);
 633         dir->i_size += sz_change;
 634         dir_ui->ui_size = dir->i_size;
 635         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 636         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 0, 0);
 637         if (err)
 638                 goto out_cancel;
 639         unlock_2_inodes(dir, inode);
 640
 641         ubifs_release_budget(c, &req);
 642         d_instantiate(dentry, inode);
 643         return 0;
 644
 645 out_cancel:
 646         dir->i_size -= sz_change;
 647         dir_ui->ui_size = dir->i_size;
 648         drop_nlink(inode);
 649         if (inode->i_nlink == 0)
 650                 ubifs_add_orphan(c, inode->i_ino);
 651         unlock_2_inodes(dir, inode);
 652         ubifs_release_budget(c, &req);
 653         iput(inode);
 654         return err;
 655 }
 656
 657 static int ubifs_unlink(struct inode *dir, struct dentry *dentry)
 658 {
 659         struct ubifs_info *c = dir->i_sb->s_fs_info;
 660         struct inode *inode = d_inode(dentry);
 661         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 662         int sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 663         int err, budgeted = 1;
 664         struct ubifs_budget_req req = { .mod_dent = 1, .dirtied_ino = 2 };
 665         unsigned int saved_nlink = inode->i_nlink;
 666
 667         /*
 668          * Budget request settings: deletion direntry, deletion inode (+1 for
 669          * @dirtied_ino), changing the parent directory inode. If budgeting
 670          * fails, go ahead anyway because we have extra space reserved for
 671          * deletions.
 672          */
 673
 674         dbg_gen("dent '%pd' from ino %lu (nlink %d) in dir ino %lu",
 675                 dentry, inode->i_ino,
 676                 inode->i_nlink, dir->i_ino);
 677         ubifs_assert(inode_is_locked(dir));
 678         ubifs_assert(inode_is_locked(inode));
 679         err = dbg_check_synced_i_size(c, inode);
 680         if (err)
 681                 return err;
 682
 683         err = ubifs_budget_space(c, &req);
 684         if (err) {
 685                 if (err != -ENOSPC)
 686                         return err;
 687                 budgeted = 0;
 688         }
 689
 690         lock_2_inodes(dir, inode);
 691         inode->i_ctime = ubifs_current_time(dir);
 692         drop_nlink(inode);
 693         dir->i_size -= sz_change;
 694         dir_ui->ui_size = dir->i_size;
 695         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 696         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 1, 0);
 697         if (err)
 698                 goto out_cancel;
 699         unlock_2_inodes(dir, inode);
 700
 701         if (budgeted)
 702                 ubifs_release_budget(c, &req);
 703         else {
 704                 /* We've deleted something - clean the "no space" flags */
 705                 c->bi.nospace = c->bi.nospace_rp = 0;
 706                 smp_wmb();
 707         }
 708         return 0;
 709
 710 out_cancel:
 711         dir->i_size += sz_change;
 712         dir_ui->ui_size = dir->i_size;
 713         set_nlink(inode, saved_nlink);
 714         unlock_2_inodes(dir, inode);
 715         if (budgeted)
 716                 ubifs_release_budget(c, &req);
 717         return err;
 718 }
 719
 720 /**
 721  * check_dir_empty - check if a directory is empty or not.
 722  * @c: UBIFS file-system description object
 723  * @dir: VFS inode object of the directory to check
 724  *
 725  * This function checks if directory @dir is empty. Returns zero if the
 726  * directory is empty, %-ENOTEMPTY if it is not, and other negative error codes
 727  * in case of of errors.
 728  */
 729 static int check_dir_empty(struct ubifs_info *c, struct inode *dir)
 730 {
 731         struct qstr nm = { .name = NULL };
 732         struct ubifs_dent_node *dent;
 733         union ubifs_key key;
 734         int err;
 735
 736         lowest_dent_key(c, &key, dir->i_ino);
 737         dent = ubifs_tnc_next_ent(c, &key, &nm);
 738         if (IS_ERR(dent)) {
 739                 err = PTR_ERR(dent);
 740                 if (err == -ENOENT)
 741                         err = 0;
 742         } else {
 743                 kfree(dent);
 744                 err = -ENOTEMPTY;
 745         }
 746         return err;
 747 }
 748
 749 static int ubifs_rmdir(struct inode *dir, struct dentry *dentry)
 750 {
 751         struct ubifs_info *c = dir->i_sb->s_fs_info;
 752         struct inode *inode = d_inode(dentry);
 753         int sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 754         int err, budgeted = 1;
 755         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 756         struct ubifs_budget_req req = { .mod_dent = 1, .dirtied_ino = 2 };
 757
 758         /*
 759          * Budget request settings: deletion direntry, deletion inode and
 760          * changing the parent inode. If budgeting fails, go ahead anyway
 761          * because we have extra space reserved for deletions.
 762          */
 763
 764         dbg_gen("directory '%pd', ino %lu in dir ino %lu", dentry,
 765                 inode->i_ino, dir->i_ino);
 766         ubifs_assert(inode_is_locked(dir));
 767         ubifs_assert(inode_is_locked(inode));
 768         err = check_dir_empty(c, d_inode(dentry));
 769         if (err)
 770                 return err;
 771
 772         err = ubifs_budget_space(c, &req);
 773         if (err) {
 774                 if (err != -ENOSPC)
 775                         return err;
 776                 budgeted = 0;
 777         }
 778
 779         lock_2_inodes(dir, inode);
 780         inode->i_ctime = ubifs_current_time(dir);
 781         clear_nlink(inode);
 782         drop_nlink(dir);
 783         dir->i_size -= sz_change;
 784         dir_ui->ui_size = dir->i_size;
 785         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 786         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 1, 0);
 787         if (err)
 788                 goto out_cancel;
 789         unlock_2_inodes(dir, inode);
 790
 791         if (budgeted)
 792                 ubifs_release_budget(c, &req);
 793         else {
 794                 /* We've deleted something - clean the "no space" flags */
 795                 c->bi.nospace = c->bi.nospace_rp = 0;
 796                 smp_wmb();
 797         }
 798         return 0;
 799
 800 out_cancel:
 801         dir->i_size += sz_change;
 802         dir_ui->ui_size = dir->i_size;
 803         inc_nlink(dir);
 804         set_nlink(inode, 2);
 805         unlock_2_inodes(dir, inode);
 806         if (budgeted)
 807                 ubifs_release_budget(c, &req);
 808         return err;
 809 }
 810
 811 static int ubifs_mkdir(struct inode *dir, struct dentry *dentry, umode_t mode)
 812 {
 813         struct inode *inode;
 814         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 815         struct ubifs_info *c = dir->i_sb->s_fs_info;
 816         int err, sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 817         struct ubifs_budget_req req = { .new_ino = 1, .new_dent = 1 };
 818
 819         /*
 820          * Budget request settings: new inode, new direntry and changing parent
 821          * directory inode.
 822          */
 823
 824         dbg_gen("dent '%pd', mode %#hx in dir ino %lu",
 825                 dentry, mode, dir->i_ino);
 826
 827         err = ubifs_budget_space(c, &req);
 828         if (err)
 829                 return err;
 830
 831         inode = ubifs_new_inode(c, dir, S_IFDIR | mode);
 832         if (IS_ERR(inode)) {
 833                 err = PTR_ERR(inode);
 834                 goto out_budg;
 835         }
 836
 837         err = ubifs_init_security(dir, inode, &dentry->d_name);
 838         if (err)
 839                 goto out_inode;
 840
 841         mutex_lock(&dir_ui->ui_mutex);
 842         insert_inode_hash(inode);
 843         inc_nlink(inode);
 844         inc_nlink(dir);
 845         dir->i_size += sz_change;
 846         dir_ui->ui_size = dir->i_size;
 847         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 848         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 0, 0);
 849         if (err) {
 850                 ubifs_err(c, "cannot create directory, error %d", err);
 851                 goto out_cancel;
 852         }
 853         mutex_unlock(&dir_ui->ui_mutex);
 854
 855         ubifs_release_budget(c, &req);
 856         d_instantiate(dentry, inode);
 857         return 0;
 858
 859 out_cancel:
 860         dir->i_size -= sz_change;
 861         dir_ui->ui_size = dir->i_size;
 862         drop_nlink(dir);
 863         mutex_unlock(&dir_ui->ui_mutex);
 864 out_inode:
 865         make_bad_inode(inode);
 866         iput(inode);
 867 out_budg:
 868         ubifs_release_budget(c, &req);
 869         return err;
 870 }
 871
 872 static int ubifs_mknod(struct inode *dir, struct dentry *dentry,
 873                        umode_t mode, dev_t rdev)
 874 {
 875         struct inode *inode;
 876         struct ubifs_inode *ui;
 877         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 878         struct ubifs_info *c = dir->i_sb->s_fs_info;
 879         union ubifs_dev_desc *dev = NULL;
 880         int sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 881         int err, devlen = 0;
 882         struct ubifs_budget_req req = { .new_ino = 1, .new_dent = 1,
 883                                         .new_ino_d = ALIGN(devlen, 8),
 884                                         .dirtied_ino = 1 };
 885
 886         /*
 887          * Budget request settings: new inode, new direntry and changing parent
 888          * directory inode.
 889          */
 890
 891         dbg_gen("dent '%pd' in dir ino %lu", dentry, dir->i_ino);
 892
 893         if (S_ISBLK(mode) || S_ISCHR(mode)) {
 894                 dev = kmalloc(sizeof(union ubifs_dev_desc), GFP_NOFS);
 895                 if (!dev)
 896                         return -ENOMEM;
 897                 devlen = ubifs_encode_dev(dev, rdev);
 898         }
 899
 900         err = ubifs_budget_space(c, &req);
 901         if (err) {
 902                 kfree(dev);
 903                 return err;
 904         }
 905
 906         inode = ubifs_new_inode(c, dir, mode);
 907         if (IS_ERR(inode)) {
 908                 kfree(dev);
 909                 err = PTR_ERR(inode);
 910                 goto out_budg;
 911         }
 912
 913         init_special_inode(inode, inode->i_mode, rdev);
 914         inode->i_size = ubifs_inode(inode)->ui_size = devlen;
 915         ui = ubifs_inode(inode);
 916         ui->data = dev;
 917         ui->data_len = devlen;
 918
 919         err = ubifs_init_security(dir, inode, &dentry->d_name);
 920         if (err)
 921                 goto out_inode;
 922
 923         mutex_lock(&dir_ui->ui_mutex);
 924         dir->i_size += sz_change;
 925         dir_ui->ui_size = dir->i_size;
 926         dir->i_mtime = dir->i_ctime = inode->i_ctime;
 927         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 0, 0);
 928         if (err)
 929                 goto out_cancel;
 930         mutex_unlock(&dir_ui->ui_mutex);
 931
 932         ubifs_release_budget(c, &req);
 933         insert_inode_hash(inode);
 934         d_instantiate(dentry, inode);
 935         return 0;
 936
 937 out_cancel:
 938         dir->i_size -= sz_change;
 939         dir_ui->ui_size = dir->i_size;
 940         mutex_unlock(&dir_ui->ui_mutex);
 941 out_inode:
 942         make_bad_inode(inode);
 943         iput(inode);
 944 out_budg:
 945         ubifs_release_budget(c, &req);
 946         return err;
 947 }
 948
 949 static int ubifs_symlink(struct inode *dir, struct dentry *dentry,
 950                          const char *symname)
 951 {
 952         struct inode *inode;
 953         struct ubifs_inode *ui;
 954         struct ubifs_inode *dir_ui = ubifs_inode(dir);
 955         struct ubifs_info *c = dir->i_sb->s_fs_info;
 956         int err, len = strlen(symname);
 957         int sz_change = CALC_DENT_SIZE(dentry->d_name.len);
 958         struct ubifs_budget_req req = { .new_ino = 1, .new_dent = 1,
 959                                         .new_ino_d = ALIGN(len, 8),
 960                                         .dirtied_ino = 1 };
 961
 962         /*
 963          * Budget request settings: new inode, new direntry and changing parent
 964          * directory inode.
 965          */
 966
 967         dbg_gen("dent '%pd', target '%s' in dir ino %lu", dentry,
 968                 symname, dir->i_ino);
 969
 970         if (len > UBIFS_MAX_INO_DATA)
 971                 return -ENAMETOOLONG;
 972
 973         err = ubifs_budget_space(c, &req);
 974         if (err)
 975                 return err;
 976
 977         inode = ubifs_new_inode(c, dir, S_IFLNK | S_IRWXUGO);
 978         if (IS_ERR(inode)) {
 979                 err = PTR_ERR(inode);
 980                 goto out_budg;
 981         }
 982
 983         ui = ubifs_inode(inode);
 984         ui->data = kmalloc(len + 1, GFP_NOFS);
 985         if (!ui->data) {
 986                 err = -ENOMEM;
 987                 goto out_inode;
 988         }
 989
 990         memcpy(ui->data, symname, len);
 991         ((char *)ui->data)[len] = '\0';
 992         inode->i_link = ui->data;
 993         /*
 994          * The terminating zero byte is not written to the flash media and it
 995          * is put just to make later in-memory string processing simpler. Thus,
 996          * data length is @len, not @len + %1.
 997          */
 998         ui->data_len = len;
 999         inode->i_size = ubifs_inode(inode)->ui_size = len;
1000
1001         err = ubifs_init_security(dir, inode, &dentry->d_name);
1002         if (err)
1003                 goto out_inode;
1004
1005         mutex_lock(&dir_ui->ui_mutex);
1006         dir->i_size += sz_change;
1007         dir_ui->ui_size = dir->i_size;
1008         dir->i_mtime = dir->i_ctime = inode->i_ctime;
1009         err = ubifs_jnl_update(c, dir, &dentry->d_name, inode, 0, 0);
1010         if (err)
1011                 goto out_cancel;
1012         mutex_unlock(&dir_ui->ui_mutex);
1013
1014         ubifs_release_budget(c, &req);
1015         insert_inode_hash(inode);
1016         d_instantiate(dentry, inode);
1017         return 0;
1018
1019 out_cancel:
1020         dir->i_size -= sz_change;
1021         dir_ui->ui_size = dir->i_size;
1022         mutex_unlock(&dir_ui->ui_mutex);
1023 out_inode:
1024         make_bad_inode(inode);
1025         iput(inode);
1026 out_budg:
1027         ubifs_release_budget(c, &req);
1028         return err;
1029 }
1030
1031 /**
1032  * lock_4_inodes - a wrapper for locking three UBIFS inodes.
1033  * @inode1: first inode
1034  * @inode2: second inode
1035  * @inode3: third inode
1036  * @inode4: fouth inode
1037  *
1038  * This function is used for 'ubifs_rename()' and @inode1 may be the same as
1039  * @inode2 whereas @inode3 and @inode4 may be %NULL.
1040  *
1041  * We do not implement any tricks to guarantee strict lock ordering, because
1042  * VFS has already done it for us on the @i_mutex. So this is just a simple
1043  * wrapper function.
1044  */
1045 static void lock_4_inodes(struct inode *inode1, struct inode *inode2,
1046                           struct inode *inode3, struct inode *inode4)
1047 {
1048         mutex_lock_nested(&ubifs_inode(inode1)->ui_mutex, WB_MUTEX_1);
1049         if (inode2 != inode1)
1050                 mutex_lock_nested(&ubifs_inode(inode2)->ui_mutex, WB_MUTEX_2);
1051         if (inode3)
1052                 mutex_lock_nested(&ubifs_inode(inode3)->ui_mutex, WB_MUTEX_3);
1053         if (inode4)
1054                 mutex_lock_nested(&ubifs_inode(inode4)->ui_mutex, WB_MUTEX_4);
1055 }
1056
1057 /**
1058  * unlock_4_inodes - a wrapper for unlocking three UBIFS inodes for rename.
1059  * @inode1: first inode
1060  * @inode2: second inode
1061  * @inode3: third inode
1062  * @inode4: fouth inode
1063  */
1064 static void unlock_4_inodes(struct inode *inode1, struct inode *inode2,
1065                             struct inode *inode3, struct inode *inode4)
1066 {
1067         if (inode4)
1068                 mutex_unlock(&ubifs_inode(inode4)->ui_mutex);
1069         if (inode3)
1070                 mutex_unlock(&ubifs_inode(inode3)->ui_mutex);
1071         if (inode1 != inode2)
1072                 mutex_unlock(&ubifs_inode(inode2)->ui_mutex);
1073         mutex_unlock(&ubifs_inode(inode1)->ui_mutex);
1074 }
1075
1076 static int do_rename(struct inode *old_dir, struct dentry *old_dentry,
1077                      struct inode *new_dir, struct dentry *new_dentry,
1078                      unsigned int flags)
1079 {
1080         struct ubifs_info *c = old_dir->i_sb->s_fs_info;
1081         struct inode *old_inode = d_inode(old_dentry);
1082         struct inode *new_inode = d_inode(new_dentry);
1083         struct inode *whiteout = NULL;
1084         struct ubifs_inode *old_inode_ui = ubifs_inode(old_inode);
1085         struct ubifs_inode *whiteout_ui = NULL;
1086         int err, release, sync = 0, move = (new_dir != old_dir);
1087         int is_dir = S_ISDIR(old_inode->i_mode);
1088         int unlink = !!new_inode;
1089         int new_sz = CALC_DENT_SIZE(new_dentry->d_name.len);
1090         int old_sz = CALC_DENT_SIZE(old_dentry->d_name.len);
1091         struct ubifs_budget_req req = { .new_dent = 1, .mod_dent = 1,
1092                                         .dirtied_ino = 3 };
1093         struct ubifs_budget_req ino_req = { .dirtied_ino = 1,
1094                         .dirtied_ino_d = ALIGN(old_inode_ui->data_len, 8) };
1095         struct timespec time;
1096         unsigned int uninitialized_var(saved_nlink);
1097
1098         /*
1099          * Budget request settings: deletion direntry, new direntry, removing
1100          * the old inode, and changing old and new parent directory inodes.
1101          *
1102          * However, this operation also marks the target inode as dirty and
1103          * does not write it, so we allocate budget for the target inode
1104          * separately.
1105          */
1106
1107         dbg_gen("dent '%pd' ino %lu in dir ino %lu to dent '%pd' in dir ino %lu flags 0x%x",
1108                 old_dentry, old_inode->i_ino, old_dir->i_ino,
1109                 new_dentry, new_dir->i_ino, flags);
1110
1111         if (unlink)
1112                 ubifs_assert(inode_is_locked(new_inode));
1113
1114         if (unlink && is_dir) {
1115                 err = check_dir_empty(c, new_inode);
1116                 if (err)
1117                         return err;
1118         }
1119
1120         err = ubifs_budget_space(c, &req);
1121         if (err)
1122                 return err;
1123         err = ubifs_budget_space(c, &ino_req);
1124         if (err) {
1125                 ubifs_release_budget(c, &req);
1126                 return err;
1127         }
1128
1129         if (flags & RENAME_WHITEOUT) {
1130                 union ubifs_dev_desc *dev = NULL;
1131
1132                 dev = kmalloc(sizeof(union ubifs_dev_desc), GFP_NOFS);
1133                 if (!dev) {
1134                         ubifs_release_budget(c, &req);
1135                         ubifs_release_budget(c, &ino_req);
1136                         return -ENOMEM;
1137                 }
1138
1139                 err = do_tmpfile(old_dir, old_dentry, S_IFCHR | WHITEOUT_MODE, &whiteout);
1140                 if (err) {
1141                         ubifs_release_budget(c, &req);
1142                         ubifs_release_budget(c, &ino_req);
1143                         kfree(dev);
1144                         return err;
1145                 }
1146
1147                 spin_lock(&whiteout->i_lock);
1148                 whiteout->i_state |= I_LINKABLE;
1149                 spin_unlock(&whiteout->i_lock);
1150
1151                 whiteout_ui = ubifs_inode(whiteout);
1152                 whiteout_ui->data = dev;
1153                 whiteout_ui->data_len = ubifs_encode_dev(dev, MKDEV(0, 0));
1154                 ubifs_assert(!whiteout_ui->dirty);
1155         }
1156
1157         lock_4_inodes(old_dir, new_dir, new_inode, whiteout);
1158
1159         /*
1160          * Like most other Unix systems, set the @i_ctime for inodes on a
1161          * rename.
1162          */
1163         time = ubifs_current_time(old_dir);
1164         old_inode->i_ctime = time;
1165
1166         /* We must adjust parent link count when renaming directories */
1167         if (is_dir) {
1168                 if (move) {
1169                         /*
1170                          * @old_dir loses a link because we are moving
1171                          * @old_inode to a different directory.
1172                          */
1173                         drop_nlink(old_dir);
1174                         /*
1175                          * @new_dir only gains a link if we are not also
1176                          * overwriting an existing directory.
1177                          */
1178                         if (!unlink)
1179                                 inc_nlink(new_dir);
1180                 } else {
1181                         /*
1182                          * @old_inode is not moving to a different directory,
1183                          * but @old_dir still loses a link if we are
1184                          * overwriting an existing directory.
1185                          */
1186                         if (unlink)
1187                                 drop_nlink(old_dir);
1188                 }
1189         }
1190
1191         old_dir->i_size -= old_sz;
1192         ubifs_inode(old_dir)->ui_size = old_dir->i_size;
1193         old_dir->i_mtime = old_dir->i_ctime = time;
1194         new_dir->i_mtime = new_dir->i_ctime = time;
1195
1196         /*
1197          * And finally, if we unlinked a direntry which happened to have the
1198          * same name as the moved direntry, we have to decrement @i_nlink of
1199          * the unlinked inode and change its ctime.
1200          */
1201         if (unlink) {
1202                 /*
1203                  * Directories cannot have hard-links, so if this is a
1204                  * directory, just clear @i_nlink.
1205                  */
1206                 saved_nlink = new_inode->i_nlink;
1207                 if (is_dir)
1208                         clear_nlink(new_inode);
1209                 else
1210                         drop_nlink(new_inode);
1211                 new_inode->i_ctime = time;
1212         } else {
1213                 new_dir->i_size += new_sz;
1214                 ubifs_inode(new_dir)->ui_size = new_dir->i_size;
1215         }
1216
1217         /*
1218          * Do not ask 'ubifs_jnl_rename()' to flush write-buffer if @old_inode
1219          * is dirty, because this will be done later on at the end of
1220          * 'ubifs_rename()'.
1221          */
1222         if (IS_SYNC(old_inode)) {
1223                 sync = IS_DIRSYNC(old_dir) || IS_DIRSYNC(new_dir);
1224                 if (unlink && IS_SYNC(new_inode))
1225                         sync = 1;
1226         }
1227
1228         if (whiteout) {
1229                 struct ubifs_budget_req wht_req = { .dirtied_ino = 1,
1230                                 .dirtied_ino_d = \
1231                                 ALIGN(ubifs_inode(whiteout)->data_len, 8) };
1232
1233                 err = ubifs_budget_space(c, &wht_req);
1234                 if (err) {
1235                         ubifs_release_budget(c, &req);
1236                         ubifs_release_budget(c, &ino_req);
1237                         kfree(whiteout_ui->data);
1238                         whiteout_ui->data_len = 0;
1239                         iput(whiteout);
1240                         return err;
1241                 }
1242
1243                 inc_nlink(whiteout);
1244                 mark_inode_dirty(whiteout);
1245
1246                 spin_lock(&whiteout->i_lock);
1247                 whiteout->i_state &= ~I_LINKABLE;
1248                 spin_unlock(&whiteout->i_lock);
1249
1250                 iput(whiteout);
1251         }
1252
1253         err = ubifs_jnl_rename(c, old_dir, old_dentry, new_dir, new_dentry, whiteout,
1254                                sync);
1255         if (err)
1256                 goto out_cancel;
1257
1258         unlock_4_inodes(old_dir, new_dir, new_inode, whiteout);
1259         ubifs_release_budget(c, &req);
1260
1261         mutex_lock(&old_inode_ui->ui_mutex);
1262         release = old_inode_ui->dirty;
1263         mark_inode_dirty_sync(old_inode);
1264         mutex_unlock(&old_inode_ui->ui_mutex);
1265
1266         if (release)
1267                 ubifs_release_budget(c, &ino_req);
1268         if (IS_SYNC(old_inode))
1269                 err = old_inode->i_sb->s_op->write_inode(old_inode, NULL);
1270         return err;
1271
1272 out_cancel:
1273         if (unlink) {
1274                 set_nlink(new_inode, saved_nlink);
1275         } else {
1276                 new_dir->i_size -= new_sz;
1277                 ubifs_inode(new_dir)->ui_size = new_dir->i_size;
1278         }
1279         old_dir->i_size += old_sz;
1280         ubifs_inode(old_dir)->ui_size = old_dir->i_size;
1281         if (is_dir) {
1282                 if (move) {
1283                         inc_nlink(old_dir);
1284                         if (!unlink)
1285                                 drop_nlink(new_dir);
1286                 } else {
1287                         if (unlink)
1288                                 inc_nlink(old_dir);
1289                 }
1290         }
1291         if (whiteout) {
1292                 drop_nlink(whiteout);
1293                 iput(whiteout);
1294         }
1295         unlock_4_inodes(old_dir, new_dir, new_inode, whiteout);
1296         ubifs_release_budget(c, &ino_req);
1297         ubifs_release_budget(c, &req);
1298         return err;
1299 }
1300
1301 static int ubifs_xrename(struct inode *old_dir, struct dentry *old_dentry,
1302                         struct inode *new_dir, struct dentry *new_dentry)
1303 {
1304         struct ubifs_info *c = old_dir->i_sb->s_fs_info;
1305         struct ubifs_budget_req req = { .new_dent = 1, .mod_dent = 1,
1306                                 .dirtied_ino = 2 };
1307         int sync = IS_DIRSYNC(old_dir) || IS_DIRSYNC(new_dir);
1308         struct inode *fst_inode = d_inode(old_dentry);
1309         struct inode *snd_inode = d_inode(new_dentry);
1310         struct timespec time;
1311         int err;
1312
1313         ubifs_assert(fst_inode && snd_inode);
1314
1315         lock_4_inodes(old_dir, new_dir, NULL, NULL);
1316
1317         time = ubifs_current_time(old_dir);
1318         fst_inode->i_ctime = time;
1319         snd_inode->i_ctime = time;
1320         old_dir->i_mtime = old_dir->i_ctime = time;
1321         new_dir->i_mtime = new_dir->i_ctime = time;
1322
1323         if (old_dir != new_dir) {
1324                 if (S_ISDIR(fst_inode->i_mode) && !S_ISDIR(snd_inode->i_mode)) {
1325                         inc_nlink(new_dir);
1326                         drop_nlink(old_dir);
1327                 }
1328                 else if (!S_ISDIR(fst_inode->i_mode) && S_ISDIR(snd_inode->i_mode)) {
1329                         drop_nlink(new_dir);
1330                         inc_nlink(old_dir);
1331                 }
1332         }
1333
1334         err = ubifs_jnl_xrename(c, old_dir, old_dentry, new_dir, new_dentry,
1335                                 sync);
1336
1337         unlock_4_inodes(old_dir, new_dir, NULL, NULL);
1338         ubifs_release_budget(c, &req);
1339
1340         return err;
1341 }
1342
1343 static int ubifs_rename(struct inode *old_dir, struct dentry *old_dentry,
1344                         struct inode *new_dir, struct dentry *new_dentry,
1345                         unsigned int flags)
1346 {
1347         if (flags & ~(RENAME_NOREPLACE | RENAME_WHITEOUT | RENAME_EXCHANGE))
1348                 return -EINVAL;
1349
1350         ubifs_assert(inode_is_locked(old_dir));
1351         ubifs_assert(inode_is_locked(new_dir));
1352
1353         if (flags & RENAME_EXCHANGE)
1354                 return ubifs_xrename(old_dir, old_dentry, new_dir, new_dentry);
1355
1356         return do_rename(old_dir, old_dentry, new_dir, new_dentry, flags);
1357 }
1358
1359 int ubifs_getattr(struct vfsmount *mnt, struct dentry *dentry,
1360                   struct kstat *stat)
1361 {
1362         loff_t size;
1363         struct inode *inode = d_inode(dentry);
1364         struct ubifs_inode *ui = ubifs_inode(inode);
1365
1366         mutex_lock(&ui->ui_mutex);
1367         generic_fillattr(inode, stat);
1368         stat->blksize = UBIFS_BLOCK_SIZE;
1369         stat->size = ui->ui_size;
1370
1371         /*
1372          * Unfortunately, the 'stat()' system call was designed for block
1373          * device based file systems, and it is not appropriate for UBIFS,
1374          * because UBIFS does not have notion of "block". For example, it is
1375          * difficult to tell how many block a directory takes - it actually
1376          * takes less than 300 bytes, but we have to round it to block size,
1377          * which introduces large mistake. This makes utilities like 'du' to
1378          * report completely senseless numbers. This is the reason why UBIFS
1379          * goes the same way as JFFS2 - it reports zero blocks for everything
1380          * but regular files, which makes more sense than reporting completely
1381          * wrong sizes.
1382          */
1383         if (S_ISREG(inode->i_mode)) {
1384                 size = ui->xattr_size;
1385                 size += stat->size;
1386                 size = ALIGN(size, UBIFS_BLOCK_SIZE);
1387                 /*
1388                  * Note, user-space expects 512-byte blocks count irrespectively
1389                  * of what was reported in @stat->size.
1390                  */
1391                 stat->blocks = size >> 9;
1392         } else
1393                 stat->blocks = 0;
1394         mutex_unlock(&ui->ui_mutex);
1395         return 0;
1396 }
1397
1398 const struct inode_operations ubifs_dir_inode_operations = {
1399         .lookup      = ubifs_lookup,
1400         .create      = ubifs_create,
1401         .link        = ubifs_link,
1402         .symlink     = ubifs_symlink,
1403         .unlink      = ubifs_unlink,
1404         .mkdir       = ubifs_mkdir,
1405         .rmdir       = ubifs_rmdir,
1406         .mknod       = ubifs_mknod,
1407         .rename      = ubifs_rename,
1408         .setattr     = ubifs_setattr,
1409         .getattr     = ubifs_getattr,
1410         .listxattr   = ubifs_listxattr,
1411 #ifdef CONFIG_UBIFS_ATIME_SUPPORT
1412         .update_time = ubifs_update_time,
1413 #endif
1414         .tmpfile     = ubifs_tmpfile,
1415 };
1416
1417 const struct file_operations ubifs_dir_operations = {
1418         .llseek         = generic_file_llseek,
1419         .release        = ubifs_dir_release,
1420         .read           = generic_read_dir,
1421         .iterate_shared = ubifs_readdir,
1422         .fsync          = ubifs_fsync,
1423         .unlocked_ioctl = ubifs_ioctl,
1424 #ifdef CONFIG_COMPAT
1425         .compat_ioctl   = ubifs_compat_ioctl,
1426 #endif
1427 };