projects / deliverable / linux.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
btrfs: make close_ctree return void
[deliverable/linux.git] / fs / btrfs / scrub.c
CommitLineData
a2de733c 1/*
b6bfebc1 2 * Copyright (C) 2011, 2012 STRATO. All rights reserved.
a2de733c
AJ		 3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
17 */
18
a2de733c 19#include <linux/blkdev.h>
558540c1 20#include <linux/ratelimit.h>
a2de733c
AJ		 21#include "ctree.h"
22#include "volumes.h"
23#include "disk-io.h"
24#include "ordered-data.h"
0ef8e451 25#include "transaction.h"
558540c1 26#include "backref.h"
5da6fcbc 27#include "extent_io.h"
ff023aac 28#include "dev-replace.h"
21adbd5c 29#include "check-integrity.h"
606686ee 30#include "rcu-string.h"
53b381b3 31#include "raid56.h"
a2de733c
AJ		 32
33/*
34 * This is only the first step towards a full-features scrub. It reads all
35 * extent and super block and verifies the checksums. In case a bad checksum
36 * is found or the extent cannot be read, good data will be written back if
37 * any can be found.
38 *
39 * Future enhancements:
a2de733c
AJ		 40 * - In case an unrepairable extent is encountered, track which files are
41 * affected and report them
a2de733c 42 * - track and record media errors, throw out bad devices
a2de733c 43 * - add a mode to also read unallocated space
a2de733c
AJ		 44 */
45
b5d67f64 46struct scrub_block;
d9d181c1 47struct scrub_ctx;
a2de733c 48
ff023aac
SB		 49/*
50 * the following three values only influence the performance.
51 * The last one configures the number of parallel and outstanding I/O
52 * operations. The first two values configure an upper limit for the number
53 * of (dynamically allocated) pages that are added to a bio.
54 */
55#define SCRUB_PAGES_PER_RD_BIO 32 /* 128k per bio */
56#define SCRUB_PAGES_PER_WR_BIO 32 /* 128k per bio */
57#define SCRUB_BIOS_PER_SCTX 64 /* 8MB per device in flight */
7a9e9987
SB		 58
59/*
60 * the following value times PAGE_SIZE needs to be large enough to match the
61 * largest node/leaf/sector size that shall be supported.
62 * Values larger than BTRFS_STRIPE_LEN are not supported.
63 */
b5d67f64 64#define SCRUB_MAX_PAGES_PER_BLOCK 16 /* 64k per node/leaf/sector */
a2de733c
AJ		 65
66struct scrub_page {
b5d67f64
SB		 67 struct scrub_block *sblock;
68 struct page *page;
442a4f63 69 struct btrfs_device *dev;
a2de733c
AJ		 70 u64 flags; /* extent flags */
71 u64 generation;
b5d67f64
SB		 72 u64 logical;
73 u64 physical;
ff023aac 74 u64 physical_for_dev_replace;
7a9e9987 75 atomic_t ref_count;
b5d67f64
SB		 76 struct {
77 unsigned int mirror_num:8;
78 unsigned int have_csum:1;
79 unsigned int io_error:1;
80 };
a2de733c
AJ		 81 u8 csum[BTRFS_CSUM_SIZE];
82};
83
84struct scrub_bio {
85 int index;
d9d181c1 86 struct scrub_ctx *sctx;
a36cf8b8 87 struct btrfs_device *dev;
a2de733c
AJ		 88 struct bio *bio;
89 int err;
90 u64 logical;
91 u64 physical;
ff023aac
SB		 92#if SCRUB_PAGES_PER_WR_BIO >= SCRUB_PAGES_PER_RD_BIO
93 struct scrub_page *pagev[SCRUB_PAGES_PER_WR_BIO];
94#else
95 struct scrub_page *pagev[SCRUB_PAGES_PER_RD_BIO];
96#endif
b5d67f64 97 int page_count;
a2de733c
AJ		 98 int next_free;
99 struct btrfs_work work;
100};
101
b5d67f64 102struct scrub_block {
7a9e9987 103 struct scrub_page *pagev[SCRUB_MAX_PAGES_PER_BLOCK];
b5d67f64
SB		 104 int page_count;
105 atomic_t outstanding_pages;
106 atomic_t ref_count; /* free mem on transition to zero */
d9d181c1 107 struct scrub_ctx *sctx;
b5d67f64
SB		 108 struct {
109 unsigned int header_error:1;
110 unsigned int checksum_error:1;
111 unsigned int no_io_error_seen:1;
442a4f63 112 unsigned int generation_error:1; /* also sets header_error */
b5d67f64
SB		 113 };
114};
115
ff023aac
SB		 116struct scrub_wr_ctx {
117 struct scrub_bio *wr_curr_bio;
118 struct btrfs_device *tgtdev;
119 int pages_per_wr_bio; /* <= SCRUB_PAGES_PER_WR_BIO */
120 atomic_t flush_all_writes;
121 struct mutex wr_lock;
122};
123
d9d181c1 124struct scrub_ctx {
ff023aac 125 struct scrub_bio *bios[SCRUB_BIOS_PER_SCTX];
a36cf8b8 126 struct btrfs_root *dev_root;
a2de733c
AJ		 127 int first_free;
128 int curr;
b6bfebc1
SB		 129 atomic_t bios_in_flight;
130 atomic_t workers_pending;
a2de733c
AJ		 131 spinlock_t list_lock;
132 wait_queue_head_t list_wait;
133 u16 csum_size;
134 struct list_head csum_list;
135 atomic_t cancel_req;
8628764e 136 int readonly;
ff023aac 137 int pages_per_rd_bio;
b5d67f64
SB		 138 u32 sectorsize;
139 u32 nodesize;
140 u32 leafsize;
63a212ab
SB		 141
142 int is_dev_replace;
ff023aac 143 struct scrub_wr_ctx wr_ctx;
63a212ab 144
a2de733c
AJ		 145 /*
146 * statistics
147 */
148 struct btrfs_scrub_progress stat;
149 spinlock_t stat_lock;
150};
151
0ef8e451 152struct scrub_fixup_nodatasum {
d9d181c1 153 struct scrub_ctx *sctx;
a36cf8b8 154 struct btrfs_device *dev;
0ef8e451
JS		 155 u64 logical;
156 struct btrfs_root *root;
157 struct btrfs_work work;
158 int mirror_num;
159};
160
652f25a2
JB		 161struct scrub_nocow_inode {
162 u64 inum;
163 u64 offset;
164 u64 root;
165 struct list_head list;
166};
167
ff023aac
SB		 168struct scrub_copy_nocow_ctx {
169 struct scrub_ctx *sctx;
170 u64 logical;
171 u64 len;
172 int mirror_num;
173 u64 physical_for_dev_replace;
652f25a2 174 struct list_head inodes;
ff023aac
SB		 175 struct btrfs_work work;
176};
177
558540c1
JS		 178struct scrub_warning {
179 struct btrfs_path *path;
180 u64 extent_item_size;
181 char *scratch_buf;
182 char *msg_buf;
183 const char *errstr;
184 sector_t sector;
185 u64 logical;
186 struct btrfs_device *dev;
187 int msg_bufsize;
188 int scratch_bufsize;
189};
190
b5d67f64 191
b6bfebc1
SB		 192static void scrub_pending_bio_inc(struct scrub_ctx *sctx);
193static void scrub_pending_bio_dec(struct scrub_ctx *sctx);
194static void scrub_pending_trans_workers_inc(struct scrub_ctx *sctx);
195static void scrub_pending_trans_workers_dec(struct scrub_ctx *sctx);
b5d67f64 196static int scrub_handle_errored_block(struct scrub_block *sblock_to_check);
d9d181c1 197static int scrub_setup_recheck_block(struct scrub_ctx *sctx,
3ec706c8 198 struct btrfs_fs_info *fs_info,
ff023aac 199 struct scrub_block *original_sblock,
b5d67f64 200 u64 length, u64 logical,
ff023aac 201 struct scrub_block *sblocks_for_recheck);
34f5c8e9
SB		 202static void scrub_recheck_block(struct btrfs_fs_info *fs_info,
203 struct scrub_block *sblock, int is_metadata,
204 int have_csum, u8 *csum, u64 generation,
205 u16 csum_size);
b5d67f64
SB		 206static void scrub_recheck_block_checksum(struct btrfs_fs_info *fs_info,
207 struct scrub_block *sblock,
208 int is_metadata, int have_csum,
209 const u8 *csum, u64 generation,
210 u16 csum_size);
b5d67f64
SB		 211static int scrub_repair_block_from_good_copy(struct scrub_block *sblock_bad,
212 struct scrub_block *sblock_good,
213 int force_write);
214static int scrub_repair_page_from_good_copy(struct scrub_block *sblock_bad,
215 struct scrub_block *sblock_good,
216 int page_num, int force_write);
ff023aac
SB		 217static void scrub_write_block_to_dev_replace(struct scrub_block *sblock);
218static int scrub_write_page_to_dev_replace(struct scrub_block *sblock,
219 int page_num);
b5d67f64
SB		 220static int scrub_checksum_data(struct scrub_block *sblock);
221static int scrub_checksum_tree_block(struct scrub_block *sblock);
222static int scrub_checksum_super(struct scrub_block *sblock);
223static void scrub_block_get(struct scrub_block *sblock);
224static void scrub_block_put(struct scrub_block *sblock);
7a9e9987
SB		 225static void scrub_page_get(struct scrub_page *spage);
226static void scrub_page_put(struct scrub_page *spage);
ff023aac
SB		 227static int scrub_add_page_to_rd_bio(struct scrub_ctx *sctx,
228 struct scrub_page *spage);
d9d181c1 229static int scrub_pages(struct scrub_ctx *sctx, u64 logical, u64 len,
a36cf8b8 230 u64 physical, struct btrfs_device *dev, u64 flags,
ff023aac
SB		 231 u64 gen, int mirror_num, u8 *csum, int force,
232 u64 physical_for_dev_replace);
1623edeb 233static void scrub_bio_end_io(struct bio *bio, int err);
b5d67f64
SB		 234static void scrub_bio_end_io_worker(struct btrfs_work *work);
235static void scrub_block_complete(struct scrub_block *sblock);
ff023aac
SB		 236static void scrub_remap_extent(struct btrfs_fs_info *fs_info,
237 u64 extent_logical, u64 extent_len,
238 u64 *extent_physical,
239 struct btrfs_device **extent_dev,
240 int *extent_mirror_num);
241static int scrub_setup_wr_ctx(struct scrub_ctx *sctx,
242 struct scrub_wr_ctx *wr_ctx,
243 struct btrfs_fs_info *fs_info,
244 struct btrfs_device *dev,
245 int is_dev_replace);
246static void scrub_free_wr_ctx(struct scrub_wr_ctx *wr_ctx);
247static int scrub_add_page_to_wr_bio(struct scrub_ctx *sctx,
248 struct scrub_page *spage);
249static void scrub_wr_submit(struct scrub_ctx *sctx);
250static void scrub_wr_bio_end_io(struct bio *bio, int err);
251static void scrub_wr_bio_end_io_worker(struct btrfs_work *work);
252static int write_page_nocow(struct scrub_ctx *sctx,
253 u64 physical_for_dev_replace, struct page *page);
254static int copy_nocow_pages_for_inode(u64 inum, u64 offset, u64 root,
652f25a2 255 struct scrub_copy_nocow_ctx *ctx);
ff023aac
SB		 256static int copy_nocow_pages(struct scrub_ctx *sctx, u64 logical, u64 len,
257 int mirror_num, u64 physical_for_dev_replace);
258static void copy_nocow_pages_worker(struct btrfs_work *work);
cb7ab021 259static void __scrub_blocked_if_needed(struct btrfs_fs_info *fs_info);
3cb0929a 260static void scrub_blocked_if_needed(struct btrfs_fs_info *fs_info);
1623edeb
SB		 261
262
b6bfebc1
SB		 263static void scrub_pending_bio_inc(struct scrub_ctx *sctx)
264{
265 atomic_inc(&sctx->bios_in_flight);
266}
267
268static void scrub_pending_bio_dec(struct scrub_ctx *sctx)
269{
270 atomic_dec(&sctx->bios_in_flight);
271 wake_up(&sctx->list_wait);
272}
273
cb7ab021 274static void __scrub_blocked_if_needed(struct btrfs_fs_info *fs_info)
3cb0929a
WS		 275{
276 while (atomic_read(&fs_info->scrub_pause_req)) {
277 mutex_unlock(&fs_info->scrub_lock);
278 wait_event(fs_info->scrub_pause_wait,
279 atomic_read(&fs_info->scrub_pause_req) == 0);
280 mutex_lock(&fs_info->scrub_lock);
281 }
282}
283
cb7ab021
WS		 284static void scrub_blocked_if_needed(struct btrfs_fs_info *fs_info)
285{
286 atomic_inc(&fs_info->scrubs_paused);
287 wake_up(&fs_info->scrub_pause_wait);
288
289 mutex_lock(&fs_info->scrub_lock);
290 __scrub_blocked_if_needed(fs_info);
291 atomic_dec(&fs_info->scrubs_paused);
292 mutex_unlock(&fs_info->scrub_lock);
293
294 wake_up(&fs_info->scrub_pause_wait);
295}
296
b6bfebc1
SB		 297/*
298 * used for workers that require transaction commits (i.e., for the
299 * NOCOW case)
300 */
301static void scrub_pending_trans_workers_inc(struct scrub_ctx *sctx)
302{
303 struct btrfs_fs_info *fs_info = sctx->dev_root->fs_info;
304
305 /*
306 * increment scrubs_running to prevent cancel requests from
307 * completing as long as a worker is running. we must also
308 * increment scrubs_paused to prevent deadlocking on pause
309 * requests used for transactions commits (as the worker uses a
310 * transaction context). it is safe to regard the worker
311 * as paused for all matters practical. effectively, we only
312 * avoid cancellation requests from completing.
313 */
314 mutex_lock(&fs_info->scrub_lock);
315 atomic_inc(&fs_info->scrubs_running);
316 atomic_inc(&fs_info->scrubs_paused);
317 mutex_unlock(&fs_info->scrub_lock);
32a44789
WS		 318
319 /*
320 * check if @scrubs_running=@scrubs_paused condition
321 * inside wait_event() is not an atomic operation.
322 * which means we may inc/dec @scrub_running/paused
323 * at any time. Let's wake up @scrub_pause_wait as
324 * much as we can to let commit transaction blocked less.
325 */
326 wake_up(&fs_info->scrub_pause_wait);
327
b6bfebc1
SB		 328 atomic_inc(&sctx->workers_pending);
329}
330
331/* used for workers that require transaction commits */
332static void scrub_pending_trans_workers_dec(struct scrub_ctx *sctx)
333{
334 struct btrfs_fs_info *fs_info = sctx->dev_root->fs_info;
335
336 /*
337 * see scrub_pending_trans_workers_inc() why we're pretending
338 * to be paused in the scrub counters
339 */
340 mutex_lock(&fs_info->scrub_lock);
341 atomic_dec(&fs_info->scrubs_running);
342 atomic_dec(&fs_info->scrubs_paused);
343 mutex_unlock(&fs_info->scrub_lock);
344 atomic_dec(&sctx->workers_pending);
345 wake_up(&fs_info->scrub_pause_wait);
346 wake_up(&sctx->list_wait);
347}
348
d9d181c1 349static void scrub_free_csums(struct scrub_ctx *sctx)
a2de733c 350{
d9d181c1 351 while (!list_empty(&sctx->csum_list)) {
a2de733c 352 struct btrfs_ordered_sum *sum;
d9d181c1 353 sum = list_first_entry(&sctx->csum_list,
a2de733c
AJ		 354 struct btrfs_ordered_sum, list);
355 list_del(&sum->list);
356 kfree(sum);
357 }
358}
359
d9d181c1 360static noinline_for_stack void scrub_free_ctx(struct scrub_ctx *sctx)
a2de733c
AJ		 361{
362 int i;
a2de733c 363
d9d181c1 364 if (!sctx)
a2de733c
AJ		 365 return;
366
ff023aac
SB		 367 scrub_free_wr_ctx(&sctx->wr_ctx);
368
b5d67f64 369 /* this can happen when scrub is cancelled */
d9d181c1
SB		 370 if (sctx->curr != -1) {
371 struct scrub_bio *sbio = sctx->bios[sctx->curr];
b5d67f64
SB		 372
373 for (i = 0; i < sbio->page_count; i++) {
ff023aac 374 WARN_ON(!sbio->pagev[i]->page);
b5d67f64
SB		 375 scrub_block_put(sbio->pagev[i]->sblock);
376 }
377 bio_put(sbio->bio);
378 }
379
ff023aac 380 for (i = 0; i < SCRUB_BIOS_PER_SCTX; ++i) {
d9d181c1 381 struct scrub_bio *sbio = sctx->bios[i];
a2de733c
AJ		 382
383 if (!sbio)
384 break;
a2de733c
AJ		 385 kfree(sbio);
386 }
387
d9d181c1
SB		 388 scrub_free_csums(sctx);
389 kfree(sctx);
a2de733c
AJ		 390}
391
392static noinline_for_stack
63a212ab 393struct scrub_ctx *scrub_setup_ctx(struct btrfs_device *dev, int is_dev_replace)
a2de733c 394{
d9d181c1 395 struct scrub_ctx *sctx;
a2de733c 396 int i;
a2de733c 397 struct btrfs_fs_info *fs_info = dev->dev_root->fs_info;
ff023aac
SB		 398 int pages_per_rd_bio;
399 int ret;
a2de733c 400
ff023aac
SB		 401 /*
402 * the setting of pages_per_rd_bio is correct for scrub but might
403 * be wrong for the dev_replace code where we might read from
404 * different devices in the initial huge bios. However, that
405 * code is able to correctly handle the case when adding a page
406 * to a bio fails.
407 */
408 if (dev->bdev)
409 pages_per_rd_bio = min_t(int, SCRUB_PAGES_PER_RD_BIO,
410 bio_get_nr_vecs(dev->bdev));
411 else
412 pages_per_rd_bio = SCRUB_PAGES_PER_RD_BIO;
d9d181c1
SB		 413 sctx = kzalloc(sizeof(*sctx), GFP_NOFS);
414 if (!sctx)
a2de733c 415 goto nomem;
63a212ab 416 sctx->is_dev_replace = is_dev_replace;
ff023aac 417 sctx->pages_per_rd_bio = pages_per_rd_bio;
d9d181c1 418 sctx->curr = -1;
a36cf8b8 419 sctx->dev_root = dev->dev_root;
ff023aac 420 for (i = 0; i < SCRUB_BIOS_PER_SCTX; ++i) {
a2de733c
AJ		 421 struct scrub_bio *sbio;
422
423 sbio = kzalloc(sizeof(*sbio), GFP_NOFS);
424 if (!sbio)
425 goto nomem;
d9d181c1 426 sctx->bios[i] = sbio;
a2de733c 427
a2de733c 428 sbio->index = i;
d9d181c1 429 sbio->sctx = sctx;
b5d67f64 430 sbio->page_count = 0;
9e0af237
LB		 431 btrfs_init_work(&sbio->work, btrfs_scrub_helper,
432 scrub_bio_end_io_worker, NULL, NULL);
a2de733c 433
ff023aac 434 if (i != SCRUB_BIOS_PER_SCTX - 1)
d9d181c1 435 sctx->bios[i]->next_free = i + 1;
0ef8e451 436 else
d9d181c1
SB		 437 sctx->bios[i]->next_free = -1;
438 }
439 sctx->first_free = 0;
440 sctx->nodesize = dev->dev_root->nodesize;
441 sctx->leafsize = dev->dev_root->leafsize;
442 sctx->sectorsize = dev->dev_root->sectorsize;
b6bfebc1
SB		 443 atomic_set(&sctx->bios_in_flight, 0);
444 atomic_set(&sctx->workers_pending, 0);
d9d181c1
SB		 445 atomic_set(&sctx->cancel_req, 0);
446 sctx->csum_size = btrfs_super_csum_size(fs_info->super_copy);
447 INIT_LIST_HEAD(&sctx->csum_list);
448
449 spin_lock_init(&sctx->list_lock);
450 spin_lock_init(&sctx->stat_lock);
451 init_waitqueue_head(&sctx->list_wait);
ff023aac
SB		 452
453 ret = scrub_setup_wr_ctx(sctx, &sctx->wr_ctx, fs_info,
454 fs_info->dev_replace.tgtdev, is_dev_replace);
455 if (ret) {
456 scrub_free_ctx(sctx);
457 return ERR_PTR(ret);
458 }
d9d181c1 459 return sctx;
a2de733c
AJ		 460
461nomem:
d9d181c1 462 scrub_free_ctx(sctx);
a2de733c
AJ		 463 return ERR_PTR(-ENOMEM);
464}
465
ff023aac
SB		 466static int scrub_print_warning_inode(u64 inum, u64 offset, u64 root,
467 void *warn_ctx)
558540c1
JS		 468{
469 u64 isize;
470 u32 nlink;
471 int ret;
472 int i;
473 struct extent_buffer *eb;
474 struct btrfs_inode_item *inode_item;
ff023aac 475 struct scrub_warning *swarn = warn_ctx;
558540c1
JS		 476 struct btrfs_fs_info *fs_info = swarn->dev->dev_root->fs_info;
477 struct inode_fs_paths *ipath = NULL;
478 struct btrfs_root *local_root;
479 struct btrfs_key root_key;
480
481 root_key.objectid = root;
482 root_key.type = BTRFS_ROOT_ITEM_KEY;
483 root_key.offset = (u64)-1;
484 local_root = btrfs_read_fs_root_no_name(fs_info, &root_key);
485 if (IS_ERR(local_root)) {
486 ret = PTR_ERR(local_root);
487 goto err;
488 }
489
490 ret = inode_item_info(inum, 0, local_root, swarn->path);
491 if (ret) {
492 btrfs_release_path(swarn->path);
493 goto err;
494 }
495
496 eb = swarn->path->nodes[0];
497 inode_item = btrfs_item_ptr(eb, swarn->path->slots[0],
498 struct btrfs_inode_item);
499 isize = btrfs_inode_size(eb, inode_item);
500 nlink = btrfs_inode_nlink(eb, inode_item);
501 btrfs_release_path(swarn->path);
502
503 ipath = init_ipath(4096, local_root, swarn->path);
26bdef54
DC		 504 if (IS_ERR(ipath)) {
505 ret = PTR_ERR(ipath);
506 ipath = NULL;
507 goto err;
508 }
558540c1
JS		 509 ret = paths_from_inode(inum, ipath);
510
511 if (ret < 0)
512 goto err;
513
514 /*
515 * we deliberately ignore the bit ipath might have been too small to
516 * hold all of the paths here
517 */
518 for (i = 0; i < ipath->fspath->elem_cnt; ++i)
efe120a0 519 printk_in_rcu(KERN_WARNING "BTRFS: %s at logical %llu on dev "
558540c1
JS		 520 "%s, sector %llu, root %llu, inode %llu, offset %llu, "
521 "length %llu, links %u (path: %s)\n", swarn->errstr,
606686ee 522 swarn->logical, rcu_str_deref(swarn->dev->name),
558540c1
JS		 523 (unsigned long long)swarn->sector, root, inum, offset,
524 min(isize - offset, (u64)PAGE_SIZE), nlink,
745c4d8e 525 (char *)(unsigned long)ipath->fspath->val[i]);
558540c1
JS		 526
527 free_ipath(ipath);
528 return 0;
529
530err:
efe120a0 531 printk_in_rcu(KERN_WARNING "BTRFS: %s at logical %llu on dev "
558540c1
JS		 532 "%s, sector %llu, root %llu, inode %llu, offset %llu: path "
533 "resolving failed with ret=%d\n", swarn->errstr,
606686ee 534 swarn->logical, rcu_str_deref(swarn->dev->name),
558540c1
JS		 535 (unsigned long long)swarn->sector, root, inum, offset, ret);
536
537 free_ipath(ipath);
538 return 0;
539}
540
b5d67f64 541static void scrub_print_warning(const char *errstr, struct scrub_block *sblock)
558540c1 542{
a36cf8b8
SB		 543 struct btrfs_device *dev;
544 struct btrfs_fs_info *fs_info;
558540c1
JS		 545 struct btrfs_path *path;
546 struct btrfs_key found_key;
547 struct extent_buffer *eb;
548 struct btrfs_extent_item *ei;
549 struct scrub_warning swarn;
69917e43
LB		 550 unsigned long ptr = 0;
551 u64 extent_item_pos;
552 u64 flags = 0;
558540c1 553 u64 ref_root;
69917e43 554 u32 item_size;
558540c1 555 u8 ref_level;
558540c1 556 const int bufsize = 4096;
69917e43 557 int ret;
558540c1 558
a36cf8b8 559 WARN_ON(sblock->page_count < 1);
7a9e9987 560 dev = sblock->pagev[0]->dev;
a36cf8b8
SB		 561 fs_info = sblock->sctx->dev_root->fs_info;
562
558540c1
JS		 563 path = btrfs_alloc_path();
564
565 swarn.scratch_buf = kmalloc(bufsize, GFP_NOFS);
566 swarn.msg_buf = kmalloc(bufsize, GFP_NOFS);
7a9e9987
SB		 567 swarn.sector = (sblock->pagev[0]->physical) >> 9;
568 swarn.logical = sblock->pagev[0]->logical;
558540c1 569 swarn.errstr = errstr;
a36cf8b8 570 swarn.dev = NULL;
558540c1
JS		 571 swarn.msg_bufsize = bufsize;
572 swarn.scratch_bufsize = bufsize;
573
574 if (!path || !swarn.scratch_buf || !swarn.msg_buf)
575 goto out;
576
69917e43
LB		 577 ret = extent_from_logical(fs_info, swarn.logical, path, &found_key,
578 &flags);
558540c1
JS		 579 if (ret < 0)
580 goto out;
581
4692cf58 582 extent_item_pos = swarn.logical - found_key.objectid;
558540c1
JS		 583 swarn.extent_item_size = found_key.offset;
584
585 eb = path->nodes[0];
586 ei = btrfs_item_ptr(eb, path->slots[0], struct btrfs_extent_item);
587 item_size = btrfs_item_size_nr(eb, path->slots[0]);
588
69917e43 589 if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK) {
558540c1 590 do {
6eda71d0
LB		 591 ret = tree_backref_for_extent(&ptr, eb, &found_key, ei,
592 item_size, &ref_root,
593 &ref_level);
606686ee 594 printk_in_rcu(KERN_WARNING
efe120a0 595 "BTRFS: %s at logical %llu on dev %s, "
558540c1 596 "sector %llu: metadata %s (level %d) in tree "
606686ee
JB		 597 "%llu\n", errstr, swarn.logical,
598 rcu_str_deref(dev->name),
558540c1
JS		 599 (unsigned long long)swarn.sector,
600 ref_level ? "node" : "leaf",
601 ret < 0 ? -1 : ref_level,
602 ret < 0 ? -1 : ref_root);
603 } while (ret != 1);
d8fe29e9 604 btrfs_release_path(path);
558540c1 605 } else {
d8fe29e9 606 btrfs_release_path(path);
558540c1 607 swarn.path = path;
a36cf8b8 608 swarn.dev = dev;
7a3ae2f8
JS		 609 iterate_extent_inodes(fs_info, found_key.objectid,
610 extent_item_pos, 1,
558540c1
JS		 611 scrub_print_warning_inode, &swarn);
612 }
613
614out:
615 btrfs_free_path(path);
616 kfree(swarn.scratch_buf);
617 kfree(swarn.msg_buf);
618}
619
ff023aac 620static int scrub_fixup_readpage(u64 inum, u64 offset, u64 root, void *fixup_ctx)
0ef8e451 621{
5da6fcbc 622 struct page *page = NULL;
0ef8e451 623 unsigned long index;
ff023aac 624 struct scrub_fixup_nodatasum *fixup = fixup_ctx;
0ef8e451 625 int ret;
5da6fcbc 626 int corrected = 0;
0ef8e451 627 struct btrfs_key key;
5da6fcbc 628 struct inode *inode = NULL;
6f1c3605 629 struct btrfs_fs_info *fs_info;
0ef8e451
JS		 630 u64 end = offset + PAGE_SIZE - 1;
631 struct btrfs_root *local_root;
6f1c3605 632 int srcu_index;
0ef8e451
JS		 633
634 key.objectid = root;
635 key.type = BTRFS_ROOT_ITEM_KEY;
636 key.offset = (u64)-1;
6f1c3605
LB		 637
638 fs_info = fixup->root->fs_info;
639 srcu_index = srcu_read_lock(&fs_info->subvol_srcu);
640
641 local_root = btrfs_read_fs_root_no_name(fs_info, &key);
642 if (IS_ERR(local_root)) {
643 srcu_read_unlock(&fs_info->subvol_srcu, srcu_index);
0ef8e451 644 return PTR_ERR(local_root);
6f1c3605 645 }
0ef8e451
JS		 646
647 key.type = BTRFS_INODE_ITEM_KEY;
648 key.objectid = inum;
649 key.offset = 0;
6f1c3605
LB		 650 inode = btrfs_iget(fs_info->sb, &key, local_root, NULL);
651 srcu_read_unlock(&fs_info->subvol_srcu, srcu_index);
0ef8e451
JS		 652 if (IS_ERR(inode))
653 return PTR_ERR(inode);
654
0ef8e451
JS		 655 index = offset >> PAGE_CACHE_SHIFT;
656
657 page = find_or_create_page(inode->i_mapping, index, GFP_NOFS);
5da6fcbc
JS		 658 if (!page) {
659 ret = -ENOMEM;
660 goto out;
661 }
662
663 if (PageUptodate(page)) {
5da6fcbc
JS		 664 if (PageDirty(page)) {
665 /*
666 * we need to write the data to the defect sector. the
667 * data that was in that sector is not in memory,
668 * because the page was modified. we must not write the
669 * modified page to that sector.
670 *
671 * TODO: what could be done here: wait for the delalloc
672 * runner to write out that page (might involve
673 * COW) and see whether the sector is still
674 * referenced afterwards.
675 *
676 * For the meantime, we'll treat this error
677 * incorrectable, although there is a chance that a
678 * later scrub will find the bad sector again and that
679 * there's no dirty page in memory, then.
680 */
681 ret = -EIO;
682 goto out;
683 }
3ec706c8
SB		 684 fs_info = BTRFS_I(inode)->root->fs_info;
685 ret = repair_io_failure(fs_info, offset, PAGE_SIZE,
5da6fcbc
JS		 686 fixup->logical, page,
687 fixup->mirror_num);
688 unlock_page(page);
689 corrected = !ret;
690 } else {
691 /*
692 * we need to get good data first. the general readpage path
693 * will call repair_io_failure for us, we just have to make
694 * sure we read the bad mirror.
695 */
696 ret = set_extent_bits(&BTRFS_I(inode)->io_tree, offset, end,
697 EXTENT_DAMAGED, GFP_NOFS);
698 if (ret) {
699 /* set_extent_bits should give proper error */
700 WARN_ON(ret > 0);
701 if (ret > 0)
702 ret = -EFAULT;
703 goto out;
704 }
705
706 ret = extent_read_full_page(&BTRFS_I(inode)->io_tree, page,
707 btrfs_get_extent,
708 fixup->mirror_num);
709 wait_on_page_locked(page);
710
711 corrected = !test_range_bit(&BTRFS_I(inode)->io_tree, offset,
712 end, EXTENT_DAMAGED, 0, NULL);
713 if (!corrected)
714 clear_extent_bits(&BTRFS_I(inode)->io_tree, offset, end,
715 EXTENT_DAMAGED, GFP_NOFS);
716 }
717
718out:
719 if (page)
720 put_page(page);
7fb18a06
TK		 721
722 iput(inode);
0ef8e451
JS		 723
724 if (ret < 0)
725 return ret;
726
727 if (ret == 0 && corrected) {
728 /*
729 * we only need to call readpage for one of the inodes belonging
730 * to this extent. so make iterate_extent_inodes stop
731 */
732 return 1;
733 }
734
735 return -EIO;
736}
737
738static void scrub_fixup_nodatasum(struct btrfs_work *work)
739{
740 int ret;
741 struct scrub_fixup_nodatasum *fixup;
d9d181c1 742 struct scrub_ctx *sctx;
0ef8e451 743 struct btrfs_trans_handle *trans = NULL;
0ef8e451
JS		 744 struct btrfs_path *path;
745 int uncorrectable = 0;
746
747 fixup = container_of(work, struct scrub_fixup_nodatasum, work);
d9d181c1 748 sctx = fixup->sctx;
0ef8e451
JS		 749
750 path = btrfs_alloc_path();
751 if (!path) {
d9d181c1
SB		 752 spin_lock(&sctx->stat_lock);
753 ++sctx->stat.malloc_errors;
754 spin_unlock(&sctx->stat_lock);
0ef8e451
JS		 755 uncorrectable = 1;
756 goto out;
757 }
758
759 trans = btrfs_join_transaction(fixup->root);
760 if (IS_ERR(trans)) {
761 uncorrectable = 1;
762 goto out;
763 }
764
765 /*
766 * the idea is to trigger a regular read through the standard path. we
767 * read a page from the (failed) logical address by specifying the
768 * corresponding copynum of the failed sector. thus, that readpage is
769 * expected to fail.
770 * that is the point where on-the-fly error correction will kick in
771 * (once it's finished) and rewrite the failed sector if a good copy
772 * can be found.
773 */
774 ret = iterate_inodes_from_logical(fixup->logical, fixup->root->fs_info,
775 path, scrub_fixup_readpage,
776 fixup);
777 if (ret < 0) {
778 uncorrectable = 1;
779 goto out;
780 }
781 WARN_ON(ret != 1);
782
d9d181c1
SB		 783 spin_lock(&sctx->stat_lock);
784 ++sctx->stat.corrected_errors;
785 spin_unlock(&sctx->stat_lock);
0ef8e451
JS		 786
787out:
788 if (trans && !IS_ERR(trans))
789 btrfs_end_transaction(trans, fixup->root);
790 if (uncorrectable) {
d9d181c1
SB		 791 spin_lock(&sctx->stat_lock);
792 ++sctx->stat.uncorrectable_errors;
793 spin_unlock(&sctx->stat_lock);
ff023aac
SB		 794 btrfs_dev_replace_stats_inc(
795 &sctx->dev_root->fs_info->dev_replace.
796 num_uncorrectable_read_errors);
efe120a0
FH		 797 printk_ratelimited_in_rcu(KERN_ERR "BTRFS: "
798 "unable to fixup (nodatasum) error at logical %llu on dev %s\n",
c1c9ff7c 799 fixup->logical, rcu_str_deref(fixup->dev->name));
0ef8e451
JS		 800 }
801
802 btrfs_free_path(path);
803 kfree(fixup);
804
b6bfebc1 805 scrub_pending_trans_workers_dec(sctx);
0ef8e451
JS		 806}
807
a2de733c 808/*
b5d67f64
SB		 809 * scrub_handle_errored_block gets called when either verification of the
810 * pages failed or the bio failed to read, e.g. with EIO. In the latter
811 * case, this function handles all pages in the bio, even though only one
812 * may be bad.
813 * The goal of this function is to repair the errored block by using the
814 * contents of one of the mirrors.
a2de733c 815 */
b5d67f64 816static int scrub_handle_errored_block(struct scrub_block *sblock_to_check)
a2de733c 817{
d9d181c1 818 struct scrub_ctx *sctx = sblock_to_check->sctx;
a36cf8b8 819 struct btrfs_device *dev;
b5d67f64
SB		 820 struct btrfs_fs_info *fs_info;
821 u64 length;
822 u64 logical;
823 u64 generation;
824 unsigned int failed_mirror_index;
825 unsigned int is_metadata;
826 unsigned int have_csum;
827 u8 *csum;
828 struct scrub_block *sblocks_for_recheck; /* holds one for each mirror */
829 struct scrub_block *sblock_bad;
830 int ret;
831 int mirror_index;
832 int page_num;
833 int success;
558540c1 834 static DEFINE_RATELIMIT_STATE(_rs, DEFAULT_RATELIMIT_INTERVAL,
b5d67f64
SB		 835 DEFAULT_RATELIMIT_BURST);
836
837 BUG_ON(sblock_to_check->page_count < 1);
a36cf8b8 838 fs_info = sctx->dev_root->fs_info;
4ded4f63
SB		 839 if (sblock_to_check->pagev[0]->flags & BTRFS_EXTENT_FLAG_SUPER) {
840 /*
841 * if we find an error in a super block, we just report it.
842 * They will get written with the next transaction commit
843 * anyway
844 */
845 spin_lock(&sctx->stat_lock);
846 ++sctx->stat.super_errors;
847 spin_unlock(&sctx->stat_lock);
848 return 0;
849 }
b5d67f64 850 length = sblock_to_check->page_count * PAGE_SIZE;
7a9e9987
SB		 851 logical = sblock_to_check->pagev[0]->logical;
852 generation = sblock_to_check->pagev[0]->generation;
853 BUG_ON(sblock_to_check->pagev[0]->mirror_num < 1);
854 failed_mirror_index = sblock_to_check->pagev[0]->mirror_num - 1;
855 is_metadata = !(sblock_to_check->pagev[0]->flags &
b5d67f64 856 BTRFS_EXTENT_FLAG_DATA);
7a9e9987
SB		 857 have_csum = sblock_to_check->pagev[0]->have_csum;
858 csum = sblock_to_check->pagev[0]->csum;
859 dev = sblock_to_check->pagev[0]->dev;
13db62b7 860
ff023aac
SB		 861 if (sctx->is_dev_replace && !is_metadata && !have_csum) {
862 sblocks_for_recheck = NULL;
863 goto nodatasum_case;
864 }
865
b5d67f64
SB		 866 /*
867 * read all mirrors one after the other. This includes to
868 * re-read the extent or metadata block that failed (that was
869 * the cause that this fixup code is called) another time,
870 * page by page this time in order to know which pages
871 * caused I/O errors and which ones are good (for all mirrors).
872 * It is the goal to handle the situation when more than one
873 * mirror contains I/O errors, but the errors do not
874 * overlap, i.e. the data can be repaired by selecting the
875 * pages from those mirrors without I/O error on the
876 * particular pages. One example (with blocks >= 2 * PAGE_SIZE)
877 * would be that mirror #1 has an I/O error on the first page,
878 * the second page is good, and mirror #2 has an I/O error on
879 * the second page, but the first page is good.
880 * Then the first page of the first mirror can be repaired by
881 * taking the first page of the second mirror, and the
882 * second page of the second mirror can be repaired by
883 * copying the contents of the 2nd page of the 1st mirror.
884 * One more note: if the pages of one mirror contain I/O
885 * errors, the checksum cannot be verified. In order to get
886 * the best data for repairing, the first attempt is to find
887 * a mirror without I/O errors and with a validated checksum.
888 * Only if this is not possible, the pages are picked from
889 * mirrors with I/O errors without considering the checksum.
890 * If the latter is the case, at the end, the checksum of the
891 * repaired area is verified in order to correctly maintain
892 * the statistics.
893 */
894
895 sblocks_for_recheck = kzalloc(BTRFS_MAX_MIRRORS *
896 sizeof(*sblocks_for_recheck),
897 GFP_NOFS);
898 if (!sblocks_for_recheck) {
d9d181c1
SB		 899 spin_lock(&sctx->stat_lock);
900 sctx->stat.malloc_errors++;
901 sctx->stat.read_errors++;
902 sctx->stat.uncorrectable_errors++;
903 spin_unlock(&sctx->stat_lock);
a36cf8b8 904 btrfs_dev_stat_inc_and_print(dev, BTRFS_DEV_STAT_READ_ERRS);
b5d67f64 905 goto out;
a2de733c
AJ		 906 }
907
b5d67f64 908 /* setup the context, map the logical blocks and alloc the pages */
ff023aac 909 ret = scrub_setup_recheck_block(sctx, fs_info, sblock_to_check, length,
b5d67f64
SB		 910 logical, sblocks_for_recheck);
911 if (ret) {
d9d181c1
SB		 912 spin_lock(&sctx->stat_lock);
913 sctx->stat.read_errors++;
914 sctx->stat.uncorrectable_errors++;
915 spin_unlock(&sctx->stat_lock);
a36cf8b8 916 btrfs_dev_stat_inc_and_print(dev, BTRFS_DEV_STAT_READ_ERRS);
b5d67f64
SB		 917 goto out;
918 }
919 BUG_ON(failed_mirror_index >= BTRFS_MAX_MIRRORS);
920 sblock_bad = sblocks_for_recheck + failed_mirror_index;
13db62b7 921
b5d67f64 922 /* build and submit the bios for the failed mirror, check checksums */
34f5c8e9
SB		 923 scrub_recheck_block(fs_info, sblock_bad, is_metadata, have_csum,
924 csum, generation, sctx->csum_size);
a2de733c 925
b5d67f64
SB		 926 if (!sblock_bad->header_error && !sblock_bad->checksum_error &&
927 sblock_bad->no_io_error_seen) {
928 /*
929 * the error disappeared after reading page by page, or
930 * the area was part of a huge bio and other parts of the
931 * bio caused I/O errors, or the block layer merged several
932 * read requests into one and the error is caused by a
933 * different bio (usually one of the two latter cases is
934 * the cause)
935 */
d9d181c1
SB		 936 spin_lock(&sctx->stat_lock);
937 sctx->stat.unverified_errors++;
938 spin_unlock(&sctx->stat_lock);
a2de733c 939
ff023aac
SB		 940 if (sctx->is_dev_replace)
941 scrub_write_block_to_dev_replace(sblock_bad);
b5d67f64 942 goto out;
a2de733c 943 }
a2de733c 944
b5d67f64 945 if (!sblock_bad->no_io_error_seen) {
d9d181c1
SB		 946 spin_lock(&sctx->stat_lock);
947 sctx->stat.read_errors++;
948 spin_unlock(&sctx->stat_lock);
b5d67f64
SB		 949 if (__ratelimit(&_rs))
950 scrub_print_warning("i/o error", sblock_to_check);
a36cf8b8 951 btrfs_dev_stat_inc_and_print(dev, BTRFS_DEV_STAT_READ_ERRS);
b5d67f64 952 } else if (sblock_bad->checksum_error) {
d9d181c1
SB		 953 spin_lock(&sctx->stat_lock);
954 sctx->stat.csum_errors++;
955 spin_unlock(&sctx->stat_lock);
b5d67f64
SB		 956 if (__ratelimit(&_rs))
957 scrub_print_warning("checksum error", sblock_to_check);
a36cf8b8 958 btrfs_dev_stat_inc_and_print(dev,
442a4f63 959 BTRFS_DEV_STAT_CORRUPTION_ERRS);
b5d67f64 960 } else if (sblock_bad->header_error) {
d9d181c1
SB		 961 spin_lock(&sctx->stat_lock);
962 sctx->stat.verify_errors++;
963 spin_unlock(&sctx->stat_lock);
b5d67f64
SB		 964 if (__ratelimit(&_rs))
965 scrub_print_warning("checksum/header error",
966 sblock_to_check);
442a4f63 967 if (sblock_bad->generation_error)
a36cf8b8 968 btrfs_dev_stat_inc_and_print(dev,
442a4f63
SB		 969 BTRFS_DEV_STAT_GENERATION_ERRS);
970 else
a36cf8b8 971 btrfs_dev_stat_inc_and_print(dev,
442a4f63 972 BTRFS_DEV_STAT_CORRUPTION_ERRS);
b5d67f64 973 }
a2de733c 974
33ef30ad
ID		 975 if (sctx->readonly) {
976 ASSERT(!sctx->is_dev_replace);
977 goto out;
978 }
a2de733c 979
b5d67f64
SB		 980 if (!is_metadata && !have_csum) {
981 struct scrub_fixup_nodatasum *fixup_nodatasum;
a2de733c 982
ff023aac
SB		 983nodatasum_case:
984 WARN_ON(sctx->is_dev_replace);
985
b5d67f64
SB		 986 /*
987 * !is_metadata and !have_csum, this means that the data
988 * might not be COW'ed, that it might be modified
989 * concurrently. The general strategy to work on the
990 * commit root does not help in the case when COW is not
991 * used.
992 */
993 fixup_nodatasum = kzalloc(sizeof(*fixup_nodatasum), GFP_NOFS);
994 if (!fixup_nodatasum)
995 goto did_not_correct_error;
d9d181c1 996 fixup_nodatasum->sctx = sctx;
a36cf8b8 997 fixup_nodatasum->dev = dev;
b5d67f64
SB		 998 fixup_nodatasum->logical = logical;
999 fixup_nodatasum->root = fs_info->extent_root;
1000 fixup_nodatasum->mirror_num = failed_mirror_index + 1;
b6bfebc1 1001 scrub_pending_trans_workers_inc(sctx);
9e0af237
LB		 1002 btrfs_init_work(&fixup_nodatasum->work, btrfs_scrub_helper,
1003 scrub_fixup_nodatasum, NULL, NULL);
0339ef2f
QW		 1004 btrfs_queue_work(fs_info->scrub_workers,
1005 &fixup_nodatasum->work);
b5d67f64 1006 goto out;
a2de733c
AJ		 1007 }
1008
b5d67f64
SB		 1009 /*
1010 * now build and submit the bios for the other mirrors, check
cb2ced73
SB		 1011 * checksums.
1012 * First try to pick the mirror which is completely without I/O
b5d67f64
SB		 1013 * errors and also does not have a checksum error.
1014 * If one is found, and if a checksum is present, the full block
1015 * that is known to contain an error is rewritten. Afterwards
1016 * the block is known to be corrected.
1017 * If a mirror is found which is completely correct, and no
1018 * checksum is present, only those pages are rewritten that had
1019 * an I/O error in the block to be repaired, since it cannot be
1020 * determined, which copy of the other pages is better (and it
1021 * could happen otherwise that a correct page would be
1022 * overwritten by a bad one).
1023 */
1024 for (mirror_index = 0;
1025 mirror_index < BTRFS_MAX_MIRRORS &&
1026 sblocks_for_recheck[mirror_index].page_count > 0;
1027 mirror_index++) {
cb2ced73 1028 struct scrub_block *sblock_other;
b5d67f64 1029
cb2ced73
SB		 1030 if (mirror_index == failed_mirror_index)
1031 continue;
1032 sblock_other = sblocks_for_recheck + mirror_index;
1033
1034 /* build and submit the bios, check checksums */
34f5c8e9
SB		 1035 scrub_recheck_block(fs_info, sblock_other, is_metadata,
1036 have_csum, csum, generation,
1037 sctx->csum_size);
1038
1039 if (!sblock_other->header_error &&
b5d67f64
SB		 1040 !sblock_other->checksum_error &&
1041 sblock_other->no_io_error_seen) {
ff023aac
SB		 1042 if (sctx->is_dev_replace) {
1043 scrub_write_block_to_dev_replace(sblock_other);
1044 } else {
1045 int force_write = is_metadata || have_csum;
1046
1047 ret = scrub_repair_block_from_good_copy(
1048 sblock_bad, sblock_other,
1049 force_write);
1050 }
b5d67f64
SB		 1051 if (0 == ret)
1052 goto corrected_error;
1053 }
1054 }
a2de733c
AJ		 1055
1056 /*
ff023aac
SB		 1057 * for dev_replace, pick good pages and write to the target device.
1058 */
1059 if (sctx->is_dev_replace) {
1060 success = 1;
1061 for (page_num = 0; page_num < sblock_bad->page_count;
1062 page_num++) {
1063 int sub_success;
1064
1065 sub_success = 0;
1066 for (mirror_index = 0;
1067 mirror_index < BTRFS_MAX_MIRRORS &&
1068 sblocks_for_recheck[mirror_index].page_count > 0;
1069 mirror_index++) {
1070 struct scrub_block *sblock_other =
1071 sblocks_for_recheck + mirror_index;
1072 struct scrub_page *page_other =
1073 sblock_other->pagev[page_num];
1074
1075 if (!page_other->io_error) {
1076 ret = scrub_write_page_to_dev_replace(
1077 sblock_other, page_num);
1078 if (ret == 0) {
1079 /* succeeded for this page */
1080 sub_success = 1;
1081 break;
1082 } else {
1083 btrfs_dev_replace_stats_inc(
1084 &sctx->dev_root->
1085 fs_info->dev_replace.
1086 num_write_errors);
1087 }
1088 }
1089 }
1090
1091 if (!sub_success) {
1092 /*
1093 * did not find a mirror to fetch the page
1094 * from. scrub_write_page_to_dev_replace()
1095 * handles this case (page->io_error), by
1096 * filling the block with zeros before
1097 * submitting the write request
1098 */
1099 success = 0;
1100 ret = scrub_write_page_to_dev_replace(
1101 sblock_bad, page_num);
1102 if (ret)
1103 btrfs_dev_replace_stats_inc(
1104 &sctx->dev_root->fs_info->
1105 dev_replace.num_write_errors);
1106 }
1107 }
1108
1109 goto out;
1110 }
1111
1112 /*
1113 * for regular scrub, repair those pages that are errored.
1114 * In case of I/O errors in the area that is supposed to be
b5d67f64
SB		 1115 * repaired, continue by picking good copies of those pages.
1116 * Select the good pages from mirrors to rewrite bad pages from
1117 * the area to fix. Afterwards verify the checksum of the block
1118 * that is supposed to be repaired. This verification step is
1119 * only done for the purpose of statistic counting and for the
1120 * final scrub report, whether errors remain.
1121 * A perfect algorithm could make use of the checksum and try
1122 * all possible combinations of pages from the different mirrors
1123 * until the checksum verification succeeds. For example, when
1124 * the 2nd page of mirror #1 faces I/O errors, and the 2nd page
1125 * of mirror #2 is readable but the final checksum test fails,
1126 * then the 2nd page of mirror #3 could be tried, whether now
1127 * the final checksum succeedes. But this would be a rare
1128 * exception and is therefore not implemented. At least it is
1129 * avoided that the good copy is overwritten.
1130 * A more useful improvement would be to pick the sectors
1131 * without I/O error based on sector sizes (512 bytes on legacy
1132 * disks) instead of on PAGE_SIZE. Then maybe 512 byte of one
1133 * mirror could be repaired by taking 512 byte of a different
1134 * mirror, even if other 512 byte sectors in the same PAGE_SIZE
1135 * area are unreadable.
a2de733c 1136 */
a2de733c 1137
b5d67f64
SB		 1138 /* can only fix I/O errors from here on */
1139 if (sblock_bad->no_io_error_seen)
1140 goto did_not_correct_error;
1141
1142 success = 1;
1143 for (page_num = 0; page_num < sblock_bad->page_count; page_num++) {
7a9e9987 1144 struct scrub_page *page_bad = sblock_bad->pagev[page_num];
b5d67f64
SB		 1145
1146 if (!page_bad->io_error)
a2de733c 1147 continue;
b5d67f64
SB		 1148
1149 for (mirror_index = 0;
1150 mirror_index < BTRFS_MAX_MIRRORS &&
1151 sblocks_for_recheck[mirror_index].page_count > 0;
1152 mirror_index++) {
1153 struct scrub_block *sblock_other = sblocks_for_recheck +
1154 mirror_index;
7a9e9987
SB		 1155 struct scrub_page *page_other = sblock_other->pagev[
1156 page_num];
b5d67f64
SB		 1157
1158 if (!page_other->io_error) {
1159 ret = scrub_repair_page_from_good_copy(
1160 sblock_bad, sblock_other, page_num, 0);
1161 if (0 == ret) {
1162 page_bad->io_error = 0;
1163 break; /* succeeded for this page */
1164 }
1165 }
96e36920 1166 }
a2de733c 1167
b5d67f64
SB		 1168 if (page_bad->io_error) {
1169 /* did not find a mirror to copy the page from */
1170 success = 0;
1171 }
a2de733c 1172 }
a2de733c 1173
b5d67f64
SB		 1174 if (success) {
1175 if (is_metadata || have_csum) {
1176 /*
1177 * need to verify the checksum now that all
1178 * sectors on disk are repaired (the write
1179 * request for data to be repaired is on its way).
1180 * Just be lazy and use scrub_recheck_block()
1181 * which re-reads the data before the checksum
1182 * is verified, but most likely the data comes out
1183 * of the page cache.
1184 */
34f5c8e9
SB		 1185 scrub_recheck_block(fs_info, sblock_bad,
1186 is_metadata, have_csum, csum,
1187 generation, sctx->csum_size);
1188 if (!sblock_bad->header_error &&
b5d67f64
SB		 1189 !sblock_bad->checksum_error &&
1190 sblock_bad->no_io_error_seen)
1191 goto corrected_error;
1192 else
1193 goto did_not_correct_error;
1194 } else {
1195corrected_error:
d9d181c1
SB		 1196 spin_lock(&sctx->stat_lock);
1197 sctx->stat.corrected_errors++;
1198 spin_unlock(&sctx->stat_lock);
606686ee 1199 printk_ratelimited_in_rcu(KERN_ERR
efe120a0 1200 "BTRFS: fixed up error at logical %llu on dev %s\n",
c1c9ff7c 1201 logical, rcu_str_deref(dev->name));
8628764e 1202 }
b5d67f64
SB		 1203 } else {
1204did_not_correct_error:
d9d181c1
SB		 1205 spin_lock(&sctx->stat_lock);
1206 sctx->stat.uncorrectable_errors++;
1207 spin_unlock(&sctx->stat_lock);
606686ee 1208 printk_ratelimited_in_rcu(KERN_ERR
efe120a0 1209 "BTRFS: unable to fixup (regular) error at logical %llu on dev %s\n",
c1c9ff7c 1210 logical, rcu_str_deref(dev->name));
96e36920 1211 }
a2de733c 1212
b5d67f64
SB		 1213out:
1214 if (sblocks_for_recheck) {
1215 for (mirror_index = 0; mirror_index < BTRFS_MAX_MIRRORS;
1216 mirror_index++) {
1217 struct scrub_block *sblock = sblocks_for_recheck +
1218 mirror_index;
1219 int page_index;
1220
7a9e9987
SB		 1221 for (page_index = 0; page_index < sblock->page_count;
1222 page_index++) {
1223 sblock->pagev[page_index]->sblock = NULL;
1224 scrub_page_put(sblock->pagev[page_index]);
1225 }
b5d67f64
SB		 1226 }
1227 kfree(sblocks_for_recheck);
1228 }
a2de733c 1229
b5d67f64
SB		 1230 return 0;
1231}
a2de733c 1232
d9d181c1 1233static int scrub_setup_recheck_block(struct scrub_ctx *sctx,
3ec706c8 1234 struct btrfs_fs_info *fs_info,
ff023aac 1235 struct scrub_block *original_sblock,
b5d67f64
SB		 1236 u64 length, u64 logical,
1237 struct scrub_block *sblocks_for_recheck)
1238{
1239 int page_index;
1240 int mirror_index;
1241 int ret;
1242
1243 /*
7a9e9987 1244 * note: the two members ref_count and outstanding_pages
b5d67f64
SB		 1245 * are not used (and not set) in the blocks that are used for
1246 * the recheck procedure
1247 */
1248
1249 page_index = 0;
1250 while (length > 0) {
1251 u64 sublen = min_t(u64, length, PAGE_SIZE);
1252 u64 mapped_length = sublen;
1253 struct btrfs_bio *bbio = NULL;
a2de733c 1254
b5d67f64
SB		 1255 /*
1256 * with a length of PAGE_SIZE, each returned stripe
1257 * represents one mirror
1258 */
29a8d9a0
SB		 1259 ret = btrfs_map_block(fs_info, REQ_GET_READ_MIRRORS, logical,
1260 &mapped_length, &bbio, 0);
b5d67f64
SB		 1261 if (ret || !bbio || mapped_length < sublen) {
1262 kfree(bbio);
1263 return -EIO;
1264 }
a2de733c 1265
ff023aac 1266 BUG_ON(page_index >= SCRUB_PAGES_PER_RD_BIO);
b5d67f64
SB		 1267 for (mirror_index = 0; mirror_index < (int)bbio->num_stripes;
1268 mirror_index++) {
1269 struct scrub_block *sblock;
1270 struct scrub_page *page;
1271
1272 if (mirror_index >= BTRFS_MAX_MIRRORS)
1273 continue;
1274
1275 sblock = sblocks_for_recheck + mirror_index;
7a9e9987
SB		 1276 sblock->sctx = sctx;
1277 page = kzalloc(sizeof(*page), GFP_NOFS);
1278 if (!page) {
1279leave_nomem:
d9d181c1
SB		 1280 spin_lock(&sctx->stat_lock);
1281 sctx->stat.malloc_errors++;
1282 spin_unlock(&sctx->stat_lock);
cf93dcce 1283 kfree(bbio);
b5d67f64
SB		 1284 return -ENOMEM;
1285 }
7a9e9987
SB		 1286 scrub_page_get(page);
1287 sblock->pagev[page_index] = page;
1288 page->logical = logical;
1289 page->physical = bbio->stripes[mirror_index].physical;
ff023aac
SB		 1290 BUG_ON(page_index >= original_sblock->page_count);
1291 page->physical_for_dev_replace =
1292 original_sblock->pagev[page_index]->
1293 physical_for_dev_replace;
7a9e9987
SB		 1294 /* for missing devices, dev->bdev is NULL */
1295 page->dev = bbio->stripes[mirror_index].dev;
1296 page->mirror_num = mirror_index + 1;
b5d67f64 1297 sblock->page_count++;
7a9e9987
SB		 1298 page->page = alloc_page(GFP_NOFS);
1299 if (!page->page)
1300 goto leave_nomem;
b5d67f64
SB		 1301 }
1302 kfree(bbio);
1303 length -= sublen;
1304 logical += sublen;
1305 page_index++;
1306 }
1307
1308 return 0;
96e36920
ID		 1309}
1310
b5d67f64
SB		 1311/*
1312 * this function will check the on disk data for checksum errors, header
1313 * errors and read I/O errors. If any I/O errors happen, the exact pages
1314 * which are errored are marked as being bad. The goal is to enable scrub
1315 * to take those pages that are not errored from all the mirrors so that
1316 * the pages that are errored in the just handled mirror can be repaired.
1317 */
34f5c8e9
SB		 1318static void scrub_recheck_block(struct btrfs_fs_info *fs_info,
1319 struct scrub_block *sblock, int is_metadata,
1320 int have_csum, u8 *csum, u64 generation,
1321 u16 csum_size)
96e36920 1322{
b5d67f64 1323 int page_num;
96e36920 1324
b5d67f64
SB		 1325 sblock->no_io_error_seen = 1;
1326 sblock->header_error = 0;
1327 sblock->checksum_error = 0;
96e36920 1328
b5d67f64
SB		 1329 for (page_num = 0; page_num < sblock->page_count; page_num++) {
1330 struct bio *bio;
7a9e9987 1331 struct scrub_page *page = sblock->pagev[page_num];
b5d67f64 1332
442a4f63 1333 if (page->dev->bdev == NULL) {
ea9947b4
SB		 1334 page->io_error = 1;
1335 sblock->no_io_error_seen = 0;
1336 continue;
1337 }
1338
7a9e9987 1339 WARN_ON(!page->page);
9be3395b 1340 bio = btrfs_io_bio_alloc(GFP_NOFS, 1);
34f5c8e9
SB		 1341 if (!bio) {
1342 page->io_error = 1;
1343 sblock->no_io_error_seen = 0;
1344 continue;
1345 }
442a4f63 1346 bio->bi_bdev = page->dev->bdev;
4f024f37 1347 bio->bi_iter.bi_sector = page->physical >> 9;
b5d67f64 1348
34f5c8e9 1349 bio_add_page(bio, page->page, PAGE_SIZE, 0);
33879d45 1350 if (btrfsic_submit_bio_wait(READ, bio))
b5d67f64 1351 sblock->no_io_error_seen = 0;
33879d45 1352
b5d67f64
SB		 1353 bio_put(bio);
1354 }
96e36920 1355
b5d67f64
SB		 1356 if (sblock->no_io_error_seen)
1357 scrub_recheck_block_checksum(fs_info, sblock, is_metadata,
1358 have_csum, csum, generation,
1359 csum_size);
1360
34f5c8e9 1361 return;
a2de733c
AJ		 1362}
1363
b5d67f64
SB		 1364static void scrub_recheck_block_checksum(struct btrfs_fs_info *fs_info,
1365 struct scrub_block *sblock,
1366 int is_metadata, int have_csum,
1367 const u8 *csum, u64 generation,
1368 u16 csum_size)
a2de733c 1369{
b5d67f64
SB		 1370 int page_num;
1371 u8 calculated_csum[BTRFS_CSUM_SIZE];
1372 u32 crc = ~(u32)0;
b5d67f64
SB		 1373 void *mapped_buffer;
1374
7a9e9987 1375 WARN_ON(!sblock->pagev[0]->page);
b5d67f64
SB		 1376 if (is_metadata) {
1377 struct btrfs_header *h;
1378
7a9e9987 1379 mapped_buffer = kmap_atomic(sblock->pagev[0]->page);
b5d67f64
SB		 1380 h = (struct btrfs_header *)mapped_buffer;
1381
3cae210f 1382 if (sblock->pagev[0]->logical != btrfs_stack_header_bytenr(h) ||
b5d67f64
SB		 1383 memcmp(h->fsid, fs_info->fsid, BTRFS_UUID_SIZE) ||
1384 memcmp(h->chunk_tree_uuid, fs_info->chunk_tree_uuid,
442a4f63 1385 BTRFS_UUID_SIZE)) {
b5d67f64 1386 sblock->header_error = 1;
3cae210f 1387 } else if (generation != btrfs_stack_header_generation(h)) {
442a4f63
SB		 1388 sblock->header_error = 1;
1389 sblock->generation_error = 1;
1390 }
b5d67f64
SB		 1391 csum = h->csum;
1392 } else {
1393 if (!have_csum)
1394 return;
a2de733c 1395
7a9e9987 1396 mapped_buffer = kmap_atomic(sblock->pagev[0]->page);
b5d67f64 1397 }
a2de733c 1398
b5d67f64
SB		 1399 for (page_num = 0;;) {
1400 if (page_num == 0 && is_metadata)
b0496686 1401 crc = btrfs_csum_data(
b5d67f64
SB		 1402 ((u8 *)mapped_buffer) + BTRFS_CSUM_SIZE,
1403 crc, PAGE_SIZE - BTRFS_CSUM_SIZE);
1404 else
b0496686 1405 crc = btrfs_csum_data(mapped_buffer, crc, PAGE_SIZE);
b5d67f64 1406
9613bebb 1407 kunmap_atomic(mapped_buffer);
b5d67f64
SB		 1408 page_num++;
1409 if (page_num >= sblock->page_count)
1410 break;
7a9e9987 1411 WARN_ON(!sblock->pagev[page_num]->page);
b5d67f64 1412
7a9e9987 1413 mapped_buffer = kmap_atomic(sblock->pagev[page_num]->page);
b5d67f64
SB		 1414 }
1415
1416 btrfs_csum_final(crc, calculated_csum);
1417 if (memcmp(calculated_csum, csum, csum_size))
1418 sblock->checksum_error = 1;
a2de733c
AJ		 1419}
1420
b5d67f64
SB		 1421static int scrub_repair_block_from_good_copy(struct scrub_block *sblock_bad,
1422 struct scrub_block *sblock_good,
1423 int force_write)
1424{
1425 int page_num;
1426 int ret = 0;
96e36920 1427
b5d67f64
SB		 1428 for (page_num = 0; page_num < sblock_bad->page_count; page_num++) {
1429 int ret_sub;
96e36920 1430
b5d67f64
SB		 1431 ret_sub = scrub_repair_page_from_good_copy(sblock_bad,
1432 sblock_good,
1433 page_num,
1434 force_write);
1435 if (ret_sub)
1436 ret = ret_sub;
a2de733c 1437 }
b5d67f64
SB		 1438
1439 return ret;
1440}
1441
1442static int scrub_repair_page_from_good_copy(struct scrub_block *sblock_bad,
1443 struct scrub_block *sblock_good,
1444 int page_num, int force_write)
1445{
7a9e9987
SB		 1446 struct scrub_page *page_bad = sblock_bad->pagev[page_num];
1447 struct scrub_page *page_good = sblock_good->pagev[page_num];
b5d67f64 1448
7a9e9987
SB		 1449 BUG_ON(page_bad->page == NULL);
1450 BUG_ON(page_good->page == NULL);
b5d67f64
SB		 1451 if (force_write || sblock_bad->header_error ||
1452 sblock_bad->checksum_error || page_bad->io_error) {
1453 struct bio *bio;
1454 int ret;
b5d67f64 1455
ff023aac 1456 if (!page_bad->dev->bdev) {
efe120a0
FH		 1457 printk_ratelimited(KERN_WARNING "BTRFS: "
1458 "scrub_repair_page_from_good_copy(bdev == NULL) "
1459 "is unexpected!\n");
ff023aac
SB		 1460 return -EIO;
1461 }
1462
9be3395b 1463 bio = btrfs_io_bio_alloc(GFP_NOFS, 1);
e627ee7b
TI		 1464 if (!bio)
1465 return -EIO;
442a4f63 1466 bio->bi_bdev = page_bad->dev->bdev;
4f024f37 1467 bio->bi_iter.bi_sector = page_bad->physical >> 9;
b5d67f64
SB		 1468
1469 ret = bio_add_page(bio, page_good->page, PAGE_SIZE, 0);
1470 if (PAGE_SIZE != ret) {
1471 bio_put(bio);
1472 return -EIO;
13db62b7 1473 }
b5d67f64 1474
33879d45 1475 if (btrfsic_submit_bio_wait(WRITE, bio)) {
442a4f63
SB		 1476 btrfs_dev_stat_inc_and_print(page_bad->dev,
1477 BTRFS_DEV_STAT_WRITE_ERRS);
ff023aac
SB		 1478 btrfs_dev_replace_stats_inc(
1479 &sblock_bad->sctx->dev_root->fs_info->
1480 dev_replace.num_write_errors);
442a4f63
SB		 1481 bio_put(bio);
1482 return -EIO;
1483 }
b5d67f64 1484 bio_put(bio);
a2de733c
AJ		 1485 }
1486
b5d67f64
SB		 1487 return 0;
1488}
1489
ff023aac
SB		 1490static void scrub_write_block_to_dev_replace(struct scrub_block *sblock)
1491{
1492 int page_num;
1493
1494 for (page_num = 0; page_num < sblock->page_count; page_num++) {
1495 int ret;
1496
1497 ret = scrub_write_page_to_dev_replace(sblock, page_num);
1498 if (ret)
1499 btrfs_dev_replace_stats_inc(
1500 &sblock->sctx->dev_root->fs_info->dev_replace.
1501 num_write_errors);
1502 }
1503}
1504
1505static int scrub_write_page_to_dev_replace(struct scrub_block *sblock,
1506 int page_num)
1507{
1508 struct scrub_page *spage = sblock->pagev[page_num];
1509
1510 BUG_ON(spage->page == NULL);
1511 if (spage->io_error) {
1512 void *mapped_buffer = kmap_atomic(spage->page);
1513
1514 memset(mapped_buffer, 0, PAGE_CACHE_SIZE);
1515 flush_dcache_page(spage->page);
1516 kunmap_atomic(mapped_buffer);
1517 }
1518 return scrub_add_page_to_wr_bio(sblock->sctx, spage);
1519}
1520
1521static int scrub_add_page_to_wr_bio(struct scrub_ctx *sctx,
1522 struct scrub_page *spage)
1523{
1524 struct scrub_wr_ctx *wr_ctx = &sctx->wr_ctx;
1525 struct scrub_bio *sbio;
1526 int ret;
1527
1528 mutex_lock(&wr_ctx->wr_lock);
1529again:
1530 if (!wr_ctx->wr_curr_bio) {
1531 wr_ctx->wr_curr_bio = kzalloc(sizeof(*wr_ctx->wr_curr_bio),
1532 GFP_NOFS);
1533 if (!wr_ctx->wr_curr_bio) {
1534 mutex_unlock(&wr_ctx->wr_lock);
1535 return -ENOMEM;
1536 }
1537 wr_ctx->wr_curr_bio->sctx = sctx;
1538 wr_ctx->wr_curr_bio->page_count = 0;
1539 }
1540 sbio = wr_ctx->wr_curr_bio;
1541 if (sbio->page_count == 0) {
1542 struct bio *bio;
1543
1544 sbio->physical = spage->physical_for_dev_replace;
1545 sbio->logical = spage->logical;
1546 sbio->dev = wr_ctx->tgtdev;
1547 bio = sbio->bio;
1548 if (!bio) {
9be3395b 1549 bio = btrfs_io_bio_alloc(GFP_NOFS, wr_ctx->pages_per_wr_bio);
ff023aac
SB		 1550 if (!bio) {
1551 mutex_unlock(&wr_ctx->wr_lock);
1552 return -ENOMEM;
1553 }
1554 sbio->bio = bio;
1555 }
1556
1557 bio->bi_private = sbio;
1558 bio->bi_end_io = scrub_wr_bio_end_io;
1559 bio->bi_bdev = sbio->dev->bdev;
4f024f37 1560 bio->bi_iter.bi_sector = sbio->physical >> 9;
ff023aac
SB		 1561 sbio->err = 0;
1562 } else if (sbio->physical + sbio->page_count * PAGE_SIZE !=
1563 spage->physical_for_dev_replace ||
1564 sbio->logical + sbio->page_count * PAGE_SIZE !=
1565 spage->logical) {
1566 scrub_wr_submit(sctx);
1567 goto again;
1568 }
1569
1570 ret = bio_add_page(sbio->bio, spage->page, PAGE_SIZE, 0);
1571 if (ret != PAGE_SIZE) {
1572 if (sbio->page_count < 1) {
1573 bio_put(sbio->bio);
1574 sbio->bio = NULL;
1575 mutex_unlock(&wr_ctx->wr_lock);
1576 return -EIO;
1577 }
1578 scrub_wr_submit(sctx);
1579 goto again;
1580 }
1581
1582 sbio->pagev[sbio->page_count] = spage;
1583 scrub_page_get(spage);
1584 sbio->page_count++;
1585 if (sbio->page_count == wr_ctx->pages_per_wr_bio)
1586 scrub_wr_submit(sctx);
1587 mutex_unlock(&wr_ctx->wr_lock);
1588
1589 return 0;
1590}
1591
1592static void scrub_wr_submit(struct scrub_ctx *sctx)
1593{
1594 struct scrub_wr_ctx *wr_ctx = &sctx->wr_ctx;
1595 struct scrub_bio *sbio;
1596
1597 if (!wr_ctx->wr_curr_bio)
1598 return;
1599
1600 sbio = wr_ctx->wr_curr_bio;
1601 wr_ctx->wr_curr_bio = NULL;
1602 WARN_ON(!sbio->bio->bi_bdev);
1603 scrub_pending_bio_inc(sctx);
1604 /* process all writes in a single worker thread. Then the block layer
1605 * orders the requests before sending them to the driver which
1606 * doubled the write performance on spinning disks when measured
1607 * with Linux 3.5 */
1608 btrfsic_submit_bio(WRITE, sbio->bio);
1609}
1610
1611static void scrub_wr_bio_end_io(struct bio *bio, int err)
1612{
1613 struct scrub_bio *sbio = bio->bi_private;
1614 struct btrfs_fs_info *fs_info = sbio->dev->dev_root->fs_info;
1615
1616 sbio->err = err;
1617 sbio->bio = bio;
1618
9e0af237
LB		 1619 btrfs_init_work(&sbio->work, btrfs_scrubwrc_helper,
1620 scrub_wr_bio_end_io_worker, NULL, NULL);
0339ef2f 1621 btrfs_queue_work(fs_info->scrub_wr_completion_workers, &sbio->work);
ff023aac
SB		 1622}
1623
1624static void scrub_wr_bio_end_io_worker(struct btrfs_work *work)
1625{
1626 struct scrub_bio *sbio = container_of(work, struct scrub_bio, work);
1627 struct scrub_ctx *sctx = sbio->sctx;
1628 int i;
1629
1630 WARN_ON(sbio->page_count > SCRUB_PAGES_PER_WR_BIO);
1631 if (sbio->err) {
1632 struct btrfs_dev_replace *dev_replace =
1633 &sbio->sctx->dev_root->fs_info->dev_replace;
1634
1635 for (i = 0; i < sbio->page_count; i++) {
1636 struct scrub_page *spage = sbio->pagev[i];
1637
1638 spage->io_error = 1;
1639 btrfs_dev_replace_stats_inc(&dev_replace->
1640 num_write_errors);
1641 }
1642 }
1643
1644 for (i = 0; i < sbio->page_count; i++)
1645 scrub_page_put(sbio->pagev[i]);
1646
1647 bio_put(sbio->bio);
1648 kfree(sbio);
1649 scrub_pending_bio_dec(sctx);
1650}
1651
1652static int scrub_checksum(struct scrub_block *sblock)
b5d67f64
SB		 1653{
1654 u64 flags;
1655 int ret;
1656
7a9e9987
SB		 1657 WARN_ON(sblock->page_count < 1);
1658 flags = sblock->pagev[0]->flags;
b5d67f64
SB		 1659 ret = 0;
1660 if (flags & BTRFS_EXTENT_FLAG_DATA)
1661 ret = scrub_checksum_data(sblock);
1662 else if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK)
1663 ret = scrub_checksum_tree_block(sblock);
1664 else if (flags & BTRFS_EXTENT_FLAG_SUPER)
1665 (void)scrub_checksum_super(sblock);
1666 else
1667 WARN_ON(1);
1668 if (ret)
1669 scrub_handle_errored_block(sblock);
ff023aac
SB		 1670
1671 return ret;
a2de733c
AJ		 1672}
1673
b5d67f64 1674static int scrub_checksum_data(struct scrub_block *sblock)
a2de733c 1675{
d9d181c1 1676 struct scrub_ctx *sctx = sblock->sctx;
a2de733c 1677 u8 csum[BTRFS_CSUM_SIZE];
b5d67f64
SB		 1678 u8 *on_disk_csum;
1679 struct page *page;
1680 void *buffer;
a2de733c
AJ		 1681 u32 crc = ~(u32)0;
1682 int fail = 0;
b5d67f64
SB		 1683 u64 len;
1684 int index;
a2de733c 1685
b5d67f64 1686 BUG_ON(sblock->page_count < 1);
7a9e9987 1687 if (!sblock->pagev[0]->have_csum)
a2de733c
AJ		 1688 return 0;
1689
7a9e9987
SB		 1690 on_disk_csum = sblock->pagev[0]->csum;
1691 page = sblock->pagev[0]->page;
9613bebb 1692 buffer = kmap_atomic(page);
b5d67f64 1693
d9d181c1 1694 len = sctx->sectorsize;
b5d67f64
SB		 1695 index = 0;
1696 for (;;) {
1697 u64 l = min_t(u64, len, PAGE_SIZE);
1698
b0496686 1699 crc = btrfs_csum_data(buffer, crc, l);
9613bebb 1700 kunmap_atomic(buffer);
b5d67f64
SB		 1701 len -= l;
1702 if (len == 0)
1703 break;
1704 index++;
1705 BUG_ON(index >= sblock->page_count);
7a9e9987
SB		 1706 BUG_ON(!sblock->pagev[index]->page);
1707 page = sblock->pagev[index]->page;
9613bebb 1708 buffer = kmap_atomic(page);
b5d67f64
SB		 1709 }
1710
a2de733c 1711 btrfs_csum_final(crc, csum);
d9d181c1 1712 if (memcmp(csum, on_disk_csum, sctx->csum_size))
a2de733c
AJ		 1713 fail = 1;
1714
a2de733c
AJ		 1715 return fail;
1716}
1717
b5d67f64 1718static int scrub_checksum_tree_block(struct scrub_block *sblock)
a2de733c 1719{
d9d181c1 1720 struct scrub_ctx *sctx = sblock->sctx;
a2de733c 1721 struct btrfs_header *h;
a36cf8b8 1722 struct btrfs_root *root = sctx->dev_root;
a2de733c 1723 struct btrfs_fs_info *fs_info = root->fs_info;
b5d67f64
SB		 1724 u8 calculated_csum[BTRFS_CSUM_SIZE];
1725 u8 on_disk_csum[BTRFS_CSUM_SIZE];
1726 struct page *page;
1727 void *mapped_buffer;
1728 u64 mapped_size;
1729 void *p;
a2de733c
AJ		 1730 u32 crc = ~(u32)0;
1731 int fail = 0;
1732 int crc_fail = 0;
b5d67f64
SB		 1733 u64 len;
1734 int index;
1735
1736 BUG_ON(sblock->page_count < 1);
7a9e9987 1737 page = sblock->pagev[0]->page;
9613bebb 1738 mapped_buffer = kmap_atomic(page);
b5d67f64 1739 h = (struct btrfs_header *)mapped_buffer;
d9d181c1 1740 memcpy(on_disk_csum, h->csum, sctx->csum_size);
a2de733c
AJ		 1741
1742 /*
1743 * we don't use the getter functions here, as we
1744 * a) don't have an extent buffer and
1745 * b) the page is already kmapped
1746 */
a2de733c 1747
3cae210f 1748 if (sblock->pagev[0]->logical != btrfs_stack_header_bytenr(h))
a2de733c
AJ		 1749 ++fail;
1750
3cae210f 1751 if (sblock->pagev[0]->generation != btrfs_stack_header_generation(h))
a2de733c
AJ		 1752 ++fail;
1753
1754 if (memcmp(h->fsid, fs_info->fsid, BTRFS_UUID_SIZE))
1755 ++fail;
1756
1757 if (memcmp(h->chunk_tree_uuid, fs_info->chunk_tree_uuid,
1758 BTRFS_UUID_SIZE))
1759 ++fail;
1760
ff023aac 1761 WARN_ON(sctx->nodesize != sctx->leafsize);
d9d181c1 1762 len = sctx->nodesize - BTRFS_CSUM_SIZE;
b5d67f64
SB		 1763 mapped_size = PAGE_SIZE - BTRFS_CSUM_SIZE;
1764 p = ((u8 *)mapped_buffer) + BTRFS_CSUM_SIZE;
1765 index = 0;
1766 for (;;) {
1767 u64 l = min_t(u64, len, mapped_size);
1768
b0496686 1769 crc = btrfs_csum_data(p, crc, l);
9613bebb 1770 kunmap_atomic(mapped_buffer);
b5d67f64
SB		 1771 len -= l;
1772 if (len == 0)
1773 break;
1774 index++;
1775 BUG_ON(index >= sblock->page_count);
7a9e9987
SB		 1776 BUG_ON(!sblock->pagev[index]->page);
1777 page = sblock->pagev[index]->page;
9613bebb 1778 mapped_buffer = kmap_atomic(page);
b5d67f64
SB		 1779 mapped_size = PAGE_SIZE;
1780 p = mapped_buffer;
1781 }
1782
1783 btrfs_csum_final(crc, calculated_csum);
d9d181c1 1784 if (memcmp(calculated_csum, on_disk_csum, sctx->csum_size))
a2de733c
AJ		 1785 ++crc_fail;
1786
a2de733c
AJ		 1787 return fail || crc_fail;
1788}
1789
b5d67f64 1790static int scrub_checksum_super(struct scrub_block *sblock)
a2de733c
AJ		 1791{
1792 struct btrfs_super_block *s;
d9d181c1 1793 struct scrub_ctx *sctx = sblock->sctx;
a36cf8b8 1794 struct btrfs_root *root = sctx->dev_root;
a2de733c 1795 struct btrfs_fs_info *fs_info = root->fs_info;
b5d67f64
SB		 1796 u8 calculated_csum[BTRFS_CSUM_SIZE];
1797 u8 on_disk_csum[BTRFS_CSUM_SIZE];
1798 struct page *page;
1799 void *mapped_buffer;
1800 u64 mapped_size;
1801 void *p;
a2de733c 1802 u32 crc = ~(u32)0;
442a4f63
SB		 1803 int fail_gen = 0;
1804 int fail_cor = 0;
b5d67f64
SB		 1805 u64 len;
1806 int index;
a2de733c 1807
b5d67f64 1808 BUG_ON(sblock->page_count < 1);
7a9e9987 1809 page = sblock->pagev[0]->page;
9613bebb 1810 mapped_buffer = kmap_atomic(page);
b5d67f64 1811 s = (struct btrfs_super_block *)mapped_buffer;
d9d181c1 1812 memcpy(on_disk_csum, s->csum, sctx->csum_size);
a2de733c 1813
3cae210f 1814 if (sblock->pagev[0]->logical != btrfs_super_bytenr(s))
442a4f63 1815 ++fail_cor;
a2de733c 1816
3cae210f 1817 if (sblock->pagev[0]->generation != btrfs_super_generation(s))
442a4f63 1818 ++fail_gen;
a2de733c
AJ		 1819
1820 if (memcmp(s->fsid, fs_info->fsid, BTRFS_UUID_SIZE))
442a4f63 1821 ++fail_cor;
a2de733c 1822
b5d67f64
SB		 1823 len = BTRFS_SUPER_INFO_SIZE - BTRFS_CSUM_SIZE;
1824 mapped_size = PAGE_SIZE - BTRFS_CSUM_SIZE;
1825 p = ((u8 *)mapped_buffer) + BTRFS_CSUM_SIZE;
1826 index = 0;
1827 for (;;) {
1828 u64 l = min_t(u64, len, mapped_size);
1829
b0496686 1830 crc = btrfs_csum_data(p, crc, l);
9613bebb 1831 kunmap_atomic(mapped_buffer);
b5d67f64
SB		 1832 len -= l;
1833 if (len == 0)
1834 break;
1835 index++;
1836 BUG_ON(index >= sblock->page_count);
7a9e9987
SB		 1837 BUG_ON(!sblock->pagev[index]->page);
1838 page = sblock->pagev[index]->page;
9613bebb 1839 mapped_buffer = kmap_atomic(page);
b5d67f64
SB		 1840 mapped_size = PAGE_SIZE;
1841 p = mapped_buffer;
1842 }
1843
1844 btrfs_csum_final(crc, calculated_csum);
d9d181c1 1845 if (memcmp(calculated_csum, on_disk_csum, sctx->csum_size))
442a4f63 1846 ++fail_cor;
a2de733c 1847
442a4f63 1848 if (fail_cor + fail_gen) {
a2de733c
AJ		 1849 /*
1850 * if we find an error in a super block, we just report it.
1851 * They will get written with the next transaction commit
1852 * anyway
1853 */
d9d181c1
SB		 1854 spin_lock(&sctx->stat_lock);
1855 ++sctx->stat.super_errors;
1856 spin_unlock(&sctx->stat_lock);
442a4f63 1857 if (fail_cor)
7a9e9987 1858 btrfs_dev_stat_inc_and_print(sblock->pagev[0]->dev,
442a4f63
SB		 1859 BTRFS_DEV_STAT_CORRUPTION_ERRS);
1860 else
7a9e9987 1861 btrfs_dev_stat_inc_and_print(sblock->pagev[0]->dev,
442a4f63 1862 BTRFS_DEV_STAT_GENERATION_ERRS);
a2de733c
AJ		 1863 }
1864
442a4f63 1865 return fail_cor + fail_gen;
a2de733c
AJ		 1866}
1867
b5d67f64
SB		 1868static void scrub_block_get(struct scrub_block *sblock)
1869{
1870 atomic_inc(&sblock->ref_count);
1871}
1872
1873static void scrub_block_put(struct scrub_block *sblock)
1874{
1875 if (atomic_dec_and_test(&sblock->ref_count)) {
1876 int i;
1877
1878 for (i = 0; i < sblock->page_count; i++)
7a9e9987 1879 scrub_page_put(sblock->pagev[i]);
b5d67f64
SB		 1880 kfree(sblock);
1881 }
1882}
1883
7a9e9987
SB		 1884static void scrub_page_get(struct scrub_page *spage)
1885{
1886 atomic_inc(&spage->ref_count);
1887}
1888
1889static void scrub_page_put(struct scrub_page *spage)
1890{
1891 if (atomic_dec_and_test(&spage->ref_count)) {
1892 if (spage->page)
1893 __free_page(spage->page);
1894 kfree(spage);
1895 }
1896}
1897
d9d181c1 1898static void scrub_submit(struct scrub_ctx *sctx)
a2de733c
AJ		 1899{
1900 struct scrub_bio *sbio;
1901
d9d181c1 1902 if (sctx->curr == -1)
1623edeb 1903 return;
a2de733c 1904
d9d181c1
SB		 1905 sbio = sctx->bios[sctx->curr];
1906 sctx->curr = -1;
b6bfebc1 1907 scrub_pending_bio_inc(sctx);
a2de733c 1908
ff023aac
SB		 1909 if (!sbio->bio->bi_bdev) {
1910 /*
1911 * this case should not happen. If btrfs_map_block() is
1912 * wrong, it could happen for dev-replace operations on
1913 * missing devices when no mirrors are available, but in
1914 * this case it should already fail the mount.
1915 * This case is handled correctly (but _very_ slowly).
1916 */
1917 printk_ratelimited(KERN_WARNING
efe120a0 1918 "BTRFS: scrub_submit(bio bdev == NULL) is unexpected!\n");
ff023aac
SB		 1919 bio_endio(sbio->bio, -EIO);
1920 } else {
1921 btrfsic_submit_bio(READ, sbio->bio);
1922 }
a2de733c
AJ		 1923}
1924
ff023aac
SB		 1925static int scrub_add_page_to_rd_bio(struct scrub_ctx *sctx,
1926 struct scrub_page *spage)
a2de733c 1927{
b5d67f64 1928 struct scrub_block *sblock = spage->sblock;
a2de733c 1929 struct scrub_bio *sbio;
69f4cb52 1930 int ret;
a2de733c
AJ		 1931
1932again:
1933 /*
1934 * grab a fresh bio or wait for one to become available
1935 */
d9d181c1
SB		 1936 while (sctx->curr == -1) {
1937 spin_lock(&sctx->list_lock);
1938 sctx->curr = sctx->first_free;
1939 if (sctx->curr != -1) {
1940 sctx->first_free = sctx->bios[sctx->curr]->next_free;
1941 sctx->bios[sctx->curr]->next_free = -1;
1942 sctx->bios[sctx->curr]->page_count = 0;
1943 spin_unlock(&sctx->list_lock);
a2de733c 1944 } else {
d9d181c1
SB		 1945 spin_unlock(&sctx->list_lock);
1946 wait_event(sctx->list_wait, sctx->first_free != -1);
a2de733c
AJ		 1947 }
1948 }
d9d181c1 1949 sbio = sctx->bios[sctx->curr];
b5d67f64 1950 if (sbio->page_count == 0) {
69f4cb52
AJ		 1951 struct bio *bio;
1952
b5d67f64
SB		 1953 sbio->physical = spage->physical;
1954 sbio->logical = spage->logical;
a36cf8b8 1955 sbio->dev = spage->dev;
b5d67f64
SB		 1956 bio = sbio->bio;
1957 if (!bio) {
9be3395b 1958 bio = btrfs_io_bio_alloc(GFP_NOFS, sctx->pages_per_rd_bio);
b5d67f64
SB		 1959 if (!bio)
1960 return -ENOMEM;
1961 sbio->bio = bio;
1962 }
69f4cb52
AJ		 1963
1964 bio->bi_private = sbio;
1965 bio->bi_end_io = scrub_bio_end_io;
a36cf8b8 1966 bio->bi_bdev = sbio->dev->bdev;
4f024f37 1967 bio->bi_iter.bi_sector = sbio->physical >> 9;
69f4cb52 1968 sbio->err = 0;
b5d67f64
SB		 1969 } else if (sbio->physical + sbio->page_count * PAGE_SIZE !=
1970 spage->physical ||
1971 sbio->logical + sbio->page_count * PAGE_SIZE !=
a36cf8b8
SB		 1972 spage->logical ||
1973 sbio->dev != spage->dev) {
d9d181c1 1974 scrub_submit(sctx);
a2de733c
AJ		 1975 goto again;
1976 }
69f4cb52 1977
b5d67f64
SB		 1978 sbio->pagev[sbio->page_count] = spage;
1979 ret = bio_add_page(sbio->bio, spage->page, PAGE_SIZE, 0);
1980 if (ret != PAGE_SIZE) {
1981 if (sbio->page_count < 1) {
1982 bio_put(sbio->bio);
1983 sbio->bio = NULL;
1984 return -EIO;
1985 }
d9d181c1 1986 scrub_submit(sctx);
69f4cb52
AJ		 1987 goto again;
1988 }
1989
ff023aac 1990 scrub_block_get(sblock); /* one for the page added to the bio */
b5d67f64
SB		 1991 atomic_inc(&sblock->outstanding_pages);
1992 sbio->page_count++;
ff023aac 1993 if (sbio->page_count == sctx->pages_per_rd_bio)
d9d181c1 1994 scrub_submit(sctx);
b5d67f64
SB		 1995
1996 return 0;
1997}
1998
d9d181c1 1999static int scrub_pages(struct scrub_ctx *sctx, u64 logical, u64 len,
a36cf8b8 2000 u64 physical, struct btrfs_device *dev, u64 flags,
ff023aac
SB		 2001 u64 gen, int mirror_num, u8 *csum, int force,
2002 u64 physical_for_dev_replace)
b5d67f64
SB		 2003{
2004 struct scrub_block *sblock;
2005 int index;
2006
2007 sblock = kzalloc(sizeof(*sblock), GFP_NOFS);
2008 if (!sblock) {
d9d181c1
SB		 2009 spin_lock(&sctx->stat_lock);
2010 sctx->stat.malloc_errors++;
2011 spin_unlock(&sctx->stat_lock);
b5d67f64 2012 return -ENOMEM;
a2de733c 2013 }
b5d67f64 2014
7a9e9987
SB		 2015 /* one ref inside this function, plus one for each page added to
2016 * a bio later on */
b5d67f64 2017 atomic_set(&sblock->ref_count, 1);
d9d181c1 2018 sblock->sctx = sctx;
b5d67f64
SB		 2019 sblock->no_io_error_seen = 1;
2020
2021 for (index = 0; len > 0; index++) {
7a9e9987 2022 struct scrub_page *spage;
b5d67f64
SB		 2023 u64 l = min_t(u64, len, PAGE_SIZE);
2024
7a9e9987
SB		 2025 spage = kzalloc(sizeof(*spage), GFP_NOFS);
2026 if (!spage) {
2027leave_nomem:
d9d181c1
SB		 2028 spin_lock(&sctx->stat_lock);
2029 sctx->stat.malloc_errors++;
2030 spin_unlock(&sctx->stat_lock);
7a9e9987 2031 scrub_block_put(sblock);
b5d67f64
SB		 2032 return -ENOMEM;
2033 }
7a9e9987
SB		 2034 BUG_ON(index >= SCRUB_MAX_PAGES_PER_BLOCK);
2035 scrub_page_get(spage);
2036 sblock->pagev[index] = spage;
b5d67f64 2037 spage->sblock = sblock;
a36cf8b8 2038 spage->dev = dev;
b5d67f64
SB		 2039 spage->flags = flags;
2040 spage->generation = gen;
2041 spage->logical = logical;
2042 spage->physical = physical;
ff023aac 2043 spage->physical_for_dev_replace = physical_for_dev_replace;
b5d67f64
SB		 2044 spage->mirror_num = mirror_num;
2045 if (csum) {
2046 spage->have_csum = 1;
d9d181c1 2047 memcpy(spage->csum, csum, sctx->csum_size);
b5d67f64
SB		 2048 } else {
2049 spage->have_csum = 0;
2050 }
2051 sblock->page_count++;
7a9e9987
SB		 2052 spage->page = alloc_page(GFP_NOFS);
2053 if (!spage->page)
2054 goto leave_nomem;
b5d67f64
SB		 2055 len -= l;
2056 logical += l;
2057 physical += l;
ff023aac 2058 physical_for_dev_replace += l;
b5d67f64
SB		 2059 }
2060
7a9e9987 2061 WARN_ON(sblock->page_count == 0);
b5d67f64 2062 for (index = 0; index < sblock->page_count; index++) {
7a9e9987 2063 struct scrub_page *spage = sblock->pagev[index];
1bc87793
AJ		 2064 int ret;
2065
ff023aac 2066 ret = scrub_add_page_to_rd_bio(sctx, spage);
b5d67f64
SB		 2067 if (ret) {
2068 scrub_block_put(sblock);
1bc87793 2069 return ret;
b5d67f64 2070 }
1bc87793 2071 }
a2de733c 2072
b5d67f64 2073 if (force)
d9d181c1 2074 scrub_submit(sctx);
a2de733c 2075
b5d67f64
SB		 2076 /* last one frees, either here or in bio completion for last page */
2077 scrub_block_put(sblock);
a2de733c
AJ		 2078 return 0;
2079}
2080
b5d67f64
SB		 2081static void scrub_bio_end_io(struct bio *bio, int err)
2082{
2083 struct scrub_bio *sbio = bio->bi_private;
a36cf8b8 2084 struct btrfs_fs_info *fs_info = sbio->dev->dev_root->fs_info;
b5d67f64
SB		 2085
2086 sbio->err = err;
2087 sbio->bio = bio;
2088
0339ef2f 2089 btrfs_queue_work(fs_info->scrub_workers, &sbio->work);
b5d67f64
SB		 2090}
2091
2092static void scrub_bio_end_io_worker(struct btrfs_work *work)
2093{
2094 struct scrub_bio *sbio = container_of(work, struct scrub_bio, work);
d9d181c1 2095 struct scrub_ctx *sctx = sbio->sctx;
b5d67f64
SB		 2096 int i;
2097
ff023aac 2098 BUG_ON(sbio->page_count > SCRUB_PAGES_PER_RD_BIO);
b5d67f64
SB		 2099 if (sbio->err) {
2100 for (i = 0; i < sbio->page_count; i++) {
2101 struct scrub_page *spage = sbio->pagev[i];
2102
2103 spage->io_error = 1;
2104 spage->sblock->no_io_error_seen = 0;
2105 }
2106 }
2107
2108 /* now complete the scrub_block items that have all pages completed */
2109 for (i = 0; i < sbio->page_count; i++) {
2110 struct scrub_page *spage = sbio->pagev[i];
2111 struct scrub_block *sblock = spage->sblock;
2112
2113 if (atomic_dec_and_test(&sblock->outstanding_pages))
2114 scrub_block_complete(sblock);
2115 scrub_block_put(sblock);
2116 }
2117
b5d67f64
SB		 2118 bio_put(sbio->bio);
2119 sbio->bio = NULL;
d9d181c1
SB		 2120 spin_lock(&sctx->list_lock);
2121 sbio->next_free = sctx->first_free;
2122 sctx->first_free = sbio->index;
2123 spin_unlock(&sctx->list_lock);
ff023aac
SB		 2124
2125 if (sctx->is_dev_replace &&
2126 atomic_read(&sctx->wr_ctx.flush_all_writes)) {
2127 mutex_lock(&sctx->wr_ctx.wr_lock);
2128 scrub_wr_submit(sctx);
2129 mutex_unlock(&sctx->wr_ctx.wr_lock);
2130 }
2131
b6bfebc1 2132 scrub_pending_bio_dec(sctx);
b5d67f64
SB		 2133}
2134
2135static void scrub_block_complete(struct scrub_block *sblock)
2136{
ff023aac 2137 if (!sblock->no_io_error_seen) {
b5d67f64 2138 scrub_handle_errored_block(sblock);
ff023aac
SB		 2139 } else {
2140 /*
2141 * if has checksum error, write via repair mechanism in
2142 * dev replace case, otherwise write here in dev replace
2143 * case.
2144 */
2145 if (!scrub_checksum(sblock) && sblock->sctx->is_dev_replace)
2146 scrub_write_block_to_dev_replace(sblock);
2147 }
b5d67f64
SB		 2148}
2149
d9d181c1 2150static int scrub_find_csum(struct scrub_ctx *sctx, u64 logical, u64 len,
a2de733c
AJ		 2151 u8 *csum)
2152{
2153 struct btrfs_ordered_sum *sum = NULL;
f51a4a18 2154 unsigned long index;
a2de733c 2155 unsigned long num_sectors;
a2de733c 2156
d9d181c1
SB		 2157 while (!list_empty(&sctx->csum_list)) {
2158 sum = list_first_entry(&sctx->csum_list,
a2de733c
AJ		 2159 struct btrfs_ordered_sum, list);
2160 if (sum->bytenr > logical)
2161 return 0;
2162 if (sum->bytenr + sum->len > logical)
2163 break;
2164
d9d181c1 2165 ++sctx->stat.csum_discards;
a2de733c
AJ		 2166 list_del(&sum->list);
2167 kfree(sum);
2168 sum = NULL;
2169 }
2170 if (!sum)
2171 return 0;
2172
f51a4a18 2173 index = ((u32)(logical - sum->bytenr)) / sctx->sectorsize;
d9d181c1 2174 num_sectors = sum->len / sctx->sectorsize;
f51a4a18
MX		 2175 memcpy(csum, sum->sums + index, sctx->csum_size);
2176 if (index == num_sectors - 1) {
a2de733c
AJ		 2177 list_del(&sum->list);
2178 kfree(sum);
2179 }
f51a4a18 2180 return 1;
a2de733c
AJ		 2181}
2182
2183/* scrub extent tries to collect up to 64 kB for each bio */
d9d181c1 2184static int scrub_extent(struct scrub_ctx *sctx, u64 logical, u64 len,
a36cf8b8 2185 u64 physical, struct btrfs_device *dev, u64 flags,
ff023aac 2186 u64 gen, int mirror_num, u64 physical_for_dev_replace)
a2de733c
AJ		 2187{
2188 int ret;
2189 u8 csum[BTRFS_CSUM_SIZE];
b5d67f64
SB		 2190 u32 blocksize;
2191
2192 if (flags & BTRFS_EXTENT_FLAG_DATA) {
d9d181c1
SB		 2193 blocksize = sctx->sectorsize;
2194 spin_lock(&sctx->stat_lock);
2195 sctx->stat.data_extents_scrubbed++;
2196 sctx->stat.data_bytes_scrubbed += len;
2197 spin_unlock(&sctx->stat_lock);
b5d67f64 2198 } else if (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK) {
ff023aac 2199 WARN_ON(sctx->nodesize != sctx->leafsize);
d9d181c1
SB		 2200 blocksize = sctx->nodesize;
2201 spin_lock(&sctx->stat_lock);
2202 sctx->stat.tree_extents_scrubbed++;
2203 sctx->stat.tree_bytes_scrubbed += len;
2204 spin_unlock(&sctx->stat_lock);
b5d67f64 2205 } else {
d9d181c1 2206 blocksize = sctx->sectorsize;
ff023aac 2207 WARN_ON(1);
b5d67f64 2208 }
a2de733c
AJ		 2209
2210 while (len) {
b5d67f64 2211 u64 l = min_t(u64, len, blocksize);
a2de733c
AJ		 2212 int have_csum = 0;
2213
2214 if (flags & BTRFS_EXTENT_FLAG_DATA) {
2215 /* push csums to sbio */
d9d181c1 2216 have_csum = scrub_find_csum(sctx, logical, l, csum);
a2de733c 2217 if (have_csum == 0)
d9d181c1 2218 ++sctx->stat.no_csum;
ff023aac
SB		 2219 if (sctx->is_dev_replace && !have_csum) {
2220 ret = copy_nocow_pages(sctx, logical, l,
2221 mirror_num,
2222 physical_for_dev_replace);
2223 goto behind_scrub_pages;
2224 }
a2de733c 2225 }
a36cf8b8 2226 ret = scrub_pages(sctx, logical, l, physical, dev, flags, gen,
ff023aac
SB		 2227 mirror_num, have_csum ? csum : NULL, 0,
2228 physical_for_dev_replace);
2229behind_scrub_pages:
a2de733c
AJ		 2230 if (ret)
2231 return ret;
2232 len -= l;
2233 logical += l;
2234 physical += l;
ff023aac 2235 physical_for_dev_replace += l;
a2de733c
AJ		 2236 }
2237 return 0;
2238}
2239
3b080b25
WS		 2240/*
2241 * Given a physical address, this will calculate it's
2242 * logical offset. if this is a parity stripe, it will return
2243 * the most left data stripe's logical offset.
2244 *
2245 * return 0 if it is a data stripe, 1 means parity stripe.
2246 */
2247static int get_raid56_logic_offset(u64 physical, int num,
2248 struct map_lookup *map, u64 *offset)
2249{
2250 int i;
2251 int j = 0;
2252 u64 stripe_nr;
2253 u64 last_offset;
2254 int stripe_index;
2255 int rot;
2256
2257 last_offset = (physical - map->stripes[num].physical) *
2258 nr_data_stripes(map);
2259 *offset = last_offset;
2260 for (i = 0; i < nr_data_stripes(map); i++) {
2261 *offset = last_offset + i * map->stripe_len;
2262
2263 stripe_nr = *offset;
2264 do_div(stripe_nr, map->stripe_len);
2265 do_div(stripe_nr, nr_data_stripes(map));
2266
2267 /* Work out the disk rotation on this stripe-set */
2268 rot = do_div(stripe_nr, map->num_stripes);
2269 /* calculate which stripe this data locates */
2270 rot += i;
e4fbaee2 2271 stripe_index = rot % map->num_stripes;
3b080b25
WS		 2272 if (stripe_index == num)
2273 return 0;
2274 if (stripe_index < num)
2275 j++;
2276 }
2277 *offset = last_offset + j * map->stripe_len;
2278 return 1;
2279}
2280
d9d181c1 2281static noinline_for_stack int scrub_stripe(struct scrub_ctx *sctx,
a36cf8b8
SB		 2282 struct map_lookup *map,
2283 struct btrfs_device *scrub_dev,
ff023aac
SB		 2284 int num, u64 base, u64 length,
2285 int is_dev_replace)
a2de733c
AJ		 2286{
2287 struct btrfs_path *path;
a36cf8b8 2288 struct btrfs_fs_info *fs_info = sctx->dev_root->fs_info;
a2de733c
AJ		 2289 struct btrfs_root *root = fs_info->extent_root;
2290 struct btrfs_root *csum_root = fs_info->csum_root;
2291 struct btrfs_extent_item *extent;
e7786c3a 2292 struct blk_plug plug;
a2de733c
AJ		 2293 u64 flags;
2294 int ret;
2295 int slot;
a2de733c 2296 u64 nstripes;
a2de733c
AJ		 2297 struct extent_buffer *l;
2298 struct btrfs_key key;
2299 u64 physical;
2300 u64 logical;
625f1c8d 2301 u64 logic_end;
3b080b25 2302 u64 physical_end;
a2de733c 2303 u64 generation;
e12fa9cd 2304 int mirror_num;
7a26285e
AJ		 2305 struct reada_control *reada1;
2306 struct reada_control *reada2;
2307 struct btrfs_key key_start;
2308 struct btrfs_key key_end;
a2de733c
AJ		 2309 u64 increment = map->stripe_len;
2310 u64 offset;
ff023aac
SB		 2311 u64 extent_logical;
2312 u64 extent_physical;
2313 u64 extent_len;
2314 struct btrfs_device *extent_dev;
2315 int extent_mirror_num;
3b080b25 2316 int stop_loop = 0;
53b381b3 2317
a2de733c 2318 nstripes = length;
3b080b25 2319 physical = map->stripes[num].physical;
a2de733c
AJ		 2320 offset = 0;
2321 do_div(nstripes, map->stripe_len);
2322 if (map->type & BTRFS_BLOCK_GROUP_RAID0) {
2323 offset = map->stripe_len * num;
2324 increment = map->stripe_len * map->num_stripes;
193ea74b 2325 mirror_num = 1;
a2de733c
AJ		 2326 } else if (map->type & BTRFS_BLOCK_GROUP_RAID10) {
2327 int factor = map->num_stripes / map->sub_stripes;
2328 offset = map->stripe_len * (num / map->sub_stripes);
2329 increment = map->stripe_len * factor;
193ea74b 2330 mirror_num = num % map->sub_stripes + 1;
a2de733c
AJ		 2331 } else if (map->type & BTRFS_BLOCK_GROUP_RAID1) {
2332 increment = map->stripe_len;
193ea74b 2333 mirror_num = num % map->num_stripes + 1;
a2de733c
AJ		 2334 } else if (map->type & BTRFS_BLOCK_GROUP_DUP) {
2335 increment = map->stripe_len;
193ea74b 2336 mirror_num = num % map->num_stripes + 1;
3b080b25
WS		 2337 } else if (map->type & (BTRFS_BLOCK_GROUP_RAID5 |
2338 BTRFS_BLOCK_GROUP_RAID6)) {
2339 get_raid56_logic_offset(physical, num, map, &offset);
2340 increment = map->stripe_len * nr_data_stripes(map);
2341 mirror_num = 1;
a2de733c
AJ		 2342 } else {
2343 increment = map->stripe_len;
193ea74b 2344 mirror_num = 1;
a2de733c
AJ		 2345 }
2346
2347 path = btrfs_alloc_path();
2348 if (!path)
2349 return -ENOMEM;
2350
b5d67f64
SB		 2351 /*
2352 * work on commit root. The related disk blocks are static as
2353 * long as COW is applied. This means, it is save to rewrite
2354 * them to repair disk errors without any race conditions
2355 */
a2de733c
AJ		 2356 path->search_commit_root = 1;
2357 path->skip_locking = 1;
2358
2359 /*
7a26285e
AJ		 2360 * trigger the readahead for extent tree csum tree and wait for
2361 * completion. During readahead, the scrub is officially paused
2362 * to not hold off transaction commits
a2de733c
AJ		 2363 */
2364 logical = base + offset;
3b080b25
WS		 2365 physical_end = physical + nstripes * map->stripe_len;
2366 if (map->type & (BTRFS_BLOCK_GROUP_RAID5 |
2367 BTRFS_BLOCK_GROUP_RAID6)) {
2368 get_raid56_logic_offset(physical_end, num,
2369 map, &logic_end);
2370 logic_end += base;
2371 } else {
2372 logic_end = logical + increment * nstripes;
2373 }
d9d181c1 2374 wait_event(sctx->list_wait,
b6bfebc1 2375 atomic_read(&sctx->bios_in_flight) == 0);
cb7ab021 2376 scrub_blocked_if_needed(fs_info);
7a26285e
AJ		 2377
2378 /* FIXME it might be better to start readahead at commit root */
2379 key_start.objectid = logical;
2380 key_start.type = BTRFS_EXTENT_ITEM_KEY;
2381 key_start.offset = (u64)0;
3b080b25 2382 key_end.objectid = logic_end;
3173a18f
JB		 2383 key_end.type = BTRFS_METADATA_ITEM_KEY;
2384 key_end.offset = (u64)-1;
7a26285e
AJ		 2385 reada1 = btrfs_reada_add(root, &key_start, &key_end);
2386
2387 key_start.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
2388 key_start.type = BTRFS_EXTENT_CSUM_KEY;
2389 key_start.offset = logical;
2390 key_end.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
2391 key_end.type = BTRFS_EXTENT_CSUM_KEY;
3b080b25 2392 key_end.offset = logic_end;
7a26285e
AJ		 2393 reada2 = btrfs_reada_add(csum_root, &key_start, &key_end);
2394
2395 if (!IS_ERR(reada1))
2396 btrfs_reada_wait(reada1);
2397 if (!IS_ERR(reada2))
2398 btrfs_reada_wait(reada2);
2399
a2de733c
AJ		 2400
2401 /*
2402 * collect all data csums for the stripe to avoid seeking during
2403 * the scrub. This might currently (crc32) end up to be about 1MB
2404 */
e7786c3a 2405 blk_start_plug(&plug);
a2de733c 2406
a2de733c
AJ		 2407 /*
2408 * now find all extents for each stripe and scrub them
2409 */
a2de733c 2410 ret = 0;
3b080b25
WS		 2411 while (physical < physical_end) {
2412 /* for raid56, we skip parity stripe */
2413 if (map->type & (BTRFS_BLOCK_GROUP_RAID5 |
2414 BTRFS_BLOCK_GROUP_RAID6)) {
2415 ret = get_raid56_logic_offset(physical, num,
2416 map, &logical);
2417 logical += base;
2418 if (ret)
2419 goto skip;
2420 }
a2de733c
AJ		 2421 /*
2422 * canceled?
2423 */
2424 if (atomic_read(&fs_info->scrub_cancel_req) ||
d9d181c1 2425 atomic_read(&sctx->cancel_req)) {
a2de733c
AJ		 2426 ret = -ECANCELED;
2427 goto out;
2428 }
2429 /*
2430 * check to see if we have to pause
2431 */
2432 if (atomic_read(&fs_info->scrub_pause_req)) {
2433 /* push queued extents */
ff023aac 2434 atomic_set(&sctx->wr_ctx.flush_all_writes, 1);
d9d181c1 2435 scrub_submit(sctx);
ff023aac
SB		 2436 mutex_lock(&sctx->wr_ctx.wr_lock);
2437 scrub_wr_submit(sctx);
2438 mutex_unlock(&sctx->wr_ctx.wr_lock);
d9d181c1 2439 wait_event(sctx->list_wait,
b6bfebc1 2440 atomic_read(&sctx->bios_in_flight) == 0);
ff023aac 2441 atomic_set(&sctx->wr_ctx.flush_all_writes, 0);
3cb0929a 2442 scrub_blocked_if_needed(fs_info);
a2de733c
AJ		 2443 }
2444
7c76edb7
WS		 2445 if (btrfs_fs_incompat(fs_info, SKINNY_METADATA))
2446 key.type = BTRFS_METADATA_ITEM_KEY;
2447 else
2448 key.type = BTRFS_EXTENT_ITEM_KEY;
a2de733c 2449 key.objectid = logical;
625f1c8d 2450 key.offset = (u64)-1;
a2de733c
AJ		 2451
2452 ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2453 if (ret < 0)
2454 goto out;
3173a18f 2455
8c51032f 2456 if (ret > 0) {
ade2e0b3 2457 ret = btrfs_previous_extent_item(root, path, 0);
a2de733c
AJ		 2458 if (ret < 0)
2459 goto out;
8c51032f
AJ		 2460 if (ret > 0) {
2461 /* there's no smaller item, so stick with the
2462 * larger one */
2463 btrfs_release_path(path);
2464 ret = btrfs_search_slot(NULL, root, &key,
2465 path, 0, 0);
2466 if (ret < 0)
2467 goto out;
2468 }
a2de733c
AJ		 2469 }
2470
625f1c8d 2471 stop_loop = 0;
a2de733c 2472 while (1) {
3173a18f
JB		 2473 u64 bytes;
2474
a2de733c
AJ		 2475 l = path->nodes[0];
2476 slot = path->slots[0];
2477 if (slot >= btrfs_header_nritems(l)) {
2478 ret = btrfs_next_leaf(root, path);
2479 if (ret == 0)
2480 continue;
2481 if (ret < 0)
2482 goto out;
2483
625f1c8d 2484 stop_loop = 1;
a2de733c
AJ		 2485 break;
2486 }
2487 btrfs_item_key_to_cpu(l, &key, slot);
2488
3173a18f
JB		 2489 if (key.type == BTRFS_METADATA_ITEM_KEY)
2490 bytes = root->leafsize;
2491 else
2492 bytes = key.offset;
2493
2494 if (key.objectid + bytes <= logical)
a2de733c
AJ		 2495 goto next;
2496
625f1c8d
LB		 2497 if (key.type != BTRFS_EXTENT_ITEM_KEY &&
2498 key.type != BTRFS_METADATA_ITEM_KEY)
2499 goto next;
a2de733c 2500
625f1c8d
LB		 2501 if (key.objectid >= logical + map->stripe_len) {
2502 /* out of this device extent */
2503 if (key.objectid >= logic_end)
2504 stop_loop = 1;
2505 break;
2506 }
a2de733c
AJ		 2507
2508 extent = btrfs_item_ptr(l, slot,
2509 struct btrfs_extent_item);
2510 flags = btrfs_extent_flags(l, extent);
2511 generation = btrfs_extent_generation(l, extent);
2512
2513 if (key.objectid < logical &&
2514 (flags & BTRFS_EXTENT_FLAG_TREE_BLOCK)) {
efe120a0
FH		 2515 btrfs_err(fs_info,
2516 "scrub: tree block %llu spanning "
2517 "stripes, ignored. logical=%llu",
c1c9ff7c 2518 key.objectid, logical);
a2de733c
AJ		 2519 goto next;
2520 }
2521
625f1c8d
LB		 2522again:
2523 extent_logical = key.objectid;
2524 extent_len = bytes;
2525
a2de733c
AJ		 2526 /*
2527 * trim extent to this stripe
2528 */
625f1c8d
LB		 2529 if (extent_logical < logical) {
2530 extent_len -= logical - extent_logical;
2531 extent_logical = logical;
a2de733c 2532 }
625f1c8d 2533 if (extent_logical + extent_len >
a2de733c 2534 logical + map->stripe_len) {
625f1c8d
LB		 2535 extent_len = logical + map->stripe_len -
2536 extent_logical;
a2de733c
AJ		 2537 }
2538
625f1c8d 2539 extent_physical = extent_logical - logical + physical;
ff023aac
SB		 2540 extent_dev = scrub_dev;
2541 extent_mirror_num = mirror_num;
2542 if (is_dev_replace)
2543 scrub_remap_extent(fs_info, extent_logical,
2544 extent_len, &extent_physical,
2545 &extent_dev,
2546 &extent_mirror_num);
625f1c8d
LB		 2547
2548 ret = btrfs_lookup_csums_range(csum_root, logical,
2549 logical + map->stripe_len - 1,
2550 &sctx->csum_list, 1);
2551 if (ret)
2552 goto out;
2553
ff023aac
SB		 2554 ret = scrub_extent(sctx, extent_logical, extent_len,
2555 extent_physical, extent_dev, flags,
2556 generation, extent_mirror_num,
115930cb 2557 extent_logical - logical + physical);
a2de733c
AJ		 2558 if (ret)
2559 goto out;
2560
d88d46c6 2561 scrub_free_csums(sctx);
625f1c8d
LB		 2562 if (extent_logical + extent_len <
2563 key.objectid + bytes) {
3b080b25
WS		 2564 if (map->type & (BTRFS_BLOCK_GROUP_RAID5 |
2565 BTRFS_BLOCK_GROUP_RAID6)) {
2566 /*
2567 * loop until we find next data stripe
2568 * or we have finished all stripes.
2569 */
2570 do {
2571 physical += map->stripe_len;
2572 ret = get_raid56_logic_offset(
2573 physical, num,
2574 map, &logical);
2575 logical += base;
2576 } while (physical < physical_end && ret);
2577 } else {
2578 physical += map->stripe_len;
2579 logical += increment;
2580 }
625f1c8d
LB		 2581 if (logical < key.objectid + bytes) {
2582 cond_resched();
2583 goto again;
2584 }
2585
3b080b25 2586 if (physical >= physical_end) {
625f1c8d
LB		 2587 stop_loop = 1;
2588 break;
2589 }
2590 }
a2de733c
AJ		 2591next:
2592 path->slots[0]++;
2593 }
71267333 2594 btrfs_release_path(path);
3b080b25 2595skip:
a2de733c
AJ		 2596 logical += increment;
2597 physical += map->stripe_len;
d9d181c1 2598 spin_lock(&sctx->stat_lock);
625f1c8d
LB		 2599 if (stop_loop)
2600 sctx->stat.last_physical = map->stripes[num].physical +
2601 length;
2602 else
2603 sctx->stat.last_physical = physical;
d9d181c1 2604 spin_unlock(&sctx->stat_lock);
625f1c8d
LB		 2605 if (stop_loop)
2606 break;
a2de733c 2607 }
ff023aac 2608out:
a2de733c 2609 /* push queued extents */
d9d181c1 2610 scrub_submit(sctx);
ff023aac
SB		 2611 mutex_lock(&sctx->wr_ctx.wr_lock);
2612 scrub_wr_submit(sctx);
2613 mutex_unlock(&sctx->wr_ctx.wr_lock);
a2de733c 2614
e7786c3a 2615 blk_finish_plug(&plug);
a2de733c
AJ		 2616 btrfs_free_path(path);
2617 return ret < 0 ? ret : 0;
2618}
2619
d9d181c1 2620static noinline_for_stack int scrub_chunk(struct scrub_ctx *sctx,
a36cf8b8
SB		 2621 struct btrfs_device *scrub_dev,
2622 u64 chunk_tree, u64 chunk_objectid,
2623 u64 chunk_offset, u64 length,
ff023aac 2624 u64 dev_offset, int is_dev_replace)
a2de733c
AJ		 2625{
2626 struct btrfs_mapping_tree *map_tree =
a36cf8b8 2627 &sctx->dev_root->fs_info->mapping_tree;
a2de733c
AJ		 2628 struct map_lookup *map;
2629 struct extent_map *em;
2630 int i;
ff023aac 2631 int ret = 0;
a2de733c
AJ		 2632
2633 read_lock(&map_tree->map_tree.lock);
2634 em = lookup_extent_mapping(&map_tree->map_tree, chunk_offset, 1);
2635 read_unlock(&map_tree->map_tree.lock);
2636
2637 if (!em)
2638 return -EINVAL;
2639
2640 map = (struct map_lookup *)em->bdev;
2641 if (em->start != chunk_offset)
2642 goto out;
2643
2644 if (em->len < length)
2645 goto out;
2646
2647 for (i = 0; i < map->num_stripes; ++i) {
a36cf8b8 2648 if (map->stripes[i].dev->bdev == scrub_dev->bdev &&
859acaf1 2649 map->stripes[i].physical == dev_offset) {
a36cf8b8 2650 ret = scrub_stripe(sctx, map, scrub_dev, i,
ff023aac
SB		 2651 chunk_offset, length,
2652 is_dev_replace);
a2de733c
AJ		 2653 if (ret)
2654 goto out;
2655 }
2656 }
2657out:
2658 free_extent_map(em);
2659
2660 return ret;
2661}
2662
2663static noinline_for_stack
a36cf8b8 2664int scrub_enumerate_chunks(struct scrub_ctx *sctx,
ff023aac
SB		 2665 struct btrfs_device *scrub_dev, u64 start, u64 end,
2666 int is_dev_replace)
a2de733c
AJ		 2667{
2668 struct btrfs_dev_extent *dev_extent = NULL;
2669 struct btrfs_path *path;
a36cf8b8 2670 struct btrfs_root *root = sctx->dev_root;
a2de733c
AJ		 2671 struct btrfs_fs_info *fs_info = root->fs_info;
2672 u64 length;
2673 u64 chunk_tree;
2674 u64 chunk_objectid;
2675 u64 chunk_offset;
2676 int ret;
2677 int slot;
2678 struct extent_buffer *l;
2679 struct btrfs_key key;
2680 struct btrfs_key found_key;
2681 struct btrfs_block_group_cache *cache;
ff023aac 2682 struct btrfs_dev_replace *dev_replace = &fs_info->dev_replace;
a2de733c
AJ		 2683
2684 path = btrfs_alloc_path();
2685 if (!path)
2686 return -ENOMEM;
2687
2688 path->reada = 2;
2689 path->search_commit_root = 1;
2690 path->skip_locking = 1;
2691
a36cf8b8 2692 key.objectid = scrub_dev->devid;
a2de733c
AJ		 2693 key.offset = 0ull;
2694 key.type = BTRFS_DEV_EXTENT_KEY;
2695
a2de733c
AJ		 2696 while (1) {
2697 ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
2698 if (ret < 0)
8c51032f
AJ		 2699 break;
2700 if (ret > 0) {
2701 if (path->slots[0] >=
2702 btrfs_header_nritems(path->nodes[0])) {
2703 ret = btrfs_next_leaf(root, path);
2704 if (ret)
2705 break;
2706 }
2707 }
a2de733c
AJ		 2708
2709 l = path->nodes[0];
2710 slot = path->slots[0];
2711
2712 btrfs_item_key_to_cpu(l, &found_key, slot);
2713
a36cf8b8 2714 if (found_key.objectid != scrub_dev->devid)
a2de733c
AJ		 2715 break;
2716
8c51032f 2717 if (btrfs_key_type(&found_key) != BTRFS_DEV_EXTENT_KEY)
a2de733c
AJ		 2718 break;
2719
2720 if (found_key.offset >= end)
2721 break;
2722
2723 if (found_key.offset < key.offset)
2724 break;
2725
2726 dev_extent = btrfs_item_ptr(l, slot, struct btrfs_dev_extent);
2727 length = btrfs_dev_extent_length(l, dev_extent);
2728
ced96edc
QW		 2729 if (found_key.offset + length <= start)
2730 goto skip;
a2de733c
AJ		 2731
2732 chunk_tree = btrfs_dev_extent_chunk_tree(l, dev_extent);
2733 chunk_objectid = btrfs_dev_extent_chunk_objectid(l, dev_extent);
2734 chunk_offset = btrfs_dev_extent_chunk_offset(l, dev_extent);
2735
2736 /*
2737 * get a reference on the corresponding block group to prevent
2738 * the chunk from going away while we scrub it
2739 */
2740 cache = btrfs_lookup_block_group(fs_info, chunk_offset);
ced96edc
QW		 2741
2742 /* some chunks are removed but not committed to disk yet,
2743 * continue scrubbing */
2744 if (!cache)
2745 goto skip;
2746
ff023aac
SB		 2747 dev_replace->cursor_right = found_key.offset + length;
2748 dev_replace->cursor_left = found_key.offset;
2749 dev_replace->item_needs_writeback = 1;
a36cf8b8 2750 ret = scrub_chunk(sctx, scrub_dev, chunk_tree, chunk_objectid,
ff023aac
SB		 2751 chunk_offset, length, found_key.offset,
2752 is_dev_replace);
2753
2754 /*
2755 * flush, submit all pending read and write bios, afterwards
2756 * wait for them.
2757 * Note that in the dev replace case, a read request causes
2758 * write requests that are submitted in the read completion
2759 * worker. Therefore in the current situation, it is required
2760 * that all write requests are flushed, so that all read and
2761 * write requests are really completed when bios_in_flight
2762 * changes to 0.
2763 */
2764 atomic_set(&sctx->wr_ctx.flush_all_writes, 1);
2765 scrub_submit(sctx);
2766 mutex_lock(&sctx->wr_ctx.wr_lock);
2767 scrub_wr_submit(sctx);
2768 mutex_unlock(&sctx->wr_ctx.wr_lock);
2769
2770 wait_event(sctx->list_wait,
2771 atomic_read(&sctx->bios_in_flight) == 0);
12cf9372
WS		 2772 atomic_inc(&fs_info->scrubs_paused);
2773 wake_up(&fs_info->scrub_pause_wait);
2774
2775 /*
2776 * must be called before we decrease @scrub_paused.
2777 * make sure we don't block transaction commit while
2778 * we are waiting pending workers finished.
2779 */
ff023aac
SB		 2780 wait_event(sctx->list_wait,
2781 atomic_read(&sctx->workers_pending) == 0);
12cf9372
WS		 2782 atomic_set(&sctx->wr_ctx.flush_all_writes, 0);
2783
2784 mutex_lock(&fs_info->scrub_lock);
2785 __scrub_blocked_if_needed(fs_info);
2786 atomic_dec(&fs_info->scrubs_paused);
2787 mutex_unlock(&fs_info->scrub_lock);
2788 wake_up(&fs_info->scrub_pause_wait);
ff023aac 2789
a2de733c
AJ		 2790 btrfs_put_block_group(cache);
2791 if (ret)
2792 break;
af1be4f8
SB		 2793 if (is_dev_replace &&
2794 atomic64_read(&dev_replace->num_write_errors) > 0) {
ff023aac
SB		 2795 ret = -EIO;
2796 break;
2797 }
2798 if (sctx->stat.malloc_errors > 0) {
2799 ret = -ENOMEM;
2800 break;
2801 }
a2de733c 2802
539f358a
ID		 2803 dev_replace->cursor_left = dev_replace->cursor_right;
2804 dev_replace->item_needs_writeback = 1;
ced96edc 2805skip:
a2de733c 2806 key.offset = found_key.offset + length;
71267333 2807 btrfs_release_path(path);
a2de733c
AJ		 2808 }
2809
a2de733c 2810 btrfs_free_path(path);
8c51032f
AJ		 2811
2812 /*
2813 * ret can still be 1 from search_slot or next_leaf,
2814 * that's not an error
2815 */
2816 return ret < 0 ? ret : 0;
a2de733c
AJ		 2817}
2818
a36cf8b8
SB		 2819static noinline_for_stack int scrub_supers(struct scrub_ctx *sctx,
2820 struct btrfs_device *scrub_dev)
a2de733c
AJ		 2821{
2822 int i;
2823 u64 bytenr;
2824 u64 gen;
2825 int ret;
a36cf8b8 2826 struct btrfs_root *root = sctx->dev_root;
a2de733c 2827
87533c47 2828 if (test_bit(BTRFS_FS_STATE_ERROR, &root->fs_info->fs_state))
79787eaa
JM		 2829 return -EIO;
2830
a2de733c
AJ		 2831 gen = root->fs_info->last_trans_committed;
2832
2833 for (i = 0; i < BTRFS_SUPER_MIRROR_MAX; i++) {
2834 bytenr = btrfs_sb_offset(i);
a36cf8b8 2835 if (bytenr + BTRFS_SUPER_INFO_SIZE > scrub_dev->total_bytes)
a2de733c
AJ		 2836 break;
2837
d9d181c1 2838 ret = scrub_pages(sctx, bytenr, BTRFS_SUPER_INFO_SIZE, bytenr,
a36cf8b8 2839 scrub_dev, BTRFS_EXTENT_FLAG_SUPER, gen, i,
ff023aac 2840 NULL, 1, bytenr);
a2de733c
AJ		 2841 if (ret)
2842 return ret;
2843 }
b6bfebc1 2844 wait_event(sctx->list_wait, atomic_read(&sctx->bios_in_flight) == 0);
a2de733c
AJ		 2845
2846 return 0;
2847}
2848
2849/*
2850 * get a reference count on fs_info->scrub_workers. start worker if necessary
2851 */
ff023aac
SB		 2852static noinline_for_stack int scrub_workers_get(struct btrfs_fs_info *fs_info,
2853 int is_dev_replace)
a2de733c 2854{
0dc3b84a 2855 int ret = 0;
0339ef2f
QW		 2856 int flags = WQ_FREEZABLE | WQ_UNBOUND;
2857 int max_active = fs_info->thread_pool_size;
a2de733c 2858
632dd772 2859 if (fs_info->scrub_workers_refcnt == 0) {
ff023aac 2860 if (is_dev_replace)
0339ef2f
QW		 2861 fs_info->scrub_workers =
2862 btrfs_alloc_workqueue("btrfs-scrub", flags,
2863 1, 4);
ff023aac 2864 else
0339ef2f
QW		 2865 fs_info->scrub_workers =
2866 btrfs_alloc_workqueue("btrfs-scrub", flags,
2867 max_active, 4);
2868 if (!fs_info->scrub_workers) {
2869 ret = -ENOMEM;
0dc3b84a 2870 goto out;
0339ef2f
QW		 2871 }
2872 fs_info->scrub_wr_completion_workers =
2873 btrfs_alloc_workqueue("btrfs-scrubwrc", flags,
2874 max_active, 2);
2875 if (!fs_info->scrub_wr_completion_workers) {
2876 ret = -ENOMEM;
ff023aac 2877 goto out;
0339ef2f
QW		 2878 }
2879 fs_info->scrub_nocow_workers =
2880 btrfs_alloc_workqueue("btrfs-scrubnc", flags, 1, 0);
2881 if (!fs_info->scrub_nocow_workers) {
2882 ret = -ENOMEM;
ff023aac 2883 goto out;
0339ef2f 2884 }
632dd772 2885 }
a2de733c 2886 ++fs_info->scrub_workers_refcnt;
0dc3b84a 2887out:
0dc3b84a 2888 return ret;
a2de733c
AJ		 2889}
2890
aa1b8cd4 2891static noinline_for_stack void scrub_workers_put(struct btrfs_fs_info *fs_info)
a2de733c 2892{
ff023aac 2893 if (--fs_info->scrub_workers_refcnt == 0) {
0339ef2f
QW		 2894 btrfs_destroy_workqueue(fs_info->scrub_workers);
2895 btrfs_destroy_workqueue(fs_info->scrub_wr_completion_workers);
2896 btrfs_destroy_workqueue(fs_info->scrub_nocow_workers);
ff023aac 2897 }
a2de733c 2898 WARN_ON(fs_info->scrub_workers_refcnt < 0);
a2de733c
AJ		 2899}
2900
aa1b8cd4
SB		 2901int btrfs_scrub_dev(struct btrfs_fs_info *fs_info, u64 devid, u64 start,
2902 u64 end, struct btrfs_scrub_progress *progress,
63a212ab 2903 int readonly, int is_dev_replace)
a2de733c 2904{
d9d181c1 2905 struct scrub_ctx *sctx;
a2de733c
AJ		 2906 int ret;
2907 struct btrfs_device *dev;
5d68da3b 2908 struct rcu_string *name;
a2de733c 2909
aa1b8cd4 2910 if (btrfs_fs_closing(fs_info))
a2de733c
AJ		 2911 return -EINVAL;
2912
2913 /*
2914 * check some assumptions
2915 */
aa1b8cd4 2916 if (fs_info->chunk_root->nodesize != fs_info->chunk_root->leafsize) {
efe120a0
FH		 2917 btrfs_err(fs_info,
2918 "scrub: size assumption nodesize == leafsize (%d == %d) fails",
aa1b8cd4
SB		 2919 fs_info->chunk_root->nodesize,
2920 fs_info->chunk_root->leafsize);
b5d67f64
SB		 2921 return -EINVAL;
2922 }
2923
aa1b8cd4 2924 if (fs_info->chunk_root->nodesize > BTRFS_STRIPE_LEN) {
b5d67f64
SB		 2925 /*
2926 * in this case scrub is unable to calculate the checksum
2927 * the way scrub is implemented. Do not handle this
2928 * situation at all because it won't ever happen.
2929 */
efe120a0
FH		 2930 btrfs_err(fs_info,
2931 "scrub: size assumption nodesize <= BTRFS_STRIPE_LEN (%d <= %d) fails",
aa1b8cd4 2932 fs_info->chunk_root->nodesize, BTRFS_STRIPE_LEN);
b5d67f64
SB		 2933 return -EINVAL;
2934 }
2935
aa1b8cd4 2936 if (fs_info->chunk_root->sectorsize != PAGE_SIZE) {
b5d67f64 2937 /* not supported for data w/o checksums */
efe120a0
FH		 2938 btrfs_err(fs_info,
2939 "scrub: size assumption sectorsize != PAGE_SIZE "
2940 "(%d != %lu) fails",
27f9f023 2941 fs_info->chunk_root->sectorsize, PAGE_SIZE);
a2de733c
AJ		 2942 return -EINVAL;
2943 }
2944
7a9e9987
SB		 2945 if (fs_info->chunk_root->nodesize >
2946 PAGE_SIZE * SCRUB_MAX_PAGES_PER_BLOCK ||
2947 fs_info->chunk_root->sectorsize >
2948 PAGE_SIZE * SCRUB_MAX_PAGES_PER_BLOCK) {
2949 /*
2950 * would exhaust the array bounds of pagev member in
2951 * struct scrub_block
2952 */
efe120a0
FH		 2953 btrfs_err(fs_info, "scrub: size assumption nodesize and sectorsize "
2954 "<= SCRUB_MAX_PAGES_PER_BLOCK (%d <= %d && %d <= %d) fails",
7a9e9987
SB		 2955 fs_info->chunk_root->nodesize,
2956 SCRUB_MAX_PAGES_PER_BLOCK,
2957 fs_info->chunk_root->sectorsize,
2958 SCRUB_MAX_PAGES_PER_BLOCK);
2959 return -EINVAL;
2960 }
2961
a2de733c 2962
aa1b8cd4
SB		 2963 mutex_lock(&fs_info->fs_devices->device_list_mutex);
2964 dev = btrfs_find_device(fs_info, devid, NULL, NULL);
63a212ab 2965 if (!dev || (dev->missing && !is_dev_replace)) {
aa1b8cd4 2966 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
a2de733c
AJ		 2967 return -ENODEV;
2968 }
a2de733c 2969
5d68da3b
MX		 2970 if (!is_dev_replace && !readonly && !dev->writeable) {
2971 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
2972 rcu_read_lock();
2973 name = rcu_dereference(dev->name);
2974 btrfs_err(fs_info, "scrub: device %s is not writable",
2975 name->str);
2976 rcu_read_unlock();
2977 return -EROFS;
2978 }
2979
3b7a016f 2980 mutex_lock(&fs_info->scrub_lock);
63a212ab 2981 if (!dev->in_fs_metadata || dev->is_tgtdev_for_dev_replace) {
a2de733c 2982 mutex_unlock(&fs_info->scrub_lock);
aa1b8cd4 2983 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
aa1b8cd4 2984 return -EIO;
a2de733c
AJ		 2985 }
2986
8dabb742
SB		 2987 btrfs_dev_replace_lock(&fs_info->dev_replace);
2988 if (dev->scrub_device ||
2989 (!is_dev_replace &&
2990 btrfs_dev_replace_is_ongoing(&fs_info->dev_replace))) {
2991 btrfs_dev_replace_unlock(&fs_info->dev_replace);
a2de733c 2992 mutex_unlock(&fs_info->scrub_lock);
aa1b8cd4 2993 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
a2de733c
AJ		 2994 return -EINPROGRESS;
2995 }
8dabb742 2996 btrfs_dev_replace_unlock(&fs_info->dev_replace);
3b7a016f
WS		 2997
2998 ret = scrub_workers_get(fs_info, is_dev_replace);
2999 if (ret) {
3000 mutex_unlock(&fs_info->scrub_lock);
3001 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
3002 return ret;
3003 }
3004
63a212ab 3005 sctx = scrub_setup_ctx(dev, is_dev_replace);
d9d181c1 3006 if (IS_ERR(sctx)) {
a2de733c 3007 mutex_unlock(&fs_info->scrub_lock);
aa1b8cd4
SB		 3008 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
3009 scrub_workers_put(fs_info);
d9d181c1 3010 return PTR_ERR(sctx);
a2de733c 3011 }
d9d181c1
SB		 3012 sctx->readonly = readonly;
3013 dev->scrub_device = sctx;
3cb0929a 3014 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
a2de733c 3015
3cb0929a
WS		 3016 /*
3017 * checking @scrub_pause_req here, we can avoid
3018 * race between committing transaction and scrubbing.
3019 */
cb7ab021 3020 __scrub_blocked_if_needed(fs_info);
a2de733c
AJ		 3021 atomic_inc(&fs_info->scrubs_running);
3022 mutex_unlock(&fs_info->scrub_lock);
a2de733c 3023
ff023aac 3024 if (!is_dev_replace) {
9b011adf
WS		 3025 /*
3026 * by holding device list mutex, we can
3027 * kick off writing super in log tree sync.
3028 */
3cb0929a 3029 mutex_lock(&fs_info->fs_devices->device_list_mutex);
ff023aac 3030 ret = scrub_supers(sctx, dev);
3cb0929a 3031 mutex_unlock(&fs_info->fs_devices->device_list_mutex);
ff023aac 3032 }
a2de733c
AJ		 3033
3034 if (!ret)
ff023aac
SB		 3035 ret = scrub_enumerate_chunks(sctx, dev, start, end,
3036 is_dev_replace);
a2de733c 3037
b6bfebc1 3038 wait_event(sctx->list_wait, atomic_read(&sctx->bios_in_flight) == 0);
a2de733c
AJ		 3039 atomic_dec(&fs_info->scrubs_running);
3040 wake_up(&fs_info->scrub_pause_wait);
3041
b6bfebc1 3042 wait_event(sctx->list_wait, atomic_read(&sctx->workers_pending) == 0);
0ef8e451 3043
a2de733c 3044 if (progress)
d9d181c1 3045 memcpy(progress, &sctx->stat, sizeof(*progress));
a2de733c
AJ		 3046
3047 mutex_lock(&fs_info->scrub_lock);
3048 dev->scrub_device = NULL;
3b7a016f 3049 scrub_workers_put(fs_info);
a2de733c
AJ		 3050 mutex_unlock(&fs_info->scrub_lock);
3051
d9d181c1 3052 scrub_free_ctx(sctx);
a2de733c
AJ		 3053
3054 return ret;
3055}
3056
143bede5 3057void btrfs_scrub_pause(struct btrfs_root *root)
a2de733c
AJ		 3058{
3059 struct btrfs_fs_info *fs_info = root->fs_info;
3060
3061 mutex_lock(&fs_info->scrub_lock);
3062 atomic_inc(&fs_info->scrub_pause_req);
3063 while (atomic_read(&fs_info->scrubs_paused) !=
3064 atomic_read(&fs_info->scrubs_running)) {
3065 mutex_unlock(&fs_info->scrub_lock);
3066 wait_event(fs_info->scrub_pause_wait,
3067 atomic_read(&fs_info->scrubs_paused) ==
3068 atomic_read(&fs_info->scrubs_running));
3069 mutex_lock(&fs_info->scrub_lock);
3070 }
3071 mutex_unlock(&fs_info->scrub_lock);
a2de733c
AJ		 3072}
3073
143bede5 3074void btrfs_scrub_continue(struct btrfs_root *root)
a2de733c
AJ		 3075{
3076 struct btrfs_fs_info *fs_info = root->fs_info;
3077
3078 atomic_dec(&fs_info->scrub_pause_req);
3079 wake_up(&fs_info->scrub_pause_wait);
a2de733c
AJ		 3080}
3081
aa1b8cd4 3082int btrfs_scrub_cancel(struct btrfs_fs_info *fs_info)
a2de733c 3083{
a2de733c
AJ		 3084 mutex_lock(&fs_info->scrub_lock);
3085 if (!atomic_read(&fs_info->scrubs_running)) {
3086 mutex_unlock(&fs_info->scrub_lock);
3087 return -ENOTCONN;
3088 }
3089
3090 atomic_inc(&fs_info->scrub_cancel_req);
3091 while (atomic_read(&fs_info->scrubs_running)) {
3092 mutex_unlock(&fs_info->scrub_lock);
3093 wait_event(fs_info->scrub_pause_wait,
3094 atomic_read(&fs_info->scrubs_running) == 0);
3095 mutex_lock(&fs_info->scrub_lock);
3096 }
3097 atomic_dec(&fs_info->scrub_cancel_req);
3098 mutex_unlock(&fs_info->scrub_lock);
3099
3100 return 0;
3101}
3102
aa1b8cd4
SB		 3103int btrfs_scrub_cancel_dev(struct btrfs_fs_info *fs_info,
3104 struct btrfs_device *dev)
49b25e05 3105{
d9d181c1 3106 struct scrub_ctx *sctx;
a2de733c
AJ		 3107
3108 mutex_lock(&fs_info->scrub_lock);
d9d181c1
SB		 3109 sctx = dev->scrub_device;
3110 if (!sctx) {
a2de733c
AJ		 3111 mutex_unlock(&fs_info->scrub_lock);
3112 return -ENOTCONN;
3113 }
d9d181c1 3114 atomic_inc(&sctx->cancel_req);
a2de733c
AJ		 3115 while (dev->scrub_device) {
3116 mutex_unlock(&fs_info->scrub_lock);
3117 wait_event(fs_info->scrub_pause_wait,
3118 dev->scrub_device == NULL);
3119 mutex_lock(&fs_info->scrub_lock);
3120 }
3121 mutex_unlock(&fs_info->scrub_lock);
3122
3123 return 0;
3124}
1623edeb 3125
a2de733c
AJ		 3126int btrfs_scrub_progress(struct btrfs_root *root, u64 devid,
3127 struct btrfs_scrub_progress *progress)
3128{
3129 struct btrfs_device *dev;
d9d181c1 3130 struct scrub_ctx *sctx = NULL;
a2de733c
AJ		 3131
3132 mutex_lock(&root->fs_info->fs_devices->device_list_mutex);
aa1b8cd4 3133 dev = btrfs_find_device(root->fs_info, devid, NULL, NULL);
a2de733c 3134 if (dev)
d9d181c1
SB		 3135 sctx = dev->scrub_device;
3136 if (sctx)
3137 memcpy(progress, &sctx->stat, sizeof(*progress));
a2de733c
AJ		 3138 mutex_unlock(&root->fs_info->fs_devices->device_list_mutex);
3139
d9d181c1 3140 return dev ? (sctx ? 0 : -ENOTCONN) : -ENODEV;
a2de733c 3141}
ff023aac
SB		 3142
3143static void scrub_remap_extent(struct btrfs_fs_info *fs_info,
3144 u64 extent_logical, u64 extent_len,
3145 u64 *extent_physical,
3146 struct btrfs_device **extent_dev,
3147 int *extent_mirror_num)
3148{
3149 u64 mapped_length;
3150 struct btrfs_bio *bbio = NULL;
3151 int ret;
3152
3153 mapped_length = extent_len;
3154 ret = btrfs_map_block(fs_info, READ, extent_logical,
3155 &mapped_length, &bbio, 0);
3156 if (ret || !bbio || mapped_length < extent_len ||
3157 !bbio->stripes[0].dev->bdev) {
3158 kfree(bbio);
3159 return;
3160 }
3161
3162 *extent_physical = bbio->stripes[0].physical;
3163 *extent_mirror_num = bbio->mirror_num;
3164 *extent_dev = bbio->stripes[0].dev;
3165 kfree(bbio);
3166}
3167
3168static int scrub_setup_wr_ctx(struct scrub_ctx *sctx,
3169 struct scrub_wr_ctx *wr_ctx,
3170 struct btrfs_fs_info *fs_info,
3171 struct btrfs_device *dev,
3172 int is_dev_replace)
3173{
3174 WARN_ON(wr_ctx->wr_curr_bio != NULL);
3175
3176 mutex_init(&wr_ctx->wr_lock);
3177 wr_ctx->wr_curr_bio = NULL;
3178 if (!is_dev_replace)
3179 return 0;
3180
3181 WARN_ON(!dev->bdev);
3182 wr_ctx->pages_per_wr_bio = min_t(int, SCRUB_PAGES_PER_WR_BIO,
3183 bio_get_nr_vecs(dev->bdev));
3184 wr_ctx->tgtdev = dev;
3185 atomic_set(&wr_ctx->flush_all_writes, 0);
3186 return 0;
3187}
3188
3189static void scrub_free_wr_ctx(struct scrub_wr_ctx *wr_ctx)
3190{
3191 mutex_lock(&wr_ctx->wr_lock);
3192 kfree(wr_ctx->wr_curr_bio);
3193 wr_ctx->wr_curr_bio = NULL;
3194 mutex_unlock(&wr_ctx->wr_lock);
3195}
3196
3197static int copy_nocow_pages(struct scrub_ctx *sctx, u64 logical, u64 len,
3198 int mirror_num, u64 physical_for_dev_replace)
3199{
3200 struct scrub_copy_nocow_ctx *nocow_ctx;
3201 struct btrfs_fs_info *fs_info = sctx->dev_root->fs_info;
3202
3203 nocow_ctx = kzalloc(sizeof(*nocow_ctx), GFP_NOFS);
3204 if (!nocow_ctx) {
3205 spin_lock(&sctx->stat_lock);
3206 sctx->stat.malloc_errors++;
3207 spin_unlock(&sctx->stat_lock);
3208 return -ENOMEM;
3209 }
3210
3211 scrub_pending_trans_workers_inc(sctx);
3212
3213 nocow_ctx->sctx = sctx;
3214 nocow_ctx->logical = logical;
3215 nocow_ctx->len = len;
3216 nocow_ctx->mirror_num = mirror_num;
3217 nocow_ctx->physical_for_dev_replace = physical_for_dev_replace;
9e0af237
LB		 3218 btrfs_init_work(&nocow_ctx->work, btrfs_scrubnc_helper,
3219 copy_nocow_pages_worker, NULL, NULL);
652f25a2 3220 INIT_LIST_HEAD(&nocow_ctx->inodes);
0339ef2f
QW		 3221 btrfs_queue_work(fs_info->scrub_nocow_workers,
3222 &nocow_ctx->work);
ff023aac
SB		 3223
3224 return 0;
3225}
3226
652f25a2
JB		 3227static int record_inode_for_nocow(u64 inum, u64 offset, u64 root, void *ctx)
3228{
3229 struct scrub_copy_nocow_ctx *nocow_ctx = ctx;
3230 struct scrub_nocow_inode *nocow_inode;
3231
3232 nocow_inode = kzalloc(sizeof(*nocow_inode), GFP_NOFS);
3233 if (!nocow_inode)
3234 return -ENOMEM;
3235 nocow_inode->inum = inum;
3236 nocow_inode->offset = offset;
3237 nocow_inode->root = root;
3238 list_add_tail(&nocow_inode->list, &nocow_ctx->inodes);
3239 return 0;
3240}
3241
3242#define COPY_COMPLETE 1
3243
ff023aac
SB		 3244static void copy_nocow_pages_worker(struct btrfs_work *work)
3245{
3246 struct scrub_copy_nocow_ctx *nocow_ctx =
3247 container_of(work, struct scrub_copy_nocow_ctx, work);
3248 struct scrub_ctx *sctx = nocow_ctx->sctx;
3249 u64 logical = nocow_ctx->logical;
3250 u64 len = nocow_ctx->len;
3251 int mirror_num = nocow_ctx->mirror_num;
3252 u64 physical_for_dev_replace = nocow_ctx->physical_for_dev_replace;
3253 int ret;
3254 struct btrfs_trans_handle *trans = NULL;
3255 struct btrfs_fs_info *fs_info;
3256 struct btrfs_path *path;
3257 struct btrfs_root *root;
3258 int not_written = 0;
3259
3260 fs_info = sctx->dev_root->fs_info;
3261 root = fs_info->extent_root;
3262
3263 path = btrfs_alloc_path();
3264 if (!path) {
3265 spin_lock(&sctx->stat_lock);
3266 sctx->stat.malloc_errors++;
3267 spin_unlock(&sctx->stat_lock);
3268 not_written = 1;
3269 goto out;
3270 }
3271
3272 trans = btrfs_join_transaction(root);
3273 if (IS_ERR(trans)) {
3274 not_written = 1;
3275 goto out;
3276 }
3277
3278 ret = iterate_inodes_from_logical(logical, fs_info, path,
652f25a2 3279 record_inode_for_nocow, nocow_ctx);
ff023aac 3280 if (ret != 0 && ret != -ENOENT) {
efe120a0
FH		 3281 btrfs_warn(fs_info, "iterate_inodes_from_logical() failed: log %llu, "
3282 "phys %llu, len %llu, mir %u, ret %d",
118a0a25
GU		 3283 logical, physical_for_dev_replace, len, mirror_num,
3284 ret);
ff023aac
SB		 3285 not_written = 1;
3286 goto out;
3287 }
3288
652f25a2
JB		 3289 btrfs_end_transaction(trans, root);
3290 trans = NULL;
3291 while (!list_empty(&nocow_ctx->inodes)) {
3292 struct scrub_nocow_inode *entry;
3293 entry = list_first_entry(&nocow_ctx->inodes,
3294 struct scrub_nocow_inode,
3295 list);
3296 list_del_init(&entry->list);
3297 ret = copy_nocow_pages_for_inode(entry->inum, entry->offset,
3298 entry->root, nocow_ctx);
3299 kfree(entry);
3300 if (ret == COPY_COMPLETE) {
3301 ret = 0;
3302 break;
3303 } else if (ret) {
3304 break;
3305 }
3306 }
ff023aac 3307out:
652f25a2
JB		 3308 while (!list_empty(&nocow_ctx->inodes)) {
3309 struct scrub_nocow_inode *entry;
3310 entry = list_first_entry(&nocow_ctx->inodes,
3311 struct scrub_nocow_inode,
3312 list);
3313 list_del_init(&entry->list);
3314 kfree(entry);
3315 }
ff023aac
SB		 3316 if (trans && !IS_ERR(trans))
3317 btrfs_end_transaction(trans, root);
3318 if (not_written)
3319 btrfs_dev_replace_stats_inc(&fs_info->dev_replace.
3320 num_uncorrectable_read_errors);
3321
3322 btrfs_free_path(path);
3323 kfree(nocow_ctx);
3324
3325 scrub_pending_trans_workers_dec(sctx);
3326}
3327
652f25a2
JB		 3328static int copy_nocow_pages_for_inode(u64 inum, u64 offset, u64 root,
3329 struct scrub_copy_nocow_ctx *nocow_ctx)
ff023aac 3330{
826aa0a8 3331 struct btrfs_fs_info *fs_info = nocow_ctx->sctx->dev_root->fs_info;
ff023aac 3332 struct btrfs_key key;
826aa0a8
MX		 3333 struct inode *inode;
3334 struct page *page;
ff023aac 3335 struct btrfs_root *local_root;
652f25a2
JB		 3336 struct btrfs_ordered_extent *ordered;
3337 struct extent_map *em;
3338 struct extent_state *cached_state = NULL;
3339 struct extent_io_tree *io_tree;
ff023aac 3340 u64 physical_for_dev_replace;
652f25a2
JB		 3341 u64 len = nocow_ctx->len;
3342 u64 lockstart = offset, lockend = offset + len - 1;
826aa0a8 3343 unsigned long index;
6f1c3605 3344 int srcu_index;
652f25a2
JB		 3345 int ret = 0;
3346 int err = 0;
ff023aac
SB		 3347
3348 key.objectid = root;
3349 key.type = BTRFS_ROOT_ITEM_KEY;
3350 key.offset = (u64)-1;
6f1c3605
LB		 3351
3352 srcu_index = srcu_read_lock(&fs_info->subvol_srcu);
3353
ff023aac 3354 local_root = btrfs_read_fs_root_no_name(fs_info, &key);
6f1c3605
LB		 3355 if (IS_ERR(local_root)) {
3356 srcu_read_unlock(&fs_info->subvol_srcu, srcu_index);
ff023aac 3357 return PTR_ERR(local_root);
6f1c3605 3358 }
ff023aac
SB		 3359
3360 key.type = BTRFS_INODE_ITEM_KEY;
3361 key.objectid = inum;
3362 key.offset = 0;
3363 inode = btrfs_iget(fs_info->sb, &key, local_root, NULL);
6f1c3605 3364 srcu_read_unlock(&fs_info->subvol_srcu, srcu_index);
ff023aac
SB		 3365 if (IS_ERR(inode))
3366 return PTR_ERR(inode);
3367
edd1400b
MX		 3368 /* Avoid truncate/dio/punch hole.. */
3369 mutex_lock(&inode->i_mutex);
3370 inode_dio_wait(inode);
3371
ff023aac 3372 physical_for_dev_replace = nocow_ctx->physical_for_dev_replace;
652f25a2
JB		 3373 io_tree = &BTRFS_I(inode)->io_tree;
3374
3375 lock_extent_bits(io_tree, lockstart, lockend, 0, &cached_state);
3376 ordered = btrfs_lookup_ordered_range(inode, lockstart, len);
3377 if (ordered) {
3378 btrfs_put_ordered_extent(ordered);
3379 goto out_unlock;
3380 }
3381
3382 em = btrfs_get_extent(inode, NULL, 0, lockstart, len, 0);
3383 if (IS_ERR(em)) {
3384 ret = PTR_ERR(em);
3385 goto out_unlock;
3386 }
3387
3388 /*
3389 * This extent does not actually cover the logical extent anymore,
3390 * move on to the next inode.
3391 */
3392 if (em->block_start > nocow_ctx->logical ||
3393 em->block_start + em->block_len < nocow_ctx->logical + len) {
3394 free_extent_map(em);
3395 goto out_unlock;
3396 }
3397 free_extent_map(em);
3398
ff023aac 3399 while (len >= PAGE_CACHE_SIZE) {
ff023aac 3400 index = offset >> PAGE_CACHE_SHIFT;
edd1400b 3401again:
ff023aac
SB		 3402 page = find_or_create_page(inode->i_mapping, index, GFP_NOFS);
3403 if (!page) {
efe120a0 3404 btrfs_err(fs_info, "find_or_create_page() failed");
ff023aac 3405 ret = -ENOMEM;
826aa0a8 3406 goto out;
ff023aac
SB		 3407 }
3408
3409 if (PageUptodate(page)) {
3410 if (PageDirty(page))
3411 goto next_page;
3412 } else {
3413 ClearPageError(page);
652f25a2
JB		 3414 err = extent_read_full_page_nolock(io_tree, page,
3415 btrfs_get_extent,
3416 nocow_ctx->mirror_num);
826aa0a8
MX		 3417 if (err) {
3418 ret = err;
ff023aac
SB		 3419 goto next_page;
3420 }
edd1400b 3421
26b25891 3422 lock_page(page);
edd1400b
MX		 3423 /*
3424 * If the page has been remove from the page cache,
3425 * the data on it is meaningless, because it may be
3426 * old one, the new data may be written into the new
3427 * page in the page cache.
3428 */
3429 if (page->mapping != inode->i_mapping) {
652f25a2 3430 unlock_page(page);
edd1400b
MX		 3431 page_cache_release(page);
3432 goto again;
3433 }
ff023aac
SB		 3434 if (!PageUptodate(page)) {
3435 ret = -EIO;
3436 goto next_page;
3437 }
3438 }
826aa0a8
MX		 3439 err = write_page_nocow(nocow_ctx->sctx,
3440 physical_for_dev_replace, page);
3441 if (err)
3442 ret = err;
ff023aac 3443next_page:
826aa0a8
MX		 3444 unlock_page(page);
3445 page_cache_release(page);
3446
3447 if (ret)
3448 break;
3449
ff023aac
SB		 3450 offset += PAGE_CACHE_SIZE;
3451 physical_for_dev_replace += PAGE_CACHE_SIZE;
3452 len -= PAGE_CACHE_SIZE;
3453 }
652f25a2
JB		 3454 ret = COPY_COMPLETE;
3455out_unlock:
3456 unlock_extent_cached(io_tree, lockstart, lockend, &cached_state,
3457 GFP_NOFS);
826aa0a8 3458out:
edd1400b 3459 mutex_unlock(&inode->i_mutex);
826aa0a8 3460 iput(inode);
ff023aac
SB		 3461 return ret;
3462}
3463
3464static int write_page_nocow(struct scrub_ctx *sctx,
3465 u64 physical_for_dev_replace, struct page *page)
3466{
3467 struct bio *bio;
3468 struct btrfs_device *dev;
3469 int ret;
ff023aac
SB		 3470
3471 dev = sctx->wr_ctx.tgtdev;
3472 if (!dev)
3473 return -EIO;
3474 if (!dev->bdev) {
3475 printk_ratelimited(KERN_WARNING
efe120a0 3476 "BTRFS: scrub write_page_nocow(bdev == NULL) is unexpected!\n");
ff023aac
SB		 3477 return -EIO;
3478 }
9be3395b 3479 bio = btrfs_io_bio_alloc(GFP_NOFS, 1);
ff023aac
SB		 3480 if (!bio) {
3481 spin_lock(&sctx->stat_lock);
3482 sctx->stat.malloc_errors++;
3483 spin_unlock(&sctx->stat_lock);
3484 return -ENOMEM;
3485 }
4f024f37
KO		 3486 bio->bi_iter.bi_size = 0;
3487 bio->bi_iter.bi_sector = physical_for_dev_replace >> 9;
ff023aac
SB		 3488 bio->bi_bdev = dev->bdev;
3489 ret = bio_add_page(bio, page, PAGE_CACHE_SIZE, 0);
3490 if (ret != PAGE_CACHE_SIZE) {
3491leave_with_eio:
3492 bio_put(bio);
3493 btrfs_dev_stat_inc_and_print(dev, BTRFS_DEV_STAT_WRITE_ERRS);
3494 return -EIO;
3495 }
ff023aac 3496
33879d45 3497 if (btrfsic_submit_bio_wait(WRITE_SYNC, bio))
ff023aac
SB		 3498 goto leave_with_eio;
3499
3500 bio_put(bio);
3501 return 0;
3502}
Linux kernel - Deliverables
This page took 0.359461 seconds and 5 git commands to generate.