packaging: Enable LTO and set visibility to hidden
[platform/upstream/btrfs-progs.git] / cmds-scrub.c
index b7aa809..6b909f2 100644 (file)
@@ -17,6 +17,7 @@
  */
 
 #include "kerncompat.h"
+#include "androidcompat.h"
 
 #include <sys/ioctl.h>
 #include <sys/wait.h>
@@ -34,6 +35,7 @@
 #include <ctype.h>
 #include <signal.h>
 #include <stdarg.h>
+#include <limits.h>
 
 #include "ctree.h"
 #include "ioctl.h"
@@ -42,6 +44,7 @@
 #include "disk-io.h"
 
 #include "commands.h"
+#include "help.h"
 
 static const char * const scrub_cmd_group_usage[] = {
        "btrfs scrub <command> [options] <path>|<device>",
@@ -126,11 +129,6 @@ static void print_scrub_full(struct btrfs_scrub_progress *sp)
        printf("\tlast_physical: %lld\n", sp->last_physical);
 }
 
-#define ERR(test, ...) do {                    \
-       if (test)                               \
-               fprintf(stderr, __VA_ARGS__);   \
-} while (0)
-
 #define PRINT_SCRUB_ERROR(test, desc) do {     \
        if (test)                               \
                printf(" %s=%llu", desc, test); \
@@ -252,17 +250,15 @@ static void _print_scrub_ss(struct scrub_stats *ss)
        hours = ss->duration / (60 * 60);
        gmtime_r(&seconds, &tm);
        strftime(t, sizeof(t), "%M:%S", &tm);
