summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/common/comp_opt.c47
-rw-r--r--lib/common/parse_size.c12
2 files changed, 17 insertions, 42 deletions
diff --git a/lib/common/comp_opt.c b/lib/common/comp_opt.c
index f257036..66d562e 100644
--- a/lib/common/comp_opt.c
+++ b/lib/common/comp_opt.c
@@ -74,39 +74,6 @@ static int find_lzo_alg(sqfs_compressor_config_t *cfg, const char *name)
return -1;
}
-static int get_size_value(const char *value, sqfs_u32 *out, sqfs_u32 block_size)
-{
- int i;
-
- for (i = 0; isdigit(value[i]) && i < 9; ++i)
- ;
-
- if (i < 1 || i > 9)
- return -1;
-
- *out = atol(value);
-
- switch (value[i]) {
- case '\0':
- break;
- case 'm':
- case 'M':
- *out <<= 20;
- break;
- case 'k':
- case 'K':
- *out <<= 10;
- break;
- case '%':
- *out = (*out * block_size) / 100;
- break;
- default:
- return -1;
- }
-
- return 0;
-}
-
enum {
OPT_WINDOW = 0,
OPT_LEVEL,
@@ -125,7 +92,7 @@ int compressor_cfg_init_options(sqfs_compressor_config_t *cfg,
E_SQFS_COMPRESSOR id,
size_t block_size, char *options)
{
- size_t num_flags = 0, min_level = 0, max_level = 0, level;
+ size_t num_flags = 0, min_level = 0, max_level = 0, level, dict_size;
const flag_t *flags = NULL;
char *subopts, *value;
int i, opt;
@@ -234,10 +201,12 @@ int compressor_cfg_init_options(sqfs_compressor_config_t *cfg,
if (value == NULL)
goto fail_value;
- if (get_size_value(value, &cfg->opt.xz.dict_size,
- cfg->block_size)) {
- goto fail_dict;
+ if (parse_size("Parsing XZ dictionary size",
+ &dict_size, value, cfg->block_size)) {
+ return -1;
}
+
+ cfg->opt.xz.dict_size = dict_size;
break;
default:
if (set_flag(cfg, value, flags, num_flags))
@@ -265,10 +234,6 @@ fail_value:
fprintf(stderr, "Missing value for compressor option '%s'.\n",
token[opt]);
return -1;
-fail_dict:
- fputs("Dictionary size must be a number with the optional "
- "suffix 'm','k' or '%'.\n", stderr);
- return -1;
}
typedef void (*compressor_help_fun_t)(void);
diff --git a/lib/common/parse_size.c b/lib/common/parse_size.c
index aede38a..1285d3b 100644
--- a/lib/common/parse_size.c
+++ b/lib/common/parse_size.c
@@ -9,7 +9,8 @@
#include <ctype.h>
#include <limits.h>
-int parse_size(const char *what, size_t *out, const char *str)
+int parse_size(const char *what, size_t *out, const char *str,
+ size_t reference)
{
const char *in = str;
size_t acc = 0, x;
@@ -49,6 +50,15 @@ int parse_size(const char *what, size_t *out, const char *str)
acc <<= 30;
++in;
break;
+ case '%':
+ if (reference == 0)
+ goto fail_suffix;
+
+ if (SZ_MUL_OV(acc, reference, &acc))
+ goto fail_ov;
+
+ acc /= 100;
+ break;
case '\0':
break;
default: