LU-3335 scrub: control OI scrub on OST from user space

[fs/lustre-release.git] / lustre / utils / lustre_lfsck.c

diff --git a/lustre/utils/lustre_lfsck.c b/lustre/utils/lustre_lfsck.c
index 403219a..20ea2be 100644 (file)
--- a/lustre/utils/lustre_lfsck.c
+++ b/lustre/utils/lustre_lfsck.c
@@ -20,7 +20,7 @@
  * GPL HEADER END
  */
 /*
- * Copyright (c) 2012 Whamcloud, Inc.
+ * Copyright (c) 2012, 2013, Intel Corporation.
  */
 /*
  * lustre/utils/lustre_lfsck.c
@@ -62,28 +62,42 @@ static struct option long_opt_stop[] = {
        {0,             0,                 0,   0}
 };
 
-struct lfsck_types_names {
-       char   *name;
-       __u16   type;
+struct lfsck_type_name {
+       char            *name;
+       int              namelen;
+       enum lfsck_type  type;
 };
 
-static struct lfsck_types_names lfsck_types_names[3] = {
-       { "layout",     LT_LAYOUT },
-       { "DNE",        LT_DNE },
-       { 0,            0 }
+static struct lfsck_type_name lfsck_types_names[] = {
+       { "layout",     6,      LT_LAYOUT },
+       { "namespace",  9,      LT_NAMESPACE},
+       { 0,            0,      0 }
 };
 
+static inline int lfsck_name2type(const char *name, int namelen)
+{
+       int i = 0;
+
+       while (lfsck_types_names[i].name != NULL) {
+               if (namelen == lfsck_types_names[i].namelen &&
+                   strncmp(lfsck_types_names[i].name, name, namelen) == 0)
+                       return lfsck_types_names[i].type;
+               i++;
+       }
+       return 0;
+}
+
 static void usage_start(void)
 {
        fprintf(stderr, "Start LFSCK.\n"
                "SYNOPSIS:\n"
-               "lfsck_start <-M | --device MDT_device>\n"
+               "lfsck_start <-M | --device [MDT,OST]_device>\n"
                "            [-e | --error error_handle] [-h | --help]\n"
                "            [-n | --dryrun switch] [-r | --reset]\n"
                "            [-s | --speed speed_limit]\n"
                "            [-t | --type lfsck_type[,lfsck_type...]]\n"
                "OPTIONS:\n"
-               "-M: The MDT device to start LFSCK on.\n"
+               "-M: The device to start LFSCK/scrub on.\n"
                "-e: Error handle, 'continue'(default) or 'abort'.\n"
                "-h: Help information.\n"
                "-n: Check without modification. 'off'(default) or 'on'.\n"
@@ -98,9 +112,9 @@ static void usage_stop(void)
 {
        fprintf(stderr, "Stop LFSCK.\n"
                "SYNOPSIS:\n"
-               "lfsck_stop <-M | --device MDT_device> [-h | --help]\n"
+               "lfsck_stop <-M | --device [MDT,OST]_device> [-h | --help]\n"
                "OPTIONS:\n"
-               "-M: The MDT device to stop LFSCK on.\n"
+               "-M: The device to stop LFSCK/scrub on.\n"
                "-h: Help information.\n");
 }
 
@@ -109,7 +123,7 @@ static int lfsck_pack_dev(struct obd_ioctl_data *data, char *device, char *arg)
        int len = strlen(arg) + 1;
 
        if (len > MAX_OBD_NAME) {
-               fprintf(stderr, "MDT device name is too long. "
+               fprintf(stderr, "device name is too long. "
                        "Valid length should be less than %d\n", MAX_OBD_NAME);
                return -EINVAL;
        }
@@ -128,7 +142,7 @@ int jt_lfsck_start(int argc, char **argv)
        char device[MAX_OBD_NAME];
        struct lfsck_start start;
        char *optstring = "M:e:hn:rs:t:";
-       int opt, index, rc, val, i;
+       int opt, index, rc, val, i, type;
 
        memset(&data, 0, sizeof(data));
        memset(&start, 0, sizeof(start));
@@ -197,29 +211,25 @@ int jt_lfsck_start(int argc, char **argv)
 
                                c = *p;
                                *p = 0;
-                               for (i = 0; i < 3; i++) {
-                                       if (strcmp(str,
-                                                  lfsck_types_names[i].name)
-                                                  == 0) {
-                                               start.ls_active |=
-                                                   lfsck_types_names[i].type;
-                                               break;
-                                       }
+                               type = lfsck_name2type(str, strlen(str));
+                               if (type == 0) {
+                                       fprintf(stderr, "Invalid type (%s).\n"
+                                               "The valid value should be "
+                                               "'layout' or 'namespace'.\n",
+                                               str);
+                                       *p = c;
+                                       return -EINVAL;
                                }
+
                                *p = c;
                                str = p;
 
-                               if (i >= 3 ) {
-                                       fprintf(stderr, "Invalid LFSCK type.\n"
-                                               "The valid value should be "
-                                               "'layout' or 'DNE'.\n");
-                                       return -EINVAL;
-                               }
+                               start.ls_active |= type;
                        }
                        if (start.ls_active == 0) {
                                fprintf(stderr, "Miss LFSCK type(s).\n"
                                        "The valid value should be "
-                                       "'layout' or 'DNE'.\n");
+                                       "'layout' or 'namespace'.\n");
                                return -EINVAL;
                        }
                        break;
@@ -231,9 +241,15 @@ int jt_lfsck_start(int argc, char **argv)
        }
 
        if (data.ioc_inlbuf4 == NULL) {
-               fprintf(stderr,
-                       "Must sepcify MDT device to start LFSCK.\n");
-               return -EINVAL;
+               if (lcfg_get_devname() != NULL) {
+                       rc = lfsck_pack_dev(&data, device, lcfg_get_devname());
+                       if (rc != 0)
+                               return rc;
+               } else {
+                       fprintf(stderr,
+                               "Must specify device to start LFSCK.\n");
+                       return -EINVAL;
+               }
        }
 
        data.ioc_inlbuf1 = (char *)&start;
@@ -253,14 +269,16 @@ int jt_lfsck_start(int argc, char **argv)
 
        obd_ioctl_unpack(&data, buf, sizeof(rawbuf));
        if (start.ls_active == 0) {
-               printf("Started LFSCK on the MDT device %s", device);
+               printf("Started LFSCK on the device %s", device);
        } else {
-               printf("Started LFSCK on the MDT device %s:", device);
-               for (i = 0; i < 2; i++) {
+               printf("Started LFSCK on the device %s:", device);
+               i = 0;
+               while (lfsck_types_names[i].name != NULL) {
                        if (start.ls_active & lfsck_types_names[i].type) {
                                printf(" %s", lfsck_types_names[i].name);
                                start.ls_active &= ~lfsck_types_names[i].type;
                        }
+                       i++;
                }
                if (start.ls_active != 0)
                        printf(" unknown(0x%x)", start.ls_active);
@@ -301,9 +319,15 @@ int jt_lfsck_stop(int argc, char **argv)
        }
 
        if (data.ioc_inlbuf4 == NULL) {
-               fprintf(stderr,
-                       "Must sepcify MDT device to stop LFSCK.\n");
-               return -EINVAL;
+               if (lcfg_get_devname() != NULL) {
+                       rc = lfsck_pack_dev(&data, device, lcfg_get_devname());
+                       if (rc != 0)
+                               return rc;
+               } else {
+                       fprintf(stderr,
+                               "Must sepcify device to stop LFSCK.\n");
+                       return -EINVAL;
+               }
        }
 
        memset(buf, 0, sizeof(rawbuf));
@@ -319,6 +343,6 @@ int jt_lfsck_stop(int argc, char **argv)
                return rc;
        }
 
-       printf("Stopped LFSCK on the MDT device %s.\n", device);
+       printf("Stopped LFSCK on the device %s.\n", device);
        return 0;
 }