src/azdhttracker/azdhttrackermodule.h

   1 /***************************************************************************
   2  *                                                                         *
   3  *   Copyright (C) 2007 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 #ifndef _azdhttracker_azdhttrackermodule_h
  18 #define _azdhttracker_azdhttrackermodule_h
  19
  20 #include <tairon/core/module.h>
  21
  22 using Tairon::Core::String;
  23
  24 namespace Tairon
  25 {
  26
  27 namespace Net
  28 {
  29
  30 class Timer;
  31
  32 }; // namespace Net
  33
  34 }; // namespace Tairon
  35
  36 namespace Tairent
  37 {
  38
  39 namespace Core
  40 {
  41
  42 class BEncode;
  43
  44 }; // namespace Core
  45
  46 namespace Main
  47 {
  48
  49 struct TorrentStruct;
  50
  51 }; // namespace Main
  52
  53 namespace AzDHTTracker
  54 {
  55
  56 /** \brief Implementation of DHT tracker.
  57  */
  58 class AzDHTTrackerModule : public Tairon::Core::Module
  59 {
  60         public:
  61                 /** Constructs an AzDHTTrackerModule object.
  62                  */
  63                 AzDHTTrackerModule();
  64
  65                 /** Destroys the object.
  66                  */
  67                 virtual ~AzDHTTrackerModule();
  68
  69         private:
  70                 /** Calls given method for all served torrents.
  71                  */
  72                 void forAllTorrents(void (AzDHTTrackerModule::*method)(const String &, Tairent::Main::TorrentStruct *));
  73
  74                 /** Requests list of peers from the DHT.
  75                  */
  76                 void getPeers(const String &infoHash, Tairent::Main::TorrentStruct *torrent);
  77
  78                 /** Registers our torrents in the DHT and requests peers.
  79                  */
  80                 void init();
  81
  82                 /** Registers torrent in the DHT.
  83                  */
  84                 void registerTorrent(const String &infoHash, Tairent::Main::TorrentStruct *torrent);
  85
  86                 /** Removes torrent's announce from the DHT.
  87                  */
  88                 void removeTorrent(const String &infoHash, Tairent::Main::TorrentStruct *torrent);
  89
  90                 /** Requests list of peers for all served torrents.
  91                  */
  92                 void timeout();
  93
  94                 /** Processes list of values supplied by the DHT.
  95                  *
  96                  * \param key The key for which the values were requested.
  97                  * \param values Usually list of values.
  98                  */
  99                 void valuesRead(const String &key, const Tairent::Core::BEncode &values);
 100
 101         private:
 102                 /** Timer for getting peers in regular intervals.
 103                  */
 104                 Tairon::Net::Timer *timer;
 105 };
 106
 107 }; // namespace AzDHTTracker
 108
 109 }; // namespace Tairent
 110
 111 #endif
 112
 113 // vim: ai sw=4 ts=4 noet fdm=marker