xfs: preserve DIFLAG2_NREXT64 when setting other inode attributes
[platform/kernel/linux-starfive.git] / fs / ext4 / namei.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  *  linux/fs/ext4/namei.c
4  *
5  * Copyright (C) 1992, 1993, 1994, 1995
6  * Remy Card (card@masi.ibp.fr)
7  * Laboratoire MASI - Institut Blaise Pascal
8  * Universite Pierre et Marie Curie (Paris VI)
9  *
10  *  from
11  *
12  *  linux/fs/minix/namei.c
13  *
14  *  Copyright (C) 1991, 1992  Linus Torvalds
15  *
16  *  Big-endian to little-endian byte-swapping/bitmaps by
17  *        David S. Miller (davem@caip.rutgers.edu), 1995
18  *  Directory entry file type support and forward compatibility hooks
19  *      for B-tree directories by Theodore Ts'o (tytso@mit.edu), 1998
20  *  Hash Tree Directory indexing (c)
21  *      Daniel Phillips, 2001
22  *  Hash Tree Directory indexing porting
23  *      Christopher Li, 2002
24  *  Hash Tree Directory indexing cleanup
25  *      Theodore Ts'o, 2002
26  */
27
28 #include <linux/fs.h>
29 #include <linux/pagemap.h>
30 #include <linux/time.h>
31 #include <linux/fcntl.h>
32 #include <linux/stat.h>
33 #include <linux/string.h>
34 #include <linux/quotaops.h>
35 #include <linux/buffer_head.h>
36 #include <linux/bio.h>
37 #include <linux/iversion.h>
38 #include <linux/unicode.h>
39 #include "ext4.h"
40 #include "ext4_jbd2.h"
41
42 #include "xattr.h"
43 #include "acl.h"
44
45 #include <trace/events/ext4.h>
46 /*
47  * define how far ahead to read directories while searching them.
48  */
49 #define NAMEI_RA_CHUNKS  2
50 #define NAMEI_RA_BLOCKS  4
51 #define NAMEI_RA_SIZE        (NAMEI_RA_CHUNKS * NAMEI_RA_BLOCKS)
52
53 static struct buffer_head *ext4_append(handle_t *handle,
54                                         struct inode *inode,
55                                         ext4_lblk_t *block)
56 {
57         struct buffer_head *bh;
58         int err;
59
60         if (unlikely(EXT4_SB(inode->i_sb)->s_max_dir_size_kb &&
61                      ((inode->i_size >> 10) >=
62                       EXT4_SB(inode->i_sb)->s_max_dir_size_kb)))
63                 return ERR_PTR(-ENOSPC);
64
65         *block = inode->i_size >> inode->i_sb->s_blocksize_bits;
66
67         bh = ext4_bread(handle, inode, *block, EXT4_GET_BLOCKS_CREATE);
68         if (IS_ERR(bh))
69                 return bh;
70         inode->i_size += inode->i_sb->s_blocksize;
71         EXT4_I(inode)->i_disksize = inode->i_size;
72         BUFFER_TRACE(bh, "get_write_access");
73         err = ext4_journal_get_write_access(handle, inode->i_sb, bh,
74                                             EXT4_JTR_NONE);
75         if (err) {
76                 brelse(bh);
77                 ext4_std_error(inode->i_sb, err);
78                 return ERR_PTR(err);
79         }
80         return bh;
81 }
82
83 static int ext4_dx_csum_verify(struct inode *inode,
84                                struct ext4_dir_entry *dirent);
85
86 /*
87  * Hints to ext4_read_dirblock regarding whether we expect a directory
88  * block being read to be an index block, or a block containing
89  * directory entries (and if the latter, whether it was found via a
90  * logical block in an htree index block).  This is used to control
91  * what sort of sanity checkinig ext4_read_dirblock() will do on the
92  * directory block read from the storage device.  EITHER will means
93  * the caller doesn't know what kind of directory block will be read,
94  * so no specific verification will be done.
95  */
96 typedef enum {
97         EITHER, INDEX, DIRENT, DIRENT_HTREE
98 } dirblock_type_t;
99
100 #define ext4_read_dirblock(inode, block, type) \
101         __ext4_read_dirblock((inode), (block), (type), __func__, __LINE__)
102
103 static struct buffer_head *__ext4_read_dirblock(struct inode *inode,
104                                                 ext4_lblk_t block,
105                                                 dirblock_type_t type,
106                                                 const char *func,
107                                                 unsigned int line)
108 {
109         struct buffer_head *bh;
110         struct ext4_dir_entry *dirent;
111         int is_dx_block = 0;
112
113         if (ext4_simulate_fail(inode->i_sb, EXT4_SIM_DIRBLOCK_EIO))
114                 bh = ERR_PTR(-EIO);
115         else
116                 bh = ext4_bread(NULL, inode, block, 0);
117         if (IS_ERR(bh)) {
118                 __ext4_warning(inode->i_sb, func, line,
119                                "inode #%lu: lblock %lu: comm %s: "
120                                "error %ld reading directory block",
121                                inode->i_ino, (unsigned long)block,
122                                current->comm, PTR_ERR(bh));
123
124                 return bh;
125         }
126         if (!bh && (type == INDEX || type == DIRENT_HTREE)) {
127                 ext4_error_inode(inode, func, line, block,
128                                  "Directory hole found for htree %s block",
129                                  (type == INDEX) ? "index" : "leaf");
130                 return ERR_PTR(-EFSCORRUPTED);
131         }
132         if (!bh)
133                 return NULL;
134         dirent = (struct ext4_dir_entry *) bh->b_data;
135         /* Determine whether or not we have an index block */
136         if (is_dx(inode)) {
137                 if (block == 0)
138                         is_dx_block = 1;
139                 else if (ext4_rec_len_from_disk(dirent->rec_len,
140                                                 inode->i_sb->s_blocksize) ==
141                          inode->i_sb->s_blocksize)
142                         is_dx_block = 1;
143         }
144         if (!is_dx_block && type == INDEX) {
145                 ext4_error_inode(inode, func, line, block,
146                        "directory leaf block found instead of index block");
147                 brelse(bh);
148                 return ERR_PTR(-EFSCORRUPTED);
149         }
150         if (!ext4_has_metadata_csum(inode->i_sb) ||
151             buffer_verified(bh))
152                 return bh;
153
154         /*
155          * An empty leaf block can get mistaken for a index block; for
156          * this reason, we can only check the index checksum when the
157          * caller is sure it should be an index block.
158          */
159         if (is_dx_block && type == INDEX) {
160                 if (ext4_dx_csum_verify(inode, dirent) &&
161                     !ext4_simulate_fail(inode->i_sb, EXT4_SIM_DIRBLOCK_CRC))
162                         set_buffer_verified(bh);
163                 else {
164                         ext4_error_inode_err(inode, func, line, block,
165                                              EFSBADCRC,
166                                              "Directory index failed checksum");
167                         brelse(bh);
168                         return ERR_PTR(-EFSBADCRC);
169                 }
170         }
171         if (!is_dx_block) {
172                 if (ext4_dirblock_csum_verify(inode, bh) &&
173                     !ext4_simulate_fail(inode->i_sb, EXT4_SIM_DIRBLOCK_CRC))
174                         set_buffer_verified(bh);
175                 else {
176                         ext4_error_inode_err(inode, func, line, block,
177                                              EFSBADCRC,
178                                              "Directory block failed checksum");
179                         brelse(bh);
180                         return ERR_PTR(-EFSBADCRC);
181                 }
182         }
183         return bh;
184 }
185
186 #ifdef DX_DEBUG
187 #define dxtrace(command) command
188 #else
189 #define dxtrace(command)
190 #endif
191
192 struct fake_dirent
193 {
194         __le32 inode;
195         __le16 rec_len;
196         u8 name_len;
197         u8 file_type;
198 };
199
200 struct dx_countlimit
201 {
202         __le16 limit;
203         __le16 count;
204 };
205
206 struct dx_entry
207 {
208         __le32 hash;
209         __le32 block;
210 };
211
212 /*
213  * dx_root_info is laid out so that if it should somehow get overlaid by a
214  * dirent the two low bits of the hash version will be zero.  Therefore, the
215  * hash version mod 4 should never be 0.  Sincerely, the paranoia department.
216  */
217
218 struct dx_root
219 {
220         struct fake_dirent dot;
221         char dot_name[4];
222         struct fake_dirent dotdot;
223         char dotdot_name[4];
224         struct dx_root_info
225         {
226                 __le32 reserved_zero;
227                 u8 hash_version;
228                 u8 info_length; /* 8 */
229                 u8 indirect_levels;
230                 u8 unused_flags;
231         }
232         info;
233         struct dx_entry entries[];
234 };
235
236 struct dx_node
237 {
238         struct fake_dirent fake;
239         struct dx_entry entries[];
240 };
241
242
243 struct dx_frame
244 {
245         struct buffer_head *bh;
246         struct dx_entry *entries;
247         struct dx_entry *at;
248 };
249
250 struct dx_map_entry
251 {
252         u32 hash;
253         u16 offs;
254         u16 size;
255 };
256
257 /*
258  * This goes at the end of each htree block.
259  */
260 struct dx_tail {
261         u32 dt_reserved;
262         __le32 dt_checksum;     /* crc32c(uuid+inum+dirblock) */
263 };
264
265 static inline ext4_lblk_t dx_get_block(struct dx_entry *entry);
266 static void dx_set_block(struct dx_entry *entry, ext4_lblk_t value);
267 static inline unsigned dx_get_hash(struct dx_entry *entry);
268 static void dx_set_hash(struct dx_entry *entry, unsigned value);
269 static unsigned dx_get_count(struct dx_entry *entries);
270 static unsigned dx_get_limit(struct dx_entry *entries);
271 static void dx_set_count(struct dx_entry *entries, unsigned value);
272 static void dx_set_limit(struct dx_entry *entries, unsigned value);
273 static unsigned dx_root_limit(struct inode *dir, unsigned infosize);
274 static unsigned dx_node_limit(struct inode *dir);
275 static struct dx_frame *dx_probe(struct ext4_filename *fname,
276                                  struct inode *dir,
277                                  struct dx_hash_info *hinfo,
278                                  struct dx_frame *frame);
279 static void dx_release(struct dx_frame *frames);
280 static int dx_make_map(struct inode *dir, struct buffer_head *bh,
281                        struct dx_hash_info *hinfo,
282                        struct dx_map_entry *map_tail);
283 static void dx_sort_map(struct dx_map_entry *map, unsigned count);
284 static struct ext4_dir_entry_2 *dx_move_dirents(struct inode *dir, char *from,
285                                         char *to, struct dx_map_entry *offsets,
286                                         int count, unsigned int blocksize);
287 static struct ext4_dir_entry_2 *dx_pack_dirents(struct inode *dir, char *base,
288                                                 unsigned int blocksize);
289 static void dx_insert_block(struct dx_frame *frame,
290                                         u32 hash, ext4_lblk_t block);
291 static int ext4_htree_next_block(struct inode *dir, __u32 hash,
292                                  struct dx_frame *frame,
293                                  struct dx_frame *frames,
294                                  __u32 *start_hash);
295 static struct buffer_head * ext4_dx_find_entry(struct inode *dir,
296                 struct ext4_filename *fname,
297                 struct ext4_dir_entry_2 **res_dir);
298 static int ext4_dx_add_entry(handle_t *handle, struct ext4_filename *fname,
299                              struct inode *dir, struct inode *inode);
300
301 /* checksumming functions */
302 void ext4_initialize_dirent_tail(struct buffer_head *bh,
303                                  unsigned int blocksize)
304 {
305         struct ext4_dir_entry_tail *t = EXT4_DIRENT_TAIL(bh->b_data, blocksize);
306
307         memset(t, 0, sizeof(struct ext4_dir_entry_tail));
308         t->det_rec_len = ext4_rec_len_to_disk(
309                         sizeof(struct ext4_dir_entry_tail), blocksize);
310         t->det_reserved_ft = EXT4_FT_DIR_CSUM;
311 }
312
313 /* Walk through a dirent block to find a checksum "dirent" at the tail */
314 static struct ext4_dir_entry_tail *get_dirent_tail(struct inode *inode,
315                                                    struct buffer_head *bh)
316 {
317         struct ext4_dir_entry_tail *t;
318
319 #ifdef PARANOID
320         struct ext4_dir_entry *d, *top;
321
322         d = (struct ext4_dir_entry *)bh->b_data;
323         top = (struct ext4_dir_entry *)(bh->b_data +
324                 (EXT4_BLOCK_SIZE(inode->i_sb) -
325                  sizeof(struct ext4_dir_entry_tail)));
326         while (d < top && d->rec_len)
327                 d = (struct ext4_dir_entry *)(((void *)d) +
328                     le16_to_cpu(d->rec_len));
329
330         if (d != top)
331                 return NULL;
332
333         t = (struct ext4_dir_entry_tail *)d;
334 #else
335         t = EXT4_DIRENT_TAIL(bh->b_data, EXT4_BLOCK_SIZE(inode->i_sb));
336 #endif
337
338         if (t->det_reserved_zero1 ||
339             le16_to_cpu(t->det_rec_len) != sizeof(struct ext4_dir_entry_tail) ||
340             t->det_reserved_zero2 ||
341             t->det_reserved_ft != EXT4_FT_DIR_CSUM)
342                 return NULL;
343
344         return t;
345 }
346
347 static __le32 ext4_dirblock_csum(struct inode *inode, void *dirent, int size)
348 {
349         struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);
350         struct ext4_inode_info *ei = EXT4_I(inode);
351         __u32 csum;
352
353         csum = ext4_chksum(sbi, ei->i_csum_seed, (__u8 *)dirent, size);
354         return cpu_to_le32(csum);
355 }
356
357 #define warn_no_space_for_csum(inode)                                   \
358         __warn_no_space_for_csum((inode), __func__, __LINE__)
359
360 static void __warn_no_space_for_csum(struct inode *inode, const char *func,
361                                      unsigned int line)
362 {
363         __ext4_warning_inode(inode, func, line,
364                 "No space for directory leaf checksum. Please run e2fsck -D.");
365 }
366
367 int ext4_dirblock_csum_verify(struct inode *inode, struct buffer_head *bh)
368 {
369         struct ext4_dir_entry_tail *t;
370
371         if (!ext4_has_metadata_csum(inode->i_sb))
372                 return 1;
373
374         t = get_dirent_tail(inode, bh);
375         if (!t) {
376                 warn_no_space_for_csum(inode);
377                 return 0;
378         }
379
380         if (t->det_checksum != ext4_dirblock_csum(inode, bh->b_data,
381                                                   (char *)t - bh->b_data))
382                 return 0;
383
384         return 1;
385 }
386
387 static void ext4_dirblock_csum_set(struct inode *inode,
388                                  struct buffer_head *bh)
389 {
390         struct ext4_dir_entry_tail *t;
391
392         if (!ext4_has_metadata_csum(inode->i_sb))
393                 return;
394
395         t = get_dirent_tail(inode, bh);
396         if (!t) {
397                 warn_no_space_for_csum(inode);
398                 return;
399         }
400
401         t->det_checksum = ext4_dirblock_csum(inode, bh->b_data,
402                                              (char *)t - bh->b_data);
403 }
404
405 int ext4_handle_dirty_dirblock(handle_t *handle,
406                                struct inode *inode,
407                                struct buffer_head *bh)
408 {
409         ext4_dirblock_csum_set(inode, bh);
410         return ext4_handle_dirty_metadata(handle, inode, bh);
411 }
412
413 static struct dx_countlimit *get_dx_countlimit(struct inode *inode,
414                                                struct ext4_dir_entry *dirent,
415                                                int *offset)
416 {
417         struct ext4_dir_entry *dp;
418         struct dx_root_info *root;
419         int count_offset;
420
421         if (le16_to_cpu(dirent->rec_len) == EXT4_BLOCK_SIZE(inode->i_sb))
422                 count_offset = 8;
423         else if (le16_to_cpu(dirent->rec_len) == 12) {
424                 dp = (struct ext4_dir_entry *)(((void *)dirent) + 12);
425                 if (le16_to_cpu(dp->rec_len) !=
426                     EXT4_BLOCK_SIZE(inode->i_sb) - 12)
427                         return NULL;
428                 root = (struct dx_root_info *)(((void *)dp + 12));
429                 if (root->reserved_zero ||
430                     root->info_length != sizeof(struct dx_root_info))
431                         return NULL;
432                 count_offset = 32;
433         } else
434                 return NULL;
435
436         if (offset)
437                 *offset = count_offset;
438         return (struct dx_countlimit *)(((void *)dirent) + count_offset);
439 }
440
441 static __le32 ext4_dx_csum(struct inode *inode, struct ext4_dir_entry *dirent,
442                            int count_offset, int count, struct dx_tail *t)
443 {
444         struct ext4_sb_info *sbi = EXT4_SB(inode->i_sb);
445         struct ext4_inode_info *ei = EXT4_I(inode);
446         __u32 csum;
447         int size;
448         __u32 dummy_csum = 0;
449         int offset = offsetof(struct dx_tail, dt_checksum);
450
451         size = count_offset + (count * sizeof(struct dx_entry));
452         csum = ext4_chksum(sbi, ei->i_csum_seed, (__u8 *)dirent, size);
453         csum = ext4_chksum(sbi, csum, (__u8 *)t, offset);
454         csum = ext4_chksum(sbi, csum, (__u8 *)&dummy_csum, sizeof(dummy_csum));
455
456         return cpu_to_le32(csum);
457 }
458
459 static int ext4_dx_csum_verify(struct inode *inode,
460                                struct ext4_dir_entry *dirent)
461 {
462         struct dx_countlimit *c;
463         struct dx_tail *t;
464         int count_offset, limit, count;
465
466         if (!ext4_has_metadata_csum(inode->i_sb))
467                 return 1;
468
469         c = get_dx_countlimit(inode, dirent, &count_offset);
470         if (!c) {
471                 EXT4_ERROR_INODE(inode, "dir seems corrupt?  Run e2fsck -D.");
472                 return 0;
473         }
474         limit = le16_to_cpu(c->limit);
475         count = le16_to_cpu(c->count);
476         if (count_offset + (limit * sizeof(struct dx_entry)) >
477             EXT4_BLOCK_SIZE(inode->i_sb) - sizeof(struct dx_tail)) {
478                 warn_no_space_for_csum(inode);
479                 return 0;
480         }
481         t = (struct dx_tail *)(((struct dx_entry *)c) + limit);
482
483         if (t->dt_checksum != ext4_dx_csum(inode, dirent, count_offset,
484                                             count, t))
485                 return 0;
486         return 1;
487 }
488
489 static void ext4_dx_csum_set(struct inode *inode, struct ext4_dir_entry *dirent)
490 {
491         struct dx_countlimit *c;
492         struct dx_tail *t;
493         int count_offset, limit, count;
494
495         if (!ext4_has_metadata_csum(inode->i_sb))
496                 return;
497
498         c = get_dx_countlimit(inode, dirent, &count_offset);
499         if (!c) {
500                 EXT4_ERROR_INODE(inode, "dir seems corrupt?  Run e2fsck -D.");
501                 return;
502         }
503         limit = le16_to_cpu(c->limit);
504         count = le16_to_cpu(c->count);
505         if (count_offset + (limit * sizeof(struct dx_entry)) >
506             EXT4_BLOCK_SIZE(inode->i_sb) - sizeof(struct dx_tail)) {
507                 warn_no_space_for_csum(inode);
508                 return;
509         }
510         t = (struct dx_tail *)(((struct dx_entry *)c) + limit);
511
512         t->dt_checksum = ext4_dx_csum(inode, dirent, count_offset, count, t);
513 }
514
515 static inline int ext4_handle_dirty_dx_node(handle_t *handle,
516                                             struct inode *inode,
517                                             struct buffer_head *bh)
518 {
519         ext4_dx_csum_set(inode, (struct ext4_dir_entry *)bh->b_data);
520         return ext4_handle_dirty_metadata(handle, inode, bh);
521 }
522
523 /*
524  * p is at least 6 bytes before the end of page
525  */
526 static inline struct ext4_dir_entry_2 *
527 ext4_next_entry(struct ext4_dir_entry_2 *p, unsigned long blocksize)
528 {
529         return (struct ext4_dir_entry_2 *)((char *)p +
530                 ext4_rec_len_from_disk(p->rec_len, blocksize));
531 }
532
533 /*
534  * Future: use high four bits of block for coalesce-on-delete flags
535  * Mask them off for now.
536  */
537
538 static inline ext4_lblk_t dx_get_block(struct dx_entry *entry)
539 {
540         return le32_to_cpu(entry->block) & 0x0fffffff;
541 }
542
543 static inline void dx_set_block(struct dx_entry *entry, ext4_lblk_t value)
544 {
545         entry->block = cpu_to_le32(value);
546 }
547
548 static inline unsigned dx_get_hash(struct dx_entry *entry)
549 {
550         return le32_to_cpu(entry->hash);
551 }
552
553 static inline void dx_set_hash(struct dx_entry *entry, unsigned value)
554 {
555         entry->hash = cpu_to_le32(value);
556 }
557
558 static inline unsigned dx_get_count(struct dx_entry *entries)
559 {
560         return le16_to_cpu(((struct dx_countlimit *) entries)->count);
561 }
562
563 static inline unsigned dx_get_limit(struct dx_entry *entries)
564 {
565         return le16_to_cpu(((struct dx_countlimit *) entries)->limit);
566 }
567
568 static inline void dx_set_count(struct dx_entry *entries, unsigned value)
569 {
570         ((struct dx_countlimit *) entries)->count = cpu_to_le16(value);
571 }
572
573 static inline void dx_set_limit(struct dx_entry *entries, unsigned value)
574 {
575         ((struct dx_countlimit *) entries)->limit = cpu_to_le16(value);
576 }
577
578 static inline unsigned dx_root_limit(struct inode *dir, unsigned infosize)
579 {
580         unsigned int entry_space = dir->i_sb->s_blocksize -
581                         ext4_dir_rec_len(1, NULL) -
582                         ext4_dir_rec_len(2, NULL) - infosize;
583
584         if (ext4_has_metadata_csum(dir->i_sb))
585                 entry_space -= sizeof(struct dx_tail);
586         return entry_space / sizeof(struct dx_entry);
587 }
588
589 static inline unsigned dx_node_limit(struct inode *dir)
590 {
591         unsigned int entry_space = dir->i_sb->s_blocksize -
592                         ext4_dir_rec_len(0, dir);
593
594         if (ext4_has_metadata_csum(dir->i_sb))
595                 entry_space -= sizeof(struct dx_tail);
596         return entry_space / sizeof(struct dx_entry);
597 }
598
599 /*
600  * Debug
601  */
602 #ifdef DX_DEBUG
603 static void dx_show_index(char * label, struct dx_entry *entries)
604 {
605         int i, n = dx_get_count (entries);
606         printk(KERN_DEBUG "%s index", label);
607         for (i = 0; i < n; i++) {
608                 printk(KERN_CONT " %x->%lu",
609                        i ? dx_get_hash(entries + i) : 0,
610                        (unsigned long)dx_get_block(entries + i));
611         }
612         printk(KERN_CONT "\n");
613 }
614
615 struct stats
616 {
617         unsigned names;
618         unsigned space;
619         unsigned bcount;
620 };
621
622 static struct stats dx_show_leaf(struct inode *dir,
623                                 struct dx_hash_info *hinfo,
624                                 struct ext4_dir_entry_2 *de,
625                                 int size, int show_names)
626 {
627         unsigned names = 0, space = 0;
628         char *base = (char *) de;
629         struct dx_hash_info h = *hinfo;
630
631         printk("names: ");
632         while ((char *) de < base + size)
633         {
634                 if (de->inode)
635                 {
636                         if (show_names)
637                         {
638 #ifdef CONFIG_FS_ENCRYPTION
639                                 int len;
640                                 char *name;
641                                 struct fscrypt_str fname_crypto_str =
642                                         FSTR_INIT(NULL, 0);
643                                 int res = 0;
644
645                                 name  = de->name;
646                                 len = de->name_len;
647                                 if (!IS_ENCRYPTED(dir)) {
648                                         /* Directory is not encrypted */
649                                         ext4fs_dirhash(dir, de->name,
650                                                 de->name_len, &h);
651                                         printk("%*.s:(U)%x.%u ", len,
652                                                name, h.hash,
653                                                (unsigned) ((char *) de
654                                                            - base));
655                                 } else {
656                                         struct fscrypt_str de_name =
657                                                 FSTR_INIT(name, len);
658
659                                         /* Directory is encrypted */
660                                         res = fscrypt_fname_alloc_buffer(
661                                                 len, &fname_crypto_str);
662                                         if (res)
663                                                 printk(KERN_WARNING "Error "
664                                                         "allocating crypto "
665                                                         "buffer--skipping "
666                                                         "crypto\n");
667                                         res = fscrypt_fname_disk_to_usr(dir,
668                                                 0, 0, &de_name,
669                                                 &fname_crypto_str);
670                                         if (res) {
671                                                 printk(KERN_WARNING "Error "
672                                                         "converting filename "
673                                                         "from disk to usr"
674                                                         "\n");
675                                                 name = "??";
676                                                 len = 2;
677                                         } else {
678                                                 name = fname_crypto_str.name;
679                                                 len = fname_crypto_str.len;
680                                         }
681                                         if (IS_CASEFOLDED(dir))
682                                                 h.hash = EXT4_DIRENT_HASH(de);
683                                         else
684                                                 ext4fs_dirhash(dir, de->name,
685                                                        de->name_len, &h);
686                                         printk("%*.s:(E)%x.%u ", len, name,
687                                                h.hash, (unsigned) ((char *) de
688                                                                    - base));
689                                         fscrypt_fname_free_buffer(
690                                                         &fname_crypto_str);
691                                 }
692 #else
693                                 int len = de->name_len;
694                                 char *name = de->name;
695                                 ext4fs_dirhash(dir, de->name, de->name_len, &h);
696                                 printk("%*.s:%x.%u ", len, name, h.hash,
697                                        (unsigned) ((char *) de - base));
698 #endif
699                         }
700                         space += ext4_dir_rec_len(de->name_len, dir);
701                         names++;
702                 }
703                 de = ext4_next_entry(de, size);
704         }
705         printk(KERN_CONT "(%i)\n", names);
706         return (struct stats) { names, space, 1 };
707 }
708
709 struct stats dx_show_entries(struct dx_hash_info *hinfo, struct inode *dir,
710                              struct dx_entry *entries, int levels)
711 {
712         unsigned blocksize = dir->i_sb->s_blocksize;
713         unsigned count = dx_get_count(entries), names = 0, space = 0, i;
714         unsigned bcount = 0;
715         struct buffer_head *bh;
716         printk("%i indexed blocks...\n", count);
717         for (i = 0; i < count; i++, entries++)
718         {
719                 ext4_lblk_t block = dx_get_block(entries);
720                 ext4_lblk_t hash  = i ? dx_get_hash(entries): 0;
721                 u32 range = i < count - 1? (dx_get_hash(entries + 1) - hash): ~hash;
722                 struct stats stats;
723                 printk("%s%3u:%03u hash %8x/%8x ",levels?"":"   ", i, block, hash, range);
724                 bh = ext4_bread(NULL,dir, block, 0);
725                 if (!bh || IS_ERR(bh))
726                         continue;
727                 stats = levels?
728                    dx_show_entries(hinfo, dir, ((struct dx_node *) bh->b_data)->entries, levels - 1):
729                    dx_show_leaf(dir, hinfo, (struct ext4_dir_entry_2 *)
730                         bh->b_data, blocksize, 0);
731                 names += stats.names;
732                 space += stats.space;
733                 bcount += stats.bcount;
734                 brelse(bh);
735         }
736         if (bcount)
737                 printk(KERN_DEBUG "%snames %u, fullness %u (%u%%)\n",
738                        levels ? "" : "   ", names, space/bcount,
739                        (space/bcount)*100/blocksize);
740         return (struct stats) { names, space, bcount};
741 }
742
743 /*
744  * Linear search cross check
745  */
746 static inline void htree_rep_invariant_check(struct dx_entry *at,
747                                              struct dx_entry *target,
748                                              u32 hash, unsigned int n)
749 {
750         while (n--) {
751                 dxtrace(printk(KERN_CONT ","));
752                 if (dx_get_hash(++at) > hash) {
753                         at--;
754                         break;
755                 }
756         }
757         ASSERT(at == target - 1);
758 }
759 #else /* DX_DEBUG */
760 static inline void htree_rep_invariant_check(struct dx_entry *at,
761                                              struct dx_entry *target,
762                                              u32 hash, unsigned int n)
763 {
764 }
765 #endif /* DX_DEBUG */
766
767 /*
768  * Probe for a directory leaf block to search.
769  *
770  * dx_probe can return ERR_BAD_DX_DIR, which means there was a format
771  * error in the directory index, and the caller should fall back to
772  * searching the directory normally.  The callers of dx_probe **MUST**
773  * check for this error code, and make sure it never gets reflected
774  * back to userspace.
775  */
776 static struct dx_frame *
777 dx_probe(struct ext4_filename *fname, struct inode *dir,
778          struct dx_hash_info *hinfo, struct dx_frame *frame_in)
779 {
780         unsigned count, indirect, level, i;
781         struct dx_entry *at, *entries, *p, *q, *m;
782         struct dx_root *root;
783         struct dx_frame *frame = frame_in;
784         struct dx_frame *ret_err = ERR_PTR(ERR_BAD_DX_DIR);
785         u32 hash;
786         ext4_lblk_t block;
787         ext4_lblk_t blocks[EXT4_HTREE_LEVEL];
788
789         memset(frame_in, 0, EXT4_HTREE_LEVEL * sizeof(frame_in[0]));
790         frame->bh = ext4_read_dirblock(dir, 0, INDEX);
791         if (IS_ERR(frame->bh))
792                 return (struct dx_frame *) frame->bh;
793
794         root = (struct dx_root *) frame->bh->b_data;
795         if (root->info.hash_version != DX_HASH_TEA &&
796             root->info.hash_version != DX_HASH_HALF_MD4 &&
797             root->info.hash_version != DX_HASH_LEGACY &&
798             root->info.hash_version != DX_HASH_SIPHASH) {
799                 ext4_warning_inode(dir, "Unrecognised inode hash code %u",
800                                    root->info.hash_version);
801                 goto fail;
802         }
803         if (ext4_hash_in_dirent(dir)) {
804                 if (root->info.hash_version != DX_HASH_SIPHASH) {
805                         ext4_warning_inode(dir,
806                                 "Hash in dirent, but hash is not SIPHASH");
807                         goto fail;
808                 }
809         } else {
810                 if (root->info.hash_version == DX_HASH_SIPHASH) {
811                         ext4_warning_inode(dir,
812                                 "Hash code is SIPHASH, but hash not in dirent");
813                         goto fail;
814                 }
815         }
816         if (fname)
817                 hinfo = &fname->hinfo;
818         hinfo->hash_version = root->info.hash_version;
819         if (hinfo->hash_version <= DX_HASH_TEA)
820                 hinfo->hash_version += EXT4_SB(dir->i_sb)->s_hash_unsigned;
821         hinfo->seed = EXT4_SB(dir->i_sb)->s_hash_seed;
822         /* hash is already computed for encrypted casefolded directory */
823         if (fname && fname_name(fname) &&
824                                 !(IS_ENCRYPTED(dir) && IS_CASEFOLDED(dir)))
825                 ext4fs_dirhash(dir, fname_name(fname), fname_len(fname), hinfo);
826         hash = hinfo->hash;
827
828         if (root->info.unused_flags & 1) {
829                 ext4_warning_inode(dir, "Unimplemented hash flags: %#06x",
830                                    root->info.unused_flags);
831                 goto fail;
832         }
833
834         indirect = root->info.indirect_levels;
835         if (indirect >= ext4_dir_htree_level(dir->i_sb)) {
836                 ext4_warning(dir->i_sb,
837                              "Directory (ino: %lu) htree depth %#06x exceed"
838                              "supported value", dir->i_ino,
839                              ext4_dir_htree_level(dir->i_sb));
840                 if (ext4_dir_htree_level(dir->i_sb) < EXT4_HTREE_LEVEL) {
841                         ext4_warning(dir->i_sb, "Enable large directory "
842                                                 "feature to access it");
843                 }
844                 goto fail;
845         }
846
847         entries = (struct dx_entry *)(((char *)&root->info) +
848                                       root->info.info_length);
849
850         if (dx_get_limit(entries) != dx_root_limit(dir,
851                                                    root->info.info_length)) {
852                 ext4_warning_inode(dir, "dx entry: limit %u != root limit %u",
853                                    dx_get_limit(entries),
854                                    dx_root_limit(dir, root->info.info_length));
855                 goto fail;
856         }
857
858         dxtrace(printk("Look up %x", hash));
859         level = 0;
860         blocks[0] = 0;
861         while (1) {
862                 count = dx_get_count(entries);
863                 if (!count || count > dx_get_limit(entries)) {
864                         ext4_warning_inode(dir,
865                                            "dx entry: count %u beyond limit %u",
866                                            count, dx_get_limit(entries));
867                         goto fail;
868                 }
869
870                 p = entries + 1;
871                 q = entries + count - 1;
872                 while (p <= q) {
873                         m = p + (q - p) / 2;
874                         dxtrace(printk(KERN_CONT "."));
875                         if (dx_get_hash(m) > hash)
876                                 q = m - 1;
877                         else
878                                 p = m + 1;
879                 }
880
881                 htree_rep_invariant_check(entries, p, hash, count - 1);
882
883                 at = p - 1;
884                 dxtrace(printk(KERN_CONT " %x->%u\n",
885                                at == entries ? 0 : dx_get_hash(at),
886                                dx_get_block(at)));
887                 frame->entries = entries;
888                 frame->at = at;
889
890                 block = dx_get_block(at);
891                 for (i = 0; i <= level; i++) {
892                         if (blocks[i] == block) {
893                                 ext4_warning_inode(dir,
894                                         "dx entry: tree cycle block %u points back to block %u",
895                                         blocks[level], block);
896                                 goto fail;
897                         }
898                 }
899                 if (++level > indirect)
900                         return frame;
901                 blocks[level] = block;
902                 frame++;
903                 frame->bh = ext4_read_dirblock(dir, block, INDEX);
904                 if (IS_ERR(frame->bh)) {
905                         ret_err = (struct dx_frame *) frame->bh;
906                         frame->bh = NULL;
907                         goto fail;
908                 }
909
910                 entries = ((struct dx_node *) frame->bh->b_data)->entries;
911
912                 if (dx_get_limit(entries) != dx_node_limit(dir)) {
913                         ext4_warning_inode(dir,
914                                 "dx entry: limit %u != node limit %u",
915                                 dx_get_limit(entries), dx_node_limit(dir));
916                         goto fail;
917                 }
918         }
919 fail:
920         while (frame >= frame_in) {
921                 brelse(frame->bh);
922                 frame--;
923         }
924
925         if (ret_err == ERR_PTR(ERR_BAD_DX_DIR))
926                 ext4_warning_inode(dir,
927                         "Corrupt directory, running e2fsck is recommended");
928         return ret_err;
929 }
930
931 static void dx_release(struct dx_frame *frames)
932 {
933         struct dx_root_info *info;
934         int i;
935         unsigned int indirect_levels;
936
937         if (frames[0].bh == NULL)
938                 return;
939
940         info = &((struct dx_root *)frames[0].bh->b_data)->info;
941         /* save local copy, "info" may be freed after brelse() */
942         indirect_levels = info->indirect_levels;
943         for (i = 0; i <= indirect_levels; i++) {
944                 if (frames[i].bh == NULL)
945                         break;
946                 brelse(frames[i].bh);
947                 frames[i].bh = NULL;
948         }
949 }
950
951 /*
952  * This function increments the frame pointer to search the next leaf
953  * block, and reads in the necessary intervening nodes if the search
954  * should be necessary.  Whether or not the search is necessary is
955  * controlled by the hash parameter.  If the hash value is even, then
956  * the search is only continued if the next block starts with that
957  * hash value.  This is used if we are searching for a specific file.
958  *
959  * If the hash value is HASH_NB_ALWAYS, then always go to the next block.
960  *
961  * This function returns 1 if the caller should continue to search,
962  * or 0 if it should not.  If there is an error reading one of the
963  * index blocks, it will a negative error code.
964  *
965  * If start_hash is non-null, it will be filled in with the starting
966  * hash of the next page.
967  */
968 static int ext4_htree_next_block(struct inode *dir, __u32 hash,
969                                  struct dx_frame *frame,
970                                  struct dx_frame *frames,
971                                  __u32 *start_hash)
972 {
973         struct dx_frame *p;
974         struct buffer_head *bh;
975         int num_frames = 0;
976         __u32 bhash;
977
978         p = frame;
979         /*
980          * Find the next leaf page by incrementing the frame pointer.
981          * If we run out of entries in the interior node, loop around and
982          * increment pointer in the parent node.  When we break out of
983          * this loop, num_frames indicates the number of interior
984          * nodes need to be read.
985          */
986         while (1) {
987                 if (++(p->at) < p->entries + dx_get_count(p->entries))
988                         break;
989                 if (p == frames)
990                         return 0;
991                 num_frames++;
992                 p--;
993         }
994
995         /*
996          * If the hash is 1, then continue only if the next page has a
997          * continuation hash of any value.  This is used for readdir
998          * handling.  Otherwise, check to see if the hash matches the
999          * desired continuation hash.  If it doesn't, return since
1000          * there's no point to read in the successive index pages.
1001          */
1002         bhash = dx_get_hash(p->at);
1003         if (start_hash)
1004                 *start_hash = bhash;
1005         if ((hash & 1) == 0) {
1006                 if ((bhash & ~1) != hash)
1007                         return 0;
1008         }
1009         /*
1010          * If the hash is HASH_NB_ALWAYS, we always go to the next
1011          * block so no check is necessary
1012          */
1013         while (num_frames--) {
1014                 bh = ext4_read_dirblock(dir, dx_get_block(p->at), INDEX);
1015                 if (IS_ERR(bh))
1016                         return PTR_ERR(bh);
1017                 p++;
1018                 brelse(p->bh);
1019                 p->bh = bh;
1020                 p->at = p->entries = ((struct dx_node *) bh->b_data)->entries;
1021         }
1022         return 1;
1023 }
1024
1025
1026 /*
1027  * This function fills a red-black tree with information from a
1028  * directory block.  It returns the number directory entries loaded
1029  * into the tree.  If there is an error it is returned in err.
1030  */
1031 static int htree_dirblock_to_tree(struct file *dir_file,
1032                                   struct inode *dir, ext4_lblk_t block,
1033                                   struct dx_hash_info *hinfo,
1034                                   __u32 start_hash, __u32 start_minor_hash)
1035 {
1036         struct buffer_head *bh;
1037         struct ext4_dir_entry_2 *de, *top;
1038         int err = 0, count = 0;
1039         struct fscrypt_str fname_crypto_str = FSTR_INIT(NULL, 0), tmp_str;
1040         int csum = ext4_has_metadata_csum(dir->i_sb);
1041
1042         dxtrace(printk(KERN_INFO "In htree dirblock_to_tree: block %lu\n",
1043                                                         (unsigned long)block));
1044         bh = ext4_read_dirblock(dir, block, DIRENT_HTREE);
1045         if (IS_ERR(bh))
1046                 return PTR_ERR(bh);
1047
1048         de = (struct ext4_dir_entry_2 *) bh->b_data;
1049         /* csum entries are not larger in the casefolded encrypted case */
1050         top = (struct ext4_dir_entry_2 *) ((char *) de +
1051                                            dir->i_sb->s_blocksize -
1052                                            ext4_dir_rec_len(0,
1053                                                            csum ? NULL : dir));
1054         /* Check if the directory is encrypted */
1055         if (IS_ENCRYPTED(dir)) {
1056                 err = fscrypt_prepare_readdir(dir);
1057                 if (err < 0) {
1058                         brelse(bh);
1059                         return err;
1060                 }
1061                 err = fscrypt_fname_alloc_buffer(EXT4_NAME_LEN,
1062                                                  &fname_crypto_str);
1063                 if (err < 0) {
1064                         brelse(bh);
1065                         return err;
1066                 }
1067         }
1068
1069         for (; de < top; de = ext4_next_entry(de, dir->i_sb->s_blocksize)) {
1070                 if (ext4_check_dir_entry(dir, NULL, de, bh,
1071                                 bh->b_data, bh->b_size,
1072                                 (block<<EXT4_BLOCK_SIZE_BITS(dir->i_sb))
1073                                          + ((char *)de - bh->b_data))) {
1074                         /* silently ignore the rest of the block */
1075                         break;
1076                 }
1077                 if (ext4_hash_in_dirent(dir)) {
1078                         if (de->name_len && de->inode) {
1079                                 hinfo->hash = EXT4_DIRENT_HASH(de);
1080                                 hinfo->minor_hash = EXT4_DIRENT_MINOR_HASH(de);
1081                         } else {
1082                                 hinfo->hash = 0;
1083                                 hinfo->minor_hash = 0;
1084                         }
1085                 } else {
1086                         ext4fs_dirhash(dir, de->name, de->name_len, hinfo);
1087                 }
1088                 if ((hinfo->hash < start_hash) ||
1089                     ((hinfo->hash == start_hash) &&
1090                      (hinfo->minor_hash < start_minor_hash)))
1091                         continue;
1092                 if (de->inode == 0)
1093                         continue;
1094                 if (!IS_ENCRYPTED(dir)) {
1095                         tmp_str.name = de->name;
1096                         tmp_str.len = de->name_len;
1097                         err = ext4_htree_store_dirent(dir_file,
1098                                    hinfo->hash, hinfo->minor_hash, de,
1099                                    &tmp_str);
1100                 } else {
1101                         int save_len = fname_crypto_str.len;
1102                         struct fscrypt_str de_name = FSTR_INIT(de->name,
1103                                                                 de->name_len);
1104
1105                         /* Directory is encrypted */
1106                         err = fscrypt_fname_disk_to_usr(dir, hinfo->hash,
1107                                         hinfo->minor_hash, &de_name,
1108                                         &fname_crypto_str);
1109                         if (err) {
1110                                 count = err;
1111                                 goto errout;
1112                         }
1113                         err = ext4_htree_store_dirent(dir_file,
1114                                    hinfo->hash, hinfo->minor_hash, de,
1115                                         &fname_crypto_str);
1116                         fname_crypto_str.len = save_len;
1117                 }
1118                 if (err != 0) {
1119                         count = err;
1120                         goto errout;
1121                 }
1122                 count++;
1123         }
1124 errout:
1125         brelse(bh);
1126         fscrypt_fname_free_buffer(&fname_crypto_str);
1127         return count;
1128 }
1129
1130
1131 /*
1132  * This function fills a red-black tree with information from a
1133  * directory.  We start scanning the directory in hash order, starting
1134  * at start_hash and start_minor_hash.
1135  *
1136  * This function returns the number of entries inserted into the tree,
1137  * or a negative error code.
1138  */
1139 int ext4_htree_fill_tree(struct file *dir_file, __u32 start_hash,
1140                          __u32 start_minor_hash, __u32 *next_hash)
1141 {
1142         struct dx_hash_info hinfo;
1143         struct ext4_dir_entry_2 *de;
1144         struct dx_frame frames[EXT4_HTREE_LEVEL], *frame;
1145         struct inode *dir;
1146         ext4_lblk_t block;
1147         int count = 0;
1148         int ret, err;
1149         __u32 hashval;
1150         struct fscrypt_str tmp_str;
1151
1152         dxtrace(printk(KERN_DEBUG "In htree_fill_tree, start hash: %x:%x\n",
1153                        start_hash, start_minor_hash));
1154         dir = file_inode(dir_file);
1155         if (!(ext4_test_inode_flag(dir, EXT4_INODE_INDEX))) {
1156                 if (ext4_hash_in_dirent(dir))
1157                         hinfo.hash_version = DX_HASH_SIPHASH;
1158                 else
1159                         hinfo.hash_version =
1160                                         EXT4_SB(dir->i_sb)->s_def_hash_version;
1161                 if (hinfo.hash_version <= DX_HASH_TEA)
1162                         hinfo.hash_version +=
1163                                 EXT4_SB(dir->i_sb)->s_hash_unsigned;
1164                 hinfo.seed = EXT4_SB(dir->i_sb)->s_hash_seed;
1165                 if (ext4_has_inline_data(dir)) {
1166                         int has_inline_data = 1;
1167                         count = ext4_inlinedir_to_tree(dir_file, dir, 0,
1168                                                        &hinfo, start_hash,
1169                                                        start_minor_hash,
1170                                                        &has_inline_data);
1171                         if (has_inline_data) {
1172                                 *next_hash = ~0;
1173                                 return count;
1174                         }
1175                 }
1176                 count = htree_dirblock_to_tree(dir_file, dir, 0, &hinfo,
1177                                                start_hash, start_minor_hash);
1178                 *next_hash = ~0;
1179                 return count;
1180         }
1181         hinfo.hash = start_hash;
1182         hinfo.minor_hash = 0;
1183         frame = dx_probe(NULL, dir, &hinfo, frames);
1184         if (IS_ERR(frame))
1185                 return PTR_ERR(frame);
1186
1187         /* Add '.' and '..' from the htree header */
1188         if (!start_hash && !start_minor_hash) {
1189                 de = (struct ext4_dir_entry_2 *) frames[0].bh->b_data;
1190                 tmp_str.name = de->name;
1191                 tmp_str.len = de->name_len;
1192                 err = ext4_htree_store_dirent(dir_file, 0, 0,
1193                                               de, &tmp_str);
1194                 if (err != 0)
1195                         goto errout;
1196                 count++;
1197         }
1198         if (start_hash < 2 || (start_hash ==2 && start_minor_hash==0)) {
1199                 de = (struct ext4_dir_entry_2 *) frames[0].bh->b_data;
1200                 de = ext4_next_entry(de, dir->i_sb->s_blocksize);
1201                 tmp_str.name = de->name;
1202                 tmp_str.len = de->name_len;
1203                 err = ext4_htree_store_dirent(dir_file, 2, 0,
1204                                               de, &tmp_str);
1205                 if (err != 0)
1206                         goto errout;
1207                 count++;
1208         }
1209
1210         while (1) {
1211                 if (fatal_signal_pending(current)) {
1212                         err = -ERESTARTSYS;
1213                         goto errout;
1214                 }
1215                 cond_resched();
1216                 block = dx_get_block(frame->at);
1217                 ret = htree_dirblock_to_tree(dir_file, dir, block, &hinfo,
1218                                              start_hash, start_minor_hash);
1219                 if (ret < 0) {
1220                         err = ret;
1221                         goto errout;
1222                 }
1223                 count += ret;
1224                 hashval = ~0;
1225                 ret = ext4_htree_next_block(dir, HASH_NB_ALWAYS,
1226                                             frame, frames, &hashval);
1227                 *next_hash = hashval;
1228                 if (ret < 0) {
1229                         err = ret;
1230                         goto errout;
1231                 }
1232                 /*
1233                  * Stop if:  (a) there are no more entries, or
1234                  * (b) we have inserted at least one entry and the
1235                  * next hash value is not a continuation
1236                  */
1237                 if ((ret == 0) ||
1238                     (count && ((hashval & 1) == 0)))
1239                         break;
1240         }
1241         dx_release(frames);
1242         dxtrace(printk(KERN_DEBUG "Fill tree: returned %d entries, "
1243                        "next hash: %x\n", count, *next_hash));
1244         return count;
1245 errout:
1246         dx_release(frames);
1247         return (err);
1248 }
1249
1250 static inline int search_dirblock(struct buffer_head *bh,
1251                                   struct inode *dir,
1252                                   struct ext4_filename *fname,
1253                                   unsigned int offset,
1254                                   struct ext4_dir_entry_2 **res_dir)
1255 {
1256         return ext4_search_dir(bh, bh->b_data, dir->i_sb->s_blocksize, dir,
1257                                fname, offset, res_dir);
1258 }
1259
1260 /*
1261  * Directory block splitting, compacting
1262  */
1263
1264 /*
1265  * Create map of hash values, offsets, and sizes, stored at end of block.
1266  * Returns number of entries mapped.
1267  */
1268 static int dx_make_map(struct inode *dir, struct buffer_head *bh,
1269                        struct dx_hash_info *hinfo,
1270                        struct dx_map_entry *map_tail)
1271 {
1272         int count = 0;
1273         struct ext4_dir_entry_2 *de = (struct ext4_dir_entry_2 *)bh->b_data;
1274         unsigned int buflen = bh->b_size;
1275         char *base = bh->b_data;
1276         struct dx_hash_info h = *hinfo;
1277
1278         if (ext4_has_metadata_csum(dir->i_sb))
1279                 buflen -= sizeof(struct ext4_dir_entry_tail);
1280
1281         while ((char *) de < base + buflen) {
1282                 if (ext4_check_dir_entry(dir, NULL, de, bh, base, buflen,
1283                                          ((char *)de) - base))
1284                         return -EFSCORRUPTED;
1285                 if (de->name_len && de->inode) {
1286                         if (ext4_hash_in_dirent(dir))
1287                                 h.hash = EXT4_DIRENT_HASH(de);
1288                         else
1289                                 ext4fs_dirhash(dir, de->name, de->name_len, &h);
1290                         map_tail--;
1291                         map_tail->hash = h.hash;
1292                         map_tail->offs = ((char *) de - base)>>2;
1293                         map_tail->size = le16_to_cpu(de->rec_len);
1294                         count++;
1295                         cond_resched();
1296                 }
1297                 de = ext4_next_entry(de, dir->i_sb->s_blocksize);
1298         }
1299         return count;
1300 }
1301
1302 /* Sort map by hash value */
1303 static void dx_sort_map (struct dx_map_entry *map, unsigned count)
1304 {
1305         struct dx_map_entry *p, *q, *top = map + count - 1;
1306         int more;
1307         /* Combsort until bubble sort doesn't suck */
1308         while (count > 2) {
1309                 count = count*10/13;
1310                 if (count - 9 < 2) /* 9, 10 -> 11 */
1311                         count = 11;
1312                 for (p = top, q = p - count; q >= map; p--, q--)
1313                         if (p->hash < q->hash)
1314                                 swap(*p, *q);
1315         }
1316         /* Garden variety bubble sort */
1317         do {
1318                 more = 0;
1319                 q = top;
1320                 while (q-- > map) {
1321                         if (q[1].hash >= q[0].hash)
1322                                 continue;
1323                         swap(*(q+1), *q);
1324                         more = 1;
1325                 }
1326         } while(more);
1327 }
1328
1329 static void dx_insert_block(struct dx_frame *frame, u32 hash, ext4_lblk_t block)
1330 {
1331         struct dx_entry *entries = frame->entries;
1332         struct dx_entry *old = frame->at, *new = old + 1;
1333         int count = dx_get_count(entries);
1334
1335         ASSERT(count < dx_get_limit(entries));
1336         ASSERT(old < entries + count);
1337         memmove(new + 1, new, (char *)(entries + count) - (char *)(new));
1338         dx_set_hash(new, hash);
1339         dx_set_block(new, block);
1340         dx_set_count(entries, count + 1);
1341 }
1342
1343 #if IS_ENABLED(CONFIG_UNICODE)
1344 /*
1345  * Test whether a case-insensitive directory entry matches the filename
1346  * being searched for.  If quick is set, assume the name being looked up
1347  * is already in the casefolded form.
1348  *
1349  * Returns: 0 if the directory entry matches, more than 0 if it
1350  * doesn't match or less than zero on error.
1351  */
1352 static int ext4_ci_compare(const struct inode *parent, const struct qstr *name,
1353                            u8 *de_name, size_t de_name_len, bool quick)
1354 {
1355         const struct super_block *sb = parent->i_sb;
1356         const struct unicode_map *um = sb->s_encoding;
1357         struct fscrypt_str decrypted_name = FSTR_INIT(NULL, de_name_len);
1358         struct qstr entry = QSTR_INIT(de_name, de_name_len);
1359         int ret;
1360
1361         if (IS_ENCRYPTED(parent)) {
1362                 const struct fscrypt_str encrypted_name =
1363                                 FSTR_INIT(de_name, de_name_len);
1364
1365                 decrypted_name.name = kmalloc(de_name_len, GFP_KERNEL);
1366                 if (!decrypted_name.name)
1367                         return -ENOMEM;
1368                 ret = fscrypt_fname_disk_to_usr(parent, 0, 0, &encrypted_name,
1369                                                 &decrypted_name);
1370                 if (ret < 0)
1371                         goto out;
1372                 entry.name = decrypted_name.name;
1373                 entry.len = decrypted_name.len;
1374         }
1375
1376         if (quick)
1377                 ret = utf8_strncasecmp_folded(um, name, &entry);
1378         else
1379                 ret = utf8_strncasecmp(um, name, &entry);
1380         if (ret < 0) {
1381                 /* Handle invalid character sequence as either an error
1382                  * or as an opaque byte sequence.
1383                  */
1384                 if (sb_has_strict_encoding(sb))
1385                         ret = -EINVAL;
1386                 else if (name->len != entry.len)
1387                         ret = 1;
1388                 else
1389                         ret = !!memcmp(name->name, entry.name, entry.len);
1390         }
1391 out:
1392         kfree(decrypted_name.name);
1393         return ret;
1394 }
1395
1396 int ext4_fname_setup_ci_filename(struct inode *dir, const struct qstr *iname,
1397                                   struct ext4_filename *name)
1398 {
1399         struct fscrypt_str *cf_name = &name->cf_name;
1400         struct dx_hash_info *hinfo = &name->hinfo;
1401         int len;
1402
1403         if (!IS_CASEFOLDED(dir) || !dir->i_sb->s_encoding ||
1404             (IS_ENCRYPTED(dir) && !fscrypt_has_encryption_key(dir))) {
1405                 cf_name->name = NULL;
1406                 return 0;
1407         }
1408
1409         cf_name->name = kmalloc(EXT4_NAME_LEN, GFP_NOFS);
1410         if (!cf_name->name)
1411                 return -ENOMEM;
1412
1413         len = utf8_casefold(dir->i_sb->s_encoding,
1414                             iname, cf_name->name,
1415                             EXT4_NAME_LEN);
1416         if (len <= 0) {
1417                 kfree(cf_name->name);
1418                 cf_name->name = NULL;
1419         }
1420         cf_name->len = (unsigned) len;
1421         if (!IS_ENCRYPTED(dir))
1422                 return 0;
1423
1424         hinfo->hash_version = DX_HASH_SIPHASH;
1425         hinfo->seed = NULL;
1426         if (cf_name->name)
1427                 ext4fs_dirhash(dir, cf_name->name, cf_name->len, hinfo);
1428         else
1429                 ext4fs_dirhash(dir, iname->name, iname->len, hinfo);
1430         return 0;
1431 }
1432 #endif
1433
1434 /*
1435  * Test whether a directory entry matches the filename being searched for.
1436  *
1437  * Return: %true if the directory entry matches, otherwise %false.
1438  */
1439 static bool ext4_match(struct inode *parent,
1440                               const struct ext4_filename *fname,
1441                               struct ext4_dir_entry_2 *de)
1442 {
1443         struct fscrypt_name f;
1444
1445         if (!de->inode)
1446                 return false;
1447
1448         f.usr_fname = fname->usr_fname;
1449         f.disk_name = fname->disk_name;
1450 #ifdef CONFIG_FS_ENCRYPTION
1451         f.crypto_buf = fname->crypto_buf;
1452 #endif
1453
1454 #if IS_ENABLED(CONFIG_UNICODE)
1455         if (parent->i_sb->s_encoding && IS_CASEFOLDED(parent) &&
1456             (!IS_ENCRYPTED(parent) || fscrypt_has_encryption_key(parent))) {
1457                 if (fname->cf_name.name) {
1458                         struct qstr cf = {.name = fname->cf_name.name,
1459                                           .len = fname->cf_name.len};
1460                         if (IS_ENCRYPTED(parent)) {
1461                                 if (fname->hinfo.hash != EXT4_DIRENT_HASH(de) ||
1462                                         fname->hinfo.minor_hash !=
1463                                                 EXT4_DIRENT_MINOR_HASH(de)) {
1464
1465                                         return false;
1466                                 }
1467                         }
1468                         return !ext4_ci_compare(parent, &cf, de->name,
1469                                                         de->name_len, true);
1470                 }
1471                 return !ext4_ci_compare(parent, fname->usr_fname, de->name,
1472                                                 de->name_len, false);
1473         }
1474 #endif
1475
1476         return fscrypt_match_name(&f, de->name, de->name_len);
1477 }
1478
1479 /*
1480  * Returns 0 if not found, -1 on failure, and 1 on success
1481  */
1482 int ext4_search_dir(struct buffer_head *bh, char *search_buf, int buf_size,
1483                     struct inode *dir, struct ext4_filename *fname,
1484                     unsigned int offset, struct ext4_dir_entry_2 **res_dir)
1485 {
1486         struct ext4_dir_entry_2 * de;
1487         char * dlimit;
1488         int de_len;
1489
1490         de = (struct ext4_dir_entry_2 *)search_buf;
1491         dlimit = search_buf + buf_size;
1492         while ((char *) de < dlimit - EXT4_BASE_DIR_LEN) {
1493                 /* this code is executed quadratically often */
1494                 /* do minimal checking `by hand' */
1495                 if (de->name + de->name_len <= dlimit &&
1496                     ext4_match(dir, fname, de)) {
1497                         /* found a match - just to be sure, do
1498                          * a full check */
1499                         if (ext4_check_dir_entry(dir, NULL, de, bh, search_buf,
1500                                                  buf_size, offset))
1501                                 return -1;
1502                         *res_dir = de;
1503                         return 1;
1504                 }
1505                 /* prevent looping on a bad block */
1506                 de_len = ext4_rec_len_from_disk(de->rec_len,
1507                                                 dir->i_sb->s_blocksize);
1508                 if (de_len <= 0)
1509                         return -1;
1510                 offset += de_len;
1511                 de = (struct ext4_dir_entry_2 *) ((char *) de + de_len);
1512         }
1513         return 0;
1514 }
1515
1516 static int is_dx_internal_node(struct inode *dir, ext4_lblk_t block,
1517                                struct ext4_dir_entry *de)
1518 {
1519         struct super_block *sb = dir->i_sb;
1520
1521         if (!is_dx(dir))
1522                 return 0;
1523         if (block == 0)
1524                 return 1;
1525         if (de->inode == 0 &&
1526             ext4_rec_len_from_disk(de->rec_len, sb->s_blocksize) ==
1527                         sb->s_blocksize)
1528                 return 1;
1529         return 0;
1530 }
1531
1532 /*
1533  *      __ext4_find_entry()
1534  *
1535  * finds an entry in the specified directory with the wanted name. It
1536  * returns the cache buffer in which the entry was found, and the entry
1537  * itself (as a parameter - res_dir). It does NOT read the inode of the
1538  * entry - you'll have to do that yourself if you want to.
1539  *
1540  * The returned buffer_head has ->b_count elevated.  The caller is expected
1541  * to brelse() it when appropriate.
1542  */
1543 static struct buffer_head *__ext4_find_entry(struct inode *dir,
1544                                              struct ext4_filename *fname,
1545                                              struct ext4_dir_entry_2 **res_dir,
1546                                              int *inlined)
1547 {
1548         struct super_block *sb;
1549         struct buffer_head *bh_use[NAMEI_RA_SIZE];
1550         struct buffer_head *bh, *ret = NULL;
1551         ext4_lblk_t start, block;
1552         const u8 *name = fname->usr_fname->name;
1553         size_t ra_max = 0;      /* Number of bh's in the readahead
1554                                    buffer, bh_use[] */
1555         size_t ra_ptr = 0;      /* Current index into readahead
1556                                    buffer */
1557         ext4_lblk_t  nblocks;
1558         int i, namelen, retval;
1559
1560         *res_dir = NULL;
1561         sb = dir->i_sb;
1562         namelen = fname->usr_fname->len;
1563         if (namelen > EXT4_NAME_LEN)
1564                 return NULL;
1565
1566         if (ext4_has_inline_data(dir)) {
1567                 int has_inline_data = 1;
1568                 ret = ext4_find_inline_entry(dir, fname, res_dir,
1569                                              &has_inline_data);
1570                 if (has_inline_data) {
1571                         if (inlined)
1572                                 *inlined = 1;
1573                         goto cleanup_and_exit;
1574                 }
1575         }
1576
1577         if ((namelen <= 2) && (name[0] == '.') &&
1578             (name[1] == '.' || name[1] == '\0')) {
1579                 /*
1580                  * "." or ".." will only be in the first block
1581                  * NFS may look up ".."; "." should be handled by the VFS
1582                  */
1583                 block = start = 0;
1584                 nblocks = 1;
1585                 goto restart;
1586         }
1587         if (is_dx(dir)) {
1588                 ret = ext4_dx_find_entry(dir, fname, res_dir);
1589                 /*
1590                  * On success, or if the error was file not found,
1591                  * return.  Otherwise, fall back to doing a search the
1592                  * old fashioned way.
1593                  */
1594                 if (!IS_ERR(ret) || PTR_ERR(ret) != ERR_BAD_DX_DIR)
1595                         goto cleanup_and_exit;
1596                 dxtrace(printk(KERN_DEBUG "ext4_find_entry: dx failed, "
1597                                "falling back\n"));
1598                 ret = NULL;
1599         }
1600         nblocks = dir->i_size >> EXT4_BLOCK_SIZE_BITS(sb);
1601         if (!nblocks) {
1602                 ret = NULL;
1603                 goto cleanup_and_exit;
1604         }
1605         start = EXT4_I(dir)->i_dir_start_lookup;
1606         if (start >= nblocks)
1607                 start = 0;
1608         block = start;
1609 restart:
1610         do {
1611                 /*
1612                  * We deal with the read-ahead logic here.
1613                  */
1614                 cond_resched();
1615                 if (ra_ptr >= ra_max) {
1616                         /* Refill the readahead buffer */
1617                         ra_ptr = 0;
1618                         if (block < start)
1619                                 ra_max = start - block;
1620                         else
1621                                 ra_max = nblocks - block;
1622                         ra_max = min(ra_max, ARRAY_SIZE(bh_use));
1623                         retval = ext4_bread_batch(dir, block, ra_max,
1624                                                   false /* wait */, bh_use);
1625                         if (retval) {
1626                                 ret = ERR_PTR(retval);
1627                                 ra_max = 0;
1628                                 goto cleanup_and_exit;
1629                         }
1630                 }
1631                 if ((bh = bh_use[ra_ptr++]) == NULL)
1632                         goto next;
1633                 wait_on_buffer(bh);
1634                 if (!buffer_uptodate(bh)) {
1635                         EXT4_ERROR_INODE_ERR(dir, EIO,
1636                                              "reading directory lblock %lu",
1637                                              (unsigned long) block);
1638                         brelse(bh);
1639                         ret = ERR_PTR(-EIO);
1640                         goto cleanup_and_exit;
1641                 }
1642                 if (!buffer_verified(bh) &&
1643                     !is_dx_internal_node(dir, block,
1644                                          (struct ext4_dir_entry *)bh->b_data) &&
1645                     !ext4_dirblock_csum_verify(dir, bh)) {
1646                         EXT4_ERROR_INODE_ERR(dir, EFSBADCRC,
1647                                              "checksumming directory "
1648                                              "block %lu", (unsigned long)block);
1649                         brelse(bh);
1650                         ret = ERR_PTR(-EFSBADCRC);
1651                         goto cleanup_and_exit;
1652                 }
1653                 set_buffer_verified(bh);
1654                 i = search_dirblock(bh, dir, fname,
1655                             block << EXT4_BLOCK_SIZE_BITS(sb), res_dir);
1656                 if (i == 1) {
1657                         EXT4_I(dir)->i_dir_start_lookup = block;
1658                         ret = bh;
1659                         goto cleanup_and_exit;
1660                 } else {
1661                         brelse(bh);
1662                         if (i < 0)
1663                                 goto cleanup_and_exit;
1664                 }
1665         next:
1666                 if (++block >= nblocks)
1667                         block = 0;
1668         } while (block != start);
1669
1670         /*
1671          * If the directory has grown while we were searching, then
1672          * search the last part of the directory before giving up.
1673          */
1674         block = nblocks;
1675         nblocks = dir->i_size >> EXT4_BLOCK_SIZE_BITS(sb);
1676         if (block < nblocks) {
1677                 start = 0;
1678                 goto restart;
1679         }
1680
1681 cleanup_and_exit:
1682         /* Clean up the read-ahead blocks */
1683         for (; ra_ptr < ra_max; ra_ptr++)
1684                 brelse(bh_use[ra_ptr]);
1685         return ret;
1686 }
1687
1688 static struct buffer_head *ext4_find_entry(struct inode *dir,
1689                                            const struct qstr *d_name,
1690                                            struct ext4_dir_entry_2 **res_dir,
1691                                            int *inlined)
1692 {
1693         int err;
1694         struct ext4_filename fname;
1695         struct buffer_head *bh;
1696
1697         err = ext4_fname_setup_filename(dir, d_name, 1, &fname);
1698         if (err == -ENOENT)
1699                 return NULL;
1700         if (err)
1701                 return ERR_PTR(err);
1702
1703         bh = __ext4_find_entry(dir, &fname, res_dir, inlined);
1704
1705         ext4_fname_free_filename(&fname);
1706         return bh;
1707 }
1708
1709 static struct buffer_head *ext4_lookup_entry(struct inode *dir,
1710                                              struct dentry *dentry,
1711                                              struct ext4_dir_entry_2 **res_dir)
1712 {
1713         int err;
1714         struct ext4_filename fname;
1715         struct buffer_head *bh;
1716
1717         err = ext4_fname_prepare_lookup(dir, dentry, &fname);
1718         generic_set_encrypted_ci_d_ops(dentry);
1719         if (err == -ENOENT)
1720                 return NULL;
1721         if (err)
1722                 return ERR_PTR(err);
1723
1724         bh = __ext4_find_entry(dir, &fname, res_dir, NULL);
1725
1726         ext4_fname_free_filename(&fname);
1727         return bh;
1728 }
1729
1730 static struct buffer_head * ext4_dx_find_entry(struct inode *dir,
1731                         struct ext4_filename *fname,
1732                         struct ext4_dir_entry_2 **res_dir)
1733 {
1734         struct super_block * sb = dir->i_sb;
1735         struct dx_frame frames[EXT4_HTREE_LEVEL], *frame;
1736         struct buffer_head *bh;
1737         ext4_lblk_t block;
1738         int retval;
1739
1740 #ifdef CONFIG_FS_ENCRYPTION
1741         *res_dir = NULL;
1742 #endif
1743         frame = dx_probe(fname, dir, NULL, frames);
1744         if (IS_ERR(frame))
1745                 return (struct buffer_head *) frame;
1746         do {
1747                 block = dx_get_block(frame->at);
1748                 bh = ext4_read_dirblock(dir, block, DIRENT_HTREE);
1749                 if (IS_ERR(bh))
1750                         goto errout;
1751
1752                 retval = search_dirblock(bh, dir, fname,
1753                                          block << EXT4_BLOCK_SIZE_BITS(sb),
1754                                          res_dir);
1755                 if (retval == 1)
1756                         goto success;
1757                 brelse(bh);
1758                 if (retval == -1) {
1759                         bh = ERR_PTR(ERR_BAD_DX_DIR);
1760                         goto errout;
1761                 }
1762
1763                 /* Check to see if we should continue to search */
1764                 retval = ext4_htree_next_block(dir, fname->hinfo.hash, frame,
1765                                                frames, NULL);
1766                 if (retval < 0) {
1767                         ext4_warning_inode(dir,
1768                                 "error %d reading directory index block",
1769                                 retval);
1770                         bh = ERR_PTR(retval);
1771                         goto errout;
1772                 }
1773         } while (retval == 1);
1774
1775         bh = NULL;
1776 errout:
1777         dxtrace(printk(KERN_DEBUG "%s not found\n", fname->usr_fname->name));
1778 success:
1779         dx_release(frames);
1780         return bh;
1781 }
1782
1783 static struct dentry *ext4_lookup(struct inode *dir, struct dentry *dentry, unsigned int flags)
1784 {
1785         struct inode *inode;
1786         struct ext4_dir_entry_2 *de;
1787         struct buffer_head *bh;
1788
1789         if (dentry->d_name.len > EXT4_NAME_LEN)
1790                 return ERR_PTR(-ENAMETOOLONG);
1791
1792         bh = ext4_lookup_entry(dir, dentry, &de);
1793         if (IS_ERR(bh))
1794                 return ERR_CAST(bh);
1795         inode = NULL;
1796         if (bh) {
1797                 __u32 ino = le32_to_cpu(de->inode);
1798                 brelse(bh);
1799                 if (!ext4_valid_inum(dir->i_sb, ino)) {
1800                         EXT4_ERROR_INODE(dir, "bad inode number: %u", ino);
1801                         return ERR_PTR(-EFSCORRUPTED);
1802                 }
1803                 if (unlikely(ino == dir->i_ino)) {
1804                         EXT4_ERROR_INODE(dir, "'%pd' linked to parent dir",
1805                                          dentry);
1806                         return ERR_PTR(-EFSCORRUPTED);
1807                 }
1808                 inode = ext4_iget(dir->i_sb, ino, EXT4_IGET_NORMAL);
1809                 if (inode == ERR_PTR(-ESTALE)) {
1810                         EXT4_ERROR_INODE(dir,
1811                                          "deleted inode referenced: %u",
1812                                          ino);
1813                         return ERR_PTR(-EFSCORRUPTED);
1814                 }
1815                 if (!IS_ERR(inode) && IS_ENCRYPTED(dir) &&
1816                     (S_ISDIR(inode->i_mode) || S_ISLNK(inode->i_mode)) &&
1817                     !fscrypt_has_permitted_context(dir, inode)) {
1818                         ext4_warning(inode->i_sb,
1819                                      "Inconsistent encryption contexts: %lu/%lu",
1820                                      dir->i_ino, inode->i_ino);
1821                         iput(inode);
1822                         return ERR_PTR(-EPERM);
1823                 }
1824         }
1825
1826 #if IS_ENABLED(CONFIG_UNICODE)
1827         if (!inode && IS_CASEFOLDED(dir)) {
1828                 /* Eventually we want to call d_add_ci(dentry, NULL)
1829                  * for negative dentries in the encoding case as
1830                  * well.  For now, prevent the negative dentry
1831                  * from being cached.
1832                  */
1833                 return NULL;
1834         }
1835 #endif
1836         return d_splice_alias(inode, dentry);
1837 }
1838
1839
1840 struct dentry *ext4_get_parent(struct dentry *child)
1841 {
1842         __u32 ino;
1843         struct ext4_dir_entry_2 * de;
1844         struct buffer_head *bh;
1845
1846         bh = ext4_find_entry(d_inode(child), &dotdot_name, &de, NULL);
1847         if (IS_ERR(bh))
1848                 return ERR_CAST(bh);
1849         if (!bh)
1850                 return ERR_PTR(-ENOENT);
1851         ino = le32_to_cpu(de->inode);
1852         brelse(bh);
1853
1854         if (!ext4_valid_inum(child->d_sb, ino)) {
1855                 EXT4_ERROR_INODE(d_inode(child),
1856                                  "bad parent inode number: %u", ino);
1857                 return ERR_PTR(-EFSCORRUPTED);
1858         }
1859
1860         return d_obtain_alias(ext4_iget(child->d_sb, ino, EXT4_IGET_NORMAL));
1861 }
1862
1863 /*
1864  * Move count entries from end of map between two memory locations.
1865  * Returns pointer to last entry moved.
1866  */
1867 static struct ext4_dir_entry_2 *
1868 dx_move_dirents(struct inode *dir, char *from, char *to,
1869                 struct dx_map_entry *map, int count,
1870                 unsigned blocksize)
1871 {
1872         unsigned rec_len = 0;
1873
1874         while (count--) {
1875                 struct ext4_dir_entry_2 *de = (struct ext4_dir_entry_2 *)
1876                                                 (from + (map->offs<<2));
1877                 rec_len = ext4_dir_rec_len(de->name_len, dir);
1878
1879                 memcpy (to, de, rec_len);
1880                 ((struct ext4_dir_entry_2 *) to)->rec_len =
1881                                 ext4_rec_len_to_disk(rec_len, blocksize);
1882
1883                 /* wipe dir_entry excluding the rec_len field */
1884                 de->inode = 0;
1885                 memset(&de->name_len, 0, ext4_rec_len_from_disk(de->rec_len,
1886                                                                 blocksize) -
1887                                          offsetof(struct ext4_dir_entry_2,
1888                                                                 name_len));
1889
1890                 map++;
1891                 to += rec_len;
1892         }
1893         return (struct ext4_dir_entry_2 *) (to - rec_len);
1894 }
1895
1896 /*
1897  * Compact each dir entry in the range to the minimal rec_len.
1898  * Returns pointer to last entry in range.
1899  */
1900 static struct ext4_dir_entry_2 *dx_pack_dirents(struct inode *dir, char *base,
1901                                                         unsigned int blocksize)
1902 {
1903         struct ext4_dir_entry_2 *next, *to, *prev, *de = (struct ext4_dir_entry_2 *) base;
1904         unsigned rec_len = 0;
1905
1906         prev = to = de;
1907         while ((char*)de < base + blocksize) {
1908                 next = ext4_next_entry(de, blocksize);
1909                 if (de->inode && de->name_len) {
1910                         rec_len = ext4_dir_rec_len(de->name_len, dir);
1911                         if (de > to)
1912                                 memmove(to, de, rec_len);
1913                         to->rec_len = ext4_rec_len_to_disk(rec_len, blocksize);
1914                         prev = to;
1915                         to = (struct ext4_dir_entry_2 *) (((char *) to) + rec_len);
1916                 }
1917                 de = next;
1918         }
1919         return prev;
1920 }
1921
1922 /*
1923  * Split a full leaf block to make room for a new dir entry.
1924  * Allocate a new block, and move entries so that they are approx. equally full.
1925  * Returns pointer to de in block into which the new entry will be inserted.
1926  */
1927 static struct ext4_dir_entry_2 *do_split(handle_t *handle, struct inode *dir,
1928                         struct buffer_head **bh,struct dx_frame *frame,
1929                         struct dx_hash_info *hinfo)
1930 {
1931         unsigned blocksize = dir->i_sb->s_blocksize;
1932         unsigned count, continued;
1933         struct buffer_head *bh2;
1934         ext4_lblk_t newblock;
1935         u32 hash2;
1936         struct dx_map_entry *map;
1937         char *data1 = (*bh)->b_data, *data2;
1938         unsigned split, move, size;
1939         struct ext4_dir_entry_2 *de = NULL, *de2;
1940         int     csum_size = 0;
1941         int     err = 0, i;
1942
1943         if (ext4_has_metadata_csum(dir->i_sb))
1944                 csum_size = sizeof(struct ext4_dir_entry_tail);
1945
1946         bh2 = ext4_append(handle, dir, &newblock);
1947         if (IS_ERR(bh2)) {
1948                 brelse(*bh);
1949                 *bh = NULL;
1950                 return (struct ext4_dir_entry_2 *) bh2;
1951         }
1952
1953         BUFFER_TRACE(*bh, "get_write_access");
1954         err = ext4_journal_get_write_access(handle, dir->i_sb, *bh,
1955                                             EXT4_JTR_NONE);
1956         if (err)
1957                 goto journal_error;
1958
1959         BUFFER_TRACE(frame->bh, "get_write_access");
1960         err = ext4_journal_get_write_access(handle, dir->i_sb, frame->bh,
1961                                             EXT4_JTR_NONE);
1962         if (err)
1963                 goto journal_error;
1964
1965         data2 = bh2->b_data;
1966
1967         /* create map in the end of data2 block */
1968         map = (struct dx_map_entry *) (data2 + blocksize);
1969         count = dx_make_map(dir, *bh, hinfo, map);
1970         if (count < 0) {
1971                 err = count;
1972                 goto journal_error;
1973         }
1974         map -= count;
1975         dx_sort_map(map, count);
1976         /* Ensure that neither split block is over half full */
1977         size = 0;
1978         move = 0;
1979         for (i = count-1; i >= 0; i--) {
1980                 /* is more than half of this entry in 2nd half of the block? */
1981                 if (size + map[i].size/2 > blocksize/2)
1982                         break;
1983                 size += map[i].size;
1984                 move++;
1985         }
1986         /*
1987          * map index at which we will split
1988          *
1989          * If the sum of active entries didn't exceed half the block size, just
1990          * split it in half by count; each resulting block will have at least
1991          * half the space free.
1992          */
1993         if (i > 0)
1994                 split = count - move;
1995         else
1996                 split = count/2;
1997
1998         hash2 = map[split].hash;
1999         continued = hash2 == map[split - 1].hash;
2000         dxtrace(printk(KERN_INFO "Split block %lu at %x, %i/%i\n",
2001                         (unsigned long)dx_get_block(frame->at),
2002                                         hash2, split, count-split));
2003
2004         /* Fancy dance to stay within two buffers */
2005         de2 = dx_move_dirents(dir, data1, data2, map + split, count - split,
2006                               blocksize);
2007         de = dx_pack_dirents(dir, data1, blocksize);
2008         de->rec_len = ext4_rec_len_to_disk(data1 + (blocksize - csum_size) -
2009                                            (char *) de,
2010                                            blocksize);
2011         de2->rec_len = ext4_rec_len_to_disk(data2 + (blocksize - csum_size) -
2012                                             (char *) de2,
2013                                             blocksize);
2014         if (csum_size) {
2015                 ext4_initialize_dirent_tail(*bh, blocksize);
2016                 ext4_initialize_dirent_tail(bh2, blocksize);
2017         }
2018
2019         dxtrace(dx_show_leaf(dir, hinfo, (struct ext4_dir_entry_2 *) data1,
2020                         blocksize, 1));
2021         dxtrace(dx_show_leaf(dir, hinfo, (struct ext4_dir_entry_2 *) data2,
2022                         blocksize, 1));
2023
2024         /* Which block gets the new entry? */
2025         if (hinfo->hash >= hash2) {
2026                 swap(*bh, bh2);
2027                 de = de2;
2028         }
2029         dx_insert_block(frame, hash2 + continued, newblock);
2030         err = ext4_handle_dirty_dirblock(handle, dir, bh2);
2031         if (err)
2032                 goto journal_error;
2033         err = ext4_handle_dirty_dx_node(handle, dir, frame->bh);
2034         if (err)
2035                 goto journal_error;
2036         brelse(bh2);
2037         dxtrace(dx_show_index("frame", frame->entries));
2038         return de;
2039
2040 journal_error:
2041         brelse(*bh);
2042         brelse(bh2);
2043         *bh = NULL;
2044         ext4_std_error(dir->i_sb, err);
2045         return ERR_PTR(err);
2046 }
2047
2048 int ext4_find_dest_de(struct inode *dir, struct inode *inode,
2049                       struct buffer_head *bh,
2050                       void *buf, int buf_size,
2051                       struct ext4_filename *fname,
2052                       struct ext4_dir_entry_2 **dest_de)
2053 {
2054         struct ext4_dir_entry_2 *de;
2055         unsigned short reclen = ext4_dir_rec_len(fname_len(fname), dir);
2056         int nlen, rlen;
2057         unsigned int offset = 0;
2058         char *top;
2059
2060         de = buf;
2061         top = buf + buf_size - reclen;
2062         while ((char *) de <= top) {
2063                 if (ext4_check_dir_entry(dir, NULL, de, bh,
2064                                          buf, buf_size, offset))
2065                         return -EFSCORRUPTED;
2066                 if (ext4_match(dir, fname, de))
2067                         return -EEXIST;
2068                 nlen = ext4_dir_rec_len(de->name_len, dir);
2069                 rlen = ext4_rec_len_from_disk(de->rec_len, buf_size);
2070                 if ((de->inode ? rlen - nlen : rlen) >= reclen)
2071                         break;
2072                 de = (struct ext4_dir_entry_2 *)((char *)de + rlen);
2073                 offset += rlen;
2074         }
2075         if ((char *) de > top)
2076                 return -ENOSPC;
2077
2078         *dest_de = de;
2079         return 0;
2080 }
2081
2082 void ext4_insert_dentry(struct inode *dir,
2083                         struct inode *inode,
2084                         struct ext4_dir_entry_2 *de,
2085                         int buf_size,
2086                         struct ext4_filename *fname)
2087 {
2088
2089         int nlen, rlen;
2090
2091         nlen = ext4_dir_rec_len(de->name_len, dir);
2092         rlen = ext4_rec_len_from_disk(de->rec_len, buf_size);
2093         if (de->inode) {
2094                 struct ext4_dir_entry_2 *de1 =
2095                         (struct ext4_dir_entry_2 *)((char *)de + nlen);
2096                 de1->rec_len = ext4_rec_len_to_disk(rlen - nlen, buf_size);
2097                 de->rec_len = ext4_rec_len_to_disk(nlen, buf_size);
2098                 de = de1;
2099         }
2100         de->file_type = EXT4_FT_UNKNOWN;
2101         de->inode = cpu_to_le32(inode->i_ino);
2102         ext4_set_de_type(inode->i_sb, de, inode->i_mode);
2103         de->name_len = fname_len(fname);
2104         memcpy(de->name, fname_name(fname), fname_len(fname));
2105         if (ext4_hash_in_dirent(dir)) {
2106                 struct dx_hash_info *hinfo = &fname->hinfo;
2107
2108                 EXT4_DIRENT_HASHES(de)->hash = cpu_to_le32(hinfo->hash);
2109                 EXT4_DIRENT_HASHES(de)->minor_hash =
2110                                                 cpu_to_le32(hinfo->minor_hash);
2111         }
2112 }
2113
2114 /*
2115  * Add a new entry into a directory (leaf) block.  If de is non-NULL,
2116  * it points to a directory entry which is guaranteed to be large
2117  * enough for new directory entry.  If de is NULL, then
2118  * add_dirent_to_buf will attempt search the directory block for
2119  * space.  It will return -ENOSPC if no space is available, and -EIO
2120  * and -EEXIST if directory entry already exists.
2121  */
2122 static int add_dirent_to_buf(handle_t *handle, struct ext4_filename *fname,
2123                              struct inode *dir,
2124                              struct inode *inode, struct ext4_dir_entry_2 *de,
2125                              struct buffer_head *bh)
2126 {
2127         unsigned int    blocksize = dir->i_sb->s_blocksize;
2128         int             csum_size = 0;
2129         int             err, err2;
2130
2131         if (ext4_has_metadata_csum(inode->i_sb))
2132                 csum_size = sizeof(struct ext4_dir_entry_tail);
2133
2134         if (!de) {
2135                 err = ext4_find_dest_de(dir, inode, bh, bh->b_data,
2136                                         blocksize - csum_size, fname, &de);
2137                 if (err)
2138                         return err;
2139         }
2140         BUFFER_TRACE(bh, "get_write_access");
2141         err = ext4_journal_get_write_access(handle, dir->i_sb, bh,
2142                                             EXT4_JTR_NONE);
2143         if (err) {
2144                 ext4_std_error(dir->i_sb, err);
2145                 return err;
2146         }
2147
2148         /* By now the buffer is marked for journaling */
2149         ext4_insert_dentry(dir, inode, de, blocksize, fname);
2150
2151         /*
2152          * XXX shouldn't update any times until successful
2153          * completion of syscall, but too many callers depend
2154          * on this.
2155          *
2156          * XXX similarly, too many callers depend on
2157          * ext4_new_inode() setting the times, but error
2158          * recovery deletes the inode, so the worst that can
2159          * happen is that the times are slightly out of date
2160          * and/or different from the directory change time.
2161          */
2162         dir->i_mtime = dir->i_ctime = current_time(dir);
2163         ext4_update_dx_flag(dir);
2164         inode_inc_iversion(dir);
2165         err2 = ext4_mark_inode_dirty(handle, dir);
2166         BUFFER_TRACE(bh, "call ext4_handle_dirty_metadata");
2167         err = ext4_handle_dirty_dirblock(handle, dir, bh);
2168         if (err)
2169                 ext4_std_error(dir->i_sb, err);
2170         return err ? err : err2;
2171 }
2172
2173 /*
2174  * This converts a one block unindexed directory to a 3 block indexed
2175  * directory, and adds the dentry to the indexed directory.
2176  */
2177 static int make_indexed_dir(handle_t *handle, struct ext4_filename *fname,
2178                             struct inode *dir,
2179                             struct inode *inode, struct buffer_head *bh)
2180 {
2181         struct buffer_head *bh2;
2182         struct dx_root  *root;
2183         struct dx_frame frames[EXT4_HTREE_LEVEL], *frame;
2184         struct dx_entry *entries;
2185         struct ext4_dir_entry_2 *de, *de2;
2186         char            *data2, *top;
2187         unsigned        len;
2188         int             retval;
2189         unsigned        blocksize;
2190         ext4_lblk_t  block;
2191         struct fake_dirent *fde;
2192         int csum_size = 0;
2193
2194         if (ext4_has_metadata_csum(inode->i_sb))
2195                 csum_size = sizeof(struct ext4_dir_entry_tail);
2196
2197         blocksize =  dir->i_sb->s_blocksize;
2198         dxtrace(printk(KERN_DEBUG "Creating index: inode %lu\n", dir->i_ino));
2199         BUFFER_TRACE(bh, "get_write_access");
2200         retval = ext4_journal_get_write_access(handle, dir->i_sb, bh,
2201                                                EXT4_JTR_NONE);
2202         if (retval) {
2203                 ext4_std_error(dir->i_sb, retval);
2204                 brelse(bh);
2205                 return retval;
2206         }
2207         root = (struct dx_root *) bh->b_data;
2208
2209         /* The 0th block becomes the root, move the dirents out */
2210         fde = &root->dotdot;
2211         de = (struct ext4_dir_entry_2 *)((char *)fde +
2212                 ext4_rec_len_from_disk(fde->rec_len, blocksize));
2213         if ((char *) de >= (((char *) root) + blocksize)) {
2214                 EXT4_ERROR_INODE(dir, "invalid rec_len for '..'");
2215                 brelse(bh);
2216                 return -EFSCORRUPTED;
2217         }
2218         len = ((char *) root) + (blocksize - csum_size) - (char *) de;
2219
2220         /* Allocate new block for the 0th block's dirents */
2221         bh2 = ext4_append(handle, dir, &block);
2222         if (IS_ERR(bh2)) {
2223                 brelse(bh);
2224                 return PTR_ERR(bh2);
2225         }
2226         ext4_set_inode_flag(dir, EXT4_INODE_INDEX);
2227         data2 = bh2->b_data;
2228
2229         memcpy(data2, de, len);
2230         memset(de, 0, len); /* wipe old data */
2231         de = (struct ext4_dir_entry_2 *) data2;
2232         top = data2 + len;
2233         while ((char *)(de2 = ext4_next_entry(de, blocksize)) < top)
2234                 de = de2;
2235         de->rec_len = ext4_rec_len_to_disk(data2 + (blocksize - csum_size) -
2236                                            (char *) de, blocksize);
2237
2238         if (csum_size)
2239                 ext4_initialize_dirent_tail(bh2, blocksize);
2240
2241         /* Initialize the root; the dot dirents already exist */
2242         de = (struct ext4_dir_entry_2 *) (&root->dotdot);
2243         de->rec_len = ext4_rec_len_to_disk(
2244                         blocksize - ext4_dir_rec_len(2, NULL), blocksize);
2245         memset (&root->info, 0, sizeof(root->info));
2246         root->info.info_length = sizeof(root->info);
2247         if (ext4_hash_in_dirent(dir))
2248                 root->info.hash_version = DX_HASH_SIPHASH;
2249         else
2250                 root->info.hash_version =
2251                                 EXT4_SB(dir->i_sb)->s_def_hash_version;
2252
2253         entries = root->entries;
2254         dx_set_block(entries, 1);
2255         dx_set_count(entries, 1);
2256         dx_set_limit(entries, dx_root_limit(dir, sizeof(root->info)));
2257
2258         /* Initialize as for dx_probe */
2259         fname->hinfo.hash_version = root->info.hash_version;
2260         if (fname->hinfo.hash_version <= DX_HASH_TEA)
2261                 fname->hinfo.hash_version += EXT4_SB(dir->i_sb)->s_hash_unsigned;
2262         fname->hinfo.seed = EXT4_SB(dir->i_sb)->s_hash_seed;
2263
2264         /* casefolded encrypted hashes are computed on fname setup */
2265         if (!ext4_hash_in_dirent(dir))
2266                 ext4fs_dirhash(dir, fname_name(fname),
2267                                 fname_len(fname), &fname->hinfo);
2268
2269         memset(frames, 0, sizeof(frames));
2270         frame = frames;
2271         frame->entries = entries;
2272         frame->at = entries;
2273         frame->bh = bh;
2274
2275         retval = ext4_handle_dirty_dx_node(handle, dir, frame->bh);
2276         if (retval)
2277                 goto out_frames;
2278         retval = ext4_handle_dirty_dirblock(handle, dir, bh2);
2279         if (retval)
2280                 goto out_frames;
2281
2282         de = do_split(handle,dir, &bh2, frame, &fname->hinfo);
2283         if (IS_ERR(de)) {
2284                 retval = PTR_ERR(de);
2285                 goto out_frames;
2286         }
2287
2288         retval = add_dirent_to_buf(handle, fname, dir, inode, de, bh2);
2289 out_frames:
2290         /*
2291          * Even if the block split failed, we have to properly write
2292          * out all the changes we did so far. Otherwise we can end up
2293          * with corrupted filesystem.
2294          */
2295         if (retval)
2296                 ext4_mark_inode_dirty(handle, dir);
2297         dx_release(frames);
2298         brelse(bh2);
2299         return retval;
2300 }
2301
2302 /*
2303  *      ext4_add_entry()
2304  *
2305  * adds a file entry to the specified directory, using the same
2306  * semantics as ext4_find_entry(). It returns NULL if it failed.
2307  *
2308  * NOTE!! The inode part of 'de' is left at 0 - which means you
2309  * may not sleep between calling this and putting something into
2310  * the entry, as someone else might have used it while you slept.
2311  */
2312 static int ext4_add_entry(handle_t *handle, struct dentry *dentry,
2313                           struct inode *inode)
2314 {
2315         struct inode *dir = d_inode(dentry->d_parent);
2316         struct buffer_head *bh = NULL;
2317         struct ext4_dir_entry_2 *de;
2318         struct super_block *sb;
2319         struct ext4_filename fname;
2320         int     retval;
2321         int     dx_fallback=0;
2322         unsigned blocksize;
2323         ext4_lblk_t block, blocks;
2324         int     csum_size = 0;
2325
2326         if (ext4_has_metadata_csum(inode->i_sb))
2327                 csum_size = sizeof(struct ext4_dir_entry_tail);
2328
2329         sb = dir->i_sb;
2330         blocksize = sb->s_blocksize;
2331         if (!dentry->d_name.len)
2332                 return -EINVAL;
2333
2334         if (fscrypt_is_nokey_name(dentry))
2335                 return -ENOKEY;
2336
2337 #if IS_ENABLED(CONFIG_UNICODE)
2338         if (sb_has_strict_encoding(sb) && IS_CASEFOLDED(dir) &&
2339             sb->s_encoding && utf8_validate(sb->s_encoding, &dentry->d_name))
2340                 return -EINVAL;
2341 #endif
2342
2343         retval = ext4_fname_setup_filename(dir, &dentry->d_name, 0, &fname);
2344         if (retval)
2345                 return retval;
2346
2347         if (ext4_has_inline_data(dir)) {
2348                 retval = ext4_try_add_inline_entry(handle, &fname, dir, inode);
2349                 if (retval < 0)
2350                         goto out;
2351                 if (retval == 1) {
2352                         retval = 0;
2353                         goto out;
2354                 }
2355         }
2356
2357         if (is_dx(dir)) {
2358                 retval = ext4_dx_add_entry(handle, &fname, dir, inode);
2359                 if (!retval || (retval != ERR_BAD_DX_DIR))
2360                         goto out;
2361                 /* Can we just ignore htree data? */
2362                 if (ext4_has_metadata_csum(sb)) {
2363                         EXT4_ERROR_INODE(dir,
2364                                 "Directory has corrupted htree index.");
2365                         retval = -EFSCORRUPTED;
2366                         goto out;
2367                 }
2368                 ext4_clear_inode_flag(dir, EXT4_INODE_INDEX);
2369                 dx_fallback++;
2370                 retval = ext4_mark_inode_dirty(handle, dir);
2371                 if (unlikely(retval))
2372                         goto out;
2373         }
2374         blocks = dir->i_size >> sb->s_blocksize_bits;
2375         for (block = 0; block < blocks; block++) {
2376                 bh = ext4_read_dirblock(dir, block, DIRENT);
2377                 if (bh == NULL) {
2378                         bh = ext4_bread(handle, dir, block,
2379                                         EXT4_GET_BLOCKS_CREATE);
2380                         goto add_to_new_block;
2381                 }
2382                 if (IS_ERR(bh)) {
2383                         retval = PTR_ERR(bh);
2384                         bh = NULL;
2385                         goto out;
2386                 }
2387                 retval = add_dirent_to_buf(handle, &fname, dir, inode,
2388                                            NULL, bh);
2389                 if (retval != -ENOSPC)
2390                         goto out;
2391
2392                 if (blocks == 1 && !dx_fallback &&
2393                     ext4_has_feature_dir_index(sb)) {
2394                         retval = make_indexed_dir(handle, &fname, dir,
2395                                                   inode, bh);
2396                         bh = NULL; /* make_indexed_dir releases bh */
2397                         goto out;
2398                 }
2399                 brelse(bh);
2400         }
2401         bh = ext4_append(handle, dir, &block);
2402 add_to_new_block:
2403         if (IS_ERR(bh)) {
2404                 retval = PTR_ERR(bh);
2405                 bh = NULL;
2406                 goto out;
2407         }
2408         de = (struct ext4_dir_entry_2 *) bh->b_data;
2409         de->inode = 0;
2410         de->rec_len = ext4_rec_len_to_disk(blocksize - csum_size, blocksize);
2411
2412         if (csum_size)
2413                 ext4_initialize_dirent_tail(bh, blocksize);
2414
2415         retval = add_dirent_to_buf(handle, &fname, dir, inode, de, bh);
2416 out:
2417         ext4_fname_free_filename(&fname);
2418         brelse(bh);
2419         if (retval == 0)
2420                 ext4_set_inode_state(inode, EXT4_STATE_NEWENTRY);
2421         return retval;
2422 }
2423
2424 /*
2425  * Returns 0 for success, or a negative error value
2426  */
2427 static int ext4_dx_add_entry(handle_t *handle, struct ext4_filename *fname,
2428                              struct inode *dir, struct inode *inode)
2429 {
2430         struct dx_frame frames[EXT4_HTREE_LEVEL], *frame;
2431         struct dx_entry *entries, *at;
2432         struct buffer_head *bh;
2433         struct super_block *sb = dir->i_sb;
2434         struct ext4_dir_entry_2 *de;
2435         int restart;
2436         int err;
2437
2438 again:
2439         restart = 0;
2440         frame = dx_probe(fname, dir, NULL, frames);
2441         if (IS_ERR(frame))
2442                 return PTR_ERR(frame);
2443         entries = frame->entries;
2444         at = frame->at;
2445         bh = ext4_read_dirblock(dir, dx_get_block(frame->at), DIRENT_HTREE);
2446         if (IS_ERR(bh)) {
2447                 err = PTR_ERR(bh);
2448                 bh = NULL;
2449                 goto cleanup;
2450         }
2451
2452         BUFFER_TRACE(bh, "get_write_access");
2453         err = ext4_journal_get_write_access(handle, sb, bh, EXT4_JTR_NONE);
2454         if (err)
2455                 goto journal_error;
2456
2457         err = add_dirent_to_buf(handle, fname, dir, inode, NULL, bh);
2458         if (err != -ENOSPC)
2459                 goto cleanup;
2460
2461         err = 0;
2462         /* Block full, should compress but for now just split */
2463         dxtrace(printk(KERN_DEBUG "using %u of %u node entries\n",
2464                        dx_get_count(entries), dx_get_limit(entries)));
2465         /* Need to split index? */
2466         if (dx_get_count(entries) == dx_get_limit(entries)) {
2467                 ext4_lblk_t newblock;
2468                 int levels = frame - frames + 1;
2469                 unsigned int icount;
2470                 int add_level = 1;
2471                 struct dx_entry *entries2;
2472                 struct dx_node *node2;
2473                 struct buffer_head *bh2;
2474
2475                 while (frame > frames) {
2476                         if (dx_get_count((frame - 1)->entries) <
2477                             dx_get_limit((frame - 1)->entries)) {
2478                                 add_level = 0;
2479                                 break;
2480                         }
2481                         frame--; /* split higher index block */
2482                         at = frame->at;
2483                         entries = frame->entries;
2484                         restart = 1;
2485                 }
2486                 if (add_level && levels == ext4_dir_htree_level(sb)) {
2487                         ext4_warning(sb, "Directory (ino: %lu) index full, "
2488                                          "reach max htree level :%d",
2489                                          dir->i_ino, levels);
2490                         if (ext4_dir_htree_level(sb) < EXT4_HTREE_LEVEL) {
2491                                 ext4_warning(sb, "Large directory feature is "
2492                                                  "not enabled on this "
2493                                                  "filesystem");
2494                         }
2495                         err = -ENOSPC;
2496                         goto cleanup;
2497                 }
2498                 icount = dx_get_count(entries);
2499                 bh2 = ext4_append(handle, dir, &newblock);
2500                 if (IS_ERR(bh2)) {
2501                         err = PTR_ERR(bh2);
2502                         goto cleanup;
2503                 }
2504                 node2 = (struct dx_node *)(bh2->b_data);
2505                 entries2 = node2->entries;
2506                 memset(&node2->fake, 0, sizeof(struct fake_dirent));
2507                 node2->fake.rec_len = ext4_rec_len_to_disk(sb->s_blocksize,
2508                                                            sb->s_blocksize);
2509                 BUFFER_TRACE(frame->bh, "get_write_access");
2510                 err = ext4_journal_get_write_access(handle, sb, frame->bh,
2511                                                     EXT4_JTR_NONE);
2512                 if (err)
2513                         goto journal_error;
2514                 if (!add_level) {
2515                         unsigned icount1 = icount/2, icount2 = icount - icount1;
2516                         unsigned hash2 = dx_get_hash(entries + icount1);
2517                         dxtrace(printk(KERN_DEBUG "Split index %i/%i\n",
2518                                        icount1, icount2));
2519
2520                         BUFFER_TRACE(frame->bh, "get_write_access"); /* index root */
2521                         err = ext4_journal_get_write_access(handle, sb,
2522                                                             (frame - 1)->bh,
2523                                                             EXT4_JTR_NONE);
2524                         if (err)
2525                                 goto journal_error;
2526
2527                         memcpy((char *) entries2, (char *) (entries + icount1),
2528                                icount2 * sizeof(struct dx_entry));
2529                         dx_set_count(entries, icount1);
2530                         dx_set_count(entries2, icount2);
2531                         dx_set_limit(entries2, dx_node_limit(dir));
2532
2533                         /* Which index block gets the new entry? */
2534                         if (at - entries >= icount1) {
2535                                 frame->at = at - entries - icount1 + entries2;
2536                                 frame->entries = entries = entries2;
2537                                 swap(frame->bh, bh2);
2538                         }
2539                         dx_insert_block((frame - 1), hash2, newblock);
2540                         dxtrace(dx_show_index("node", frame->entries));
2541                         dxtrace(dx_show_index("node",
2542                                ((struct dx_node *) bh2->b_data)->entries));
2543                         err = ext4_handle_dirty_dx_node(handle, dir, bh2);
2544                         if (err)
2545                                 goto journal_error;
2546                         brelse (bh2);
2547                         err = ext4_handle_dirty_dx_node(handle, dir,
2548                                                    (frame - 1)->bh);
2549                         if (err)
2550                                 goto journal_error;
2551                         err = ext4_handle_dirty_dx_node(handle, dir,
2552                                                         frame->bh);
2553                         if (restart || err)
2554                                 goto journal_error;
2555                 } else {
2556                         struct dx_root *dxroot;
2557                         memcpy((char *) entries2, (char *) entries,
2558                                icount * sizeof(struct dx_entry));
2559                         dx_set_limit(entries2, dx_node_limit(dir));
2560
2561                         /* Set up root */
2562                         dx_set_count(entries, 1);
2563                         dx_set_block(entries + 0, newblock);
2564                         dxroot = (struct dx_root *)frames[0].bh->b_data;
2565                         dxroot->info.indirect_levels += 1;
2566                         dxtrace(printk(KERN_DEBUG
2567                                        "Creating %d level index...\n",
2568                                        dxroot->info.indirect_levels));
2569                         err = ext4_handle_dirty_dx_node(handle, dir, frame->bh);
2570                         if (err)
2571                                 goto journal_error;
2572                         err = ext4_handle_dirty_dx_node(handle, dir, bh2);
2573                         brelse(bh2);
2574                         restart = 1;
2575                         goto journal_error;
2576                 }
2577         }
2578         de = do_split(handle, dir, &bh, frame, &fname->hinfo);
2579         if (IS_ERR(de)) {
2580                 err = PTR_ERR(de);
2581                 goto cleanup;
2582         }
2583         err = add_dirent_to_buf(handle, fname, dir, inode, de, bh);
2584         goto cleanup;
2585
2586 journal_error:
2587         ext4_std_error(dir->i_sb, err); /* this is a no-op if err == 0 */
2588 cleanup:
2589         brelse(bh);
2590         dx_release(frames);
2591         /* @restart is true means htree-path has been changed, we need to
2592          * repeat dx_probe() to find out valid htree-path
2593          */
2594         if (restart && err == 0)
2595                 goto again;
2596         return err;
2597 }
2598
2599 /*
2600  * ext4_generic_delete_entry deletes a directory entry by merging it
2601  * with the previous entry
2602  */
2603 int ext4_generic_delete_entry(struct inode *dir,
2604                               struct ext4_dir_entry_2 *de_del,
2605                               struct buffer_head *bh,
2606                               void *entry_buf,
2607                               int buf_size,
2608                               int csum_size)
2609 {
2610         struct ext4_dir_entry_2 *de, *pde;
2611         unsigned int blocksize = dir->i_sb->s_blocksize;
2612         int i;
2613
2614         i = 0;
2615         pde = NULL;
2616         de = entry_buf;
2617         while (i < buf_size - csum_size) {
2618                 if (ext4_check_dir_entry(dir, NULL, de, bh,
2619                                          entry_buf, buf_size, i))
2620                         return -EFSCORRUPTED;
2621                 if (de == de_del)  {
2622                         if (pde) {
2623                                 pde->rec_len = ext4_rec_len_to_disk(
2624                                         ext4_rec_len_from_disk(pde->rec_len,
2625                                                                blocksize) +
2626                                         ext4_rec_len_from_disk(de->rec_len,
2627                                                                blocksize),
2628                                         blocksize);
2629
2630                                 /* wipe entire dir_entry */
2631                                 memset(de, 0, ext4_rec_len_from_disk(de->rec_len,
2632                                                                 blocksize));
2633                         } else {
2634                                 /* wipe dir_entry excluding the rec_len field */
2635                                 de->inode = 0;
2636                                 memset(&de->name_len, 0,
2637                                         ext4_rec_len_from_disk(de->rec_len,
2638                                                                 blocksize) -
2639                                         offsetof(struct ext4_dir_entry_2,
2640                                                                 name_len));
2641                         }
2642
2643                         inode_inc_iversion(dir);
2644                         return 0;
2645                 }
2646                 i += ext4_rec_len_from_disk(de->rec_len, blocksize);
2647                 pde = de;
2648                 de = ext4_next_entry(de, blocksize);
2649         }
2650         return -ENOENT;
2651 }
2652
2653 static int ext4_delete_entry(handle_t *handle,
2654                              struct inode *dir,
2655                              struct ext4_dir_entry_2 *de_del,
2656                              struct buffer_head *bh)
2657 {
2658         int err, csum_size = 0;
2659
2660         if (ext4_has_inline_data(dir)) {
2661                 int has_inline_data = 1;
2662                 err = ext4_delete_inline_entry(handle, dir, de_del, bh,
2663                                                &has_inline_data);
2664                 if (has_inline_data)
2665                         return err;
2666         }
2667
2668         if (ext4_has_metadata_csum(dir->i_sb))
2669                 csum_size = sizeof(struct ext4_dir_entry_tail);
2670
2671         BUFFER_TRACE(bh, "get_write_access");
2672         err = ext4_journal_get_write_access(handle, dir->i_sb, bh,
2673                                             EXT4_JTR_NONE);
2674         if (unlikely(err))
2675                 goto out;
2676
2677         err = ext4_generic_delete_entry(dir, de_del, bh, bh->b_data,
2678                                         dir->i_sb->s_blocksize, csum_size);
2679         if (err)
2680                 goto out;
2681
2682         BUFFER_TRACE(bh, "call ext4_handle_dirty_metadata");
2683         err = ext4_handle_dirty_dirblock(handle, dir, bh);
2684         if (unlikely(err))
2685                 goto out;
2686
2687         return 0;
2688 out:
2689         if (err != -ENOENT)
2690                 ext4_std_error(dir->i_sb, err);
2691         return err;
2692 }
2693
2694 /*
2695  * Set directory link count to 1 if nlinks > EXT4_LINK_MAX, or if nlinks == 2
2696  * since this indicates that nlinks count was previously 1 to avoid overflowing
2697  * the 16-bit i_links_count field on disk.  Directories with i_nlink == 1 mean
2698  * that subdirectory link counts are not being maintained accurately.
2699  *
2700  * The caller has already checked for i_nlink overflow in case the DIR_LINK
2701  * feature is not enabled and returned -EMLINK.  The is_dx() check is a proxy
2702  * for checking S_ISDIR(inode) (since the INODE_INDEX feature will not be set
2703  * on regular files) and to avoid creating huge/slow non-HTREE directories.
2704  */
2705 static void ext4_inc_count(struct inode *inode)
2706 {
2707         inc_nlink(inode);
2708         if (is_dx(inode) &&
2709             (inode->i_nlink > EXT4_LINK_MAX || inode->i_nlink == 2))
2710                 set_nlink(inode, 1);
2711 }
2712
2713 /*
2714  * If a directory had nlink == 1, then we should let it be 1. This indicates
2715  * directory has >EXT4_LINK_MAX subdirs.
2716  */
2717 static void ext4_dec_count(struct inode *inode)
2718 {
2719         if (!S_ISDIR(inode->i_mode) || inode->i_nlink > 2)
2720                 drop_nlink(inode);
2721 }
2722
2723
2724 /*
2725  * Add non-directory inode to a directory. On success, the inode reference is
2726  * consumed by dentry is instantiation. This is also indicated by clearing of
2727  * *inodep pointer. On failure, the caller is responsible for dropping the
2728  * inode reference in the safe context.
2729  */
2730 static int ext4_add_nondir(handle_t *handle,
2731                 struct dentry *dentry, struct inode **inodep)
2732 {
2733         struct inode *dir = d_inode(dentry->d_parent);
2734         struct inode *inode = *inodep;
2735         int err = ext4_add_entry(handle, dentry, inode);
2736         if (!err) {
2737                 err = ext4_mark_inode_dirty(handle, inode);
2738                 if (IS_DIRSYNC(dir))
2739                         ext4_handle_sync(handle);
2740                 d_instantiate_new(dentry, inode);
2741                 *inodep = NULL;
2742                 return err;
2743         }
2744         drop_nlink(inode);
2745         ext4_orphan_add(handle, inode);
2746         unlock_new_inode(inode);
2747         return err;
2748 }
2749
2750 /*
2751  * By the time this is called, we already have created
2752  * the directory cache entry for the new file, but it
2753  * is so far negative - it has no inode.
2754  *
2755  * If the create succeeds, we fill in the inode information
2756  * with d_instantiate().
2757  */
2758 static int ext4_create(struct user_namespace *mnt_userns, struct inode *dir,
2759                        struct dentry *dentry, umode_t mode, bool excl)
2760 {
2761         handle_t *handle;
2762         struct inode *inode;
2763         int err, credits, retries = 0;
2764
2765         err = dquot_initialize(dir);
2766         if (err)
2767                 return err;
2768
2769         credits = (EXT4_DATA_TRANS_BLOCKS(dir->i_sb) +
2770                    EXT4_INDEX_EXTRA_TRANS_BLOCKS + 3);
2771 retry:
2772         inode = ext4_new_inode_start_handle(mnt_userns, dir, mode, &dentry->d_name,
2773                                             0, NULL, EXT4_HT_DIR, credits);
2774         handle = ext4_journal_current_handle();
2775         err = PTR_ERR(inode);
2776         if (!IS_ERR(inode)) {
2777                 inode->i_op = &ext4_file_inode_operations;
2778                 inode->i_fop = &ext4_file_operations;
2779                 ext4_set_aops(inode);
2780                 err = ext4_add_nondir(handle, dentry, &inode);
2781                 if (!err)
2782                         ext4_fc_track_create(handle, dentry);
2783         }
2784         if (handle)
2785                 ext4_journal_stop(handle);
2786         if (!IS_ERR_OR_NULL(inode))
2787                 iput(inode);
2788         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
2789                 goto retry;
2790         return err;
2791 }
2792
2793 static int ext4_mknod(struct user_namespace *mnt_userns, struct inode *dir,
2794                       struct dentry *dentry, umode_t mode, dev_t rdev)
2795 {
2796         handle_t *handle;
2797         struct inode *inode;
2798         int err, credits, retries = 0;
2799
2800         err = dquot_initialize(dir);
2801         if (err)
2802                 return err;
2803
2804         credits = (EXT4_DATA_TRANS_BLOCKS(dir->i_sb) +
2805                    EXT4_INDEX_EXTRA_TRANS_BLOCKS + 3);
2806 retry:
2807         inode = ext4_new_inode_start_handle(mnt_userns, dir, mode, &dentry->d_name,
2808                                             0, NULL, EXT4_HT_DIR, credits);
2809         handle = ext4_journal_current_handle();
2810         err = PTR_ERR(inode);
2811         if (!IS_ERR(inode)) {
2812                 init_special_inode(inode, inode->i_mode, rdev);
2813                 inode->i_op = &ext4_special_inode_operations;
2814                 err = ext4_add_nondir(handle, dentry, &inode);
2815                 if (!err)
2816                         ext4_fc_track_create(handle, dentry);
2817         }
2818         if (handle)
2819                 ext4_journal_stop(handle);
2820         if (!IS_ERR_OR_NULL(inode))
2821                 iput(inode);
2822         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
2823                 goto retry;
2824         return err;
2825 }
2826
2827 static int ext4_tmpfile(struct user_namespace *mnt_userns, struct inode *dir,
2828                         struct dentry *dentry, umode_t mode)
2829 {
2830         handle_t *handle;
2831         struct inode *inode;
2832         int err, retries = 0;
2833
2834         err = dquot_initialize(dir);
2835         if (err)
2836                 return err;
2837
2838 retry:
2839         inode = ext4_new_inode_start_handle(mnt_userns, dir, mode,
2840                                             NULL, 0, NULL,
2841                                             EXT4_HT_DIR,
2842                         EXT4_MAXQUOTAS_INIT_BLOCKS(dir->i_sb) +
2843                           4 + EXT4_XATTR_TRANS_BLOCKS);
2844         handle = ext4_journal_current_handle();
2845         err = PTR_ERR(inode);
2846         if (!IS_ERR(inode)) {
2847                 inode->i_op = &ext4_file_inode_operations;
2848                 inode->i_fop = &ext4_file_operations;
2849                 ext4_set_aops(inode);
2850                 d_tmpfile(dentry, inode);
2851                 err = ext4_orphan_add(handle, inode);
2852                 if (err)
2853                         goto err_unlock_inode;
2854                 mark_inode_dirty(inode);
2855                 unlock_new_inode(inode);
2856         }
2857         if (handle)
2858                 ext4_journal_stop(handle);
2859         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
2860                 goto retry;
2861         return err;
2862 err_unlock_inode:
2863         ext4_journal_stop(handle);
2864         unlock_new_inode(inode);
2865         return err;
2866 }
2867
2868 struct ext4_dir_entry_2 *ext4_init_dot_dotdot(struct inode *inode,
2869                           struct ext4_dir_entry_2 *de,
2870                           int blocksize, int csum_size,
2871                           unsigned int parent_ino, int dotdot_real_len)
2872 {
2873         de->inode = cpu_to_le32(inode->i_ino);
2874         de->name_len = 1;
2875         de->rec_len = ext4_rec_len_to_disk(ext4_dir_rec_len(de->name_len, NULL),
2876                                            blocksize);
2877         strcpy(de->name, ".");
2878         ext4_set_de_type(inode->i_sb, de, S_IFDIR);
2879
2880         de = ext4_next_entry(de, blocksize);
2881         de->inode = cpu_to_le32(parent_ino);
2882         de->name_len = 2;
2883         if (!dotdot_real_len)
2884                 de->rec_len = ext4_rec_len_to_disk(blocksize -
2885                                         (csum_size + ext4_dir_rec_len(1, NULL)),
2886                                         blocksize);
2887         else
2888                 de->rec_len = ext4_rec_len_to_disk(
2889                                         ext4_dir_rec_len(de->name_len, NULL),
2890                                         blocksize);
2891         strcpy(de->name, "..");
2892         ext4_set_de_type(inode->i_sb, de, S_IFDIR);
2893
2894         return ext4_next_entry(de, blocksize);
2895 }
2896
2897 int ext4_init_new_dir(handle_t *handle, struct inode *dir,
2898                              struct inode *inode)
2899 {
2900         struct buffer_head *dir_block = NULL;
2901         struct ext4_dir_entry_2 *de;
2902         ext4_lblk_t block = 0;
2903         unsigned int blocksize = dir->i_sb->s_blocksize;
2904         int csum_size = 0;
2905         int err;
2906
2907         if (ext4_has_metadata_csum(dir->i_sb))
2908                 csum_size = sizeof(struct ext4_dir_entry_tail);
2909
2910         if (ext4_test_inode_state(inode, EXT4_STATE_MAY_INLINE_DATA)) {
2911                 err = ext4_try_create_inline_dir(handle, dir, inode);
2912                 if (err < 0 && err != -ENOSPC)
2913                         goto out;
2914                 if (!err)
2915                         goto out;
2916         }
2917
2918         inode->i_size = 0;
2919         dir_block = ext4_append(handle, inode, &block);
2920         if (IS_ERR(dir_block))
2921                 return PTR_ERR(dir_block);
2922         de = (struct ext4_dir_entry_2 *)dir_block->b_data;
2923         ext4_init_dot_dotdot(inode, de, blocksize, csum_size, dir->i_ino, 0);
2924         set_nlink(inode, 2);
2925         if (csum_size)
2926                 ext4_initialize_dirent_tail(dir_block, blocksize);
2927
2928         BUFFER_TRACE(dir_block, "call ext4_handle_dirty_metadata");
2929         err = ext4_handle_dirty_dirblock(handle, inode, dir_block);
2930         if (err)
2931                 goto out;
2932         set_buffer_verified(dir_block);
2933 out:
2934         brelse(dir_block);
2935         return err;
2936 }
2937
2938 static int ext4_mkdir(struct user_namespace *mnt_userns, struct inode *dir,
2939                       struct dentry *dentry, umode_t mode)
2940 {
2941         handle_t *handle;
2942         struct inode *inode;
2943         int err, err2 = 0, credits, retries = 0;
2944
2945         if (EXT4_DIR_LINK_MAX(dir))
2946                 return -EMLINK;
2947
2948         err = dquot_initialize(dir);
2949         if (err)
2950                 return err;
2951
2952         credits = (EXT4_DATA_TRANS_BLOCKS(dir->i_sb) +
2953                    EXT4_INDEX_EXTRA_TRANS_BLOCKS + 3);
2954 retry:
2955         inode = ext4_new_inode_start_handle(mnt_userns, dir, S_IFDIR | mode,
2956                                             &dentry->d_name,
2957                                             0, NULL, EXT4_HT_DIR, credits);
2958         handle = ext4_journal_current_handle();
2959         err = PTR_ERR(inode);
2960         if (IS_ERR(inode))
2961                 goto out_stop;
2962
2963         inode->i_op = &ext4_dir_inode_operations;
2964         inode->i_fop = &ext4_dir_operations;
2965         err = ext4_init_new_dir(handle, dir, inode);
2966         if (err)
2967                 goto out_clear_inode;
2968         err = ext4_mark_inode_dirty(handle, inode);
2969         if (!err)
2970                 err = ext4_add_entry(handle, dentry, inode);
2971         if (err) {
2972 out_clear_inode:
2973                 clear_nlink(inode);
2974                 ext4_orphan_add(handle, inode);
2975                 unlock_new_inode(inode);
2976                 err2 = ext4_mark_inode_dirty(handle, inode);
2977                 if (unlikely(err2))
2978                         err = err2;
2979                 ext4_journal_stop(handle);
2980                 iput(inode);
2981                 goto out_retry;
2982         }
2983         ext4_inc_count(dir);
2984
2985         ext4_update_dx_flag(dir);
2986         err = ext4_mark_inode_dirty(handle, dir);
2987         if (err)
2988                 goto out_clear_inode;
2989         d_instantiate_new(dentry, inode);
2990         ext4_fc_track_create(handle, dentry);
2991         if (IS_DIRSYNC(dir))
2992                 ext4_handle_sync(handle);
2993
2994 out_stop:
2995         if (handle)
2996                 ext4_journal_stop(handle);
2997 out_retry:
2998         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
2999                 goto retry;
3000         return err;
3001 }
3002
3003 /*
3004  * routine to check that the specified directory is empty (for rmdir)
3005  */
3006 bool ext4_empty_dir(struct inode *inode)
3007 {
3008         unsigned int offset;
3009         struct buffer_head *bh;
3010         struct ext4_dir_entry_2 *de;
3011         struct super_block *sb;
3012
3013         if (ext4_has_inline_data(inode)) {
3014                 int has_inline_data = 1;
3015                 int ret;
3016
3017                 ret = empty_inline_dir(inode, &has_inline_data);
3018                 if (has_inline_data)
3019                         return ret;
3020         }
3021
3022         sb = inode->i_sb;
3023         if (inode->i_size < ext4_dir_rec_len(1, NULL) +
3024                                         ext4_dir_rec_len(2, NULL)) {
3025                 EXT4_ERROR_INODE(inode, "invalid size");
3026                 return false;
3027         }
3028         /* The first directory block must not be a hole,
3029          * so treat it as DIRENT_HTREE
3030          */
3031         bh = ext4_read_dirblock(inode, 0, DIRENT_HTREE);
3032         if (IS_ERR(bh))
3033                 return false;
3034
3035         de = (struct ext4_dir_entry_2 *) bh->b_data;
3036         if (ext4_check_dir_entry(inode, NULL, de, bh, bh->b_data, bh->b_size,
3037                                  0) ||
3038             le32_to_cpu(de->inode) != inode->i_ino || strcmp(".", de->name)) {
3039                 ext4_warning_inode(inode, "directory missing '.'");
3040                 brelse(bh);
3041                 return false;
3042         }
3043         offset = ext4_rec_len_from_disk(de->rec_len, sb->s_blocksize);
3044         de = ext4_next_entry(de, sb->s_blocksize);
3045         if (ext4_check_dir_entry(inode, NULL, de, bh, bh->b_data, bh->b_size,
3046                                  offset) ||
3047             le32_to_cpu(de->inode) == 0 || strcmp("..", de->name)) {
3048                 ext4_warning_inode(inode, "directory missing '..'");
3049                 brelse(bh);
3050                 return false;
3051         }
3052         offset += ext4_rec_len_from_disk(de->rec_len, sb->s_blocksize);
3053         while (offset < inode->i_size) {
3054                 if (!(offset & (sb->s_blocksize - 1))) {
3055                         unsigned int lblock;
3056                         brelse(bh);
3057                         lblock = offset >> EXT4_BLOCK_SIZE_BITS(sb);
3058                         bh = ext4_read_dirblock(inode, lblock, EITHER);
3059                         if (bh == NULL) {
3060                                 offset += sb->s_blocksize;
3061                                 continue;
3062                         }
3063                         if (IS_ERR(bh))
3064                                 return false;
3065                 }
3066                 de = (struct ext4_dir_entry_2 *) (bh->b_data +
3067                                         (offset & (sb->s_blocksize - 1)));
3068                 if (ext4_check_dir_entry(inode, NULL, de, bh,
3069                                          bh->b_data, bh->b_size, offset)) {
3070                         offset = (offset | (sb->s_blocksize - 1)) + 1;
3071                         continue;
3072                 }
3073                 if (le32_to_cpu(de->inode)) {
3074                         brelse(bh);
3075                         return false;
3076                 }
3077                 offset += ext4_rec_len_from_disk(de->rec_len, sb->s_blocksize);
3078         }
3079         brelse(bh);
3080         return true;
3081 }
3082
3083 static int ext4_rmdir(struct inode *dir, struct dentry *dentry)
3084 {
3085         int retval;
3086         struct inode *inode;
3087         struct buffer_head *bh;
3088         struct ext4_dir_entry_2 *de;
3089         handle_t *handle = NULL;
3090
3091         if (unlikely(ext4_forced_shutdown(EXT4_SB(dir->i_sb))))
3092                 return -EIO;
3093
3094         /* Initialize quotas before so that eventual writes go in
3095          * separate transaction */
3096         retval = dquot_initialize(dir);
3097         if (retval)
3098                 return retval;
3099         retval = dquot_initialize(d_inode(dentry));
3100         if (retval)
3101                 return retval;
3102
3103         retval = -ENOENT;
3104         bh = ext4_find_entry(dir, &dentry->d_name, &de, NULL);
3105         if (IS_ERR(bh))
3106                 return PTR_ERR(bh);
3107         if (!bh)
3108                 goto end_rmdir;
3109
3110         inode = d_inode(dentry);
3111
3112         retval = -EFSCORRUPTED;
3113         if (le32_to_cpu(de->inode) != inode->i_ino)
3114                 goto end_rmdir;
3115
3116         retval = -ENOTEMPTY;
3117         if (!ext4_empty_dir(inode))
3118                 goto end_rmdir;
3119
3120         handle = ext4_journal_start(dir, EXT4_HT_DIR,
3121                                     EXT4_DATA_TRANS_BLOCKS(dir->i_sb));
3122         if (IS_ERR(handle)) {
3123                 retval = PTR_ERR(handle);
3124                 handle = NULL;
3125                 goto end_rmdir;
3126         }
3127
3128         if (IS_DIRSYNC(dir))
3129                 ext4_handle_sync(handle);
3130
3131         retval = ext4_delete_entry(handle, dir, de, bh);
3132         if (retval)
3133                 goto end_rmdir;
3134         if (!EXT4_DIR_LINK_EMPTY(inode))
3135                 ext4_warning_inode(inode,
3136                              "empty directory '%.*s' has too many links (%u)",
3137                              dentry->d_name.len, dentry->d_name.name,
3138                              inode->i_nlink);
3139         inode_inc_iversion(inode);
3140         clear_nlink(inode);
3141         /* There's no need to set i_disksize: the fact that i_nlink is
3142          * zero will ensure that the right thing happens during any
3143          * recovery. */
3144         inode->i_size = 0;
3145         ext4_orphan_add(handle, inode);
3146         inode->i_ctime = dir->i_ctime = dir->i_mtime = current_time(inode);
3147         retval = ext4_mark_inode_dirty(handle, inode);
3148         if (retval)
3149                 goto end_rmdir;
3150         ext4_dec_count(dir);
3151         ext4_update_dx_flag(dir);
3152         ext4_fc_track_unlink(handle, dentry);
3153         retval = ext4_mark_inode_dirty(handle, dir);
3154
3155 #if IS_ENABLED(CONFIG_UNICODE)
3156         /* VFS negative dentries are incompatible with Encoding and
3157          * Case-insensitiveness. Eventually we'll want avoid
3158          * invalidating the dentries here, alongside with returning the
3159          * negative dentries at ext4_lookup(), when it is better
3160          * supported by the VFS for the CI case.
3161          */
3162         if (IS_CASEFOLDED(dir))
3163                 d_invalidate(dentry);
3164 #endif
3165
3166 end_rmdir:
3167         brelse(bh);
3168         if (handle)
3169                 ext4_journal_stop(handle);
3170         return retval;
3171 }
3172
3173 int __ext4_unlink(handle_t *handle, struct inode *dir, const struct qstr *d_name,
3174                   struct inode *inode)
3175 {
3176         int retval = -ENOENT;
3177         struct buffer_head *bh;
3178         struct ext4_dir_entry_2 *de;
3179         int skip_remove_dentry = 0;
3180
3181         bh = ext4_find_entry(dir, d_name, &de, NULL);
3182         if (IS_ERR(bh))
3183                 return PTR_ERR(bh);
3184
3185         if (!bh)
3186                 return -ENOENT;
3187
3188         if (le32_to_cpu(de->inode) != inode->i_ino) {
3189                 /*
3190                  * It's okay if we find dont find dentry which matches
3191                  * the inode. That's because it might have gotten
3192                  * renamed to a different inode number
3193                  */
3194                 if (EXT4_SB(inode->i_sb)->s_mount_state & EXT4_FC_REPLAY)
3195                         skip_remove_dentry = 1;
3196                 else
3197                         goto out;
3198         }
3199
3200         if (IS_DIRSYNC(dir))
3201                 ext4_handle_sync(handle);
3202
3203         if (!skip_remove_dentry) {
3204                 retval = ext4_delete_entry(handle, dir, de, bh);
3205                 if (retval)
3206                         goto out;
3207                 dir->i_ctime = dir->i_mtime = current_time(dir);
3208                 ext4_update_dx_flag(dir);
3209                 retval = ext4_mark_inode_dirty(handle, dir);
3210                 if (retval)
3211                         goto out;
3212         } else {
3213                 retval = 0;
3214         }
3215         if (inode->i_nlink == 0)
3216                 ext4_warning_inode(inode, "Deleting file '%.*s' with no links",
3217                                    d_name->len, d_name->name);
3218         else
3219                 drop_nlink(inode);
3220         if (!inode->i_nlink)
3221                 ext4_orphan_add(handle, inode);
3222         inode->i_ctime = current_time(inode);
3223         retval = ext4_mark_inode_dirty(handle, inode);
3224
3225 out:
3226         brelse(bh);
3227         return retval;
3228 }
3229
3230 static int ext4_unlink(struct inode *dir, struct dentry *dentry)
3231 {
3232         handle_t *handle;
3233         int retval;
3234
3235         if (unlikely(ext4_forced_shutdown(EXT4_SB(dir->i_sb))))
3236                 return -EIO;
3237
3238         trace_ext4_unlink_enter(dir, dentry);
3239         /*
3240          * Initialize quotas before so that eventual writes go
3241          * in separate transaction
3242          */
3243         retval = dquot_initialize(dir);
3244         if (retval)
3245                 goto out_trace;
3246         retval = dquot_initialize(d_inode(dentry));
3247         if (retval)
3248                 goto out_trace;
3249
3250         handle = ext4_journal_start(dir, EXT4_HT_DIR,
3251                                     EXT4_DATA_TRANS_BLOCKS(dir->i_sb));
3252         if (IS_ERR(handle)) {
3253                 retval = PTR_ERR(handle);
3254                 goto out_trace;
3255         }
3256
3257         retval = __ext4_unlink(handle, dir, &dentry->d_name, d_inode(dentry));
3258         if (!retval)
3259                 ext4_fc_track_unlink(handle, dentry);
3260 #if IS_ENABLED(CONFIG_UNICODE)
3261         /* VFS negative dentries are incompatible with Encoding and
3262          * Case-insensitiveness. Eventually we'll want avoid
3263          * invalidating the dentries here, alongside with returning the
3264          * negative dentries at ext4_lookup(), when it is  better
3265          * supported by the VFS for the CI case.
3266          */
3267         if (IS_CASEFOLDED(dir))
3268                 d_invalidate(dentry);
3269 #endif
3270         if (handle)
3271                 ext4_journal_stop(handle);
3272
3273 out_trace:
3274         trace_ext4_unlink_exit(dentry, retval);
3275         return retval;
3276 }
3277
3278 static int ext4_init_symlink_block(handle_t *handle, struct inode *inode,
3279                                    struct fscrypt_str *disk_link)
3280 {
3281         struct buffer_head *bh;
3282         char *kaddr;
3283         int err = 0;
3284
3285         bh = ext4_bread(handle, inode, 0, EXT4_GET_BLOCKS_CREATE);
3286         if (IS_ERR(bh))
3287                 return PTR_ERR(bh);
3288
3289         BUFFER_TRACE(bh, "get_write_access");
3290         err = ext4_journal_get_write_access(handle, inode->i_sb, bh, EXT4_JTR_NONE);
3291         if (err)
3292                 goto out;
3293
3294         kaddr = (char *)bh->b_data;
3295         memcpy(kaddr, disk_link->name, disk_link->len);
3296         inode->i_size = disk_link->len - 1;
3297         EXT4_I(inode)->i_disksize = inode->i_size;
3298         err = ext4_handle_dirty_metadata(handle, inode, bh);
3299 out:
3300         brelse(bh);
3301         return err;
3302 }
3303
3304 static int ext4_symlink(struct user_namespace *mnt_userns, struct inode *dir,
3305                         struct dentry *dentry, const char *symname)
3306 {
3307         handle_t *handle;
3308         struct inode *inode;
3309         int err, len = strlen(symname);
3310         int credits;
3311         struct fscrypt_str disk_link;
3312         int retries = 0;
3313
3314         if (unlikely(ext4_forced_shutdown(EXT4_SB(dir->i_sb))))
3315                 return -EIO;
3316
3317         err = fscrypt_prepare_symlink(dir, symname, len, dir->i_sb->s_blocksize,
3318                                       &disk_link);
3319         if (err)
3320                 return err;
3321
3322         err = dquot_initialize(dir);
3323         if (err)
3324                 return err;
3325
3326         /*
3327          * EXT4_INDEX_EXTRA_TRANS_BLOCKS for addition of entry into the
3328          * directory. +3 for inode, inode bitmap, group descriptor allocation.
3329          * EXT4_DATA_TRANS_BLOCKS for the data block allocation and
3330          * modification.
3331          */
3332         credits = EXT4_DATA_TRANS_BLOCKS(dir->i_sb) +
3333                   EXT4_INDEX_EXTRA_TRANS_BLOCKS + 3;
3334 retry:
3335         inode = ext4_new_inode_start_handle(mnt_userns, dir, S_IFLNK|S_IRWXUGO,
3336                                             &dentry->d_name, 0, NULL,
3337                                             EXT4_HT_DIR, credits);
3338         handle = ext4_journal_current_handle();
3339         if (IS_ERR(inode)) {
3340                 if (handle)
3341                         ext4_journal_stop(handle);
3342                 err = PTR_ERR(inode);
3343                 goto out_retry;
3344         }
3345
3346         if (IS_ENCRYPTED(inode)) {
3347                 err = fscrypt_encrypt_symlink(inode, symname, len, &disk_link);
3348                 if (err)
3349                         goto err_drop_inode;
3350                 inode->i_op = &ext4_encrypted_symlink_inode_operations;
3351         } else {
3352                 if ((disk_link.len > EXT4_N_BLOCKS * 4)) {
3353                         inode->i_op = &ext4_symlink_inode_operations;
3354                 } else {
3355                         inode->i_op = &ext4_fast_symlink_inode_operations;
3356                         inode->i_link = (char *)&EXT4_I(inode)->i_data;
3357                 }
3358         }
3359
3360         if ((disk_link.len > EXT4_N_BLOCKS * 4)) {
3361                 /* alloc symlink block and fill it */
3362                 err = ext4_init_symlink_block(handle, inode, &disk_link);
3363                 if (err)
3364                         goto err_drop_inode;
3365         } else {
3366                 /* clear the extent format for fast symlink */
3367                 ext4_clear_inode_flag(inode, EXT4_INODE_EXTENTS);
3368                 memcpy((char *)&EXT4_I(inode)->i_data, disk_link.name,
3369                        disk_link.len);
3370                 inode->i_size = disk_link.len - 1;
3371                 EXT4_I(inode)->i_disksize = inode->i_size;
3372         }
3373         err = ext4_add_nondir(handle, dentry, &inode);
3374         if (handle)
3375                 ext4_journal_stop(handle);
3376         iput(inode);
3377         goto out_retry;
3378
3379 err_drop_inode:
3380         clear_nlink(inode);
3381         ext4_orphan_add(handle, inode);
3382         unlock_new_inode(inode);
3383         if (handle)
3384                 ext4_journal_stop(handle);
3385         iput(inode);
3386 out_retry:
3387         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
3388                 goto retry;
3389         if (disk_link.name != (unsigned char *)symname)
3390                 kfree(disk_link.name);
3391         return err;
3392 }
3393
3394 int __ext4_link(struct inode *dir, struct inode *inode, struct dentry *dentry)
3395 {
3396         handle_t *handle;
3397         int err, retries = 0;
3398 retry:
3399         handle = ext4_journal_start(dir, EXT4_HT_DIR,
3400                 (EXT4_DATA_TRANS_BLOCKS(dir->i_sb) +
3401                  EXT4_INDEX_EXTRA_TRANS_BLOCKS) + 1);
3402         if (IS_ERR(handle))
3403                 return PTR_ERR(handle);
3404
3405         if (IS_DIRSYNC(dir))
3406                 ext4_handle_sync(handle);
3407
3408         inode->i_ctime = current_time(inode);
3409         ext4_inc_count(inode);
3410         ihold(inode);
3411
3412         err = ext4_add_entry(handle, dentry, inode);
3413         if (!err) {
3414                 err = ext4_mark_inode_dirty(handle, inode);
3415                 /* this can happen only for tmpfile being
3416                  * linked the first time
3417                  */
3418                 if (inode->i_nlink == 1)
3419                         ext4_orphan_del(handle, inode);
3420                 d_instantiate(dentry, inode);
3421                 ext4_fc_track_link(handle, dentry);
3422         } else {
3423                 drop_nlink(inode);
3424                 iput(inode);
3425         }
3426         ext4_journal_stop(handle);
3427         if (err == -ENOSPC && ext4_should_retry_alloc(dir->i_sb, &retries))
3428                 goto retry;
3429         return err;
3430 }
3431
3432 static int ext4_link(struct dentry *old_dentry,
3433                      struct inode *dir, struct dentry *dentry)
3434 {
3435         struct inode *inode = d_inode(old_dentry);
3436         int err;
3437
3438         if (inode->i_nlink >= EXT4_LINK_MAX)
3439                 return -EMLINK;
3440
3441         err = fscrypt_prepare_link(old_dentry, dir, dentry);
3442         if (err)
3443                 return err;
3444
3445         if ((ext4_test_inode_flag(dir, EXT4_INODE_PROJINHERIT)) &&
3446             (!projid_eq(EXT4_I(dir)->i_projid,
3447                         EXT4_I(old_dentry->d_inode)->i_projid)))
3448                 return -EXDEV;
3449
3450         err = dquot_initialize(dir);
3451         if (err)
3452                 return err;
3453         return __ext4_link(dir, inode, dentry);
3454 }
3455
3456 /*
3457  * Try to find buffer head where contains the parent block.
3458  * It should be the inode block if it is inlined or the 1st block
3459  * if it is a normal dir.
3460  */
3461 static struct buffer_head *ext4_get_first_dir_block(handle_t *handle,
3462                                         struct inode *inode,
3463                                         int *retval,
3464                                         struct ext4_dir_entry_2 **parent_de,
3465                                         int *inlined)
3466 {
3467         struct buffer_head *bh;
3468
3469         if (!ext4_has_inline_data(inode)) {
3470                 struct ext4_dir_entry_2 *de;
3471                 unsigned int offset;
3472
3473                 /* The first directory block must not be a hole, so
3474                  * treat it as DIRENT_HTREE
3475                  */
3476                 bh = ext4_read_dirblock(inode, 0, DIRENT_HTREE);
3477                 if (IS_ERR(bh)) {
3478                         *retval = PTR_ERR(bh);
3479                         return NULL;
3480                 }
3481
3482                 de = (struct ext4_dir_entry_2 *) bh->b_data;
3483                 if (ext4_check_dir_entry(inode, NULL, de, bh, bh->b_data,
3484                                          bh->b_size, 0) ||
3485                     le32_to_cpu(de->inode) != inode->i_ino ||
3486                     strcmp(".", de->name)) {
3487                         EXT4_ERROR_INODE(inode, "directory missing '.'");
3488                         brelse(bh);
3489                         *retval = -EFSCORRUPTED;
3490                         return NULL;
3491                 }
3492                 offset = ext4_rec_len_from_disk(de->rec_len,
3493                                                 inode->i_sb->s_blocksize);
3494                 de = ext4_next_entry(de, inode->i_sb->s_blocksize);
3495                 if (ext4_check_dir_entry(inode, NULL, de, bh, bh->b_data,
3496                                          bh->b_size, offset) ||
3497                     le32_to_cpu(de->inode) == 0 || strcmp("..", de->name)) {
3498                         EXT4_ERROR_INODE(inode, "directory missing '..'");
3499                         brelse(bh);
3500                         *retval = -EFSCORRUPTED;
3501                         return NULL;
3502                 }
3503                 *parent_de = de;
3504
3505                 return bh;
3506         }
3507
3508         *inlined = 1;
3509         return ext4_get_first_inline_block(inode, parent_de, retval);
3510 }
3511
3512 struct ext4_renament {
3513         struct inode *dir;
3514         struct dentry *dentry;
3515         struct inode *inode;
3516         bool is_dir;
3517         int dir_nlink_delta;
3518
3519         /* entry for "dentry" */
3520         struct buffer_head *bh;
3521         struct ext4_dir_entry_2 *de;
3522         int inlined;
3523
3524         /* entry for ".." in inode if it's a directory */
3525         struct buffer_head *dir_bh;
3526         struct ext4_dir_entry_2 *parent_de;
3527         int dir_inlined;
3528 };
3529
3530 static int ext4_rename_dir_prepare(handle_t *handle, struct ext4_renament *ent)
3531 {
3532         int retval;
3533
3534         ent->dir_bh = ext4_get_first_dir_block(handle, ent->inode,
3535                                               &retval, &ent->parent_de,
3536                                               &ent->dir_inlined);
3537         if (!ent->dir_bh)
3538                 return retval;
3539         if (le32_to_cpu(ent->parent_de->inode) != ent->dir->i_ino)
3540                 return -EFSCORRUPTED;
3541         BUFFER_TRACE(ent->dir_bh, "get_write_access");
3542         return ext4_journal_get_write_access(handle, ent->dir->i_sb,
3543                                              ent->dir_bh, EXT4_JTR_NONE);
3544 }
3545
3546 static int ext4_rename_dir_finish(handle_t *handle, struct ext4_renament *ent,
3547                                   unsigned dir_ino)
3548 {
3549         int retval;
3550
3551         ent->parent_de->inode = cpu_to_le32(dir_ino);
3552         BUFFER_TRACE(ent->dir_bh, "call ext4_handle_dirty_metadata");
3553         if (!ent->dir_inlined) {
3554                 if (is_dx(ent->inode)) {
3555                         retval = ext4_handle_dirty_dx_node(handle,
3556                                                            ent->inode,
3557                                                            ent->dir_bh);
3558                 } else {
3559                         retval = ext4_handle_dirty_dirblock(handle, ent->inode,
3560                                                             ent->dir_bh);
3561                 }
3562         } else {
3563                 retval = ext4_mark_inode_dirty(handle, ent->inode);
3564         }
3565         if (retval) {
3566                 ext4_std_error(ent->dir->i_sb, retval);
3567                 return retval;
3568         }
3569         return 0;
3570 }
3571
3572 static int ext4_setent(handle_t *handle, struct ext4_renament *ent,
3573                        unsigned ino, unsigned file_type)
3574 {
3575         int retval, retval2;
3576
3577         BUFFER_TRACE(ent->bh, "get write access");
3578         retval = ext4_journal_get_write_access(handle, ent->dir->i_sb, ent->bh,
3579                                                EXT4_JTR_NONE);
3580         if (retval)
3581                 return retval;
3582         ent->de->inode = cpu_to_le32(ino);
3583         if (ext4_has_feature_filetype(ent->dir->i_sb))
3584                 ent->de->file_type = file_type;
3585         inode_inc_iversion(ent->dir);
3586         ent->dir->i_ctime = ent->dir->i_mtime =
3587                 current_time(ent->dir);
3588         retval = ext4_mark_inode_dirty(handle, ent->dir);
3589         BUFFER_TRACE(ent->bh, "call ext4_handle_dirty_metadata");
3590         if (!ent->inlined) {
3591                 retval2 = ext4_handle_dirty_dirblock(handle, ent->dir, ent->bh);
3592                 if (unlikely(retval2)) {
3593                         ext4_std_error(ent->dir->i_sb, retval2);
3594                         return retval2;
3595                 }
3596         }
3597         return retval;
3598 }
3599
3600 static void ext4_resetent(handle_t *handle, struct ext4_renament *ent,
3601                           unsigned ino, unsigned file_type)
3602 {
3603         struct ext4_renament old = *ent;
3604         int retval = 0;
3605
3606         /*
3607          * old->de could have moved from under us during make indexed dir,
3608          * so the old->de may no longer valid and need to find it again
3609          * before reset old inode info.
3610          */
3611         old.bh = ext4_find_entry(old.dir, &old.dentry->d_name, &old.de, NULL);
3612         if (IS_ERR(old.bh))
3613                 retval = PTR_ERR(old.bh);
3614         if (!old.bh)
3615                 retval = -ENOENT;
3616         if (retval) {
3617                 ext4_std_error(old.dir->i_sb, retval);
3618                 return;
3619         }
3620
3621         ext4_setent(handle, &old, ino, file_type);
3622         brelse(old.bh);
3623 }
3624
3625 static int ext4_find_delete_entry(handle_t *handle, struct inode *dir,
3626                                   const struct qstr *d_name)
3627 {
3628         int retval = -ENOENT;
3629         struct buffer_head *bh;
3630         struct ext4_dir_entry_2 *de;
3631
3632         bh = ext4_find_entry(dir, d_name, &de, NULL);
3633         if (IS_ERR(bh))
3634                 return PTR_ERR(bh);
3635         if (bh) {
3636                 retval = ext4_delete_entry(handle, dir, de, bh);
3637                 brelse(bh);
3638         }
3639         return retval;
3640 }
3641
3642 static void ext4_rename_delete(handle_t *handle, struct ext4_renament *ent,
3643                                int force_reread)
3644 {
3645         int retval;
3646         /*
3647          * ent->de could have moved from under us during htree split, so make
3648          * sure that we are deleting the right entry.  We might also be pointing
3649          * to a stale entry in the unused part of ent->bh so just checking inum
3650          * and the name isn't enough.
3651          */
3652         if (le32_to_cpu(ent->de->inode) != ent->inode->i_ino ||
3653             ent->de->name_len != ent->dentry->d_name.len ||
3654             strncmp(ent->de->name, ent->dentry->d_name.name,
3655                     ent->de->name_len) ||
3656             force_reread) {
3657                 retval = ext4_find_delete_entry(handle, ent->dir,
3658                                                 &ent->dentry->d_name);
3659         } else {
3660                 retval = ext4_delete_entry(handle, ent->dir, ent->de, ent->bh);
3661                 if (retval == -ENOENT) {
3662                         retval = ext4_find_delete_entry(handle, ent->dir,
3663                                                         &ent->dentry->d_name);
3664                 }
3665         }
3666
3667         if (retval) {
3668                 ext4_warning_inode(ent->dir,
3669                                    "Deleting old file: nlink %d, error=%d",
3670                                    ent->dir->i_nlink, retval);
3671         }
3672 }
3673
3674 static void ext4_update_dir_count(handle_t *handle, struct ext4_renament *ent)
3675 {
3676         if (ent->dir_nlink_delta) {
3677                 if (ent->dir_nlink_delta == -1)
3678                         ext4_dec_count(ent->dir);
3679                 else
3680                         ext4_inc_count(ent->dir);
3681                 ext4_mark_inode_dirty(handle, ent->dir);
3682         }
3683 }
3684
3685 static struct inode *ext4_whiteout_for_rename(struct user_namespace *mnt_userns,
3686                                               struct ext4_renament *ent,
3687                                               int credits, handle_t **h)
3688 {
3689         struct inode *wh;
3690         handle_t *handle;
3691         int retries = 0;
3692
3693         /*
3694          * for inode block, sb block, group summaries,
3695          * and inode bitmap
3696          */
3697         credits += (EXT4_MAXQUOTAS_TRANS_BLOCKS(ent->dir->i_sb) +
3698                     EXT4_XATTR_TRANS_BLOCKS + 4);
3699 retry:
3700         wh = ext4_new_inode_start_handle(mnt_userns, ent->dir,
3701                                          S_IFCHR | WHITEOUT_MODE,
3702                                          &ent->dentry->d_name, 0, NULL,
3703                                          EXT4_HT_DIR, credits);
3704
3705         handle = ext4_journal_current_handle();
3706         if (IS_ERR(wh)) {
3707                 if (handle)
3708                         ext4_journal_stop(handle);
3709                 if (PTR_ERR(wh) == -ENOSPC &&
3710                     ext4_should_retry_alloc(ent->dir->i_sb, &retries))
3711                         goto retry;
3712         } else {
3713                 *h = handle;
3714                 init_special_inode(wh, wh->i_mode, WHITEOUT_DEV);
3715                 wh->i_op = &ext4_special_inode_operations;
3716         }
3717         return wh;
3718 }
3719
3720 /*
3721  * Anybody can rename anything with this: the permission checks are left to the
3722  * higher-level routines.
3723  *
3724  * n.b.  old_{dentry,inode) refers to the source dentry/inode
3725  * while new_{dentry,inode) refers to the destination dentry/inode
3726  * This comes from rename(const char *oldpath, const char *newpath)
3727  */
3728 static int ext4_rename(struct user_namespace *mnt_userns, struct inode *old_dir,
3729                        struct dentry *old_dentry, struct inode *new_dir,
3730                        struct dentry *new_dentry, unsigned int flags)
3731 {
3732         handle_t *handle = NULL;
3733         struct ext4_renament old = {
3734                 .dir = old_dir,
3735                 .dentry = old_dentry,
3736                 .inode = d_inode(old_dentry),
3737         };
3738         struct ext4_renament new = {
3739                 .dir = new_dir,
3740                 .dentry = new_dentry,
3741                 .inode = d_inode(new_dentry),
3742         };
3743         int force_reread;
3744         int retval;
3745         struct inode *whiteout = NULL;
3746         int credits;
3747         u8 old_file_type;
3748
3749         if (new.inode && new.inode->i_nlink == 0) {
3750                 EXT4_ERROR_INODE(new.inode,
3751                                  "target of rename is already freed");
3752                 return -EFSCORRUPTED;
3753         }
3754
3755         if ((ext4_test_inode_flag(new_dir, EXT4_INODE_PROJINHERIT)) &&
3756             (!projid_eq(EXT4_I(new_dir)->i_projid,
3757                         EXT4_I(old_dentry->d_inode)->i_projid)))
3758                 return -EXDEV;
3759
3760         retval = dquot_initialize(old.dir);
3761         if (retval)
3762                 return retval;
3763         retval = dquot_initialize(new.dir);
3764         if (retval)
3765                 return retval;
3766
3767         /* Initialize quotas before so that eventual writes go
3768          * in separate transaction */
3769         if (new.inode) {
3770                 retval = dquot_initialize(new.inode);
3771                 if (retval)
3772                         return retval;
3773         }
3774
3775         old.bh = ext4_find_entry(old.dir, &old.dentry->d_name, &old.de, NULL);
3776         if (IS_ERR(old.bh))
3777                 return PTR_ERR(old.bh);
3778         /*
3779          *  Check for inode number is _not_ due to possible IO errors.
3780          *  We might rmdir the source, keep it as pwd of some process
3781          *  and merrily kill the link to whatever was created under the
3782          *  same name. Goodbye sticky bit ;-<
3783          */
3784         retval = -ENOENT;
3785         if (!old.bh || le32_to_cpu(old.de->inode) != old.inode->i_ino)
3786                 goto release_bh;
3787
3788         new.bh = ext4_find_entry(new.dir, &new.dentry->d_name,
3789                                  &new.de, &new.inlined);
3790         if (IS_ERR(new.bh)) {
3791                 retval = PTR_ERR(new.bh);
3792                 new.bh = NULL;
3793                 goto release_bh;
3794         }
3795         if (new.bh) {
3796                 if (!new.inode) {
3797                         brelse(new.bh);
3798                         new.bh = NULL;
3799                 }
3800         }
3801         if (new.inode && !test_opt(new.dir->i_sb, NO_AUTO_DA_ALLOC))
3802                 ext4_alloc_da_blocks(old.inode);
3803
3804         credits = (2 * EXT4_DATA_TRANS_BLOCKS(old.dir->i_sb) +
3805                    EXT4_INDEX_EXTRA_TRANS_BLOCKS + 2);
3806         if (!(flags & RENAME_WHITEOUT)) {
3807                 handle = ext4_journal_start(old.dir, EXT4_HT_DIR, credits);
3808                 if (IS_ERR(handle)) {
3809                         retval = PTR_ERR(handle);
3810                         goto release_bh;
3811                 }
3812         } else {
3813                 whiteout = ext4_whiteout_for_rename(mnt_userns, &old, credits, &handle);
3814                 if (IS_ERR(whiteout)) {
3815                         retval = PTR_ERR(whiteout);
3816                         goto release_bh;
3817                 }
3818         }
3819
3820         old_file_type = old.de->file_type;
3821         if (IS_DIRSYNC(old.dir) || IS_DIRSYNC(new.dir))
3822                 ext4_handle_sync(handle);
3823
3824         if (S_ISDIR(old.inode->i_mode)) {
3825                 if (new.inode) {
3826                         retval = -ENOTEMPTY;
3827                         if (!ext4_empty_dir(new.inode))
3828                                 goto end_rename;
3829                 } else {
3830                         retval = -EMLINK;
3831                         if (new.dir != old.dir && EXT4_DIR_LINK_MAX(new.dir))
3832                                 goto end_rename;
3833                 }
3834                 retval = ext4_rename_dir_prepare(handle, &old);
3835                 if (retval)
3836                         goto end_rename;
3837         }
3838         /*
3839          * If we're renaming a file within an inline_data dir and adding or
3840          * setting the new dirent causes a conversion from inline_data to
3841          * extents/blockmap, we need to force the dirent delete code to
3842          * re-read the directory, or else we end up trying to delete a dirent
3843          * from what is now the extent tree root (or a block map).
3844          */
3845         force_reread = (new.dir->i_ino == old.dir->i_ino &&
3846                         ext4_test_inode_flag(new.dir, EXT4_INODE_INLINE_DATA));
3847
3848         if (whiteout) {
3849                 /*
3850                  * Do this before adding a new entry, so the old entry is sure
3851                  * to be still pointing to the valid old entry.
3852                  */
3853                 retval = ext4_setent(handle, &old, whiteout->i_ino,
3854                                      EXT4_FT_CHRDEV);
3855                 if (retval)
3856                         goto end_rename;
3857                 retval = ext4_mark_inode_dirty(handle, whiteout);
3858                 if (unlikely(retval))
3859                         goto end_rename;
3860
3861         }
3862         if (!new.bh) {
3863                 retval = ext4_add_entry(handle, new.dentry, old.inode);
3864                 if (retval)
3865                         goto end_rename;
3866         } else {
3867                 retval = ext4_setent(handle, &new,
3868                                      old.inode->i_ino, old_file_type);
3869                 if (retval)
3870                         goto end_rename;
3871         }
3872         if (force_reread)
3873                 force_reread = !ext4_test_inode_flag(new.dir,
3874                                                      EXT4_INODE_INLINE_DATA);
3875
3876         /*
3877          * Like most other Unix systems, set the ctime for inodes on a
3878          * rename.
3879          */
3880         old.inode->i_ctime = current_time(old.inode);
3881         retval = ext4_mark_inode_dirty(handle, old.inode);
3882         if (unlikely(retval))
3883                 goto end_rename;
3884
3885         if (!whiteout) {
3886                 /*
3887                  * ok, that's it
3888                  */
3889                 ext4_rename_delete(handle, &old, force_reread);
3890         }
3891
3892         if (new.inode) {
3893                 ext4_dec_count(new.inode);
3894                 new.inode->i_ctime = current_time(new.inode);
3895         }
3896         old.dir->i_ctime = old.dir->i_mtime = current_time(old.dir);
3897         ext4_update_dx_flag(old.dir);
3898         if (old.dir_bh) {
3899                 retval = ext4_rename_dir_finish(handle, &old, new.dir->i_ino);
3900                 if (retval)
3901                         goto end_rename;
3902
3903                 ext4_dec_count(old.dir);
3904                 if (new.inode) {
3905                         /* checked ext4_empty_dir above, can't have another
3906                          * parent, ext4_dec_count() won't work for many-linked
3907                          * dirs */
3908                         clear_nlink(new.inode);
3909                 } else {
3910                         ext4_inc_count(new.dir);
3911                         ext4_update_dx_flag(new.dir);
3912                         retval = ext4_mark_inode_dirty(handle, new.dir);
3913                         if (unlikely(retval))
3914                                 goto end_rename;
3915                 }
3916         }
3917         retval = ext4_mark_inode_dirty(handle, old.dir);
3918         if (unlikely(retval))
3919                 goto end_rename;
3920
3921         if (S_ISDIR(old.inode->i_mode)) {
3922                 /*
3923                  * We disable fast commits here that's because the
3924                  * replay code is not yet capable of changing dot dot
3925                  * dirents in directories.
3926                  */
3927                 ext4_fc_mark_ineligible(old.inode->i_sb,
3928                         EXT4_FC_REASON_RENAME_DIR, handle);
3929         } else {
3930                 struct super_block *sb = old.inode->i_sb;
3931
3932                 if (new.inode)
3933                         ext4_fc_track_unlink(handle, new.dentry);
3934                 if (test_opt2(sb, JOURNAL_FAST_COMMIT) &&
3935                     !(EXT4_SB(sb)->s_mount_state & EXT4_FC_REPLAY) &&
3936                     !(ext4_test_mount_flag(sb, EXT4_MF_FC_INELIGIBLE))) {
3937                         __ext4_fc_track_link(handle, old.inode, new.dentry);
3938                         __ext4_fc_track_unlink(handle, old.inode, old.dentry);
3939                         if (whiteout)
3940                                 __ext4_fc_track_create(handle, whiteout,
3941                                                        old.dentry);
3942                 }
3943         }
3944
3945         if (new.inode) {
3946                 retval = ext4_mark_inode_dirty(handle, new.inode);
3947                 if (unlikely(retval))
3948                         goto end_rename;
3949                 if (!new.inode->i_nlink)
3950                         ext4_orphan_add(handle, new.inode);
3951         }
3952         retval = 0;
3953
3954 end_rename:
3955         if (whiteout) {
3956                 if (retval) {
3957                         ext4_resetent(handle, &old,
3958                                       old.inode->i_ino, old_file_type);
3959                         drop_nlink(whiteout);
3960                         ext4_orphan_add(handle, whiteout);
3961                 }
3962                 unlock_new_inode(whiteout);
3963                 ext4_journal_stop(handle);
3964                 iput(whiteout);
3965         } else {
3966                 ext4_journal_stop(handle);
3967         }
3968 release_bh:
3969         brelse(old.dir_bh);
3970         brelse(old.bh);
3971         brelse(new.bh);
3972         return retval;
3973 }
3974
3975 static int ext4_cross_rename(struct inode *old_dir, struct dentry *old_dentry,
3976                              struct inode *new_dir, struct dentry *new_dentry)
3977 {
3978         handle_t *handle = NULL;
3979         struct ext4_renament old = {
3980                 .dir = old_dir,
3981                 .dentry = old_dentry,
3982                 .inode = d_inode(old_dentry),
3983         };
3984         struct ext4_renament new = {
3985                 .dir = new_dir,
3986                 .dentry = new_dentry,
3987                 .inode = d_inode(new_dentry),
3988         };
3989         u8 new_file_type;
3990         int retval;
3991         struct timespec64 ctime;
3992
3993         if ((ext4_test_inode_flag(new_dir, EXT4_INODE_PROJINHERIT) &&
3994              !projid_eq(EXT4_I(new_dir)->i_projid,
3995                         EXT4_I(old_dentry->d_inode)->i_projid)) ||
3996             (ext4_test_inode_flag(old_dir, EXT4_INODE_PROJINHERIT) &&
3997              !projid_eq(EXT4_I(old_dir)->i_projid,
3998                         EXT4_I(new_dentry->d_inode)->i_projid)))
3999                 return -EXDEV;
4000
4001         retval = dquot_initialize(old.dir);
4002         if (retval)
4003                 return retval;
4004         retval = dquot_initialize(new.dir);
4005         if (retval)
4006                 return retval;
4007
4008         old.bh = ext4_find_entry(old.dir, &old.dentry->d_name,
4009                                  &old.de, &old.inlined);
4010         if (IS_ERR(old.bh))
4011                 return PTR_ERR(old.bh);
4012         /*
4013          *  Check for inode number is _not_ due to possible IO errors.
4014          *  We might rmdir the source, keep it as pwd of some process
4015          *  and merrily kill the link to whatever was created under the
4016          *  same name. Goodbye sticky bit ;-<
4017          */
4018         retval = -ENOENT;
4019         if (!old.bh || le32_to_cpu(old.de->inode) != old.inode->i_ino)
4020                 goto end_rename;
4021
4022         new.bh = ext4_find_entry(new.dir, &new.dentry->d_name,
4023                                  &new.de, &new.inlined);
4024         if (IS_ERR(new.bh)) {
4025                 retval = PTR_ERR(new.bh);
4026                 new.bh = NULL;
4027                 goto end_rename;
4028         }
4029
4030         /* RENAME_EXCHANGE case: old *and* new must both exist */
4031         if (!new.bh || le32_to_cpu(new.de->inode) != new.inode->i_ino)
4032                 goto end_rename;
4033
4034         handle = ext4_journal_start(old.dir, EXT4_HT_DIR,
4035                 (2 * EXT4_DATA_TRANS_BLOCKS(old.dir->i_sb) +
4036                  2 * EXT4_INDEX_EXTRA_TRANS_BLOCKS + 2));
4037         if (IS_ERR(handle)) {
4038                 retval = PTR_ERR(handle);
4039                 handle = NULL;
4040                 goto end_rename;
4041         }
4042
4043         if (IS_DIRSYNC(old.dir) || IS_DIRSYNC(new.dir))
4044                 ext4_handle_sync(handle);
4045
4046         if (S_ISDIR(old.inode->i_mode)) {
4047                 old.is_dir = true;
4048                 retval = ext4_rename_dir_prepare(handle, &old);
4049                 if (retval)
4050                         goto end_rename;
4051         }
4052         if (S_ISDIR(new.inode->i_mode)) {
4053                 new.is_dir = true;
4054                 retval = ext4_rename_dir_prepare(handle, &new);
4055                 if (retval)
4056                         goto end_rename;
4057         }
4058
4059         /*
4060          * Other than the special case of overwriting a directory, parents'
4061          * nlink only needs to be modified if this is a cross directory rename.
4062          */
4063         if (old.dir != new.dir && old.is_dir != new.is_dir) {
4064                 old.dir_nlink_delta = old.is_dir ? -1 : 1;
4065                 new.dir_nlink_delta = -old.dir_nlink_delta;
4066                 retval = -EMLINK;
4067                 if ((old.dir_nlink_delta > 0 && EXT4_DIR_LINK_MAX(old.dir)) ||
4068                     (new.dir_nlink_delta > 0 && EXT4_DIR_LINK_MAX(new.dir)))
4069                         goto end_rename;
4070         }
4071
4072         new_file_type = new.de->file_type;
4073         retval = ext4_setent(handle, &new, old.inode->i_ino, old.de->file_type);
4074         if (retval)
4075                 goto end_rename;
4076
4077         retval = ext4_setent(handle, &old, new.inode->i_ino, new_file_type);
4078         if (retval)
4079                 goto end_rename;
4080
4081         /*
4082          * Like most other Unix systems, set the ctime for inodes on a
4083          * rename.
4084          */
4085         ctime = current_time(old.inode);
4086         old.inode->i_ctime = ctime;
4087         new.inode->i_ctime = ctime;
4088         retval = ext4_mark_inode_dirty(handle, old.inode);
4089         if (unlikely(retval))
4090                 goto end_rename;
4091         retval = ext4_mark_inode_dirty(handle, new.inode);
4092         if (unlikely(retval))
4093                 goto end_rename;
4094         ext4_fc_mark_ineligible(new.inode->i_sb,
4095                                 EXT4_FC_REASON_CROSS_RENAME, handle);
4096         if (old.dir_bh) {
4097                 retval = ext4_rename_dir_finish(handle, &old, new.dir->i_ino);
4098                 if (retval)
4099                         goto end_rename;
4100         }
4101         if (new.dir_bh) {
4102                 retval = ext4_rename_dir_finish(handle, &new, old.dir->i_ino);
4103                 if (retval)
4104                         goto end_rename;
4105         }
4106         ext4_update_dir_count(handle, &old);
4107         ext4_update_dir_count(handle, &new);
4108         retval = 0;
4109
4110 end_rename:
4111         brelse(old.dir_bh);
4112         brelse(new.dir_bh);
4113         brelse(old.bh);
4114         brelse(new.bh);
4115         if (handle)
4116                 ext4_journal_stop(handle);
4117         return retval;
4118 }
4119
4120 static int ext4_rename2(struct user_namespace *mnt_userns,
4121                         struct inode *old_dir, struct dentry *old_dentry,
4122                         struct inode *new_dir, struct dentry *new_dentry,
4123                         unsigned int flags)
4124 {
4125         int err;
4126
4127         if (unlikely(ext4_forced_shutdown(EXT4_SB(old_dir->i_sb))))
4128                 return -EIO;
4129
4130         if (flags & ~(RENAME_NOREPLACE | RENAME_EXCHANGE | RENAME_WHITEOUT))
4131                 return -EINVAL;
4132
4133         err = fscrypt_prepare_rename(old_dir, old_dentry, new_dir, new_dentry,
4134                                      flags);
4135         if (err)
4136                 return err;
4137
4138         if (flags & RENAME_EXCHANGE) {
4139                 return ext4_cross_rename(old_dir, old_dentry,
4140                                          new_dir, new_dentry);
4141         }
4142
4143         return ext4_rename(mnt_userns, old_dir, old_dentry, new_dir, new_dentry, flags);
4144 }
4145
4146 /*
4147  * directories can handle most operations...
4148  */
4149 const struct inode_operations ext4_dir_inode_operations = {
4150         .create         = ext4_create,
4151         .lookup         = ext4_lookup,
4152         .link           = ext4_link,
4153         .unlink         = ext4_unlink,
4154         .symlink        = ext4_symlink,
4155         .mkdir          = ext4_mkdir,
4156         .rmdir          = ext4_rmdir,
4157         .mknod          = ext4_mknod,
4158         .tmpfile        = ext4_tmpfile,
4159         .rename         = ext4_rename2,
4160         .setattr        = ext4_setattr,
4161         .getattr        = ext4_getattr,
4162         .listxattr      = ext4_listxattr,
4163         .get_acl        = ext4_get_acl,
4164         .set_acl        = ext4_set_acl,
4165         .fiemap         = ext4_fiemap,
4166         .fileattr_get   = ext4_fileattr_get,
4167         .fileattr_set   = ext4_fileattr_set,
4168 };
4169
4170 const struct inode_operations ext4_special_inode_operations = {
4171         .setattr        = ext4_setattr,
4172         .getattr        = ext4_getattr,
4173         .listxattr      = ext4_listxattr,
4174         .get_acl        = ext4_get_acl,
4175         .set_acl        = ext4_set_acl,
4176 };