/
SDL_getenv.c
247 lines (218 loc) · 5.55 KB
1
2
/*
SDL - Simple DirectMedia Layer
3
Copyright (C) 1997-2006 Sam Lantinga
4
5
This library is free software; you can redistribute it and/or
6
modify it under the terms of the GNU Lesser General Public
7
License as published by the Free Software Foundation; either
8
version 2.1 of the License, or (at your option) any later version.
9
10
11
12
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
13
Lesser General Public License for more details.
14
15
16
17
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
18
19
Sam Lantinga
20
slouken@libsdl.org
21
*/
22
#include "SDL_config.h"
23
24
#include "SDL_stdinc.h"
25
26
#ifndef HAVE_GETENV
27
28
#if defined(__WIN32__) && !defined(_WIN32_WCE)
29
30
31
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
32
33
34
35
36
37
38
39
40
41
42
43
44
/* Note this isn't thread-safe! */
static char *SDL_envmem = NULL; /* Ugh, memory leak */
static DWORD SDL_envmemlen = 0;
/* Put a variable of the form "name=value" into the environment */
int SDL_putenv(const char *variable)
{
DWORD bufferlen;
char *value;
const char *sep;
45
sep = SDL_strchr(variable, '=');
46
47
48
if ( sep == NULL ) {
return -1;
}
49
bufferlen = SDL_strlen(variable)+1;
50
if ( bufferlen > SDL_envmemlen ) {
51
char *newmem = (char *)SDL_realloc(SDL_envmem, bufferlen);
52
53
54
55
56
57
if ( newmem == NULL ) {
return -1;
}
SDL_envmem = newmem;
SDL_envmemlen = bufferlen;
}
58
SDL_strlcpy(SDL_envmem, variable, bufferlen);
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
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)
{
DWORD bufferlen;
bufferlen = GetEnvironmentVariable(name, SDL_envmem, SDL_envmemlen);
if ( bufferlen == 0 ) {
return NULL;
}
if ( bufferlen > SDL_envmemlen ) {
77
char *newmem = (char *)SDL_realloc(SDL_envmem, bufferlen);
78
79
80
81
82
83
84
85
86
87
88
89
if ( newmem == NULL ) {
return NULL;
}
SDL_envmem = newmem;
SDL_envmemlen = bufferlen;
GetEnvironmentVariable(name, SDL_envmem, SDL_envmemlen);
}
return SDL_envmem;
}
#else /* roll our own */
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
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 */
116
new_variable = SDL_strdup(variable);
117
118
119
120
121
122
123
124
125
126
127
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 ) {
128
if ( SDL_strncmp(SDL_env[i], name, len) == 0 ) {
129
130
131
132
133
break;
}
}
/* If we found it, just replace the entry */
if ( SDL_env[i] ) {
134
SDL_free(SDL_env[i]);
135
136
137
138
139
140
141
SDL_env[i] = new_variable;
added = 1;
}
}
/* Didn't find it in the environment, expand and add */
if ( ! added ) {
142
new_env = SDL_realloc(SDL_env, (i+2)*sizeof(char *));
143
144
145
146
147
148
if ( new_env ) {
SDL_env = new_env;
SDL_env[i++] = new_variable;
SDL_env[i++] = (char *)0;
added = 1;
} else {
149
SDL_free(new_variable);
150
151
152
153
154
155
156
157
158
159
160
161
162
}
}
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 ) {
163
len = SDL_strlen(name);
164
for ( i=0; SDL_env[i] && !value; ++i ) {
165
if ( (SDL_strncmp(SDL_env[i], name, len) == 0) &&
166
167
168
169
170
171
172
173
(SDL_env[i][len] == '=') ) {
value = &SDL_env[i][len+1];
}
}
}
return value;
}
174
#endif /* __WIN32__ */
175
176
177
#endif /* !HAVE_GETENV */
178
179
180
181
182
183
184
185
186
#ifdef TEST_MAIN
#include <stdio.h>
int main(int argc, char *argv[])
{
char *value;
printf("Checking for non-existent variable... ");
fflush(stdout);
187
if ( ! SDL_getenv("EXISTS") ) {
188
189
190
191
192
193
printf("okay\n");
} else {
printf("failed\n");
}
printf("Setting FIRST=VALUE1 in the environment... ");
fflush(stdout);
194
if ( SDL_putenv("FIRST=VALUE1") == 0 ) {
195
196
197
198
199
200
printf("okay\n");
} else {
printf("failed\n");
}
printf("Getting FIRST from the environment... ");
fflush(stdout);
201
202
value = SDL_getenv("FIRST");
if ( value && (SDL_strcmp(value, "VALUE1") == 0) ) {
203
204
205
206
207
208
printf("okay\n");
} else {
printf("failed\n");
}
printf("Setting SECOND=VALUE2 in the environment... ");
fflush(stdout);
209
if ( SDL_putenv("SECOND=VALUE2") == 0 ) {
210
211
212
213
214
215
printf("okay\n");
} else {
printf("failed\n");
}
printf("Getting SECOND from the environment... ");
fflush(stdout);
216
217
value = SDL_getenv("SECOND");
if ( value && (SDL_strcmp(value, "VALUE2") == 0) ) {
218
219
220
221
222
223
printf("okay\n");
} else {
printf("failed\n");
}
printf("Setting FIRST=NOVALUE in the environment... ");
fflush(stdout);
224
if ( SDL_putenv("FIRST=NOVALUE") == 0 ) {
225
226
227
228
229
230
printf("okay\n");
} else {
printf("failed\n");
}
printf("Getting FIRST from the environment... ");
fflush(stdout);
231
232
value = SDL_getenv("FIRST");
if ( value && (SDL_strcmp(value, "NOVALUE") == 0) ) {
233
234
235
236
237
238
printf("okay\n");
} else {
printf("failed\n");
}
printf("Checking for non-existent variable... ");
fflush(stdout);
239
if ( ! SDL_getenv("EXISTS") ) {
240
241
242
243
244
245
246
printf("okay\n");
} else {
printf("failed\n");
}
return(0);
}
#endif /* TEST_MAIN */