rescrubbing only previously detected error block

Signed-off-by: TulsiJain <tulsi.jain@delphix.com>

Author: TulsiJain

.gitignore

@@ -62,4 +62,7 @@ cscope.*
 *.orig
 *.log
 venv
+# Ignore Eclipse files
+/.cproject
+/.project
 

cmd/zpool/zpool_main.c

@@ -381,7 +381,8 @@ get_usage(zpool_help_t idx)
 		return (gettext("\tinitialize [-c | -s] <pool> "
 		    "[<device> ...]\n"));
 	case HELP_SCRUB:
-		return (gettext("\tscrub [-s | -p] <pool> ...\n"));
+		return (gettext("\tscrub [-s | -p | -e | -e -s | -e -p] <pool>"
+		    " ...\n"));
 	case HELP_RESILVER:
 		return (gettext("\tresilver <pool> ...\n"));
 	case HELP_TRIM:
@@ -6691,7 +6692,11 @@ scrub_callback(zpool_handle_t *zhp, void *data)
 		return (1);
 	}
 
-	err = zpool_scan(zhp, cb->cb_type, cb->cb_scrub_cmd);
+	if (cb->cb_type == POOL_RESCRUB) {
+		err = zpool_rescrub(zhp, cb->cb_type, cb->cb_scrub_cmd);
+	} else {
+		err = zpool_scan(zhp, cb->cb_type, cb->cb_scrub_cmd);
+	}
 
 	if (err == 0 && zpool_has_checkpoint(zhp) &&
 	    cb->cb_type == POOL_SCAN_SCRUB) {
@@ -6706,6 +6711,7 @@ scrub_callback(zpool_handle_t *zhp, void *data)
 /*
  * zpool scrub [-s | -p] <pool> ...
  *
+ *	-e	For error blocks. Starts/Resumes error blocks scrubbing.
  *	-s	Stop.  Stops any in-progress scrub.
  *	-p	Pause. Pause in-progress scrub.
  */
@@ -6718,27 +6724,52 @@ zpool_do_scrub(int argc, char **argv)
 	cb.cb_type = POOL_SCAN_SCRUB;
 	cb.cb_scrub_cmd = POOL_SCRUB_NORMAL;
 
+	boolean_t is_error_scrub = B_FALSE;
+	boolean_t is_pause = B_FALSE;
+	boolean_t is_stop = B_FALSE;
 	/* check options */
-	while ((c = getopt(argc, argv, "sp")) != -1) {
+	while ((c = getopt(argc, argv, "eps")) != -1) {
 		switch (c) {
-		case 's':
-			cb.cb_type = POOL_SCAN_NONE;
+		case 'e': {
+			is_error_scrub = B_TRUE;
 			break;
-		case 'p':
-			cb.cb_scrub_cmd = POOL_SCRUB_PAUSE;
+		}
+		case 's': {
+			is_stop = B_TRUE;
+			break;
+		}
+		case 'p': {
+			is_pause = B_TRUE;
 			break;
+		}
 		case '?':
 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
 			    optopt);
 			usage(B_FALSE);
 		}
 	}
 
-	if (cb.cb_type == POOL_SCAN_NONE &&
-	    cb.cb_scrub_cmd == POOL_SCRUB_PAUSE) {
-		(void) fprintf(stderr, gettext("invalid option combination: "
-		    "-s and -p are mutually exclusive\n"));
+	if (is_pause && is_stop) {
+		(void) fprintf(stderr, gettext("invalid option "
+		    "combination :-s and -p are mutually exclusive\n"));
 		usage(B_FALSE);
+	} else {
+		if (is_error_scrub) {
+			cb.cb_type = POOL_RESCRUB;
+			if (is_pause) {
+				cb.cb_scrub_cmd = POOL_RESCRUB_PAUSE;
+			} else if (is_stop) {
+				cb.cb_scrub_cmd = POOL_RESCRUB_STOP;
+			} else {
+				cb.cb_scrub_cmd = POOL_RESCRUB_NORMAL;
+			}
+		} else {
+			if (is_pause) {
+				cb.cb_scrub_cmd = POOL_SCRUB_PAUSE;
+			} else if (is_stop) {
+				cb.cb_type = POOL_SCAN_NONE;
+			}
+		}
 	}
 
 	cb.cb_argc = argc;
@@ -6912,6 +6943,69 @@ zpool_do_trim(int argc, char **argv)
 	return (error);
 }
 
+static void
+print_err_scrub_status(pool_scan_stat_t *ps)
+{
+	time_t start, end, pause;
+	uint64_t total_secs_left;
+	uint64_t secs_left, mins_left, hours_left, days_left;
+	uint64_t examined, to_be_examined;
+
+	(void) printf(gettext(" error scrub: "));
+
+	if (ps == NULL || ps->pss_error_scrub_func == POOL_SCAN_NONE ||
+	    ps->pss_error_scrub_func >= POOL_SCAN_FUNCS) {
+		(void) printf(gettext("no error scrubbing requested\n"));
+		return;
+	}
+
+	start = ps->pss_error_scrub_start;
+	end = ps->pss_error_scrub_end;
+	pause = ps->pss_pass_error_scrub_pause;
+	examined = ps->pss_error_scrub_examined;
+	to_be_examined = ps->pss_error_scrub_to_be_examined;
+
+	assert(ps->pss_error_scrub_func == POOL_RESCRUB);
+
+	if (ps->pss_error_scrub_state == DSS_FINISHED) {
+		total_secs_left = end - start;
+		days_left = total_secs_left / 60 / 60 / 24;
+		hours_left = (total_secs_left / 60 / 60) % 24;
+		mins_left = (total_secs_left / 60) % 60;
+		secs_left = (total_secs_left % 60);
+
+		(void) printf(gettext("scrubbed %llu error blocks in %llu days "
+		    "%02llu:%02llu:%02llu on %s"), (u_longlong_t)examined,
+		    (u_longlong_t)days_left, (u_longlong_t)hours_left,
+		    (u_longlong_t)mins_left, (u_longlong_t)secs_left,
+		    ctime(&end));
+
+		return;
+	} else if (ps->pss_error_scrub_state == DSS_CANCELED) {
+		(void) printf(gettext("error scrub canceled on %s"),
+		    ctime(&end));
+		return;
+	}
+	assert(ps->pss_error_scrub_state == DSS_RESCRUBING);
+
+	/* Error scrub is in progress. */
+	if (pause == 0) {
+		(void) printf(gettext("error scrub in progress since %s"),
+		    ctime(&start));
+	} else {
+		(void) printf(gettext("error scrub paused since %s"),
+		    ctime(&pause));
+		(void) printf(gettext("\terror scrub started on %s"),
+		    ctime(&start));
+	}
+
+	double fraction_done = (double)examined / (to_be_examined + examined);
+	(void) printf(gettext("\t%.2f%% done, issused i/o for %llu error"
+	    " blocks"), 100 * fraction_done, (u_longlong_t)examined);
+
+	(void) printf(gettext("\n"));
+}
+
 /*
  * Print out detailed scrub status.
  */
@@ -7687,6 +7781,7 @@ status_callback(zpool_handle_t *zhp, void *data)
 		    ZPOOL_CONFIG_REMOVAL_STATS, (uint64_t **)&prs, &c);
 
 		print_scan_status(ps);
