static tb_bool_t tb_environment_set_impl(tb_char_t const* name, tb_char_t const* value)
{
// check
tb_assert_and_check_return_val(name, tb_false);
// done
tb_bool_t ok = tb_false;
tb_size_t size = 0;
tb_wchar_t* value_w = tb_null;
tb_size_t value_n = 0;
do
{
// make name
tb_wchar_t name_w[512] = {0};
tb_size_t name_n = tb_atow(name_w, name, tb_arrayn(name_w));
tb_assert_and_check_break(name_n);
// exists value?
if (value)
{
// make value
value_n = tb_strlen(value);
value_w = (tb_wchar_t*)tb_malloc0(sizeof(tb_wchar_t) * (value_n + 1));
tb_assert_and_check_break(value_w);
// init value
if (!tb_atow(value_w, value, value_n + 1)) break;
// set it
if (!tb_kernel32()->SetEnvironmentVariableW(name_w, value_w)) break;
}
// remove this variable
else
{
// remove it
if (!tb_kernel32()->SetEnvironmentVariableW(name_w, tb_null)) break;
}
// ok
ok = tb_true;
} while (0);
// exit data
if (value_w) tb_free(value_w);
value_w = tb_null;
// ok?
return ok;
}
/* //////////////////////////////////////////////////////////////////////////////////////
* private implementation
*/
static tb_char_t* tb_environment_get_impl(tb_char_t const* name, tb_size_t* psize)
{
// check
tb_assert_and_check_return_val(name, 0);
// done
tb_bool_t ok = tb_false;
tb_size_t size = 0;
tb_size_t maxn = 256;
tb_char_t* value = tb_null;
tb_wchar_t* value_w = tb_null;
do
{
// make value_w
value_w = (tb_wchar_t*)tb_malloc0(sizeof(tb_wchar_t) * maxn);
tb_assert_and_check_break(value_w);
// make name
tb_wchar_t name_w[512];
tb_size_t name_n = tb_atow(name_w, name, tb_arrayn(name_w));
tb_assert_and_check_break(name_n != -1);
// get it
size = (tb_size_t)tb_kernel32()->GetEnvironmentVariableW(name_w, value_w, (DWORD)maxn);
if (!size)
{
// error?
if (ERROR_ENVVAR_NOT_FOUND == GetLastError())
{
// trace
tb_trace_d("environment variable(%s) does not exist", name);
}
break;
}
else if (size > maxn)
{
// grow space
value_w = (tb_wchar_t*)tb_ralloc(value_w, sizeof(tb_wchar_t) * (size + 1));
tb_assert_and_check_break(value_w);
// get it
size = (tb_size_t)tb_kernel32()->GetEnvironmentVariableW(name_w, value_w, (DWORD)size + 1);
tb_assert_and_check_break(size);
}
// make value
value = (tb_char_t*)tb_malloc0(sizeof(tb_char_t) * (size + 1));
tb_assert_and_check_break(value);
// save value
if ((size = tb_wtoa(value, value_w, size)) == -1) break;
// save size
if (psize) *psize = size;
// ok
ok = tb_true;
} while (0);
// failed?
if (!ok)
{
// exit value
if (value) tb_free(value);
value = tb_null;
}
// exit value_w
if (value_w) tb_free(value_w);
value_w = tb_null;
// ok?
return value;
}