diff options
Diffstat (limited to 'src/UTMetadataPostDownloadHandler.cc')
| -rw-r--r-- | src/UTMetadataPostDownloadHandler.cc | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/src/UTMetadataPostDownloadHandler.cc b/src/UTMetadataPostDownloadHandler.cc new file mode 100644 index 00000000..14f8ce6e --- /dev/null +++ b/src/UTMetadataPostDownloadHandler.cc @@ -0,0 +1,109 @@ +/* <!-- copyright */ +/* + * aria2 - The high speed download utility + * + * Copyright (C) 2009 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 "UTMetadataPostDownloadHandler.h" +#include "BDE.h" +#include "bittorrent_helper.h" +#include "RequestGroup.h" +#include "download_helper.h" +#include "RecoverableException.h" +#include "A2STR.h" +#include "DownloadContext.h" +#include "Logger.h" +#include "util.h" +#include "a2functional.h" +#include "DiskAdaptor.h" +#include "PieceStorage.h" + +namespace aria2 { + +bool UTMetadataPostDownloadHandler::Criteria::match +(const RequestGroup* requestGroup) const +{ + SharedHandle<DownloadContext> dctx = + requestGroup->getDownloadContext(); + if(dctx->hasAttribute(bittorrent::BITTORRENT)) { + const BDE& attrs = dctx->getAttribute(bittorrent::BITTORRENT); + if(!attrs.containsKey(bittorrent::METADATA)) { + return true; + } + } + return false; +} + +UTMetadataPostDownloadHandler::UTMetadataPostDownloadHandler() +{ + setCriteria(SharedHandle<Criteria>(new Criteria())); +} + +void UTMetadataPostDownloadHandler::getNextRequestGroups +(std::deque<SharedHandle<RequestGroup> >& groups, RequestGroup* requestGroup) +{ + SharedHandle<DownloadContext> dctx = requestGroup->getDownloadContext(); + const BDE& attrs = dctx->getAttribute(bittorrent::BITTORRENT); + std::string torrent = + strconcat("d4:info", + util::toString(requestGroup->getPieceStorage()->getDiskAdaptor()), + "e"); + try { + std::deque<SharedHandle<RequestGroup> > newRgs; + createRequestGroupForBitTorrent(newRgs, requestGroup->getOption(), + std::deque<std::string>(), torrent); + if(attrs.containsKey(bittorrent::ANNOUNCE_LIST)) { + for(std::deque<SharedHandle<RequestGroup> >::const_iterator i = + newRgs.begin(); i != newRgs.end(); ++i) { + SharedHandle<DownloadContext> newDctx = (*i)->getDownloadContext(); + if(!newDctx->hasAttribute(bittorrent::BITTORRENT)) { + continue; + } + BDE& newAttrs = newDctx->getAttribute(bittorrent::BITTORRENT); + if(attrs[bittorrent::INFO_HASH].s() != + newAttrs[bittorrent::INFO_HASH].s()) { + continue; + } + if(!newAttrs.containsKey(bittorrent::ANNOUNCE_LIST)) { + newAttrs[bittorrent::ANNOUNCE_LIST] = + attrs[bittorrent::ANNOUNCE_LIST]; + } + break; + } + } + + groups.insert(groups.end(), newRgs.begin(), newRgs.end()); + } catch(RecoverableException& e) { + _logger->error("Failed to parse BitTorrent metadata.", e); + } +} + +} // namespace aria2 |