248 lines
5.6 KiB
C
248 lines
5.6 KiB
C
/*
|
|
SDL - Simple DirectMedia Layer
|
|
Copyright (C) 1997-2012 Sam Lantinga
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
Sam Lantinga
|
|
slouken@libsdl.org
|
|
*/
|
|
#include "SDL_config.h"
|
|
|
|
#include "SDL_stdinc.h"
|
|
|
|
#ifndef HAVE_GETENV
|
|
|
|
#if defined(__WIN32__) && !defined(_WIN32_WCE) && !defined(__SYMBIAN32__)
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
|
|
/* Note this isn't thread-safe! */
|
|
|
|
static char *SDL_envmem = NULL; /* Ugh, memory leak */
|
|
static size_t SDL_envmemlen = 0;
|
|
|
|
/* Put a variable of the form "name=value" into the environment */
|
|
int SDL_putenv(const char *variable)
|
|
{
|
|
size_t bufferlen;
|
|
char *value;
|
|
const char *sep;
|
|
|
|
sep = SDL_strchr(variable, '=');
|
|
if ( sep == NULL ) {
|
|
return -1;
|
|
}
|
|
bufferlen = SDL_strlen(variable)+1;
|
|
if ( bufferlen > SDL_envmemlen ) {
|
|
char *newmem = (char *)SDL_realloc(SDL_envmem, bufferlen);
|
|
if ( newmem == NULL ) {
|
|
return -1;
|
|
}
|
|
SDL_envmem = newmem;
|
|
SDL_envmemlen = bufferlen;
|
|
}
|
|
SDL_strlcpy(SDL_envmem, variable, bufferlen);
|
|
value = SDL_envmem + (sep - variable);
|
|
*value++ = '\0';
|
|
if ( !SetEnvironmentVariable(SDL_envmem, *value ? value : NULL) ) {
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/* Retrieve a variable named "name" from the environment */
|
|
char *SDL_getenv(const char *name)
|
|
{
|
|
size_t bufferlen;
|
|
|
|
bufferlen = GetEnvironmentVariable(name, SDL_envmem, (DWORD)SDL_envmemlen);
|
|
if ( bufferlen == 0 ) {
|
|
return NULL;
|
|
}
|
|
if ( bufferlen > SDL_envmemlen ) {
|
|
char *newmem = (char *)SDL_realloc(SDL_envmem, bufferlen);
|
|
if ( newmem == NULL ) {
|
|
return NULL;
|
|
}
|
|
SDL_envmem = newmem;
|
|
SDL_envmemlen = bufferlen;
|
|
GetEnvironmentVariable(name, SDL_envmem, (DWORD)SDL_envmemlen);
|
|
}
|
|
return SDL_envmem;
|
|
}
|
|
|
|
#else /* roll our own */
|
|
|
|
static char **SDL_env = (char **)0;
|
|
|
|
/* Put a variable of the form "name=value" into the environment */
|
|
int SDL_putenv(const char *variable)
|
|
{
|
|
const char *name, *value;
|
|
int added;
|
|
int len, i;
|
|
char **new_env;
|
|
char *new_variable;
|
|
|
|
/* A little error checking */
|
|
if ( ! variable ) {
|
|
return(-1);
|
|
}
|
|
name = variable;
|
|
for ( value=variable; *value && (*value != '='); ++value ) {
|
|
/* Keep looking for '=' */ ;
|
|
}
|
|
if ( *value ) {
|
|
++value;
|
|
} else {
|
|
return(-1);
|
|
}
|
|
|
|
/* Allocate memory for the variable */
|
|
new_variable = SDL_strdup(variable);
|
|
if ( ! new_variable ) {
|
|
return(-1);
|
|
}
|
|
|
|
/* Actually put it into the environment */
|
|
added = 0;
|
|
i = 0;
|
|
if ( SDL_env ) {
|
|
/* Check to see if it's already there... */
|
|
len = (value - name);
|
|
for ( ; SDL_env[i]; ++i ) {
|
|
if ( SDL_strncmp(SDL_env[i], name, len) == 0 ) {
|
|
break;
|
|
}
|
|
}
|
|
/* If we found it, just replace the entry */
|
|
if ( SDL_env[i] ) {
|
|
SDL_free(SDL_env[i]);
|
|
SDL_env[i] = new_variable;
|
|
added = 1;
|
|
}
|
|
}
|
|
|
|
/* Didn't find it in the environment, expand and add */
|
|
if ( ! added ) {
|
|
new_env = SDL_realloc(SDL_env, (i+2)*sizeof(char *));
|
|
if ( new_env ) {
|
|
SDL_env = new_env;
|
|
SDL_env[i++] = new_variable;
|
|
SDL_env[i++] = (char *)0;
|
|
added = 1;
|
|
} else {
|
|
SDL_free(new_variable);
|
|
}
|
|
}
|
|
return (added ? 0 : -1);
|
|
}
|
|
|
|
/* Retrieve a variable named "name" from the environment */
|
|
char *SDL_getenv(const char *name)
|
|
{
|
|
int len, i;
|
|
char *value;
|
|
|
|
value = (char *)0;
|
|
if ( SDL_env ) {
|
|
len = SDL_strlen(name);
|
|
for ( i=0; SDL_env[i] && !value; ++i ) {
|
|
if ( (SDL_strncmp(SDL_env[i], name, len) == 0) &&
|
|
(SDL_env[i][len] == '=') ) {
|
|
value = &SDL_env[i][len+1];
|
|
}
|
|
}
|
|
}
|
|
return value;
|
|
}
|
|
|
|
#endif /* __WIN32__ */
|
|
|
|
#endif /* !HAVE_GETENV */
|
|
|
|
#ifdef TEST_MAIN
|
|
#include <stdio.h>
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
char *value;
|
|
|
|
printf("Checking for non-existent variable... ");
|
|
fflush(stdout);
|
|
if ( ! SDL_getenv("EXISTS") ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Setting FIRST=VALUE1 in the environment... ");
|
|
fflush(stdout);
|
|
if ( SDL_putenv("FIRST=VALUE1") == 0 ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Getting FIRST from the environment... ");
|
|
fflush(stdout);
|
|
value = SDL_getenv("FIRST");
|
|
if ( value && (SDL_strcmp(value, "VALUE1") == 0) ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Setting SECOND=VALUE2 in the environment... ");
|
|
fflush(stdout);
|
|
if ( SDL_putenv("SECOND=VALUE2") == 0 ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Getting SECOND from the environment... ");
|
|
fflush(stdout);
|
|
value = SDL_getenv("SECOND");
|
|
if ( value && (SDL_strcmp(value, "VALUE2") == 0) ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Setting FIRST=NOVALUE in the environment... ");
|
|
fflush(stdout);
|
|
if ( SDL_putenv("FIRST=NOVALUE") == 0 ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Getting FIRST from the environment... ");
|
|
fflush(stdout);
|
|
value = SDL_getenv("FIRST");
|
|
if ( value && (SDL_strcmp(value, "NOVALUE") == 0) ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
printf("Checking for non-existent variable... ");
|
|
fflush(stdout);
|
|
if ( ! SDL_getenv("EXISTS") ) {
|
|
printf("okay\n");
|
|
} else {
|
|
printf("failed\n");
|
|
}
|
|
return(0);
|
|
}
|
|
#endif /* TEST_MAIN */
|
|
|