diff options
Diffstat (limited to 'src/PeerAbstractCommand.cc')
| -rw-r--r-- | src/PeerAbstractCommand.cc | 181 |
1 files changed, 181 insertions, 0 deletions
diff --git a/src/PeerAbstractCommand.cc b/src/PeerAbstractCommand.cc new file mode 100644 index 00000000..6cbd615c --- /dev/null +++ b/src/PeerAbstractCommand.cc @@ -0,0 +1,181 @@ +/* <!-- copyright */ +/* + * aria2 - a simple utility for downloading files faster + * + * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ +/* copyright --> */ +#include "PeerAbstractCommand.h" +#include "DlAbortEx.h" +#include "DlRetryEx.h" +#include <sys/time.h> +#include "Util.h" +#include "message.h" +#include "prefs.h" + +PeerAbstractCommand::PeerAbstractCommand(int cuid, Peer* peer, TorrentDownloadEngine* e, Socket* s): + Command(cuid), e(e), peer(peer), checkSocketIsReadable(false), checkSocketIsWritable(false) { + + if(s != NULL) { + socket = new Socket(*s); + setReadCheckSocket(socket); + } else { + socket = NULL; + } + this->checkPoint.tv_sec = 0; + this->checkPoint.tv_usec = 0; + + e->torrentMan->connections++; +} + +PeerAbstractCommand::~PeerAbstractCommand() { + setReadCheckSocket(NULL); + setWriteCheckSocket(NULL); + if(socket != NULL) { + delete(socket); + } + e->torrentMan->connections--; +} + +void PeerAbstractCommand::updateCheckPoint() { + gettimeofday(&checkPoint, NULL); +} + +bool PeerAbstractCommand::isTimeoutDetected() { + struct timeval now; + gettimeofday(&now, NULL); + if(checkPoint.tv_sec == 0 && checkPoint.tv_usec == 0) { + checkPoint = now; + return false; + } else { + long long int elapsed = Util::difftv(now, checkPoint); + if(elapsed >= e->option->getAsInt(PREF_TIMEOUT)*1000000) { + return true; + } else { + return false; + } + } +} + +bool PeerAbstractCommand::execute() { + try { + if(e->torrentMan->downloadComplete()) { + return true; + } + beforeSocketCheck(); + if(checkSocketIsReadable && !readCheckTarget->isReadable(0) + || checkSocketIsWritable && !writeCheckTarget->isWritable(0)) { + if(isTimeoutDetected()) { + // TODO + checkPoint.tv_sec = 0; + checkPoint.tv_usec = 0; + throw new DlRetryEx(EX_TIME_OUT); + } + e->commands.push(this); + return false; + } + updateCheckPoint(); + bool returnValue = executeInternal(); + //e->torrentMan->updatePeer(peer); + return returnValue; + } catch(Exception* err) { + e->logger->error(MSG_DOWNLOAD_ABORTED, err, cuid); + onAbort(err); + delete(err); + return prepareForNextPeer(0); + } catch(DlRetryEx* err) { + e->logger->error(MSG_RESTARTING_DOWNLOAD, err, cuid); + peer->tryCount++; + bool isAbort = e->option->getAsInt(PREF_MAX_TRIES) != 0 && + peer->tryCount >= e->option->getAsInt(PREF_MAX_TRIES); + int tryCount = peer->tryCount; + if(isAbort) { + onAbort(err); + } + delete(err); + if(isAbort) { + e->logger->error(MSG_MAX_TRY, cuid, tryCount); + return true; + } else { + return prepareForRetry(e->option->getAsInt(PREF_RETRY_WAIT)); + } + } +} + +// TODO this method removed when PeerBalancerCommand is implemented +bool PeerAbstractCommand::prepareForNextPeer(int wait) { + return true; +} + +bool PeerAbstractCommand::prepareForRetry(int wait) { + return true; +} + +void PeerAbstractCommand::onAbort(Exception* ex) { + peer->error = 1; + peer->tryCount = 0; + peer->cuid = 0; + peer->amChocking = true; + peer->amInterested = false; + peer->peerChoking = true; + peer->peerInterested = false; + e->logger->debug("CUID#%d - peer %s:%d banned.", cuid, peer->ipaddr.c_str(), peer->port); +} + +void PeerAbstractCommand::setReadCheckSocket(Socket* socket) { + if(socket == NULL) { + if(checkSocketIsReadable) { + e->deleteSocketForReadCheck(readCheckTarget); + checkSocketIsReadable = false; + readCheckTarget = NULL; + } + } else { + if(checkSocketIsReadable) { + if(readCheckTarget != socket) { + e->deleteSocketForReadCheck(readCheckTarget); + e->addSocketForReadCheck(socket); + readCheckTarget = socket; + } + } else { + e->addSocketForReadCheck(socket); + checkSocketIsReadable = true; + readCheckTarget = socket; + } + } +} + +void PeerAbstractCommand::setWriteCheckSocket(Socket* socket) { + if(socket == NULL) { + if(checkSocketIsWritable) { + e->deleteSocketForWriteCheck(writeCheckTarget); + checkSocketIsWritable = false; + writeCheckTarget = NULL; + } + } else { + if(checkSocketIsWritable) { + if(writeCheckTarget != socket) { + e->deleteSocketForWriteCheck(writeCheckTarget); + e->addSocketForWriteCheck(socket); + writeCheckTarget = socket; + } + } else { + e->addSocketForWriteCheck(socket); + checkSocketIsWritable = true; + writeCheckTarget = socket; + } + } +} |