Make tree2 perform reference counting.

Fix bits of the treebuilder to perform reference counting correctly in the face of *result not pointing to the same object as the node passed in to the treebuilder client callbacks.

svn path=/trunk/hubbub/; revision=4666

1 files changed, 5 insertions, 5 deletions

diff --git a/src/treebuilder/before_html.c b/src/treebuilder/before_html.c
index ebfc92f..8401087 100644
--- a/src/treebuilder/before_html.c
+++ b/src/treebuilder/before_html.c
@@ -104,21 +104,21 @@ bool handle_before_html(hubbub_treebuilder *treebuilder,
 					html);
 		}
 
+		treebuilder->tree_handler->unref_node(
+				treebuilder->tree_handler->ctx,
+				html);
+
 		/* We can't use element_stack_push() here, as it 
 		 * assumes that current_node is pointing at the index 
 		 * before the one to insert at. For the first entry in 
 		 * the stack, this does not hold so we must insert
 		 * manually. */
 		treebuilder->context.element_stack[0].type = HTML;
-		treebuilder->context.element_stack[0].node = html;
+		treebuilder->context.element_stack[0].node = appended;
 		treebuilder->context.current_node = 0;
 
 		/** \todo cache selection algorithm */
 
-		treebuilder->tree_handler->unref_node(
-				treebuilder->tree_handler->ctx,
-				appended);
-
 		treebuilder->context.mode = BEFORE_HEAD;
 	}