GNU Linux-libre 5.10.215-gnu1
[releases.git] / fs / nilfs2 / the_nilfs.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * the_nilfs.c - the_nilfs shared structure.
4  *
5  * Copyright (C) 2005-2008 Nippon Telegraph and Telephone Corporation.
6  *
7  * Written by Ryusuke Konishi.
8  *
9  */
10
11 #include <linux/buffer_head.h>
12 #include <linux/slab.h>
13 #include <linux/blkdev.h>
14 #include <linux/backing-dev.h>
15 #include <linux/random.h>
16 #include <linux/log2.h>
17 #include <linux/crc32.h>
18 #include "nilfs.h"
19 #include "segment.h"
20 #include "alloc.h"
21 #include "cpfile.h"
22 #include "sufile.h"
23 #include "dat.h"
24 #include "segbuf.h"
25
26
27 static int nilfs_valid_sb(struct nilfs_super_block *sbp);
28
29 void nilfs_set_last_segment(struct the_nilfs *nilfs,
30                             sector_t start_blocknr, u64 seq, __u64 cno)
31 {
32         spin_lock(&nilfs->ns_last_segment_lock);
33         nilfs->ns_last_pseg = start_blocknr;
34         nilfs->ns_last_seq = seq;
35         nilfs->ns_last_cno = cno;
36
37         if (!nilfs_sb_dirty(nilfs)) {
38                 if (nilfs->ns_prev_seq == nilfs->ns_last_seq)
39                         goto stay_cursor;
40
41                 set_nilfs_sb_dirty(nilfs);
42         }
43         nilfs->ns_prev_seq = nilfs->ns_last_seq;
44
45  stay_cursor:
46         spin_unlock(&nilfs->ns_last_segment_lock);
47 }
48
49 /**
50  * alloc_nilfs - allocate a nilfs object
51  * @sb: super block instance
52  *
53  * Return Value: On success, pointer to the_nilfs is returned.
54  * On error, NULL is returned.
55  */
56 struct the_nilfs *alloc_nilfs(struct super_block *sb)
57 {
58         struct the_nilfs *nilfs;
59
60         nilfs = kzalloc(sizeof(*nilfs), GFP_KERNEL);
61         if (!nilfs)
62                 return NULL;
63
64         nilfs->ns_sb = sb;
65         nilfs->ns_bdev = sb->s_bdev;
66         atomic_set(&nilfs->ns_ndirtyblks, 0);
67         init_rwsem(&nilfs->ns_sem);
68         mutex_init(&nilfs->ns_snapshot_mount_mutex);
69         INIT_LIST_HEAD(&nilfs->ns_dirty_files);
70         INIT_LIST_HEAD(&nilfs->ns_gc_inodes);
71         spin_lock_init(&nilfs->ns_inode_lock);
72         spin_lock_init(&nilfs->ns_next_gen_lock);
73         spin_lock_init(&nilfs->ns_last_segment_lock);
74         nilfs->ns_cptree = RB_ROOT;
75         spin_lock_init(&nilfs->ns_cptree_lock);
76         init_rwsem(&nilfs->ns_segctor_sem);
77         nilfs->ns_sb_update_freq = NILFS_SB_FREQ;
78
79         return nilfs;
80 }
81
82 /**
83  * destroy_nilfs - destroy nilfs object
84  * @nilfs: nilfs object to be released
85  */
86 void destroy_nilfs(struct the_nilfs *nilfs)
87 {
88         might_sleep();
89         if (nilfs_init(nilfs)) {
90                 brelse(nilfs->ns_sbh[0]);
91                 brelse(nilfs->ns_sbh[1]);
92         }
93         kfree(nilfs);
94 }
95
96 static int nilfs_load_super_root(struct the_nilfs *nilfs,
97                                  struct super_block *sb, sector_t sr_block)
98 {
99         struct buffer_head *bh_sr;
100         struct nilfs_super_root *raw_sr;
101         struct nilfs_super_block **sbp = nilfs->ns_sbp;
102         struct nilfs_inode *rawi;
103         unsigned int dat_entry_size, segment_usage_size, checkpoint_size;
104         unsigned int inode_size;
105         int err;
106
107         err = nilfs_read_super_root_block(nilfs, sr_block, &bh_sr, 1);
108         if (unlikely(err))
109                 return err;
110
111         down_read(&nilfs->ns_sem);
112         dat_entry_size = le16_to_cpu(sbp[0]->s_dat_entry_size);
113         checkpoint_size = le16_to_cpu(sbp[0]->s_checkpoint_size);
114         segment_usage_size = le16_to_cpu(sbp[0]->s_segment_usage_size);
115         up_read(&nilfs->ns_sem);
116
117         inode_size = nilfs->ns_inode_size;
118
119         rawi = (void *)bh_sr->b_data + NILFS_SR_DAT_OFFSET(inode_size);
120         err = nilfs_dat_read(sb, dat_entry_size, rawi, &nilfs->ns_dat);
121         if (err)
122                 goto failed;
123
124         rawi = (void *)bh_sr->b_data + NILFS_SR_CPFILE_OFFSET(inode_size);
125         err = nilfs_cpfile_read(sb, checkpoint_size, rawi, &nilfs->ns_cpfile);
126         if (err)
127                 goto failed_dat;
128
129         rawi = (void *)bh_sr->b_data + NILFS_SR_SUFILE_OFFSET(inode_size);
130         err = nilfs_sufile_read(sb, segment_usage_size, rawi,
131                                 &nilfs->ns_sufile);
132         if (err)
133                 goto failed_cpfile;
134
135         raw_sr = (struct nilfs_super_root *)bh_sr->b_data;
136         nilfs->ns_nongc_ctime = le64_to_cpu(raw_sr->sr_nongc_ctime);
137
138  failed:
139         brelse(bh_sr);
140         return err;
141
142  failed_cpfile:
143         iput(nilfs->ns_cpfile);
144
145  failed_dat:
146         iput(nilfs->ns_dat);
147         goto failed;
148 }
149
150 static void nilfs_init_recovery_info(struct nilfs_recovery_info *ri)
151 {
152         memset(ri, 0, sizeof(*ri));
153         INIT_LIST_HEAD(&ri->ri_used_segments);
154 }
155
156 static void nilfs_clear_recovery_info(struct nilfs_recovery_info *ri)
157 {
158         nilfs_dispose_segment_list(&ri->ri_used_segments);
159 }
160
161 /**
162  * nilfs_store_log_cursor - load log cursor from a super block
163  * @nilfs: nilfs object
164  * @sbp: buffer storing super block to be read
165  *
166  * nilfs_store_log_cursor() reads the last position of the log
167  * containing a super root from a given super block, and initializes
168  * relevant information on the nilfs object preparatory for log
169  * scanning and recovery.
170  */
171 static int nilfs_store_log_cursor(struct the_nilfs *nilfs,
172                                   struct nilfs_super_block *sbp)
173 {
174         int ret = 0;
175
176         nilfs->ns_last_pseg = le64_to_cpu(sbp->s_last_pseg);
177         nilfs->ns_last_cno = le64_to_cpu(sbp->s_last_cno);
178         nilfs->ns_last_seq = le64_to_cpu(sbp->s_last_seq);
179
180         nilfs->ns_prev_seq = nilfs->ns_last_seq;
181         nilfs->ns_seg_seq = nilfs->ns_last_seq;
182         nilfs->ns_segnum =
183                 nilfs_get_segnum_of_block(nilfs, nilfs->ns_last_pseg);
184         nilfs->ns_cno = nilfs->ns_last_cno + 1;
185         if (nilfs->ns_segnum >= nilfs->ns_nsegments) {
186                 nilfs_err(nilfs->ns_sb,
187                           "pointed segment number is out of range: segnum=%llu, nsegments=%lu",
188                           (unsigned long long)nilfs->ns_segnum,
189                           nilfs->ns_nsegments);
190                 ret = -EINVAL;
191         }
192         return ret;
193 }
194
195 /**
196  * nilfs_get_blocksize - get block size from raw superblock data
197  * @sb: super block instance
198  * @sbp: superblock raw data buffer
199  * @blocksize: place to store block size
200  *
201  * nilfs_get_blocksize() calculates the block size from the block size
202  * exponent information written in @sbp and stores it in @blocksize,
203  * or aborts with an error message if it's too large.
204  *
205  * Return Value: On success, 0 is returned. If the block size is too
206  * large, -EINVAL is returned.
207  */
208 static int nilfs_get_blocksize(struct super_block *sb,
209                                struct nilfs_super_block *sbp, int *blocksize)
210 {
211         unsigned int shift_bits = le32_to_cpu(sbp->s_log_block_size);
212
213         if (unlikely(shift_bits >
214                      ilog2(NILFS_MAX_BLOCK_SIZE) - BLOCK_SIZE_BITS)) {
215                 nilfs_err(sb, "too large filesystem blocksize: 2 ^ %u KiB",
216                           shift_bits);
217                 return -EINVAL;
218         }
219         *blocksize = BLOCK_SIZE << shift_bits;
220         return 0;
221 }
222
223 /**
224  * load_nilfs - load and recover the nilfs
225  * @nilfs: the_nilfs structure to be released
226  * @sb: super block isntance used to recover past segment
227  *
228  * load_nilfs() searches and load the latest super root,
229  * attaches the last segment, and does recovery if needed.
230  * The caller must call this exclusively for simultaneous mounts.
231  */
232 int load_nilfs(struct the_nilfs *nilfs, struct super_block *sb)
233 {
234         struct nilfs_recovery_info ri;
235         unsigned int s_flags = sb->s_flags;
236         int really_read_only = bdev_read_only(nilfs->ns_bdev);
237         int valid_fs = nilfs_valid_fs(nilfs);
238         int err;
239
240         if (!valid_fs) {
241                 nilfs_warn(sb, "mounting unchecked fs");
242                 if (s_flags & SB_RDONLY) {
243                         nilfs_info(sb,
244                                    "recovery required for readonly filesystem");
245                         nilfs_info(sb,
246                                    "write access will be enabled during recovery");
247                 }
248         }
249
250         nilfs_init_recovery_info(&ri);
251
252         err = nilfs_search_super_root(nilfs, &ri);
253         if (unlikely(err)) {
254                 struct nilfs_super_block **sbp = nilfs->ns_sbp;
255                 int blocksize;
256
257                 if (err != -EINVAL)
258                         goto scan_error;
259
260                 if (!nilfs_valid_sb(sbp[1])) {
261                         nilfs_warn(sb,
262                                    "unable to fall back to spare super block");
263                         goto scan_error;
264                 }
265                 nilfs_info(sb, "trying rollback from an earlier position");
266
267                 /*
268                  * restore super block with its spare and reconfigure
269                  * relevant states of the nilfs object.
270                  */
271                 memcpy(sbp[0], sbp[1], nilfs->ns_sbsize);
272                 nilfs->ns_crc_seed = le32_to_cpu(sbp[0]->s_crc_seed);
273                 nilfs->ns_sbwtime = le64_to_cpu(sbp[0]->s_wtime);
274
275                 /* verify consistency between two super blocks */
276                 err = nilfs_get_blocksize(sb, sbp[0], &blocksize);
277                 if (err)
278                         goto scan_error;
279
280                 if (blocksize != nilfs->ns_blocksize) {
281                         nilfs_warn(sb,
282                                    "blocksize differs between two super blocks (%d != %d)",
283                                    blocksize, nilfs->ns_blocksize);
284                         err = -EINVAL;
285                         goto scan_error;
286                 }
287
288                 err = nilfs_store_log_cursor(nilfs, sbp[0]);
289                 if (err)
290                         goto scan_error;
291
292                 /* drop clean flag to allow roll-forward and recovery */
293                 nilfs->ns_mount_state &= ~NILFS_VALID_FS;
294                 valid_fs = 0;
295
296                 err = nilfs_search_super_root(nilfs, &ri);
297                 if (err)
298                         goto scan_error;
299         }
300
301         err = nilfs_load_super_root(nilfs, sb, ri.ri_super_root);
302         if (unlikely(err)) {
303                 nilfs_err(sb, "error %d while loading super root", err);
304                 goto failed;
305         }
306
307         err = nilfs_sysfs_create_device_group(sb);
308         if (unlikely(err))
309                 goto sysfs_error;
310
311         if (valid_fs)
312                 goto skip_recovery;
313
314         if (s_flags & SB_RDONLY) {
315                 __u64 features;
316
317                 if (nilfs_test_opt(nilfs, NORECOVERY)) {
318                         nilfs_info(sb,
319                                    "norecovery option specified, skipping roll-forward recovery");
320                         goto skip_recovery;
321                 }
322                 features = le64_to_cpu(nilfs->ns_sbp[0]->s_feature_compat_ro) &
323                         ~NILFS_FEATURE_COMPAT_RO_SUPP;
324                 if (features) {
325                         nilfs_err(sb,
326                                   "couldn't proceed with recovery because of unsupported optional features (%llx)",
327                                   (unsigned long long)features);
328                         err = -EROFS;
329                         goto failed_unload;
330                 }
331                 if (really_read_only) {
332                         nilfs_err(sb,
333                                   "write access unavailable, cannot proceed");
334                         err = -EROFS;
335                         goto failed_unload;
336                 }
337                 sb->s_flags &= ~SB_RDONLY;
338         } else if (nilfs_test_opt(nilfs, NORECOVERY)) {
339                 nilfs_err(sb,
340                           "recovery cancelled because norecovery option was specified for a read/write mount");
341                 err = -EINVAL;
342                 goto failed_unload;
343         }
344
345         err = nilfs_salvage_orphan_logs(nilfs, sb, &ri);
346         if (err)
347                 goto failed_unload;
348
349         down_write(&nilfs->ns_sem);
350         nilfs->ns_mount_state |= NILFS_VALID_FS; /* set "clean" flag */
351         err = nilfs_cleanup_super(sb);
352         up_write(&nilfs->ns_sem);
353
354         if (err) {
355                 nilfs_err(sb,
356                           "error %d updating super block. recovery unfinished.",
357                           err);
358                 goto failed_unload;
359         }
360         nilfs_info(sb, "recovery complete");
361
362  skip_recovery:
363         nilfs_clear_recovery_info(&ri);
364         sb->s_flags = s_flags;
365         return 0;
366
367  scan_error:
368         nilfs_err(sb, "error %d while searching super root", err);
369         goto failed;
370
371  failed_unload:
372         nilfs_sysfs_delete_device_group(nilfs);
373
374  sysfs_error:
375         iput(nilfs->ns_cpfile);
376         iput(nilfs->ns_sufile);
377         iput(nilfs->ns_dat);
378
379  failed:
380         nilfs_clear_recovery_info(&ri);
381         sb->s_flags = s_flags;
382         return err;
383 }
384
385 static unsigned long long nilfs_max_size(unsigned int blkbits)
386 {
387         unsigned int max_bits;
388         unsigned long long res = MAX_LFS_FILESIZE; /* page cache limit */
389
390         max_bits = blkbits + NILFS_BMAP_KEY_BIT; /* bmap size limit */
391         if (max_bits < 64)
392                 res = min_t(unsigned long long, res, (1ULL << max_bits) - 1);
393         return res;
394 }
395
396 /**
397  * nilfs_nrsvsegs - calculate the number of reserved segments
398  * @nilfs: nilfs object
399  * @nsegs: total number of segments
400  */
401 unsigned long nilfs_nrsvsegs(struct the_nilfs *nilfs, unsigned long nsegs)
402 {
403         return max_t(unsigned long, NILFS_MIN_NRSVSEGS,
404                      DIV_ROUND_UP(nsegs * nilfs->ns_r_segments_percentage,
405                                   100));
406 }
407
408 /**
409  * nilfs_max_segment_count - calculate the maximum number of segments
410  * @nilfs: nilfs object
411  */
412 static u64 nilfs_max_segment_count(struct the_nilfs *nilfs)
413 {
414         u64 max_count = U64_MAX;
415
416         do_div(max_count, nilfs->ns_blocks_per_segment);
417         return min_t(u64, max_count, ULONG_MAX);
418 }
419
420 void nilfs_set_nsegments(struct the_nilfs *nilfs, unsigned long nsegs)
421 {
422         nilfs->ns_nsegments = nsegs;
423         nilfs->ns_nrsvsegs = nilfs_nrsvsegs(nilfs, nsegs);
424 }
425
426 static int nilfs_store_disk_layout(struct the_nilfs *nilfs,
427                                    struct nilfs_super_block *sbp)
428 {
429         u64 nsegments, nblocks;
430
431         if (le32_to_cpu(sbp->s_rev_level) < NILFS_MIN_SUPP_REV) {
432                 nilfs_err(nilfs->ns_sb,
433                           "unsupported revision (superblock rev.=%d.%d, current rev.=%d.%d). Please check the version of mkfs.nilfs(2).",
434                           le32_to_cpu(sbp->s_rev_level),
435                           le16_to_cpu(sbp->s_minor_rev_level),
436                           NILFS_CURRENT_REV, NILFS_MINOR_REV);
437                 return -EINVAL;
438         }
439         nilfs->ns_sbsize = le16_to_cpu(sbp->s_bytes);
440         if (nilfs->ns_sbsize > BLOCK_SIZE)
441                 return -EINVAL;
442
443         nilfs->ns_inode_size = le16_to_cpu(sbp->s_inode_size);
444         if (nilfs->ns_inode_size > nilfs->ns_blocksize) {
445                 nilfs_err(nilfs->ns_sb, "too large inode size: %d bytes",
446                           nilfs->ns_inode_size);
447                 return -EINVAL;
448         } else if (nilfs->ns_inode_size < NILFS_MIN_INODE_SIZE) {
449                 nilfs_err(nilfs->ns_sb, "too small inode size: %d bytes",
450                           nilfs->ns_inode_size);
451                 return -EINVAL;
452         }
453
454         nilfs->ns_first_ino = le32_to_cpu(sbp->s_first_ino);
455
456         nilfs->ns_blocks_per_segment = le32_to_cpu(sbp->s_blocks_per_segment);
457         if (nilfs->ns_blocks_per_segment < NILFS_SEG_MIN_BLOCKS) {
458                 nilfs_err(nilfs->ns_sb, "too short segment: %lu blocks",
459                           nilfs->ns_blocks_per_segment);
460                 return -EINVAL;
461         }
462
463         nilfs->ns_first_data_block = le64_to_cpu(sbp->s_first_data_block);
464         nilfs->ns_r_segments_percentage =
465                 le32_to_cpu(sbp->s_r_segments_percentage);
466         if (nilfs->ns_r_segments_percentage < 1 ||
467             nilfs->ns_r_segments_percentage > 99) {
468                 nilfs_err(nilfs->ns_sb,
469                           "invalid reserved segments percentage: %lu",
470                           nilfs->ns_r_segments_percentage);
471                 return -EINVAL;
472         }
473
474         nsegments = le64_to_cpu(sbp->s_nsegments);
475         if (nsegments > nilfs_max_segment_count(nilfs)) {
476                 nilfs_err(nilfs->ns_sb,
477                           "segment count %llu exceeds upper limit (%llu segments)",
478                           (unsigned long long)nsegments,
479                           (unsigned long long)nilfs_max_segment_count(nilfs));
480                 return -EINVAL;
481         }
482
483         nblocks = (u64)i_size_read(nilfs->ns_sb->s_bdev->bd_inode) >>
484                 nilfs->ns_sb->s_blocksize_bits;
485         if (nblocks) {
486                 u64 min_block_count = nsegments * nilfs->ns_blocks_per_segment;
487                 /*
488                  * To avoid failing to mount early device images without a
489                  * second superblock, exclude that block count from the
490                  * "min_block_count" calculation.
491                  */
492
493                 if (nblocks < min_block_count) {
494                         nilfs_err(nilfs->ns_sb,
495                                   "total number of segment blocks %llu exceeds device size (%llu blocks)",
496                                   (unsigned long long)min_block_count,
497                                   (unsigned long long)nblocks);
498                         return -EINVAL;
499                 }
500         }
501
502         nilfs_set_nsegments(nilfs, nsegments);
503         nilfs->ns_crc_seed = le32_to_cpu(sbp->s_crc_seed);
504         return 0;
505 }
506
507 static int nilfs_valid_sb(struct nilfs_super_block *sbp)
508 {
509         static unsigned char sum[4];
510         const int sumoff = offsetof(struct nilfs_super_block, s_sum);
511         size_t bytes;
512         u32 crc;
513
514         if (!sbp || le16_to_cpu(sbp->s_magic) != NILFS_SUPER_MAGIC)
515                 return 0;
516         bytes = le16_to_cpu(sbp->s_bytes);
517         if (bytes < sumoff + 4 || bytes > BLOCK_SIZE)
518                 return 0;
519         crc = crc32_le(le32_to_cpu(sbp->s_crc_seed), (unsigned char *)sbp,
520                        sumoff);
521         crc = crc32_le(crc, sum, 4);
522         crc = crc32_le(crc, (unsigned char *)sbp + sumoff + 4,
523                        bytes - sumoff - 4);
524         return crc == le32_to_cpu(sbp->s_sum);
525 }
526
527 /**
528  * nilfs_sb2_bad_offset - check the location of the second superblock
529  * @sbp: superblock raw data buffer
530  * @offset: byte offset of second superblock calculated from device size
531  *
532  * nilfs_sb2_bad_offset() checks if the position on the second
533  * superblock is valid or not based on the filesystem parameters
534  * stored in @sbp.  If @offset points to a location within the segment
535  * area, or if the parameters themselves are not normal, it is
536  * determined to be invalid.
537  *
538  * Return Value: true if invalid, false if valid.
539  */
540 static bool nilfs_sb2_bad_offset(struct nilfs_super_block *sbp, u64 offset)
541 {
542         unsigned int shift_bits = le32_to_cpu(sbp->s_log_block_size);
543         u32 blocks_per_segment = le32_to_cpu(sbp->s_blocks_per_segment);
544         u64 nsegments = le64_to_cpu(sbp->s_nsegments);
545         u64 index;
546
547         if (blocks_per_segment < NILFS_SEG_MIN_BLOCKS ||
548             shift_bits > ilog2(NILFS_MAX_BLOCK_SIZE) - BLOCK_SIZE_BITS)
549                 return true;
550
551         index = offset >> (shift_bits + BLOCK_SIZE_BITS);
552         do_div(index, blocks_per_segment);
553         return index < nsegments;
554 }
555
556 static void nilfs_release_super_block(struct the_nilfs *nilfs)
557 {
558         int i;
559
560         for (i = 0; i < 2; i++) {
561                 if (nilfs->ns_sbp[i]) {
562                         brelse(nilfs->ns_sbh[i]);
563                         nilfs->ns_sbh[i] = NULL;
564                         nilfs->ns_sbp[i] = NULL;
565                 }
566         }
567 }
568
569 void nilfs_fall_back_super_block(struct the_nilfs *nilfs)
570 {
571         brelse(nilfs->ns_sbh[0]);
572         nilfs->ns_sbh[0] = nilfs->ns_sbh[1];
573         nilfs->ns_sbp[0] = nilfs->ns_sbp[1];
574         nilfs->ns_sbh[1] = NULL;
575         nilfs->ns_sbp[1] = NULL;
576 }
577
578 void nilfs_swap_super_block(struct the_nilfs *nilfs)
579 {
580         struct buffer_head *tsbh = nilfs->ns_sbh[0];
581         struct nilfs_super_block *tsbp = nilfs->ns_sbp[0];
582
583         nilfs->ns_sbh[0] = nilfs->ns_sbh[1];
584         nilfs->ns_sbp[0] = nilfs->ns_sbp[1];
585         nilfs->ns_sbh[1] = tsbh;
586         nilfs->ns_sbp[1] = tsbp;
587 }
588
589 static int nilfs_load_super_block(struct the_nilfs *nilfs,
590                                   struct super_block *sb, int blocksize,
591                                   struct nilfs_super_block **sbpp)
592 {
593         struct nilfs_super_block **sbp = nilfs->ns_sbp;
594         struct buffer_head **sbh = nilfs->ns_sbh;
595         u64 sb2off, devsize = nilfs->ns_bdev->bd_inode->i_size;
596         int valid[2], swp = 0;
597
598         if (devsize < NILFS_SEG_MIN_BLOCKS * NILFS_MIN_BLOCK_SIZE + 4096) {
599                 nilfs_err(sb, "device size too small");
600                 return -EINVAL;
601         }
602         sb2off = NILFS_SB2_OFFSET_BYTES(devsize);
603
604         sbp[0] = nilfs_read_super_block(sb, NILFS_SB_OFFSET_BYTES, blocksize,
605                                         &sbh[0]);
606         sbp[1] = nilfs_read_super_block(sb, sb2off, blocksize, &sbh[1]);
607
608         if (!sbp[0]) {
609                 if (!sbp[1]) {
610                         nilfs_err(sb, "unable to read superblock");
611                         return -EIO;
612                 }
613                 nilfs_warn(sb,
614                            "unable to read primary superblock (blocksize = %d)",
615                            blocksize);
616         } else if (!sbp[1]) {
617                 nilfs_warn(sb,
618                            "unable to read secondary superblock (blocksize = %d)",
619                            blocksize);
620         }
621
622         /*
623          * Compare two super blocks and set 1 in swp if the secondary
624          * super block is valid and newer.  Otherwise, set 0 in swp.
625          */
626         valid[0] = nilfs_valid_sb(sbp[0]);
627         valid[1] = nilfs_valid_sb(sbp[1]);
628         swp = valid[1] && (!valid[0] ||
629                            le64_to_cpu(sbp[1]->s_last_cno) >
630                            le64_to_cpu(sbp[0]->s_last_cno));
631
632         if (valid[swp] && nilfs_sb2_bad_offset(sbp[swp], sb2off)) {
633                 brelse(sbh[1]);
634                 sbh[1] = NULL;
635                 sbp[1] = NULL;
636                 valid[1] = 0;
637                 swp = 0;
638         }
639         if (!valid[swp]) {
640                 nilfs_release_super_block(nilfs);
641                 nilfs_err(sb, "couldn't find nilfs on the device");
642                 return -EINVAL;
643         }
644
645         if (!valid[!swp])
646                 nilfs_warn(sb,
647                            "broken superblock, retrying with spare superblock (blocksize = %d)",
648                            blocksize);
649         if (swp)
650                 nilfs_swap_super_block(nilfs);
651
652         nilfs->ns_sbwcount = 0;
653         nilfs->ns_sbwtime = le64_to_cpu(sbp[0]->s_wtime);
654         nilfs->ns_prot_seq = le64_to_cpu(sbp[valid[1] & !swp]->s_last_seq);
655         *sbpp = sbp[0];
656         return 0;
657 }
658
659 /**
660  * init_nilfs - initialize a NILFS instance.
661  * @nilfs: the_nilfs structure
662  * @sb: super block
663  * @data: mount options
664  *
665  * init_nilfs() performs common initialization per block device (e.g.
666  * reading the super block, getting disk layout information, initializing
667  * shared fields in the_nilfs).
668  *
669  * Return Value: On success, 0 is returned. On error, a negative error
670  * code is returned.
671  */
672 int init_nilfs(struct the_nilfs *nilfs, struct super_block *sb, char *data)
673 {
674         struct nilfs_super_block *sbp;
675         int blocksize;
676         int err;
677
678         down_write(&nilfs->ns_sem);
679
680         blocksize = sb_min_blocksize(sb, NILFS_MIN_BLOCK_SIZE);
681         if (!blocksize) {
682                 nilfs_err(sb, "unable to set blocksize");
683                 err = -EINVAL;
684                 goto out;
685         }
686         err = nilfs_load_super_block(nilfs, sb, blocksize, &sbp);
687         if (err)
688                 goto out;
689
690         err = nilfs_store_magic_and_option(sb, sbp, data);
691         if (err)
692                 goto failed_sbh;
693
694         err = nilfs_check_feature_compatibility(sb, sbp);
695         if (err)
696                 goto failed_sbh;
697
698         err = nilfs_get_blocksize(sb, sbp, &blocksize);
699         if (err)
700                 goto failed_sbh;
701
702         if (blocksize < NILFS_MIN_BLOCK_SIZE) {
703                 nilfs_err(sb,
704                           "couldn't mount because of unsupported filesystem blocksize %d",
705                           blocksize);
706                 err = -EINVAL;
707                 goto failed_sbh;
708         }
709         if (sb->s_blocksize != blocksize) {
710                 int hw_blocksize = bdev_logical_block_size(sb->s_bdev);
711
712                 if (blocksize < hw_blocksize) {
713                         nilfs_err(sb,
714                                   "blocksize %d too small for device (sector-size = %d)",
715                                   blocksize, hw_blocksize);
716                         err = -EINVAL;
717                         goto failed_sbh;
718                 }
719                 nilfs_release_super_block(nilfs);
720                 if (!sb_set_blocksize(sb, blocksize)) {
721                         nilfs_err(sb, "bad blocksize %d", blocksize);
722                         err = -EINVAL;
723                         goto out;
724                 }
725
726                 err = nilfs_load_super_block(nilfs, sb, blocksize, &sbp);
727                 if (err)
728                         goto out;
729                         /*
730                          * Not to failed_sbh; sbh is released automatically
731                          * when reloading fails.
732                          */
733         }
734         nilfs->ns_blocksize_bits = sb->s_blocksize_bits;
735         nilfs->ns_blocksize = blocksize;
736
737         get_random_bytes(&nilfs->ns_next_generation,
738                          sizeof(nilfs->ns_next_generation));
739
740         err = nilfs_store_disk_layout(nilfs, sbp);
741         if (err)
742                 goto failed_sbh;
743
744         sb->s_maxbytes = nilfs_max_size(sb->s_blocksize_bits);
745
746         nilfs->ns_mount_state = le16_to_cpu(sbp->s_state);
747
748         err = nilfs_store_log_cursor(nilfs, sbp);
749         if (err)
750                 goto failed_sbh;
751
752         set_nilfs_init(nilfs);
753         err = 0;
754  out:
755         up_write(&nilfs->ns_sem);
756         return err;
757
758  failed_sbh:
759         nilfs_release_super_block(nilfs);
760         goto out;
761 }
762
763 int nilfs_discard_segments(struct the_nilfs *nilfs, __u64 *segnump,
764                             size_t nsegs)
765 {
766         sector_t seg_start, seg_end;
767         sector_t start = 0, nblocks = 0;
768         unsigned int sects_per_block;
769         __u64 *sn;
770         int ret = 0;
771
772         sects_per_block = (1 << nilfs->ns_blocksize_bits) /
773                 bdev_logical_block_size(nilfs->ns_bdev);
774         for (sn = segnump; sn < segnump + nsegs; sn++) {
775                 nilfs_get_segment_range(nilfs, *sn, &seg_start, &seg_end);
776
777                 if (!nblocks) {
778                         start = seg_start;
779                         nblocks = seg_end - seg_start + 1;
780                 } else if (start + nblocks == seg_start) {
781                         nblocks += seg_end - seg_start + 1;
782                 } else {
783                         ret = blkdev_issue_discard(nilfs->ns_bdev,
784                                                    start * sects_per_block,
785                                                    nblocks * sects_per_block,
786                                                    GFP_NOFS, 0);
787                         if (ret < 0)
788                                 return ret;
789                         nblocks = 0;
790                 }
791         }
792         if (nblocks)
793                 ret = blkdev_issue_discard(nilfs->ns_bdev,
794                                            start * sects_per_block,
795                                            nblocks * sects_per_block,
796                                            GFP_NOFS, 0);
797         return ret;
798 }
799
800 int nilfs_count_free_blocks(struct the_nilfs *nilfs, sector_t *nblocks)
801 {
802         unsigned long ncleansegs;
803
804         ncleansegs = nilfs_sufile_get_ncleansegs(nilfs->ns_sufile);
805         *nblocks = (sector_t)ncleansegs * nilfs->ns_blocks_per_segment;
806         return 0;
807 }
808
809 int nilfs_near_disk_full(struct the_nilfs *nilfs)
810 {
811         unsigned long ncleansegs, nincsegs;
812
813         ncleansegs = nilfs_sufile_get_ncleansegs(nilfs->ns_sufile);
814         nincsegs = atomic_read(&nilfs->ns_ndirtyblks) /
815                 nilfs->ns_blocks_per_segment + 1;
816
817         return ncleansegs <= nilfs->ns_nrsvsegs + nincsegs;
818 }
819
820 struct nilfs_root *nilfs_lookup_root(struct the_nilfs *nilfs, __u64 cno)
821 {
822         struct rb_node *n;
823         struct nilfs_root *root;
824
825         spin_lock(&nilfs->ns_cptree_lock);
826         n = nilfs->ns_cptree.rb_node;
827         while (n) {
828                 root = rb_entry(n, struct nilfs_root, rb_node);
829
830                 if (cno < root->cno) {
831                         n = n->rb_left;
832                 } else if (cno > root->cno) {
833                         n = n->rb_right;
834                 } else {
835                         refcount_inc(&root->count);
836                         spin_unlock(&nilfs->ns_cptree_lock);
837                         return root;
838                 }
839         }
840         spin_unlock(&nilfs->ns_cptree_lock);
841
842         return NULL;
843 }
844
845 struct nilfs_root *
846 nilfs_find_or_create_root(struct the_nilfs *nilfs, __u64 cno)
847 {
848         struct rb_node **p, *parent;
849         struct nilfs_root *root, *new;
850         int err;
851
852         root = nilfs_lookup_root(nilfs, cno);
853         if (root)
854                 return root;
855
856         new = kzalloc(sizeof(*root), GFP_KERNEL);
857         if (!new)
858                 return NULL;
859
860         spin_lock(&nilfs->ns_cptree_lock);
861
862         p = &nilfs->ns_cptree.rb_node;
863         parent = NULL;
864
865         while (*p) {
866                 parent = *p;
867                 root = rb_entry(parent, struct nilfs_root, rb_node);
868
869                 if (cno < root->cno) {
870                         p = &(*p)->rb_left;
871                 } else if (cno > root->cno) {
872                         p = &(*p)->rb_right;
873                 } else {
874                         refcount_inc(&root->count);
875                         spin_unlock(&nilfs->ns_cptree_lock);
876                         kfree(new);
877                         return root;
878                 }
879         }
880
881         new->cno = cno;
882         new->ifile = NULL;
883         new->nilfs = nilfs;
884         refcount_set(&new->count, 1);
885         atomic64_set(&new->inodes_count, 0);
886         atomic64_set(&new->blocks_count, 0);
887
888         rb_link_node(&new->rb_node, parent, p);
889         rb_insert_color(&new->rb_node, &nilfs->ns_cptree);
890
891         spin_unlock(&nilfs->ns_cptree_lock);
892
893         err = nilfs_sysfs_create_snapshot_group(new);
894         if (err) {
895                 kfree(new);
896                 new = NULL;
897         }
898
899         return new;
900 }
901
902 void nilfs_put_root(struct nilfs_root *root)
903 {
904         struct the_nilfs *nilfs = root->nilfs;
905
906         if (refcount_dec_and_lock(&root->count, &nilfs->ns_cptree_lock)) {
907                 rb_erase(&root->rb_node, &nilfs->ns_cptree);
908                 spin_unlock(&nilfs->ns_cptree_lock);
909
910                 nilfs_sysfs_delete_snapshot_group(root);
911                 iput(root->ifile);
912
913                 kfree(root);
914         }
915 }