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 | ** |
11 | ** |
12 | **=========================================================*/ |
13 | |
14 | /* Depends on SetEnvironmentVariableW (because we're implmenting the wide |
15 | version) and strcmp() */ |
16 | |
17 | #define UNICODE |
18 | #include <palsuite.h> |
19 | |
20 | int __cdecl main(int argc, char *argv[]) { |
21 | |
22 | /* Define some buffers needed for the function */ |
23 | WCHAR * pResultBuffer = NULL; |
24 | WCHAR SomeEnvironmentVariable[] = {'P','A','L','T','E','S','T','\0'}; |
25 | WCHAR TheEnvironmentValue[] = {'T','E','S','T','\0'}; |
26 | int size; |
27 | |
28 | /* |
29 | * Initialize the PAL and return FAILURE if this fails |
30 | */ |
31 | |
32 | if(0 != (PAL_Initialize(argc, argv))) |
33 | { |
34 | return FAIL; |
35 | } |
36 | |
37 | SetEnvironmentVariable(SomeEnvironmentVariable, |
38 | TheEnvironmentValue); |
39 | |
40 | |
41 | /* Normal case, PATH should fit into this buffer */ |
42 | size = GetEnvironmentVariable(convert("PALTEST" ), // Variable Name |
43 | pResultBuffer, // Buffer for Value |
44 | 0); // Buffer size |
45 | |
46 | pResultBuffer = (WCHAR*)malloc(size*sizeof(WCHAR)); |
47 | |
48 | GetEnvironmentVariable(convert("PALTEST" ), |
49 | pResultBuffer, |
50 | size); |
51 | |
52 | if(wcsncmp(pResultBuffer,convert("TEST" ),wcslen(pResultBuffer) * 2) != 0) |
53 | { |
54 | Fail("ERROR: The value in the buffer should have been 'TEST' but was " |
55 | "really '%s'." ,convertC(pResultBuffer)); |
56 | } |
57 | |
58 | free(pResultBuffer); |
59 | |
60 | PAL_Terminate(); |
61 | return PASS; |
62 | } |
63 | |
64 | |