diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2010-01-05 16:01:46 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2010-01-05 16:01:46 +0000 |
| commit | 4db349c1f32b4a313d12d114137432441633d00a (patch) | |
| tree | 417451971e264834312cfd8339c79930d680a5d1 /src/DefaultExtensionMessageFactory.cc | |
| parent | 338e8aca09331d3e7fa4e1e8e1f9d10260986772 (diff) | |
2010-01-06 Tatsuhiro Tsujikawa <t-tujikawa@users.sourceforge.net>
Untabify. Fixed one line in copyright which is longer than 80
columns.
* src
* test
Diffstat (limited to 'src/DefaultExtensionMessageFactory.cc')
| -rw-r--r-- | src/DefaultExtensionMessageFactory.cc | 84 |
1 files changed, 42 insertions, 42 deletions
diff --git a/src/DefaultExtensionMessageFactory.cc b/src/DefaultExtensionMessageFactory.cc index 6dae3cd4..d9f87aef 100644 --- a/src/DefaultExtensionMessageFactory.cc +++ b/src/DefaultExtensionMessageFactory.cc @@ -16,7 +16,7 @@ * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA * * In addition, as a special exception, the copyright holders give * permission to link the code of portions of this program with the @@ -82,77 +82,77 @@ DefaultExtensionMessageFactory::createMessage(const unsigned char* data, size_t std::string extensionName = _registry->getExtensionName(extensionMessageID); if(extensionName.empty()) { throw DL_ABORT_EX - (StringFormat("No extension registered for extended message ID %u", - extensionMessageID).str()); + (StringFormat("No extension registered for extended message ID %u", + extensionMessageID).str()); } if(extensionName == "ut_pex") { // uTorrent compatible Peer-Exchange UTPexExtensionMessageHandle m = - UTPexExtensionMessage::create(data, length); + UTPexExtensionMessage::create(data, length); m->setPeerStorage(_peerStorage); return m; } else if(extensionName == "ut_metadata") { if(length == 0) { - throw DL_ABORT_EX(StringFormat(MSG_TOO_SMALL_PAYLOAD_SIZE, - "ut_metadata", length).str()); + throw DL_ABORT_EX(StringFormat(MSG_TOO_SMALL_PAYLOAD_SIZE, + "ut_metadata", length).str()); } size_t end; BDE dict = bencode::decode(data+1, length-1, end); if(!dict.isDict()) { - throw DL_ABORT_EX("Bad ut_metadata: dictionary not found"); + throw DL_ABORT_EX("Bad ut_metadata: dictionary not found"); } const BDE& msgType = dict["msg_type"]; if(!msgType.isInteger()) { - throw DL_ABORT_EX("Bad ut_metadata: msg_type not found"); + throw DL_ABORT_EX("Bad ut_metadata: msg_type not found"); } const BDE& index = dict["piece"]; if(!index.isInteger()) { - throw DL_ABORT_EX("Bad ut_metadata: piece not found"); + throw DL_ABORT_EX("Bad ut_metadata: piece not found"); } switch(msgType.i()) { case 0: { - SharedHandle<UTMetadataRequestExtensionMessage> m - (new UTMetadataRequestExtensionMessage(extensionMessageID)); - m->setIndex(index.i()); - m->setDownloadContext(_dctx); - m->setPeer(_peer); - m->setBtMessageFactory(_messageFactory); - m->setBtMessageDispatcher(_dispatcher); - return m; + SharedHandle<UTMetadataRequestExtensionMessage> m + (new UTMetadataRequestExtensionMessage(extensionMessageID)); + m->setIndex(index.i()); + m->setDownloadContext(_dctx); + m->setPeer(_peer); + m->setBtMessageFactory(_messageFactory); + m->setBtMessageDispatcher(_dispatcher); + return m; } case 1: { - if(end == length) { - throw DL_ABORT_EX("Bad ut_metadata data: data not found"); - } - const BDE& totalSize = dict["total_size"]; - if(!totalSize.isInteger()) { - throw DL_ABORT_EX("Bad ut_metadata data: total_size not found"); - } - SharedHandle<UTMetadataDataExtensionMessage> m - (new UTMetadataDataExtensionMessage(extensionMessageID)); - m->setIndex(index.i()); - m->setTotalSize(totalSize.i()); - m->setData(std::string(&data[1+end], &data[length])); - m->setUTMetadataRequestTracker(_tracker); - m->setPieceStorage(_dctx->getOwnerRequestGroup()->getPieceStorage()); - m->setDownloadContext(_dctx); - return m; + if(end == length) { + throw DL_ABORT_EX("Bad ut_metadata data: data not found"); + } + const BDE& totalSize = dict["total_size"]; + if(!totalSize.isInteger()) { + throw DL_ABORT_EX("Bad ut_metadata data: total_size not found"); + } + SharedHandle<UTMetadataDataExtensionMessage> m + (new UTMetadataDataExtensionMessage(extensionMessageID)); + m->setIndex(index.i()); + m->setTotalSize(totalSize.i()); + m->setData(std::string(&data[1+end], &data[length])); + m->setUTMetadataRequestTracker(_tracker); + m->setPieceStorage(_dctx->getOwnerRequestGroup()->getPieceStorage()); + m->setDownloadContext(_dctx); + return m; } case 2: { - SharedHandle<UTMetadataRejectExtensionMessage> m - (new UTMetadataRejectExtensionMessage(extensionMessageID)); - m->setIndex(index.i()); - // No need to inject tracker because peer will be disconnected. - return m; + SharedHandle<UTMetadataRejectExtensionMessage> m + (new UTMetadataRejectExtensionMessage(extensionMessageID)); + m->setIndex(index.i()); + // No need to inject tracker because peer will be disconnected. + return m; } default: - throw DL_ABORT_EX(StringFormat("Bad ut_metadata: unknown msg_type=%u", - msgType.i()).str()); + throw DL_ABORT_EX(StringFormat("Bad ut_metadata: unknown msg_type=%u", + msgType.i()).str()); } } else { throw DL_ABORT_EX - (StringFormat("Unsupported extension message received. extensionMessageID=%u, extensionName=%s", - extensionMessageID, extensionName.c_str()).str()); + (StringFormat("Unsupported extension message received. extensionMessageID=%u, extensionName=%s", + extensionMessageID, extensionName.c_str()).str()); } } } |
