source: trunk/gtk/tr_core.c @ 4035

Last change on this file since 4035 was 4035, checked in by charles, 15 years ago

fix gtk+ crash. reported, tracked, and patched by serge in ticket #512

  • Property svn:keywords set to Date Rev Author Id
File size: 19.3 KB
Line 
1/******************************************************************************
2 * $Id: tr_core.c 4035 2007-12-01 20:07:53Z charles $
3 *
4 * Copyright (c) 2007 Transmission authors and contributors
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *****************************************************************************/
24
25#include <inttypes.h>
26#include <stdio.h>
27#include <string.h>
28
29#include <gtk/gtk.h>
30#include <glib/gi18n.h>
31
32#include <libtransmission/transmission.h>
33#include <libtransmission/utils.h>
34
35#include "conf.h"
36#include "tr_core.h"
37#include "tr_prefs.h"
38#include "tr_torrent.h"
39#include "util.h"
40
41static void
42tr_core_marshal_err( GClosure * closure, GValue * ret SHUTUP, guint count,
43                     const GValue * vals, gpointer hint SHUTUP,
44                     gpointer marshal )
45{
46    typedef void (*TRMarshalErr)
47        ( gpointer, enum tr_core_err, const char *, gpointer );
48    TRMarshalErr     callback;
49    GCClosure      * cclosure = (GCClosure*) closure;
50    enum tr_core_err errcode;
51    const char     * errstr;
52    gpointer         inst, gdata;
53
54    g_return_if_fail( 3 == count );
55
56    inst    = g_value_peek_pointer( vals );
57    errcode = g_value_get_int( vals + 1 );
58    errstr  = g_value_get_string( vals + 2 );
59    gdata   = closure->data;
60
61    callback = (TRMarshalErr) ( NULL == marshal ?
62                                cclosure->callback : marshal );
63    callback( inst, errcode, errstr, gdata );
64}
65
66static void
67tr_core_marshal_prompt( GClosure * closure, GValue * ret SHUTUP, guint count,
68                        const GValue * vals, gpointer hint SHUTUP,
69                        gpointer marshal )
70{
71    typedef void (*TRMarshalPrompt)
72        ( gpointer, GList *, enum tr_torrent_action, gboolean, gpointer );
73    TRMarshalPrompt        callback;
74    GCClosure            * cclosure = (GCClosure*) closure;
75    GList                * paths;
76    enum tr_torrent_action action;
77    gboolean               paused;
78    gpointer               inst, gdata;
79
80    g_return_if_fail( 4 == count );
81
82    inst    = g_value_peek_pointer( vals );
83    paths   = g_value_get_pointer( vals + 1 );
84    action  = g_value_get_int( vals + 2 );
85    paused  = g_value_get_boolean( vals + 3 );
86    gdata   = closure->data;
87
88    callback = (TRMarshalPrompt) ( NULL == marshal ?
89                                   cclosure->callback : marshal );
90    callback( inst, paths, action, paused, gdata );
91}
92
93static void
94tr_core_marshal_data( GClosure * closure, GValue * ret SHUTUP, guint count,
95                      const GValue * vals, gpointer hint SHUTUP,
96                      gpointer marshal )
97{
98    typedef void (*TRMarshalPrompt)
99        ( gpointer, uint8_t *, size_t, gboolean, gpointer );
100    TRMarshalPrompt        callback;
101    GCClosure            * cclosure = (GCClosure*) closure;
102    uint8_t              * data;
103    size_t                 size;
104    gboolean               paused;
105    gpointer               inst, gdata;
106
107    g_return_if_fail( 4 == count );
108
109    inst    = g_value_peek_pointer( vals );
110    data    = (uint8_t *) g_value_get_string( vals + 1 );
111    size    = g_value_get_uint( vals + 2 );
112    paused  = g_value_get_boolean( vals + 3 );
113    gdata   = closure->data;
114
115    callback = (TRMarshalPrompt) ( NULL == marshal ?
116                                   cclosure->callback : marshal );
117    callback( inst, data, size, paused, gdata );
118}
119
120static void
121tr_core_dispose( GObject * obj )
122{
123    TrCore       * self = (TrCore *) obj;
124    GObjectClass * parent;
125
126    if( self->disposed )
127        return;
128
129    self->disposed = TRUE;
130    pref_save( NULL );
131    parent = g_type_class_peek( g_type_parent( TR_CORE_TYPE ) );
132    parent->dispose( obj );
133}
134
135
136static void
137tr_core_class_init( gpointer g_class, gpointer g_class_data SHUTUP )
138{
139    GObjectClass * gobject_class;
140    TrCoreClass  * core_class;
141
142    gobject_class = G_OBJECT_CLASS( g_class );
143    gobject_class->dispose = tr_core_dispose;
144
145    core_class = TR_CORE_CLASS( g_class );
146    core_class->errsig = g_signal_new( "error", G_TYPE_FROM_CLASS( g_class ),
147                                       G_SIGNAL_RUN_LAST, 0, NULL, NULL,
148                                       tr_core_marshal_err, G_TYPE_NONE,
149                                       2, G_TYPE_INT, G_TYPE_STRING );
150    core_class->promptsig = g_signal_new( "directory-prompt",
151                                          G_TYPE_FROM_CLASS( g_class ),
152                                          G_SIGNAL_RUN_LAST, 0, NULL, NULL,
153                                          tr_core_marshal_prompt, G_TYPE_NONE,
154                                          3, G_TYPE_POINTER, G_TYPE_INT,
155                                          G_TYPE_BOOLEAN );
156    core_class->promptdatasig = g_signal_new( "directory-prompt-data",
157                                              G_TYPE_FROM_CLASS( g_class ),
158                                              G_SIGNAL_RUN_LAST, 0, NULL, NULL,
159                                              tr_core_marshal_data,
160                                              G_TYPE_NONE, 3, G_TYPE_STRING,
161                                              G_TYPE_UINT, G_TYPE_BOOLEAN );
162    core_class->quitsig = g_signal_new( "quit", G_TYPE_FROM_CLASS( g_class ),
163                                        G_SIGNAL_RUN_LAST, 0, NULL, NULL,
164                                        g_cclosure_marshal_VOID__VOID,
165                                        G_TYPE_NONE, 0 );
166    core_class->prefsig = g_signal_new( "prefs-changed",
167                                        G_TYPE_FROM_CLASS( g_class ),
168                                        G_SIGNAL_RUN_LAST, 0, NULL, NULL,
169                                        g_cclosure_marshal_VOID__STRING,
170                                        G_TYPE_NONE, 1, G_TYPE_STRING );
171}
172
173static int
174compareProgress( GtkTreeModel   * model,
175                 GtkTreeIter    * a,
176                 GtkTreeIter    * b,
177                 gpointer         user_data UNUSED )
178{
179    int ia, ib;
180    gfloat rateUpA, rateUpB;
181    gfloat rateDownA, rateDownB;
182    gfloat percentDoneA, percentDoneB;
183    guint64 uploadedEverA, uploadedEverB;
184
185    gtk_tree_model_get( model, a, MC_PROG_D, &percentDoneA,
186                                  MC_DRATE, &rateDownA,
187                                  MC_URATE, &rateUpA,
188                                  MC_UP, &uploadedEverA,
189                                  -1 );
190    gtk_tree_model_get( model, b, MC_PROG_D, &percentDoneB,
191                                  MC_DRATE, &rateDownB,
192                                  MC_URATE, &rateUpB,
193                                  MC_UP, &uploadedEverB,
194                                  -1 );
195
196    ia = (int)( rateUpA + rateDownA );
197    ib = (int)( rateUpB + rateDownB );
198    if( ia != ib )
199        return ia - ib;
200
201    ia = (int)( 100.0 * percentDoneA );
202    ib = (int)( 100.0 * percentDoneB );
203    if( ia != ib )
204        return ia - ib;
205
206    if( uploadedEverA != uploadedEverB )
207        return uploadedEverA < uploadedEverB ? -1 : 1;
208
209    return 0;
210}
211
212#define STR_REVERSE "reverse-"
213#define STR_PROGRESS "progress"
214#define STR_NAME "name"
215
216static void
217onSortColumnChanged( GtkTreeSortable * sortable, gpointer unused UNUSED )
218{
219    int column;
220    GtkSortType order;
221    if( gtk_tree_sortable_get_sort_column_id( sortable, &column, &order ) )
222    {
223        GString * gstr = g_string_new( NULL );
224        switch( column ) {
225            case MC_PROG_D: g_string_assign( gstr, STR_PROGRESS ); break;
226            default: g_string_assign( gstr, STR_NAME ); break;
227        }
228        if( order == GTK_SORT_DESCENDING )
229            g_string_prepend( gstr, STR_REVERSE );
230        pref_string_set( PREF_KEY_SORT_COLUMN, gstr->str );
231        g_string_free( gstr, TRUE );
232    }
233}
234
235void
236tr_core_set_sort_column_from_prefs( TrCore * core )
237{
238    char * val = pref_string_get( PREF_KEY_SORT_COLUMN );
239    char * freeme = val;
240    gint column;
241    GtkSortType order = GTK_SORT_ASCENDING;
242    if( g_str_has_prefix( val, STR_REVERSE ) ) {
243        order = GTK_SORT_DESCENDING;
244        val += strlen( STR_REVERSE );
245    }
246    if( !strcmp( val, STR_PROGRESS ) )
247        column = MC_PROG_D;
248    else /* default */
249        column = MC_NAME;
250    gtk_tree_sortable_set_sort_column_id ( GTK_TREE_SORTABLE( core->model ), column, order );
251    g_free( freeme );
252}
253
254static void
255tr_core_init( GTypeInstance * instance, gpointer g_class SHUTUP )
256{
257    TrCore * self = (TrCore *) instance;
258    GtkListStore * store;
259
260    /* column types for the model used to store torrent information */
261    /* keep this in sync with the enum near the bottom of tr_core.h */
262    GType types[] =
263    {
264        /* info->name, info->totalSize, info->hashString, status, */
265        G_TYPE_STRING, G_TYPE_UINT64,   G_TYPE_STRING,    G_TYPE_INT,
266        /* error,   errorString,   percentComplete, percentDone,  rateDownload, rateUpload, */
267        G_TYPE_INT, G_TYPE_STRING, G_TYPE_FLOAT,    G_TYPE_FLOAT, G_TYPE_FLOAT, G_TYPE_FLOAT,
268        /* eta,     peersConnected, peersUploading, peersDownloading, seeders, */
269        G_TYPE_INT, G_TYPE_INT,     G_TYPE_INT,     G_TYPE_INT,       G_TYPE_INT,
270        /* leechers, completedFromTracker, downloaded,    uploaded */
271        G_TYPE_INT,  G_TYPE_INT,           G_TYPE_UINT64, G_TYPE_UINT64,
272        /* ratio,      left,          TrTorrent object, ID for IPC */
273        G_TYPE_FLOAT,  G_TYPE_UINT64, TR_TORRENT_TYPE,  G_TYPE_INT,
274    };
275
276#ifdef REFDBG
277    fprintf( stderr, "core    %p init\n", self );
278#endif
279
280    /* create the model used to store torrent data */
281    g_assert( ALEN( types ) == MC_ROW_COUNT );
282    store = gtk_list_store_newv( MC_ROW_COUNT, types );
283    g_signal_connect( store, "sort-column-changed", G_CALLBACK(onSortColumnChanged), NULL );
284
285    gtk_tree_sortable_set_sort_func( GTK_TREE_SORTABLE(store),
286                                     MC_PROG_D,
287                                     compareProgress,
288                                     NULL, NULL );
289
290    self->model    = GTK_TREE_MODEL( store );
291    self->handle   = tr_init( "gtk" );
292    self->nextid   = 1;
293    self->quitting = FALSE;
294    self->disposed = FALSE;
295}
296
297GType
298tr_core_get_type( void )
299{
300    static GType type = 0;
301
302    if( 0 == type )
303    {
304        static const GTypeInfo info =
305        {
306            sizeof( TrCoreClass ),
307            NULL,                       /* base_init */
308            NULL,                       /* base_finalize */
309            tr_core_class_init,         /* class_init */
310            NULL,                       /* class_finalize */
311            NULL,                       /* class_data */
312            sizeof( TrCore ),
313            0,                          /* n_preallocs */
314            tr_core_init,               /* instance_init */
315            NULL,
316        };
317        type = g_type_register_static( G_TYPE_OBJECT, "TrCore", &info, 0 );
318    }
319
320    return type;
321}
322
323/**
324***
325**/
326
327TrCore *
328tr_core_new( void )
329{
330    return g_object_new( TR_CORE_TYPE, NULL );
331}
332
333GtkTreeModel *
334tr_core_model( TrCore * self )
335{
336    g_return_val_if_fail (TR_IS_CORE(self), NULL);
337
338    return self->disposed ? NULL : self->model;
339}
340
341tr_handle *
342tr_core_handle( TrCore * self )
343{
344    g_return_val_if_fail (TR_IS_CORE(self), NULL);
345
346    return self->disposed ? NULL : self->handle;
347}
348
349static void
350tr_core_insert( TrCore * self, TrTorrent * tor )
351{
352    const tr_info * inf = tr_torrent_info( tor );
353    gtk_list_store_insert_with_values( GTK_LIST_STORE( self->model ), NULL, 0, 
354                                       MC_NAME,    inf->name,
355                                       MC_SIZE,    inf->totalSize,
356                                       MC_HASH,    inf->hashString,
357                                       MC_TORRENT, tor,
358                                       MC_ID,      self->nextid,
359                                       -1);
360    g_object_unref( tor );
361    self->nextid++;
362}
363
364int
365tr_core_load( TrCore * self, gboolean paused )
366{
367    int i;
368    int count = 0;
369    tr_torrent ** torrents;
370    char * path;
371
372    TR_IS_CORE( self );
373
374    path = getdownloaddir( );
375
376    torrents = tr_loadTorrents ( self->handle, path, paused, &count );
377    for( i=0; i<count; ++i )
378        tr_core_insert( self, tr_torrent_new_preexisting( torrents[i] ) );
379    tr_free( torrents );
380
381    g_free( path );
382    return count;
383}
384
385gboolean
386tr_core_add( TrCore * self, const char * path, enum tr_torrent_action act,
387             gboolean paused )
388{
389    GList * list;
390    int     ret;
391
392    TR_IS_CORE( self );
393
394    list = g_list_append( NULL, (void *) path );
395    ret  = tr_core_add_list( self, list, act, paused );
396    g_list_free( list );
397
398    return 1 == ret;
399}
400
401static void
402tr_core_errsig( TrCore * self, enum tr_core_err type, const char * msg )
403{
404    TrCoreClass * class;
405
406    class = g_type_class_peek( TR_CORE_TYPE );
407    g_signal_emit( self, class->errsig, 0, type, msg );
408}
409
410gboolean
411tr_core_add_dir( TrCore * self, const char * path, const char * dir,
412                 enum tr_torrent_action act, gboolean paused )
413{
414    TrTorrent * tor;
415    char      * errstr;
416
417    TR_IS_CORE( self );
418
419    errstr = NULL;
420    tor = tr_torrent_new( self->handle, path, dir, act, paused, &errstr );
421    if( NULL == tor )
422    {
423        tr_core_errsig( self, TR_CORE_ERR_ADD_TORRENT, errstr );
424        g_free( errstr );
425        return FALSE;
426    }
427    g_assert( NULL == errstr );
428
429    tr_core_insert( self, tor );
430
431    return TRUE;
432}
433
434int
435tr_core_add_list( TrCore * self, GList * paths, enum tr_torrent_action act,
436                  gboolean paused )
437{
438    char * dir;
439    int count;
440
441    TR_IS_CORE( self );
442
443    if( pref_flag_get( PREF_KEY_DIR_ASK ) )
444    {
445        TrCoreClass * class = g_type_class_peek( TR_CORE_TYPE );
446        g_signal_emit( self, class->promptsig, 0, paths, act, paused );
447        return 0;
448    }
449
450    dir = getdownloaddir();
451    count = 0;
452    for( ; paths; paths=paths->next )
453        if( tr_core_add_dir( self, paths->data, dir, act, paused ) )
454            count++;
455
456    g_free( dir );
457    return count;
458}
459
460gboolean
461tr_core_add_data( TrCore * self, uint8_t * data, size_t size, gboolean paused )
462{
463    gboolean ret;
464    char * path;
465    TR_IS_CORE( self );
466
467    if( pref_flag_get( PREF_KEY_DIR_ASK ) )
468    {
469        TrCoreClass * class = g_type_class_peek( TR_CORE_TYPE );
470        g_signal_emit( self, class->promptdatasig, 0, data, size, paused );
471        return FALSE;
472    }
473
474    path = getdownloaddir( );
475    ret = tr_core_add_data_dir( self, data, size, path, paused );
476    g_free( path );
477    return ret;
478}
479
480gboolean
481tr_core_add_data_dir( TrCore * self, uint8_t * data, size_t size,
482                      const char * dir, gboolean paused )
483{
484    TrTorrent * tor;
485    char      * errstr = NULL;
486
487    TR_IS_CORE( self );
488
489    tor = tr_torrent_new_with_data( self->handle, data, size, dir,
490                                    paused, &errstr );
491    if( NULL == tor )
492    {
493        tr_core_errsig( self, TR_CORE_ERR_ADD_TORRENT, errstr );
494        g_free( errstr );
495        return FALSE;
496    }
497    g_assert( NULL == errstr );
498
499    tr_core_insert( self, tor );
500
501    return TRUE;
502}
503
504void
505tr_core_torrents_added( TrCore * self )
506{
507    TR_IS_CORE( self );
508
509    tr_core_update( self );
510    tr_core_errsig( self, TR_CORE_ERR_NO_MORE_TORRENTS, NULL );
511}
512
513void
514tr_core_delete_torrent( TrCore * self, GtkTreeIter * iter )
515{
516    TrTorrent * tor;
517
518    TR_IS_CORE( self );
519
520    gtk_tree_model_get( self->model, iter, MC_TORRENT, &tor, -1 );
521    gtk_list_store_remove( GTK_LIST_STORE( self->model ), iter );
522    tr_torrentRemoveSaved( tr_torrent_handle( tor ) );
523
524    tr_torrent_sever( tor );
525}
526
527static gboolean
528update_foreach( GtkTreeModel * model,
529                GtkTreePath  * path UNUSED,
530                GtkTreeIter  * iter,
531                gpointer       data UNUSED)
532{
533    TrTorrent * tor;
534    const tr_stat * st;
535
536    gtk_tree_model_get( model, iter, MC_TORRENT, &tor, -1 );
537    st = tr_torrent_stat( tor );
538    tr_torrent_check_seeding_cap ( tor );
539    g_object_unref( tor );
540
541    gtk_list_store_set( GTK_LIST_STORE( model ), iter,
542                        MC_STAT,        st->status,
543                        MC_ERR,         st->error,
544                        MC_TERR,        st->errorString,
545                        MC_PROG_C,      st->percentComplete,
546                        MC_PROG_D,      st->percentDone,
547                        MC_DRATE,       st->rateDownload,
548                        MC_URATE,       st->rateUpload,
549                        MC_ETA,         st->eta,
550                        MC_PEERS,       st->peersConnected,
551                        MC_UPEERS,      st->peersGettingFromUs,
552                        MC_DPEERS,      st->peersSendingToUs,
553                        MC_SEED,        st->seeders,
554                        MC_LEECH,       st->leechers,
555                        MC_DONE,        st->completedFromTracker,
556                        MC_DOWN,        st->downloadedEver,
557                        MC_UP,          st->uploadedEver,
558                        MC_RATIO,       st->ratio,
559                        MC_LEFT,        st->leftUntilDone,
560                        -1 );
561
562    return FALSE;
563}
564
565void
566tr_core_update( TrCore * self )
567{
568    int column;
569    GtkSortType order;
570    GtkTreeSortable * sortable;
571
572    /* pause sorting */
573    sortable = GTK_TREE_SORTABLE( self->model );
574    gtk_tree_sortable_get_sort_column_id( sortable, &column, &order );
575    gtk_tree_sortable_set_sort_column_id( sortable, GTK_TREE_SORTABLE_UNSORTED_SORT_COLUMN_ID, order );
576
577    /* refresh the model */
578    gtk_tree_model_foreach( self->model, update_foreach, NULL );
579
580    /* resume sorting */
581    gtk_tree_sortable_set_sort_column_id( sortable, column, order );
582}
583
584void
585tr_core_quit( TrCore * self )
586{
587    TrCoreClass * class;
588
589    TR_IS_CORE( self );
590
591    class = g_type_class_peek( TR_CORE_TYPE );
592    g_signal_emit( self, class->quitsig, 0 );
593}
594
595/**
596***  Prefs
597**/
598
599static void
600commitPrefsChange( TrCore * self, const char * key )
601{
602    TrCoreClass * class = g_type_class_peek( TR_CORE_TYPE );
603    pref_save( NULL );
604    g_signal_emit( self, class->prefsig, 0, key );
605}
606
607void
608tr_core_set_pref( TrCore * self, const char * key, const char * newval )
609{
610    char * oldval = pref_string_get( key );
611    if( tr_strcmp( oldval, newval ) )
612    {
613        pref_string_set( key, newval );
614        commitPrefsChange( self, key );
615    }
616    g_free( oldval );
617}
618
619void
620tr_core_set_pref_bool( TrCore * self, const char * key, gboolean newval )
621{
622    const gboolean oldval = pref_flag_get( key );
623    if( oldval != newval )
624    {
625        pref_flag_set( key, newval );
626        commitPrefsChange( self, key );
627    }
628}
629
630void
631tr_core_set_pref_int( TrCore * self, const char * key, int newval )
632{
633    const int oldval = pref_int_get( key );
634    if( oldval != newval )
635    {
636        pref_int_set( key, newval );
637        commitPrefsChange( self, key );
638    }
639}
Note: See TracBrowser for help on using the repository browser.