Commit 95047b05 authored by Kees Cook's avatar Kees Cook

pstore: Refactor compression initialization

This refactors compression initialization slightly to better handle
getting potentially called twice (via early pstore_register() calls
and later pstore_init()) and improves the comments and reporting to be
more verbose.
Signed-off-by: default avatarKees Cook <keescook@chromium.org>
Tested-by: default avatarGuenter Roeck <groeck@chromium.org>
parent 41603165
...@@ -274,36 +274,56 @@ static int pstore_decompress(void *in, void *out, ...@@ -274,36 +274,56 @@ static int pstore_decompress(void *in, void *out,
static void allocate_buf_for_compression(void) static void allocate_buf_for_compression(void)
{ {
struct crypto_comp *ctx;
int size;
char *buf;
/* Skip if not built-in or compression backend not selected yet. */
if (!IS_ENABLED(CONFIG_PSTORE_COMPRESS) || !zbackend) if (!IS_ENABLED(CONFIG_PSTORE_COMPRESS) || !zbackend)
return; return;
/* Skip if no pstore backend yet or compression init already done. */
if (!psinfo || tfm)
return;
if (!crypto_has_comp(zbackend->name, 0, 0)) { if (!crypto_has_comp(zbackend->name, 0, 0)) {
pr_err("No %s compression\n", zbackend->name); pr_err("Unknown compression: %s\n", zbackend->name);
return; return;
} }
big_oops_buf_sz = zbackend->zbufsize(psinfo->bufsize); size = zbackend->zbufsize(psinfo->bufsize);
if (big_oops_buf_sz <= 0) if (size <= 0) {
pr_err("Invalid compression size for %s: %d\n",
zbackend->name, size);
return; return;
}
big_oops_buf = kmalloc(big_oops_buf_sz, GFP_KERNEL); buf = kmalloc(size, GFP_KERNEL);
if (!big_oops_buf) { if (!buf) {
pr_err("allocate compression buffer error!\n"); pr_err("Failed %d byte compression buffer allocation for: %s\n",
size, zbackend->name);
return; return;
} }
tfm = crypto_alloc_comp(zbackend->name, 0, 0); ctx = crypto_alloc_comp(zbackend->name, 0, 0);
if (IS_ERR_OR_NULL(tfm)) { if (IS_ERR_OR_NULL(ctx)) {
kfree(big_oops_buf); kfree(buf);
big_oops_buf = NULL; pr_err("crypto_alloc_comp('%s') failed: %ld\n", zbackend->name,
pr_err("crypto_alloc_comp() failed!\n"); PTR_ERR(ctx));
return; return;
} }
/* A non-NULL big_oops_buf indicates compression is available. */
tfm = ctx;
big_oops_buf_sz = size;
big_oops_buf = buf;
pr_info("Using compression: %s\n", zbackend->name);
} }
static void free_buf_for_compression(void) static void free_buf_for_compression(void)
{ {
if (IS_ENABLED(CONFIG_PSTORE_COMPRESS) && !IS_ERR_OR_NULL(tfm)) if (IS_ENABLED(CONFIG_PSTORE_COMPRESS) && tfm)
crypto_free_comp(tfm); crypto_free_comp(tfm);
kfree(big_oops_buf); kfree(big_oops_buf);
big_oops_buf = NULL; big_oops_buf = NULL;
...@@ -774,7 +794,6 @@ void __init pstore_choose_compression(void) ...@@ -774,7 +794,6 @@ void __init pstore_choose_compression(void)
for (step = zbackends; step->name; step++) { for (step = zbackends; step->name; step++) {
if (!strcmp(compress, step->name)) { if (!strcmp(compress, step->name)) {
zbackend = step; zbackend = step;
pr_info("using %s compression\n", zbackend->name);
return; return;
} }
} }
...@@ -791,7 +810,6 @@ static int __init pstore_init(void) ...@@ -791,7 +810,6 @@ static int __init pstore_init(void)
* initialize compression because crypto was not ready. If so, * initialize compression because crypto was not ready. If so,
* initialize compression now. * initialize compression now.
*/ */
if (psinfo && !tfm)
allocate_buf_for_compression(); allocate_buf_for_compression();
ret = pstore_init_fs(); ret = pstore_init_fs();
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment