Do not bomb out on unusual xml

In particular, properly handle directy Type child of XRD element.

As usually, thanks to Joseph Smarr of Plaxo for spotting it ;-)

Signed-off-by: Michael Krelin <hacker@klever.net>

1 files changed, 15 insertions, 4 deletions

diff --git a/lib/discovery.cc b/lib/discovery.cc
index cbdceb3..f8a3988 100644
--- a/lib/discovery.cc
+++ b/lib/discovery.cc
@@ -451,31 +451,42 @@ namespace opkele {
 	    void end_element(const XML_Char *n) {
 		if(skipping<0) return;
 		if(skipping) {
 		    --skipping; return;
 		}
 		if(is_qelement(n,NSURI_XRD "\tType")) {
-		    assert(xrd); assert(xrd_service); assert(cdata==&cdata_buf);
+		    if(xrd && xrd_service) {
+			assert(cdata==&cdata_buf);
 		    xrd_service->types.insert(cdata_buf);
+		    }
 		}else if(is_qelement(n,NSURI_XRD "\tService")) {
-		    assert(xrd); assert(xrd_service);
+		    if(!(xrd && xrd_service)) {
+			skipping = -1;
+		    }else{
 		    assert(!pt_stack.empty());
 		    assert(pt_stack.back()==(NSURI_XRD "\tService"));
 		    pt_stack.pop_back();
 		    xrd_service = 0;
+		    }
 		}else if(is_qelement(n,NSURI_XRD "\tStatus")) {
-		    assert(xrd);
+		    if(!xrd) {
+			skipping=-1;
+		    }else{
 		    if(is_qelement(pt_stack.back().c_str(),n)) {
 			assert(cdata==&status_string);
 			pt_stack.pop_back();
 			if(status_code!=100)
 			    skipping = -1;
 		    }
+		    }
 		}else if(is_qelement(n,NSURI_XRD "\tExpires")) {
-		    assert(xrd);
+		    if(!xrd) {
+			skipping=-1;
+		    }else{
 		    xrd->expires = util::w3c_to_time(cdata_buf);
+		    }
 		}else if((xmode&xmode_html) && is_element(n,"head")) {
 		    skipping = -1;
 		}
 		cdata = 0;
 	    }
 	    void character_data(const XML_Char *s,int l) {