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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
/* Copyright (c) 2012-2025. The SimGrid Team. All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
* under the terms of the license (GNU LGPL) which comes with this package. */
#ifndef BITTORRENT_PEER_HPP
#define BITTORRENT_PEER_HPP
#include "s4u-bittorrent.hpp"
#include <set>
#include <unordered_map>
class Connection {
public:
int id; // Peer id
simgrid::s4u::Mailbox* mailbox_;
unsigned int bitfield = 0U; // Fields
double peer_speed = 0;
double last_unchoke = 0;
int current_piece = -1;
bool am_interested = false; // Indicates if we are interested in something the peer has
bool interested = false; // Indicates if the peer is interested in one of our pieces
bool choked_upload = true; // Indicates if the peer is choked for the current peer
bool choked_download = true; // Indicates if the peer has choked the current peer
explicit Connection(int id) : id(id), mailbox_(simgrid::s4u::Mailbox::by_name(std::to_string(id))){};
void addSpeedValue(double speed) { peer_speed = peer_speed * 0.6 + speed * 0.4; }
bool hasPiece(unsigned int piece) const { return bitfield & 1U << piece; }
};
class Peer {
int id;
double deadline;
simgrid::xbt::random::XbtRandom random;
simgrid::s4u::Mailbox* mailbox_;
std::unordered_map<int, Connection> connected_peers;
std::set<Connection*> active_peers; // active peers list
unsigned int bitfield_ = 0; // list of pieces the peer has.
unsigned long long bitfield_blocks = 0; // list of blocks the peer has.
std::vector<short> pieces_count; // number of peers that have each piece.
unsigned int current_pieces = 0; // current pieces the peer is downloading
double begin_receive_time = 0; // time when the receiving communication has begun, useful for calculating host speed.
int round_ = 0; // current round for the chocking algorithm.
simgrid::s4u::CommPtr comm_received = nullptr; // current comm
Message* message = nullptr; // current message being received
public:
explicit Peer(std::vector<std::string> args);
Peer(const Peer&) = delete;
Peer& operator=(const Peer&) = delete;
void operator()();
std::string getStatus() const;
bool hasFinished() const;
int nbInterestedPeers() const;
bool isInterestedBy(const Connection* remote_peer) const;
bool isInterestedByFree(const Connection* remote_peer) const;
void updateActivePeersSet(Connection* remote_peer);
void updateInterestedAfterReceive();
void updateChokedPeers();
bool hasNotPiece(unsigned int piece) const { return not(bitfield_ & 1U << piece); }
bool remotePeerHasMissingPiece(const Connection* remote_peer, unsigned int piece) const
{
return hasNotPiece(piece) && remote_peer->hasPiece(piece);
}
bool hasCompletedPiece(unsigned int piece) const;
unsigned int countPieces(unsigned int bitfield) const;
/** Check that a piece is not currently being download by the peer. */
bool isNotDownloadingPiece(unsigned int piece) const { return not(current_pieces & 1U << piece); }
int partiallyDownloadedPiece(const Connection* remote_peer) const;
void updatePiecesCountFromBitfield(unsigned int bitfield);
void removeCurrentPiece(Connection* remote_peer, unsigned int current_piece);
void updateBitfieldBlocks(int piece, int block_index, int block_length);
int getFirstMissingBlockFrom(int piece) const;
int selectPieceToDownload(const Connection* remote_peer);
void requestNewPieceTo(Connection* remote_peer);
bool getPeersFromTracker();
void sendMessage(simgrid::s4u::Mailbox* mailbox, MessageType type, uint64_t size);
void sendBitfield(simgrid::s4u::Mailbox* mailbox);
void sendPiece(simgrid::s4u::Mailbox* mailbox, unsigned int piece, int block_index, int block_length);
void sendHandshakeToAllPeers();
void sendHaveToAllPeers(unsigned int piece);
void sendRequestTo(Connection* remote_peer, unsigned int piece);
void handleMessage();
void leech();
void seed();
};
#endif /* BITTORRENT_PEER_HPP */
|
