reactos/rostests/apitests/w32knapi/ntuser/NtUserEnumDisplaySettings.c

static struct
{
	DEVMODEW devmode;
	CHAR buffer[0xffff];
} data;

INT
TEST_NtUserEnumDisplaySettings(PTESTINFO pti)
{
	UNICODE_STRING usDeviceName;
	WCHAR szName[] = L"DISPLAY";
	NTSTATUS Status;
	INT i;

	SetLastError(ERROR_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 0, 0, 0);
	TEST(Status == STATUS_ACCESS_VIOLATION);
	TEST(GetLastError() == ERROR_SUCCESS);

	data.devmode.dmDriverExtra = 0;
	for (i = 0; i < 2 * sizeof(DEVMODEW); i++)
	{
		data.devmode.dmSize = i;
		Status = NtUserEnumDisplaySettings(NULL, 1000, (DEVMODEW*)&data, 0);
		if (i != sizeof(DEVMODEW))
		{
			TEST(Status == STATUS_BUFFER_TOO_SMALL);
		}
	}
	TEST(GetLastError() == ERROR_SUCCESS);

	usDeviceName.Buffer = NULL;
	usDeviceName.Length = 0;
	usDeviceName.MaximumLength = 0;
	Status = NtUserEnumDisplaySettings(&usDeviceName, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_BUFFER_TOO_SMALL);
	Status = NtUserEnumDisplaySettings(&usDeviceName, -4, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_BUFFER_TOO_SMALL);

	data.devmode.dmSize = sizeof(DEVMODEW);
	data.devmode.dmDriverExtra = 0xffff;
	Status = NtUserEnumDisplaySettings(NULL, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);

	data.devmode.dmSize = sizeof(DEVMODEW);
	data.devmode.dmDriverExtra = 0;
	Status = NtUserEnumDisplaySettings(NULL, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);

	usDeviceName.Buffer = NULL;
	usDeviceName.Length = 0;
	usDeviceName.MaximumLength = 0;
	Status = NtUserEnumDisplaySettings(&usDeviceName, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_1);
	Status = NtUserEnumDisplaySettings(&usDeviceName, -4, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_1);

	Status = NtUserEnumDisplaySettings(NULL, 0, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 1, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 2, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 4, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 8, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);

	Status = NtUserEnumDisplaySettings(NULL, 247, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, 248, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_2);

	Status = NtUserEnumDisplaySettings(NULL, -1, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, -2, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, -3, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_SUCCESS);
	Status = NtUserEnumDisplaySettings(NULL, -4, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_2);

	Status = NtUserEnumDisplaySettings(&usDeviceName, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_1);

	usDeviceName.Buffer = szName;
	usDeviceName.Length = wcslen(szName);
	usDeviceName.MaximumLength = usDeviceName.Length;
	Status = NtUserEnumDisplaySettings(&usDeviceName, ENUM_CURRENT_SETTINGS, (DEVMODEW*)&data, 0);
	TEST(Status == STATUS_INVALID_PARAMETER_1);

	Status = NtUserEnumDisplaySettings(&usDeviceName, 1000, (DEVMODEW*)&data, 123456);
	TEST(Status == STATUS_INVALID_PARAMETER_1);

	TEST(GetLastError() == ERROR_SUCCESS);

	return APISTATUS_NORMAL;
}