mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
314 lines
9.4 KiB
314 lines
9.4 KiB
/***
|
|
*putenv.c - put an environment variable into the environment
|
|
*
|
|
* Copyright (c) 1985-2001, Microsoft Corporation. All rights reserved.
|
|
*
|
|
*Purpose:
|
|
* defines _putenv() - adds a new variable to environment; does not
|
|
* change global environment, only the process' environment.
|
|
*
|
|
*Revision History:
|
|
* 08-08-84 RN initial version
|
|
* 02-23-88 SKS check for environment containing only the NULL string
|
|
* 05-31-88 PHG Merged DLL and normal versions
|
|
* 07-14-88 JCR Much simplified since (1) __setenvp always uses heap, and
|
|
* (2) envp array and env strings are in seperate heap blocks
|
|
* 07-03-89 PHG Now "option=" string removes string from environment
|
|
* 08-17-89 GJF Removed _NEAR_, _LOAD_DS and fixed indents.
|
|
* 09-14-89 KRS Don't give error if 'option' not defined in "option=".
|
|
* 11-20-89 GJF Added const to arg type. Also, fixed copyright.
|
|
* 03-15-90 GJF Made the calling type _CALLTYPE1, added #include
|
|
* <cruntime.h> and removed #include <register.h>.
|
|
* 04-05-90 GJF Made findenv() _CALLTYPE4.
|
|
* 04-26-90 JCR Bug fix if environ is NULL (stubbed out _setenvp)
|
|
* 07-25-90 SBM Removed redundant include (stdio.h)
|
|
* 10-04-90 GJF New-style function declarators.
|
|
* 01-21-91 GJF ANSI naming.
|
|
* 02-06-91 SRW Added _WIN32_ conditional for SetEnvironmentVariable
|
|
* 02-18-91 SRW Changed _WIN32_ conditional for SetEnvironmentVariable
|
|
* to be in addition to old logic instead of replacement
|
|
* 04-23-92 GJF Made findenv insensitive to the case of name for Win32.
|
|
* Also added support for 'current drive' environment
|
|
* strings in Win32.
|
|
* 04-29-92 GJF Repackaged so that _putenv_lk could be easily added for
|
|
* for Win32.
|
|
* 05-05-92 DJM POSIX not supported.
|
|
* 04-06-93 SKS Replace _CRTAPI* with __cdecl
|
|
* 06-05-92 PLM Added _MAC_
|
|
* 11-24-93 CFW Rip out Cruiser, disallow "=C:=C:\foo" format putenvs.
|
|
* 11-29-93 CFW Wide char enable, convert between wide and narrow
|
|
* types. Mucho code moved to setenv.c
|
|
* 12-07-93 CFW Change _TCHAR to _TSCHAR.
|
|
* 01-15-94 CFW Use _tcsnicoll for global match.
|
|
* 09-06-94 CFW Replace MTHREAD with _MT.
|
|
* 01-10-95 CFW Debug CRT allocs.
|
|
* 02-14-95 CFW Debug CRT allocs for Mac version.
|
|
* 02-16-95 JWM Spliced _WIN32 & Mac versions.
|
|
* 06-01-95 CFW Copy environment string before passing to _crt[w]setenv.
|
|
* 07-09-97 GJF Added a check that the environment initialization has
|
|
* been executed. Also, cleaned up the format a bit and
|
|
* got rid of obsolete _CALLTYPE* macros.
|
|
* 03-03-98 RKP Added 64 bit support.
|
|
* 03-05-98 GJF Exception-safe locking.
|
|
* 08-28-98 GJF Use CP_ACP instead of CP_OEMCP.
|
|
* 05-17-99 PML Remove all Macintosh support.
|
|
* 05-25-99 GJF Free up buffers allocated to hold env string when there
|
|
* there is a failure.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifndef _POSIX_
|
|
|
|
#include <windows.h>
|
|
#include <cruntime.h>
|
|
#include <internal.h>
|
|
#include <mtdll.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <tchar.h>
|
|
#include <dbgint.h>
|
|
|
|
#ifndef CRTDLL
|
|
|
|
/*
|
|
* Flag checked by getenv() and _putenv() to determine if the environment has
|
|
* been initialized.
|
|
*/
|
|
extern int __env_initialized;
|
|
|
|
#endif
|
|
|
|
/***
|
|
*int _putenv(option) - add/replace/remove variable in environment
|
|
*
|
|
*Purpose:
|
|
* option should be of the form "option=value". If a string with the
|
|
* given option part already exists, it is replaced with the given
|
|
* string; otherwise the given string is added to the environment.
|
|
* If the string is of the form "option=", then the string is
|
|
* removed from the environment, if it exists. If the string has
|
|
* no equals sign, error is returned.
|
|
*
|
|
*Entry:
|
|
* char *option - option string to set in the environment list.
|
|
* should be of the form "option=value".
|
|
*
|
|
*Exit:
|
|
* returns 0 if OK, -1 if fails.
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*Warning:
|
|
* This code will not work if variables are removed from the
|
|
* environment by deleting them from environ[]. Use _putenv("option=")
|
|
* to remove a variable.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifdef _MT
|
|
|
|
#ifdef WPRFLAG
|
|
int __cdecl _wputenv (
|
|
#else
|
|
int __cdecl _putenv (
|
|
#endif
|
|
const _TSCHAR *option
|
|
)
|
|
{
|
|
int retval;
|
|
|
|
_mlock( _ENV_LOCK );
|
|
|
|
__try {
|
|
#ifdef WPRFLAG
|
|
retval = _wputenv_lk(option);
|
|
#else
|
|
retval = _putenv_lk(option);
|
|
#endif
|
|
}
|
|
__finally {
|
|
_munlock( _ENV_LOCK );
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
#ifdef WPRFLAG
|
|
int __cdecl _wputenv_lk (
|
|
#else
|
|
int __cdecl _putenv_lk (
|
|
#endif
|
|
const _TSCHAR *option
|
|
)
|
|
|
|
#else /* ndef _MT */
|
|
|
|
#ifdef WPRFLAG
|
|
int __cdecl _wputenv (
|
|
#else
|
|
int __cdecl _putenv (
|
|
#endif
|
|
const _TSCHAR *option
|
|
)
|
|
|
|
#endif /* _MT */
|
|
|
|
{
|
|
int size;
|
|
_TSCHAR * newoption;
|
|
|
|
#ifndef CRTDLL
|
|
/*
|
|
* Make sure the environment is initialized.
|
|
*/
|
|
if ( !__env_initialized )
|
|
return -1;
|
|
#endif /* CRTDLL */
|
|
|
|
/*
|
|
* At startup, we obtain the 'native' flavor of environment strings
|
|
* from the OS. So a "main" program has _environ and a "wmain" has
|
|
* _wenviron loaded at startup. Only when the user gets or puts the
|
|
* 'other' flavor do we convert it.
|
|
*/
|
|
|
|
/* copy the new environent string */
|
|
if ( (newoption = (_TSCHAR *)_malloc_crt((_tcslen(option)+1) *
|
|
sizeof(_TSCHAR))) == NULL )
|
|
return -1;
|
|
|
|
_tcscpy(newoption, option);
|
|
|
|
#ifdef WPRFLAG
|
|
if ( __crtwsetenv(newoption, 1) != 0 )
|
|
{
|
|
_free_crt(newoption);
|
|
return -1;
|
|
}
|
|
|
|
/* If other environment type exists, set it */
|
|
if (_environ)
|
|
{
|
|
char *mboption;
|
|
|
|
/* find out how much space is needed */
|
|
if ( (size = WideCharToMultiByte(CP_ACP, 0, option, -1, NULL,
|
|
0, NULL, NULL)) == 0 )
|
|
return -1;
|
|
|
|
/* allocate space for variable */
|
|
if ((mboption = (char *) _malloc_crt(size * sizeof(char))) == NULL)
|
|
return -1;
|
|
|
|
/* convert it */
|
|
if ( WideCharToMultiByte(CP_ACP, 0, option, -1, mboption, size,
|
|
NULL, NULL) == 0 )
|
|
{
|
|
_free_crt(mboption);
|
|
return -1;
|
|
}
|
|
|
|
/* set it - this is not primary call, so set primary == 0 */
|
|
if ( __crtsetenv(mboption, 0) != 0 )
|
|
{
|
|
_free_crt(mboption);
|
|
return -1;
|
|
}
|
|
}
|
|
#else
|
|
/* Set requested environment type, primary call */
|
|
if ( __crtsetenv(newoption, 1) != 0 )
|
|
{
|
|
_free_crt(newoption);
|
|
return -1;
|
|
}
|
|
|
|
/* If other environment type exists, set it */
|
|
if (_wenviron)
|
|
{
|
|
wchar_t *woption;
|
|
|
|
/* find out how much space is needed */
|
|
if ( (size = MultiByteToWideChar(CP_ACP, 0, option, -1, NULL, 0))
|
|
== 0 )
|
|
return -1;
|
|
|
|
/* allocate space for variable */
|
|
if ( (woption = (wchar_t *) _malloc_crt(size * sizeof(wchar_t)))
|
|
== NULL )
|
|
return -1;
|
|
|
|
/* convert it */
|
|
if ( MultiByteToWideChar(CP_ACP, 0, option, -1, woption, size)
|
|
== 0 )
|
|
{
|
|
_free_crt(woption);
|
|
return -1;
|
|
}
|
|
|
|
/* set it - this is not primary call, so set primary == 0 */
|
|
if ( __crtwsetenv(woption, 0) != 0 )
|
|
{
|
|
_free_crt(woption);
|
|
return -1;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
/***
|
|
*int findenv(name, len) - [STATIC]
|
|
*
|
|
*Purpose:
|
|
* Scan for the given string within the environment
|
|
*
|
|
*Entry:
|
|
*
|
|
*Exit:
|
|
* Returns the offset in "environ[]" of the given variable
|
|
* Returns the negative of the length of environ[] if not found.
|
|
* Returns 0 if the environment is empty.
|
|
*
|
|
* [NOTE: That a 0 return can mean that the environment is empty
|
|
* or that the string was found as the first entry in the array.]
|
|
*
|
|
*Exceptions:
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifdef WPRFLAG
|
|
static int __cdecl wfindenv (
|
|
#else
|
|
static int __cdecl findenv (
|
|
#endif
|
|
const _TSCHAR *name,
|
|
int len
|
|
)
|
|
{
|
|
_TSCHAR **env;
|
|
|
|
for ( env = _tenviron ; *env != NULL ; env++ ) {
|
|
/*
|
|
* See if first len characters match, up to case
|
|
*/
|
|
if ( _tcsnicoll(name, *env, len) == 0 )
|
|
/*
|
|
* the next character of the environment string must
|
|
* be an '=' or a '\0'
|
|
*/
|
|
if ( (*env)[len] == _T('=') || (*env)[len] == _T('\0') )
|
|
return(int)(env - _tenviron);
|
|
//
|
|
// We cannot break here since findenv must report the total number of strings.
|
|
// else
|
|
// break;
|
|
}
|
|
|
|
return(-(int)(env - _tenviron));
|
|
}
|
|
|
|
#endif /* _POSIX_ */
|