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