env-posix.cpp revision a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fc
/* $Id$ */
/** @file
* innotek Portable Runtime - Environment, Posix.
*/
/*
* Copyright (C) 2006-2007 innotek GmbH
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* you can redistribute it and/or modify it under the terms of the GNU
* General Public License (GPL) as published by the Free Software
* Foundation, in version 2 as it comes in the "COPYING" file of the
* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
*
* The contents of this file may alternatively be used under the terms
* of the Common Development and Distribution License Version 1.0
* (CDDL) only, as it comes in the "COPYING.CDDL" file of the
* VirtualBox OSE distribution, in which case the provisions of the
* CDDL are applicable instead of those of the GPL.
*
* You may elect to license modified versions of this file under the
* terms and conditions of either the GPL or the CDDL or both.
*/
/*******************************************************************************
* Header Files *
*******************************************************************************/
#include <iprt/env.h>
#include <iprt/string.h>
#include <iprt/alloca.h>
#include <iprt/assert.h>
#include <stdlib.h>
#include <errno.h>
RTDECL(bool) RTEnvExist(const char *pszVar)
{
return getenv(pszVar) != NULL;
}
RTDECL(const char *) RTEnvGet(const char *pszVar)
{
return getenv(pszVar);
}
RTDECL(int) RTEnvPut(const char *pszVarEqualValue)
{
/** @todo putenv is a source memory leaks. deal with this on a per system basis. */
if (!putenv((char *)pszVarEqualValue))
return 0;
return RTErrConvertFromErrno(errno);
}
RTDECL(int) RTEnvSet(const char *pszVar, const char *pszValue)
{
#if defined(_MSC_VER)
/* make a local copy and feed it to putenv. */
const size_t cchVar = strlen(pszVar);
const size_t cchValue = strlen(pszValue);
char *pszTmp = (char *)alloca(cchVar + cchValue + 2 + !*pszValue);
memcpy(pszTmp, pszVar, cchVar);
pszTmp[cchVar] = '=';
if (*pszValue)
memcpy(pszTmp + cchVar + 1, pszValue, cchValue + 1);
else
{
pszTmp[cchVar + 1] = ' '; /* wrong, but putenv will remove it otherwise. */
pszTmp[cchVar + 2] = '\0';
}
if (!putenv(pszTmp))
return 0;
return RTErrConvertFromErrno(errno);
#else
if (!setenv(pszVar, pszValue, 1))
return VINF_SUCCESS;
return RTErrConvertFromErrno(errno);
#endif
}
RTDECL(int) RTEnvUnset(const char *pszVar)
{
AssertReturn(!strchr(pszVar, '='), VERR_INVALID_PARAMETER);
/* Check that it exists first. */
if (!RTEnvExist(pszVar))
return VINF_ENV_VAR_NOT_FOUND;
/* Ok, try remove it. */
if (!putenv((char *)pszVar))
return VINF_SUCCESS;
return RTErrConvertFromErrno(errno);
}