mirror of
https://github.com/fish-shell/fish-shell.git
synced 2024-11-27 11:43:37 +08:00
bd5a16d213
darcs-hash:20070127165911-ac50b-72747760d767a6918152c2e4f254febb19ab1205.gz
1545 lines
26 KiB
C
1545 lines
26 KiB
C
/** \file env.c
|
|
Functions for setting and getting environment variables.
|
|
*/
|
|
#include "config.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <wchar.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <locale.h>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <pwd.h>
|
|
|
|
#if HAVE_NCURSES_H
|
|
#include <ncurses.h>
|
|
#else
|
|
#include <curses.h>
|
|
#endif
|
|
|
|
#if HAVE_TERMIO_H
|
|
#include <termio.h>
|
|
#endif
|
|
|
|
#if HAVE_TERM_H
|
|
#include <term.h>
|
|
#elif HAVE_NCURSES_TERM_H
|
|
#include <ncurses/term.h>
|
|
#endif
|
|
|
|
#if HAVE_LIBINTL_H
|
|
#include <libintl.h>
|
|
#endif
|
|
|
|
#include <errno.h>
|
|
|
|
|
|
#include "fallback.h"
|
|
#include "util.h"
|
|
|
|
#include "wutil.h"
|
|
#include "proc.h"
|
|
#include "common.h"
|
|
#include "env.h"
|
|
#include "sanity.h"
|
|
#include "expand.h"
|
|
#include "history.h"
|
|
#include "reader.h"
|
|
#include "parser.h"
|
|
#include "env_universal.h"
|
|
#include "input_common.h"
|
|
#include "event.h"
|
|
|
|
#include "complete.h"
|
|
|
|
/**
|
|
Command used to start fishd
|
|
*/
|
|
#define FISHD_CMD L"fishd ^/tmp/fishd.log.%s"
|
|
|
|
/**
|
|
Value denoting a null string
|
|
*/
|
|
#define ENV_NULL L"\x1d"
|
|
|
|
/**
|
|
At init, we read all the environment variables from this array.
|
|
*/
|
|
extern char **environ;
|
|
/**
|
|
This should be the same thing as \c environ, but it is possible only one of the two work...
|
|
*/
|
|
extern char **__environ;
|
|
|
|
|
|
/**
|
|
Struct representing one level in the function variable stack
|
|
*/
|
|
typedef struct env_node
|
|
{
|
|
/**
|
|
Variable table
|
|
*/
|
|
hash_table_t env;
|
|
/**
|
|
Does this node imply a new variable scope? If yes, all
|
|
non-global variables below this one in the stack are
|
|
invisible. If new_scope is set for the global variable node,
|
|
the universe will explode.
|
|
*/
|
|
int new_scope;
|
|
/**
|
|
Does this node contain any variables which are exported to subshells
|
|
*/
|
|
int export;
|
|
|
|
/**
|
|
Pointer to next level
|
|
*/
|
|
struct env_node *next;
|
|
}
|
|
env_node_t;
|
|
|
|
/**
|
|
A variable entry. Stores the value of a variable and whether it
|
|
should be exported. Obviously, it needs to be allocated large
|
|
enough to fit the value string.
|
|
*/
|
|
typedef struct var_entry
|
|
{
|
|
int export; /**< Whether the variable should be exported */
|
|
size_t size; /**< The maximum length (excluding the NULL) that will fit into this var_entry_t */
|
|
|
|
#if __STDC_VERSION__ < 199901L
|
|
wchar_t val[1]; /**< The value of the variable */
|
|
#else
|
|
wchar_t val[]; /**< The value of the variable */
|
|
#endif
|
|
}
|
|
var_entry_t;
|
|
|
|
/**
|
|
Top node on the function stack
|
|
*/
|
|
static env_node_t *top=0;
|
|
|
|
/**
|
|
Bottom node on the function stack
|
|
*/
|
|
static env_node_t *global_env = 0;
|
|
|
|
|
|
/**
|
|
Table for global variables
|
|
*/
|
|
static hash_table_t *global;
|
|
|
|
/**
|
|
Table of variables that may not be set using the set command.
|
|
*/
|
|
static hash_table_t env_read_only;
|
|
|
|
/**
|
|
Table of variables whose value is dynamically calculated, such as umask, status, etc
|
|
*/
|
|
static hash_table_t env_electric;
|
|
|
|
/**
|
|
Exported variable array used by execv
|
|
*/
|
|
static char **export_arr=0;
|
|
|
|
/**
|
|
Buffer used for storing string contents for export_arr
|
|
*/
|
|
static buffer_t export_buffer;
|
|
|
|
|
|
/**
|
|
Flag for checking if we need to regenerate the exported variable
|
|
array
|
|
*/
|
|
static int has_changed = 1;
|
|
|
|
/**
|
|
This stringbuffer is used to store the value of dynamically
|
|
generated variables, such as history.
|
|
*/
|
|
static string_buffer_t dyn_var;
|
|
|
|
/**
|
|
Variable used by env_get_names to communicate auxiliary information
|
|
to add_key_to_hash
|
|
*/
|
|
static int get_names_show_exported;
|
|
/**
|
|
Variable used by env_get_names to communicate auxiliary information
|
|
to add_key_to_hash
|
|
*/
|
|
static int get_names_show_unexported;
|
|
|
|
/**
|
|
List of all locale variable names
|
|
*/
|
|
static const wchar_t *locale_variable[] =
|
|
{
|
|
L"LANG",
|
|
L"LC_ALL",
|
|
L"LC_COLLATE",
|
|
L"LC_CTYPE",
|
|
L"LC_MESSAGES",
|
|
L"LC_MONETARY",
|
|
L"LC_NUMERIC",
|
|
L"LC_TIME",
|
|
(void *)0
|
|
}
|
|
;
|
|
|
|
/**
|
|
Free hash key and hash value
|
|
*/
|
|
static void clear_hash_entry( void *key, void *data )
|
|
{
|
|
var_entry_t *entry = (var_entry_t *)data;
|
|
if( entry->export )
|
|
{
|
|
has_changed = 1;
|
|
}
|
|
|
|
free( (void *)key );
|
|
free( (void *)data );
|
|
}
|
|
|
|
/**
|
|
When fishd isn't started, this function is provided to
|
|
env_universal as a callback, it tries to start up fishd. It's
|
|
implementation is a bit of a hack, since it evaluates a bit of
|
|
shellscript, and it might be used at times when that might not be
|
|
the best idea.
|
|
*/
|
|
static void start_fishd()
|
|
{
|
|
string_buffer_t cmd;
|
|
struct passwd *pw;
|
|
|
|
sb_init( &cmd );
|
|
pw = getpwuid(getuid());
|
|
|
|
debug( 3, L"Spawning new copy of fishd" );
|
|
|
|
if( !pw )
|
|
{
|
|
debug( 0, _( L"Could not get user information" ) );
|
|
return;
|
|
}
|
|
|
|
sb_printf( &cmd, FISHD_CMD, pw->pw_name );
|
|
|
|
eval( (wchar_t *)cmd.buff,
|
|
0,
|
|
TOP );
|
|
sb_destroy( &cmd );
|
|
}
|
|
|
|
/**
|
|
Return the current umask value.
|
|
*/
|
|
static mode_t get_umask()
|
|
{
|
|
mode_t res;
|
|
res = umask( 0 );
|
|
umask( res );
|
|
return res;
|
|
}
|
|
|
|
/**
|
|
Checks if the specified variable is a locale variable
|
|
*/
|
|
static int is_locale( const wchar_t *key )
|
|
{
|
|
int i;
|
|
for( i=0; locale_variable[i]; i++ )
|
|
{
|
|
if( wcscmp(locale_variable[i], key ) == 0 )
|
|
{
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
Properly sets all locale information
|
|
*/
|
|
static void handle_locale()
|
|
{
|
|
const wchar_t *lc_all = env_get( L"LC_ALL" );
|
|
const wchar_t *lang;
|
|
int i;
|
|
wchar_t *old = wcsdup(wsetlocale( LC_MESSAGES, (void *)0 ));
|
|
|
|
/*
|
|
Array of locale constants corresponding to the local variable names defined in locale_variable
|
|
*/
|
|
static const int cat[] =
|
|
{
|
|
0,
|
|
LC_ALL,
|
|
LC_COLLATE,
|
|
LC_CTYPE,
|
|
LC_MESSAGES,
|
|
LC_MONETARY,
|
|
LC_NUMERIC,
|
|
LC_TIME
|
|
}
|
|
;
|
|
|
|
if( lc_all )
|
|
{
|
|
wsetlocale( LC_ALL, lc_all );
|
|
}
|
|
else
|
|
{
|
|
lang = env_get( L"LANG" );
|
|
if( lang )
|
|
{
|
|
wsetlocale( LC_ALL, lang );
|
|
}
|
|
|
|
for( i=2; locale_variable[i]; i++ )
|
|
{
|
|
const wchar_t *val = env_get( locale_variable[i] );
|
|
|
|
if( val )
|
|
{
|
|
wsetlocale( cat[i], val );
|
|
}
|
|
}
|
|
}
|
|
|
|
if( wcscmp( wsetlocale( LC_MESSAGES, (void *)0 ), old ) != 0 )
|
|
{
|
|
|
|
/*
|
|
Try to make change known to gettext. Both changing
|
|
_nl_msg_cat_cntr and calling dcgettext might potentially
|
|
tell some gettext implementation that the translation
|
|
strings should be reloaded. We do both and hope for the
|
|
best.
|
|
*/
|
|
|
|
extern int _nl_msg_cat_cntr;
|
|
_nl_msg_cat_cntr++;
|
|
|
|
dcgettext( "fish", "Changing language to English", LC_MESSAGES );
|
|
|
|
if( is_interactive )
|
|
{
|
|
debug( 0, _(L"Changing language to English") );
|
|
}
|
|
}
|
|
free( old );
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
Universal variable callback function. This function makes sure the
|
|
proper events are triggered when an event occurs.
|
|
*/
|
|
static void universal_callback( int type,
|
|
const wchar_t *name,
|
|
const wchar_t *val )
|
|
{
|
|
wchar_t *str=0;
|
|
|
|
if( is_locale( name ) )
|
|
{
|
|
handle_locale();
|
|
}
|
|
|
|
switch( type )
|
|
{
|
|
case SET:
|
|
case SET_EXPORT:
|
|
{
|
|
str=L"SET";
|
|
break;
|
|
}
|
|
|
|
case ERASE:
|
|
{
|
|
str=L"ERASE";
|
|
break;
|
|
}
|
|
}
|
|
|
|
if( str )
|
|
{
|
|
event_t ev;
|
|
|
|
has_changed=1;
|
|
|
|
ev.type=EVENT_VARIABLE;
|
|
ev.param1.variable=name;
|
|
ev.function_name=0;
|
|
|
|
al_init( &ev.arguments );
|
|
al_push( &ev.arguments, L"VARIABLE" );
|
|
al_push( &ev.arguments, str );
|
|
al_push( &ev.arguments, name );
|
|
event_fire( &ev );
|
|
al_destroy( &ev.arguments );
|
|
}
|
|
}
|
|
|
|
/**
|
|
Make sure the PATH variable contains the essaential directories
|
|
*/
|
|
static void setup_path()
|
|
{
|
|
wchar_t *path;
|
|
|
|
int i, j;
|
|
array_list_t l;
|
|
|
|
const wchar_t *path_el[] =
|
|
{
|
|
L"/bin",
|
|
L"/usr/bin",
|
|
PREFIX L"/bin",
|
|
0
|
|
}
|
|
;
|
|
|
|
path = env_get( L"PATH" );
|
|
if( !path )
|
|
{
|
|
env_set( L"PATH", 0, ENV_EXPORT | ENV_GLOBAL );
|
|
path=0;
|
|
}
|
|
|
|
al_init( &l );
|
|
|
|
if( path )
|
|
{
|
|
tokenize_variable_array( path, &l );
|
|
}
|
|
|
|
for( j=0; path_el[j]; j++ )
|
|
{
|
|
|
|
int has_el=0;
|
|
|
|
for( i=0; i<al_get_count( &l); i++ )
|
|
{
|
|
wchar_t * el = (wchar_t *)al_get( &l, i );
|
|
size_t len = wcslen( el );
|
|
|
|
while( (len > 0) && (el[len-1]==L'/') )
|
|
{
|
|
len--;
|
|
}
|
|
|
|
if( (wcslen( path_el[j] ) == len) &&
|
|
(wcsncmp( el, path_el[j], len)==0) )
|
|
{
|
|
has_el = 1;
|
|
}
|
|
}
|
|
|
|
if( !has_el )
|
|
{
|
|
string_buffer_t b;
|
|
|
|
debug( 3, L"directory %ls was missing", path_el[j] );
|
|
|
|
sb_init( &b );
|
|
if( path )
|
|
{
|
|
sb_append( &b, path );
|
|
}
|
|
|
|
sb_append2( &b,
|
|
ARRAY_SEP_STR,
|
|
path_el[j],
|
|
(void *)0 );
|
|
|
|
env_set( L"PATH", (wchar_t *)b.buff, ENV_GLOBAL | ENV_EXPORT );
|
|
|
|
sb_destroy( &b );
|
|
|
|
al_foreach( &l, &free );
|
|
path = env_get( L"PATH" );
|
|
al_truncate( &l, 0 );
|
|
tokenize_variable_array( path, &l );
|
|
}
|
|
}
|
|
|
|
al_foreach( &l, &free );
|
|
al_destroy( &l );
|
|
}
|
|
|
|
static void env_set_defaults()
|
|
{
|
|
|
|
if( !env_get( L"USER" ) )
|
|
{
|
|
struct passwd *pw = getpwuid( getuid());
|
|
wchar_t *unam = str2wcs( pw->pw_name );
|
|
env_set( L"USER", unam, ENV_GLOBAL );
|
|
free( unam );
|
|
}
|
|
|
|
if( !env_get( L"HOME" ) )
|
|
{
|
|
wchar_t *unam = env_get( L"USER" );
|
|
char *unam_narrow = wcs2str( unam );
|
|
struct passwd *pw = getpwnam( unam_narrow );
|
|
wchar_t *dir = str2wcs( pw->pw_dir );
|
|
env_set( L"HOME", dir, ENV_GLOBAL );
|
|
free( dir );
|
|
free( unam_narrow );
|
|
}
|
|
|
|
}
|
|
|
|
void env_init()
|
|
{
|
|
char **p;
|
|
struct passwd *pw;
|
|
wchar_t *uname;
|
|
wchar_t *version;
|
|
|
|
sb_init( &dyn_var );
|
|
b_init( &export_buffer );
|
|
|
|
/*
|
|
These variables can not be altered directly by the user
|
|
*/
|
|
hash_init( &env_read_only, &hash_wcs_func, &hash_wcs_cmp );
|
|
|
|
hash_put( &env_read_only, L"status", L"" );
|
|
hash_put( &env_read_only, L"history", L"" );
|
|
hash_put( &env_read_only, L"version", L"" );
|
|
hash_put( &env_read_only, L"_", L"" );
|
|
hash_put( &env_read_only, L"LINES", L"" );
|
|
hash_put( &env_read_only, L"COLUMNS", L"" );
|
|
hash_put( &env_read_only, L"PWD", L"" );
|
|
|
|
/*
|
|
Names of all dynamically calculated variables
|
|
*/
|
|
hash_init( &env_electric, &hash_wcs_func, &hash_wcs_cmp );
|
|
hash_put( &env_electric, L"history", L"" );
|
|
hash_put( &env_electric, L"status", L"" );
|
|
hash_put( &env_electric, L"umask", L"" );
|
|
|
|
/*
|
|
HOME and USER should be writeable by root, since this can be a
|
|
convenient way to install software.
|
|
*/
|
|
if( getuid() != 0 )
|
|
{
|
|
hash_put( &env_read_only, L"HOME", L"" );
|
|
hash_put( &env_read_only, L"USER", L"" );
|
|
}
|
|
|
|
top = malloc( sizeof(env_node_t) );
|
|
top->next = 0;
|
|
top->new_scope = 0;
|
|
top->export=0;
|
|
hash_init( &top->env, &hash_wcs_func, &hash_wcs_cmp );
|
|
global_env = top;
|
|
global = &top->env;
|
|
|
|
/*
|
|
Now the environemnt variable handling is set up, the next step
|
|
is to insert valid data
|
|
*/
|
|
|
|
/*
|
|
Import environment variables
|
|
*/
|
|
for( p=environ?environ:__environ; p && *p; p++ )
|
|
{
|
|
wchar_t *key, *val;
|
|
wchar_t *pos;
|
|
|
|
key = str2wcs(*p);
|
|
|
|
if( !key )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
val = wcschr( key, L'=' );
|
|
|
|
if( val == 0 )
|
|
{
|
|
env_set( key, L"", ENV_EXPORT );
|
|
}
|
|
else
|
|
{
|
|
*val = L'\0';
|
|
val++;
|
|
pos=val;
|
|
//fwprintf( stderr, L"Set $%ls to %ls\n", key, val );
|
|
while( *pos )
|
|
{
|
|
if( *pos == L':' )
|
|
{
|
|
*pos = ARRAY_SEP;
|
|
}
|
|
pos++;
|
|
}
|
|
|
|
env_set( key, val, ENV_EXPORT | ENV_GLOBAL );
|
|
}
|
|
free(key);
|
|
}
|
|
|
|
/*
|
|
Set up the PATH variable
|
|
*/
|
|
setup_path();
|
|
|
|
/*
|
|
Set up the USER variable
|
|
*/
|
|
pw = getpwuid( getuid() );
|
|
if( pw )
|
|
{
|
|
uname = str2wcs( pw->pw_name );
|
|
env_set( L"USER", uname, ENV_GLOBAL | ENV_EXPORT );
|
|
free( uname );
|
|
}
|
|
|
|
/*
|
|
Set up the version variable
|
|
*/
|
|
version = str2wcs( PACKAGE_VERSION );
|
|
env_set( L"version", version, ENV_GLOBAL );
|
|
free( version );
|
|
|
|
env_universal_init( env_get( L"FISHD_SOCKET_DIR"),
|
|
env_get( L"USER" ),
|
|
&start_fishd,
|
|
&universal_callback );
|
|
|
|
/*
|
|
Set correct defaults for e.g. USER and HOME variables
|
|
*/
|
|
env_set_defaults();
|
|
}
|
|
|
|
void env_destroy()
|
|
{
|
|
env_universal_destroy();
|
|
|
|
sb_destroy( &dyn_var );
|
|
|
|
b_destroy( &export_buffer );
|
|
|
|
while( &top->env != global )
|
|
{
|
|
env_pop();
|
|
}
|
|
|
|
hash_destroy( &env_read_only );
|
|
|
|
hash_destroy( &env_electric );
|
|
|
|
hash_foreach( global, &clear_hash_entry );
|
|
hash_destroy( global );
|
|
free( top );
|
|
|
|
free( export_arr );
|
|
|
|
}
|
|
|
|
/**
|
|
Search all visible scopes in order for the specified key. Return
|
|
the first scope in which it was found.
|
|
*/
|
|
static env_node_t *env_get_node( const wchar_t *key )
|
|
{
|
|
var_entry_t* res;
|
|
env_node_t *env = top;
|
|
|
|
|
|
while( env != 0 )
|
|
{
|
|
res = (var_entry_t *) hash_get( &env->env,
|
|
key );
|
|
if( res != 0 )
|
|
{
|
|
return env;
|
|
}
|
|
|
|
if( env->new_scope )
|
|
{
|
|
env = global_env;
|
|
}
|
|
else
|
|
{
|
|
env = env->next;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int env_set( const wchar_t *key,
|
|
const wchar_t *val,
|
|
int var_mode )
|
|
{
|
|
int free_val = 0;
|
|
var_entry_t *entry;
|
|
env_node_t *node;
|
|
int has_changed_old = has_changed;
|
|
int has_changed_new = 0;
|
|
var_entry_t *e=0;
|
|
int done=0;
|
|
|
|
event_t ev;
|
|
int is_universal = 0;
|
|
|
|
CHECK( key, ENV_INVALID );
|
|
|
|
if( (var_mode & ENV_USER ) &&
|
|
hash_get( &env_read_only, key ) )
|
|
{
|
|
return ENV_PERM;
|
|
}
|
|
|
|
if( wcscmp( key, L"umask" ) == 0)
|
|
{
|
|
wchar_t *end;
|
|
int mask;
|
|
|
|
/*
|
|
Set the new umask
|
|
*/
|
|
if( val && wcslen(val) )
|
|
{
|
|
errno=0;
|
|
mask = wcstol( val, &end, 8 );
|
|
|
|
if( !errno && (!*end) && (mask <= 0777) && (mask >= 0) )
|
|
{
|
|
umask( mask );
|
|
}
|
|
}
|
|
/*
|
|
Do not actually create a umask variable, on env_get, it will
|
|
be calculated dynamically
|
|
*/
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
Zero element arrays are internaly not coded as null but as this
|
|
placeholder string
|
|
*/
|
|
if( !val )
|
|
{
|
|
val = ENV_NULL;
|
|
}
|
|
|
|
if( var_mode & ENV_UNIVERSAL )
|
|
{
|
|
int export = 0;
|
|
|
|
if( !(var_mode & ENV_EXPORT ) &&
|
|
!(var_mode & ENV_UNEXPORT ) )
|
|
{
|
|
env_universal_get_export( key );
|
|
}
|
|
else
|
|
{
|
|
export = (var_mode & ENV_EXPORT );
|
|
}
|
|
|
|
env_universal_set( key, val, export );
|
|
is_universal = 1;
|
|
|
|
}
|
|
else
|
|
{
|
|
|
|
node = env_get_node( key );
|
|
if( node && &node->env != 0 )
|
|
{
|
|
e = (var_entry_t *) hash_get( &node->env,
|
|
key );
|
|
|
|
if( e->export )
|
|
{
|
|
has_changed_new = 1;
|
|
}
|
|
}
|
|
|
|
if( (var_mode & ENV_LOCAL) ||
|
|
(var_mode & ENV_GLOBAL) )
|
|
{
|
|
node = ( var_mode & ENV_GLOBAL )?global_env:top;
|
|
}
|
|
else
|
|
{
|
|
if( node )
|
|
{
|
|
if( !(var_mode & ENV_EXPORT ) &&
|
|
!(var_mode & ENV_UNEXPORT ) )
|
|
{
|
|
var_mode = e->export?ENV_EXPORT:0;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if( !proc_had_barrier)
|
|
{
|
|
proc_had_barrier=1;
|
|
env_universal_barrier();
|
|
}
|
|
|
|
if( env_universal_get( key ) )
|
|
{
|
|
int export = 0;
|
|
|
|
if( !(var_mode & ENV_EXPORT ) &&
|
|
!(var_mode & ENV_UNEXPORT ) )
|
|
{
|
|
env_universal_get_export( key );
|
|
}
|
|
else
|
|
{
|
|
export = (var_mode & ENV_EXPORT );
|
|
}
|
|
|
|
env_universal_set( key, val, export );
|
|
is_universal = 1;
|
|
|
|
done = 1;
|
|
|
|
}
|
|
else
|
|
{
|
|
/*
|
|
New variable with unspecified scope. The default
|
|
scope is the innermost scope that is shadowing,
|
|
which will be either the current function or the
|
|
global scope.
|
|
*/
|
|
node = top;
|
|
while( node->next && !node->new_scope )
|
|
{
|
|
node = node->next;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if( !done )
|
|
{
|
|
void *k, *v;
|
|
var_entry_t *old_entry;
|
|
size_t val_len = wcslen(val);
|
|
|
|
hash_remove( &node->env, key, &k, &v );
|
|
|
|
/*
|
|
Try to reuse previous key string
|
|
*/
|
|
if( !k )
|
|
{
|
|
k = wcsdup(key);
|
|
}
|
|
|
|
old_entry = (var_entry_t *)v;
|
|
if( old_entry && old_entry->size >= val_len )
|
|
{
|
|
entry = old_entry;
|
|
|
|
if( !!(var_mode & ENV_EXPORT) || entry->export )
|
|
{
|
|
entry->export = !!(var_mode & ENV_EXPORT);
|
|
has_changed_new = 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
free( v );
|
|
|
|
entry = malloc( sizeof( var_entry_t ) +
|
|
sizeof(wchar_t )*(val_len+1));
|
|
|
|
if( !entry )
|
|
{
|
|
DIE_MEM();
|
|
}
|
|
|
|
entry->size = val_len;
|
|
|
|
if( var_mode & ENV_EXPORT)
|
|
{
|
|
entry->export = 1;
|
|
has_changed_new = 1;
|
|
}
|
|
else
|
|
{
|
|
entry->export = 0;
|
|
}
|
|
|
|
}
|
|
|
|
wcscpy( entry->val, val );
|
|
|
|
hash_put( &node->env, k, entry );
|
|
|
|
if( entry->export )
|
|
{
|
|
node->export=1;
|
|
}
|
|
|
|
if( free_val )
|
|
{
|
|
free((void *)val);
|
|
}
|
|
|
|
has_changed = has_changed_old || has_changed_new;
|
|
}
|
|
|
|
}
|
|
|
|
if( !is_universal )
|
|
{
|
|
ev.type=EVENT_VARIABLE;
|
|
ev.param1.variable = key;
|
|
ev.function_name = 0;
|
|
|
|
al_init( &ev.arguments );
|
|
al_push( &ev.arguments, L"VARIABLE" );
|
|
al_push( &ev.arguments, L"SET" );
|
|
al_push( &ev.arguments, key );
|
|
|
|
// debug( 1, L"env_set: fire events on variable %ls", key );
|
|
event_fire( &ev );
|
|
// debug( 1, L"env_set: return from event firing" );
|
|
al_destroy( &ev.arguments );
|
|
}
|
|
|
|
if( is_locale( key ) )
|
|
{
|
|
handle_locale();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/**
|
|
Attempt to remove/free the specified key/value pair from the
|
|
specified hash table.
|
|
|
|
\return zero if the variable was not found, non-zero otherwise
|
|
*/
|
|
static int try_remove( env_node_t *n,
|
|
const wchar_t *key,
|
|
int var_mode )
|
|
{
|
|
void *old_key_void, *old_val_void;
|
|
wchar_t *old_key, *old_val;
|
|
|
|
if( n == 0 )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
hash_remove( &n->env,
|
|
key,
|
|
&old_key_void,
|
|
&old_val_void );
|
|
|
|
old_key = (wchar_t *)old_key_void;
|
|
old_val = (wchar_t *)old_val_void;
|
|
|
|
if( old_key != 0 )
|
|
{
|
|
var_entry_t * v = (var_entry_t *)old_val;
|
|
if( v->export )
|
|
{
|
|
has_changed = 1;
|
|
}
|
|
|
|
free(old_key);
|
|
free(old_val);
|
|
return 1;
|
|
}
|
|
|
|
if( var_mode & ENV_LOCAL )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
if( n->new_scope )
|
|
{
|
|
return try_remove( global_env, key, var_mode );
|
|
}
|
|
else
|
|
{
|
|
return try_remove( n->next, key, var_mode );
|
|
}
|
|
}
|
|
|
|
|
|
int env_remove( const wchar_t *key, int var_mode )
|
|
{
|
|
env_node_t *first_node;
|
|
int erased = 0;
|
|
|
|
CHECK( key, 1 );
|
|
|
|
if( (var_mode & ENV_USER ) &&
|
|
hash_get( &env_read_only, key ) )
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
first_node = top;
|
|
|
|
if( ! (var_mode & ENV_UNIVERSAL ) )
|
|
{
|
|
|
|
if( var_mode & ENV_GLOBAL )
|
|
{
|
|
first_node = global_env;
|
|
}
|
|
|
|
if( try_remove( first_node, key, var_mode ) )
|
|
{
|
|
event_t ev;
|
|
|
|
ev.type=EVENT_VARIABLE;
|
|
ev.param1.variable=key;
|
|
ev.function_name=0;
|
|
|
|
al_init( &ev.arguments );
|
|
al_push( &ev.arguments, L"VARIABLE" );
|
|
al_push( &ev.arguments, L"ERASE" );
|
|
al_push( &ev.arguments, key );
|
|
|
|
event_fire( &ev );
|
|
|
|
al_destroy( &ev.arguments );
|
|
erased = 1;
|
|
}
|
|
}
|
|
|
|
if( !erased &&
|
|
!(var_mode & ENV_GLOBAL) &&
|
|
!(var_mode & ENV_LOCAL) )
|
|
{
|
|
erased = !env_universal_remove( key );
|
|
}
|
|
|
|
if( is_locale( key ) )
|
|
{
|
|
handle_locale();
|
|
}
|
|
|
|
return !erased;
|
|
}
|
|
|
|
|
|
wchar_t *env_get( const wchar_t *key )
|
|
{
|
|
var_entry_t *res;
|
|
env_node_t *env = top;
|
|
wchar_t *item;
|
|
|
|
CHECK( key, 0 );
|
|
|
|
if( wcscmp( key, L"history" ) == 0 )
|
|
{
|
|
wchar_t *current;
|
|
int i;
|
|
int add_current=0;
|
|
sb_clear( &dyn_var );
|
|
|
|
current = reader_get_buffer();
|
|
if( current && wcslen( current ) )
|
|
{
|
|
add_current=1;
|
|
sb_append( &dyn_var, current );
|
|
}
|
|
|
|
for( i=add_current;; i++ )
|
|
{
|
|
wchar_t *next = history_get( i-add_current );
|
|
if( !next )
|
|
{
|
|
break;
|
|
}
|
|
|
|
if( i!=0)
|
|
{
|
|
sb_append( &dyn_var, ARRAY_SEP_STR );
|
|
}
|
|
|
|
sb_append( &dyn_var, next );
|
|
}
|
|
|
|
return (wchar_t *)dyn_var.buff;
|
|
}
|
|
else if( wcscmp( key, L"COLUMNS" )==0 )
|
|
{
|
|
sb_clear( &dyn_var );
|
|
sb_printf( &dyn_var, L"%d", common_get_width() );
|
|
return (wchar_t *)dyn_var.buff;
|
|
}
|
|
else if( wcscmp( key, L"LINES" )==0 )
|
|
{
|
|
sb_clear( &dyn_var );
|
|
sb_printf( &dyn_var, L"%d", common_get_height() );
|
|
return (wchar_t *)dyn_var.buff;
|
|
}
|
|
else if( wcscmp( key, L"status" )==0 )
|
|
{
|
|
sb_clear( &dyn_var );
|
|
sb_printf( &dyn_var, L"%d", proc_get_last_status() );
|
|
return (wchar_t *)dyn_var.buff;
|
|
}
|
|
else if( wcscmp( key, L"umask" )==0 )
|
|
{
|
|
sb_clear( &dyn_var );
|
|
sb_printf( &dyn_var, L"0%0.3o", get_umask() );
|
|
return (wchar_t *)dyn_var.buff;
|
|
}
|
|
|
|
while( env != 0 )
|
|
{
|
|
res = (var_entry_t *) hash_get( &env->env,
|
|
key );
|
|
if( res != 0 )
|
|
{
|
|
if( wcscmp( res->val, ENV_NULL )==0)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
return res->val;
|
|
}
|
|
}
|
|
|
|
if( env->new_scope )
|
|
{
|
|
env = global_env;
|
|
}
|
|
else
|
|
{
|
|
env = env->next;
|
|
}
|
|
}
|
|
if( !proc_had_barrier)
|
|
{
|
|
proc_had_barrier=1;
|
|
env_universal_barrier();
|
|
}
|
|
|
|
item = env_universal_get( key );
|
|
|
|
if( !item || (wcscmp( item, ENV_NULL )==0))
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
return item;
|
|
}
|
|
}
|
|
|
|
int env_exist( const wchar_t *key, int mode )
|
|
{
|
|
var_entry_t *res;
|
|
env_node_t *env;
|
|
wchar_t *item=0;
|
|
|
|
CHECK( key, 0 );
|
|
|
|
/*
|
|
Read only variables all exist, and they are all global. A local
|
|
version can not exist.
|
|
*/
|
|
if( ! (mode & ENV_LOCAL) && ! (mode & ENV_UNIVERSAL) )
|
|
{
|
|
if( hash_get( &env_read_only, key ) || hash_get( &env_electric, key ) )
|
|
{
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if( ! (mode & ENV_UNIVERSAL) )
|
|
{
|
|
env = (mode & ENV_GLOBAL)?global_env:top;
|
|
|
|
while( env != 0 )
|
|
{
|
|
res = (var_entry_t *) hash_get( &env->env,
|
|
key );
|
|
if( res != 0 )
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
if( mode & ENV_LOCAL )
|
|
{
|
|
break;
|
|
}
|
|
|
|
if( env->new_scope )
|
|
{
|
|
env = global_env;
|
|
}
|
|
else
|
|
{
|
|
env = env->next;
|
|
}
|
|
}
|
|
}
|
|
|
|
if( ! (mode & ENV_LOCAL) && ! (mode & ENV_GLOBAL) )
|
|
{
|
|
if( !proc_had_barrier)
|
|
{
|
|
proc_had_barrier=1;
|
|
env_universal_barrier();
|
|
}
|
|
|
|
item = env_universal_get( key );
|
|
|
|
}
|
|
return item != 0;
|
|
|
|
}
|
|
|
|
/**
|
|
Returns true if the specified scope or any non-shadowed non-global subscopes contain an exported variable.
|
|
*/
|
|
static int local_scope_exports( env_node_t *n )
|
|
{
|
|
|
|
if( n==global_env )
|
|
return 0;
|
|
|
|
if( n->export )
|
|
return 1;
|
|
|
|
if( n->new_scope )
|
|
return 0;
|
|
|
|
return local_scope_exports( n->next );
|
|
}
|
|
|
|
void env_push( int new_scope )
|
|
{
|
|
env_node_t *node = malloc( sizeof(env_node_t) );
|
|
node->next = top;
|
|
node->export=0;
|
|
hash_init( &node->env, &hash_wcs_func, &hash_wcs_cmp );
|
|
node->new_scope=new_scope;
|
|
if( new_scope )
|
|
{
|
|
has_changed |= local_scope_exports(top);
|
|
}
|
|
top = node;
|
|
|
|
}
|
|
|
|
|
|
void env_pop()
|
|
{
|
|
if( &top->env != global )
|
|
{
|
|
int i;
|
|
int locale_changed = 0;
|
|
|
|
env_node_t *killme = top;
|
|
|
|
for( i=0; locale_variable[i]; i++ )
|
|
{
|
|
if( hash_get( &killme->env, locale_variable[i] ) )
|
|
{
|
|
locale_changed = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if( killme->new_scope )
|
|
{
|
|
has_changed |= killme->export || local_scope_exports( killme->next );
|
|
}
|
|
|
|
top = top->next;
|
|
hash_foreach( &killme->env, &clear_hash_entry );
|
|
hash_destroy( &killme->env );
|
|
free( killme );
|
|
|
|
if( locale_changed )
|
|
handle_locale();
|
|
|
|
}
|
|
else
|
|
{
|
|
debug( 0,
|
|
_( L"Tried to pop empty environment stack." ) );
|
|
sanity_lose();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
Function used with hash_foreach to insert keys of one table into
|
|
another
|
|
*/
|
|
static void add_key_to_hash( void *key,
|
|
void *data,
|
|
void *aux )
|
|
{
|
|
var_entry_t *e = (var_entry_t *)data;
|
|
if( ( e->export && get_names_show_exported) ||
|
|
( !e->export && get_names_show_unexported) )
|
|
{
|
|
hash_put( (hash_table_t *)aux, key, 0 );
|
|
}
|
|
}
|
|
|
|
/**
|
|
Add key to hashtable
|
|
*/
|
|
static void add_to_hash( void *k, void *aux )
|
|
{
|
|
hash_put( (hash_table_t *)aux,
|
|
k,
|
|
0 );
|
|
}
|
|
|
|
/**
|
|
Add key to list
|
|
*/
|
|
static void add_key_to_list( void * key,
|
|
void * val,
|
|
void *aux )
|
|
{
|
|
al_push( (array_list_t *)aux, key );
|
|
}
|
|
|
|
|
|
void env_get_names( array_list_t *l, int flags )
|
|
{
|
|
int show_local = flags & ENV_LOCAL;
|
|
int show_global = flags & ENV_GLOBAL;
|
|
int show_universal = flags & ENV_UNIVERSAL;
|
|
|
|
hash_table_t names;
|
|
env_node_t *n=top;
|
|
|
|
CHECK( l, );
|
|
|
|
get_names_show_exported =
|
|
flags & ENV_EXPORT|| (!(flags & ENV_UNEXPORT));
|
|
get_names_show_unexported =
|
|
flags & ENV_UNEXPORT|| (!(flags & ENV_EXPORT));
|
|
|
|
if( !show_local && !show_global && !show_universal )
|
|
{
|
|
show_local =show_universal = show_global=1;
|
|
}
|
|
|
|
hash_init( &names, &hash_wcs_func, &hash_wcs_cmp );
|
|
|
|
if( show_local )
|
|
{
|
|
while( n )
|
|
{
|
|
if( n == global_env )
|
|
break;
|
|
|
|
hash_foreach2( &n->env,
|
|
add_key_to_hash,
|
|
&names );
|
|
|
|
if( n->new_scope )
|
|
break;
|
|
else
|
|
n = n->next;
|
|
|
|
}
|
|
}
|
|
|
|
if( show_global )
|
|
{
|
|
hash_foreach2( &global_env->env,
|
|
add_key_to_hash,
|
|
&names );
|
|
|
|
if( get_names_show_unexported )
|
|
hash_foreach2( &env_electric, &add_key_to_list, l );
|
|
|
|
if( get_names_show_exported )
|
|
{
|
|
al_push( l, L"COLUMNS" );
|
|
al_push( l, L"LINES" );
|
|
}
|
|
|
|
}
|
|
|
|
if( show_universal )
|
|
{
|
|
array_list_t uni_list;
|
|
al_init( &uni_list );
|
|
|
|
env_universal_get_names( &uni_list,
|
|
get_names_show_exported,
|
|
get_names_show_unexported );
|
|
|
|
al_foreach2( &uni_list, &add_to_hash, &names );
|
|
al_destroy( &uni_list );
|
|
}
|
|
|
|
hash_get_keys( &names, l );
|
|
hash_destroy( &names );
|
|
}
|
|
|
|
/**
|
|
Function used by env_export_arr to iterate over hashtable of variables
|
|
*/
|
|
static void export_func1( void *k, void *v, void *aux )
|
|
{
|
|
var_entry_t *val_entry = (var_entry_t *)v;
|
|
hash_table_t *h = (hash_table_t *)aux;
|
|
|
|
hash_remove( h, k, 0, 0 );
|
|
|
|
if( val_entry->export && wcscmp( val_entry->val, ENV_NULL ) )
|
|
{
|
|
hash_put( h, k, val_entry->val );
|
|
}
|
|
|
|
}
|
|
|
|
static void get_exported( env_node_t *n, hash_table_t *h )
|
|
{
|
|
if( !n )
|
|
return;
|
|
|
|
if( n->new_scope )
|
|
get_exported( global_env, h );
|
|
else
|
|
get_exported( n->next, h );
|
|
|
|
hash_foreach2( &n->env, &export_func1, h );
|
|
}
|
|
|
|
|
|
/**
|
|
Function used by env_export_arr to iterate over hashtable of variables
|
|
*/
|
|
static void export_func2( void *k, void *v, void *aux )
|
|
{
|
|
wchar_t *key = (wchar_t *)k;
|
|
wchar_t *val = (wchar_t *)v;
|
|
|
|
char *ks = wcs2str( key );
|
|
char *vs = wcs2str( val );
|
|
|
|
char *pos = vs;
|
|
|
|
buffer_t *out = (buffer_t *)aux;
|
|
|
|
if( !ks || !vs )
|
|
{
|
|
DIE_MEM();
|
|
}
|
|
|
|
/*
|
|
Make arrays into colon-separated lists
|
|
*/
|
|
while( *pos )
|
|
{
|
|
if( *pos == ARRAY_SEP )
|
|
*pos = ':';
|
|
pos++;
|
|
}
|
|
int nil = 0;
|
|
|
|
b_append( out, ks, strlen(ks) );
|
|
b_append( out, "=", 1 );
|
|
b_append( out, vs, strlen(vs) );
|
|
b_append( out, &nil, 1 );
|
|
|
|
free( ks );
|
|
free( vs );
|
|
}
|
|
|
|
char **env_export_arr( int recalc )
|
|
{
|
|
if( recalc && !proc_had_barrier)
|
|
{
|
|
proc_had_barrier=1;
|
|
env_universal_barrier();
|
|
}
|
|
|
|
if( has_changed )
|
|
{
|
|
array_list_t uni;
|
|
hash_table_t vals;
|
|
int prev_was_null=1;
|
|
int pos=0;
|
|
int i;
|
|
|
|
debug( 4, L"env_export_arr() recalc" );
|
|
|
|
hash_init( &vals, &hash_wcs_func, &hash_wcs_cmp );
|
|
|
|
get_exported( top, &vals );
|
|
|
|
al_init( &uni );
|
|
env_universal_get_names( &uni, 1, 0 );
|
|
for( i=0; i<al_get_count( &uni ); i++ )
|
|
{
|
|
wchar_t *key = (wchar_t *)al_get( &uni, i );
|
|
wchar_t *val = env_universal_get( key );
|
|
if( wcscmp( val, ENV_NULL) && !hash_get( &vals, key ) )
|
|
hash_put( &vals, key, val );
|
|
}
|
|
al_destroy( &uni );
|
|
|
|
export_buffer.used=0;
|
|
|
|
hash_foreach2( &vals, &export_func2, &export_buffer );
|
|
hash_destroy( &vals );
|
|
|
|
export_arr = realloc( export_arr,
|
|
sizeof(char *)*(hash_get_count( &vals) + 1) );
|
|
|
|
for( i=0; i<export_buffer.used; i++ )
|
|
{
|
|
if( prev_was_null )
|
|
{
|
|
export_arr[pos++]= &export_buffer.buff[i];
|
|
debug( 3, L"%s", &export_buffer.buff[i]);
|
|
}
|
|
prev_was_null = (export_buffer.buff[i]==0);
|
|
}
|
|
export_arr[pos]=0;
|
|
has_changed=0;
|
|
|
|
}
|
|
return export_arr;
|
|
}
|