/
Tracker.h
62 lines (43 loc) · 971 Bytes
/
Tracker.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#ifndef __TORRENTSTREAM_TRACKER_H
#define __TORRENTSTREAM_TRACKER_H
namespace TorrentStream
{
enum class TrackerEvent
{
None = 0,
Started,
Stopped,
Completed
};
class Tracker
{
public:
Tracker(const std::string& announceURL, const std::string& infoHash, const std::string& peerID, int port, Client* client);
~Tracker();
std::unique_ptr<Peer> FetchPeer()
{
std::unique_lock<std::mutex> _(m_Mutex);
if (m_Peers.size() == 0)
{
return nullptr;
}
auto peer = std::move(m_Peers.back());
m_Peers.pop_back();
return peer;
}
private:
Client* m_Client;
void RunThread();
void Update(TrackerEvent event = TrackerEvent::None);
std::string m_AnnounceURL;
std::string m_PeerID;
std::string m_InfoHash;
std::set<std::string> m_Known;
std::vector<std::unique_ptr<Peer>> m_Peers;
bool m_Started = false;
int m_Port = 0;
std::mutex m_Mutex;
std::unique_ptr<std::thread> m_Thread;
};
}
#endif