Swiftdiff options
| author | Remko Tronçon <git@el-tramo.be> | 2010-11-01 20:18:36 (GMT) |
|---|---|---|
| committer | Remko Tronçon <git@el-tramo.be> | 2010-11-03 12:17:42 (GMT) |
| commit | 63a8085f02218d2dd2b51acf29280624d6317610 (patch) | |
| tree | fe0019cf3d9cdc15f27732062a326dcdc7c2b11c /Swiften/FileTransfer/IBBReceiveSession.cpp | |
| parent | f03b5eee1d92f686f4f2ccc8a0b10c18bac05e72 (diff) | |
| download | swift-63a8085f02218d2dd2b51acf29280624d6317610.zip swift-63a8085f02218d2dd2b51acf29280624d6317610.tar.bz2 | |
Added incoming IBB session.
Diffstat (limited to 'Swiften/FileTransfer/IBBReceiveSession.cpp')
| -rw-r--r-- | Swiften/FileTransfer/IBBReceiveSession.cpp | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/Swiften/FileTransfer/IBBReceiveSession.cpp b/Swiften/FileTransfer/IBBReceiveSession.cpp new file mode 100644 index 0000000..9eed21d --- /dev/null +++ b/Swiften/FileTransfer/IBBReceiveSession.cpp @@ -0,0 +1,69 @@ +/* + * Copyright (c) 2010 Remko Tronçon + * Licensed under the GNU General Public License v3. + * See Documentation/Licenses/GPLv3.txt for more information. + */ + +#include "Swiften/FileTransfer/IBBReceiveSession.h" + +#include <boost/bind.hpp> + +#include "Swiften/Queries/IQRouter.h" +#include "Swiften/FileTransfer/IBBRequest.h" +#include "Swiften/FileTransfer/BytestreamException.h" + +namespace Swift { + +IBBReceiveSession::IBBReceiveSession(const String& id, const JID& from, size_t size, WriteBytestream::ref bytestream, IQRouter* router) : SetResponder<IBB>(router), id(id), from(from), size(size), bytestream(bytestream), router(router), sequenceNumber(0), active(false), receivedSize(0) { +} + +IBBReceiveSession::~IBBReceiveSession() { +} + +void IBBReceiveSession::start() { + active = true; +} + +void IBBReceiveSession::stop() { + if (active && router->isAvailable()) { + IBBRequest::create(from, IBB::createIBBClose(id), router)->send(); + } + finish(boost::optional<FileTransferError>()); +} + +void IBBReceiveSession::finish(boost::optional<FileTransferError> error) { + active = false; + onFinished(error); +} + +bool IBBReceiveSession::handleSetRequest(const JID& from, const JID&, const String& id, IBB::ref ibb) { + if (from == this->from && ibb->getStreamID() == id) { + if (ibb->getAction() == IBB::Data) { + if (sequenceNumber == ibb->getSequenceNumber()) { + bytestream->write(ibb->getData()); + receivedSize += ibb->getData().getSize(); + if (receivedSize >= size) { + if (receivedSize > size) { + std::cerr << "Warning: Received more data than expected" << std::endl; + } + finish(boost::optional<FileTransferError>()); + } + } + else { + sendError(from, id, ErrorPayload::NotAcceptable, ErrorPayload::Cancel); + finish(FileTransferError(FileTransferError::ClosedError)); + } + } + else if (ibb->getAction() == IBB::Open) { + sendResponse(from, id, IBB::ref()); + } + else if (ibb->getAction() == IBB::Close) { + sendResponse(from, id, IBB::ref()); + finish(FileTransferError(FileTransferError::ClosedError)); + } + return true; + } + return false; +} + +} |