0
0
mirror of https://github.com/vim/vim.git synced 2025-09-24 03:44:06 -04:00

patch 8.0.0167: str2nr()/str2float() fail with negative values

Problem:    str2nr() and str2float() do not always work with negative values.
Solution:   Be more flexible about handling signs. (LemonBoy, closes #1332)
            Add more tests.
This commit is contained in:
Bram Moolenaar
2017-01-10 16:12:29 +01:00
parent 03c60c1573
commit 08243d26d2
6 changed files with 47 additions and 3 deletions

View File

@@ -2091,6 +2091,7 @@ test_arglist \
test_delete \ test_delete \
test_diffmode \ test_diffmode \
test_digraph \ test_digraph \
test_functions \
test_display \ test_display \
test_ex_undo \ test_ex_undo \
test_execute_func \ test_execute_func \

View File

@@ -11066,10 +11066,13 @@ f_sqrt(typval_T *argvars, typval_T *rettv)
f_str2float(typval_T *argvars, typval_T *rettv) f_str2float(typval_T *argvars, typval_T *rettv)
{ {
char_u *p = skipwhite(get_tv_string(&argvars[0])); char_u *p = skipwhite(get_tv_string(&argvars[0]));
int isneg = (*p == '-');
if (*p == '+') if (*p == '+' || *p == '-')
p = skipwhite(p + 1); p = skipwhite(p + 1);
(void)string2float(p, &rettv->vval.v_float); (void)string2float(p, &rettv->vval.v_float);
if (isneg)
rettv->vval.v_float *= -1;
rettv->v_type = VAR_FLOAT; rettv->v_type = VAR_FLOAT;
} }
#endif #endif
@@ -11084,6 +11087,7 @@ f_str2nr(typval_T *argvars, typval_T *rettv)
char_u *p; char_u *p;
varnumber_T n; varnumber_T n;
int what; int what;
int isneg;
if (argvars[1].v_type != VAR_UNKNOWN) if (argvars[1].v_type != VAR_UNKNOWN)
{ {
@@ -11096,7 +11100,8 @@ f_str2nr(typval_T *argvars, typval_T *rettv)
} }
p = skipwhite(get_tv_string(&argvars[0])); p = skipwhite(get_tv_string(&argvars[0]));
if (*p == '+') isneg = (*p == '-');
if (*p == '+' || *p == '-')
p = skipwhite(p + 1); p = skipwhite(p + 1);
switch (base) switch (base)
{ {
@@ -11106,7 +11111,11 @@ f_str2nr(typval_T *argvars, typval_T *rettv)
default: what = 0; default: what = 0;
} }
vim_str2nr(p, NULL, NULL, what, &n, NULL, 0); vim_str2nr(p, NULL, NULL, what, &n, NULL, 0);
rettv->vval.v_number = n; if (isneg)
rettv->vval.v_number = -n;
else
rettv->vval.v_number = n;
} }
#ifdef HAVE_STRFTIME #ifdef HAVE_STRFTIME

View File

@@ -18,6 +18,7 @@ source test_filter_cmd.vim
source test_filter_map.vim source test_filter_map.vim
source test_float_func.vim source test_float_func.vim
source test_fnamemodify.vim source test_fnamemodify.vim
source test_functions.vim
source test_glob2regpat.vim source test_glob2regpat.vim
source test_goto.vim source test_goto.vim
source test_help_tagjump.vim source test_help_tagjump.vim

View File

@@ -165,9 +165,22 @@ endfunc
func Test_str2float() func Test_str2float()
call assert_equal('1.0', string(str2float('1'))) call assert_equal('1.0', string(str2float('1')))
call assert_equal('1.0', string(str2float(' 1 ')))
call assert_equal('1.0', string(str2float(' 1.0 ')))
call assert_equal('1.23', string(str2float('1.23'))) call assert_equal('1.23', string(str2float('1.23')))
call assert_equal('1.23', string(str2float('1.23abc'))) call assert_equal('1.23', string(str2float('1.23abc')))
call assert_equal('1.0e40', string(str2float('1e40'))) call assert_equal('1.0e40', string(str2float('1e40')))
call assert_equal('1.0', string(str2float('+1')))
call assert_equal('1.0', string(str2float('+1')))
call assert_equal('1.0', string(str2float(' +1 ')))
call assert_equal('1.0', string(str2float(' + 1 ')))
call assert_equal('-1.0', string(str2float('-1')))
call assert_equal('-1.0', string(str2float('-1')))
call assert_equal('-1.0', string(str2float(' -1 ')))
call assert_equal('-1.0', string(str2float(' - 1 ')))
call assert_equal('inf', string(str2float('1e1000'))) call assert_equal('inf', string(str2float('1e1000')))
call assert_equal('inf', string(str2float('inf'))) call assert_equal('inf', string(str2float('inf')))
call assert_equal('-inf', string(str2float('-inf'))) call assert_equal('-inf', string(str2float('-inf')))

View File

@@ -0,0 +1,18 @@
" Tests for various functions.
func Test_str2nr()
call assert_equal(0, str2nr(''))
call assert_equal(1, str2nr('1'))
call assert_equal(1, str2nr(' 1 '))
call assert_equal(1, str2nr('+1'))
call assert_equal(1, str2nr('+ 1'))
call assert_equal(1, str2nr(' + 1 '))
call assert_equal(-1, str2nr('-1'))
call assert_equal(-1, str2nr('- 1'))
call assert_equal(-1, str2nr(' - 1 '))
call assert_equal(123456789, str2nr('123456789'))
call assert_equal(-123456789, str2nr('-123456789'))
endfunc

View File

@@ -764,6 +764,8 @@ static char *(features[]) =
static int included_patches[] = static int included_patches[] =
{ /* Add new patch number below this line */ { /* Add new patch number below this line */
/**/
167,
/**/ /**/
166, 166,
/**/ /**/