Annotation of embedaddon/curl/docs/examples/xmlstream.c, revision 1.1.1.1

1.1       misho       1: /***************************************************************************
                      2:  *                                  _   _ ____  _
                      3:  *  Project                     ___| | | |  _ \| |
                      4:  *                             / __| | | | |_) | |
                      5:  *                            | (__| |_| |  _ <| |___
                      6:  *                             \___|\___/|_| \_\_____|
                      7:  *
                      8:  * Copyright (C) 1998 - 2018, Daniel Stenberg, <daniel@haxx.se>, et al.
                      9:  *
                     10:  * This software is licensed as described in the file COPYING, which
                     11:  * you should have received as part of this distribution. The terms
                     12:  * are also available at https://curl.haxx.se/docs/copyright.html.
                     13:  *
                     14:  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
                     15:  * copies of the Software, and permit persons to whom the Software is
                     16:  * furnished to do so, under the terms of the COPYING file.
                     17:  *
                     18:  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
                     19:  * KIND, either express or implied.
                     20:  *
                     21:  ***************************************************************************/
                     22: /* <DESC>
                     23:  * Stream-parse a document using the streaming Expat parser.
                     24:  * </DESC>
                     25:  */
                     26: /* Written by David Strauss
                     27:  *
                     28:  * Expat => https://libexpat.github.io/
                     29:  *
                     30:  * gcc -Wall -I/usr/local/include xmlstream.c -lcurl -lexpat -o xmlstream
                     31:  *
                     32:  */
                     33: 
                     34: #include <stdio.h>
                     35: #include <stdlib.h>
                     36: #include <string.h>
                     37: #include <assert.h>
                     38: 
                     39: #include <expat.h>
                     40: #include <curl/curl.h>
                     41: 
                     42: struct MemoryStruct {
                     43:   char *memory;
                     44:   size_t size;
                     45: };
                     46: 
                     47: struct ParserStruct {
                     48:   int ok;
                     49:   size_t tags;
                     50:   size_t depth;
                     51:   struct MemoryStruct characters;
                     52: };
                     53: 
                     54: static void startElement(void *userData, const XML_Char *name,
                     55:                          const XML_Char **atts)
                     56: {
                     57:   struct ParserStruct *state = (struct ParserStruct *) userData;
                     58:   state->tags++;
                     59:   state->depth++;
                     60: 
                     61:   /* Get a clean slate for reading in character data. */
                     62:   free(state->characters.memory);
                     63:   state->characters.memory = NULL;
                     64:   state->characters.size = 0;
                     65: }
                     66: 
                     67: static void characterDataHandler(void *userData, const XML_Char *s, int len)
                     68: {
                     69:   struct ParserStruct *state = (struct ParserStruct *) userData;
                     70:   struct MemoryStruct *mem = &state->characters;
                     71: 
                     72:   char *ptr = realloc(mem->memory, mem->size + len + 1);
                     73:   if(!ptr) {
                     74:     /* Out of memory. */
                     75:     fprintf(stderr, "Not enough memory (realloc returned NULL).\n");
                     76:     state->ok = 0;
                     77:     return;
                     78:   }
                     79: 
                     80:   mem->memory = ptr;
                     81:   memcpy(&(mem->memory[mem->size]), s, len);
                     82:   mem->size += len;
                     83:   mem->memory[mem->size] = 0;
                     84: }
                     85: 
                     86: static void endElement(void *userData, const XML_Char *name)
                     87: {
                     88:   struct ParserStruct *state = (struct ParserStruct *) userData;
                     89:   state->depth--;
                     90: 
                     91:   printf("%5lu   %10lu   %s\n", state->depth, state->characters.size, name);
                     92: }
                     93: 
                     94: static size_t parseStreamCallback(void *contents, size_t length, size_t nmemb,
                     95:                                   void *userp)
                     96: {
                     97:   XML_Parser parser = (XML_Parser) userp;
                     98:   size_t real_size = length * nmemb;
                     99:   struct ParserStruct *state = (struct ParserStruct *) XML_GetUserData(parser);
                    100: 
                    101:   /* Only parse if we're not already in a failure state. */
                    102:   if(state->ok && XML_Parse(parser, contents, real_size, 0) == 0) {
                    103:     int error_code = XML_GetErrorCode(parser);
                    104:     fprintf(stderr, "Parsing response buffer of length %lu failed"
                    105:             " with error code %d (%s).\n",
                    106:             real_size, error_code, XML_ErrorString(error_code));
                    107:     state->ok = 0;
                    108:   }
                    109: 
                    110:   return real_size;
                    111: }
                    112: 
                    113: int main(void)
                    114: {
                    115:   CURL *curl_handle;
                    116:   CURLcode res;
                    117:   XML_Parser parser;
                    118:   struct ParserStruct state;
                    119: 
                    120:   /* Initialize the state structure for parsing. */
                    121:   memset(&state, 0, sizeof(struct ParserStruct));
                    122:   state.ok = 1;
                    123: 
                    124:   /* Initialize a namespace-aware parser. */
                    125:   parser = XML_ParserCreateNS(NULL, '\0');
                    126:   XML_SetUserData(parser, &state);
                    127:   XML_SetElementHandler(parser, startElement, endElement);
                    128:   XML_SetCharacterDataHandler(parser, characterDataHandler);
                    129: 
                    130:   /* Initialize a libcurl handle. */
                    131:   curl_global_init(CURL_GLOBAL_DEFAULT);
                    132:   curl_handle = curl_easy_init();
                    133:   curl_easy_setopt(curl_handle, CURLOPT_URL,
                    134:                    "https://www.w3schools.com/xml/simple.xml");
                    135:   curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, parseStreamCallback);
                    136:   curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *)parser);
                    137: 
                    138:   printf("Depth   Characters   Closing Tag\n");
                    139: 
                    140:   /* Perform the request and any follow-up parsing. */
                    141:   res = curl_easy_perform(curl_handle);
                    142:   if(res != CURLE_OK) {
                    143:     fprintf(stderr, "curl_easy_perform() failed: %s\n",
                    144:             curl_easy_strerror(res));
                    145:   }
                    146:   else if(state.ok) {
                    147:     /* Expat requires one final call to finalize parsing. */
                    148:     if(XML_Parse(parser, NULL, 0, 1) == 0) {
                    149:       int error_code = XML_GetErrorCode(parser);
                    150:       fprintf(stderr, "Finalizing parsing failed with error code %d (%s).\n",
                    151:               error_code, XML_ErrorString(error_code));
                    152:     }
                    153:     else {
                    154:       printf("                     --------------\n");
                    155:       printf("                     %lu tags total\n", state.tags);
                    156:     }
                    157:   }
                    158: 
                    159:   /* Clean up. */
                    160:   free(state.characters.memory);
                    161:   XML_ParserFree(parser);
                    162:   curl_easy_cleanup(curl_handle);
                    163:   curl_global_cleanup();
                    164: 
                    165:   return 0;
                    166: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>