Milestone 5: deliver embedded RDP sessions and lifecycle hardening
This commit is contained in:
160
third_party/FreeRDP/winpr/libwinpr/utils/test/TestIni.c
vendored
Normal file
160
third_party/FreeRDP/winpr/libwinpr/utils/test/TestIni.c
vendored
Normal file
@@ -0,0 +1,160 @@
|
||||
|
||||
#include <winpr/crt.h>
|
||||
#include <winpr/ini.h>
|
||||
|
||||
static const char TEST_INI_01[] = "; This is a sample .ini config file\n"
|
||||
"\n"
|
||||
"[first_section]\n"
|
||||
"one = 1\n"
|
||||
"five = 5\n"
|
||||
"animal = BIRD\n"
|
||||
"\n"
|
||||
"[second_section]\n"
|
||||
"path = \"/usr/local/bin\"\n"
|
||||
"URL = \"http://www.example.com/~username\"\n"
|
||||
"\n";
|
||||
|
||||
static const char TEST_INI_02[] = "[FreeRDS]\n"
|
||||
"prefix=\"/usr/local\"\n"
|
||||
"bindir=\"bin\"\n"
|
||||
"sbindir=\"sbin\"\n"
|
||||
"libdir=\"lib\"\n"
|
||||
"datarootdir=\"share\"\n"
|
||||
"localstatedir=\"var\"\n"
|
||||
"sysconfdir=\"etc\"\n"
|
||||
"\n";
|
||||
|
||||
static const char TEST_INI_03[] = "[FreeRDS]\n"
|
||||
"prefix=\"/usr/local\"\n"
|
||||
"bindir=\"bin\"\n"
|
||||
"# some illegal string\n"
|
||||
"sbindir=\"sbin\"\n"
|
||||
"libdir=\"lib\"\n"
|
||||
"invalid key-value pair\n"
|
||||
"datarootdir=\"share\"\n"
|
||||
"localstatedir=\"var\"\n"
|
||||
"sysconfdir=\"etc\"\n"
|
||||
"\n";
|
||||
|
||||
int TestIni(int argc, char* argv[])
|
||||
{
|
||||
int rc = -1;
|
||||
size_t nKeys = 0;
|
||||
size_t nSections = 0;
|
||||
UINT32 iValue = 0;
|
||||
wIniFile* ini = nullptr;
|
||||
const char* sValue = nullptr;
|
||||
char** keyNames = nullptr;
|
||||
char** sectionNames = nullptr;
|
||||
|
||||
WINPR_UNUSED(argc);
|
||||
WINPR_UNUSED(argv);
|
||||
|
||||
/* First Sample */
|
||||
ini = IniFile_New();
|
||||
if (!ini)
|
||||
goto fail;
|
||||
|
||||
if (IniFile_ReadBuffer(ini, TEST_INI_01) < 0)
|
||||
goto fail;
|
||||
|
||||
free((void*)sectionNames);
|
||||
sectionNames = IniFile_GetSectionNames(ini, &nSections);
|
||||
if (!sectionNames && (nSections > 0))
|
||||
goto fail;
|
||||
|
||||
for (size_t i = 0; i < nSections; i++)
|
||||
{
|
||||
free((void*)keyNames);
|
||||
keyNames = IniFile_GetSectionKeyNames(ini, sectionNames[i], &nKeys);
|
||||
printf("[%s]\n", sectionNames[i]);
|
||||
if (!keyNames && (nKeys > 0))
|
||||
goto fail;
|
||||
for (size_t j = 0; j < nKeys; j++)
|
||||
{
|
||||
sValue = IniFile_GetKeyValueString(ini, sectionNames[i], keyNames[j]);
|
||||
printf("%s = %s\n", keyNames[j], sValue);
|
||||
}
|
||||
}
|
||||
|
||||
iValue = IniFile_GetKeyValueInt(ini, "first_section", "one");
|
||||
|
||||
if (iValue != 1)
|
||||
{
|
||||
printf("IniFile_GetKeyValueInt failure\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
iValue = IniFile_GetKeyValueInt(ini, "first_section", "five");
|
||||
|
||||
if (iValue != 5)
|
||||
{
|
||||
printf("IniFile_GetKeyValueInt failure\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
sValue = IniFile_GetKeyValueString(ini, "first_section", "animal");
|
||||
|
||||
if (strcmp(sValue, "BIRD") != 0)
|
||||
{
|
||||
printf("IniFile_GetKeyValueString failure\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
sValue = IniFile_GetKeyValueString(ini, "second_section", "path");
|
||||
|
||||
if (strcmp(sValue, "/usr/local/bin") != 0)
|
||||
{
|
||||
printf("IniFile_GetKeyValueString failure\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
sValue = IniFile_GetKeyValueString(ini, "second_section", "URL");
|
||||
|
||||
if (strcmp(sValue, "http://www.example.com/~username") != 0)
|
||||
{
|
||||
printf("IniFile_GetKeyValueString failure\n");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
IniFile_Free(ini);
|
||||
/* Second Sample */
|
||||
ini = IniFile_New();
|
||||
if (!ini)
|
||||
goto fail;
|
||||
if (IniFile_ReadBuffer(ini, TEST_INI_02) < 0)
|
||||
goto fail;
|
||||
free((void*)sectionNames);
|
||||
sectionNames = IniFile_GetSectionNames(ini, &nSections);
|
||||
if (!sectionNames && (nSections > 0))
|
||||
goto fail;
|
||||
|
||||
for (size_t i = 0; i < nSections; i++)
|
||||
{
|
||||
free((void*)keyNames);
|
||||
keyNames = IniFile_GetSectionKeyNames(ini, sectionNames[i], &nKeys);
|
||||
printf("[%s]\n", sectionNames[i]);
|
||||
|
||||
if (!keyNames && (nKeys > 0))
|
||||
goto fail;
|
||||
for (size_t j = 0; j < nKeys; j++)
|
||||
{
|
||||
sValue = IniFile_GetKeyValueString(ini, sectionNames[i], keyNames[j]);
|
||||
printf("%s = %s\n", keyNames[j], sValue);
|
||||
}
|
||||
}
|
||||
|
||||
IniFile_Free(ini);
|
||||
/* Third sample - invalid input */
|
||||
ini = IniFile_New();
|
||||
|
||||
if (IniFile_ReadBuffer(ini, TEST_INI_03) != -1)
|
||||
goto fail;
|
||||
|
||||
rc = 0;
|
||||
fail:
|
||||
free((void*)keyNames);
|
||||
free((void*)sectionNames);
|
||||
IniFile_Free(ini);
|
||||
return rc;
|
||||
}
|
||||
Reference in New Issue
Block a user