From: Haoqin Huang Writing dict or out-of-range level for backends that don't support them is currently silently accepted but has no effect. For example, "algo=lzo dict=/data/dict" succeeds but lzo_setup_params() ignores the dict entirely. Add caps, level_min and level_max to zcomp_ops and validate user-supplied parameters in algorithm_params_store() before storing, printing an error on failure so the user knows what went wrong. Signed-off-by: Haoqin Huang Signed-off-by: Rongwei Wang --- drivers/block/zram/backend_842.c | 1 + drivers/block/zram/backend_deflate.c | 3 +++ drivers/block/zram/backend_lz4.c | 3 +++ drivers/block/zram/backend_lz4hc.c | 3 +++ drivers/block/zram/backend_lzo.c | 1 + drivers/block/zram/backend_lzorle.c | 1 + drivers/block/zram/backend_zstd.c | 3 +++ drivers/block/zram/zcomp.c | 23 +++++++++++++++++++++++ drivers/block/zram/zcomp.h | 8 ++++++++ drivers/block/zram/zram_drv.c | 19 +++++++++++++++++++ 10 files changed, 65 insertions(+) diff --git a/drivers/block/zram/backend_842.c b/drivers/block/zram/backend_842.c index 10d9d5c60f53..d796ebda1fa0 100644 --- a/drivers/block/zram/backend_842.c +++ b/drivers/block/zram/backend_842.c @@ -57,5 +57,6 @@ const struct zcomp_ops backend_842 = { .destroy_ctx = destroy_842, .setup_params = setup_params_842, .release_params = release_params_842, + .caps = 0, .name = "842", }; diff --git a/drivers/block/zram/backend_deflate.c b/drivers/block/zram/backend_deflate.c index f92a52a720d1..cedc3daad33a 100644 --- a/drivers/block/zram/backend_deflate.c +++ b/drivers/block/zram/backend_deflate.c @@ -144,5 +144,8 @@ const struct zcomp_ops backend_deflate = { .destroy_ctx = deflate_destroy, .setup_params = deflate_setup_params, .release_params = deflate_release_params, + .caps = ZCOMP_CAP_LEVEL, + .level_min = Z_DEFAULT_COMPRESSION, + .level_max = Z_BEST_COMPRESSION, .name = "deflate", }; diff --git a/drivers/block/zram/backend_lz4.c b/drivers/block/zram/backend_lz4.c index c449d511ba86..bd1e5ca4d134 100644 --- a/drivers/block/zram/backend_lz4.c +++ b/drivers/block/zram/backend_lz4.c @@ -146,5 +146,8 @@ const struct zcomp_ops backend_lz4 = { .destroy_ctx = lz4_destroy, .setup_params = lz4_setup_params, .release_params = lz4_release_params, + .caps = ZCOMP_CAP_DICT | ZCOMP_CAP_LEVEL, + .level_min = LZ4_ACCELERATION_DEFAULT, + .level_max = 65535, .name = "lz4", }; diff --git a/drivers/block/zram/backend_lz4hc.c b/drivers/block/zram/backend_lz4hc.c index f6a336acfe20..0e0d7c68a7d4 100644 --- a/drivers/block/zram/backend_lz4hc.c +++ b/drivers/block/zram/backend_lz4hc.c @@ -124,5 +124,8 @@ const struct zcomp_ops backend_lz4hc = { .destroy_ctx = lz4hc_destroy, .setup_params = lz4hc_setup_params, .release_params = lz4hc_release_params, + .caps = ZCOMP_CAP_DICT | ZCOMP_CAP_LEVEL, + .level_min = LZ4HC_MIN_CLEVEL, + .level_max = LZ4HC_MAX_CLEVEL, .name = "lz4hc", }; diff --git a/drivers/block/zram/backend_lzo.c b/drivers/block/zram/backend_lzo.c index 4c906beaae6b..965f007e2ca8 100644 --- a/drivers/block/zram/backend_lzo.c +++ b/drivers/block/zram/backend_lzo.c @@ -55,5 +55,6 @@ const struct zcomp_ops backend_lzo = { .destroy_ctx = lzo_destroy, .setup_params = lzo_setup_params, .release_params = lzo_release_params, + .caps = 0, .name = "lzo", }; diff --git a/drivers/block/zram/backend_lzorle.c b/drivers/block/zram/backend_lzorle.c index 10640c96cbfc..757b4598be03 100644 --- a/drivers/block/zram/backend_lzorle.c +++ b/drivers/block/zram/backend_lzorle.c @@ -55,5 +55,6 @@ const struct zcomp_ops backend_lzorle = { .destroy_ctx = lzorle_destroy, .setup_params = lzorle_setup_params, .release_params = lzorle_release_params, + .caps = 0, .name = "lzo-rle", }; diff --git a/drivers/block/zram/backend_zstd.c b/drivers/block/zram/backend_zstd.c index 2584f47c9b3c..0fbd2460883a 100644 --- a/drivers/block/zram/backend_zstd.c +++ b/drivers/block/zram/backend_zstd.c @@ -212,5 +212,8 @@ const struct zcomp_ops backend_zstd = { .destroy_ctx = zstd_destroy, .setup_params = zstd_setup_params, .release_params = zstd_release_params, + .caps = ZCOMP_CAP_DICT | ZCOMP_CAP_LEVEL, + .level_min = (int)-ZSTD_TARGETLENGTH_MAX, + .level_max = ZSTD_MAX_CLEVEL, .name = "zstd", }; diff --git a/drivers/block/zram/zcomp.c b/drivers/block/zram/zcomp.c index 974c4691887e..15de28b50d42 100644 --- a/drivers/block/zram/zcomp.c +++ b/drivers/block/zram/zcomp.c @@ -9,6 +9,9 @@ #include #include #include +#include +#include +#include #include "zcomp.h" @@ -94,6 +97,26 @@ const char *zcomp_lookup_backend_name(const char *comp) return NULL; } +unsigned int zcomp_get_caps(const char *comp) +{ + const struct zcomp_ops *backend = lookup_backend_ops(comp); + + return backend ? backend->caps : 0; +} + +int zcomp_validate_level(const char *comp, s32 level) +{ + const struct zcomp_ops *backend = lookup_backend_ops(comp); + + if (!backend) + return -EINVAL; + if (!(backend->caps & ZCOMP_CAP_LEVEL)) + return -EOPNOTSUPP; + if (level < backend->level_min || level > backend->level_max) + return -EINVAL; + return 0; +} + /* show available compressors */ ssize_t zcomp_available_show(const char *comp, char *buf, ssize_t at) { diff --git a/drivers/block/zram/zcomp.h b/drivers/block/zram/zcomp.h index 81a0f3f6ff48..16f812a94c67 100644 --- a/drivers/block/zram/zcomp.h +++ b/drivers/block/zram/zcomp.h @@ -7,6 +7,9 @@ #define ZCOMP_PARAM_NOT_SET INT_MIN +#define ZCOMP_CAP_DICT BIT(0) /* dictionary support */ +#define ZCOMP_CAP_LEVEL BIT(1) /* adjustable compression level */ + struct deflate_params { s32 winbits; }; @@ -66,6 +69,9 @@ struct zcomp_ops { int (*setup_params)(struct zcomp_params *params); void (*release_params)(struct zcomp_params *params); + unsigned int caps; + s32 level_min; + s32 level_max; const char *name; }; @@ -81,6 +87,8 @@ int zcomp_cpu_up_prepare(unsigned int cpu, struct hlist_node *node); int zcomp_cpu_dead(unsigned int cpu, struct hlist_node *node); ssize_t zcomp_available_show(const char *comp, char *buf, ssize_t at); const char *zcomp_lookup_backend_name(const char *comp); +unsigned int zcomp_get_caps(const char *comp); +int zcomp_validate_level(const char *comp, s32 level); struct zcomp *zcomp_create(const char *alg, struct zcomp_params *params); void zcomp_destroy(struct zcomp *comp); diff --git a/drivers/block/zram/zram_drv.c b/drivers/block/zram/zram_drv.c index 9ea7ba9d1ed0..d86f4eb06d58 100644 --- a/drivers/block/zram/zram_drv.c +++ b/drivers/block/zram/zram_drv.c @@ -1796,6 +1796,25 @@ static ssize_t algorithm_params_store(struct device *dev, return -EINVAL; } + if (zram->comp_algs[prio]) { + unsigned int caps = zcomp_get_caps(zram->comp_algs[prio]); + + if (dict_path && !(caps & ZCOMP_CAP_DICT)) { + pr_err("zram: %s does not support dictionary\n", + zram->comp_algs[prio]); + return -EOPNOTSUPP; + } + + if (level != ZCOMP_PARAM_NOT_SET) { + ret = zcomp_validate_level(zram->comp_algs[prio], level); + if (ret) { + pr_err("zram: invalid level for %s\n", + zram->comp_algs[prio]); + return ret; + } + } + } + ret = comp_params_store(zram, prio, level, dict_path, &deflate_params); return ret ? ret : len; } -- 2.43.7