transfer.cpp

   1 /*
   2  *  transfer.cpp
   3  *  some transfer-scope code
   4  *
   5  *  Created by Victor Grishchenko on 10/6/09.
   6  *  Copyright 2009 Delft University of Technology. All rights reserved.
   7  *
   8  */
   9 #include <errno.h>
  10 #include <string>
  11 #include <sstream>
  12 #include "swift.h"
  13
  14 #include "ext/seq_picker.cpp" // FIXME FIXME FIXME FIXME
  15
  16 using namespace swift;
  17
  18 std::vector<FileTransfer*> FileTransfer::files(20);
  19
  20 #define BINHASHSIZE (sizeof(bin64_t)+sizeof(Sha1Hash))
  21
  22 // FIXME: separate Bootstrap() and Download(), then Size(), Progress(), SeqProgress()
  23
  24 FileTransfer::FileTransfer (const char* filename, const Sha1Hash& _root_hash) :
  25     file_(filename,_root_hash), hs_in_offset_(0)
  26 {
  27     if (files.size()<fd()+1)
  28         files.resize(fd()+1);
  29     files[fd()] = this;
  30     picker_ = new SeqPiecePicker(this);
  31     picker_->Randomize(rand()&63);
  32     init_time_ = Datagram::Time();
  33 }
  34
  35
  36 void    Channel::CloseTransfer (FileTransfer* trans) {
  37     for(int i=0; i<Channel::channels.size(); i++)
  38         if (Channel::channels[i] && Channel::channels[i]->transfer_==trans)
  39             delete Channel::channels[i];
  40 }
  41
  42
  43 FileTransfer::~FileTransfer ()
  44 {
  45     Channel::CloseTransfer(this);
  46     files[fd()] = NULL;
  47     delete picker_;
  48 }
  49
  50
  51 FileTransfer* FileTransfer::Find (const Sha1Hash& root_hash) {
  52     for(int i=0; i<files.size(); i++)
  53         if (files[i] && files[i]->root_hash()==root_hash)
  54             return files[i];
  55     return NULL;
  56 }
  57
  58
  59 void            FileTransfer::OnPexIn (const Address& addr) {
  60     for(int i=0; i<hs_in_.size(); i++) {
  61         Channel* c = Channel::channel(hs_in_[i]);
  62         if (c && c->transfer().fd()==this->fd() && c->peer()==addr)
  63             return; // already connected
  64     }
  65     if (hs_in_.size()<20) {
  66         new Channel(this,Channel::default_socket(),addr);
  67     } else {
  68         pex_in_.push_back(addr);
  69         if (pex_in_.size()>1000)
  70             pex_in_.pop_front();
  71     }
  72 }
  73
  74
  75 int        FileTransfer::RevealChannel (int& pex_out_) { // FIXME brainfuck
  76     pex_out_ -= hs_in_offset_;
  77     if (pex_out_<0)
  78         pex_out_ = 0;
  79     while (pex_out_<hs_in_.size()) {
  80         Channel* c = Channel::channel(hs_in_[pex_out_]);
  81         if (c && c->transfer().fd()==this->fd()) {
  82             if (c->is_established()) {
  83                 pex_out_ += hs_in_offset_ + 1;
  84                 return c->id();
  85             } else
  86                 pex_out_++;
  87         } else {
  88             hs_in_[pex_out_] = hs_in_[0];
  89             hs_in_.pop_front();
  90             hs_in_offset_++;
  91         }
  92     }
  93     pex_out_ += hs_in_offset_;
  94     return -1;
  95 }
  96