/* * This file is part of LibDOM. * Licensed under the MIT License, * http://www.opensource.org/licenses/mit-license.php * * Copyright 2010 Michael Drake */ /* * Load an HTML file into LibDOM with Hubbub and print out the DOM structure. * * This example demonstrates the following: * * 1. Using LibDOM's Hubbub binding to read an HTML file into LibDOM. * 2. Walking around the DOM tree. * 3. Accessing DOM node attributes. * * Example input: *

NetSurf

NetSurf is awesome!

Hubbub

Hubbub is too.

Big time.

* * Example output: * * HTML * +-BODY * | +-H1 class="woo" * | +-P * | | +-EM * | +-DIV * | | +-H2 * | | +-P * | | +-P * */ #define _GNU_SOURCE /* for strndup */ #include #include #include #include #include #include #include #include #define UNUSED(x) ((x)=(x)) void *test_realloc(void *ptr, size_t len, void *pw) { UNUSED(pw); return realloc(ptr, len); } void test_msg(uint32_t severity, void *ctx, const char *msg, ...) { va_list l; UNUSED(ctx); va_start(l, msg); fprintf(stderr, "%d: ", severity); vfprintf(stderr, msg, l); fprintf(stderr, "\n"); } /** * Generate a LibDOM document DOM from an HTML file * * \param file The file path * \return pointer to DOM document, or NULL on error */ dom_document *create_doc_dom_from_file(char *file) { const unsigned int buffer_size = 1024; dom_hubbub_parser *parser = NULL; int handle; int chunk_length; dom_hubbub_error error; dom_document *doc; char buffer[buffer_size]; /* Create Hubbub parser */ parser = dom_hubbub_parser_create(NULL, true, test_realloc, NULL, test_msg, NULL); if (parser == NULL) { printf("Can't create Hubbub Parser\n"); return NULL; } /* Open input file */ handle = open(file, O_RDONLY); if (handle == -1) { dom_hubbub_parser_destroy(parser); printf("Can't open test input file: %s\n", file); return NULL; } /* Parse input file in chunks */ chunk_length = buffer_size; while(chunk_length == buffer_size) { chunk_length = read(handle, buffer, buffer_size); error = dom_hubbub_parser_parse_chunk(parser, buffer, chunk_length); if (error != DOM_HUBBUB_OK) { dom_hubbub_parser_destroy(parser); printf("Parsing errors occur\n"); return NULL; } } /* Done parsing file */ error = dom_hubbub_parser_completed(parser); if (error != DOM_HUBBUB_OK) { dom_hubbub_parser_destroy(parser); printf("Parsing error when construct DOM\n"); return NULL; } /* Get the document */ doc = dom_hubbub_parser_get_document(parser); /* Finished with parser */ dom_hubbub_parser_destroy(parser); /* Close input file */ if (close(handle) == -1) { printf("Can't close test input file: %s\n", file); return NULL; } return doc; } /** * Dump class attribute/value for an element node * * \param node The element node to dump class for * \return true on success, or false on error */ bool dump_dom_element_class(dom_node_internal *node) { dom_exception exc; lwc_error err; dom_string *class = NULL; dom_string *classvalue = NULL; lwc_string *lwcstr = NULL; dom_node_type type; int i; const char *string; size_t length; /* Should only have element nodes here */ exc = dom_node_get_node_type(node, &type); if (exc != DOM_NO_ERR) { printf("Exception raised for node_get_node_type\n"); return false; } assert(type == DOM_ELEMENT_NODE); /* Create a dom_string constaining "class". */ exc = dom_string_create(test_realloc, NULL, "class", 5, &class); if (exc != DOM_NO_ERR) { printf("Exception raised for dom_string_create\n"); return false; } /* Get class attribute's value */ exc = dom_element_get_attribute(node, class, &classvalue); if (exc != DOM_NO_ERR) { printf("Exception raised for element_get_attribute\n"); return false; } else if (classvalue == NULL) { /* Element has no class attribute */ return true; } /* Get attributes's lwc_string */ exc = dom_string_get_intern(classvalue, &lwcstr); if (exc != DOM_NO_ERR) { printf("Exception raised for string_get_intern\n"); return false; } /* Get string data and print class info */ string = lwc_string_data(lwcstr); length = lwc_string_length(lwcstr); printf(" class=\"%*s\"", length, string); /* Print the element's class, if it has one */ dump_dom_element_class(node); return true; } /** * Print a line in a DOM structure dump for an element * * \param node The node to dump * \param depth The node's depth * \return true on success, or false on error */ bool dump_dom_element(dom_node_internal *node, int depth) { dom_exception exc; lwc_error err; dom_string *node_name = NULL; lwc_string *lwcstr = NULL; dom_node_type type; int i; const char *string; size_t length; /* Only interested in element nodes */ exc = dom_node_get_node_type(node, &type); if (exc != DOM_NO_ERR) { printf("Exception raised for node_get_node_type\n"); return false; } else if (type != DOM_ELEMENT_NODE) { /* Nothing to print */ return true; } /* Get element name */ exc = dom_node_get_node_name(node, &node_name); if (exc != DOM_NO_ERR) { printf("Exception raised for get_node_name\n"); return false; } else if (node_name == NULL) { printf("Broken: root_name == NULL\n"); return false; } /* Get element name's lwc_string */ exc = dom_string_get_intern(node_name, &lwcstr); if (exc != DOM_NO_ERR) { printf("Exception raised for string_get_intern\n"); return false; } /* Print ASCII tree structure for current node */ if (depth > 0) { for (i = 0; i < depth; i++) { printf("| "); } printf("+-"); } /* Get string data and print element name */ string = lwc_string_data(lwcstr); length = lwc_string_length(lwcstr); printf("%*s", length, string); /* PENDING FIX: Print the element's class, if it has one */ // if (dump_dom_element_class(node) == false) { // return false; // } printf("\n"); return true; } /** * Walk though a DOM (sub)tree, in depth first order, printing DOM structure. * * \param node The root node to start from * \param depth The depth of 'node' in the (sub)tree */ bool dump_dom_structure(dom_node_internal *node, int depth) { dom_exception exc; dom_node_internal *child; /* Print this node's entry */ if (dump_dom_element(node, depth) == false) { /* There was an error; return */ return false; } /* Get the node's first child */ exc = dom_node_get_first_child(node, &child); if (exc != DOM_NO_ERR) { printf("Exception raised for node_get_first_child\n"); return false; } else if (child != NULL) { /* node has children; decend to children's depth */ depth++; /* Loop though all node's children */ do { /* Visit node's descendents */ dump_dom_structure(child, depth); /* Go to next sibling */ exc = dom_node_get_next_sibling(child, &child); if (exc != DOM_NO_ERR) { printf("Exception raised for " "node_get_next_sibling\n"); return false; } } while (child != NULL); /* No more children */ } return true; } /** * Main entry point from OS. */ int main(int argc, char **argv) { dom_exception exc; /* returned by libdom functions */ lwc_error err; /* returned by libwapacplet functions */ dom_document *doc = NULL; /* document, loaded into libdom */ dom_node_internal *root = NULL; /* root element of document */ /* Initialise the DOM library */ exc = dom_initialise(test_realloc, NULL); if (exc != DOM_NO_ERR) { printf("Failed to initialise DOM library.\n"); return EXIT_FAILURE; } /* Load up the input HTML file */ doc = create_doc_dom_from_file("files/test.html"); if (doc == NULL) { printf("Failed to load document.\n"); return EXIT_FAILURE; } /* Get root element */ exc = dom_document_get_document_element(doc, &root); if (exc != DOM_NO_ERR) { printf("Exception raised for get_document_element\n"); return EXIT_FAILURE; } else if (root == NULL) { printf("Broken: root == NULL\n"); return EXIT_FAILURE; } /* Dump DOM structure */ if (dump_dom_structure(root, 0) == false) { return EXIT_FAILURE; } /* Finalise the DOM library */ exc = dom_finalise(); if (exc != DOM_NO_ERR) { printf("Failed to finalise DOM library.\n"); return EXIT_FAILURE; } return EXIT_SUCCESS; }