610 lines
16 KiB
C
610 lines
16 KiB
C
/******************************************************************************
|
|
* $Id$
|
|
*
|
|
* Copyright (c) 2006-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 <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include <gtk/gtk.h>
|
|
#include <glib/gi18n.h>
|
|
|
|
#include <libtransmission/transmission.h>
|
|
#include <libtransmission/bencode.h>
|
|
|
|
#include "tr_prefs.h"
|
|
#include "tr_torrent.h"
|
|
#include "conf.h"
|
|
#include "util.h"
|
|
|
|
enum {
|
|
TR_TORRENT_HANDLE = 1,
|
|
TR_TORRENT_DIR,
|
|
};
|
|
|
|
static void
|
|
tr_torrent_init(GTypeInstance *instance, gpointer g_class);
|
|
static void
|
|
tr_torrent_set_property(GObject *object, guint property_id,
|
|
const GValue *value, GParamSpec *pspec);
|
|
static void
|
|
tr_torrent_get_property(GObject *object, guint property_id,
|
|
GValue *value, GParamSpec *pspec);
|
|
static void
|
|
tr_torrent_class_init(gpointer g_class, gpointer g_class_data);
|
|
static void
|
|
tr_torrent_dispose(GObject *obj);
|
|
static void
|
|
tr_torrent_set_folder(TrTorrent *tor);
|
|
|
|
static gpointer
|
|
tracker_boxed_fake_copy( gpointer boxed )
|
|
{
|
|
return boxed;
|
|
}
|
|
|
|
static void
|
|
tracker_boxed_fake_free( gpointer boxed SHUTUP )
|
|
{
|
|
}
|
|
|
|
GType
|
|
tr_tracker_boxed_get_type( void )
|
|
{
|
|
static GType type = 0;
|
|
|
|
if( 0 == type )
|
|
{
|
|
type = g_boxed_type_register_static( "TrTrackerBoxed",
|
|
tracker_boxed_fake_copy,
|
|
tracker_boxed_fake_free );
|
|
}
|
|
|
|
return type;
|
|
}
|
|
|
|
GType
|
|
tr_torrent_get_type(void) {
|
|
static GType type = 0;
|
|
|
|
if(0 == type) {
|
|
static const GTypeInfo info = {
|
|
sizeof (TrTorrentClass),
|
|
NULL, /* base_init */
|
|
NULL, /* base_finalize */
|
|
tr_torrent_class_init, /* class_init */
|
|
NULL, /* class_finalize */
|
|
NULL, /* class_data */
|
|
sizeof (TrTorrent),
|
|
0, /* n_preallocs */
|
|
tr_torrent_init, /* instance_init */
|
|
NULL,
|
|
};
|
|
type = g_type_register_static(G_TYPE_OBJECT, "TrTorrent", &info, 0);
|
|
}
|
|
return type;
|
|
}
|
|
|
|
static void
|
|
tr_torrent_class_init(gpointer g_class, gpointer g_class_data SHUTUP) {
|
|
GObjectClass *gobject_class = G_OBJECT_CLASS(g_class);
|
|
GParamSpec *pspec;
|
|
|
|
gobject_class->set_property = tr_torrent_set_property;
|
|
gobject_class->get_property = tr_torrent_get_property;
|
|
gobject_class->dispose = tr_torrent_dispose;
|
|
|
|
pspec = g_param_spec_pointer("torrent-handle", "Torrent handle",
|
|
"Torrent handle from libtransmission",
|
|
G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE);
|
|
g_object_class_install_property(gobject_class, TR_TORRENT_HANDLE, pspec);
|
|
|
|
pspec = g_param_spec_string("download-directory", "Download directory",
|
|
"Directory to download files to", NULL,
|
|
G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE);
|
|
g_object_class_install_property(gobject_class, TR_TORRENT_DIR, pspec);
|
|
}
|
|
|
|
static void
|
|
tr_torrent_init(GTypeInstance *instance, gpointer g_class SHUTUP) {
|
|
TrTorrent *self = (TrTorrent *)instance;
|
|
|
|
#ifdef REFDBG
|
|
fprintf( stderr, "torrent %p init\n", self );
|
|
#endif
|
|
|
|
self->handle = NULL;
|
|
self->lastStatTime = 0;
|
|
self->dir = NULL;
|
|
self->delfile = NULL;
|
|
self->severed = FALSE;
|
|
self->disposed = FALSE;
|
|
}
|
|
|
|
static void
|
|
tr_torrent_set_property(GObject *object, guint property_id,
|
|
const GValue *value, GParamSpec *pspec) {
|
|
TrTorrent *self = (TrTorrent*)object;
|
|
|
|
if(self->severed)
|
|
return;
|
|
|
|
switch(property_id) {
|
|
case TR_TORRENT_HANDLE:
|
|
g_assert(NULL == self->handle);
|
|
self->handle = g_value_get_pointer(value);
|
|
if(NULL != self->handle && NULL != self->dir)
|
|
tr_torrent_set_folder(self);
|
|
break;
|
|
case TR_TORRENT_DIR:
|
|
g_assert(NULL == self->dir);
|
|
self->dir = g_value_dup_string(value);
|
|
if(NULL != self->handle && NULL != self->dir)
|
|
tr_torrent_set_folder(self);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
tr_torrent_get_property(GObject *object, guint property_id,
|
|
GValue *value, GParamSpec *pspec) {
|
|
TrTorrent *self = (TrTorrent*)object;
|
|
|
|
if(self->severed)
|
|
return;
|
|
|
|
switch(property_id) {
|
|
case TR_TORRENT_HANDLE:
|
|
g_value_set_pointer(value, self->handle);
|
|
break;
|
|
case TR_TORRENT_DIR:
|
|
g_value_set_string(value, (NULL != self->dir ? self->dir :
|
|
tr_torrentGetFolder(self->handle)));
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
tr_torrent_dispose(GObject *obj) {
|
|
GObjectClass *parent = g_type_class_peek(g_type_parent(TR_TORRENT_TYPE));
|
|
TrTorrent *self = (TrTorrent*)obj;
|
|
|
|
if(self->disposed)
|
|
return;
|
|
self->disposed = TRUE;
|
|
|
|
#ifdef REFDBG
|
|
fprintf( stderr, "torrent %p dispose\n", self );
|
|
#endif
|
|
|
|
if( !self->severed )
|
|
tr_torrent_sever( self );
|
|
|
|
g_free (self->delfile);
|
|
g_free (self->dir);
|
|
|
|
/* Chain up to the parent class */
|
|
parent->dispose(obj);
|
|
}
|
|
|
|
void
|
|
tr_torrent_sever( TrTorrent * self )
|
|
{
|
|
g_return_if_fail (TR_IS_TORRENT( self ));
|
|
|
|
if( !self->severed )
|
|
{
|
|
self->severed = TRUE;
|
|
|
|
if( self->handle )
|
|
tr_torrentClose( self->handle );
|
|
}
|
|
}
|
|
|
|
tr_torrent *
|
|
tr_torrent_handle(TrTorrent *tor)
|
|
{
|
|
g_assert( TR_IS_TORRENT(tor) );
|
|
|
|
return tor->severed ? NULL : tor->handle;
|
|
}
|
|
|
|
static tr_stat*
|
|
refreshStat( TrTorrent * tor )
|
|
{
|
|
tor->lastStatTime= time( NULL );
|
|
tor->stat = *tr_torrentStat( tor->handle );
|
|
return &tor->stat;
|
|
}
|
|
|
|
const tr_stat *
|
|
tr_torrent_stat(TrTorrent *tor)
|
|
{
|
|
g_assert( TR_IS_TORRENT(tor) );
|
|
|
|
if( !tor->severed && tor->lastStatTime!=time(NULL) )
|
|
refreshStat( tor );
|
|
|
|
return &tor->stat;
|
|
}
|
|
|
|
const tr_info *
|
|
tr_torrent_info(TrTorrent *tor) {
|
|
TR_IS_TORRENT(tor);
|
|
|
|
if(tor->severed)
|
|
return NULL;
|
|
|
|
return tr_torrentInfo(tor->handle);
|
|
}
|
|
|
|
void
|
|
tr_torrent_start( TrTorrent * self )
|
|
{
|
|
TR_IS_TORRENT( self );
|
|
|
|
if( !self->severed )
|
|
tr_torrentStart( self->handle );
|
|
}
|
|
|
|
void
|
|
tr_torrent_stop( TrTorrent * self )
|
|
{
|
|
TR_IS_TORRENT( self );
|
|
|
|
if( !self->severed )
|
|
tr_torrentStop( self->handle );
|
|
}
|
|
|
|
static TrTorrent *
|
|
maketorrent( tr_torrent * handle )
|
|
{
|
|
tr_torrentDisablePex( handle, !pref_flag_get( PREF_KEY_PEX ) );
|
|
|
|
return g_object_new( TR_TORRENT_TYPE,
|
|
"torrent-handle", handle,
|
|
NULL);
|
|
}
|
|
|
|
TrTorrent*
|
|
tr_torrent_new_preexisting( tr_torrent * tor )
|
|
{
|
|
return maketorrent( tor );
|
|
}
|
|
|
|
|
|
TrTorrent *
|
|
tr_torrent_new( tr_handle * back, const char *torrent, const char *dir,
|
|
enum tr_torrent_action act, gboolean paused, char **err )
|
|
{
|
|
TrTorrent *ret;
|
|
tr_torrent *handle;
|
|
int errcode, flags;
|
|
|
|
g_assert(NULL != dir);
|
|
|
|
*err = NULL;
|
|
|
|
errcode = -1;
|
|
|
|
flags = ( TR_TOR_COPY == act || TR_TOR_MOVE == act ? TR_FLAG_SAVE : 0 );
|
|
if( paused )
|
|
flags |= TR_FLAG_PAUSED;
|
|
|
|
handle = tr_torrentInit( back, torrent, dir, flags, &errcode );
|
|
|
|
if(NULL == handle) {
|
|
switch(errcode) {
|
|
case TR_EINVALID:
|
|
*err = g_strdup_printf(_("%s: not a valid torrent file"), torrent);
|
|
break;
|
|
case TR_EDUPLICATE:
|
|
*err = g_strdup_printf(_("%s: torrent is already open"), torrent);
|
|
break;
|
|
default:
|
|
*err = g_strdup(torrent);
|
|
break;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
ret = maketorrent( handle );
|
|
|
|
if( TR_TOR_MOVE == act )
|
|
ret->delfile = g_strdup(torrent);
|
|
|
|
return ret;
|
|
}
|
|
|
|
TrTorrent *
|
|
tr_torrent_new_with_data( tr_handle * back, uint8_t * data, size_t size,
|
|
const char * dir, gboolean paused, char ** err )
|
|
{
|
|
tr_torrent * handle;
|
|
int errcode;
|
|
int flags;
|
|
|
|
g_assert( NULL != dir );
|
|
|
|
*err = NULL;
|
|
|
|
flags = TR_FLAG_SAVE;
|
|
if( paused )
|
|
flags |= TR_FLAG_PAUSED;
|
|
|
|
errcode = -1;
|
|
handle = tr_torrentInitData( back, data, size, dir, flags, &errcode );
|
|
|
|
if( NULL == handle )
|
|
{
|
|
switch( errcode )
|
|
{
|
|
case TR_EINVALID:
|
|
*err = g_strdup( _("not a valid torrent file") );
|
|
break;
|
|
case TR_EDUPLICATE:
|
|
*err = g_strdup( _("torrent is already open") );
|
|
break;
|
|
default:
|
|
*err = g_strdup( "" );
|
|
break;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
return maketorrent( handle );
|
|
}
|
|
|
|
TrTorrent *
|
|
tr_torrent_new_with_state( tr_handle * back, benc_val_t * state,
|
|
gboolean forcedpause, char ** err )
|
|
{
|
|
TrTorrent * ret;
|
|
tr_torrent * handle;
|
|
int ii, errcode;
|
|
int flags;
|
|
benc_val_t *name, *data;
|
|
char *torrent, *hash, *dir;
|
|
gboolean paused = FALSE;
|
|
gboolean seeding_cap_enabled = FALSE;
|
|
gdouble seeding_cap = 0.0;
|
|
|
|
*err = NULL;
|
|
|
|
if(TYPE_DICT != state->type)
|
|
return NULL;
|
|
|
|
torrent = hash = dir = NULL;
|
|
|
|
for(ii = 0; ii + 1 < state->val.l.count; ii += 2) {
|
|
name = state->val.l.vals + ii;
|
|
data = state->val.l.vals + ii + 1;
|
|
if(TYPE_STR == name->type &&
|
|
(TYPE_STR == data->type || TYPE_INT == data->type)) {
|
|
char * key = name->val.s.s;
|
|
char * val = data->val.s.s;
|
|
if (!strcmp (key, "torrent")) torrent = val;
|
|
else if (!strcmp (key, "hash")) hash = val;
|
|
else if (!strcmp (key, "dir")) dir = val;
|
|
else if (!strcmp (key, "paused")) paused = !!data->val.i;
|
|
else if (!strcmp (key, "seeding-cap-ratio")) seeding_cap = (data->val.i / 100.0);
|
|
else if (!strcmp (key, "seeding-cap-enabled")) seeding_cap_enabled = !!data->val.i;
|
|
}
|
|
}
|
|
|
|
if((NULL != torrent && NULL != hash) ||
|
|
(NULL == torrent && NULL == hash) || NULL == dir)
|
|
return NULL;
|
|
|
|
flags = 0;
|
|
if( paused || forcedpause )
|
|
flags |= TR_FLAG_PAUSED;
|
|
|
|
if( NULL != hash )
|
|
handle = tr_torrentInitSaved(back, hash, dir, flags, &errcode);
|
|
else
|
|
handle = tr_torrentInit(back, torrent, dir, flags, &errcode);
|
|
|
|
if(NULL == handle) {
|
|
torrent = ( NULL == hash ? torrent : hash );
|
|
switch(errcode) {
|
|
case TR_EINVALID:
|
|
*err = g_strdup_printf(_("%s: not a valid torrent file"), torrent);
|
|
break;
|
|
case TR_EDUPLICATE:
|
|
*err = g_strdup_printf(_("%s: torrent is already open"), torrent);
|
|
break;
|
|
default:
|
|
*err = g_strdup(torrent);
|
|
break;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
ret = maketorrent( handle );
|
|
ret->seeding_cap = seeding_cap;
|
|
ret->seeding_cap_enabled = seeding_cap_enabled;
|
|
return ret;
|
|
}
|
|
|
|
gboolean
|
|
tr_torrent_get_state( TrTorrent * tor, benc_val_t * state )
|
|
{
|
|
const tr_info * inf;
|
|
|
|
TR_IS_TORRENT( tor );
|
|
|
|
if( tor->severed )
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
inf = tr_torrentInfo( tor->handle );
|
|
|
|
tr_bencInit( state, TYPE_DICT );
|
|
if( tr_bencDictReserve( state, 3 ) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
if( TR_FLAG_SAVE & inf->flags )
|
|
{
|
|
tr_bencInitStr( tr_bencDictAdd( state, "hash" ),
|
|
inf->hashString, -1, 1 );
|
|
}
|
|
else
|
|
{
|
|
tr_bencInitStr( tr_bencDictAdd( state, "torrent" ),
|
|
inf->torrent, -1, 1 );
|
|
}
|
|
tr_bencInitStr( tr_bencDictAdd( state, "dir" ),
|
|
tr_torrentGetFolder( tor->handle ), -1, 1 );
|
|
tr_bencInitInt( tr_bencDictAdd( state, "paused" ),
|
|
(refreshStat(tor)->status & TR_STATUS_INACTIVE) ? 1 : 0);
|
|
tr_bencInitInt( tr_bencDictAdd( state, "seeding-cap-ratio" ),
|
|
(int)(tor->seeding_cap * 100.0)); /* two decimal places */
|
|
tr_bencInitInt( tr_bencDictAdd( state, "seeding-cap-enabled" ),
|
|
tor->seeding_cap_enabled ? 1 : 0);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/* XXX this should probably be done with a signal */
|
|
void
|
|
tr_torrent_state_saved(TrTorrent *tor) {
|
|
TR_IS_TORRENT(tor);
|
|
|
|
if(tor->severed)
|
|
return;
|
|
|
|
if(NULL != tor->delfile) {
|
|
unlink(tor->delfile);
|
|
g_free(tor->delfile);
|
|
tor->delfile = NULL;
|
|
}
|
|
}
|
|
|
|
static void
|
|
tr_torrent_set_folder(TrTorrent *tor) {
|
|
char *wd;
|
|
|
|
if(NULL != tor->dir)
|
|
tr_torrentSetFolder(tor->handle, tor->dir);
|
|
else {
|
|
wd = g_new(char, MAX_PATH_LENGTH + 1);
|
|
tr_torrentSetFolder(tor->handle,
|
|
(NULL == getcwd(wd, MAX_PATH_LENGTH + 1) ? "." : wd));
|
|
g_free(wd);
|
|
}
|
|
}
|
|
|
|
void
|
|
tr_torrent_check_seeding_cap ( TrTorrent *gtor)
|
|
{
|
|
const tr_stat * st = tr_torrent_stat( gtor );
|
|
if ((gtor->seeding_cap_enabled) && (st->ratio >= gtor->seeding_cap))
|
|
tr_torrent_stop (gtor);
|
|
}
|
|
void
|
|
tr_torrent_set_seeding_cap_ratio ( TrTorrent *gtor, gdouble ratio )
|
|
{
|
|
gtor->seeding_cap = ratio;
|
|
tr_torrent_check_seeding_cap (gtor);
|
|
}
|
|
void
|
|
tr_torrent_set_seeding_cap_enabled ( TrTorrent *gtor, gboolean b )
|
|
{
|
|
if ((gtor->seeding_cap_enabled = b))
|
|
tr_torrent_check_seeding_cap (gtor);
|
|
}
|
|
|
|
char *
|
|
tr_torrent_status_str ( TrTorrent * gtor )
|
|
{
|
|
char * top = 0;
|
|
|
|
const tr_stat * st = tr_torrent_stat( gtor );
|
|
|
|
const int tpeers = MAX (st->peersConnected, 0);
|
|
const int upeers = MAX (st->peersGettingFromUs, 0);
|
|
const int eta = st->eta;
|
|
double prog = st->percentDone * 100.0; /* [0...100] */
|
|
|
|
switch( st->status )
|
|
{
|
|
case TR_STATUS_CHECK_WAIT:
|
|
prog = st->recheckProgress * 100.0; /* [0...100] */
|
|
top = g_strdup_printf( _("Waiting to check existing files (checked %.1f%%)"), prog );
|
|
break;
|
|
|
|
case TR_STATUS_CHECK:
|
|
prog = st->recheckProgress * 100.0; /* [0...100] */
|
|
top = g_strdup_printf( _("Checking existing files (checked %.1f%%)"), prog );
|
|
break;
|
|
|
|
case TR_STATUS_DOWNLOAD:
|
|
if( eta < 0 )
|
|
top = g_strdup_printf( _("Stalled (%.1f%%)"), prog );
|
|
else {
|
|
char * timestr = readabletime(eta);
|
|
top = g_strdup_printf( _("%s remaining (%.1f%%)"), timestr, prog );
|
|
g_free(timestr);
|
|
}
|
|
break;
|
|
|
|
case TR_STATUS_DONE:
|
|
top = g_strdup_printf(
|
|
ngettext( "Uploading to %d of %d peer",
|
|
"Uploading to %d of %d peers", tpeers ),
|
|
upeers, tpeers );
|
|
break;
|
|
|
|
case TR_STATUS_SEED:
|
|
top = g_strdup_printf(
|
|
ngettext( "Seeding to %d of %d peer",
|
|
"Seeding to %d of %d peers", tpeers ),
|
|
upeers, tpeers );
|
|
break;
|
|
|
|
case TR_STATUS_STOPPING:
|
|
top = g_strdup( _("Stopping...") );
|
|
break;
|
|
|
|
case TR_STATUS_STOPPED:
|
|
top = g_strdup_printf( _("Stopped (%.1f%%)"), prog );
|
|
break;
|
|
|
|
default:
|
|
top = g_strdup_printf("Unrecognized state: %d", st->status );
|
|
break;
|
|
|
|
}
|
|
|
|
return top;
|
|
}
|