~ubuntu-branches/ubuntu/oneiric/libktorrent/oneiric

« back to all changes in this revision

Viewing changes to src/utp/connection.cpp

Committer: Bazaar Package Importer
Author(s): Modestas Vainius
Date: 2011-05-26 00:33:38 UTC
mfrom: (5.1.7 experimental)
Revision ID: james.westby@ubuntu.com-20110526003338-2r4zwyzn8bovsxay

Tags: 1.1.1-2

Release to unstable.

files added:
RoadMap

cmake/modules/FindLibGcrypt.cmake

debian/libktorrent3.install

debian/libktorrent3.symbols

examples

examples/CMakeLists.txt

examples/ktcli

examples/ktcli/CMakeLists.txt

examples/ktcli/ktcli.cpp

examples/ktcli/ktcli.h

examples/ktcli/main.cpp

src/datachecker/CMakeLists.txt

src/datachecker/tests

src/datachecker/tests/CMakeLists.txt

src/datachecker/tests/datacheckertest.cpp

src/diskio/CMakeLists.txt

src/diskio/tests

src/diskio/tests/CMakeLists.txt

src/diskio/tests/chunkmanagertest.cpp

src/download/CMakeLists.txt

src/download/streamingchunkselector.cpp

src/download/streamingchunkselector.h

src/download/tests

src/download/tests/CMakeLists.txt

src/download/tests/streamingchunkselectortest.cpp

src/mse/CMakeLists.txt

src/mse/tests

src/mse/tests/CMakeLists.txt

src/mse/tests/rc4encryptortest.cpp

src/net/serversocket.cpp

src/net/serversocket.h

src/peer/CMakeLists.txt

src/peer/superseeder.cpp

src/peer/superseeder.h

src/peer/tests

src/peer/tests/CMakeLists.txt

src/peer/tests/superseedtest.cpp

src/torrent/tests/torrentfilestreammultitest.cpp

src/torrent/tests/torrentfilestreamtest.cpp

src/torrent/torrentfilestream.cpp

src/torrent/torrentfilestream.h

src/tracker/CMakeLists.txt

src/upnp

src/upnp/CMakeLists.txt

src/upnp/httprequest.cpp

src/upnp/httprequest.h

src/upnp/soap.cpp

src/upnp/soap.h

src/upnp/upnpdescriptionparser.cpp

src/upnp/upnpdescriptionparser.h

src/upnp/upnpmcastsocket.cpp

src/upnp/upnpmcastsocket.h

src/upnp/upnprouter.cpp

src/upnp/upnprouter.h

src/util/resourcemanager.cpp

src/util/resourcemanager.h

src/util/signalcatcher.cpp

src/util/signalcatcher.h

src/util/tests/fileopstest.cpp

src/util/tests/resourcemanagertest.cpp

src/util/tests/signalcatchertest.cpp

src/utp/outputqueue.cpp

src/utp/outputqueue.h

src/utp/utpserver_p.h

testlib

testlib/CMakeLists.txt

testlib/dummytorrentcreator.cpp

testlib/dummytorrentcreator.h

files removed:
debian/libktorrent2.install

debian/libktorrent2.symbols

src/datachecker/datacheckerlistener.cpp

src/datachecker/datacheckerlistener.h

files modified:
CMakeLists.txt

ChangeLog

debian/changelog

debian/control

debian/copyright

debian/rules

po/ar/libktorrent.po

po/ast/libktorrent.po

po/be/libktorrent.po

po/bg/libktorrent.po

po/ca/libktorrent.po

po/ca@valencia/libktorrent.po

po/cs/libktorrent.po

po/da/libktorrent.po

po/de/libktorrent.po

po/el/libktorrent.po

po/en_GB/libktorrent.po

po/eo/libktorrent.po

po/es/libktorrent.po

po/et/libktorrent.po

po/eu/libktorrent.po

po/fi/libktorrent.po

po/fr/libktorrent.po

po/ga/libktorrent.po

po/gl/libktorrent.po

po/hi/libktorrent.po

po/hne/libktorrent.po

po/hr/libktorrent.po

po/hu/libktorrent.po

po/is/libktorrent.po

po/it/libktorrent.po

