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), cb_installed(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 void swift::AddProgressCallback (int transfer, TransferProgressCallback cb) {
  44     FileTransfer* trans = FileTransfer::file(transfer);
  45     if (!trans)
  46         return;
  47     trans->callbacks[trans->cb_installed++] = cb;
  48 }
  49
  50
  51 void swift::ExternallyRetrieved (int transfer,bin64_t piece) {
  52     FileTransfer* trans = FileTransfer::file(transfer);
  53     if (!trans)
  54         return;
  55     trans->ack_out().set(piece); // that easy
  56 }
  57
  58
  59 void swift::RemoveProgressCallback (int transfer, TransferProgressCallback cb) {
  60     FileTransfer* trans = FileTransfer::file(transfer);
  61     if (!trans)
  62         return;
  63     for(int i=0; i<trans->cb_installed; i++)
  64         if (trans->callbacks[i].cb==cb.cb)
  65             trans->callbacks[i]=trans->callbacks[--trans->cb_installed];
  66 }
  67
  68
  69 FileTransfer::~FileTransfer ()
  70 {
  71     Channel::CloseTransfer(this);
  72     files[fd()] = NULL;
  73     delete picker_;
  74 }
  75
  76
  77 FileTransfer* FileTransfer::Find (const Sha1Hash& root_hash) {
  78     for(int i=0; i<files.size(); i++)
  79         if (files[i] && files[i]->root_hash()==root_hash)
  80             return files[i];
  81     return NULL;
  82 }
  83
  84
  85 int       swift:: Find (Sha1Hash hash) {
  86     FileTransfer* t = FileTransfer::Find(hash);
  87     if (t)
  88         return t->fd();
  89     return -1;
  90 }
  91
  92
  93
  94 void            FileTransfer::OnPexIn (const Address& addr) {
  95     for(int i=0; i<hs_in_.size(); i++) {
  96         Channel* c = Channel::channel(hs_in_[i]);
  97         if (c && c->transfer().fd()==this->fd() && c->peer()==addr)
  98             return; // already connected
  99     }
 100     if (hs_in_.size()<20) {
 101         new Channel(this,Datagram::default_socket(),addr);
 102     } else {
 103         pex_in_.push_back(addr);
 104         if (pex_in_.size()>1000)
 105             pex_in_.pop_front();
 106     }
 107 }
 108
 109
 110 int        FileTransfer::RevealChannel (int& pex_out_) { // FIXME brainfuck
 111     pex_out_ -= hs_in_offset_;
 112     if (pex_out_<0)
 113         pex_out_ = 0;
 114     while (pex_out_<hs_in_.size()) {
 115         Channel* c = Channel::channel(hs_in_[pex_out_]);
 116         if (c && c->transfer().fd()==this->fd()) {
 117             if (c->is_established()) {
 118                 pex_out_ += hs_in_offset_ + 1;
 119                 return c->id();
 120             } else
 121                 pex_out_++;
 122         } else {
 123             hs_in_[pex_out_] = hs_in_[0];
 124             hs_in_.pop_front();
 125             hs_in_offset_++;
 126         }
 127     }
 128     pex_out_ += hs_in_offset_;
 129     return -1;
 130 }
 131