1268 lines
34 KiB
C
1268 lines
34 KiB
C
/******************************************************************************
|
|
* $Id$
|
|
*
|
|
* Copyright (c) 2005-2007 Transmission authors and contributors
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
* DEALINGS IN THE SOFTWARE.
|
|
*****************************************************************************/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include "transmission.h"
|
|
#include "fastresume.h"
|
|
#include "trcompat.h" /* for strlcpy */
|
|
#include "metainfo.h"
|
|
#include "net.h" /* tr_netNtop */
|
|
#include "shared.h"
|
|
|
|
/***
|
|
**** LOCKS
|
|
***/
|
|
|
|
void
|
|
tr_torrentReaderLock( const tr_torrent_t * tor )
|
|
{
|
|
tr_rwReaderLock ( (tr_rwlock_t*)&tor->lock );
|
|
}
|
|
|
|
void
|
|
tr_torrentReaderUnlock( const tr_torrent_t * tor )
|
|
{
|
|
tr_rwReaderUnlock ( (tr_rwlock_t*)&tor->lock );
|
|
}
|
|
|
|
void
|
|
tr_torrentWriterLock( tr_torrent_t * tor )
|
|
{
|
|
tr_rwWriterLock ( &tor->lock );
|
|
}
|
|
|
|
void
|
|
tr_torrentWriterUnlock( tr_torrent_t * tor )
|
|
{
|
|
tr_rwWriterUnlock ( &tor->lock );
|
|
}
|
|
|
|
/***
|
|
**** PER-TORRENT UL / DL SPEEDS
|
|
***/
|
|
|
|
void
|
|
tr_setUseCustomUpload( tr_torrent_t * tor, int limit )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tor->customUploadLimit = limit;
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_setUseCustomDownload( tr_torrent_t * tor, int limit )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tor->customDownloadLimit = limit;
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_setUploadLimit( tr_torrent_t * tor, int limit )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tr_rcSetLimit( tor->upload, limit );
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_setDownloadLimit( tr_torrent_t * tor, int limit )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tr_rcSetLimit( tor->download, limit );
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
/***
|
|
****
|
|
**** TORRENT INSTANTIATION
|
|
****
|
|
***/
|
|
|
|
static int
|
|
getBytePiece( const tr_info_t * info, uint64_t byteOffset )
|
|
{
|
|
assert( info != NULL );
|
|
assert( info->pieceSize != 0 );
|
|
|
|
return byteOffset / info->pieceSize;
|
|
}
|
|
|
|
static void
|
|
initFilePieces ( tr_info_t * info, int fileIndex )
|
|
{
|
|
tr_file_t * file = &info->files[fileIndex];
|
|
uint64_t firstByte, lastByte;
|
|
|
|
assert( info != NULL );
|
|
assert( 0<=fileIndex && fileIndex<info->fileCount );
|
|
|
|
file = &info->files[fileIndex];
|
|
firstByte = file->offset;
|
|
lastByte = firstByte + (file->length ? file->length-1 : 0);
|
|
file->firstPiece = getBytePiece( info, firstByte );
|
|
file->lastPiece = getBytePiece( info, lastByte );
|
|
tr_dbg( "file #%d is in pieces [%d...%d] (%s)", fileIndex, file->firstPiece, file->lastPiece, file->name );
|
|
}
|
|
|
|
static tr_priority_t
|
|
calculatePiecePriority ( const tr_torrent_t * tor,
|
|
int piece )
|
|
{
|
|
int i;
|
|
tr_priority_t priority = TR_PRI_NORMAL;
|
|
|
|
for( i=0; i<tor->info.fileCount; ++i )
|
|
{
|
|
const tr_file_t * file = &tor->info.files[i];
|
|
if ( file->firstPiece <= piece
|
|
&& file->lastPiece >= piece
|
|
&& file->priority > priority)
|
|
priority = file->priority;
|
|
}
|
|
|
|
return priority;
|
|
}
|
|
|
|
static void
|
|
tr_torrentInitFilePieces( tr_torrent_t * tor )
|
|
{
|
|
int i;
|
|
uint64_t offset = 0;
|
|
|
|
assert( tor != NULL );
|
|
|
|
for( i=0; i<tor->info.fileCount; ++i ) {
|
|
tor->info.files[i].offset = offset;
|
|
offset += tor->info.files[i].length;
|
|
initFilePieces( &tor->info, i );
|
|
}
|
|
|
|
for( i=0; i<tor->info.pieceCount; ++i )
|
|
tor->info.pieces[i].priority = calculatePiecePriority( tor, i );
|
|
}
|
|
|
|
static void torrentThreadLoop( void * );
|
|
|
|
static void
|
|
torrentRealInit( tr_handle_t * h,
|
|
tr_torrent_t * tor,
|
|
const char * destination,
|
|
int flags )
|
|
{
|
|
int i;
|
|
char name[512];
|
|
|
|
tor->info.flags |= flags;
|
|
|
|
tr_sharedLock( h->shared );
|
|
|
|
tor->destination = tr_strdup( destination );
|
|
|
|
tr_torrentInitFilePieces( tor );
|
|
|
|
tor->handle = h;
|
|
tor->id = h->id;
|
|
tor->key = h->key;
|
|
tor->azId = h->azId;
|
|
tor->hasChangedState = -1;
|
|
|
|
/* Escaped info hash for HTTP queries */
|
|
for( i = 0; i < SHA_DIGEST_LENGTH; i++ )
|
|
{
|
|
snprintf( &tor->escapedHashString[3*i],
|
|
sizeof( tor->escapedHashString ) - 3 * i,
|
|
"%%%02x", tor->info.hash[i] );
|
|
}
|
|
|
|
tor->pexDisabled = 0;
|
|
|
|
/* Block size: usually 16 ko, or less if we have to */
|
|
tor->blockSize = MIN( tor->info.pieceSize, 1 << 14 );
|
|
tor->blockCount = ( tor->info.totalSize + tor->blockSize - 1 ) /
|
|
tor->blockSize;
|
|
tor->completion = tr_cpInit( tor );
|
|
|
|
tor->thread = THREAD_EMPTY;
|
|
tr_rwInit( &tor->lock );
|
|
|
|
tor->upload = tr_rcInit();
|
|
tor->download = tr_rcInit();
|
|
tor->swarmspeed = tr_rcInit();
|
|
|
|
/* We have a new torrent */
|
|
tor->publicPort = tr_sharedGetPublicPort( h->shared );
|
|
|
|
tr_sharedUnlock( h->shared );
|
|
|
|
if( !h->isPortSet )
|
|
tr_setBindPort( h, TR_DEFAULT_PORT );
|
|
|
|
assert( !tor->downloadedCur );
|
|
assert( !tor->uploadedCur );
|
|
|
|
tor->error = TR_OK;
|
|
tor->runStatus = flags & TR_FLAG_PAUSED ? TR_RUN_STOPPED : TR_RUN_RUNNING;
|
|
tor->recheckFlag = tr_ioCheckFiles( tor, TR_RECHECK_FAST );
|
|
tor->cpStatus = tr_cpGetStatus( tor->completion );
|
|
|
|
tr_sharedLock( h->shared );
|
|
tor->next = h->torrentList;
|
|
h->torrentList = tor;
|
|
h->torrentCount++;
|
|
tr_sharedUnlock( h->shared );
|
|
|
|
snprintf( name, sizeof( name ), "torrent %p (%s)", tor, tor->info.name );
|
|
tr_threadCreate( &tor->thread, torrentThreadLoop, tor, name );
|
|
}
|
|
|
|
static int
|
|
pathIsInUse ( const tr_handle_t * h,
|
|
const char * destination,
|
|
const char * name )
|
|
{
|
|
const tr_torrent_t * tor;
|
|
|
|
for( tor=h->torrentList; tor; tor=tor->next )
|
|
if( !strcmp( destination, tor->destination )
|
|
&& !strcmp( name, tor->info.name ) )
|
|
return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
static int
|
|
hashExists( const tr_handle_t * h,
|
|
const uint8_t * hash )
|
|
{
|
|
const tr_torrent_t * tor;
|
|
|
|
for( tor=h->torrentList; tor; tor=tor->next )
|
|
if( !memcmp( hash, tor->info.hash, SHA_DIGEST_LENGTH ) )
|
|
return TRUE;
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
static int
|
|
infoCanAdd( const tr_handle_t * h,
|
|
const char * destination,
|
|
const tr_info_t * info )
|
|
{
|
|
if( hashExists( h, info->hash ) )
|
|
return TR_EDUPLICATE;
|
|
|
|
if( destination && pathIsInUse( h, destination, info->name ) )
|
|
return TR_EDUPLICATE;
|
|
|
|
return TR_OK;
|
|
}
|
|
|
|
int
|
|
tr_torrentParse( const tr_handle_t * h,
|
|
const char * path,
|
|
const char * destination,
|
|
tr_info_t * setme_info )
|
|
{
|
|
int ret;
|
|
tr_info_t tmp;
|
|
|
|
if( setme_info == NULL )
|
|
setme_info = &tmp;
|
|
|
|
memset( setme_info, 0, sizeof( tr_info_t ) );
|
|
ret = tr_metainfoParseFile( setme_info, h->tag, path, FALSE );
|
|
|
|
if( ret == TR_OK )
|
|
ret = infoCanAdd( h, destination, setme_info );
|
|
|
|
if( setme_info == &tmp )
|
|
tr_metainfoFree( &tmp );
|
|
|
|
return ret;
|
|
}
|
|
|
|
tr_torrent_t *
|
|
tr_torrentInit( tr_handle_t * h,
|
|
const char * path,
|
|
const char * destination,
|
|
int flags,
|
|
int * error )
|
|
{
|
|
int val;
|
|
tr_torrent_t * tor = NULL;
|
|
|
|
if(( val = tr_torrentParse( h, path, destination, NULL )))
|
|
*error = val;
|
|
else if(!(( tor = tr_new0( tr_torrent_t, 1 ))))
|
|
*error = TR_EOTHER;
|
|
else {
|
|
tr_metainfoParseFile( &tor->info, h->tag, path, TR_FLAG_SAVE & flags );
|
|
torrentRealInit( h, tor, destination, flags );
|
|
}
|
|
|
|
return tor;
|
|
}
|
|
|
|
int
|
|
tr_torrentParseHash( const tr_handle_t * h,
|
|
const char * hashStr,
|
|
const char * destination,
|
|
tr_info_t * setme_info )
|
|
{
|
|
int ret;
|
|
tr_info_t tmp;
|
|
|
|
if( setme_info == NULL )
|
|
setme_info = &tmp;
|
|
|
|
memset( setme_info, 0, sizeof( tr_info_t ) );
|
|
ret = tr_metainfoParseHash( setme_info, h->tag, hashStr );
|
|
|
|
if( ret == TR_OK )
|
|
ret = infoCanAdd( h, destination, setme_info );
|
|
|
|
if( setme_info == &tmp )
|
|
tr_metainfoFree( &tmp );
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
tr_torrent_t *
|
|
tr_torrentInitSaved( tr_handle_t * h,
|
|
const char * hashStr,
|
|
const char * destination,
|
|
int flags,
|
|
int * error )
|
|
{
|
|
int val;
|
|
tr_torrent_t * tor = NULL;
|
|
|
|
if(( val = tr_torrentParseHash( h, hashStr, destination, NULL )))
|
|
*error = val;
|
|
else if(!(( tor = tr_new0( tr_torrent_t, 1 ))))
|
|
*error = TR_EOTHER;
|
|
else {
|
|
tr_metainfoParseHash( &tor->info, h->tag, hashStr );
|
|
torrentRealInit( h, tor, destination, (TR_FLAG_SAVE|flags) );
|
|
}
|
|
|
|
return tor;
|
|
}
|
|
|
|
static int
|
|
tr_torrentParseData( const tr_handle_t * h,
|
|
const uint8_t * data,
|
|
size_t size,
|
|
const char * destination,
|
|
tr_info_t * setme_info )
|
|
{
|
|
int ret;
|
|
tr_info_t tmp;
|
|
|
|
if( setme_info == NULL )
|
|
setme_info = &tmp;
|
|
|
|
memset( setme_info, 0, sizeof( tr_info_t ) );
|
|
ret = tr_metainfoParseData( setme_info, h->tag, data, size, FALSE );
|
|
|
|
if( ret == TR_OK )
|
|
ret = infoCanAdd( h, destination, setme_info );
|
|
|
|
if( setme_info == &tmp )
|
|
tr_metainfoFree( &tmp );
|
|
|
|
return ret;
|
|
}
|
|
|
|
tr_torrent_t *
|
|
tr_torrentInitData( tr_handle_t * h,
|
|
const uint8_t * data,
|
|
size_t size,
|
|
const char * destination,
|
|
int flags,
|
|
int * error )
|
|
{
|
|
int val;
|
|
tr_torrent_t * tor = NULL;
|
|
|
|
if(( val = tr_torrentParseData( h, data, size, destination, NULL )))
|
|
*error = val;
|
|
else if(!(( tor = tr_new0( tr_torrent_t, 1 ))))
|
|
*error = TR_EOTHER;
|
|
else {
|
|
tr_metainfoParseData( &tor->info, h->tag, data, size, TR_FLAG_SAVE & flags );
|
|
torrentRealInit( h, tor, destination, flags );
|
|
}
|
|
|
|
return tor;
|
|
}
|
|
|
|
const tr_info_t *
|
|
tr_torrentInfo( const tr_torrent_t * tor )
|
|
{
|
|
return &tor->info;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
int tr_torrentScrape( tr_torrent_t * tor, int * s, int * l, int * d )
|
|
{
|
|
return tr_trackerScrape( tor, s, l, d );
|
|
}
|
|
|
|
void tr_torrentSetFolder( tr_torrent_t * tor, const char * path )
|
|
{
|
|
tr_free( tor->destination );
|
|
tor->destination = tr_strdup( path );
|
|
|
|
if( !tor->ioLoaded )
|
|
tor->ioLoaded = tr_ioLoadResume( tor ) == TR_OK;
|
|
}
|
|
|
|
const char* tr_torrentGetFolder( const tr_torrent_t * tor )
|
|
{
|
|
return tor->destination;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* torrentReallyStop
|
|
***********************************************************************
|
|
* Joins the download thread and frees/closes everything related to it.
|
|
**********************************************************************/
|
|
|
|
void tr_torrentDisablePex( tr_torrent_t * tor, int disable )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
|
|
if( ! ( TR_FLAG_PRIVATE & tor->info.flags ) )
|
|
{
|
|
if( tor->pexDisabled != disable )
|
|
{
|
|
int i;
|
|
tor->pexDisabled = disable;
|
|
for( i=0; i<tor->peerCount; ++i )
|
|
tr_peerSetPrivate( tor->peers[i], disable );
|
|
}
|
|
}
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
static int tr_didStateChangeTo ( tr_torrent_t * tor, int status )
|
|
{
|
|
int ret;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
if (( ret = tor->hasChangedState == status ))
|
|
tor->hasChangedState = -1;
|
|
tr_torrentWriterUnlock( tor );
|
|
|
|
return ret;
|
|
}
|
|
|
|
int tr_getIncomplete( tr_torrent_t * tor )
|
|
{
|
|
return tr_didStateChangeTo( tor, TR_CP_INCOMPLETE );
|
|
}
|
|
int tr_getDone( tr_torrent_t * tor )
|
|
{
|
|
return tr_didStateChangeTo( tor, TR_CP_DONE );
|
|
}
|
|
int tr_getComplete( tr_torrent_t * tor )
|
|
{
|
|
return tr_didStateChangeTo( tor, TR_CP_COMPLETE );
|
|
}
|
|
|
|
void tr_manualUpdate( tr_torrent_t * tor UNUSED )
|
|
{
|
|
#if 0
|
|
int peerCount, new;
|
|
uint8_t * peerCompact;
|
|
|
|
if( tor->status != TR_RUN_RUNNING )
|
|
return;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
tr_trackerAnnouncePulse( tor->tracker, &peerCount, &peerCompact, 1 );
|
|
new = 0;
|
|
if( peerCount > 0 )
|
|
{
|
|
new = tr_torrentAddCompact( tor, TR_PEER_FROM_TRACKER,
|
|
peerCompact, peerCount );
|
|
free( peerCompact );
|
|
}
|
|
tr_dbg( "got %i peers from manual announce, used %i", peerCount, new );
|
|
tr_torrentWriterUnlock( tor );
|
|
#endif
|
|
}
|
|
|
|
tr_stat_t * tr_torrentStat( tr_torrent_t * tor )
|
|
{
|
|
tr_stat_t * s;
|
|
tr_tracker_t * tc;
|
|
int i;
|
|
|
|
tr_torrentReaderLock( tor );
|
|
|
|
tor->statCur = ( tor->statCur + 1 ) % 2;
|
|
s = &tor->stats[tor->statCur];
|
|
|
|
s->error = tor->error;
|
|
memcpy( s->errorString, tor->errorString,
|
|
sizeof( s->errorString ) );
|
|
|
|
tc = tor->tracker;
|
|
s->cannotConnect = tr_trackerCannotConnect( tc );
|
|
s->tracker = tc
|
|
? tr_trackerGet( tc )
|
|
: &tor->info.trackerList[0].list[0];
|
|
|
|
/* peers... */
|
|
memset( s->peersFrom, 0, sizeof( s->peersFrom ) );
|
|
s->peersTotal = 0;
|
|
s->peersUploading = 0;
|
|
s->peersDownloading = 0;
|
|
for( i=0; i<tor->peerCount; ++i ) {
|
|
const tr_peer_t * peer = tor->peers[i];
|
|
++s->peersTotal;
|
|
if( tr_peerIsConnected( peer ) ) {
|
|
++s->peersFrom[tr_peerIsFrom(peer)];
|
|
if( tr_peerIsInterested( peer ) && !tr_peerIsChokedByUs( peer ) )
|
|
++s->peersUploading;
|
|
if( tr_peerIsInteresting( peer ) && !tr_peerIsChokingUs( peer ) )
|
|
++s->peersDownloading;
|
|
}
|
|
}
|
|
|
|
s->percentDone = tr_cpPercentDone( tor->completion );
|
|
s->percentComplete = tr_cpPercentComplete( tor->completion );
|
|
s->left = tr_cpLeftUntilDone( tor->completion );
|
|
|
|
|
|
if( tor->recheckFlag )
|
|
s->status = TR_STATUS_CHECK_WAIT;
|
|
else switch( tor->runStatus ) {
|
|
case TR_RUN_STOPPING: /* fallthrough */
|
|
case TR_RUN_STOPPING_NET_WAIT: s->status = TR_STATUS_STOPPING; break;
|
|
case TR_RUN_STOPPED: s->status = TR_STATUS_STOPPED; break;
|
|
case TR_RUN_CHECKING: s->status = TR_STATUS_CHECK; break;
|
|
case TR_RUN_RUNNING: switch( tor->cpStatus ) {
|
|
case TR_CP_INCOMPLETE: s->status = TR_STATUS_DOWNLOAD; break;
|
|
case TR_CP_DONE: s->status = TR_STATUS_DONE; break;
|
|
case TR_CP_COMPLETE: s->status = TR_STATUS_SEED; break;
|
|
}
|
|
}
|
|
|
|
s->cpStatus = tor->cpStatus;
|
|
|
|
/* tr_rcRate() doesn't make the difference between 'piece'
|
|
messages and other messages, which causes a non-zero
|
|
download rate even tough we are not downloading. So we
|
|
force it to zero not to confuse the user. */
|
|
s->rateDownload = tor->runStatus==TR_RUN_RUNNING
|
|
? tr_rcRate( tor->download )
|
|
: 0.0;
|
|
s->rateUpload = tr_rcRate( tor->upload );
|
|
|
|
s->seeders = tr_trackerSeeders( tc );
|
|
s->leechers = tr_trackerLeechers( tc );
|
|
s->completedFromTracker = tr_trackerDownloaded( tc );
|
|
|
|
s->swarmspeed = tr_rcRate( tor->swarmspeed );
|
|
|
|
s->startDate = tor->startDate;
|
|
s->activityDate = tor->activityDate;
|
|
|
|
s->eta = s->rateDownload < 0.1
|
|
? -1.0f
|
|
: (s->left / s->rateDownload / 1024.0);
|
|
|
|
s->uploaded = tor->uploadedCur + tor->uploadedPrev;
|
|
s->downloaded = tor->downloadedCur + tor->downloadedPrev;
|
|
s->downloadedValid = tr_cpDownloadedValid( tor->completion );
|
|
|
|
s->ratio = s->downloaded || s->downloadedValid
|
|
? (float)s->uploaded / (float)MAX(s->downloaded, s->downloadedValid)
|
|
: TR_RATIO_NA;
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
|
|
return s;
|
|
}
|
|
|
|
tr_peer_stat_t *
|
|
tr_torrentPeers( const tr_torrent_t * tor, int * peerCount )
|
|
{
|
|
tr_peer_stat_t * peers;
|
|
|
|
tr_torrentReaderLock( tor );
|
|
|
|
*peerCount = tor->peerCount;
|
|
|
|
peers = tr_new0( tr_peer_stat_t, tor->peerCount );
|
|
if (peers != NULL)
|
|
{
|
|
tr_peer_t * peer;
|
|
struct in_addr * addr;
|
|
int i;
|
|
for( i=0; i<tor->peerCount; ++i )
|
|
{
|
|
peer = tor->peers[i];
|
|
|
|
addr = tr_peerAddress( peer );
|
|
if( NULL != addr )
|
|
{
|
|
tr_netNtop( addr, peers[i].addr,
|
|
sizeof( peers[i].addr ) );
|
|
}
|
|
|
|
peers[i].client = tr_peerClient( peer );
|
|
peers[i].isConnected = tr_peerIsConnected( peer );
|
|
peers[i].from = tr_peerIsFrom( peer );
|
|
peers[i].progress = tr_peerProgress( peer );
|
|
peers[i].port = tr_peerPort( peer );
|
|
peers[i].isDownloading = !tr_peerIsChokingUs( peer );
|
|
peers[i].uploadToRate = tr_peerUploadRate( peer );
|
|
peers[i].isUploading = !tr_peerIsChokedByUs( peer );
|
|
peers[i].downloadFromRate = tr_peerDownloadRate( peer );
|
|
}
|
|
}
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
|
|
return peers;
|
|
}
|
|
|
|
void tr_torrentPeersFree( tr_peer_stat_t * peers, int peerCount UNUSED )
|
|
{
|
|
tr_free( peers );
|
|
}
|
|
|
|
void tr_torrentAvailability( const tr_torrent_t * tor, int8_t * tab, int size )
|
|
{
|
|
int i, j, piece;
|
|
float interval;
|
|
|
|
tr_torrentReaderLock( tor );
|
|
|
|
interval = (float)tor->info.pieceCount / (float)size;
|
|
for( i = 0; i < size; i++ )
|
|
{
|
|
piece = i * interval;
|
|
|
|
if( tr_cpPieceIsComplete( tor->completion, piece ) )
|
|
{
|
|
tab[i] = -1;
|
|
continue;
|
|
}
|
|
|
|
tab[i] = 0;
|
|
for( j = 0; j < tor->peerCount; j++ )
|
|
{
|
|
if( tr_peerHasPiece( tor->peers[j], piece ) )
|
|
{
|
|
(tab[i])++;
|
|
}
|
|
}
|
|
}
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
}
|
|
|
|
uint64_t
|
|
tr_torrentFileBytesCompleted ( const tr_torrent_t * tor, int fileIndex )
|
|
{
|
|
const tr_file_t * file = &tor->info.files[fileIndex];
|
|
const uint64_t firstBlock = file->offset / tor->blockSize;
|
|
const uint64_t firstBlockOffset = file->offset % tor->blockSize;
|
|
const uint64_t lastOffset = file->length ? (file->length-1) : 0;
|
|
const uint64_t lastBlock = (file->offset + lastOffset) / tor->blockSize;
|
|
const uint64_t lastBlockOffset = (file->offset + lastOffset) % tor->blockSize;
|
|
uint64_t haveBytes = 0;
|
|
|
|
assert( tor != NULL );
|
|
assert( 0<=fileIndex && fileIndex<tor->info.fileCount );
|
|
assert( file->offset + file->length <= tor->info.totalSize );
|
|
assert( (int)firstBlock < tor->blockCount );
|
|
assert( (int)lastBlock < tor->blockCount );
|
|
assert( firstBlock <= lastBlock );
|
|
assert( tr_blockPiece( firstBlock ) == file->firstPiece );
|
|
assert( tr_blockPiece( lastBlock ) == file->lastPiece );
|
|
|
|
if( firstBlock == lastBlock )
|
|
{
|
|
if( tr_cpBlockIsComplete( tor->completion, firstBlock ) )
|
|
haveBytes += lastBlockOffset + 1 - firstBlockOffset;
|
|
}
|
|
else
|
|
{
|
|
uint64_t i;
|
|
|
|
if( tr_cpBlockIsComplete( tor->completion, firstBlock ) )
|
|
haveBytes += tor->blockSize - firstBlockOffset;
|
|
|
|
for( i=firstBlock+1; i<lastBlock; ++i )
|
|
if( tr_cpBlockIsComplete( tor->completion, i ) )
|
|
haveBytes += tor->blockSize;
|
|
|
|
if( tr_cpBlockIsComplete( tor->completion, lastBlock ) )
|
|
haveBytes += lastBlockOffset + 1;
|
|
}
|
|
|
|
return haveBytes;
|
|
}
|
|
|
|
float
|
|
tr_torrentFileCompletion ( const tr_torrent_t * tor, int fileIndex )
|
|
{
|
|
const uint64_t c = tr_torrentFileBytesCompleted ( tor, fileIndex );
|
|
uint64_t length = tor->info.files[fileIndex].length;
|
|
|
|
if( !length )
|
|
return 1.0;
|
|
return (double)c / length;
|
|
}
|
|
|
|
float*
|
|
tr_torrentCompletion( const tr_torrent_t * tor )
|
|
{
|
|
int i;
|
|
float * f;
|
|
tr_torrentReaderLock( tor );
|
|
|
|
f = tr_new0( float, tor->info.fileCount );
|
|
for( i=0; i<tor->info.fileCount; ++i )
|
|
f[i] = tr_torrentFileCompletion ( tor, i );
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
return f;
|
|
}
|
|
|
|
void tr_torrentAmountFinished( const tr_torrent_t * tor, float * tab, int size )
|
|
{
|
|
int i;
|
|
float interval;
|
|
tr_torrentReaderLock( tor );
|
|
|
|
interval = (float)tor->info.pieceCount / (float)size;
|
|
for( i = 0; i < size; i++ )
|
|
{
|
|
int piece = i * interval;
|
|
tab[i] = tr_cpPercentBlocksInPiece( tor->completion, piece );
|
|
}
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_torrentResetTransferStats( tr_torrent_t * tor )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
|
|
tor->downloadedPrev += tor->downloadedCur;
|
|
tor->downloadedCur = 0;
|
|
tor->uploadedPrev += tor->uploadedCur;
|
|
tor->uploadedCur = 0;
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
|
|
void
|
|
tr_torrentSetHasPiece( tr_torrent_t * tor, int pieceIndex, int has )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
|
|
if( has )
|
|
tr_cpPieceAdd( tor->completion, pieceIndex );
|
|
else
|
|
tr_cpPieceRem( tor->completion, pieceIndex );
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void tr_torrentRemoveSaved( tr_torrent_t * tor )
|
|
{
|
|
tr_metainfoRemoveSaved( tor->info.hashString, tor->handle->tag );
|
|
}
|
|
|
|
void tr_torrentRecheck( tr_torrent_t * tor )
|
|
{
|
|
tor->recheckFlag = TRUE;
|
|
}
|
|
|
|
|
|
int tr_torrentAttachPeer( tr_torrent_t * tor, tr_peer_t * peer )
|
|
{
|
|
int i;
|
|
tr_peer_t * otherPeer;
|
|
|
|
assert( tor != NULL );
|
|
assert( peer != NULL );
|
|
|
|
if( tor->peerCount >= TR_MAX_PEER_COUNT )
|
|
{
|
|
tr_peerDestroy( peer );
|
|
return 0;
|
|
}
|
|
|
|
/* Don't accept two connections from the same IP */
|
|
for( i = 0; i < tor->peerCount; i++ )
|
|
{
|
|
otherPeer = tor->peers[i];
|
|
if( !memcmp( tr_peerAddress( peer ), tr_peerAddress( otherPeer ), 4 ) )
|
|
{
|
|
tr_peerDestroy( peer );
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
tr_peerSetPrivate( peer, tor->info.flags & TR_FLAG_PRIVATE ||
|
|
tor->pexDisabled );
|
|
tr_peerSetTorrent( peer, tor );
|
|
tor->peers[tor->peerCount++] = peer;
|
|
|
|
return 1;
|
|
}
|
|
|
|
int tr_torrentAddCompact( tr_torrent_t * tor, int from,
|
|
uint8_t * buf, int count )
|
|
{
|
|
struct in_addr addr;
|
|
in_port_t port;
|
|
int i, added;
|
|
tr_peer_t * peer;
|
|
|
|
added = 0;
|
|
for( i = 0; i < count; i++ )
|
|
{
|
|
memcpy( &addr, buf, 4 ); buf += 4;
|
|
memcpy( &port, buf, 2 ); buf += 2;
|
|
|
|
peer = tr_peerInit( &addr, port, -1, from );
|
|
added += tr_torrentAttachPeer( tor, peer );
|
|
}
|
|
|
|
return added;
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static void setRunState( tr_torrent_t * tor, run_status_t run )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tor->runStatus = run;
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void tr_torrentStart( tr_torrent_t * tor )
|
|
{
|
|
setRunState( tor, TR_RUN_RUNNING );
|
|
}
|
|
|
|
void tr_torrentStop( tr_torrent_t * tor )
|
|
{
|
|
setRunState( tor, TR_RUN_STOPPING );
|
|
}
|
|
|
|
void tr_torrentClose( tr_torrent_t * tor )
|
|
{
|
|
tr_torrentStop( tor );
|
|
tor->dieFlag = TRUE;
|
|
}
|
|
|
|
static void
|
|
tr_torrentFree( tr_torrent_t * tor )
|
|
{
|
|
tr_torrent_t * t;
|
|
tr_handle_t * h = tor->handle;
|
|
tr_info_t * inf = &tor->info;
|
|
|
|
tr_sharedLock( h->shared );
|
|
|
|
tr_rwClose( &tor->lock );
|
|
tr_cpClose( tor->completion );
|
|
|
|
tr_rcClose( tor->upload );
|
|
tr_rcClose( tor->download );
|
|
tr_rcClose( tor->swarmspeed );
|
|
|
|
tr_free( tor->destination );
|
|
|
|
tr_metainfoFree( inf );
|
|
|
|
if( tor == h->torrentList )
|
|
h->torrentList = tor->next;
|
|
else for( t=h->torrentList; t!=NULL; t=t->next ) {
|
|
if( t->next == tor ) {
|
|
t->next = tor->next;
|
|
break;
|
|
}
|
|
}
|
|
|
|
tr_free( tor );
|
|
|
|
h->torrentCount--;
|
|
|
|
tr_sharedUnlock( h->shared );
|
|
}
|
|
|
|
static void
|
|
torrentThreadLoop ( void * _tor )
|
|
{
|
|
static tr_lock_t checkFilesLock;
|
|
static int checkFilesLockInited = FALSE;
|
|
tr_torrent_t * tor = _tor;
|
|
|
|
/* create the check-files mutex */
|
|
if( !checkFilesLockInited ) {
|
|
checkFilesLockInited = TRUE;
|
|
tr_lockInit( &checkFilesLock );
|
|
}
|
|
|
|
/* loop until the torrent is being deleted */
|
|
while( ! ( tor->dieFlag && (tor->runStatus == TR_RUN_STOPPED) ) )
|
|
{
|
|
cp_status_t cpStatus;
|
|
|
|
/* sleep a little while */
|
|
tr_wait( tor->runStatus == TR_RUN_STOPPED ? 1600 : 600 );
|
|
|
|
/* if we're stopping... */
|
|
if( tor->runStatus == TR_RUN_STOPPING )
|
|
{
|
|
int i;
|
|
int peerCount;
|
|
uint8_t * peerCompact;
|
|
tr_torrentWriterLock( tor );
|
|
|
|
/* close the IO */
|
|
tr_ioClose( tor->io );
|
|
tor->io = NULL;
|
|
|
|
/* close the peers */
|
|
for( i=0; i<tor->peerCount; ++i )
|
|
tr_peerDestroy( tor->peers[i] );
|
|
tor->peerCount = 0;
|
|
|
|
/* resest the transfer rates */
|
|
tr_rcReset( tor->download );
|
|
tr_rcReset( tor->upload );
|
|
tr_rcReset( tor->swarmspeed );
|
|
|
|
/* tell the tracker we're stopping */
|
|
tr_trackerStopped( tor->tracker );
|
|
tr_trackerPulse( tor->tracker, &peerCount, &peerCompact );
|
|
tor->runStatus = TR_RUN_STOPPING_NET_WAIT;
|
|
tor->stopDate = tr_date();
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
if( tor->runStatus == TR_RUN_STOPPING_NET_WAIT )
|
|
{
|
|
uint64_t date;
|
|
int peerCount;
|
|
uint8_t * peerCompact;
|
|
tr_trackerPulse( tor->tracker, &peerCount, &peerCompact );
|
|
|
|
/* have we finished telling the tracker that we're stopping? */
|
|
date = tr_trackerLastResponseDate( tor->tracker );
|
|
if( date > tor->stopDate )
|
|
{
|
|
tr_torrentWriterLock( tor );
|
|
tr_trackerClose( tor->tracker );
|
|
tor->tracker = NULL;
|
|
tor->runStatus = TR_RUN_STOPPED;
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
continue;
|
|
}
|
|
|
|
/* do we need to check files? */
|
|
if( tor->recheckFlag )
|
|
{
|
|
if( !tr_lockTryLock( &checkFilesLock ) )
|
|
{
|
|
run_status_t realStatus;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
realStatus = tor->runStatus;
|
|
tor->recheckFlag = FALSE;
|
|
tor->runStatus = TR_RUN_CHECKING;
|
|
tr_torrentWriterUnlock( tor );
|
|
|
|
tr_ioCheckFiles( tor, TR_RECHECK_FORCE );
|
|
setRunState( tor, realStatus );
|
|
|
|
tr_lockUnlock( &checkFilesLock );
|
|
}
|
|
continue;
|
|
}
|
|
|
|
/* if we're paused or stopped, not much to do... */
|
|
if( tor->runStatus == TR_RUN_STOPPED )
|
|
continue;
|
|
|
|
/* ping our peers if we're running... */
|
|
if( tor->runStatus == TR_RUN_RUNNING )
|
|
{
|
|
int i;
|
|
int peerCount;
|
|
uint8_t * peerCompact;
|
|
|
|
/* starting to run... */
|
|
if( tor->io == NULL ) {
|
|
*tor->errorString = '\0';
|
|
tr_torrentResetTransferStats( tor );
|
|
tor->io = tr_ioInitFast( tor );
|
|
if( tor->io == NULL ) {
|
|
tor->recheckFlag = TRUE;
|
|
continue;
|
|
}
|
|
tor->tracker = tr_trackerInit( tor );
|
|
tor->startDate = tr_date();
|
|
}
|
|
|
|
/* refresh our completion state */
|
|
tr_torrentWriterLock( tor );
|
|
cpStatus = tr_cpGetStatus( tor->completion );
|
|
if( cpStatus != tor->cpStatus ) {
|
|
tor->cpStatus = cpStatus;
|
|
if( (cpStatus == TR_CP_COMPLETE) /* if we're complete */
|
|
&& tor->tracker!=NULL /* and we have a tracker */
|
|
&& tor->downloadedCur ) { /* and it just happened */
|
|
tr_trackerCompleted( tor->tracker ); /* tell the tracker */
|
|
tor->hasChangedState = tor->cpStatus; /* and the client */
|
|
}
|
|
tr_ioSync( tor->io );
|
|
}
|
|
tr_torrentWriterUnlock( tor );
|
|
|
|
/* ping the tracker... */
|
|
tr_trackerPulse( tor->tracker, &peerCount, &peerCompact );
|
|
if( peerCount > 0 ) {
|
|
int used = tr_torrentAddCompact( tor, TR_PEER_FROM_TRACKER,
|
|
peerCompact, peerCount );
|
|
tr_dbg( "got %i peers from announce, used %i", peerCount, used );
|
|
free( peerCompact );
|
|
}
|
|
|
|
/* Shuffle peers */
|
|
if ( tor->peerCount > 1 ) {
|
|
tr_peer_t * tmp = tor->peers[0];
|
|
memmove( tor->peers, tor->peers+1,
|
|
(tor->peerCount-1) * sizeof(void*) );
|
|
tor->peers[tor->peerCount - 1] = tmp;
|
|
}
|
|
|
|
/* receive/send messages */
|
|
tr_torrentWriterLock( tor );
|
|
for( i=0; i<tor->peerCount; ) {
|
|
tr_peer_t * peer = tor->peers[i];
|
|
int ret = tr_peerPulse( peer );
|
|
if( ret & TR_ERROR_IO_MASK ) {
|
|
tr_err( "Fatal error, stopping download (%d)", ret );
|
|
tor->runStatus = TR_RUN_STOPPING;
|
|
tor->error = ret;
|
|
strlcpy( tor->errorString,
|
|
tr_errorString(ret),
|
|
sizeof(tor->errorString) );
|
|
break;
|
|
}
|
|
if( ret ) {
|
|
tr_peerDestroy( peer );
|
|
tor->peerCount--;
|
|
memmove( &tor->peers[i], &tor->peers[i+1],
|
|
(tor->peerCount-i)*sizeof(void*) );
|
|
continue;
|
|
}
|
|
i++;
|
|
}
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
}
|
|
|
|
tr_ioClose( tor->io );
|
|
tr_torrentFree( tor );
|
|
}
|
|
|
|
|
|
/***
|
|
****
|
|
**** File prioritization
|
|
****
|
|
***/
|
|
|
|
static void
|
|
tr_torrentSetFilePriorityImpl( tr_torrent_t * tor,
|
|
int fileIndex,
|
|
tr_priority_t priority,
|
|
int doSave )
|
|
{
|
|
int i;
|
|
tr_file_t * file;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
|
|
assert( tor != NULL );
|
|
assert( 0<=fileIndex && fileIndex<tor->info.fileCount );
|
|
assert( priority==TR_PRI_LOW || priority==TR_PRI_NORMAL || priority==TR_PRI_HIGH );
|
|
|
|
file = &tor->info.files[fileIndex];
|
|
file->priority = priority;
|
|
for( i=file->firstPiece; i<=file->lastPiece; ++i )
|
|
tor->info.pieces[i].priority = calculatePiecePriority( tor, i );
|
|
|
|
tr_dbg ( "Setting file #%d (pieces %d-%d) priority to %d (%s)",
|
|
fileIndex, file->firstPiece, file->lastPiece,
|
|
priority, tor->info.files[fileIndex].name );
|
|
|
|
if( doSave )
|
|
fastResumeSave( tor );
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_torrentSetFilePriority( tr_torrent_t * tor,
|
|
int fileIndex,
|
|
tr_priority_t priority )
|
|
{
|
|
tr_torrentSetFilePriorityImpl( tor, fileIndex, priority, TRUE );
|
|
}
|
|
|
|
void
|
|
tr_torrentSetFilePriorities( tr_torrent_t * tor,
|
|
int * files,
|
|
int fileCount,
|
|
tr_priority_t priority )
|
|
{
|
|
int i;
|
|
for( i=0; i<fileCount; ++i ) {
|
|
const int fileIndex = files[i];
|
|
tr_torrentSetFilePriorityImpl( tor, fileIndex, priority, FALSE );
|
|
}
|
|
fastResumeSave( tor );
|
|
}
|
|
|
|
tr_priority_t
|
|
tr_torrentGetFilePriority( const tr_torrent_t * tor, int file )
|
|
{
|
|
tr_priority_t ret;
|
|
|
|
tr_torrentReaderLock( tor );
|
|
assert( tor != NULL );
|
|
assert( 0<=file && file<tor->info.fileCount );
|
|
ret = tor->info.files[file].priority;
|
|
tr_torrentReaderUnlock( tor );
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
tr_priority_t*
|
|
tr_torrentGetFilePriorities( const tr_torrent_t * tor )
|
|
{
|
|
int i;
|
|
tr_priority_t * p;
|
|
|
|
tr_torrentReaderLock( tor );
|
|
p = tr_new0( tr_priority_t, tor->info.fileCount );
|
|
for( i=0; i<tor->info.fileCount; ++i )
|
|
p[i] = tor->info.files[i].priority;
|
|
tr_torrentReaderUnlock( tor );
|
|
|
|
return p;
|
|
}
|
|
|
|
int
|
|
tr_torrentGetFileDL( const tr_torrent_t * tor,
|
|
int file )
|
|
{
|
|
int do_download;
|
|
tr_torrentReaderLock( tor );
|
|
|
|
assert( 0<=file && file<tor->info.fileCount );
|
|
do_download = !tor->info.files[file].dnd;
|
|
|
|
tr_torrentReaderUnlock( tor );
|
|
return do_download != 0;
|
|
}
|
|
|
|
void
|
|
tr_torrentSetFileDL( tr_torrent_t * tor,
|
|
int fileIndex,
|
|
int do_download )
|
|
{
|
|
int i;
|
|
tr_file_t * file;
|
|
const int dnd = !do_download;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
|
|
assert( 0<=fileIndex && fileIndex<tor->info.fileCount );
|
|
file = &tor->info.files[fileIndex];
|
|
file->dnd = dnd;
|
|
for( i=file->firstPiece; i<=file->lastPiece; ++i )
|
|
tor->info.pieces[i].dnd = dnd;
|
|
fastResumeSave( tor );
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|
|
|
|
void
|
|
tr_torrentSetFileDLs ( tr_torrent_t * tor,
|
|
int * files,
|
|
int fileCount,
|
|
int do_download )
|
|
{
|
|
int i, j;
|
|
const int dnd = !do_download;
|
|
|
|
tr_torrentWriterLock( tor );
|
|
|
|
for( i=0; i<fileCount; ++i ) {
|
|
const int fileIndex = files[i];
|
|
tr_file_t * file = &tor->info.files[fileIndex];
|
|
file->dnd = dnd;
|
|
for( j=file->firstPiece; j<=file->lastPiece; ++j )
|
|
tor->info.pieces[j].dnd = dnd;
|
|
}
|
|
|
|
fastResumeSave( tor );
|
|
|
|
tr_torrentWriterUnlock( tor );
|
|
}
|