src/main/hashchecker.cpp

   1 /***************************************************************************
   2  *                                                                         *
   3  *   Copyright (C) 2006 David Brodsky                                      *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or         *
   6  *   modify it under the terms of the GNU General Public License as        *
   7  *   published by the Free Software Foundation and appearing               *
   8  *   in the file LICENSE.GPL included in the packaging of this file.       *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU      *
  13  *   General Public License for more details.                              *
  14  *                                                                         *
  15  ***************************************************************************/
  16
  17 #include <tairon/core/signals.h>
  18 #include <tairon/core/threadmanager.h>
  19 #include <tairon/crypto/sha1.h>
  20
  21 #include "hashchecker.h"
  22
  23 #include "hashcheckerthread.h"
  24 #include "storage.h"
  25
  26 namespace Tairent
  27 {
  28
  29 namespace Main
  30 {
  31
  32 /* {{{ HashChecker::HashChecker(uint32_t, Storage *) */
  33 HashChecker::HashChecker(uint32_t i, Storage *s) : index(i)
  34 {
  35         piece = s->getPiece(i);
  36
  37         pieceHash = s->getMetaInfo()["pieces"].asString().substr(i * 20, 20);
  38
  39         disabler = new Tairon::Core::FunctorDisabler();
  40         disabler->incRef();
  41 }
  42 /* }}} */
  43
  44 /* {{{ HashChecker::~HashChecker() */
  45 HashChecker::~HashChecker()
  46 {
  47         disabler->decRef();
  48 }
  49 /* }}} */
  50
  51 /* {{{ HashChecker::check() */
  52 void HashChecker::check()
  53 {
  54         Tairon::Core::ThreadFunctorCaller *caller = new Tairon::Core::ThreadMethodDFunctorCaller0<HashChecker>(this, &HashChecker::checkPiece, disabler);
  55         Tairon::Core::ThreadManager::self()->getThreadByName("hash")->addFunctorCaller(caller);
  56 }
  57 /* }}} */
  58
  59 /* {{{ HashChecker::checkPiece() */
  60 void HashChecker::checkPiece()
  61 {
  62         Tairon::Crypto::SHA1 hash;
  63
  64         for (std::list<PieceStruct::FilePieceStruct>::const_iterator it = piece->pieces.begin(); it != piece->pieces.end(); ++it)
  65                 hash.update(it->mem, it->length);
  66
  67         if (hash.final() == pieceHash)
  68                 hashCorrectSignal.emit(index, this);
  69         else
  70                 hashIncorrectSignal.emit(index, this);
  71 }
  72 /* }}} */
  73
  74 /* {{{ HashChecker::destroy() */
  75 void HashChecker::destroy()
  76 {
  77         HashCheckerThread::self()->addCheckerToDelete(this);
  78 }
  79 /* }}} */
  80
  81 }; // namespace Main
  82
  83 }; // namespace Tairent
  84
  85 // vim: ai sw=4 ts=4 noet fdm=marker