TestIni.c

 
#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";
 
static const char TEST_INI_04[] = "[foobar]\r\n"
                                  "\r\n"
                                  "\r\n"
                                  "prefix=\"someprefix\"\r\n";
 
WINPR_ATTR_NODISCARD
static BOOL test04(void)
{
  BOOL rc = FALSE;
  size_t nSections = 0;
 
  /* First Sample */
  char** sectionNames = nullptr;
  wIniFile* ini = IniFile_New();
  if (!ini)
    goto fail;
 
  if (IniFile_ReadBuffer(ini, TEST_INI_04) < 0)
    goto fail;
 
  sectionNames = IniFile_GetSectionNames(ini, &nSections);
  if (!sectionNames && (nSections > 0))
    goto fail;
 
  if (nSections != 1)
    goto fail;
 
  for (size_t i = 0; i < nSections; i++)
  {
    size_t nKeys = 0;
    char** keyNames = IniFile_GetSectionKeyNames(ini, sectionNames[i], &nKeys);
    printf("[%s]\n", sectionNames[i]);
    if ((!keyNames && (nKeys > 0)) || (nKeys != 1))
    {
      free((void*)keyNames);
      goto fail;
    }
    for (size_t j = 0; j < nKeys; j++)
    {
      const char* sValue = IniFile_GetKeyValueString(ini, sectionNames[i], keyNames[j]);
      printf("%s = %s\n", keyNames[j], sValue);
    }
    free((void*)keyNames);
  }
 
  const char* sValue = IniFile_GetKeyValueString(ini, "foobar", "prefix");
 
  if (strcmp(sValue, "someprefix") != 0)
  {
    printf("IniFile_GetKeyValueString failure\n");
    goto fail;
  }
 
  rc = TRUE;
fail:
  free((void*)sectionNames);
  IniFile_Free(ini);
  return rc;
}
 
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);
 
  if (!test04())
    return -1;
 
  /* 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;
}