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

#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[])
{
	int rc = -1;
	DWORD nSize = 0;
	LPSTR lpBuffer = nullptr;
	DWORD error = 0;

	WINPR_UNUSED(argc);
	WINPR_UNUSED(argv);

	SetEnvironmentVariableA(TEST_NAME, TEST_VALUE);
	nSize = GetEnvironmentVariableA(TEST_NAME, nullptr, 0);

	/* check if value returned is len + 1 ) */
	if (nSize != strnlen(TEST_VALUE, sizeof(TEST_VALUE)) + 1)
	{
		printf("GetEnvironmentVariableA not found error\n");
		return -1;
	}

	lpBuffer = (LPSTR)malloc(nSize);

	if (!lpBuffer)
		return -1;

	nSize = GetEnvironmentVariableA(TEST_NAME, lpBuffer, nSize);

	if (nSize != strnlen(TEST_VALUE, sizeof(TEST_VALUE)))
	{
		printf("GetEnvironmentVariableA wrong size returned\n");
		goto fail;
	}

	if (strcmp(lpBuffer, TEST_VALUE) != 0)
	{
		printf("GetEnvironmentVariableA returned value doesn't match\n");
		goto fail;
	}

	nSize = GetEnvironmentVariableA("__xx__notset_", lpBuffer, nSize);
	error = GetLastError();

	if (0 != nSize || ERROR_ENVVAR_NOT_FOUND != error)
	{
		printf("GetEnvironmentVariableA not found error\n");
		goto fail;
	}

	/* clear variable */
	SetEnvironmentVariableA(TEST_NAME, nullptr);
	nSize = GetEnvironmentVariableA(TEST_VALUE, nullptr, 0);

	if (0 != nSize)
	{
		printf("SetEnvironmentVariableA failed to clear variable\n");
		goto fail;
	}

	rc = 0;
fail:
	free(lpBuffer);
	return rc;
}