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