From 68d83a8142e52643cc98d692dce54a49d9e2c386 Mon Sep 17 00:00:00 2001 From: Yang Tse Date: Sun, 14 Feb 2010 13:14:17 +0000 Subject: Overhauled test suite getpart() function. Fixing potential out of bounds stack and memory overwrites triggered with huge test case definitions. --- tests/server/testpart.c | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) (limited to 'tests/server/testpart.c') diff --git a/tests/server/testpart.c b/tests/server/testpart.c index d2be529de..a78799ca7 100644 --- a/tests/server/testpart.c +++ b/tests/server/testpart.c @@ -35,15 +35,20 @@ int main(int argc, char **argv) { + int rc; + char *part; + size_t partlen, i; + if(argc< 3) { printf("./testpart main sub\n"); } else { - size_t size; - unsigned int i; - const char *buffer = spitout(stdin, argv[1], argv[2], &size); - for(i=0; i< size; i++) - printf("%c", buffer[i]); + rc = getpart(&part, &partlen, argv[1], argv[2], stdin); + if(rc) + return(rc); + for(i = 0; i < partlen; i++) + printf("%c", part[i]); + free(part); } return 0; } -- cgit v1.2.3