po/ja/libktorrent.po

po/km/libktorrent.po

po/ku/libktorrent.po

po/lt/libktorrent.po

po/lv/libktorrent.po

po/ms/libktorrent.po

po/nb/libktorrent.po

po/nds/libktorrent.po

po/nl/libktorrent.po

po/nn/libktorrent.po

po/oc/libktorrent.po

po/pl/libktorrent.po

po/pt/libktorrent.po

po/pt_BR/libktorrent.po

po/ro/libktorrent.po

po/ru/libktorrent.po

po/se/libktorrent.po

po/si/libktorrent.po

po/sk/libktorrent.po

po/sl/libktorrent.po

po/sr/libktorrent.po

po/sr@ijekavian/libktorrent.po

po/sr@ijekavianlatin/libktorrent.po

po/sr@latin/libktorrent.po

po/sv/libktorrent.po

po/tr/libktorrent.po

po/uk/libktorrent.po

po/zh_CN/libktorrent.po

po/zh_TW/libktorrent.po

scripts/gen_pkg.rb

src/CMakeLists.txt

src/bcodec/value.h

src/datachecker/datachecker.cpp

src/datachecker/datachecker.h

src/datachecker/datacheckerjob.cpp

src/datachecker/datacheckerjob.h

src/datachecker/datacheckerthread.cpp

src/datachecker/datacheckerthread.h

src/datachecker/multidatachecker.cpp

src/datachecker/multidatachecker.h

src/datachecker/singledatachecker.cpp

src/datachecker/singledatachecker.h

src/dht/announcetask.cpp

src/dht/announcetask.h

src/dht/dht.cpp

src/dht/dht.h

src/dht/kbucket.cpp

src/dht/kbucket.h

src/dht/node.cpp

src/dht/node.h

src/dht/nodelookup.cpp

src/dht/nodelookup.h

src/dht/rpccall.cpp

src/dht/rpccall.h

src/dht/rpcmsg.cpp

src/dht/rpcmsg.h

src/dht/rpcserver.cpp

src/dht/rpcserver.h

src/dht/task.cpp

src/dht/task.h

src/diskio/cache.cpp

src/diskio/cache.h

src/diskio/cachefile.cpp

src/diskio/chunk.cpp

src/diskio/chunk.h

src/diskio/chunkmanager.cpp

src/diskio/chunkmanager.h

src/diskio/dndfile.cpp

src/diskio/movedatafilesjob.cpp

src/diskio/movedatafilesjob.h

src/diskio/multifilecache.cpp

src/diskio/multifilecache.h

src/diskio/piecedata.cpp

src/diskio/piecedata.h

src/diskio/singlefilecache.cpp

src/diskio/singlefilecache.h

src/download/chunkdownload.cpp

src/download/chunkdownload.h

src/download/chunkselector.cpp

src/download/chunkselector.h

src/download/downloader.cpp

src/download/downloader.h

src/download/httpconnection.cpp

src/download/httpconnection.h

src/download/packet.cpp

src/download/webseed.cpp

src/download/webseed.h

src/interfaces/chunkselectorinterface.cpp

src/interfaces/chunkselectorinterface.h

src/interfaces/peerinterface.cpp

src/interfaces/peerinterface.h

src/interfaces/piecedownloader.h

src/interfaces/serverinterface.cpp

src/interfaces/serverinterface.h

src/interfaces/torrentfileinterface.cpp

src/interfaces/torrentfileinterface.h

src/interfaces/torrentinterface.h

src/interfaces/trackerinterface.cpp

src/interfaces/trackerinterface.h

src/magnet/magnetdownloader.cpp

src/magnet/magnetdownloader.h

src/magnet/magnetlink.cpp

src/magnet/magnetlink.h

src/magnet/tests/magnetlinktest.cpp

src/mse/encryptedauthenticate.cpp

src/mse/encryptedauthenticate.h

src/mse/encryptedserverauthenticate.cpp

src/mse/encryptedserverauthenticate.h

src/mse/rc4encryptor.cpp

src/mse/rc4encryptor.h

src/mse/streamsocket.cpp

src/mse/streamsocket.h

src/net/CMakeLists.txt

src/net/downloadthread.h

