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