1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
/*============================================================
** Source : test3.c
**
** Purpose: Test for SetEnvironmentVariableA() function
** Create environment variables that differ only
** in case and verify that they return the appropriate
** value on the BSD environment.
**
**
===========================================================*/
#include <palsuite.h>
int __cdecl main(int argc, char *argv[])
{
#if WIN32
return PASS;
#else
/* Define some buffers needed for the function */
char * pResultBuffer = NULL;
char FirstEnvironmentVariable[] = {"PALTEST"};
char FirstEnvironmentValue[] = {"FIRST"};
char SecondEnvironmentVariable[] = {"paltest"};
char SecondEnvironmentValue[] = {"SECOND"};
DWORD size = 0;
BOOL bRc = TRUE;
/*
* Initialize the PAL and return FAILURE if this fails
*/
if(0 != (PAL_Initialize(argc, argv)))
{
return FAIL;
}
/* Set the first environment variable */
bRc = SetEnvironmentVariableA(FirstEnvironmentVariable,
FirstEnvironmentValue);
if(!bRc)
{
Fail("ERROR: SetEnvironmentVariable failed to set a "
"proper environment variable with error %u.\n",
GetLastError());
}
/* Set the second environment Variable */
bRc = SetEnvironmentVariableA(SecondEnvironmentVariable,
SecondEnvironmentValue);
if(!bRc)
{
Fail("ERROR: SetEnvironmentVariable failed to set a "
"proper environment variable with error %u.\n",
GetLastError());
}
/* Normal case, PATH should fit into this buffer */
size = GetEnvironmentVariableA(FirstEnvironmentVariable,
pResultBuffer,
0);
/* increase size to account for the null char at the end */
size = size + 1;
pResultBuffer = (char*)malloc(sizeof(char)*size);
if ( pResultBuffer == NULL )
{
Fail("ERROR: Failed to allocate memory for pResultBuffer pointer.\n");
}
/* Try to retrieve the value of the first environment variable */
GetEnvironmentVariable(FirstEnvironmentVariable,
pResultBuffer,
size);
if ( pResultBuffer == NULL )
{
free(pResultBuffer);
Fail("ERROR: GetEnvironmentVariable failed to return a value "
"from a proper environment variable with error %u.\n",
GetLastError());
}
/* Compare the strings to see that the correct variable was returned */
if(strcmp(pResultBuffer,FirstEnvironmentValue) != 0)
{
Trace("ERROR: The value in the buffer should have been '%s' but "
"was really '%s'.\n",FirstEnvironmentValue, pResultBuffer);
free(pResultBuffer);
Fail("");
}
free(pResultBuffer);
/* Reallocate the memory for the string */
pResultBuffer = (char*)malloc(sizeof(char)*size);
if ( pResultBuffer == NULL )
{
Fail("ERROR: Failed to allocate memory for pResultBuffer pointer.\n");
}
/* Try retrieving the value of the first variable, even though the
second variable has the same spelling and only differs in case */
GetEnvironmentVariable(SecondEnvironmentVariable,
pResultBuffer,
size);
if ( pResultBuffer == NULL )
{
Fail("ERROR: GetEnvironmentVariable failed to return a value "
"from a proper environment variable with error %u.\n",
GetLastError());
}
/* Compare the two strings to confirm that the right value is returned */
if(strcmp(pResultBuffer,SecondEnvironmentValue) != 0)
{
Trace("ERROR: The value in the buffer should have been '%s' but "
"was really '%s'.\n",SecondEnvironmentValue,pResultBuffer);
free(pResultBuffer);
Fail("");
}
free(pResultBuffer);
PAL_Terminate();
return PASS;
#endif
}
|
