Now, files which don't have compession property won't be defraged with
compression.

So add an option '--compress-force' to extend -c to drop nocompress flag
on files.

If the option is enable and a file doesn't have compression property:
First add a compression property which is specified by option '-c'
(default as zlib). Then do defrag.

Suggested-by: David Sterba <dste...@suse.com>
Suggested-by: Anand Jain <anand.j...@oracle.com>
Signed-off-by: Su Yue <suy.f...@cn.fujitsu.com>
---
 cmds-filesystem.c | 94 +++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 88 insertions(+), 6 deletions(-)

diff --git a/cmds-filesystem.c b/cmds-filesystem.c
index 7728430f16a1..4bbbc86a3b3b 100644
--- a/cmds-filesystem.c
+++ b/cmds-filesystem.c
@@ -37,6 +37,7 @@
 #include "list_sort.h"
 #include "disk-io.h"
 #include "help.h"
+#include "props.h"
 
 /*
  * for btrfs fi show, we maintain a hash of fsids we've already printed.
@@ -855,6 +856,20 @@ static int parse_compress_type(char *s)
        };
 }
 
+static const char *compress_type_extract(int type)
+{
+       switch (type) {
+       case BTRFS_COMPRESS_ZLIB:
+               return "zlib";
+       case BTRFS_COMPRESS_LZO:
+               return "lzo";
+       case BTRFS_COMPRESS_ZSTD:
+               return "zstd";
+       }
+
+       return NULL;
+}
+
 static const char * const cmd_filesystem_defrag_usage[] = {
        "btrfs filesystem defragment [options] <file>|<dir> [<file>|<dir>...]",
        "Defragment a file or a directory",
@@ -867,6 +882,8 @@ static const char * const cmd_filesystem_defrag_usage[] = {
        "-l len              defragment only up to len bytes",
        "-t size             target extent size hint (default: 32M)",
        "",
+       "--compress-force    drop nocompress flag on files, only work with 
option -c",
+       "",
        "Warning: most Linux kernels will break up the ref-links of COW data",
        "(e.g., files copied with 'cp --reflink', snapshots) which may cause",
        "considerable increase of space usage. See btrfs-filesystem(8) for",
@@ -877,8 +894,49 @@ static const char * const cmd_filesystem_defrag_usage[] = {
 static struct btrfs_ioctl_defrag_range_args defrag_global_range;
 static int defrag_global_verbose;
 static int defrag_global_errors;
+static int defrag_global_compress_force;
+static int drop_nocompress_property(const char *fpath)
+{
+       const char *name = "compression";
+       const char *compress_type;
+       int type = prop_object_inode;
+       int ret = 0;
+       const struct prop_handler *prop;
+
+       for (prop = prop_handlers; prop->name; prop++) {
+               if (!strcmp(prop->name, name))
+                       break;
+       }
+       if (!prop->name) {
+               error("can't find handler of compression property");
+               ret = -1;
+               goto out;
+       }
+
+       ret = prop->getter(type, fpath, name, NULL);
+       if (ret) {
+               ret = 0;
+               goto out;
+       }
+
+       compress_type = compress_type_extract(
+               defrag_global_range.compress_type);
+       if (!compress_type) {
+               error("unknown compression type %d",
+                     defrag_global_range.compress_type);
+               ret = -1;
+               goto out;
+       }
+       ret = prop->setter(type, fpath, name, compress_type);
+out:
+       if (ret)
+               error("failed to drop nocompress property on file %s", fpath);
+
+       return ret;
+}
+
 static int defrag_callback(const char *fpath, const struct stat *sb,
-               int typeflag, struct FTW *ftwbuf)
+                          int typeflag, struct FTW *ftwbuf)
 {
        int ret = 0;
        int err = 0;
@@ -887,6 +945,11 @@ static int defrag_callback(const char *fpath, const struct 
stat *sb,
        if ((typeflag == FTW_F) && S_ISREG(sb->st_mode)) {
                if (defrag_global_verbose)
                        printf("%s\n", fpath);
+               if (defrag_global_compress_force) {
+                       ret = drop_nocompress_property(fpath);
+                       if (ret)
+                               goto error;
+               }
                fd = open(fpath, O_RDWR);
                if (fd < 0) {
                        err = errno;
@@ -926,6 +989,12 @@ static int cmd_filesystem_defrag(int argc, char **argv)
        int compress_type = BTRFS_COMPRESS_NONE;
        DIR *dirstream;
 
+       enum { GETOPT_VAL_COMPRESS_FORCE = 257};
+       static const struct option long_options[] = {
+               { "compress-force", no_argument, NULL,
+                 GETOPT_VAL_COMPRESS_FORCE},
+               { NULL, 0, NULL, 0}
+       };
        /*
         * Kernel has a different default (256K) that is supposed to be safe,
         * but it does not defragment very well. The 32M will likely lead to
@@ -937,8 +1006,10 @@ static int cmd_filesystem_defrag(int argc, char **argv)
        defrag_global_errors = 0;
        defrag_global_verbose = 0;
        defrag_global_errors = 0;
+       defrag_global_compress_force = 0;
        while(1) {
-               int c = getopt(argc, argv, "vrc::fs:l:t:");
+               int c = getopt_long(argc, argv, "vrc::fs:l:t:", long_options,
+                                   NULL);
                if (c < 0)
                        break;
 
@@ -972,6 +1043,9 @@ static int cmd_filesystem_defrag(int argc, char **argv)
                case 'r':
                        recursive = 1;
                        break;
+               case GETOPT_VAL_COMPRESS_FORCE:
+                       defrag_global_compress_force = 1;
+                       break;
                default:
                        usage(cmd_filesystem_defrag_usage);
                }
@@ -987,6 +1061,8 @@ static int cmd_filesystem_defrag(int argc, char **argv)
        if (compress_type) {
                defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_COMPRESS;
                defrag_global_range.compress_type = compress_type;
+       } else if (defrag_global_compress_force) {
+               warning("Option --compress-force only works for -c");
        }
        if (flush)
                defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_START_IO;
@@ -1049,7 +1125,7 @@ static int cmd_filesystem_defrag(int argc, char **argv)
                }
                if (recursive && S_ISDIR(st.st_mode)) {
                        ret = nftw(argv[i], defrag_callback, 10,
-                                               FTW_MOUNT | FTW_PHYS);
+                                  FTW_MOUNT | FTW_PHYS);
                        if (ret == ENOTTY)
                                exit(1);
                        /* errors are handled in the callback */
@@ -1057,9 +1133,15 @@ static int cmd_filesystem_defrag(int argc, char **argv)
                } else {
                        if (defrag_global_verbose)
                                printf("%s\n", argv[i]);
-                       ret = ioctl(fd, BTRFS_IOC_DEFRAG_RANGE,
-                                       &defrag_global_range);
-                       defrag_err = errno;
+                       if (defrag_global_compress_force)
+                               ret = drop_nocompress_property(argv[i]);
+                       if (ret) {
+                               defrag_err = -ret;
+                       } else {
+                               ret = ioctl(fd, BTRFS_IOC_DEFRAG_RANGE,
+                                           &defrag_global_range);
+                               defrag_err = errno;
+                       }
                }
                close_file_or_dir(fd, dirstream);
                if (ret && defrag_err == ENOTTY) {
-- 
2.15.0



--
To unsubscribe from this list: send the line "unsubscribe linux-btrfs" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to