1 files changed, 14 insertions, 1 deletions
diff --git a/Swiften/Elements/WhiteboardPayload.h b/Swiften/Elements/WhiteboardPayload.h
index 8d40d9d..db261ca 100644
--- a/Swiften/Elements/WhiteboardPayload.h
+++ b/Swiften/Elements/WhiteboardPayload.h
@@ -13,7 +13,11 @@
 namespace Swift {
 	class WhiteboardPayload : public Payload {
 	public:
-		WhiteboardPayload() {
+		typedef boost::shared_ptr<WhiteboardPayload> ref;
+
+		enum Type {Data, SessionRequest};
+
+		WhiteboardPayload(Type type = WhiteboardPayload::Data) : type_(type) {
 		}
 
 		void setData(const std::string &data) {
@@ -24,7 +28,16 @@ namespace Swift {
 			return data_;
 		}
 
+		Type getType() const {
+			return type_;
+		}
+
+		void setType(Type type) {
+			type_ = type;
+		}
+
 	private:
 		std::string data_;
+		Type type_;
 	};
 }