f_environ: cleanup/refactor

- use os_getenvname_at_index / os_getenv
- f_getenv: empty (*p == NUL) is not null (undefined)
This commit is contained in:
Daniel Hahler 2019-07-30 11:55:55 +02:00 committed by Justin M. Keyes
parent fd66ad2262
commit d55b12ea50
3 changed files with 24 additions and 41 deletions

View File

@ -8498,48 +8498,19 @@ static void f_empty(typval_T *argvars, typval_T *rettv, FunPtr fptr)
/// "environ()" function /// "environ()" function
static void f_environ(typval_T *argvars, typval_T *rettv, FunPtr fptr) static void f_environ(typval_T *argvars, typval_T *rettv, FunPtr fptr)
{ {
int i = 0;
char_u *entry, *value;
# ifdef WIN32
extern wchar_t **_wenviron;
# else
extern char **environ;
# endif
tv_dict_alloc_ret(rettv); tv_dict_alloc_ret(rettv);
# ifdef WIN32 for (int i = 0; ; i++) {
if (*_wenviron == NULL) { // TODO(justinmk): use os_copyfullenv from #7202 ?
return; char *envname = os_getenvname_at_index((size_t)i);
if (envname == NULL) {
break;
} }
# else const char *value = os_getenv(envname);
if (*environ == NULL) { tv_dict_add_str(rettv->vval.v_dict,
return; (char *)envname, STRLEN((char *)envname),
} value == NULL ? "" : value);
# endif xfree(envname);
for (i = 0; ; i++) {
# ifdef WIN32
uint16_t *p;
if ((p = (uint16_t *)_wenviron[i]) == NULL) {
return;
}
entry = utf16_to_enc(p, NULL);
# else
if ((entry = (char_u *)environ[i]) == NULL) {
return;
}
entry = vim_strsave(entry);
# endif
if ((value = vim_strchr(entry, '=')) == NULL) {
xfree(entry);
continue;
}
*value++ = NUL;
tv_dict_add_str(rettv->vval.v_dict, (char *)entry, STRLEN((char *)entry),
(const char *)value);
xfree(entry);
} }
} }
@ -8561,12 +8532,11 @@ static void f_getenv(typval_T *argvars, typval_T *rettv, FunPtr fptr)
{ {
char_u *p = (char_u *)vim_getenv(tv_get_string(&argvars[0])); char_u *p = (char_u *)vim_getenv(tv_get_string(&argvars[0]));
if (p == NULL || *p == NUL) { if (p == NULL) {
rettv->v_type = VAR_SPECIAL; rettv->v_type = VAR_SPECIAL;
rettv->vval.v_number = kSpecialVarNull; rettv->vval.v_number = kSpecialVarNull;
return; return;
} }
p = vim_strsave(p);
rettv->vval.v_string = p; rettv->vval.v_string = p;
rettv->v_type = VAR_STRING; rettv->v_type = VAR_STRING;
} }

View File

@ -45,6 +45,7 @@ void env_init(void)
} }
/// Like getenv(), but returns NULL if the variable is empty. /// Like getenv(), but returns NULL if the variable is empty.
/// @see os_env_exists
const char *os_getenv(const char *name) const char *os_getenv(const char *name)
FUNC_ATTR_NONNULL_ALL FUNC_ATTR_NONNULL_ALL
{ {

View File

@ -0,0 +1,12 @@
local helpers = require('test.functional.helpers')(after_each)
local clear = helpers.clear
local eq = helpers.eq
local environ = helpers.funcs.environ
describe('environ()', function()
it('handles empty env variable', function()
clear({env={EMPTY_VAR=""}})
eq("", environ()['EMPTY_VAR'])
eq(nil, environ()['DOES_NOT_EXIST'])
end)
end)