Better handling of remote stream close

src/XMPPConnection.m

@@ -269,12 +269,11 @@
 		size_t length = [sock readNBytes: 512
 				      intoBuffer: buffer];
 
-		if (length < 1 && [delegate respondsToSelector:
+		[self parseBuffer: buffer
-		    @selector(connectionWasClosed:)])
+		       withLength: length];
-			[delegate connectionWasClosed: self];
 
-		[parser parseBuffer: buffer
+		if (length < 1)
-			 withLength: length];
+			return;
 	}
 }
 
@@ -282,8 +281,10 @@
 	 withLength: (size_t)length
 {
 	if (length < 1 && [delegate respondsToSelector:
-	    @selector(connectionWasClosed:)])
+	    @selector(connectionWasClosed:)]) {
 		[delegate connectionWasClosed: self];
+		return;
+	}
 
 	[parser parseBuffer: buffer
 		 withLength: length];
@@ -454,6 +455,19 @@ withCallbackBlock: (xmpp_callback_block)callback;
 		[self XMPP_handleSASL: element];
 }
 
+- (void)elementBuilder: (OFXMLElementBuilder *)builder
+  didNotExpectCloseTag: (OFString *)name
+	    withPrefix: (OFString *)prefix
+	     namespace: (OFString *)ns
+{
+	if (![name isEqual: @"stream"] || ![prefix isEqual: @"stream"] ||
+	    ![ns isEqual: XMPP_NS_STREAM]) {
+		@throw [OFMalformedXMLException
+		    exceptionWithClass: [builder class]
+				parser: nil];
+	}
+}
+
 - (void)XMPP_startStream
 {
 	/* Make sure we don't get any old events */