examples/msg/bittorrent/peer.h

   1 /* Copyright (c) 2012. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6 #ifndef BITTORRENT_PEER_H
   7 #define BITTORRENT_PEER_H
   8 #include <msg/msg.h>
   9 #include <xbt/dict.h>
  10 #include <xbt/dynar.h>
  11 #include <xbt/RngStream.h>
  12 #include "connection.h"
  13 #include "bittorrent.h"
  14
  15 /**
  16  * Peer data
  17  */
  18 typedef struct s_peer {
  19         int id; //peer id
  20
  21         int pieces; //number of pieces the peer has.
  22         char *bitfield; //list of pieces the peer has.
  23         char *bitfield_blocks; //list of blocks the peer has.
  24         short *pieces_count; //number of peers that have each piece.
  25
  26         xbt_dynar_t current_pieces; //current pieces the peer is downloading
  27         int current_piece; //current pieces
  28         int pieces_requested; //number of pieces the peer has requested
  29
  30         xbt_dict_t peers; //peers list
  31         xbt_dict_t active_peers; //active peers list
  32
  33         char mailbox[MAILBOX_SIZE]; //peer mailbox.
  34         char mailbox_tracker[MAILBOX_SIZE]; //pair mailbox while communicating with the tracker.
  35         const char *hostname; //peer hostname
  36
  37         msg_task_t task_received; //current task being received
  38         msg_comm_t comm_received; //current comm
  39
  40         int round; //current round for the chocking algortihm.
  41
  42         RngStream stream; //RngStream for
  43
  44         double begin_receive_time; //time when the receiving communication has begun, useful for calculating host speed.
  45 } s_peer_t, *peer_t;
  46
  47 /**
  48  * Peer main function
  49  */
  50 int peer(int argc, char *argv[]);
  51
  52 int get_peers_data(peer_t peer);
  53 void leech_loop(peer_t peer, double deadline);
  54 void seed_loop(peer_t peer, double deadline);
  55
  56 void peer_init(peer_t, int id, int seed);
  57 void peer_free(peer_t peer);
  58
  59 int has_finished(char *bitfield);
  60
  61 void handle_message(peer_t peer, msg_task_t task);
  62
  63 void wait_for_pieces( peer_t peer, double deadline);
  64
  65 void update_pieces_count_from_bitfield(peer_t peer, char *bitfield);
  66 void update_current_piece(peer_t peer);
  67 void update_choked_peers(peer_t peer);
  68
  69 void update_interested_after_receive(peer_t peer);
  70
  71 void update_bitfield_blocks(peer_t peer, int index, int block_index, int block_length);
  72 int piece_complete(peer_t peer, int index);
  73 int get_first_block(peer_t peer, int piece);
  74
  75
  76 void send_requests_to_peer(peer_t peer, connection_t remote_peer);
  77
  78 void send_interested_to_peers(peer_t peer);
  79 void send_handshake_all(peer_t peer);
  80
  81 void send_interested(peer_t peer, const char *mailbox);
  82
  83 void send_notinterested(peer_t peer, const char *mailbox);
  84 void send_handshake(peer_t peer, const char *mailbox);
  85 void send_bitfield(peer_t peer, const char *mailbox);
  86 void send_choked(peer_t peer, const char *mailbox);
  87 void send_unchoked(peer_t peer, const char *mailbox);
  88 void send_have(peer_t peer, int piece);
  89
  90 void send_request(peer_t peer, const char *mailbox, int piece,int block_index, int block_length);
  91 void send_piece(peer_t peer, const char *mailbox, int piece, int stalled, int block_index, int block_length);
  92
  93 int in_current_pieces(peer_t peer, int piece);
  94 #endif /* BITTORRENT_PEER_H */