3 files changed, 14 insertions, 0 deletions
diff --git a/Swiften/Queries/GenericRequest.h b/Swiften/Queries/GenericRequest.h
index be6a88d..72bbcbd 100644
--- a/Swiften/Queries/GenericRequest.h
+++ b/Swiften/Queries/GenericRequest.h
@@ -26,6 +26,11 @@ namespace Swift {
 				onResponse(boost::dynamic_pointer_cast<PAYLOAD_TYPE>(payload), error);
 			}
 
+		protected:
+			boost::shared_ptr<PAYLOAD_TYPE> getPayloadGeneric() const {
+				return boost::dynamic_pointer_cast<PAYLOAD_TYPE>(getPayload());
+			}
+
 		public:
 			boost::signal<void (boost::shared_ptr<PAYLOAD_TYPE>, const boost::optional<ErrorPayload>&)> onResponse;
 	};
diff --git a/Swiften/Queries/Request.h b/Swiften/Queries/Request.h
index d0fd8c6..450e311 100644
--- a/Swiften/Queries/Request.h
+++ b/Swiften/Queries/Request.h
@@ -38,6 +38,10 @@ namespace Swift {
 				payload_ = p;
 			}
 
+			boost::shared_ptr<Payload> getPayload() const {
+				return payload_;
+			}
+
 			virtual void handleResponse(boost::shared_ptr<Payload>, boost::optional<ErrorPayload>) = 0;
 
 		private:
diff --git a/Swiften/Queries/Requests/GetDiscoInfoRequest.h b/Swiften/Queries/Requests/GetDiscoInfoRequest.h
index fd4d9f6..9ec1050 100644
--- a/Swiften/Queries/Requests/GetDiscoInfoRequest.h
+++ b/Swiften/Queries/Requests/GetDiscoInfoRequest.h
@@ -15,5 +15,10 @@ namespace Swift {
 			GetDiscoInfoRequest(const JID& jid, IQRouter* router) :
 					GenericRequest<DiscoInfo>(IQ::Get, jid, boost::shared_ptr<DiscoInfo>(new DiscoInfo()), router) {
 			}
+
+			GetDiscoInfoRequest(const JID& jid, const String& node, IQRouter* router) :
+					GenericRequest<DiscoInfo>(IQ::Get, jid, boost::shared_ptr<DiscoInfo>(new DiscoInfo()), router) {
+				getPayloadGeneric()->setNode(node);
+			}
 	};
 }