+		print_err_scrub_status(ps);
 		print_checkpoint_scan_warning(ps, pcs);
 		print_removal_status(zhp, prs);
 		print_checkpoint_status(pcs);

include/libzfs.h

@@ -125,7 +125,10 @@ typedef enum zfs_error {
 	EZFS_THREADCREATEFAILED, /* thread create failed */
 	EZFS_POSTSPLIT_ONLINE,	/* onlining a disk after splitting it */
 	EZFS_SCRUBBING,		/* currently scrubbing */
+	EZFS_RESCRUBBING,	/* currently error scrubbing */
+	EZFS_RESCRUB_PAUSED,	/* rescrub currently paused */
 	EZFS_NO_SCRUB,		/* no active scrub */
+	EZFS_NO_RESCRUB,	/* no active rescrub */
 	EZFS_DIFF,		/* general failure of zfs diff */
 	EZFS_DIFFDATA,		/* bad zfs diff data */
 	EZFS_POOLREADONLY,	/* pool is in read-only mode */
@@ -273,6 +276,7 @@ typedef struct trimflags {
  * Functions to manipulate pool and vdev state
  */
 extern int zpool_scan(zpool_handle_t *, pool_scan_func_t, pool_scrub_cmd_t);
+extern int zpool_rescrub(zpool_handle_t *, pool_scan_func_t, pool_scrub_cmd_t);
 extern int zpool_initialize(zpool_handle_t *, pool_initialize_func_t,
     nvlist_t *);
 extern int zpool_trim(zpool_handle_t *, pool_trim_func_t, nvlist_t *,

include/libzfs_core.h

@@ -130,6 +130,8 @@ int lzc_reopen(const char *, boolean_t);
 int lzc_pool_checkpoint(const char *);
 int lzc_pool_checkpoint_discard(const char *);
 
+int lzc_ioctl(zfs_ioc_t, const char *, nvlist_t *, nvlist_t **);
+
 #ifdef	__cplusplus
 }
 #endif

include/sys/dmu.h

@@ -373,6 +373,7 @@ typedef struct dmu_buf {
 #define	DMU_POOL_DDT_STATS		"DDT-statistics"
 #define	DMU_POOL_CREATION_VERSION	"creation_version"
 #define	DMU_POOL_SCAN			"scan"
+#define	DMU_POOL_RESCRUB		"rescrub"
 #define	DMU_POOL_FREE_BPOBJ		"free_bpobj"
 #define	DMU_POOL_BPTREE_OBJ		"bptree_obj"
 #define	DMU_POOL_EMPTY_BPOBJ		"empty_bpobj"

include/sys/dsl_scan.h

@@ -29,6 +29,7 @@
 
 #include <sys/zfs_context.h>
 #include <sys/zio.h>
+#include <sys/zap.h>
 #include <sys/ddt.h>
 #include <sys/bplist.h>
 
@@ -76,6 +77,18 @@ typedef enum dsl_scan_flags {
 
 #define	DSL_SCAN_FLAGS_MASK (DSF_VISIT_DS_AGAIN)
 
+typedef struct dsl_rescrub_phys {
+	uint64_t rescrub_func; /* pool_scan_func_t */
+	uint64_t rescrub_state; /* dsl_scan_state_t */
+	zap_cursor_t rescrub_cursor; /* zap cursor */
+	uint64_t rescrub_start_time;
+	uint64_t rescrub_end_time;
+	uint64_t rescrub_to_examine; /* total blocks to be rescrubbed */
+	uint64_t rescrub_examined; /* blocks remaining so far */
+	uint64_t rescrub_errors;	/* rescrub I/O error count */
+	uint8_t rescrub_paused_flags; /* flag for paused */
+} dsl_rescrub_phys_t;
+
 /*
  * Every pool will have one dsl_scan_t and this structure will contain
  * in-memory information about the scan and a pointer to the on-disk
@@ -153,6 +166,9 @@ typedef struct dsl_scan {
 	dsl_scan_phys_t scn_phys_cached;
 	avl_tree_t scn_queue;		/* queue of datasets to scan */
 	uint64_t scn_bytes_pending;	/* outstanding data to issue */
+
+	/* members needed for syncing rescrub status to disk */
+	dsl_rescrub_phys_t rescrub_phys; /* on disk representation of rescrub */
 } dsl_scan_t;
 
 typedef struct dsl_scan_io_queue dsl_scan_io_queue_t;
@@ -166,6 +182,11 @@ int dsl_scan_cancel(struct dsl_pool *);
 int dsl_scan(struct dsl_pool *, pool_scan_func_t);
 boolean_t dsl_scan_scrubbing(const struct dsl_pool *dp);
 int dsl_scrub_set_pause_resume(const struct dsl_pool *dp, pool_scrub_cmd_t cmd);
+int dsl_rescrub_set_pause_resume(const struct dsl_pool *dp,
+    pool_scrub_cmd_t cmd);
+int dsl_rescrub(struct dsl_pool *, pool_scan_func_t);
+void dsl_rescrub_sync(struct dsl_pool *, dmu_tx_t *);
+int dsl_rescrub_cancel(struct dsl_pool *);
 void dsl_resilver_restart(struct dsl_pool *, uint64_t txg);
 boolean_t dsl_scan_resilvering(struct dsl_pool *dp);
 boolean_t dsl_dataset_unstable(struct dsl_dataset *ds);

include/sys/fs/zfs.h

@@ -899,6 +899,7 @@ typedef enum pool_scan_func {
 	POOL_SCAN_NONE,
 	POOL_SCAN_SCRUB,
 	POOL_SCAN_RESILVER,
+	POOL_RESCRUB,
 	POOL_SCAN_FUNCS
 } pool_scan_func_t;
 
@@ -908,6 +909,9 @@ typedef enum pool_scan_func {
 typedef enum pool_scrub_cmd {
 	POOL_SCRUB_NORMAL = 0,
 	POOL_SCRUB_PAUSE,
+	POOL_RESCRUB_NORMAL,
+	POOL_RESCRUB_PAUSE,
+	POOL_RESCRUB_STOP,
 	POOL_SCRUB_FLAGS_END
 } pool_scrub_cmd_t;
 
@@ -962,6 +966,16 @@ typedef struct pool_scan_stat {
 	uint64_t	pss_pass_scrub_spent_paused;
 	uint64_t	pss_pass_issued; /* issued bytes per scan pass */
 	uint64_t	pss_issued;	/* total bytes checked by scanner */
+	/* values not stored on disk */
+	uint64_t	pss_error_scrub_func;	/* pool_scan_func_t */
+	uint64_t	pss_error_scrub_state;	/* dsl_scan_state_t */
+	uint64_t	pss_error_scrub_start;	/* error scrub start time */
+	uint64_t	pss_error_scrub_end;	/* error scrub end time */
+	uint64_t	pss_error_scrub_examined;	/* scan start time */
+	uint64_t	pss_error_scrub_to_be_examined;	/* scan start time */
+	uint64_t	pss_pass_error_scrub_pause;
+	uint64_t	pss_pass_error_scrub_spent_paused;
+
 } pool_scan_stat_t;
 
 typedef struct pool_removal_stat {
@@ -983,6 +997,7 @@ typedef enum dsl_scan_state {
 	DSS_SCANNING,
 	DSS_FINISHED,
 	DSS_CANCELED,
+	DSS_RESCRUBING,
 	DSS_NUM_STATES
 } dsl_scan_state_t;
 
@@ -1277,6 +1292,7 @@ typedef enum zfs_ioc {
 	ZFS_IOC_POOL_TRIM,			/* 0x5a50 */
 	ZFS_IOC_REDACT,				/* 0x5a51 */
 	ZFS_IOC_GET_BOOKMARK_PROPS,		/* 0x5a52 */
+	ZFS_IOC_POOL_RESCRUB,			/* 0x5a53 */
 
 	/*
 	 * Linux - 3/64 numbers reserved.

include/sys/spa.h

@@ -866,6 +866,11 @@ extern int spa_scan(spa_t *spa, pool_scan_func_t func);
 extern int spa_scan_stop(spa_t *spa);
 extern int spa_scrub_pause_resume(spa_t *spa, pool_scrub_cmd_t flag);
 
+/* rescrubing */
+extern int spa_rescrub(spa_t *spa, pool_scan_func_t func);
+extern int spa_rescrub_stop(spa_t *spa);
+extern int spa_rescrub_pause_resume(spa_t *spa, pool_scrub_cmd_t flag);
+
 /* spa syncing */
 extern void spa_sync(spa_t *spa, uint64_t txg); /* only for DMU use */
 extern void spa_sync_allpools(void);
@@ -1176,6 +1181,7 @@ extern void zfs_post_remove(spa_t *spa, vdev_t *vd);
 extern void zfs_post_state_change(spa_t *spa, vdev_t *vd, uint64_t laststate);
 extern void zfs_post_autoreplace(spa_t *spa, vdev_t *vd);
 extern uint64_t spa_get_errlog_size(spa_t *spa);
+extern uint64_t spa_get_last_errlog_size(spa_t *spa);
 extern int spa_get_errlog(spa_t *spa, void *uaddr, size_t *count);
 extern void spa_errlog_rotate(spa_t *spa);
 extern void spa_errlog_drain(spa_t *spa);

include/sys/spa_impl.h

@@ -285,6 +285,8 @@ struct spa {
 	uint64_t	spa_scan_pass_exam;	/* examined bytes per pass */
 	uint64_t	spa_scan_pass_issued;	/* issued bytes per pass */
 
+	uint64_t	spa_scan_pass_rescrub_pause; /* rescrub pause time */
+	uint64_t	spa_scan_pass_rescrub_spent_paused; /* total paused */
 	/*
 	 * We are in the middle of a resilver, and another resilver
 	 * is needed once this one completes. This is set iff any

include/sys/sysevent/eventdefs.h

@@ -123,6 +123,11 @@ extern "C" {
 #define	ESC_ZFS_TRIM_CANCEL		"trim_cancel"
 #define	ESC_ZFS_TRIM_RESUME		"trim_resume"
 #define	ESC_ZFS_TRIM_SUSPEND		"trim_suspend"
+#define	ESC_ZFS_RESCRUB_START		"rescrub_start"
+#define	ESC_ZFS_RESCRUB_FINISH		"rescrub_finish"
+#define	ESC_ZFS_RESCRUB_ABORT		"rescrub_abort"
+#define	ESC_ZFS_RESCRUB_RESUME		"rescrub_resume"
+#define	ESC_ZFS_RESCRUB_PAUSED		"rescrub_paused"
 
 /*
  * datalink subclass definitions.