src/net/socket.cpp

src/net/socket.h

src/net/socketdevice.cpp

src/net/socketdevice.h

src/net/socketmonitor.cpp

src/net/socketmonitor.h

src/net/socks.cpp

src/net/socks.h

src/net/uploadthread.h

src/net/wakeuppipe.cpp

src/peer/authenticate.cpp

src/peer/authenticate.h

src/peer/authenticatebase.cpp

src/peer/authenticatebase.h

src/peer/authenticationmonitor.cpp

src/peer/authenticationmonitor.h

src/peer/chunkcounter.h

src/peer/packetreader.cpp

src/peer/packetreader.h

src/peer/packetwriter.cpp

src/peer/peer.cpp

src/peer/peer.h

src/peer/peerconnector.cpp

src/peer/peerconnector.h

src/peer/peerdownloader.cpp

src/peer/peermanager.cpp

src/peer/peermanager.h

src/peer/peeruploader.cpp

src/peer/peeruploader.h

src/peer/serverauthenticate.cpp

src/peer/serverauthenticate.h

src/peer/utpex.cpp

src/peer/utpex.h

src/torrent/CMakeLists.txt

src/torrent/advancedchokealgorithm.cpp

src/torrent/globals.cpp

src/torrent/job.cpp

src/torrent/job.h

src/torrent/jobqueue.cpp

src/torrent/server.cpp

src/torrent/server.h

src/torrent/tests/CMakeLists.txt

src/torrent/timeestimator.cpp

src/torrent/torrent.cpp

src/torrent/torrentcontrol.cpp

src/torrent/torrentcontrol.h

src/torrent/torrentfile.cpp

src/torrent/torrentfile.h

src/torrent/torrentstats.cpp

src/torrent/torrentstats.h

src/torrent/uploader.cpp

src/torrent/uploader.h

src/tracker/httpannouncejob.cpp

src/tracker/httpannouncejob.h

src/tracker/httptracker.cpp

src/tracker/udptracker.cpp

src/tracker/udptracker.h

src/tracker/udptrackersocket.cpp

src/tracker/udptrackersocket.h

src/util/CMakeLists.txt

src/util/file.cpp

src/util/file.h

src/util/fileops.cpp

src/util/fileops.h

src/util/functions.cpp

src/util/functions.h

src/util/log.cpp

src/util/log.h

src/util/pipe.cpp

src/util/sha1hashgen.cpp

src/util/sha1hashgen.h

src/util/tests/CMakeLists.txt

src/util/waitjob.cpp

src/util/waitjob.h

src/utp/connection.cpp

src/utp/connection.h

src/utp/delaywindow.cpp

src/utp/delaywindow.h

src/utp/localwindow.cpp

src/utp/localwindow.h

src/utp/pollpipe.h

src/utp/remotewindow.cpp

src/utp/remotewindow.h

src/utp/tests/congestiontest.cpp

src/utp/tests/connectiontest.cpp

src/utp/tests/connecttest.cpp

src/utp/tests/fintest.cpp

src/utp/tests/localwindowtest.cpp

src/utp/tests/packetlosstest.cpp

src/utp/tests/remotewindowtest.cpp

src/utp/tests/sendtest.cpp

src/utp/tests/sockettest.cpp

src/utp/tests/transmittest.cpp

src/utp/tests/utppolltest.cpp

src/utp/utpserver.cpp

src/utp/utpserver.h

src/utp/utpserverthread.cpp

src/utp/utpsocket.cpp

src/utp/utpsocket.h

src/version.cpp

src/version.h

Show diffs side-by-side

added added

removed removed

src/utp/connection.cpp

}

Connection::Connection(bt::Uint16 recv_connection_id, Type type, const net::Address& remote, Transmitter* transmitter)

: transmitter(transmitter)

: transmitter(transmitter),timer_id(-1)

{

stats.type = type;

stats.remote = remote;

stats.packets_sent = 0;

stats.bytes_lost = 0;

stats.packets_lost = 0;

stats.readable = stats.writeable = false;

connect(this,SIGNAL(doDelayedStartTimer()),this,SLOT(delayedStartTimer()),Qt::QueuedConnection);

if (type == INCOMING)

262

263

break;

263

264

}

