Search Site ~Apps~  btpeer   trackerbt   Socks4   The Minder   duct  ~Projects~  RobotWorld   Cgi++   IBank   SDL C++ Gui   Applesoft   Wolfshade   Legend  ~Contact~ message board  email author   visit project   homepage at   freshmeat.net

PeerHandler Class Reference Custom sockethandler for bittorrent client. More... #include <PeerHandler.h> Collaboration diagram for PeerHandler: [legend]List of all members. Public Member Functions   PeerHandler ()   PeerHandler (StdLog *)   ~PeerHandler () void  RegSession (Session *) bool  SessionExists (const std::string &hash) Session *  GetSession (const std::string &hash) const std::string &  GetTorrentDirectory () void  SetTorrentDirectory (const std::string &x) void  mkpath (const std::string &) void  Tick (time_t) port_t  GetListenPort () void  SetListenPort (port_t x) const std::string &  GetExternIP () void  SetExternIP (const std::string &x) void  dprintf (int id, char *format,...) int  GetChokeTimer () void  SetChokeTimer (int x) size_t  GetMinPeers () void  SetMinPeers (size_t x) size_t  GetMaxPeers () void  SetMaxPeers (size_t x) int  LocalChokeTime () void  SetLocalChokeTime (int x) size_t  PeerCount (const std::string &hash) bool  Connected (Peer *) pSocket *  PeerSocket (Peer *) void  SendHave (const std::string &have, size_t piece) bool  IgnoreChoke () void  SetIgnoreChoke (bool x=true) int  MaxRequestAge () void  SetMaxRequestAge (int x) void  Save () void  Show () void  Show (StatusSocket *) bool  Quit () void  SetQuit (bool x=true) void  SetDebug (int x) int  GetDebug () size_t  GetDownloaders () void  SetDownloaders (size_t x) size_t  GetOptimistic () void  SetOptimistic (size_t x) void  CheckDownloadRate () std::string  GetTime () Private Types typedef std::map< std::string, Session * >  session_m Private Member Functions   PeerHandler (const PeerHandler &) PeerHandler &  operator= (const PeerHandler &) Private Attributes session_m  m_sessions std::string  m_torrent_dir port_t  m_listen_port std::string  m_extern_ip int  m_choke_timer size_t  m_min_peers size_t  m_max_peers int  m_local_choke_time bool  m_ignore_choke int  m_max_request_age bool  m_quit int  m_debug size_t  m_downloaders size_t  m_optimistic Detailed Description Custom sockethandler for bittorrent client. Definition at line 40 of file PeerHandler.h. Member Typedef Documentation typedef std::map<std::string,Session *> PeerHandler::session_m [private] Definition at line 42 of file PeerHandler.h. Constructor & Destructor Documentation PeerHandler::PeerHandler (  )  Definition at line 47 of file PeerHandler.cpp. :SocketHandler() ,m_listen_port(0) ,m_choke_timer(300) ,m_min_peers(10) ,m_max_peers(20) ,m_local_choke_time(30) ,m_ignore_choke(false) ,m_max_request_age(30) ,m_quit(false) ,m_debug(0) ,m_downloaders(4) ,m_optimistic(1) { srand(time(NULL) * (time_t)this); } PeerHandler::PeerHandler ( StdLog *   )  Definition at line 65 of file PeerHandler.cpp. :SocketHandler(p) ,m_listen_port(0) ,m_choke_timer(300) ,m_min_peers(10) ,m_max_peers(20) ,m_local_choke_time(30) ,m_ignore_choke(false) ,m_max_request_age(30) ,m_quit(false) ,m_debug(0) ,m_downloaders(4) ,m_optimistic(1) { } PeerHandler::~PeerHandler (  )  Definition at line 82 of file PeerHandler.cpp. References m_sessions. { while (m_sessions.size()) { session_m::iterator it = m_sessions.begin(); Session *p = (*it).second; delete p; m_sessions.erase(it); } } PeerHandler::PeerHandler ( const PeerHandler &   )  [inline, private] Definition at line 117 of file PeerHandler.h. {} // copy constructor Member Function Documentation void PeerHandler::RegSession ( Session *   )  Definition at line 94 of file PeerHandler.cpp. References m_sessions. Referenced by MetainfoSocket::InitSession(). { m_sessions[p -> GetInfoHash()] = p; } bool PeerHandler::SessionExists ( const std::string &  hash  )  Definition at line 100 of file PeerHandler.cpp. References m_sessions. Referenced by MetainfoSocket::InitSession(). { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { if ((*it).first == hash && dynamic_cast<Session *>((*it).second)) return true; } return false; } Session * PeerHandler::GetSession ( const std::string &  hash  )  Definition at line 127 of file PeerHandler.cpp. References m_sessions. Referenced by FileManager::FileManager(), FileManager::GetFiles(), FileManager::ReadPiece(), FileManager::Verify(), and FileManager::Write(). { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { if ((*it).first == hash && dynamic_cast<Session *>((*it).second)) return (*it).second; } return NULL; } const std::string& PeerHandler::GetTorrentDirectory (  )  [inline] Definition at line 52 of file PeerHandler.h. References m_torrent_dir. Referenced by FileManager::FileManager(), FileManager::GetFiles(), and MetainfoSocket::InitSession(). { return m_torrent_dir; } void PeerHandler::SetTorrentDirectory ( const std::string &  x  )  [inline] Definition at line 53 of file PeerHandler.h. References m_torrent_dir. Referenced by main(). { m_torrent_dir = x; } void PeerHandler::mkpath ( const std::string &   )  Definition at line 111 of file PeerHandler.cpp. Referenced by MetainfoSocket::InitSession(), and FileManager::OpenFiles(). { for (size_t i = 0; i < path.size(); i++) { if (path[i] == '/') { #ifdef _WIN32 _mkdir(path.substr(0, i).c_str()); #else mkdir(path.substr(0, i).c_str(), 0750); #endif } } } void PeerHandler::Tick ( time_t   )  Definition at line 138 of file PeerHandler.cpp. References m_sessions. Referenced by main(). { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { Session *sess = (*it).second; if (sess) { if (now > sess -> GetTimeTracker() + sess -> GetInterval()) { tSocket *p = new tSocket(*this, sess -> GetInfoHash()); p -> SetDeleteByHandler(); Add(p); sess -> SetTimeTracker(); } sess -> Update(); } } } port_t PeerHandler::GetListenPort (  )  [inline] Definition at line 58 of file PeerHandler.h. References m_listen_port. { return m_listen_port; } void PeerHandler::SetListenPort ( port_t  x  )  [inline] Definition at line 59 of file PeerHandler.h. References m_listen_port. Referenced by main(). { m_listen_port = x; } const std::string& PeerHandler::GetExternIP (  )  [inline] Definition at line 60 of file PeerHandler.h. References m_extern_ip. { return m_extern_ip; } void PeerHandler::SetExternIP ( const std::string &  x  )  [inline] Definition at line 61 of file PeerHandler.h. References m_extern_ip. Referenced by main(). { m_extern_ip = x; } void PeerHandler::dprintf ( int  id, char *  format,   ...   ) Definition at line 158 of file PeerHandler.cpp. References cstring::c_str(). Referenced by MyLog::error(). { // char *colors = "rgybmcwlRGYBMCWL"; char *colors = "rgymcRGYBMCWL"; va_list ap; va_start(ap, format); char tmp[5000]; #ifdef _WIN32 vsprintf(tmp, format, ap); va_end(ap); printf("%s\n", tmp); #else vsnprintf(tmp, 5000, format, ap); va_end(ap); cstring str; char slask[10]; sprintf(slask, "&%c", colors[id % strlen(colors)]); str = slask; str += tmp; str += "&n"; printf("%s\n", str.c_str()); if (0) { FILE *fil = fopen("btlog","at"); if (!fil) fil = fopen("btlog","wt"); fprintf(fil,"%s\n", str.c_str()); fclose(fil); } #endif } int PeerHandler::GetChokeTimer (  )  [inline] Definition at line 65 of file PeerHandler.h. References m_choke_timer. Referenced by Session::AddConnect(). { return m_choke_timer; } void PeerHandler::SetChokeTimer ( int  x  )  [inline] Definition at line 66 of file PeerHandler.h. References m_choke_timer. Referenced by main(). { m_choke_timer = x; } size_t PeerHandler::GetMinPeers (  )  [inline] outgoing connections stop at this limit Definition at line 68 of file PeerHandler.h. References m_min_peers. Referenced by Session::Update(). { return m_min_peers; } void PeerHandler::SetMinPeers ( size_t  x  )  [inline] Definition at line 69 of file PeerHandler.h. References m_min_peers. Referenced by main(). { m_min_peers = x; } size_t PeerHandler::GetMaxPeers (  )  [inline] no more incoming connections will be accepted above this limit Definition at line 71 of file PeerHandler.h. References m_max_peers. { return m_max_peers; } void PeerHandler::SetMaxPeers ( size_t  x  )  [inline] Definition at line 72 of file PeerHandler.h. References m_max_peers. Referenced by main(). { m_max_peers = x; } int PeerHandler::LocalChokeTime (  )  [inline] Definition at line 74 of file PeerHandler.h. References m_local_choke_time. { return m_local_choke_time; } void PeerHandler::SetLocalChokeTime ( int  x  )  [inline] Definition at line 75 of file PeerHandler.h. References m_local_choke_time. { m_local_choke_time = x; } size_t PeerHandler::PeerCount ( const std::string &  hash  )  Definition at line 191 of file PeerHandler.cpp. Referenced by Session::Update(). { size_t q = 0; for (socket_m::iterator it = m_sockets.begin(); it != m_sockets.end(); it++) { pSocket *p = dynamic_cast<pSocket *>((*it).second); if (p && p -> GetHash() == hash) { q++; } } return q; } bool PeerHandler::Connected ( Peer *   )  Definition at line 206 of file PeerHandler.cpp. { for (socket_m::iterator it = m_sockets.begin(); it != m_sockets.end(); it++) { pSocket *p = dynamic_cast<pSocket *>((*it).second); if (p && p -> GetHash() == peer -> GetHash() && p -> GetRemoteAddress() == peer -> GetIP()) { return true; } } return false; } pSocket * PeerHandler::PeerSocket ( Peer *   )  Definition at line 220 of file PeerHandler.cpp. { for (socket_m::iterator it = m_sockets.begin(); it != m_sockets.end(); it++) { pSocket *p = dynamic_cast<pSocket *>((*it).second); if (p && p -> GetHash() == peer -> GetHash() && p -> GetRemoteAddress() == peer -> GetIP()) { return p; } } return NULL; } void PeerHandler::SendHave ( const std::string &  have, size_t  piece   ) Definition at line 234 of file PeerHandler.cpp. Referenced by Session::Update(). { for (socket_m::iterator it = m_sockets.begin(); it != m_sockets.end(); it++) { pSocket *p = dynamic_cast<pSocket *>((*it).second); if (p && p -> GetHash() == hash && p -> CTS() ) { p -> SendHave(piece); } } } bool PeerHandler::IgnoreChoke (  )  [inline] Definition at line 82 of file PeerHandler.h. References m_ignore_choke. Referenced by Session::Update(). { return m_ignore_choke; } void PeerHandler::SetIgnoreChoke ( bool  x = true  )  [inline] Definition at line 83 of file PeerHandler.h. References m_ignore_choke. Referenced by main(). { m_ignore_choke = x; } int PeerHandler::MaxRequestAge (  )  [inline] Definition at line 85 of file PeerHandler.h. References m_max_request_age. { return m_max_request_age; } void PeerHandler::SetMaxRequestAge ( int  x  )  [inline] Definition at line 86 of file PeerHandler.h. References m_max_request_age. Referenced by main(). { m_max_request_age = x; } void PeerHandler::Save (  )  Definition at line 247 of file PeerHandler.cpp. References m_sessions. Referenced by main(). { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { Session *sess = (*it).second; sess -> Save(); } } void PeerHandler::Show (  )  [inline] Definition at line 90 of file PeerHandler.h. References m_sessions. Referenced by main(). { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { Session *sess = (*it).second; if (sess) { sess -> PeerStatus(); } } } void PeerHandler::Show ( StatusSocket *   )  Definition at line 257 of file PeerHandler.cpp. References m_sessions. { for (session_m::iterator it = m_sessions.begin(); it != m_sessions.end(); it++) { Session *sess = (*it).second; sess -> Status(sock); } } bool PeerHandler::Quit (  )  [inline] Definition at line 103 of file PeerHandler.h. References m_quit. Referenced by main(). { return m_quit; } void PeerHandler::SetQuit ( bool  x = true  )  [inline] Definition at line 104 of file PeerHandler.h. References m_quit. { m_quit = x; } void PeerHandler::SetDebug ( int  x  )  [inline] Definition at line 106 of file PeerHandler.h. References m_debug. { m_debug = x; } int PeerHandler::GetDebug (  )  [inline] Definition at line 107 of file PeerHandler.h. References m_debug. Referenced by MyLog::error(), and Session::Update(). { return m_debug; } size_t PeerHandler::GetDownloaders (  )  [inline] Definition at line 108 of file PeerHandler.h. References m_downloaders. Referenced by CheckDownloadRate(), and Session::Update(). { return m_downloaders; } void PeerHandler::SetDownloaders ( size_t  x  )  [inline] Definition at line 109 of file PeerHandler.h. References m_downloaders. Referenced by main(). { m_downloaders = x; } size_t PeerHandler::GetOptimistic (  )  [inline] Definition at line 110 of file PeerHandler.h. References m_optimistic. Referenced by CheckDownloadRate(), and Session::Update(). { return m_optimistic; } void PeerHandler::SetOptimistic ( size_t  x  )  [inline] Definition at line 111 of file PeerHandler.h. References m_optimistic. Referenced by main(). { m_optimistic = x; } void PeerHandler::CheckDownloadRate (  )  Definition at line 267 of file PeerHandler.cpp. References DEB, GetDownloaders(), and GetOptimistic(). Referenced by Session::Update(). { // TODO: disabled return; std::map<pSocket *,size_t> mmap; size_t max = 0; DEB(printf("CheckDownloadRate()\n");) for (socket_m::iterator it = m_sockets.begin(); it != m_sockets.end(); it++) { pSocket *p = dynamic_cast<pSocket *>((*it).second); if (p) { Peer *peer = p -> GetPeer(); if (p && peer && p -> CTS() && !peer -> IsChoked()) { size_t sz; if (p -> GetDownloadRate(sz)) { mmap[p] = sz; max = MAX(max,sz); } } } } if (mmap.size() == GetDownloaders() + GetOptimistic() ) { pSocket *least = NULL; for (std::map<pSocket *,size_t>::iterator it = mmap.begin(); it != mmap.end(); it++) { pSocket *sock = (*it).first; size_t sz = (*it).second; if (sz < max) { least = sock; max = sz; } } if (least) { //DEB(printf("Kicking pSocket with download rate %d bytes/sec\n", max);) least -> SendChoke(true); } } } std::string PeerHandler::GetTime (  )  Definition at line 314 of file PeerHandler.cpp. { char slask[100]; #ifdef _WIN32 SYSTEMTIME time; ::GetLocalTime(&time); sprintf(slask, "%ld.%03ld", time.wHour * 3600 + time.wMinute + 60 + time.wSecond, time.wMilliseconds); #else struct timeval tv; struct timezone tz; gettimeofday(&tv, &tz); sprintf(slask, "%ld.%06ld", tv.tv_sec, tv.tv_usec); #endif return slask; } PeerHandler& PeerHandler::operator= ( const PeerHandler &   )  [inline, private] Definition at line 118 of file PeerHandler.h. { return *this; } // assignment operator Member Data Documentation session_m PeerHandler::m_sessions [private] Definition at line 120 of file PeerHandler.h. Referenced by GetSession(), RegSession(), Save(), SessionExists(), Show(), Tick(), and ~PeerHandler(). std::string PeerHandler::m_torrent_dir [private] Definition at line 121 of file PeerHandler.h. Referenced by GetTorrentDirectory(), and SetTorrentDirectory(). port_t PeerHandler::m_listen_port [private] Definition at line 122 of file PeerHandler.h. Referenced by GetListenPort(), and SetListenPort(). std::string PeerHandler::m_extern_ip [private] Definition at line 123 of file PeerHandler.h. Referenced by GetExternIP(), and SetExternIP(). int PeerHandler::m_choke_timer [private] Definition at line 124 of file PeerHandler.h. Referenced by GetChokeTimer(), and SetChokeTimer(). size_t PeerHandler::m_min_peers [private] Definition at line 125 of file PeerHandler.h. Referenced by GetMinPeers(), and SetMinPeers(). size_t PeerHandler::m_max_peers [private] Definition at line 126 of file PeerHandler.h. Referenced by GetMaxPeers(), and SetMaxPeers(). int PeerHandler::m_local_choke_time [private] Definition at line 127 of file PeerHandler.h. Referenced by LocalChokeTime(), and SetLocalChokeTime(). bool PeerHandler::m_ignore_choke [private] Definition at line 128 of file PeerHandler.h. Referenced by IgnoreChoke(), and SetIgnoreChoke(). int PeerHandler::m_max_request_age [private] Definition at line 129 of file PeerHandler.h. Referenced by MaxRequestAge(), and SetMaxRequestAge(). bool PeerHandler::m_quit [private] Definition at line 130 of file PeerHandler.h. Referenced by Quit(), and SetQuit(). int PeerHandler::m_debug [private] Definition at line 131 of file PeerHandler.h. Referenced by GetDebug(), and SetDebug(). size_t PeerHandler::m_downloaders [private] Definition at line 132 of file PeerHandler.h. Referenced by GetDownloaders(), and SetDownloaders(). size_t PeerHandler::m_optimistic [private] Definition at line 133 of file PeerHandler.h. Referenced by GetOptimistic(), and SetOptimistic(). The documentation for this class was generated from the following files: PeerHandler.h PeerHandler.cpp

Page, code, and content Copyright (C) 2006 by Anders Hedström

---