channel.cpp

   1 /*
   2  *  swift.cpp
   3  *  serp++
   4  *
   5  *  Created by Victor Grishchenko on 3/6/09.
   6  *  Copyright 2009 Delft University of Technology. All rights reserved.
   7  *
   8  */
   9
  10 #include <stdlib.h>
  11 #include <fcntl.h>
  12 #ifndef _WIN32
  13 #include <sys/select.h>
  14 #include <sys/time.h>
  15 #include <sys/mman.h>
  16 #include <arpa/inet.h>
  17 #include <unistd.h>
  18 #endif
  19 #include <sys/stat.h>
  20 #include <string.h>
  21
  22 //#include <glog/logging.h>
  23 #include "swift.h"
  24 #include "datagram.h"
  25
  26 using namespace std;
  27 using namespace swift;
  28
  29 swift::tint Channel::last_tick = 0;
  30 int Channel::MAX_REORDERING = 4;
  31 bool Channel::SELF_CONN_OK = false;
  32 swift::tint Channel::TIMEOUT = TINT_SEC*60;
  33 std::vector<Channel*> Channel::channels(1);
  34 SOCKET Channel::sockets[8] = {0,0,0,0,0,0,0,0};
  35 int Channel::socket_count = 0;
  36 Address Channel::tracker;
  37 tbheap Channel::send_queue;
  38 FILE* Channel::debug_file = NULL;
  39 #include "ext/simple_selector.cpp"
  40 PeerSelector* Channel::peer_selector = new SimpleSelector();
  41
  42 Channel::Channel    (FileTransfer* transfer, int socket, Address peer_addr) :
  43     transfer_(transfer), peer_(peer_addr), peer_channel_id_(0), pex_out_(0),
  44     socket_(socket==-1?sockets[0]:socket), // FIXME
  45     data_out_cap_(bin64_t::ALL), last_data_out_time_(0), last_data_in_time_(0),
  46     own_id_mentioned_(false), next_send_time_(0), last_send_time_(0),
  47     last_recv_time_(0), rtt_avg_(TINT_SEC), dev_avg_(0), dip_avg_(TINT_SEC),
  48     data_in_dbl_(bin64_t::NONE), hint_out_size_(0),
  49     cwnd_(1), send_interval_(TINT_SEC), send_control_(PING_PONG_CONTROL),
  50     sent_since_recv_(0), ack_rcvd_recent_(0), ack_not_rcvd_recent_(0),
  51     last_loss_time_(0), owd_min_bin_(0), owd_min_bin_start_(NOW),
  52     owd_cur_bin_(0), dgrams_sent_(0), dgrams_rcvd_(0),
  53     data_in_(TINT_NEVER,bin64_t::NONE)
  54 {
  55     if (peer_==Address())
  56         peer_ = tracker;
  57     this->id_ = channels.size();
  58     channels.push_back(this);
  59     transfer_->hs_in_.push_back(id_);
  60     for(int i=0; i<4; i++) {
  61         owd_min_bins_[i] = TINT_NEVER;
  62         owd_current_[i] = TINT_NEVER;
  63     }
  64     Reschedule();
  65     dprintf("%s #%u init %s\n",tintstr(),id_,peer_.str());
  66 }
  67
  68
  69 Channel::~Channel () {
  70     channels[id_] = NULL;
  71 }
  72
  73
  74 void     swift::SetTracker(const Address& tracker) {
  75     Channel::tracker = tracker;
  76 }
  77
  78
  79 int Channel::DecodeID(int scrambled) {
  80     return scrambled ^ (int)Datagram::start;
  81 }
  82 int Channel::EncodeID(int unscrambled) {
  83     return unscrambled ^ (int)Datagram::start;
  84 }
  85
  86
  87 int     swift::Listen (Address addr) {
  88     int sock = Datagram::Bind(addr);
  89     if (sock!=INVALID_SOCKET)
  90         Channel::sockets[Channel::socket_count++] = sock;
  91     return sock;
  92 }
  93
  94
  95 void    swift::Shutdown (int sock_des) {
  96     for(int i=0; i<Channel::socket_count; i++)
  97         if (sock_des==-1 || Channel::sockets[i]==sock_des) {
  98             Datagram::Close(Channel::sockets[i]);
  99             Channel::sockets[i] = Channel::sockets[--Channel::socket_count];
 100         }
 101 }
 102
 103
 104 void    swift::Loop (tint till) {
 105     Channel::Loop(till);
 106 }
 107
 108
 109 int      swift::Open (const char* filename, const Sha1Hash& hash) {
 110     FileTransfer* ft = new FileTransfer(filename, hash);
 111     if (ft && ft->file().file_descriptor()) {
 112
 113         /*if (FileTransfer::files.size()<fdes)  // FIXME duplication
 114             FileTransfer::files.resize(fdes);
 115         FileTransfer::files[fdes] = ft;*/
 116
 117         // initiate tracker connections
 118         if (Channel::tracker!=Address())
 119             new Channel(ft);
 120
 121         return ft->file().file_descriptor();
 122     } else {
 123         if (ft)
 124             delete ft;
 125         return -1;
 126     }
 127 }
 128
 129
 130 void    swift::Close (int fd) {
 131     if (fd<FileTransfer::files.size() && FileTransfer::files[fd])
 132         delete FileTransfer::files[fd];
 133 }
 134
 135
 136 void    swift::AddPeer (Address address, const Sha1Hash& root) {
 137     Channel::peer_selector->AddPeer(address,root);
 138 }
 139
 140
 141 uint64_t  swift::Size (int fdes) {
 142     if (FileTransfer::files.size()>fdes && FileTransfer::files[fdes])
 143         return FileTransfer::files[fdes]->file().size();
 144     else
 145         return 0;
 146 }
 147
 148
 149 bool  swift::IsComplete (int fdes) {
 150     if (FileTransfer::files.size()>fdes && FileTransfer::files[fdes])
 151         return FileTransfer::files[fdes]->file().is_complete();
 152     else
 153         return 0;
 154 }
 155
 156
 157 uint64_t  swift::Complete (int fdes) {
 158     if (FileTransfer::files.size()>fdes && FileTransfer::files[fdes])
 159         return FileTransfer::files[fdes]->file().complete();
 160     else
 161         return 0;
 162 }
 163
 164
 165 uint64_t  swift::SeqComplete (int fdes) {
 166     if (FileTransfer::files.size()>fdes && FileTransfer::files[fdes])
 167         return FileTransfer::files[fdes]->file().seq_complete();
 168     else
 169         return 0;
 170 }
 171
 172
 173 const Sha1Hash& swift::RootMerkleHash (int file) {
 174     FileTransfer* trans = FileTransfer::file(file);
 175     if (!trans)
 176         return Sha1Hash::ZERO;
 177     return trans->file().root_hash();
 178 }
 179
 180
 181 /**    <h2> swift handshake </h2>
 182  Basic rules:
 183  <ul>
 184  <li>    to send a datagram, a channel must be created
 185  (channels are cheap and easily recycled)
 186  <li>    a datagram must contain either the receiving
 187  channel id (scrambled) or the root hash
 188  </ul>
 189  <b>Note:</b>
 190  */