-       if (ss->finished && !ss->canceled) {
-               printf(" and finished after %02u:%s\n", hours, t);
-       } else if (ss->canceled) {
+       if (ss->in_progress)
+               printf(", running for %02u:%s\n", hours, t);
+       else if (ss->canceled)
                printf(" and was aborted after %02u:%s\n", hours, t);
-       } else {
-               if (ss->in_progress)
-                       printf(", running for %02u:%s\n", hours, t);
-               else
-                       printf(", interrupted after %02u:%s, not running\n",
-                                       hours, t);
-       }
+       else if (ss->finished)
+               printf(" and finished after %02u:%s\n", hours, t);
+       else
+               printf(", interrupted after %02u:%s, not running\n",
+                      hours, t);
 }
 
 static void print_scrub_dev(struct btrfs_ioctl_dev_info_args *di,
@@ -435,7 +431,7 @@ static int scrub_rename_file(const char *fn_base, const char *fn_local,
 /*
  * returns 0 if the key did not match (nothing was read)
  *         1 if the key did match (success)
- *        -1 if the key did match and an error occured
+ *        -1 if the key did match and an error occurred
  */
 static int scrub_kvread(int *i, int len, int avail, const char *buf,
                        const char *key, u64 *dest)
@@ -461,8 +457,8 @@ static int scrub_kvread(int *i, int len, int avail, const char *buf,
 
 #define _SCRUB_INVALID do {                                            \
        if (report_errors)                                              \
-               fprintf(stderr, "WARNING: invalid data in line %d pos " \
-                       "%d state %d (near \"%.*s\") at %s:%d\n",       \
+               warning("invalid data on line %d pos "                  \
+                       "%d state %d (near \"%.*s\") at %s:%d",         \
                        lineno, i, state, 20 > avail ? avail : 20,      \
                        l + i,  __FILE__, __LINE__);                    \
        goto skip;                                                      \
@@ -472,7 +468,7 @@ static struct scrub_file_record **scrub_read_file(int fd, int report_errors)
 {
        int avail = 0;
        int old_avail = 0;
-       char l[16 * 1024];
+       char l[SZ_16K];
        int state = 0;
        int curr = -1;
        int i = 0;
@@ -486,7 +482,10 @@ static struct scrub_file_record **scrub_read_file(int fd, int report_errors)
 
 again:
        old_avail = avail - i;
-       BUG_ON(old_avail < 0);
+       if (old_avail < 0) {
+               error("scrub record file corrupted near byte %d", i);
+               return ERR_PTR(-EINVAL);
+       }
        if (old_avail)
                memmove(l, l + i, old_avail);
        avail = read(fd, l + old_avail, sizeof(l) - old_avail);
@@ -501,12 +500,16 @@ again:
                }
                return p;
        }
-       if (avail == -1)
+       if (avail == -1) {
+               free_history(p);
                return ERR_PTR(-errno);
+       }
        avail += old_avail;
 
        i = 0;
        while (i < avail) {
+               void *tmp;
+
                switch (state) {
                case 0: /* start of file */
                        ret = scrub_kvread(&i,
@@ -533,11 +536,17 @@ again:
                                continue;
                        }
                        ++curr;
+                       tmp = p;
                        p = realloc(p, (curr + 2) * sizeof(*p));
-                       if (p)
-                               p[curr] = malloc(sizeof(**p));
-                       if (!p || !p[curr])
+                       if (!p) {
+                               free_history(tmp);
+                               return ERR_PTR(-errno);
+                       }
+                       p[curr] = malloc(sizeof(**p));
+                       if (!p[curr]) {
+                               free_history(p);
                                return ERR_PTR(-errno);
+                       }
                        memset(p[curr], 0, sizeof(**p));
                        p[curr + 1] = NULL;
                        ++state;
@@ -582,8 +591,6 @@ again:
                        ret = 0;
                        _SCRUB_KVREAD(ret, &i, data_extents_scrubbed, avail, l,
                                        &p[curr]->p);
-                       _SCRUB_KVREAD(ret, &i, data_extents_scrubbed, avail, l,
-                                       &p[curr]->p);
                        _SCRUB_KVREAD(ret, &i, tree_extents_scrubbed, avail, l,
                                        &p[curr]->p);
                        _SCRUB_KVREAD(ret, &i, data_bytes_scrubbed, avail, l,
@@ -645,7 +652,9 @@ skip:
                        } while (i < avail);
                        continue;
                }
-               BUG();
+               error("internal error: unknown parser state %d near byte %d",
+                               state, i);
+               return ERR_PTR(-EINVAL);
        }
        goto again;
 }
@@ -838,9 +847,7 @@ static void *scrub_one_dev(void *ctx)
                      IOPRIO_PRIO_VALUE(sp->ioprio_class,
                                        sp->ioprio_classdata));
        if (ret)
-               fprintf(stderr,
-                       "WARNING: setting ioprio failed: %s (ignored).\n",
-                       strerror(errno));
+               warning("setting ioprio failed: %m (ignored)");
 
        ret = ioctl(sp->fd, BTRFS_IOC_SCRUB, &sp->scrub_args);
        gettimeofday(&tv, NULL);
@@ -1137,7 +1144,6 @@ static int scrub_start(int argc, char **argv, int resume)
        int force = 0;
        int nothing_to_resume = 0;
 
-       optind = 1;
        while ((c = getopt(argc, argv, "BdqrRc:n:f")) != -1) {
                switch (c) {
                case 'B':
@@ -1185,37 +1191,29 @@ static int scrub_start(int argc, char **argv, int resume)
                do_print = 0;
 
        if (mkdir_p(datafile)) {
-               ERR(!do_quiet, "WARNING: cannot create scrub data "
-                              "file, mkdir %s failed: %s. Status recording "
-                              "disabled\n", datafile, strerror(errno));
+               warning_on(!do_quiet,
+    "cannot create scrub data file, mkdir %s failed: %m. Status recording disabled",
+                       datafile);
                do_record = 0;
        }
        free(datafile);
 
        path = argv[optind];
 
-       fdmnt = open_path_or_dev_mnt(path, &dirstream);
-
-       if (fdmnt < 0) {
-               if (errno == EINVAL)
-                       ERR(!do_quiet,
-                           "ERROR: '%s' is not a mounted btrfs device\n",
-                           path);
-               else
-                       ERR(!do_quiet, "ERROR: can't access '%s': %s\n",
-                           path, strerror(errno));
+       fdmnt = open_path_or_dev_mnt(path, &dirstream, !do_quiet);
+       if (fdmnt < 0)
                return 1;
-       }
 
        ret = get_fs_info(path, &fi_args, &di_args);
        if (ret) {
-               ERR(!do_quiet, "ERROR: getting dev info for scrub failed: "
-                   "%s\n", strerror(-ret));
+               error_on(!do_quiet,
+                       "getting dev info for scrub failed: %s",
+                        strerror(-ret));
                err = 1;
                goto out;
        }
        if (!fi_args.num_devices) {
-               ERR(!do_quiet, "ERROR: no devices found\n");
+               error_on(!do_quiet, "no devices found");
                err = 1;
                goto out;
        }
@@ -1223,13 +1221,13 @@ static int scrub_start(int argc, char **argv, int resume)
        uuid_unparse(fi_args.fsid, fsid);
        fdres = scrub_open_file_r(SCRUB_DATA_FILE, fsid);
        if (fdres < 0 && fdres != -ENOENT) {
-               ERR(!do_quiet, "WARNING: failed to open status file: "
-                   "%s\n", strerror(-fdres));
+               warning_on(!do_quiet, "failed to open status file: %s",
+                       strerror(-fdres));
        } else if (fdres >= 0) {
                past_scrubs = scrub_read_file(fdres, !do_quiet);
                if (IS_ERR(past_scrubs))
-                       ERR(!do_quiet, "WARNING: failed to read status file: "
-                           "%s\n", strerror(-PTR_ERR(past_scrubs)));
+                       warning_on(!do_quiet, "failed to read status file: %s",
+                               strerror(-PTR_ERR(past_scrubs)));
                close(fdres);
        }
 
@@ -1252,11 +1250,11 @@ static int scrub_start(int argc, char **argv, int resume)
         * single devices, there is no reason to prevent this.
         */
        if (!force && is_scrub_running_on_fs(&fi_args, di_args, past_scrubs)) {
-               ERR(!do_quiet,
-                   "ERROR: scrub is already running.\n"
-                   "To cancel use 'btrfs scrub cancel %s'.\n"
-                   "To see the status use 'btrfs scrub status [-d] %s'.\n",
-                   path, path);
+               error_on(!do_quiet,
+                       "Scrub is already running.\n"
+                       "To cancel use 'btrfs scrub cancel %s'.\n"
+                       "To see the status use 'btrfs scrub status [-d] %s'",
+                       path, path);
                err = 1;
                goto out;
        }
@@ -1266,7 +1264,7 @@ static int scrub_start(int argc, char **argv, int resume)
        spc.progress = calloc(fi_args.num_devices * 2, sizeof(*spc.progress));
 
        if (!t_devs || !sp || !spc.progress) {
-               ERR(!do_quiet, "ERROR: scrub failed: %s", strerror(errno));
+               error_on(!do_quiet, "scrub failed: %m");
                err = 1;
                goto out;
        }
@@ -1275,8 +1273,8 @@ static int scrub_start(int argc, char **argv, int resume)
                devid = di_args[i].devid;
                ret = pthread_mutex_init(&sp[i].progress_mutex, NULL);
                if (ret) {
-                       ERR(!do_quiet, "ERROR: pthread_mutex_init failed: "
-                           "%s\n", strerror(ret));
+                       error_on(!do_quiet, "pthread_mutex_init failed: %s",
+                               strerror(ret));
                        err = 1;
                        goto out;
                }
@@ -1330,7 +1328,7 @@ static int scrub_start(int argc, char **argv, int resume)
                ret = connect(prg_fd, (struct sockaddr *)&addr, sizeof(addr));
                if (!ret || errno != ECONNREFUSED) {
                        /* ... yes, so scrub must be running. error out */
-                       fprintf(stderr, "ERROR: scrub already running\n");
+                       error("scrub already running");
                        close(prg_fd);
                        prg_fd = -1;
                        goto out;
@@ -1344,10 +1342,10 @@ static int scrub_start(int argc, char **argv, int resume)
        if (ret != -1)
                ret = listen(prg_fd, 100);
        if (ret == -1) {
-               ERR(!do_quiet, "WARNING: failed to open the progress status "
-                   "socket at %s: %s. Progress cannot be queried\n",
-                   sock_path[0] ? sock_path : SCRUB_PROGRESS_SOCKET_PATH,
-                   strerror(errno));
+               warning_on(!do_quiet,
+   "failed to open the progress status socket at %s: %m. Progress cannot be queried",
+                       sock_path[0] ? sock_path :
+                       SCRUB_PROGRESS_SOCKET_PATH);
                if (prg_fd != -1) {
                        close(prg_fd);
                        prg_fd = -1;
@@ -1361,9 +1359,9 @@ static int scrub_start(int argc, char **argv, int resume)
                ret = scrub_write_progress(&spc_write_mutex, fsid, sp,
                                           fi_args.num_devices);
                if (ret) {
-                       ERR(!do_quiet, "WARNING: failed to write the progress "
-                           "status file: %s. Status recording disabled\n",
-                           strerror(-ret));
+                       warning_on(!do_quiet,
+   "failed to write the progress status file: %s. Status recording disabled",
+                               strerror(-ret));
                        do_record = 0;
                }
        }
@@ -1371,8 +1369,7 @@ static int scrub_start(int argc, char **argv, int resume)
        if (do_background) {
                pid = fork();
                if (pid == -1) {
-                       ERR(!do_quiet, "ERROR: cannot scrub, fork failed: "
-                                       "%s\n", strerror(errno));
+                       error_on(!do_quiet, "cannot scrub, fork failed: %m");
                        err = 1;
                        goto out;
                }
@@ -1390,13 +1387,13 @@ static int scrub_start(int argc, char **argv, int resume)
                        }
                        ret = wait(&stat);
                        if (ret != pid) {
-                               ERR(!do_quiet, "ERROR: wait failed: (ret=%d) "
-                                   "%s\n", ret, strerror(errno));
+                               error_on(!do_quiet, "wait failed (ret=%d): %m",
+                                       ret);
                                err = 1;
                                goto out;
                        }
                        if (!WIFEXITED(stat) || WEXITSTATUS(stat)) {
-                               ERR(!do_quiet, "ERROR: scrub process failed\n");
+                               error_on(!do_quiet, "scrub process failed");
                                err = WIFEXITED(stat) ? WEXITSTATUS(stat) : -1;
                                goto out;
                        }
@@ -1422,9 +1419,8 @@ static int scrub_start(int argc, char **argv, int resume)
                                        scrub_one_dev, &sp[i]);
                if (ret) {
                        if (do_print)
-                               fprintf(stderr, "ERROR: creating "
-                                       "scrub_one_dev[%llu] thread failed: "
-                                       "%s\n", devid, strerror(ret));
+                       error("creating scrub_one_dev[%llu] thread failed: %s",
+                               devid, strerror(ret));
                        err = 1;
                        goto out;
                }
@@ -1439,8 +1435,8 @@ static int scrub_start(int argc, char **argv, int resume)
        ret = pthread_create(&t_prog, NULL, scrub_progress_cycle, &spc);
        if (ret) {
                if (do_print)
-                       fprintf(stderr, "ERROR: creating progress thread "
-                               "failed: %s\n", strerror(ret));
+                       error("creating progress thread failed: %s",
+                               strerror(ret));
                err = 1;
                goto out;
        }
@@ -1453,27 +1449,30 @@ static int scrub_start(int argc, char **argv, int resume)
                ret = pthread_join(t_devs[i], NULL);
                if (ret) {
                        if (do_print)
-                               fprintf(stderr, "ERROR: pthread_join failed "
-                                       "for scrub_one_dev[%llu]: %s\n", devid,
-                                       strerror(ret));
+                         error("pthread_join failed for scrub_one_dev[%llu]: %s",
+                               devid, strerror(ret));
                        ++err;
                        continue;
                }
-               if (sp[i].ret && sp[i].ioctl_errno == ENODEV) {
-                       if (do_print)
-                               fprintf(stderr, "WARNING: device %lld not "
-                                       "present\n", devid);
-                       continue;
-               }
-               if (sp[i].ret && sp[i].ioctl_errno == ECANCELED) {
-                       ++err;
-               } else if (sp[i].ret) {
-                       if (do_print)
-                               fprintf(stderr, "ERROR: scrubbing %s failed "
-                                       "for device id %lld (%s)\n", path,
-                                       devid, strerror(sp[i].ioctl_errno));
-                       ++err;
-                       continue;
+               if (sp[i].ret) {
+                       switch (sp[i].ioctl_errno) {
+                       case ENODEV:
+                               if (do_print)
+                                       warning("device %lld not present",
+                                               devid);
+                               continue;
+                       case ECANCELED:
+                               ++err;
+                               break;
+                       default:
+                               if (do_print)
+               error("scrubbing %s failed for device id %lld: ret=%d, errno=%d (%s)",
+                                       path, devid,
+                                       sp[i].ret, sp[i].ioctl_errno,
+                                       strerror(sp[i].ioctl_errno));
+                               ++err;
+                               continue;
+                       }
                }
                if (sp[i].scrub_args.progress.uncorrectable_errors > 0)
                        e_uncorrectable++;
@@ -1512,23 +1511,21 @@ static int scrub_start(int argc, char **argv, int resume)
 
        /* check for errors from the handling of the progress thread */
        if (do_print && ret) {
-               fprintf(stderr, "ERROR: progress thread handling failed: %s\n",
+               error("progress thread handling failed: %s",
                        strerror(ret));
        }
 
        /* check for errors returned from the progress thread itself */
-       if (do_print && terr && terr != PTHREAD_CANCELED) {
-               fprintf(stderr, "ERROR: recording progress "
-                       "failed: %s\n", strerror(-PTR_ERR(terr)));
-       }
+       if (do_print && terr && terr != PTHREAD_CANCELED)
+               error("recording progress failed: %s",
+                       strerror(-PTR_ERR(terr)));
 
        if (do_record) {
                ret = scrub_write_progress(&spc_write_mutex, fsid, sp,
                                           fi_args.num_devices);
-               if (ret && do_print) {
-                       fprintf(stderr, "ERROR: failed to record the result: "
-                               "%s\n", strerror(-ret));
-               }
+               if (ret && do_print)
+                       error("failed to record the result: %s",
+                               strerror(-ret));
        }
 
        scrub_handle_sigint_child(-1);
@@ -1551,11 +1548,12 @@ out:
        if (nothing_to_resume)
                return 2;
        if (e_uncorrectable) {
-               ERR(!do_quiet, "ERROR: There are uncorrectable errors.\n");
+               error_on(!do_quiet, "there are uncorrectable errors");
                return 3;
        }
        if (e_correctable)
-               ERR(!do_quiet, "WARNING: errors detected during scrubbing, corrected.\n");
+               warning_on(!do_quiet,
+                       "errors detected during scrubbing, corrected");
 
        return 0;
 }
@@ -1594,20 +1592,15 @@ static int cmd_scrub_cancel(int argc, char **argv)
        int fdmnt = -1;
        DIR *dirstream = NULL;
 
-       if (check_argc_exact(argc, 2))
+       clean_args_no_options(argc, argv, cmd_scrub_cancel_usage);
+
+       if (check_argc_exact(argc - optind, 1))
                usage(cmd_scrub_cancel_usage);
 
-       path = argv[1];
+       path = argv[optind];
 
-       fdmnt = open_path_or_dev_mnt(path, &dirstream);
+       fdmnt = open_path_or_dev_mnt(path, &dirstream, 1);
        if (fdmnt < 0) {
-               if (errno == EINVAL)
-                       fprintf(stderr,
-                               "ERROR: '%s' is not a mounted btrfs device\n",
-                               path);
-               else
-                       fprintf(stderr, "ERROR: can't access '%s': %s\n",
-                               path, strerror(errno));
                ret = 1;
                goto out;
        }
@@ -1615,7 +1608,7 @@ static int cmd_scrub_cancel(int argc, char **argv)
        ret = ioctl(fdmnt, BTRFS_IOC_SCRUB_CANCEL, NULL);
 
        if (ret < 0) {
-               fprintf(stderr, "ERROR: scrub cancel failed on %s: %s\n", path,
+               error("scrub cancel failed on %s: %s", path,
                        errno == ENOTCONN ? "not running" : strerror(errno));
                if (errno == ENOTCONN)
                        ret = 2;
@@ -1683,7 +1676,6 @@ static int cmd_scrub_status(int argc, char **argv)
        int err = 0;
        DIR *dirstream = NULL;
 
-       optind = 1;
        while ((c = getopt(argc, argv, "dR")) != -1) {
                switch (c) {
                case 'd':
@@ -1703,28 +1695,19 @@ static int cmd_scrub_status(int argc, char **argv)
 
        path = argv[optind];
 
-       fdmnt = open_path_or_dev_mnt(path, &dirstream);
-
-       if (fdmnt < 0) {
-               if (errno == EINVAL)
-                       fprintf(stderr,
-                               "ERROR: '%s' is not a mounted btrfs device\n",
-                               path);
-               else
-                       fprintf(stderr, "ERROR: can't access '%s': %s\n",
-                               path, strerror(errno));
+       fdmnt = open_path_or_dev_mnt(path, &dirstream, 1);
+       if (fdmnt < 0)
                return 1;
-       }
 
        ret = get_fs_info(path, &fi_args, &di_args);
        if (ret) {
-               fprintf(stderr, "ERROR: getting dev info for scrub failed: "
-                               "%s\n", strerror(-ret));
+               error("getting dev info for scrub failed: %s",
+                       strerror(-ret));
                err = 1;
                goto out;
        }
        if (!fi_args.num_devices) {
-               fprintf(stderr, "ERROR: no devices found\n");
+               error("no devices found");
                err = 1;
                goto out;
        }
@@ -1733,9 +1716,7 @@ static int cmd_scrub_status(int argc, char **argv)
 
        fdres = socket(AF_UNIX, SOCK_STREAM, 0);
        if (fdres == -1) {
-               fprintf(stderr, "ERROR: failed to create socket to "
-                       "receive progress information: %s\n",
-                       strerror(errno));
+               error("failed to create socket to receive progress information: %m");
                err = 1;
                goto out;
        }
@@ -1748,8 +1729,8 @@ static int cmd_scrub_status(int argc, char **argv)
                close(fdres);
                fdres = scrub_open_file_r(SCRUB_DATA_FILE, fsid);
                if (fdres < 0 && fdres != -ENOENT) {
-                       fprintf(stderr, "WARNING: failed to open status file: "
-                               "%s\n", strerror(-fdres));
+                       warning("failed to open status file: %s",
+                               strerror(-fdres));
                        err = 1;
                        goto out;
                }
@@ -1758,7 +1739,7 @@ static int cmd_scrub_status(int argc, char **argv)
        if (fdres >= 0) {
                past_scrubs = scrub_read_file(fdres, 1);
                if (IS_ERR(past_scrubs))
-                       fprintf(stderr, "WARNING: failed to read status: %s\n",
+                       warning("failed to read status: %s",
                                strerror(-PTR_ERR(past_scrubs)));
        }
        in_progress = is_scrub_running_in_kernel(fdmnt, di_args, fi_args.num_devices);