fs/f2fs/dir.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * fs/f2fs/dir.c
   4  *
   5  * Copyright (c) 2012 Samsung Electronics Co., Ltd.
   6  *             http://www.samsung.com/
   7  */
   8 #include <asm/unaligned.h>
   9 #include <linux/fs.h>
  10 #include <linux/f2fs_fs.h>
  11 #include <linux/sched/signal.h>
  12 #include <linux/unicode.h>
  13 #include "f2fs.h"
  14 #include "node.h"
  15 #include "acl.h"
  16 #include "xattr.h"
  17 #include <trace/events/f2fs.h>
  18
  19 static unsigned long dir_blocks(struct inode *inode)
  20 {
  21         return ((unsigned long long) (i_size_read(inode) + PAGE_SIZE - 1))
  22                                                         >> PAGE_SHIFT;
  23 }
  24
  25 static unsigned int dir_buckets(unsigned int level, int dir_level)
  26 {
  27         if (level + dir_level < MAX_DIR_HASH_DEPTH / 2)
  28                 return 1 << (level + dir_level);
  29         else
  30                 return MAX_DIR_BUCKETS;
  31 }
  32
  33 static unsigned int bucket_blocks(unsigned int level)
  34 {
  35         if (level < MAX_DIR_HASH_DEPTH / 2)
  36                 return 2;
  37         else
  38                 return 4;
  39 }
  40
  41 static unsigned char f2fs_filetype_table[F2FS_FT_MAX] = {
  42         [F2FS_FT_UNKNOWN]       = DT_UNKNOWN,
  43         [F2FS_FT_REG_FILE]      = DT_REG,
  44         [F2FS_FT_DIR]           = DT_DIR,
  45         [F2FS_FT_CHRDEV]        = DT_CHR,
  46         [F2FS_FT_BLKDEV]        = DT_BLK,
  47         [F2FS_FT_FIFO]          = DT_FIFO,
  48         [F2FS_FT_SOCK]          = DT_SOCK,
  49         [F2FS_FT_SYMLINK]       = DT_LNK,
  50 };
  51
  52 static unsigned char f2fs_type_by_mode[S_IFMT >> S_SHIFT] = {
  53         [S_IFREG >> S_SHIFT]    = F2FS_FT_REG_FILE,
  54         [S_IFDIR >> S_SHIFT]    = F2FS_FT_DIR,
  55         [S_IFCHR >> S_SHIFT]    = F2FS_FT_CHRDEV,
  56         [S_IFBLK >> S_SHIFT]    = F2FS_FT_BLKDEV,
  57         [S_IFIFO >> S_SHIFT]    = F2FS_FT_FIFO,
  58         [S_IFSOCK >> S_SHIFT]   = F2FS_FT_SOCK,
  59         [S_IFLNK >> S_SHIFT]    = F2FS_FT_SYMLINK,
  60 };
  61
  62 static void set_de_type(struct f2fs_dir_entry *de, umode_t mode)
  63 {
  64         de->file_type = f2fs_type_by_mode[(mode & S_IFMT) >> S_SHIFT];
  65 }
  66
  67 unsigned char f2fs_get_de_type(struct f2fs_dir_entry *de)
  68 {
  69         if (de->file_type < F2FS_FT_MAX)
  70                 return f2fs_filetype_table[de->file_type];
  71         return DT_UNKNOWN;
  72 }
  73
  74 /* If @dir is casefolded, initialize @fname->cf_name from @fname->usr_fname. */
  75 int f2fs_init_casefolded_name(const struct inode *dir,
  76                               struct f2fs_filename *fname)
  77 {
  78 #ifdef CONFIG_UNICODE
  79         struct super_block *sb = dir->i_sb;
  80         struct f2fs_sb_info *sbi = F2FS_SB(sb);
  81
  82         if (IS_CASEFOLDED(dir)) {
  83                 fname->cf_name.name = f2fs_kmalloc(sbi, F2FS_NAME_LEN,
  84                                                    GFP_NOFS);
  85                 if (!fname->cf_name.name)
  86                         return -ENOMEM;
  87                 fname->cf_name.len = utf8_casefold(sb->s_encoding,
  88                                                    fname->usr_fname,
  89                                                    fname->cf_name.name,
  90                                                    F2FS_NAME_LEN);
  91                 if ((int)fname->cf_name.len <= 0) {
  92                         kfree(fname->cf_name.name);
  93                         fname->cf_name.name = NULL;
  94                         if (sb_has_strict_encoding(sb))
  95                                 return -EINVAL;
  96                         /* fall back to treating name as opaque byte sequence */
  97                 }
  98         }
  99 #endif
 100         return 0;
 101 }
 102
 103 static int __f2fs_setup_filename(const struct inode *dir,
 104                                  const struct fscrypt_name *crypt_name,
 105                                  struct f2fs_filename *fname)
 106 {
 107         int err;
 108
 109         memset(fname, 0, sizeof(*fname));
 110
 111         fname->usr_fname = crypt_name->usr_fname;
 112         fname->disk_name = crypt_name->disk_name;
 113 #ifdef CONFIG_FS_ENCRYPTION
 114         fname->crypto_buf = crypt_name->crypto_buf;
 115 #endif
 116         if (crypt_name->is_nokey_name) {
 117                 /* hash was decoded from the no-key name */
 118                 fname->hash = cpu_to_le32(crypt_name->hash);
 119         } else {
 120                 err = f2fs_init_casefolded_name(dir, fname);
 121                 if (err) {
 122                         f2fs_free_filename(fname);
 123                         return err;
 124                 }
 125                 f2fs_hash_filename(dir, fname);
 126         }
 127         return 0;
 128 }
 129
 130 /*
 131  * Prepare to search for @iname in @dir.  This is similar to
 132  * fscrypt_setup_filename(), but this also handles computing the casefolded name
 133  * and the f2fs dirhash if needed, then packing all the information about this
 134  * filename up into a 'struct f2fs_filename'.
 135  */
 136 int f2fs_setup_filename(struct inode *dir, const struct qstr *iname,
 137                         int lookup, struct f2fs_filename *fname)
 138 {
 139         struct fscrypt_name crypt_name;
 140         int err;
 141
 142         err = fscrypt_setup_filename(dir, iname, lookup, &crypt_name);
 143         if (err)
 144                 return err;
 145
 146         return __f2fs_setup_filename(dir, &crypt_name, fname);
 147 }
 148
 149 /*
 150  * Prepare to look up @dentry in @dir.  This is similar to
 151  * fscrypt_prepare_lookup(), but this also handles computing the casefolded name
 152  * and the f2fs dirhash if needed, then packing all the information about this
 153  * filename up into a 'struct f2fs_filename'.
 154  */
 155 int f2fs_prepare_lookup(struct inode *dir, struct dentry *dentry,
 156                         struct f2fs_filename *fname)
 157 {
 158         struct fscrypt_name crypt_name;
 159         int err;
 160
 161         err = fscrypt_prepare_lookup(dir, dentry, &crypt_name);
 162         if (err)
 163                 return err;
 164
 165         return __f2fs_setup_filename(dir, &crypt_name, fname);
 166 }
 167
 168 void f2fs_free_filename(struct f2fs_filename *fname)
 169 {
 170 #ifdef CONFIG_FS_ENCRYPTION
 171         kfree(fname->crypto_buf.name);
 172         fname->crypto_buf.name = NULL;
 173 #endif
 174 #ifdef CONFIG_UNICODE
 175         kfree(fname->cf_name.name);
 176         fname->cf_name.name = NULL;
 177 #endif
 178 }
 179
 180 static unsigned long dir_block_index(unsigned int level,
 181                                 int dir_level, unsigned int idx)
 182 {
 183         unsigned long i;
 184         unsigned long bidx = 0;
 185
 186         for (i = 0; i < level; i++)
 187                 bidx += dir_buckets(i, dir_level) * bucket_blocks(i);
 188         bidx += idx * bucket_blocks(level);
 189         return bidx;
 190 }
 191
 192 static struct f2fs_dir_entry *find_in_block(struct inode *dir,
 193                                 struct page *dentry_page,
 194                                 const struct f2fs_filename *fname,
 195                                 int *max_slots)
 196 {
 197         struct f2fs_dentry_block *dentry_blk;
 198         struct f2fs_dentry_ptr d;
 199
 200         dentry_blk = (struct f2fs_dentry_block *)page_address(dentry_page);
 201
 202         make_dentry_ptr_block(dir, &d, dentry_blk);
 203         return f2fs_find_target_dentry(&d, fname, max_slots);
 204 }
 205
 206 #ifdef CONFIG_UNICODE
 207 /*
 208  * Test whether a case-insensitive directory entry matches the filename
 209  * being searched for.
 210  *
 211  * Returns 1 for a match, 0 for no match, and -errno on an error.
 212  */
 213 static int f2fs_match_ci_name(const struct inode *dir, const struct qstr *name,
 214                                const u8 *de_name, u32 de_name_len)
 215 {
 216         const struct super_block *sb = dir->i_sb;
 217         const struct unicode_map *um = sb->s_encoding;
 218         struct fscrypt_str decrypted_name = FSTR_INIT(NULL, de_name_len);
 219         struct qstr entry = QSTR_INIT(de_name, de_name_len);
 220         int res;
 221
 222         if (IS_ENCRYPTED(dir)) {
 223                 const struct fscrypt_str encrypted_name =
 224                         FSTR_INIT((u8 *)de_name, de_name_len);
 225
 226                 if (WARN_ON_ONCE(!fscrypt_has_encryption_key(dir)))
 227                         return -EINVAL;
 228
 229                 decrypted_name.name = kmalloc(de_name_len, GFP_KERNEL);
 230                 if (!decrypted_name.name)
 231                         return -ENOMEM;
 232                 res = fscrypt_fname_disk_to_usr(dir, 0, 0, &encrypted_name,
 233                                                 &decrypted_name);
 234                 if (res < 0)
 235                         goto out;
 236                 entry.name = decrypted_name.name;
 237                 entry.len = decrypted_name.len;
 238         }
 239
 240         res = utf8_strncasecmp_folded(um, name, &entry);
 241         /*
 242          * In strict mode, ignore invalid names.  In non-strict mode,
 243          * fall back to treating them as opaque byte sequences.
 244          */
 245         if (res < 0 && !sb_has_strict_encoding(sb)) {
 246                 res = name->len == entry.len &&
 247                                 memcmp(name->name, entry.name, name->len) == 0;
 248         } else {
 249                 /* utf8_strncasecmp_folded returns 0 on match */
 250                 res = (res == 0);
 251         }
 252 out:
 253         kfree(decrypted_name.name);
 254         return res;
 255 }
 256 #endif /* CONFIG_UNICODE */
 257
 258 static inline int f2fs_match_name(const struct inode *dir,
 259                                    const struct f2fs_filename *fname,
 260                                    const u8 *de_name, u32 de_name_len)
 261 {
 262         struct fscrypt_name f;
 263
 264 #ifdef CONFIG_UNICODE
 265         if (fname->cf_name.name) {
 266                 struct qstr cf = FSTR_TO_QSTR(&fname->cf_name);
 267
 268                 return f2fs_match_ci_name(dir, &cf, de_name, de_name_len);
 269         }
 270 #endif
 271         f.usr_fname = fname->usr_fname;
 272         f.disk_name = fname->disk_name;
 273 #ifdef CONFIG_FS_ENCRYPTION
 274         f.crypto_buf = fname->crypto_buf;
 275 #endif
 276         return fscrypt_match_name(&f, de_name, de_name_len);
 277 }
 278
 279 struct f2fs_dir_entry *f2fs_find_target_dentry(const struct f2fs_dentry_ptr *d,
 280                         const struct f2fs_filename *fname, int *max_slots)
 281 {
 282         struct f2fs_dir_entry *de;
 283         unsigned long bit_pos = 0;
 284         int max_len = 0;
 285         int res = 0;
 286
 287         if (max_slots)
 288                 *max_slots = 0;
 289         while (bit_pos < d->max) {
 290                 if (!test_bit_le(bit_pos, d->bitmap)) {
 291                         bit_pos++;
 292                         max_len++;
 293                         continue;
 294                 }
 295
 296                 de = &d->dentry[bit_pos];
 297
 298                 if (unlikely(!de->name_len)) {
 299                         bit_pos++;
 300                         continue;
 301                 }
 302
 303                 if (de->hash_code == fname->hash) {
 304                         res = f2fs_match_name(d->inode, fname,
 305                                               d->filename[bit_pos],
 306                                               le16_to_cpu(de->name_len));
 307                         if (res < 0)
 308                                 return ERR_PTR(res);
 309                         if (res)
 310                                 goto found;
 311                 }
 312
 313                 if (max_slots && max_len > *max_slots)
 314                         *max_slots = max_len;
 315                 max_len = 0;
 316
 317                 bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
 318         }
 319
 320         de = NULL;
 321 found:
 322         if (max_slots && max_len > *max_slots)
 323                 *max_slots = max_len;
 324         return de;
 325 }
 326
 327 static struct f2fs_dir_entry *find_in_level(struct inode *dir,
 328                                         unsigned int level,
 329                                         const struct f2fs_filename *fname,
 330                                         struct page **res_page)
 331 {
 332         int s = GET_DENTRY_SLOTS(fname->disk_name.len);
 333         unsigned int nbucket, nblock;
 334         unsigned int bidx, end_block;
 335         struct page *dentry_page;
 336         struct f2fs_dir_entry *de = NULL;
 337         bool room = false;
 338         int max_slots;
 339
 340         nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
 341         nblock = bucket_blocks(level);
 342
 343         bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
 344                                le32_to_cpu(fname->hash) % nbucket);
 345         end_block = bidx + nblock;
 346
 347         for (; bidx < end_block; bidx++) {
 348                 /* no need to allocate new dentry pages to all the indices */
 349                 dentry_page = f2fs_find_data_page(dir, bidx);
 350                 if (IS_ERR(dentry_page)) {
 351                         if (PTR_ERR(dentry_page) == -ENOENT) {
 352                                 room = true;
 353                                 continue;
 354                         } else {
 355                                 *res_page = dentry_page;
 356                                 break;
 357                         }
 358                 }
 359
 360                 de = find_in_block(dir, dentry_page, fname, &max_slots);
 361                 if (IS_ERR(de)) {
 362                         *res_page = ERR_CAST(de);
 363                         de = NULL;
 364                         break;
 365                 } else if (de) {
 366                         *res_page = dentry_page;
 367                         break;
 368                 }
 369
 370                 if (max_slots >= s)
 371                         room = true;
 372                 f2fs_put_page(dentry_page, 0);
 373         }
 374
 375         if (!de && room && F2FS_I(dir)->chash != fname->hash) {
 376                 F2FS_I(dir)->chash = fname->hash;
 377                 F2FS_I(dir)->clevel = level;
 378         }
 379
 380         return de;
 381 }
 382
 383 struct f2fs_dir_entry *__f2fs_find_entry(struct inode *dir,
 384                                          const struct f2fs_filename *fname,
 385                                          struct page **res_page)
 386 {
 387         unsigned long npages = dir_blocks(dir);
 388         struct f2fs_dir_entry *de = NULL;
 389         unsigned int max_depth;
 390         unsigned int level;
 391
 392         *res_page = NULL;
 393
 394         if (f2fs_has_inline_dentry(dir)) {
 395                 de = f2fs_find_in_inline_dir(dir, fname, res_page);
 396                 goto out;
 397         }
 398
 399         if (npages == 0)
 400                 goto out;
 401
 402         max_depth = F2FS_I(dir)->i_current_depth;
 403         if (unlikely(max_depth > MAX_DIR_HASH_DEPTH)) {
 404                 f2fs_warn(F2FS_I_SB(dir), "Corrupted max_depth of %lu: %u",
 405                           dir->i_ino, max_depth);
 406                 max_depth = MAX_DIR_HASH_DEPTH;
 407                 f2fs_i_depth_write(dir, max_depth);
 408         }
 409
 410         for (level = 0; level < max_depth; level++) {
 411                 de = find_in_level(dir, level, fname, res_page);
 412                 if (de || IS_ERR(*res_page))
 413                         break;
 414         }
 415 out:
 416         /* This is to increase the speed of f2fs_create */
 417         if (!de)
 418                 F2FS_I(dir)->task = current;
 419         return de;
 420 }
 421
 422 /*
 423  * Find an entry in the specified directory with the wanted name.
 424  * It returns the page where the entry was found (as a parameter - res_page),
 425  * and the entry itself. Page is returned mapped and unlocked.
 426  * Entry is guaranteed to be valid.
 427  */
 428 struct f2fs_dir_entry *f2fs_find_entry(struct inode *dir,
 429                         const struct qstr *child, struct page **res_page)
 430 {
 431         struct f2fs_dir_entry *de = NULL;
 432         struct f2fs_filename fname;
 433         int err;
 434
 435         err = f2fs_setup_filename(dir, child, 1, &fname);
 436         if (err) {
 437                 if (err == -ENOENT)
 438                         *res_page = NULL;
 439                 else
 440                         *res_page = ERR_PTR(err);
 441                 return NULL;
 442         }
 443
 444         de = __f2fs_find_entry(dir, &fname, res_page);
 445
 446         f2fs_free_filename(&fname);
 447         return de;
 448 }
 449
 450 struct f2fs_dir_entry *f2fs_parent_dir(struct inode *dir, struct page **p)
 451 {
 452         struct qstr dotdot = QSTR_INIT("..", 2);
 453
 454         return f2fs_find_entry(dir, &dotdot, p);
 455 }
 456
 457 ino_t f2fs_inode_by_name(struct inode *dir, const struct qstr *qstr,
 458                                                         struct page **page)
 459 {
 460         ino_t res = 0;
 461         struct f2fs_dir_entry *de;
 462
 463         de = f2fs_find_entry(dir, qstr, page);
 464         if (de) {
 465                 res = le32_to_cpu(de->ino);
 466                 f2fs_put_page(*page, 0);
 467         }
 468
 469         return res;
 470 }
 471
 472 void f2fs_set_link(struct inode *dir, struct f2fs_dir_entry *de,
 473                 struct page *page, struct inode *inode)
 474 {
 475         enum page_type type = f2fs_has_inline_dentry(dir) ? NODE : DATA;
 476
 477         lock_page(page);
 478         f2fs_wait_on_page_writeback(page, type, true, true);
 479         de->ino = cpu_to_le32(inode->i_ino);
 480         set_de_type(de, inode->i_mode);
 481         set_page_dirty(page);
 482
 483         dir->i_mtime = dir->i_ctime = current_time(dir);
 484         f2fs_mark_inode_dirty_sync(dir, false);
 485         f2fs_put_page(page, 1);
 486 }
 487
 488 static void init_dent_inode(struct inode *dir, struct inode *inode,
 489                             const struct f2fs_filename *fname,
 490                             struct page *ipage)
 491 {
 492         struct f2fs_inode *ri;
 493
 494         if (!fname) /* tmpfile case? */
 495                 return;
 496
 497         f2fs_wait_on_page_writeback(ipage, NODE, true, true);
 498
 499         /* copy name info. to this inode page */
 500         ri = F2FS_INODE(ipage);
 501         ri->i_namelen = cpu_to_le32(fname->disk_name.len);
 502         memcpy(ri->i_name, fname->disk_name.name, fname->disk_name.len);
 503         if (IS_ENCRYPTED(dir)) {
 504                 file_set_enc_name(inode);
 505                 /*
 506                  * Roll-forward recovery doesn't have encryption keys available,
 507                  * so it can't compute the dirhash for encrypted+casefolded
 508                  * filenames.  Append it to i_name if possible.  Else, disable
 509                  * roll-forward recovery of the dentry (i.e., make fsync'ing the
 510                  * file force a checkpoint) by setting LOST_PINO.
 511                  */
 512                 if (IS_CASEFOLDED(dir)) {
 513                         if (fname->disk_name.len + sizeof(f2fs_hash_t) <=
 514                             F2FS_NAME_LEN)
 515                                 put_unaligned(fname->hash, (f2fs_hash_t *)
 516                                         &ri->i_name[fname->disk_name.len]);
 517                         else
 518                                 file_lost_pino(inode);
 519                 }
 520         }
 521         set_page_dirty(ipage);
 522 }
 523
 524 void f2fs_do_make_empty_dir(struct inode *inode, struct inode *parent,
 525                                         struct f2fs_dentry_ptr *d)
 526 {
 527         struct fscrypt_str dot = FSTR_INIT(".", 1);
 528         struct fscrypt_str dotdot = FSTR_INIT("..", 2);
 529
 530         /* update dirent of "." */
 531         f2fs_update_dentry(inode->i_ino, inode->i_mode, d, &dot, 0, 0);
 532
 533         /* update dirent of ".." */
 534         f2fs_update_dentry(parent->i_ino, parent->i_mode, d, &dotdot, 0, 1);
 535 }
 536
 537 static int make_empty_dir(struct inode *inode,
 538                 struct inode *parent, struct page *page)
 539 {
 540         struct page *dentry_page;
 541         struct f2fs_dentry_block *dentry_blk;
 542         struct f2fs_dentry_ptr d;
 543
 544         if (f2fs_has_inline_dentry(inode))
 545                 return f2fs_make_empty_inline_dir(inode, parent, page);
 546
 547         dentry_page = f2fs_get_new_data_page(inode, page, 0, true);
 548         if (IS_ERR(dentry_page))
 549                 return PTR_ERR(dentry_page);
 550
 551         dentry_blk = page_address(dentry_page);
 552
 553         make_dentry_ptr_block(NULL, &d, dentry_blk);
 554         f2fs_do_make_empty_dir(inode, parent, &d);
 555
 556         set_page_dirty(dentry_page);
 557         f2fs_put_page(dentry_page, 1);
 558         return 0;
 559 }
 560
 561 struct page *f2fs_init_inode_metadata(struct inode *inode, struct inode *dir,
 562                         const struct f2fs_filename *fname, struct page *dpage)
 563 {
 564         struct page *page;
 565         int err;
 566
 567         if (is_inode_flag_set(inode, FI_NEW_INODE)) {
 568                 page = f2fs_new_inode_page(inode);
 569                 if (IS_ERR(page))
 570                         return page;
 571
 572                 if (S_ISDIR(inode->i_mode)) {
 573                         /* in order to handle error case */
 574                         get_page(page);
 575                         err = make_empty_dir(inode, dir, page);
 576                         if (err) {
 577                                 lock_page(page);
 578                                 goto put_error;
 579                         }
 580                         put_page(page);
 581                 }
 582
 583                 err = f2fs_init_acl(inode, dir, page, dpage);
 584                 if (err)
 585                         goto put_error;
 586
 587                 err = f2fs_init_security(inode, dir,
 588                                          fname ? fname->usr_fname : NULL, page);
 589                 if (err)
 590                         goto put_error;
 591
 592                 if (IS_ENCRYPTED(inode)) {
 593                         err = fscrypt_set_context(inode, page);
 594                         if (err)
 595                                 goto put_error;
 596                 }
 597         } else {
 598                 page = f2fs_get_node_page(F2FS_I_SB(dir), inode->i_ino);
 599                 if (IS_ERR(page))
 600                         return page;
 601         }
 602
 603         init_dent_inode(dir, inode, fname, page);
 604
 605         /*
 606          * This file should be checkpointed during fsync.
 607          * We lost i_pino from now on.
 608          */
 609         if (is_inode_flag_set(inode, FI_INC_LINK)) {
 610                 if (!S_ISDIR(inode->i_mode))
 611                         file_lost_pino(inode);
 612                 /*
 613                  * If link the tmpfile to alias through linkat path,
 614                  * we should remove this inode from orphan list.
 615                  */
 616                 if (inode->i_nlink == 0)
 617                         f2fs_remove_orphan_inode(F2FS_I_SB(dir), inode->i_ino);
 618                 f2fs_i_links_write(inode, true);
 619         }
 620         return page;
 621
 622 put_error:
 623         clear_nlink(inode);
 624         f2fs_update_inode(inode, page);
 625         f2fs_put_page(page, 1);
 626         return ERR_PTR(err);
 627 }
 628
 629 void f2fs_update_parent_metadata(struct inode *dir, struct inode *inode,
 630                                                 unsigned int current_depth)
 631 {
 632         if (inode && is_inode_flag_set(inode, FI_NEW_INODE)) {
 633                 if (S_ISDIR(inode->i_mode))
 634                         f2fs_i_links_write(dir, true);
 635                 clear_inode_flag(inode, FI_NEW_INODE);
 636         }
 637         dir->i_mtime = dir->i_ctime = current_time(dir);
 638         f2fs_mark_inode_dirty_sync(dir, false);
 639
 640         if (F2FS_I(dir)->i_current_depth != current_depth)
 641                 f2fs_i_depth_write(dir, current_depth);
 642
 643         if (inode && is_inode_flag_set(inode, FI_INC_LINK))
 644                 clear_inode_flag(inode, FI_INC_LINK);
 645 }
 646
 647 int f2fs_room_for_filename(const void *bitmap, int slots, int max_slots)
 648 {
 649         int bit_start = 0;
 650         int zero_start, zero_end;
 651 next:
 652         zero_start = find_next_zero_bit_le(bitmap, max_slots, bit_start);
 653         if (zero_start >= max_slots)
 654                 return max_slots;
 655
 656         zero_end = find_next_bit_le(bitmap, max_slots, zero_start);
 657         if (zero_end - zero_start >= slots)
 658                 return zero_start;
 659
 660         bit_start = zero_end + 1;
 661
 662         if (zero_end + 1 >= max_slots)
 663                 return max_slots;
 664         goto next;
 665 }
 666
 667 bool f2fs_has_enough_room(struct inode *dir, struct page *ipage,
 668                           const struct f2fs_filename *fname)
 669 {
 670         struct f2fs_dentry_ptr d;
 671         unsigned int bit_pos;
 672         int slots = GET_DENTRY_SLOTS(fname->disk_name.len);
 673
 674         make_dentry_ptr_inline(dir, &d, inline_data_addr(dir, ipage));
 675
 676         bit_pos = f2fs_room_for_filename(d.bitmap, slots, d.max);
 677
 678         return bit_pos < d.max;
 679 }
 680
 681 void f2fs_update_dentry(nid_t ino, umode_t mode, struct f2fs_dentry_ptr *d,
 682                         const struct fscrypt_str *name, f2fs_hash_t name_hash,
 683                         unsigned int bit_pos)
 684 {
 685         struct f2fs_dir_entry *de;
 686         int slots = GET_DENTRY_SLOTS(name->len);
 687         int i;
 688
 689         de = &d->dentry[bit_pos];
 690         de->hash_code = name_hash;
 691         de->name_len = cpu_to_le16(name->len);
 692         memcpy(d->filename[bit_pos], name->name, name->len);
 693         de->ino = cpu_to_le32(ino);
 694         set_de_type(de, mode);
 695         for (i = 0; i < slots; i++) {
 696                 __set_bit_le(bit_pos + i, (void *)d->bitmap);
 697                 /* avoid wrong garbage data for readdir */
 698                 if (i)
 699                         (de + i)->name_len = 0;
 700         }
 701 }
 702
 703 int f2fs_add_regular_entry(struct inode *dir, const struct f2fs_filename *fname,
 704                            struct inode *inode, nid_t ino, umode_t mode)
 705 {
 706         unsigned int bit_pos;
 707         unsigned int level;
 708         unsigned int current_depth;
 709         unsigned long bidx, block;
 710         unsigned int nbucket, nblock;
 711         struct page *dentry_page = NULL;
 712         struct f2fs_dentry_block *dentry_blk = NULL;
 713         struct f2fs_dentry_ptr d;
 714         struct page *page = NULL;
 715         int slots, err = 0;
 716
 717         level = 0;
 718         slots = GET_DENTRY_SLOTS(fname->disk_name.len);
 719
 720         current_depth = F2FS_I(dir)->i_current_depth;
 721         if (F2FS_I(dir)->chash == fname->hash) {
 722                 level = F2FS_I(dir)->clevel;
 723                 F2FS_I(dir)->chash = 0;
 724         }
 725
 726 start:
 727         if (time_to_inject(F2FS_I_SB(dir), FAULT_DIR_DEPTH)) {
 728                 f2fs_show_injection_info(F2FS_I_SB(dir), FAULT_DIR_DEPTH);
 729                 return -ENOSPC;
 730         }
 731
 732         if (unlikely(current_depth == MAX_DIR_HASH_DEPTH))
 733                 return -ENOSPC;
 734
 735         /* Increase the depth, if required */
 736         if (level == current_depth)
 737                 ++current_depth;
 738
 739         nbucket = dir_buckets(level, F2FS_I(dir)->i_dir_level);
 740         nblock = bucket_blocks(level);
 741
 742         bidx = dir_block_index(level, F2FS_I(dir)->i_dir_level,
 743                                 (le32_to_cpu(fname->hash) % nbucket));
 744
 745         for (block = bidx; block <= (bidx + nblock - 1); block++) {
 746                 dentry_page = f2fs_get_new_data_page(dir, NULL, block, true);
 747                 if (IS_ERR(dentry_page))
 748                         return PTR_ERR(dentry_page);
 749
 750                 dentry_blk = page_address(dentry_page);
 751                 bit_pos = f2fs_room_for_filename(&dentry_blk->dentry_bitmap,
 752                                                 slots, NR_DENTRY_IN_BLOCK);
 753                 if (bit_pos < NR_DENTRY_IN_BLOCK)
 754                         goto add_dentry;
 755
 756                 f2fs_put_page(dentry_page, 1);
 757         }
 758
 759         /* Move to next level to find the empty slot for new dentry */
 760         ++level;
 761         goto start;
 762 add_dentry:
 763         f2fs_wait_on_page_writeback(dentry_page, DATA, true, true);
 764
 765         if (inode) {
 766                 down_write(&F2FS_I(inode)->i_sem);
 767                 page = f2fs_init_inode_metadata(inode, dir, fname, NULL);
 768                 if (IS_ERR(page)) {
 769                         err = PTR_ERR(page);
 770                         goto fail;
 771                 }
 772         }
 773
 774         make_dentry_ptr_block(NULL, &d, dentry_blk);
 775         f2fs_update_dentry(ino, mode, &d, &fname->disk_name, fname->hash,
 776                            bit_pos);
 777
 778         set_page_dirty(dentry_page);
 779
 780         if (inode) {
 781                 f2fs_i_pino_write(inode, dir->i_ino);
 782
 783                 /* synchronize inode page's data from inode cache */
 784                 if (is_inode_flag_set(inode, FI_NEW_INODE))
 785                         f2fs_update_inode(inode, page);
 786
 787                 f2fs_put_page(page, 1);
 788         }
 789
 790         f2fs_update_parent_metadata(dir, inode, current_depth);
 791 fail:
 792         if (inode)
 793                 up_write(&F2FS_I(inode)->i_sem);
 794
 795         f2fs_put_page(dentry_page, 1);
 796
 797         return err;
 798 }
 799
 800 int f2fs_add_dentry(struct inode *dir, const struct f2fs_filename *fname,
 801                     struct inode *inode, nid_t ino, umode_t mode)
 802 {
 803         int err = -EAGAIN;
 804
 805         if (f2fs_has_inline_dentry(dir))
 806                 err = f2fs_add_inline_entry(dir, fname, inode, ino, mode);
 807         if (err == -EAGAIN)
 808                 err = f2fs_add_regular_entry(dir, fname, inode, ino, mode);
 809
 810         f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
 811         return err;
 812 }
 813
 814 /*
 815  * Caller should grab and release a rwsem by calling f2fs_lock_op() and
 816  * f2fs_unlock_op().
 817  */
 818 int f2fs_do_add_link(struct inode *dir, const struct qstr *name,
 819                                 struct inode *inode, nid_t ino, umode_t mode)
 820 {
 821         struct f2fs_filename fname;
 822         struct page *page = NULL;
 823         struct f2fs_dir_entry *de = NULL;
 824         int err;
 825
 826         err = f2fs_setup_filename(dir, name, 0, &fname);
 827         if (err)
 828                 return err;
 829
 830         /*
 831          * An immature stackable filesystem shows a race condition between lookup
 832          * and create. If we have same task when doing lookup and create, it's
 833          * definitely fine as expected by VFS normally. Otherwise, let's just
 834          * verify on-disk dentry one more time, which guarantees filesystem
 835          * consistency more.
 836          */
 837         if (current != F2FS_I(dir)->task) {
 838                 de = __f2fs_find_entry(dir, &fname, &page);
 839                 F2FS_I(dir)->task = NULL;
 840         }
 841         if (de) {
 842                 f2fs_put_page(page, 0);
 843                 err = -EEXIST;
 844         } else if (IS_ERR(page)) {
 845                 err = PTR_ERR(page);
 846         } else {
 847                 err = f2fs_add_dentry(dir, &fname, inode, ino, mode);
 848         }
 849         f2fs_free_filename(&fname);
 850         return err;
 851 }
 852
 853 int f2fs_do_tmpfile(struct inode *inode, struct inode *dir)
 854 {
 855         struct page *page;
 856         int err = 0;
 857
 858         down_write(&F2FS_I(inode)->i_sem);
 859         page = f2fs_init_inode_metadata(inode, dir, NULL, NULL);
 860         if (IS_ERR(page)) {
 861                 err = PTR_ERR(page);
 862                 goto fail;
 863         }
 864         f2fs_put_page(page, 1);
 865
 866         clear_inode_flag(inode, FI_NEW_INODE);
 867         f2fs_update_time(F2FS_I_SB(inode), REQ_TIME);
 868 fail:
 869         up_write(&F2FS_I(inode)->i_sem);
 870         return err;
 871 }
 872
 873 void f2fs_drop_nlink(struct inode *dir, struct inode *inode)
 874 {
 875         struct f2fs_sb_info *sbi = F2FS_I_SB(dir);
 876
 877         down_write(&F2FS_I(inode)->i_sem);
 878
 879         if (S_ISDIR(inode->i_mode))
 880                 f2fs_i_links_write(dir, false);
 881         inode->i_ctime = current_time(inode);
 882
 883         f2fs_i_links_write(inode, false);
 884         if (S_ISDIR(inode->i_mode)) {
 885                 f2fs_i_links_write(inode, false);
 886                 f2fs_i_size_write(inode, 0);
 887         }
 888         up_write(&F2FS_I(inode)->i_sem);
 889
 890         if (inode->i_nlink == 0)
 891                 f2fs_add_orphan_inode(inode);
 892         else
 893                 f2fs_release_orphan_inode(sbi);
 894 }
 895
 896 /*
 897  * It only removes the dentry from the dentry page, corresponding name
 898  * entry in name page does not need to be touched during deletion.
 899  */
 900 void f2fs_delete_entry(struct f2fs_dir_entry *dentry, struct page *page,
 901                                         struct inode *dir, struct inode *inode)
 902 {
 903         struct  f2fs_dentry_block *dentry_blk;
 904         unsigned int bit_pos;
 905         int slots = GET_DENTRY_SLOTS(le16_to_cpu(dentry->name_len));
 906         int i;
 907
 908         f2fs_update_time(F2FS_I_SB(dir), REQ_TIME);
 909
 910         if (F2FS_OPTION(F2FS_I_SB(dir)).fsync_mode == FSYNC_MODE_STRICT)
 911                 f2fs_add_ino_entry(F2FS_I_SB(dir), dir->i_ino, TRANS_DIR_INO);
 912
 913         if (f2fs_has_inline_dentry(dir))
 914                 return f2fs_delete_inline_entry(dentry, page, dir, inode);
 915
 916         lock_page(page);
 917         f2fs_wait_on_page_writeback(page, DATA, true, true);
 918
 919         dentry_blk = page_address(page);
 920         bit_pos = dentry - dentry_blk->dentry;
 921         for (i = 0; i < slots; i++)
 922                 __clear_bit_le(bit_pos + i, &dentry_blk->dentry_bitmap);
 923
 924         /* Let's check and deallocate this dentry page */
 925         bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
 926                         NR_DENTRY_IN_BLOCK,
 927                         0);
 928         set_page_dirty(page);
 929
 930         if (bit_pos == NR_DENTRY_IN_BLOCK &&
 931                 !f2fs_truncate_hole(dir, page->index, page->index + 1)) {
 932                 f2fs_clear_page_cache_dirty_tag(page);
 933                 clear_page_dirty_for_io(page);
 934                 f2fs_clear_page_private(page);
 935                 ClearPageUptodate(page);
 936                 clear_cold_data(page);
 937                 inode_dec_dirty_pages(dir);
 938                 f2fs_remove_dirty_inode(dir);
 939         }
 940         f2fs_put_page(page, 1);
 941
 942         dir->i_ctime = dir->i_mtime = current_time(dir);
 943         f2fs_mark_inode_dirty_sync(dir, false);
 944
 945         if (inode)
 946                 f2fs_drop_nlink(dir, inode);
 947 }
 948
 949 bool f2fs_empty_dir(struct inode *dir)
 950 {
 951         unsigned long bidx;
 952         struct page *dentry_page;
 953         unsigned int bit_pos;
 954         struct f2fs_dentry_block *dentry_blk;
 955         unsigned long nblock = dir_blocks(dir);
 956
 957         if (f2fs_has_inline_dentry(dir))
 958                 return f2fs_empty_inline_dir(dir);
 959
 960         for (bidx = 0; bidx < nblock; bidx++) {
 961                 dentry_page = f2fs_get_lock_data_page(dir, bidx, false);
 962                 if (IS_ERR(dentry_page)) {
 963                         if (PTR_ERR(dentry_page) == -ENOENT)
 964                                 continue;
 965                         else
 966                                 return false;
 967                 }
 968
 969                 dentry_blk = page_address(dentry_page);
 970                 if (bidx == 0)
 971                         bit_pos = 2;
 972                 else
 973                         bit_pos = 0;
 974                 bit_pos = find_next_bit_le(&dentry_blk->dentry_bitmap,
 975                                                 NR_DENTRY_IN_BLOCK,
 976                                                 bit_pos);
 977
 978                 f2fs_put_page(dentry_page, 1);
 979
 980                 if (bit_pos < NR_DENTRY_IN_BLOCK)
 981                         return false;
 982         }
 983         return true;
 984 }
 985
 986 int f2fs_fill_dentries(struct dir_context *ctx, struct f2fs_dentry_ptr *d,
 987                         unsigned int start_pos, struct fscrypt_str *fstr)
 988 {
 989         unsigned char d_type = DT_UNKNOWN;
 990         unsigned int bit_pos;
 991         struct f2fs_dir_entry *de = NULL;
 992         struct fscrypt_str de_name = FSTR_INIT(NULL, 0);
 993         struct f2fs_sb_info *sbi = F2FS_I_SB(d->inode);
 994         struct blk_plug plug;
 995         bool readdir_ra = sbi->readdir_ra == 1;
 996         int err = 0;
 997
 998         bit_pos = ((unsigned long)ctx->pos % d->max);
 999
1000         if (readdir_ra)
1001                 blk_start_plug(&plug);
1002
1003         while (bit_pos < d->max) {
1004                 bit_pos = find_next_bit_le(d->bitmap, d->max, bit_pos);
1005                 if (bit_pos >= d->max)
1006                         break;
1007
1008                 de = &d->dentry[bit_pos];
1009                 if (de->name_len == 0) {
1010                         bit_pos++;
1011                         ctx->pos = start_pos + bit_pos;
1012                         printk_ratelimited(
1013                                 "%sF2FS-fs (%s): invalid namelen(0), ino:%u, run fsck to fix.",
1014                                 KERN_WARNING, sbi->sb->s_id,
1015                                 le32_to_cpu(de->ino));
1016                         set_sbi_flag(sbi, SBI_NEED_FSCK);
1017                         continue;
1018                 }
1019
1020                 d_type = f2fs_get_de_type(de);
1021
1022                 de_name.name = d->filename[bit_pos];
1023                 de_name.len = le16_to_cpu(de->name_len);
1024
1025                 /* check memory boundary before moving forward */
1026                 bit_pos += GET_DENTRY_SLOTS(le16_to_cpu(de->name_len));
1027                 if (unlikely(bit_pos > d->max ||
1028                                 le16_to_cpu(de->name_len) > F2FS_NAME_LEN)) {
1029                         f2fs_warn(sbi, "%s: corrupted namelen=%d, run fsck to fix.",
1030                                   __func__, le16_to_cpu(de->name_len));
1031                         set_sbi_flag(sbi, SBI_NEED_FSCK);
1032                         err = -EFSCORRUPTED;
1033                         goto out;
1034                 }
1035
1036                 if (IS_ENCRYPTED(d->inode)) {
1037                         int save_len = fstr->len;
1038
1039                         err = fscrypt_fname_disk_to_usr(d->inode,
1040                                                 (u32)le32_to_cpu(de->hash_code),
1041                                                 0, &de_name, fstr);
1042                         if (err)
1043                                 goto out;
1044
1045                         de_name = *fstr;
1046                         fstr->len = save_len;
1047                 }
1048
1049                 if (!dir_emit(ctx, de_name.name, de_name.len,
1050                                         le32_to_cpu(de->ino), d_type)) {
1051                         err = 1;
1052                         goto out;
1053                 }
1054
1055                 if (readdir_ra)
1056                         f2fs_ra_node_page(sbi, le32_to_cpu(de->ino));
1057
1058                 ctx->pos = start_pos + bit_pos;
1059         }
1060 out:
1061         if (readdir_ra)
1062                 blk_finish_plug(&plug);
1063         return err;
1064 }
1065
1066 static int f2fs_readdir(struct file *file, struct dir_context *ctx)
1067 {
1068         struct inode *inode = file_inode(file);
1069         unsigned long npages = dir_blocks(inode);
1070         struct f2fs_dentry_block *dentry_blk = NULL;
1071         struct page *dentry_page = NULL;
1072         struct file_ra_state *ra = &file->f_ra;
1073         loff_t start_pos = ctx->pos;
1074         unsigned int n = ((unsigned long)ctx->pos / NR_DENTRY_IN_BLOCK);
1075         struct f2fs_dentry_ptr d;
1076         struct fscrypt_str fstr = FSTR_INIT(NULL, 0);
1077         int err = 0;
1078
1079         if (IS_ENCRYPTED(inode)) {
1080                 err = fscrypt_prepare_readdir(inode);
1081                 if (err)
1082                         goto out;
1083
1084                 err = fscrypt_fname_alloc_buffer(F2FS_NAME_LEN, &fstr);
1085                 if (err < 0)
1086                         goto out;
1087         }
1088
1089         if (f2fs_has_inline_dentry(inode)) {
1090                 err = f2fs_read_inline_dir(file, ctx, &fstr);
1091                 goto out_free;
1092         }
1093
1094         for (; n < npages; n++, ctx->pos = n * NR_DENTRY_IN_BLOCK) {
1095
1096                 /* allow readdir() to be interrupted */
1097                 if (fatal_signal_pending(current)) {
1098                         err = -ERESTARTSYS;
1099                         goto out_free;
1100                 }
1101                 cond_resched();
1102
1103                 /* readahead for multi pages of dir */
1104                 if (npages - n > 1 && !ra_has_index(ra, n))
1105                         page_cache_sync_readahead(inode->i_mapping, ra, file, n,
1106                                 min(npages - n, (pgoff_t)MAX_DIR_RA_PAGES));
1107
1108                 dentry_page = f2fs_find_data_page(inode, n);
1109                 if (IS_ERR(dentry_page)) {
1110                         err = PTR_ERR(dentry_page);
1111                         if (err == -ENOENT) {
1112                                 err = 0;
1113                                 continue;
1114                         } else {
1115                                 goto out_free;
1116                         }
1117                 }
1118
1119                 dentry_blk = page_address(dentry_page);
1120
1121                 make_dentry_ptr_block(inode, &d, dentry_blk);
1122
1123                 err = f2fs_fill_dentries(ctx, &d,
1124                                 n * NR_DENTRY_IN_BLOCK, &fstr);
1125                 if (err) {
1126                         f2fs_put_page(dentry_page, 0);
1127                         break;
1128                 }
1129
1130                 f2fs_put_page(dentry_page, 0);
1131         }
1132 out_free:
1133         fscrypt_fname_free_buffer(&fstr);
1134 out:
1135         trace_f2fs_readdir(inode, start_pos, ctx->pos, err);
1136         return err < 0 ? err : 0;
1137 }
1138
1139 const struct file_operations f2fs_dir_operations = {
1140         .llseek         = generic_file_llseek,
1141         .read           = generic_read_dir,
1142         .iterate_shared = f2fs_readdir,
1143         .fsync          = f2fs_sync_file,
1144         .unlocked_ioctl = f2fs_ioctl,
1145 #ifdef CONFIG_COMPAT
1146         .compat_ioctl   = f2fs_compat_ioctl,
1147 #endif
1148 };