diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/argv/ntapi_tt_env_vars.c | 34 | ||||
| -rw-r--r-- | src/internal/ntapi.c | 1 | ||||
| -rw-r--r-- | src/internal/ntapi_fnapi.h | 1 |
3 files changed, 36 insertions, 0 deletions
diff --git a/src/argv/ntapi_tt_env_vars.c b/src/argv/ntapi_tt_env_vars.c index 9c62eb5..9b671b6 100644 --- a/src/argv/ntapi_tt_env_vars.c +++ b/src/argv/ntapi_tt_env_vars.c @@ -8,6 +8,40 @@ #include <ntapi/ntapi.h> #include "ntapi_impl.h" +int32_t __stdcall __ntapi_tt_get_env_var_meta_utf8( + __in char * env_var_name, + __in char ** envp, + __out nt_env_var_meta_utf8 * env_var_meta) +{ + char ** penv; + char * ch; + + /* lookup */ + for (penv=envp; *penv; penv++) { + for (ch=penv[0]; *ch && (*ch != '='); ) + ch++; + + if (*ch != '=') + return NT_STATUS_INVALID_USER_BUFFER; + + if ((ch > *penv) && !(__ntapi->tt_strncmp_multibyte( + *penv, + env_var_name, + ch - *penv))) { + /* match */ + env_var_meta->name = *penv; + env_var_meta->value = ++ch; + env_var_meta->envp_index = penv - envp; + env_var_meta->flags = 0; + + return NT_STATUS_SUCCESS; + } + } + + return NT_STATUS_NOT_FOUND; +} + + int32_t __stdcall __ntapi_tt_get_env_var_meta_utf16( __in wchar16_t * env_var_name, __in wchar16_t ** envp, diff --git a/src/internal/ntapi.c b/src/internal/ntapi.c index 0e9e6e0..1047606 100644 --- a/src/internal/ntapi.c +++ b/src/internal/ntapi.c @@ -311,6 +311,7 @@ static int32_t __fastcall __ntapi_init_once(ntapi_vtbl ** pvtbl) __ntapi->tt_parse_cmd_line_args_utf16 = __ntapi_tt_parse_cmd_line_args_utf16; __ntapi->tt_get_argv_envp_utf8 = __ntapi_tt_get_argv_envp_utf8; __ntapi->tt_get_argv_envp_utf16 = __ntapi_tt_get_argv_envp_utf16; + __ntapi->tt_get_env_var_meta_utf8 = __ntapi_tt_get_env_var_meta_utf8; __ntapi->tt_get_env_var_meta_utf16 = __ntapi_tt_get_env_var_meta_utf16; __ntapi->tt_array_copy_utf16 = __ntapi_tt_array_copy_utf16; __ntapi->tt_array_copy_utf8 = __ntapi_tt_array_copy_utf8; diff --git a/src/internal/ntapi_fnapi.h b/src/internal/ntapi_fnapi.h index 8bafdbc..05ff9ab 100644 --- a/src/internal/ntapi_fnapi.h +++ b/src/internal/ntapi_fnapi.h @@ -182,6 +182,7 @@ ntapi_tt_get_peb_env_block_utf16 __ntapi_tt_get_peb_env_block_utf16; ntapi_tt_parse_cmd_line_args_utf16 __ntapi_tt_parse_cmd_line_args_utf16; ntapi_tt_get_argv_envp_utf8 __ntapi_tt_get_argv_envp_utf8; ntapi_tt_get_argv_envp_utf16 __ntapi_tt_get_argv_envp_utf16; +ntapi_tt_get_env_var_meta_utf8 __ntapi_tt_get_env_var_meta_utf8; ntapi_tt_get_env_var_meta_utf16 __ntapi_tt_get_env_var_meta_utf16; ntapi_tt_array_copy_utf8 __ntapi_tt_array_copy_utf8; ntapi_tt_array_copy_utf16 __ntapi_tt_array_copy_utf16; |