unix/modos: getenv(): Handle non-existing envvar correctly.

pfalcon · pfalcon · commit ec314c951dbc · 2015-11-14T17:12:04.000+02:00
diff --git a/unix/modos.c b/unix/modos.c
@@ -137,6 +137,9 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_system_obj, mod_os_system);
 
 STATIC mp_obj_t mod_os_getenv(mp_obj_t var_in) {
     const char *s = getenv(mp_obj_str_get_str(var_in));
+    if (s == NULL) {
+        return mp_const_none;
+    }
     return mp_obj_new_str(s, strlen(s), false);
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_getenv_obj, mod_os_getenv);

Original file line number	Diff line number	Diff line change
`@@ -137,6 +137,9 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_system_obj, mod_os_system);`
`137`	`137`
`138`	`138`	`STATIC mp_obj_t mod_os_getenv(mp_obj_t var_in) {`
`139`	`139`	`const char *s = getenv(mp_obj_str_get_str(var_in));`
	`140`	`+ if (s == NULL) {`
	`141`	`+ return mp_const_none;`
	`142`	`+ }`
`140`	`143`	`return mp_obj_new_str(s, strlen(s), false);`
`141`	`144`	`}`
`142`	`145`	`STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_getenv_obj, mod_os_getenv);`