diff options
Diffstat (limited to 'src/HandshakeExtensionMessage.cc')
| -rw-r--r-- | src/HandshakeExtensionMessage.cc | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/src/HandshakeExtensionMessage.cc b/src/HandshakeExtensionMessage.cc new file mode 100644 index 00000000..df003b5b --- /dev/null +++ b/src/HandshakeExtensionMessage.cc @@ -0,0 +1,174 @@ +/* <!-- copyright */ +/* + * aria2 - The high speed download utility + * + * Copyright (C) 2006 Tatsuhiro Tsujikawa + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * 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 + * + * In addition, as a special exception, the copyright holders give + * permission to link the code of portions of this program with the + * OpenSSL library under certain conditions as described in each + * individual source file, and distribute linked combinations + * including the two. + * You must obey the GNU General Public License in all respects + * for all of the code used other than OpenSSL. If you modify + * file(s) with this exception, you may extend this exception to your + * version of the file(s), but you are not obligated to do so. If you + * do not wish to do so, delete this exception statement from your + * version. If you delete this exception statement from all source + * files in the program, then also delete it here. + */ +/* copyright --> */ +#include "HandshakeExtensionMessage.h" +#include "Peer.h" +#include "BtContext.h" +#include "Dictionary.h" +#include "Data.h" +#include "Util.h" +#include "BencodeVisitor.h" +#include "BtRegistry.h" +#include "MetaFileUtil.h" +#include "DlAbortEx.h" +#include "LogFactory.h" +#include "message.h" + +const string HandshakeExtensionMessage::EXTENSION_NAME = "handshake"; + +HandshakeExtensionMessage::HandshakeExtensionMessage():_tcpPort(0), + _btContext(0), + _peer(0), + _logger(LogFactory::getInstance()) +{} + +HandshakeExtensionMessage::~HandshakeExtensionMessage() {} + +string HandshakeExtensionMessage::getBencodedData() +{ + SharedHandle<Dictionary> dic = new Dictionary(); + if(!_clientVersion.empty()) { + Data* v = new Data(_clientVersion); + dic->put("v", v); + } + if(_tcpPort > 0) { + string portStr = Util::itos(_tcpPort); + Data* p = new Data(portStr, true); + dic->put("p", p); + } + Dictionary* exts = new Dictionary(); + dic->put("m", exts); + for(map<string, uint8_t>::const_iterator itr = _extensions.begin(); + itr != _extensions.end(); ++itr) { + const map<string, uint8_t>::value_type& vt = *itr; + string idStr = Util::uitos((uint32_t)vt.second); + exts->put(vt.first, new Data(idStr, true)); + } + BencodeVisitor v; + dic->accept(&v); + return v.getBencodedData(); +} + +string HandshakeExtensionMessage::toString() const +{ + string s = getExtensionName(); + if(!_clientVersion.empty()) { + s += " client="+Util::urlencode(_clientVersion); + } + if(_tcpPort > 0) { + s += ", tcpPort="+Util::itos(_tcpPort); + } + for(map<string, uint8_t>::const_iterator itr = _extensions.begin(); + itr != _extensions.end(); ++itr) { + const map<string, uint8_t>::value_type& vt = *itr; + s += ", "+vt.first+"="+Util::uitos((uint32_t)vt.second); + } + return s; +} + +void HandshakeExtensionMessage::doReceivedAction() +{ + if(_tcpPort > 0) { + _peer->port = _tcpPort; + } + for(map<string, uint8_t>::const_iterator itr = _extensions.begin(); + itr != _extensions.end(); ++itr) { + const map<string, uint8_t>::value_type& vt = *itr; + _peer->setExtension(vt.first, vt.second); + } + if(_peer->port > 0) { + // This is needed when _peer is a connection initiator, listen port of + // _peer is now available, which is initially unknown. + // If _peer is a receiver or already its port is known, _peer has to be + // already added to PeerStorage using addPeer() and call + // PeerStorage::addPeer() here does nothing and just returns false. + PEER_STORAGE(_btContext)->addPeer(_peer); + } +} + +void HandshakeExtensionMessage::setPeer(const PeerHandle& peer) +{ + _peer = peer; +} + +void HandshakeExtensionMessage::setBtContext(const BtContextHandle& btContext) +{ + _btContext = btContext; +} + +uint8_t HandshakeExtensionMessage::getExtensionMessageID(const string& name) const +{ + map<string, uint8_t>::const_iterator i = _extensions.find(name); + if(i == _extensions.end()) { + return 0; + } else { + return (*i).second; + } +} + +HandshakeExtensionMessageHandle +HandshakeExtensionMessage::create(const char* data, size_t length) +{ + if(length < 1) { + throw new DlAbortEx(MSG_TOO_SMALL_PAYLOAD_SIZE, + EXTENSION_NAME.c_str(), length); + } + HandshakeExtensionMessageHandle msg = new HandshakeExtensionMessage(); + msg->_logger->debug("Creating HandshakeExtensionMessage from %s", + Util::urlencode((const unsigned char*)data, length).c_str()); + SharedHandle<MetaEntry> root = MetaFileUtil::bdecoding(data+1, length-1); + Dictionary* d = dynamic_cast<Dictionary*>(root.get()); + if(d == 0) { + throw new DlAbortEx("Unexpected payload format for extended message handshake"); + } + const Data* p = dynamic_cast<const Data*>(d->get("p")); + if(p) { + msg->_tcpPort = p->toInt(); + } + const Data* v = dynamic_cast<const Data*>(d->get("v")); + if(v) { + msg->_clientVersion = v->toString(); + } + const Dictionary* m = dynamic_cast<const Dictionary*>(d->get("m")); + if(m) { + const Order& order = m->getOrder(); + for(Order::const_iterator i = order.begin(); i != order.end(); ++i) { + const Data* e = dynamic_cast<const Data*>(m->get(*i)); + if(e) { + msg->_extensions[*i] = e->toInt(); + } + } + } + return msg; +} |