264

265

266

checkState();

265

267

startTimer();

266

268

return stats.state;

267

269

}

268

270

271

void Connection::checkState()

272

{

273

// Check if we have become readable or writeable, and notify if necessary

274

bool r = local_wnd->size() > 0 || stats.state == CS_CLOSED;

275

bool w = remote_wnd->availableSpace() > 0 && stats.state == CS_CONNECTED;

276

bool r_changed = !stats.readable && r;

277

bool w_changed = !stats.writeable && w;

278

stats.readable = r;

279

stats.writeable = w;

280

281

mutex.unlock();

282

// Temporary unlock the mutex to avoid a deadlock

283

if (r_changed || w_changed)

284

transmitter->stateChanged(self.toStrongRef(),r_changed,w_changed);

285

mutex.lock();

286

}

287

288

269

289

void Connection::checkIfClosed()

270

290

{

271

291

// Check if we need to go to the closed state

272

292

// We can do this if all our packets have been acked and the local window

273

293

// has been fully read

274

if (remote_wnd->allPacketsAcked() && local_wnd->isEmpty())

294

if (stats.state == CS_FINISHED && remote_wnd->allPacketsAcked() && local_wnd->isEmpty())

275

295

{

276

296

stats.state = CS_CLOSED;

277

297

Out(SYS_UTP|LOG_NOTICE) << "UTP: Connection " << stats.recv_connection_id << "|" << stats.send_connection_id << " closed " << endl;

323

343

}

324

344

325

345

326

if (!transmitter->sendTo(ba,stats.remote,receiveConnectionID()))

346

if (!transmitter->sendTo(self.toStrongRef(),ba))

327

347

throw TransmissionError(__FILE__,__LINE__);

328

348

329

349

last_packet_sent = tv;

407

427

// first put data in the output buffer then send packets

408

428

bt::Uint32 ret = output_buffer.write(data,len);

409

429

sendPackets();

430

stats.writeable = !output_buffer.full();

410

431

return ret;

411

432

}

412

433

442

463

else

443

464

sendState();

444

465

}

445

446

int Connection::sendDataPacket(const QByteArray& packet)

466

467

void Connection::sendDataPacket(const QByteArray& packet, Uint16 seq_nr, const utp::TimeValue& now)

447

468

{

448

bt::Uint32 to_send = packet.size();

449

TimeValue now;

450

451

469

bt::Uint32 extension_length = 0;

452

470

bt::Uint32 sack_bits = local_wnd->selectiveAckBits();

453

471

if (sack_bits > 0)

462

480

hdr.timestamp_microseconds = now.timestampMicroSeconds();

463

481

hdr.timestamp_difference_microseconds = stats.reply_micro;

464

482

hdr.wnd_size = stats.last_window_size_transmitted = local_wnd->availableSpace();

465

hdr.seq_nr = stats.seq_nr;

483

hdr.seq_nr = seq_nr;

466

484

hdr.ack_nr = local_wnd->lastSeqNr();

467

485

hdr.write((bt::Uint8*)ba.data());

468

486

476

494

local_wnd->fillSelectiveAck(&sack);

477

495

}

478

496

479

memcpy(ba.data() + Header::size() + extension_length,packet.data(),to_send);

480

if (!transmitter->sendTo(ba,stats.remote,receiveConnectionID()))

497

memcpy(ba.data() + Header::size() + extension_length,packet.data(),packet.size());

498

if (!transmitter->sendTo(self.toStrongRef(),ba))

481

499

throw TransmissionError(__FILE__,__LINE__);

482

500

483

501

last_packet_sent = now;

484

502

stats.packets_sent++;

503

}

504

505

506

void Connection::sendDataPacket(const QByteArray& packet)

507

{

508

TimeValue now;

509

sendDataPacket(packet,stats.seq_nr,now);

510

485

511

remote_wnd->addPacket(packet,stats.seq_nr,bt::Now());

486

512

stats.seq_nr++;

487

513

startTimer();

488

return to_send;

489

514

}

490

515

491

516

void Connection::retransmit(const QByteArray& packet, Uint16 p_seq_nr)

492

517

