FreeRDP/winpr/libwinpr/environment/test/TestEnvironmentSetEnvironmentVariable.c

70 lines
1.4 KiB
C
Raw Normal View History

#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/tchar.h>
#include <winpr/environment.h>
#include <winpr/error.h>
#define TEST_NAME "WINPR_TEST_VARIABLE"
#define TEST_VALUE "WINPR_TEST_VALUE"
int TestEnvironmentSetEnvironmentVariable(int argc, char* argv[])
{
2019-04-05 14:17:24 +03:00
int rc = -1;
DWORD nSize;
2019-04-05 14:17:24 +03:00
LPSTR lpBuffer = NULL;
DWORD error = 0;
SetEnvironmentVariableA(TEST_NAME, TEST_VALUE);
nSize = GetEnvironmentVariableA(TEST_NAME, NULL, 0);
2019-04-05 14:17:24 +03:00
/* check if value returned is len + 1 ) */
if (nSize != strlen(TEST_VALUE) + 1)
{
printf("GetEnvironmentVariableA not found error\n");
return -1;
}
lpBuffer = (LPSTR) malloc(nSize);
2019-04-05 14:17:24 +03:00
if (!lpBuffer)
return -1;
nSize = GetEnvironmentVariableA(TEST_NAME, lpBuffer, nSize);
if (nSize != strlen(TEST_VALUE))
{
printf("GetEnvironmentVariableA wrong size returned\n");
2019-04-05 14:17:24 +03:00
goto fail;
}
if (strcmp(lpBuffer, TEST_VALUE) != 0)
{
printf("GetEnvironmentVariableA returned value doesn't match\n");
2019-04-05 14:17:24 +03:00
goto fail;
}
2019-04-05 14:17:24 +03:00
nSize = GetEnvironmentVariableA("__xx__notset_", lpBuffer, nSize);
error = GetLastError();
2019-04-05 14:17:24 +03:00
if (0 != nSize || ERROR_ENVVAR_NOT_FOUND != error)
{
printf("GetEnvironmentVariableA not found error\n");
2019-04-05 14:17:24 +03:00
goto fail;
}
/* clear variable */
SetEnvironmentVariableA(TEST_NAME, NULL);
nSize = GetEnvironmentVariableA(TEST_VALUE, NULL, 0);
2019-04-05 14:17:24 +03:00
if (0 != nSize)
{
printf("SetEnvironmentVariableA failed to clear variable\n");
2019-04-05 14:17:24 +03:00
goto fail;
}
2019-04-05 14:17:24 +03:00
rc = 0;
fail:
free(lpBuffer);
return rc;
}