curl/tests/server/getpart.c

190 lines
4.6 KiB
C
Raw Normal View History

2004-02-20 15:19:18 +08:00
/***************************************************************************
* _ _ ____ _
* Project ___| | | | _ \| |
* / __| | | | |_) | |
* | (__| |_| | _ <| |___
* \___|\___/|_| \_\_____|
*
* Copyright (C) 1998 - 2004, Daniel Stenberg, <daniel@haxx.se>, et al.
*
* This software is licensed as described in the file COPYING, which
* you should have received as part of this distribution. The terms
* are also available at http://curl.haxx.se/docs/copyright.html.
*
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
* copies of the Software, and permit persons to whom the Software is
* furnished to do so, under the terms of the COPYING file.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
* $Id$
***************************************************************************/
#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <stdlib.h>
2004-02-20 15:19:18 +08:00
#include "getpart.h"
2002-05-17 16:15:33 +08:00
#define EAT_SPACE(ptr) while( ptr && *ptr && isspace((int)*ptr) ) ptr++
#define EAT_WORD(ptr) while( ptr && *ptr && !isspace((int)*ptr) && ('>' != *ptr)) ptr++
#ifdef DEBUG
#define show(x) printf x
#else
#define show(x)
#endif
static
2002-02-19 09:04:46 +08:00
char *appendstring(char *string, /* original string */
char *buffer, /* to append */
2004-03-10 16:12:09 +08:00
size_t *stringlen, /* length of string */
size_t *stralloc) /* allocated size */
2002-02-19 09:04:46 +08:00
{
2004-03-10 16:12:09 +08:00
size_t len = strlen(buffer);
size_t needed_len = len + *stringlen;
2002-02-19 09:04:46 +08:00
if(needed_len >= *stralloc) {
char *newptr;
2004-03-10 16:12:09 +08:00
size_t newsize = needed_len*2; /* get twice the needed size */
newptr = realloc(string, newsize);
2002-02-19 09:04:46 +08:00
if(newptr) {
string = newptr;
*stralloc = newsize;
2002-02-19 09:04:46 +08:00
}
else
return NULL;
}
strcpy(&string[*stringlen], buffer);
*stringlen += len;
return string;
}
2004-02-20 15:19:18 +08:00
const char *spitout(FILE *stream,
const char *main,
2004-03-23 16:50:28 +08:00
const char *sub, size_t *size)
{
char buffer[8192]; /* big enough for anything */
char cmain[128]=""; /* current main section */
char csub[128]=""; /* current sub section */
char *ptr;
char *end;
char display = 0;
char *string;
2004-03-10 16:12:09 +08:00
size_t stringlen=0;
size_t stralloc=256;
enum {
STATE_OUTSIDE,
STATE_INMAIN,
STATE_INSUB,
STATE_ILLEGAL
} state = STATE_OUTSIDE;
string = (char *)malloc(stralloc);
if(!string)
return NULL;
string[0] = 0; /* zero first byte in case of no data */
while(fgets(buffer, sizeof(buffer), stream)) {
ptr = buffer;
/* pass white spaces */
EAT_SPACE(ptr);
if('<' != *ptr) {
if(display) {
show(("=> %s", buffer));
2002-02-19 09:04:46 +08:00
string = appendstring(string, buffer, &stringlen, &stralloc);
show(("* %s\n", buffer));
}
continue;
}
ptr++;
EAT_SPACE(ptr);
if('/' == *ptr) {
/* end of a section */
ptr++;
EAT_SPACE(ptr);
end = ptr;
EAT_WORD(end);
*end = 0;
if((state == STATE_INSUB) &&
!strcmp(csub, ptr)) {
/* this is the end of the currently read sub section */
state--;
csub[0]=0; /* no sub anymore */
2002-02-19 09:04:46 +08:00
display=0;
}
else if((state == STATE_INMAIN) &&
!strcmp(cmain, ptr)) {
/* this is the end of the currently read main section */
state--;
cmain[0]=0; /* no main anymore */
2002-02-19 09:04:46 +08:00
display=0;
}
}
2002-02-19 09:04:46 +08:00
else if(!display) {
/* this is the beginning of a section */
end = ptr;
EAT_WORD(end);
*end = 0;
switch(state) {
case STATE_OUTSIDE:
strcpy(cmain, ptr);
state = STATE_INMAIN;
break;
case STATE_INMAIN:
strcpy(csub, ptr);
state = STATE_INSUB;
break;
default:
break;
}
}
2002-02-19 09:04:46 +08:00
if(display) {
string = appendstring(string, buffer, &stringlen, &stralloc);
show(("* %s\n", buffer));
2002-02-19 09:04:46 +08:00
}
if((STATE_INSUB == state) &&
!strcmp(cmain, main) &&
!strcmp(csub, sub)) {
show(("* (%d bytes) %s\n", stringlen, buffer));
display = 1; /* start displaying */
}
else {
show(("%d (%s/%s): %s\n", state, cmain, csub, buffer));
display = 0; /* no display */
}
}
*size = stringlen;
return string;
}
#ifdef TEST
int main(int argc, char **argv)
{
if(argc< 3) {
printf("./moo main sub\n");
}
else {
int size;
char *buffer = spitout(stdin, argv[1], argv[2], &size);
}
return 0;
}
#endif