{

493

518

TimeValue now;

494

495

bt::Uint32 extension_length = 0;

496

bt::Uint32 sack_bits = local_wnd->selectiveAckBits();

497

if (sack_bits > 0)

498

extension_length += 2 + qMin(sack_bits / 8,(bt::Uint32)4);

499

500

QByteArray ba(Header::size() + extension_length + packet.size(),0);

501

Header hdr;

502

hdr.version = 1;

503

hdr.type = ST_DATA;

504

hdr.extension = extension_length == 0 ? 0 : SELECTIVE_ACK_ID;

505

hdr.connection_id = stats.send_connection_id;

506

hdr.timestamp_microseconds = now.timestampMicroSeconds();

507

hdr.timestamp_difference_microseconds = stats.reply_micro;

508

hdr.wnd_size = stats.last_window_size_transmitted = local_wnd->availableSpace();

509

hdr.seq_nr = p_seq_nr;

510

hdr.ack_nr = local_wnd->lastSeqNr();

511

hdr.write((bt::Uint8*)ba.data());

512

513

if (extension_length > 0)

514

{

515

bt::Uint8* ptr = (bt::Uint8*)(ba.data() + Header::size());

516

SelectiveAck sack;

517

sack.extension = ptr[0] = 0;

518

sack.length = ptr[1] = extension_length - 2;

519

sack.bitmask = ptr + 2;

520

local_wnd->fillSelectiveAck(&sack);

521

}

522

523

memcpy(ba.data() + Header::size() + extension_length,packet.data(),packet.size());

519

sendDataPacket(packet, p_seq_nr, now);

524

520

startTimer();

525

if (!transmitter->sendTo(ba,stats.remote,receiveConnectionID()))

526

throw TransmissionError(__FILE__,__LINE__);

527

528

last_packet_sent = now;

529

stats.packets_sent++;

530

521

}

531

522

532

523

bt::Uint32 Connection::bytesAvailable() const

556

547

sendState();

557

548

558

549

stats.bytes_received += ret;

550

stats.readable = local_wnd->size() > 0;

559

551

return ret;

560

552

}

561

553

562

554

563

555

bool Connection::waitUntilConnected()

564

556

{

565

mutex.lock();

557

QMutexLocker lock(&mutex);

558

if (stats.state == CS_CONNECTED)

559

return true;

560

566

561

connected.wait(&mutex);

567

bool ret = stats.state == CS_CONNECTED;

568

mutex.unlock();

569

return ret;

562

return stats.state == CS_CONNECTED;

570

563

}

571

564

572

565

573

bool Connection::waitForData()

566

bool Connection::waitForData(Uint32 timeout)

574

567

{

575

mutex.lock();

576

data_ready.wait(&mutex);

577

bool ret = local_wnd->size() > 0;

578

mutex.unlock();

579

return ret;

568

QMutexLocker lock(&mutex);

569

if (local_wnd->size() > 0)

570

return true;

571

572

data_ready.wait(&mutex,timeout == 0 ? ULONG_MAX : timeout);

573

return local_wnd->size() > 0;

580

574

}

581

575

582

576

637

631

sendState();

638

632

}

639

633

break;

640

case CS_CLOSED:

641

634

case CS_IDLE:

642

635

startTimer();

643

636

break;

637

case CS_CLOSED:

638

break;

644

639

}

640

641

checkState();

642

if (stats.state == CS_CLOSED)

643

transmitter->closed(self.toStrongRef());

645

644

}

646

645

647

646

void Connection::dumpStats()

669

668

if (QThread::currentThread() != thread())

670

669

emit doDelayedStartTimer();

671

670

else

672

timer.start(stats.timeout,this);

671

delayedStartTimer();

673

672

}

674

673

675

674

void Connection::delayedStartTimer()

676

675

{

677

timer.start(stats.timeout,this);

678

}

679

680

void Connection::timerEvent(QTimerEvent* event)

681

{

682

Q_UNUSED(event);

683

handleTimeout();

684

}

685

676

if (timer_id != -1)

677

transmitter->cancelTimer(timer_id);

678

timer_id = transmitter->scheduleTimer(self.toStrongRef(),stats.timeout);

679

}

686

680

}

687

681

Older »