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 "storage.h"
  24
  25 namespace Tairent
  26 {
  27
  28 namespace Main
  29 {
  30
  31 /* {{{ HashChecker::HashChecker(uint32_t, Storage *) */
  32 HashChecker::HashChecker(uint32_t i, Storage *s) : index(i)
  33 {
  34         piece = s->getPiece(i);
  35
  36         pieceHash = s->getMetaInfo()["pieces"].asString().substr(i * 20, 20);
  37
  38         disabler = new Tairon::Core::FunctorDisabler();
  39         disabler->incRef();
  40 }
  41 /* }}} */
  42
  43 /* {{{ HashChecker::~HashChecker() */
  44 HashChecker::~HashChecker()
  45 {
  46         disabler->decRef();
  47 }
  48 /* }}} */
  49
  50 /* {{{ HashChecker::check() */
  51 void HashChecker::check()
  52 {
  53         Tairon::Core::ThreadFunctorCaller *caller = new Tairon::Core::ThreadMethodDFunctorCaller0<HashChecker>(this, &HashChecker::checkPiece, disabler);
  54         Tairon::Core::ThreadManager::self()->getThreadByName("hash")->addFunctorCaller(caller);
  55 }
  56 /* }}} */
  57
  58 /* {{{ HashChecker::checkPiece() */
  59 void HashChecker::checkPiece()
  60 {
  61         Tairon::Crypto::SHA1 hash;
  62
  63         for (std::list<PieceStruct::FilePieceStruct>::const_iterator it = piece->pieces.begin(); it != piece->pieces.end(); ++it)
  64                 hash.update(it->mem, it->length);
  65
  66         if (hash.final() == pieceHash)
  67                 hashCorrectSignal.emit(index, this);
  68         else
  69                 hashIncorrectSignal.emit(index, this);
  70 }
  71 /* }}} */
  72
  73 }; // namespace Main
  74
  75 }; // namespace Tairent
  76
  77 // vim: ai sw=4 ts=4 noet fdm=marker