Fix misc leaks in qparams code, support ; as param separator. Add test suite

This commit is contained in:
Daniel P. Berrange 2008-05-22 23:49:36 +00:00
parent 7646d59568
commit 9dc31e159a
5 changed files with 258 additions and 8 deletions

View File

@ -1,3 +1,10 @@
Thu May 22 19:47:29 EST 2008 Daniel P. Berrange <berrange@redhat.com>
* src/qparams.c: Support ; as a param separator. Misc memory
leaks
* tests/qparamtest.c, tests/Makefile.am: Add test suite for
qparams code
Thu May 22 19:44:29 EST 2008 Daniel P. Berrange <berrange@redhat.com> Thu May 22 19:44:29 EST 2008 Daniel P. Berrange <berrange@redhat.com>
* src/qemu_conf.c: Refactor qemudBuildCommandLine to use a * src/qemu_conf.c: Refactor qemudBuildCommandLine to use a

View File

@ -166,7 +166,7 @@ struct qparam_set *
qparam_query_parse (const char *query) qparam_query_parse (const char *query)
{ {
struct qparam_set *ps; struct qparam_set *ps;
const char *name, *value, *end, *eq; const char *end, *eq;
ps = new_qparam_set (0, NULL); ps = new_qparam_set (0, NULL);
if (!ps) return NULL; if (!ps) return NULL;
@ -174,9 +174,14 @@ qparam_query_parse (const char *query)
if (!query || query[0] == '\0') return ps; if (!query || query[0] == '\0') return ps;
while (*query) { while (*query) {
char *name = NULL, *value = NULL;
/* Find the next separator, or end of the string. */ /* Find the next separator, or end of the string. */
end = strchr (query, '&'); end = strchr (query, '&');
if (!end) end = query + strlen (query); if (!end)
end = strchr (query, ';');
if (!end)
end = query + strlen (query);
/* Find the first '=' character between here and end. */ /* Find the first '=' character between here and end. */
eq = strchr (query, '='); eq = strchr (query, '=');
@ -191,7 +196,6 @@ qparam_query_parse (const char *query)
*/ */
else if (!eq) { else if (!eq) {
name = xmlURIUnescapeString (query, end - query, NULL); name = xmlURIUnescapeString (query, end - query, NULL);
value = "";
if (!name) goto out_of_memory; if (!name) goto out_of_memory;
} }
/* Or if we have "name=" here (works around annoying /* Or if we have "name=" here (works around annoying
@ -199,7 +203,6 @@ qparam_query_parse (const char *query)
*/ */
else if (eq+1 == end) { else if (eq+1 == end) {
name = xmlURIUnescapeString (query, eq - query, NULL); name = xmlURIUnescapeString (query, eq - query, NULL);
value = "";
if (!name) goto out_of_memory; if (!name) goto out_of_memory;
} }
/* If the '=' character is at the beginning then we have /* If the '=' character is at the beginning then we have
@ -211,12 +214,23 @@ qparam_query_parse (const char *query)
/* Otherwise it's "name=value". */ /* Otherwise it's "name=value". */
else { else {
name = xmlURIUnescapeString (query, eq - query, NULL); name = xmlURIUnescapeString (query, eq - query, NULL);
if (!name)
goto out_of_memory;
value = xmlURIUnescapeString (eq+1, end - (eq+1), NULL); value = xmlURIUnescapeString (eq+1, end - (eq+1), NULL);
if (!name || !value) goto out_of_memory; if (!value) {
VIR_FREE(name);
goto out_of_memory;
}
} }
/* Append to the parameter set. */ /* Append to the parameter set. */
if (append_qparam (ps, name, value) == -1) goto out_of_memory; if (append_qparam (ps, name, value ? value : "") == -1) {
VIR_FREE(name);
VIR_FREE(value);
goto out_of_memory;
}
VIR_FREE(name);
VIR_FREE(value);
next: next:
query = end; query = end;

View File

@ -14,6 +14,7 @@ qemuxml2xmltest
qemuxml2argvtest qemuxml2argvtest
nodeinfotest nodeinfotest
statstest statstest
qparamtest
*.gcda *.gcda
*.gcno *.gcno

View File

@ -41,7 +41,7 @@ EXTRA_DIST = \
noinst_PROGRAMS = xmlrpctest xml2sexprtest sexpr2xmltest virshtest conftest \ noinst_PROGRAMS = xmlrpctest xml2sexprtest sexpr2xmltest virshtest conftest \
reconnect xmconfigtest xencapstest qemuxml2argvtest qemuxml2xmltest \ reconnect xmconfigtest xencapstest qemuxml2argvtest qemuxml2xmltest \
nodeinfotest statstest nodeinfotest statstest qparamtest
test_scripts = \ test_scripts = \
daemon-conf \ daemon-conf \
@ -53,7 +53,7 @@ EXTRA_DIST += $(test_scripts)
TESTS = xml2sexprtest sexpr2xmltest virshtest test_conf.sh xmconfigtest \ TESTS = xml2sexprtest sexpr2xmltest virshtest test_conf.sh xmconfigtest \
xencapstest qemuxml2argvtest qemuxml2xmltest nodeinfotest \ xencapstest qemuxml2argvtest qemuxml2xmltest nodeinfotest \
statstest $(test_scripts) statstest qparamtest $(test_scripts)
if ENABLE_XEN_TESTS if ENABLE_XEN_TESTS
TESTS += reconnect TESTS += reconnect
endif endif
@ -130,6 +130,10 @@ statstest_SOURCES = \
statstest.c testutils.h testutils.c statstest.c testutils.h testutils.c
statstest_LDADD = $(LDADDS) statstest_LDADD = $(LDADDS)
qparamtest_SOURCES = \
qparamtest.c testutils.h testutils.c
qparamtest_LDADD = $(LDADDS)
reconnect_SOURCES = \ reconnect_SOURCES = \
reconnect.c reconnect.c
reconnect_LDADD = $(LDADDS) reconnect_LDADD = $(LDADDS)

224
tests/qparamtest.c Normal file
View File

@ -0,0 +1,224 @@
#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "testutils.h"
#include "qparams.h"
#include "util.h"
struct qparamParseDataEntry {
const char *name;
const char *value;
};
struct qparamParseData {
const char *queryIn;
const char *queryOut;
int nparams;
const struct qparamParseDataEntry *params;
};
static int
qparamParseTest(const void *data)
{
const struct qparamParseData *expect = data;
struct qparam_set *actual = qparam_query_parse(expect->queryIn);
int ret = -1, i;
if (!actual)
return -1;
if (actual->n != expect->nparams)
goto fail;
for (i = 0 ; i < actual->n ; i++) {
if (!STREQ(expect->params[i].name,
actual->p[i].name))
goto fail;
if (!STREQ(expect->params[i].value,
actual->p[i].value))
goto fail;
}
ret = 0;
fail:
free_qparam_set(actual);
return ret;
}
static int
qparamFormatTest(const void *data)
{
const struct qparamParseData *expect = data;
struct qparam_set *actual = qparam_query_parse(expect->queryIn);
char *output = NULL;
int ret = -1;
if (!actual)
return -1;
output = qparam_get_query(actual);
if (!output)
goto fail;
if (!STREQ(output, expect->queryOut))
goto fail;
ret = 0;
fail:
free(output);
free_qparam_set(actual);
return ret;
}
static int
qparamBuildTest(const void *data)
{
const struct qparamParseData *expect = data;
struct qparam_set *actual = new_qparam_set(0, NULL);
int ret = -1, i;
if (!actual)
return -1;
for (i = 0 ; i < expect->nparams ; i++) {
if (append_qparam(actual,
expect->params[i].name,
expect->params[i].value) < 0)
goto fail;
}
if (actual->n != expect->nparams)
goto fail;
for (i = 0 ; i < actual->n ; i++) {
if (!STREQ(expect->params[i].name,
actual->p[i].name))
goto fail;
if (!STREQ(expect->params[i].value,
actual->p[i].value))
goto fail;
}
ret = 0;
fail:
free_qparam_set(actual);
return ret;
}
static int
qparamTestNewVargs(const void *data ATTRIBUTE_UNUSED)
{
struct qparam_set *actual = new_qparam_set(0, "foo", "one", "bar", "two", NULL);
int ret = -1;
if (!actual)
return -1;
if (actual->n != 2)
goto fail;
if (!STREQ(actual->p[0].name, "foo"))
goto fail;
if (!STREQ(actual->p[0].value, "one"))
goto fail;
if (!STREQ(actual->p[1].name, "bar"))
goto fail;
if (!STREQ(actual->p[1].value, "two"))
goto fail;
ret = 0;
fail:
free_qparam_set(actual);
return ret;
}
static int
qparamTestAddVargs(const void *data ATTRIBUTE_UNUSED)
{
struct qparam_set *actual = new_qparam_set(0, NULL);
int ret = -1;
if (!actual)
return -1;
if (append_qparams(actual, "foo", "one", "bar", "two", NULL) < 0)
goto fail;
if (actual->n != 2)
goto fail;
if (!STREQ(actual->p[0].name, "foo"))
goto fail;
if (!STREQ(actual->p[0].value, "one"))
goto fail;
if (!STREQ(actual->p[1].name, "bar"))
goto fail;
if (!STREQ(actual->p[1].value, "two"))
goto fail;
ret = 0;
fail:
free_qparam_set(actual);
return ret;
}
static const struct qparamParseDataEntry const params1[] = { { "foo", "one" }, { "bar", "two" } };
static const struct qparamParseDataEntry const params2[] = { { "foo", "one" }, { "foo", "two" } };
static const struct qparamParseDataEntry const params3[] = { { "foo", "&one" }, { "bar", "&two" } };
static const struct qparamParseDataEntry const params4[] = { { "foo", "" } };
static const struct qparamParseDataEntry const params5[] = { { "foo", "one two" } };
static const struct qparamParseDataEntry const params6[] = { { "foo", "one" } };
int
main(void)
{
int ret = 0;
#define DO_TEST(queryIn,queryOut,params) \
do { \
struct qparamParseData info = { \
queryIn, \
queryOut ? queryOut : queryIn, \
sizeof(params)/sizeof(params[0]), \
params }; \
if (virtTestRun("Parse " queryIn, \
1, qparamParseTest, &info) < 0) \
ret = -1; \
if (virtTestRun("Format " queryIn, \
1, qparamFormatTest, &info) < 0) \
ret = -1; \
if (virtTestRun("Build " queryIn, \
1, qparamBuildTest, &info) < 0) \
ret = -1; \
} while (0)
DO_TEST("foo=one&bar=two", NULL, params1);
DO_TEST("foo=one&foo=two", NULL, params2);
DO_TEST("foo=one&&foo=two", "foo=one&foo=two", params2);
DO_TEST("foo=one;foo=two", "foo=one&foo=two", params2);
DO_TEST("foo", "foo=", params4);
DO_TEST("foo=", NULL, params4);
DO_TEST("foo=&", "foo=", params4);
DO_TEST("foo=&&", "foo=", params4);
DO_TEST("foo=one%20two", NULL, params5);
DO_TEST("=bogus&foo=one", "foo=one", params6);
if (virtTestRun("New vargs", 1, qparamTestNewVargs, NULL) < 0)
ret = -1;
if (virtTestRun("Add vargs", 1, qparamTestAddVargs, NULL) < 0)
ret = -1;
exit(ret==0 ? EXIT_SUCCESS : EXIT_FAILURE);
}