1 | // Licensed to the .NET Foundation under one or more agreements. |
2 | // The .NET Foundation licenses this file to you under the MIT license. |
3 | // See the LICENSE file in the project root for more information. |
4 | |
5 | /*============================================================ |
6 | ** |
7 | ** Source : test.c |
8 | ** |
9 | ** Purpose: Test for GetEnvironmentVariable() function |
10 | ** Set an Environment Variable, then use GetEnvironmentVariable to |
11 | ** retrieve it -- ensure that it retrieves properly. |
12 | ** |
13 | ** |
14 | **=========================================================*/ |
15 | |
16 | /* Depends on SetEnvironmentVariableW (because we're implmenting |
17 | the wide version) and strcmp() |
18 | */ |
19 | |
20 | #include <palsuite.h> |
21 | |
22 | int __cdecl main(int argc, char *argv[]) { |
23 | |
24 | /* Define some buffers needed for the function */ |
25 | char * pResultBuffer = NULL; |
26 | WCHAR SomeEnvironmentVariable[] = {'P','A','L','T','E','S','T','\0'}; |
27 | WCHAR TheEnvironmentValue[] = {'T','E','S','T','\0'}; |
28 | int size = 0; |
29 | |
30 | /* |
31 | * Initialize the PAL and return FAILURE if this fails |
32 | */ |
33 | |
34 | if(0 != (PAL_Initialize(argc, argv))) |
35 | { |
36 | return FAIL; |
37 | } |
38 | |
39 | SetEnvironmentVariableW(SomeEnvironmentVariable, |
40 | TheEnvironmentValue); |
41 | |
42 | |
43 | /* Normal case, PATH should fit into this buffer */ |
44 | size = GetEnvironmentVariable("PALTEST" , // Variable Name |
45 | pResultBuffer, // Buffer for Value |
46 | 0); // Buffer size |
47 | |
48 | pResultBuffer = (char*)malloc(size); |
49 | if ( pResultBuffer == NULL ) |
50 | { |
51 | Fail("ERROR: Failed to allocate memory for pResultBuffer pointer. " |
52 | "Can't properly exec test case without this.\n" ); |
53 | } |
54 | |
55 | |
56 | GetEnvironmentVariable("PALTEST" , |
57 | pResultBuffer, |
58 | size); |
59 | |
60 | if(strcmp(pResultBuffer,"TEST" ) != 0) |
61 | { |
62 | free(pResultBuffer); |
63 | Fail("ERROR: The value in the buffer should have been 'TEST' but " |
64 | "was really '%s'.\n" ,pResultBuffer); |
65 | |
66 | } |
67 | |
68 | free(pResultBuffer); |
69 | |
70 | PAL_Terminate(); |
71 | return PASS; |
72 | } |
73 | |
74 | |
75 | |
76 | |