Swiftdiff options
Diffstat (limited to 'Swiften/StreamStack')
| -rw-r--r-- | Swiften/StreamStack/ConnectionLayer.h | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/Swiften/StreamStack/ConnectionLayer.h b/Swiften/StreamStack/ConnectionLayer.h index 99873a0..7688f78 100644 --- a/Swiften/StreamStack/ConnectionLayer.h +++ b/Swiften/StreamStack/ConnectionLayer.h @@ -1,7 +1,7 @@ -#ifndef SWIFTEN_CONNECTIONLAYER_H -#define SWIFTEN_CONNECTIONLAYER_H +#pragma once #include <boost/signal.hpp> +#include <boost/shared_ptr.hpp> #include "Swiften/StreamStack/LowLayer.h" #include "Swiften/Network/Connection.h" @@ -9,17 +9,15 @@ namespace Swift { class ConnectionLayer : public LowLayer { public: - ConnectionLayer(Connection* connection) : connection_(connection) { - connection_->onDataRead.connect(onDataRead); + ConnectionLayer(boost::shared_ptr<Connection> connection) : connection(connection) { + connection->onDataRead.connect(onDataRead); } void writeData(const ByteArray& data) { - connection_->write(data); + connection->write(data); } private: - Connection* connection_; + boost::shared_ptr<Connection> connection; }; } - -#endif |