archie/tcl7.6/generic/tclEnv.c
2024-05-27 16:40:40 +02:00

632 lines
16 KiB
C
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
* tclEnv.c --
*
* Tcl support for environment variables, including a setenv
* procedure.
*
* Copyright (c) 1991-1994 The Regents of the University of California.
* Copyright (c) 1994-1996 Sun Microsystems, Inc.
*
* See the file "license.terms" for information on usage and redistribution
* of this file, and for a DISCLAIMER OF ALL WARRANTIES.
*
* SCCS: @(#) tclEnv.c 1.39 96/08/22 10:47:09
*/
/*
* The putenv and setenv definitions below cause any system prototypes for
* those procedures to be ignored so that there won't be a clash when the
* versions in this file are compiled.
*/
#define putenv ignore_putenv
#define setenv ignore_setenv
#include "tclInt.h"
#include "tclPort.h"
#undef putenv
#undef setenv
/*
* The structure below is used to keep track of all of the interpereters
* for which we're managing the "env" array. It's needed so that they
* can all be updated whenever an environment variable is changed
* anywhere.
*/
typedef struct EnvInterp {
Tcl_Interp *interp; /* Interpreter for which we're managing
* the env array. */
struct EnvInterp *nextPtr; /* Next in list of all such interpreters,
* or zero. */
} EnvInterp;
static EnvInterp *firstInterpPtr;
/* First in list of all managed interpreters,
* or NULL if none. */
static int environSize = 0; /* Non-zero means that the all of the
* environ-related information is malloc-ed
* and the environ array itself has this
* many total entries allocated to it (not
* all may be in use at once). Zero means
* that the environment array is in its
* original static state. */
/*
* Declarations for local procedures defined in this file:
*/
static void EnvExitProc _ANSI_ARGS_((ClientData clientData));
static void EnvInit _ANSI_ARGS_((void));
static char * EnvTraceProc _ANSI_ARGS_((ClientData clientData,
Tcl_Interp *interp, char *name1, char *name2,
int flags));
static int FindVariable _ANSI_ARGS_((CONST char *name,
int *lengthPtr));
void TclSetEnv _ANSI_ARGS_((CONST char *name,
CONST char *value));
void TclUnsetEnv _ANSI_ARGS_((CONST char *name));
/*
*----------------------------------------------------------------------
*
* TclSetupEnv --
*
* This procedure is invoked for an interpreter to make environment
* variables accessible from that interpreter via the "env"
* associative array.
*
* Results:
* None.
*
* Side effects:
* The interpreter is added to a list of interpreters managed
* by us, so that its view of envariables can be kept consistent
* with the view in other interpreters. If this is the first
* call to Tcl_SetupEnv, then additional initialization happens,
* such as copying the environment to dynamically-allocated space
* for ease of management.
*
*----------------------------------------------------------------------
*/
void
TclSetupEnv(interp)
Tcl_Interp *interp; /* Interpreter whose "env" array is to be
* managed. */
{
EnvInterp *eiPtr;
char *p, *p2;
Tcl_DString ds;
int i, sz;
/*
* First, initialize our environment-related information, if
* necessary.
*/
if (environSize == 0) {
EnvInit();
}
/*
* Next, initialize the DString we are going to use for copying
* the names of the environment variables.
*/
Tcl_DStringInit(&ds);
/*
* Next, add the interpreter to the list of those that we manage.
*/
eiPtr = (EnvInterp *) ckalloc(sizeof(EnvInterp));
eiPtr->interp = interp;
eiPtr->nextPtr = firstInterpPtr;
firstInterpPtr = eiPtr;
/*
* Store the environment variable values into the interpreter's
* "env" array, and arrange for us to be notified on future
* writes and unsets to that array.
*/
(void) Tcl_UnsetVar2(interp, "env", (char *) NULL, TCL_GLOBAL_ONLY);
for (i = 0; ; i++) {
p = environ[i];
if (p == NULL) {
break;
}
for (p2 = p; *p2 != '='; p2++) {
/* Empty loop body. */
}
sz = p2 - p;
Tcl_DStringSetLength(&ds, 0);
Tcl_DStringAppend(&ds, p, sz);
(void) Tcl_SetVar2(interp, "env", Tcl_DStringValue(&ds),
p2+1, TCL_GLOBAL_ONLY);
}
Tcl_TraceVar2(interp, "env", (char *) NULL,
TCL_GLOBAL_ONLY | TCL_TRACE_WRITES | TCL_TRACE_UNSETS,
EnvTraceProc, (ClientData) NULL);
/*
* Finally clean up the DString.
*/
Tcl_DStringFree(&ds);
}
/*
*----------------------------------------------------------------------
*
* FindVariable --
*
* Locate the entry in environ for a given name.
*
* Results:
* The return value is the index in environ of an entry with the
* name "name", or -1 if there is no such entry. The integer at
* *lengthPtr is filled in with the length of name (if a matching
* entry is found) or the length of the environ array (if no matching
* entry is found).
*
* Side effects:
* None.
*
*----------------------------------------------------------------------
*/
static int
FindVariable(name, lengthPtr)
CONST char *name; /* Name of desired environment variable. */
int *lengthPtr; /* Used to return length of name (for
* successful searches) or number of non-NULL
* entries in environ (for unsuccessful
* searches). */
{
int i;
register CONST char *p1, *p2;
for (i = 0, p1 = environ[i]; p1 != NULL; i++, p1 = environ[i]) {
for (p2 = name; *p2 == *p1; p1++, p2++) {
/* NULL loop body. */
}
if ((*p1 == '=') && (*p2 == '\0')) {
*lengthPtr = p2-name;
return i;
}
}
*lengthPtr = i;
return -1;
}
/*
*----------------------------------------------------------------------
*
* TclGetEnv --
*
* Get an environment variable or return NULL if the variable
* doesn't exist. This procedure is intended to be a
* stand-in for the UNIX "getenv" procedure so that applications
* using that procedure will interface properly to Tcl. To make
* it a stand-in, the Makefile must define "TclGetEnv" to "getenv".
*
* Results:
* ptr to value on success, NULL if error.
*
* Side effects:
* None.
*
*----------------------------------------------------------------------
*/
char *
TclGetEnv(name)
char *name; /* Name of desired environment variable. */
{
int i;
size_t len, nameLen;
char *equal;
nameLen = strlen(name);
for (i = 0; environ[i] != NULL; i++) {
equal = strchr(environ[i], '=');
if (equal == NULL) {
continue;
}
len = (size_t) (equal - environ[i]);
if ((len == nameLen) && (strncmp(name, environ[i], len) == 0)) {
/*
* The caller of this function should regard this
* as static memory.
*/
return &environ[i][len+1];
}
}
return NULL;
}
/*
*----------------------------------------------------------------------
*
* TclSetEnv --
*
* Set an environment variable, replacing an existing value
* or creating a new variable if there doesn't exist a variable
* by the given name. This procedure is intended to be a
* stand-in for the UNIX "setenv" procedure so that applications
* using that procedure will interface properly to Tcl. To make
* it a stand-in, the Makefile must define "TclSetEnv" to "setenv".
*
* Results:
* None.
*
* Side effects:
* The environ array gets updated, as do all of the interpreters
* that we manage.
*
*----------------------------------------------------------------------
*/
void
TclSetEnv(name, value)
CONST char *name; /* Name of variable whose value is to be
* set. */
CONST char *value; /* New value for variable. */
{
int index, length, nameLength;
char *p;
EnvInterp *eiPtr;
if (environSize == 0) {
EnvInit();
}
/*
* Figure out where the entry is going to go. If the name doesn't
* already exist, enlarge the array if necessary to make room. If
* the name exists, free its old entry.
*/
index = FindVariable(name, &length);
if (index == -1) {
if ((length+2) > environSize) {
char **newEnviron;
newEnviron = (char **) ckalloc((unsigned)
((length+5) * sizeof(char *)));
memcpy((VOID *) newEnviron, (VOID *) environ,
length*sizeof(char *));
ckfree((char *) environ);
environ = newEnviron;
environSize = length+5;
}
index = length;
environ[index+1] = NULL;
nameLength = strlen(name);
} else {
/*
* Compare the new value to the existing value. If they're
* the same then quit immediately (e.g. don't rewrite the
* value or propagate it to other interpreters). Otherwise,
* when there are N interpreters there will be N! propagations
* of the same value among the interpreters.
*/
if (strcmp(value, environ[index]+length+1) == 0) {
return;
}
ckfree(environ[index]);
nameLength = length;
}
/*
* Create a new entry and enter it into the table.
*/
p = (char *) ckalloc((unsigned) (nameLength + strlen(value) + 2));
environ[index] = p;
strcpy(p, name);
p += nameLength;
*p = '=';
strcpy(p+1, value);
/*
* Update all of the interpreters.
*/
for (eiPtr= firstInterpPtr; eiPtr != NULL; eiPtr = eiPtr->nextPtr) {
(void) Tcl_SetVar2(eiPtr->interp, "env", (char *) name,
p+1, TCL_GLOBAL_ONLY);
}
/*
* Update the system environment.
*/
TclSetSystemEnv(name, value);
}
/*
*----------------------------------------------------------------------
*
* Tcl_PutEnv --
*
* Set an environment variable. Similar to setenv except that
* the information is passed in a single string of the form
* NAME=value, rather than as separate name strings. This procedure
* is intended to be a stand-in for the UNIX "putenv" procedure
* so that applications using that procedure will interface
* properly to Tcl. To make it a stand-in, the Makefile will
* define "Tcl_PutEnv" to "putenv".
*
* Results:
* None.
*
* Side effects:
* The environ array gets updated, as do all of the interpreters
* that we manage.
*
*----------------------------------------------------------------------
*/
int
Tcl_PutEnv(string)
CONST char *string; /* Info about environment variable in the
* form NAME=value. */
{
int nameLength;
char *name, *value;
if (string == NULL) {
return 0;
}
/*
* Separate the string into name and value parts, then call
* TclSetEnv to do all of the real work.
*/
value = strchr(string, '=');
if (value == NULL) {
return 0;
}
nameLength = value - string;
if (nameLength == 0) {
return 0;
}
name = (char *) ckalloc((unsigned) nameLength+1);
memcpy(name, string, (size_t) nameLength);
name[nameLength] = 0;
TclSetEnv(name, value+1);
ckfree(name);
return 0;
}
/*
*----------------------------------------------------------------------
*
* TclUnsetEnv --
*
* Remove an environment variable, updating the "env" arrays
* in all interpreters managed by us. This function is intended
* to replace the UNIX "unsetenv" function (but to do this the
* Makefile must be modified to redefine "TclUnsetEnv" to
* "unsetenv".
*
* Results:
* None.
*
* Side effects:
* Interpreters are updated, as is environ.
*
*----------------------------------------------------------------------
*/
void
TclUnsetEnv(name)
CONST char *name; /* Name of variable to remove. */
{
int index, dummy;
char **envPtr;
EnvInterp *eiPtr;
if (environSize == 0) {
EnvInit();
}
/*
* Update the environ array.
*/
index = FindVariable(name, &dummy);
if (index == -1) {
return;
}
ckfree(environ[index]);
for (envPtr = environ+index+1; ; envPtr++) {
envPtr[-1] = *envPtr;
if (*envPtr == NULL) {
break;
}
}
/*
* Update all of the interpreters.
*/
for (eiPtr = firstInterpPtr; eiPtr != NULL; eiPtr = eiPtr->nextPtr) {
(void) Tcl_UnsetVar2(eiPtr->interp, "env", (char *) name,
TCL_GLOBAL_ONLY);
}
/*
* Update the system environment.
*/
TclSetSystemEnv(name, NULL);
}
/*
*----------------------------------------------------------------------
*
* EnvTraceProc --
*
* This procedure is invoked whenever an environment variable
* is modified or deleted. It propagates the change to the
* "environ" array and to any other interpreters for whom
* we're managing an "env" array.
*
* Results:
* Always returns NULL to indicate success.
*
* Side effects:
* Environment variable changes get propagated. If the whole
* "env" array is deleted, then we stop managing things for
* this interpreter (usually this happens because the whole
* interpreter is being deleted).
*
*----------------------------------------------------------------------
*/
/* ARGSUSED */
static char *
EnvTraceProc(clientData, interp, name1, name2, flags)
ClientData clientData; /* Not used. */
Tcl_Interp *interp; /* Interpreter whose "env" variable is
* being modified. */
char *name1; /* Better be "env". */
char *name2; /* Name of variable being modified, or
* NULL if whole array is being deleted. */
int flags; /* Indicates what's happening. */
{
/*
* First see if the whole "env" variable is being deleted. If
* so, just forget about this interpreter.
*/
if (name2 == NULL) {
register EnvInterp *eiPtr, *prevPtr;
if ((flags & (TCL_TRACE_UNSETS|TCL_TRACE_DESTROYED))
!= (TCL_TRACE_UNSETS|TCL_TRACE_DESTROYED)) {
panic("EnvTraceProc called with confusing arguments");
}
eiPtr = firstInterpPtr;
if (eiPtr->interp == interp) {
firstInterpPtr = eiPtr->nextPtr;
} else {
for (prevPtr = eiPtr, eiPtr = eiPtr->nextPtr; ;
prevPtr = eiPtr, eiPtr = eiPtr->nextPtr) {
if (eiPtr == NULL) {
panic("EnvTraceProc couldn't find interpreter");
}
if (eiPtr->interp == interp) {
prevPtr->nextPtr = eiPtr->nextPtr;
break;
}
}
}
ckfree((char *) eiPtr);
return NULL;
}
/*
* If a value is being set, call TclSetEnv to do all of the work.
*/
if (flags & TCL_TRACE_WRITES) {
TclSetEnv(name2, Tcl_GetVar2(interp, "env", name2, TCL_GLOBAL_ONLY));
}
if (flags & TCL_TRACE_UNSETS) {
TclUnsetEnv(name2);
}
return NULL;
}
/*
*----------------------------------------------------------------------
*
* EnvInit --
*
* This procedure is called to initialize our management
* of the environ array.
*
* Results:
* None.
*
* Side effects:
* Environ gets copied to malloc-ed storage, so that in
* the future we don't have to worry about which entries
* are malloc-ed and which are static.
*
*----------------------------------------------------------------------
*/
static void
EnvInit()
{
#ifdef MAC_TCL
environSize = TclMacCreateEnv();
#else
char **newEnviron;
int i, length;
if (environSize != 0) {
return;
}
for (length = 0; environ[length] != NULL; length++) {
/* Empty loop body. */
}
environSize = length+5;
newEnviron = (char **) ckalloc((unsigned)
(environSize * sizeof(char *)));
for (i = 0; i < length; i++) {
newEnviron[i] = (char *) ckalloc((unsigned) (strlen(environ[i]) + 1));
strcpy(newEnviron[i], environ[i]);
}
newEnviron[length] = NULL;
environ = newEnviron;
Tcl_CreateExitHandler(EnvExitProc, (ClientData) NULL);
#endif
}
/*
*----------------------------------------------------------------------
*
* EnvExitProc --
*
* This procedure is called just before the process exits. It
* frees the memory associated with environment variables.
*
* Results:
* None.
*
* Side effects:
* Memory is freed.
*
*----------------------------------------------------------------------
*/
static void
EnvExitProc(clientData)
ClientData clientData; /* Not used. */
{
char **p;
for (p = environ; *p != NULL; p++) {
ckfree(*p);
}
ckfree((char *) environ);
/*
* Note that we need to reset the environ global so the Borland C run-time
* doesn't choke on exit.
*/
environ = NULL;
}