mirror/transmission
1
0
Fork 0
mirror of https://github.com/transmission/transmission synced 2024-12-26 09:37:56 +00:00
Code Issues Releases Wiki Activity
transmission/gtk/tr-core.c

1931 lines
51 KiB
C
Raw Blame History

/******************************************************************************
* $Id$
*
* Copyright (c) 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 <math.h> /* pow () */
#include <string.h> /* strlen */
#include <gtk/gtk.h>
#include <glib/gi18n.h>
#include <gio/gio.h>
#include <event2/buffer.h>
#include <libtransmission/transmission.h>
#include <libtransmission/log.h>
#include <libtransmission/rpcimpl.h>
#include <libtransmission/utils.h> /* tr_free */
#include <libtransmission/variant.h>
#include "actions.h"
#include "conf.h"
#include "notify.h"
#include "tr-core.h"
#include "tr-prefs.h"
#include "util.h"
/***
****
***/
enum
{
ADD_ERROR_SIGNAL,
ADD_PROMPT_SIGNAL,
BLOCKLIST_SIGNAL,
BUSY_SIGNAL,
PORT_SIGNAL,
PREFS_SIGNAL,
LAST_SIGNAL
};
static guint signals[LAST_SIGNAL] = { 0 };
static void core_maybe_inhibit_hibernation (TrCore * core);
struct TrCorePrivate
{
GFileMonitor * monitor;
gulong monitor_tag;
GFile * monitor_dir;
GSList * monitor_files;
gulong monitor_idle_tag;
gboolean adding_from_watch_dir;
gboolean inhibit_allowed;
gboolean have_inhibit_cookie;
gboolean dbus_error;
guint inhibit_cookie;
gint busy_count;
GtkTreeModel * raw_model;
GtkTreeModel * sorted_model;
tr_session * session;
GStringChunk * string_chunk;
};
static int
core_is_disposed (const TrCore * core)
{
return !core || !core->priv->sorted_model;
}
G_DEFINE_TYPE (TrCore, tr_core, G_TYPE_OBJECT)
static void
core_dispose (GObject * o)
{
TrCore * core = TR_CORE (o);
if (core->priv->sorted_model != NULL)
{
g_object_unref (core->priv->sorted_model);
core->priv->sorted_model = NULL;
core->priv->raw_model = NULL;
}
G_OBJECT_CLASS (tr_core_parent_class)->dispose (o);
}
static void
core_finalize (GObject * o)
{
TrCore * core = TR_CORE (o);
g_string_chunk_free (core->priv->string_chunk);
G_OBJECT_CLASS (tr_core_parent_class)->finalize (o);
}
static void
tr_core_class_init (TrCoreClass * core_class)
{
GObjectClass * gobject_class;
GType core_type = G_TYPE_FROM_CLASS (core_class);
g_type_class_add_private (core_class, sizeof (struct TrCorePrivate));
gobject_class = G_OBJECT_CLASS (core_class);
gobject_class->dispose = core_dispose;
gobject_class->finalize = core_finalize;
signals[ADD_ERROR_SIGNAL] =
g_signal_new ("add-error", core_type,
G_SIGNAL_RUN_LAST,
G_STRUCT_OFFSET (TrCoreClass, add_error),
NULL, NULL,
g_cclosure_marshal_VOID__UINT_POINTER,
G_TYPE_NONE,
2, G_TYPE_UINT, G_TYPE_POINTER);
signals[ADD_PROMPT_SIGNAL] =
g_signal_new ("add-prompt", core_type,
G_SIGNAL_RUN_LAST,
G_STRUCT_OFFSET (TrCoreClass, add_prompt),
NULL, NULL,
g_cclosure_marshal_VOID__POINTER,
G_TYPE_NONE,
1, G_TYPE_POINTER);
signals[BUSY_SIGNAL] =
g_signal_new ("busy", core_type,
G_SIGNAL_RUN_FIRST,
G_STRUCT_OFFSET (TrCoreClass, busy),
NULL, NULL,
g_cclosure_marshal_VOID__BOOLEAN,
G_TYPE_NONE,
1, G_TYPE_BOOLEAN);
signals[BLOCKLIST_SIGNAL] =
g_signal_new ("blocklist-updated", core_type,
G_SIGNAL_RUN_FIRST,
G_STRUCT_OFFSET (TrCoreClass, blocklist_updated),
NULL, NULL,
g_cclosure_marshal_VOID__INT,
G_TYPE_NONE,
1, G_TYPE_INT);
signals[PORT_SIGNAL] =
g_signal_new ("port-tested", core_type,
G_SIGNAL_RUN_LAST,
G_STRUCT_OFFSET (TrCoreClass, port_tested),
NULL, NULL,
g_cclosure_marshal_VOID__BOOLEAN,
G_TYPE_NONE,
1, G_TYPE_BOOLEAN);
signals[PREFS_SIGNAL] =
g_signal_new ("prefs-changed", core_type,
G_SIGNAL_RUN_LAST,
G_STRUCT_OFFSET (TrCoreClass, prefs_changed),
NULL, NULL,
g_cclosure_marshal_VOID__INT,
G_TYPE_NONE,
1, G_TYPE_INT);
}
static void
tr_core_init (TrCore * core)
{
GtkListStore * store;
struct TrCorePrivate * p;
/* column types for the model used to store torrent information */
/* keep this in sync with the enum near the bottom of tr_core.h */
GType types[] = { G_TYPE_POINTER, /* collated name */
G_TYPE_POINTER, /* tr_torrent* */
G_TYPE_INT, /* torrent id */
G_TYPE_DOUBLE, /* tr_stat.pieceUploadSpeed_KBps */
G_TYPE_DOUBLE, /* tr_stat.pieceDownloadSpeed_KBps */
G_TYPE_INT, /* tr_stat.peersGettingFromUs */
G_TYPE_INT, /* tr_stat.peersSendingToUs + webseedsSendingToUs */
G_TYPE_DOUBLE, /* tr_stat.recheckProgress */
G_TYPE_BOOLEAN, /* filter.c:ACTIVITY_FILTER_ACTIVE */
G_TYPE_INT, /* tr_stat.activity */
G_TYPE_UCHAR, /* tr_stat.finished */
G_TYPE_CHAR, /* tr_priority_t */
G_TYPE_INT, /* tr_stat.queuePosition */
G_TYPE_UINT, /* build_torrent_trackers_hash () */
G_TYPE_INT, /* MC_ERROR */
G_TYPE_INT }; /* MC_ACTIVE_PEER_COUNT */
p = core->priv = G_TYPE_INSTANCE_GET_PRIVATE (core,
TR_CORE_TYPE,
struct TrCorePrivate);
/* create the model used to store torrent data */
g_assert (G_N_ELEMENTS (types) == MC_ROW_COUNT);
store = gtk_list_store_newv (MC_ROW_COUNT, types);
p->raw_model = GTK_TREE_MODEL (store);
p->sorted_model = gtk_tree_model_sort_new_with_model (p->raw_model);
p->string_chunk = g_string_chunk_new (2048);
g_object_unref (p->raw_model);
}
/***
**** EMIT SIGNALS
***/
static inline void
core_emit_blocklist_udpated (TrCore * core, int ruleCount)
{
g_signal_emit (core, signals[BLOCKLIST_SIGNAL], 0, ruleCount);
}
static inline void
core_emit_port_tested (TrCore * core, gboolean is_open)
{
g_signal_emit (core, signals[PORT_SIGNAL], 0, is_open);
}
static inline void
core_emit_err (TrCore * core, enum tr_core_err type, const char * msg)
{
g_signal_emit (core, signals[ADD_ERROR_SIGNAL], 0, type, msg);
}
static inline void
core_emit_busy (TrCore * core, gboolean is_busy)
{
g_signal_emit (core, signals[BUSY_SIGNAL], 0, is_busy);
}
void
gtr_core_pref_changed (TrCore * core, const tr_quark key)
{
g_signal_emit (core, signals[PREFS_SIGNAL], 0, key);
}
/***
****
***/
static GtkTreeModel *
core_raw_model (TrCore * core)
{
return core_is_disposed (core) ? NULL : core->priv->raw_model;
}
GtkTreeModel *
gtr_core_model (TrCore * core)
{
return core_is_disposed (core) ? NULL : core->priv->sorted_model;
}
tr_session *
gtr_core_session (TrCore * core)
{
return core_is_disposed (core) ? NULL : core->priv->session;
}
/***
**** BUSY
***/
static bool
core_is_busy (TrCore * core)
{
return core->priv->busy_count > 0;
}
static void
core_add_to_busy (TrCore * core, int addMe)
{
const bool wasBusy = core_is_busy (core);
core->priv->busy_count += addMe;
if (wasBusy != core_is_busy (core))
core_emit_busy (core, core_is_busy (core));
}
static void core_inc_busy (TrCore * core) { core_add_to_busy (core, 1); }
static void core_dec_busy (TrCore * core) { core_add_to_busy (core, -1); }
/***
****
**** SORTING THE MODEL
****
***/
static gboolean
is_valid_eta (int t)
{
return (t != TR_ETA_NOT_AVAIL) && (t != TR_ETA_UNKNOWN);
}
static int
compare_eta (int a, int b)
{
int ret;
const gboolean a_valid = is_valid_eta (a);
const gboolean b_valid = is_valid_eta (b);
if (!a_valid && !b_valid)
ret = 0;
else if (!a_valid)
ret = -1;
else if (!b_valid)
ret = 1;
else
ret = a < b ? 1 : -1;
return ret;
}
static int
compare_double (double a, double b)
{
int ret;
if (a < b)
ret = -1;
else if (a > b)
ret = 1;
else
ret = 0;
return ret;
}
static int
compare_uint64 (uint64_t a, uint64_t b)
{
int ret;
if (a < b)
ret = -1;
else if (a > b)
ret = 1;
else
ret = 0;
return ret;
}
static int
compare_int (int a, int b)
{
int ret;
if (a < b)
ret = -1;
else if (a > b)
ret = 1;
else
ret = 0;
return ret;
}
static int
compare_ratio (double a, double b)
{
int ret;
if ((int)a == TR_RATIO_INF && (int)b == TR_RATIO_INF)
ret = 0;
else if ((int)a == TR_RATIO_INF)
ret = 1;
else if ((int)b == TR_RATIO_INF)
ret = -1;
else
ret = compare_double (a, b);
return ret;
}
static int
compare_time (time_t a, time_t b)
{
int ret;
if (a < b)
ret = -1;
else if (a > b)
ret = 1;
else
ret = 0;
return ret;
}
static int
compare_by_name (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer user_data UNUSED)
{
const char *ca, *cb;
gtk_tree_model_get (m, a, MC_NAME_COLLATED, &ca, -1);
gtk_tree_model_get (m, b, MC_NAME_COLLATED, &cb, -1);
return g_strcmp0 (ca, cb);
}
static int
compare_by_queue (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer user_data UNUSED)
{
tr_torrent *ta, *tb;
const tr_stat *sa, *sb;
gtk_tree_model_get (m, a, MC_TORRENT, &ta, -1);
sa = tr_torrentStatCached (ta);
gtk_tree_model_get (m, b, MC_TORRENT, &tb, -1);
sb = tr_torrentStatCached (tb);
return sb->queuePosition - sa->queuePosition;
}
static int
compare_by_ratio (GtkTreeModel* m, GtkTreeIter * a, GtkTreeIter * b, gpointer user_data)
{
int ret = 0;
tr_torrent *ta, *tb;
const tr_stat *sa, *sb;
gtk_tree_model_get (m, a, MC_TORRENT, &ta, -1);
sa = tr_torrentStatCached (ta);
gtk_tree_model_get (m, b, MC_TORRENT, &tb, -1);
sb = tr_torrentStatCached (tb);
if (!ret)
ret = compare_ratio (sa->ratio, sb->ratio);
if (!ret)
ret = compare_by_queue (m, a, b, user_data);
return ret;
}
static int
compare_by_activity (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer user_data)
{
int ret = 0;
tr_torrent *ta, *tb;
double aUp, aDown, bUp, bDown;
gtk_tree_model_get (m, a, MC_SPEED_UP, &aUp,
MC_SPEED_DOWN, &aDown,
MC_TORRENT, &ta,
-1);
gtk_tree_model_get (m, b, MC_SPEED_UP, &bUp,
MC_SPEED_DOWN, &bDown,
MC_TORRENT, &tb,
-1);
ret = compare_double (aUp+aDown, bUp+bDown);
if (!ret)
{
const tr_stat * const sa = tr_torrentStatCached (ta);
const tr_stat * const sb = tr_torrentStatCached (tb);
ret = compare_uint64 (sa->peersSendingToUs + sa->peersGettingFromUs,
sb->peersSendingToUs + sb->peersGettingFromUs);
}
if (!ret)
ret = compare_by_queue (m, a, b, user_data);
return ret;
}
static int
compare_by_age (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer u)
{
int ret = 0;
tr_torrent *ta, *tb;
gtk_tree_model_get (m, a, MC_TORRENT, &ta, -1);
gtk_tree_model_get (m, b, MC_TORRENT, &tb, -1);
if (!ret)
ret = compare_time (tr_torrentStatCached (ta)->addedDate,
tr_torrentStatCached (tb)->addedDate);
if (!ret)
ret = compare_by_name (m, a, b, u);
return ret;
}
static int
compare_by_size (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer u)
{
int ret = 0;
tr_torrent *t;
const tr_info *ia, *ib;
gtk_tree_model_get (m, a, MC_TORRENT, &t, -1);
ia = tr_torrentInfo (t);
gtk_tree_model_get (m, b, MC_TORRENT, &t, -1);
ib = tr_torrentInfo (t);
if (!ret)
ret = compare_uint64 (ia->totalSize, ib->totalSize);
if (!ret)
ret = compare_by_name (m, a, b, u);
return ret;
}
static int
compare_by_progress (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer u)
{
int ret = 0;
tr_torrent * t;
const tr_stat *sa, *sb;
gtk_tree_model_get (m, a, MC_TORRENT, &t, -1);
sa = tr_torrentStatCached (t);
gtk_tree_model_get (m, b, MC_TORRENT, &t, -1);
sb = tr_torrentStatCached (t);
if (!ret)
ret = compare_double (sa->percentComplete, sb->percentComplete);
if (!ret)
ret = compare_double (sa->seedRatioPercentDone, sb->seedRatioPercentDone);
if (!ret)
ret = compare_by_ratio (m, a, b, u);
return ret;
}
static int
compare_by_eta (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer u)
{
int ret = 0;
tr_torrent *ta, *tb;
gtk_tree_model_get (m, a, MC_TORRENT, &ta, -1);
gtk_tree_model_get (m, b, MC_TORRENT, &tb, -1);
if (!ret)
ret = compare_eta (tr_torrentStatCached (ta)->eta,
tr_torrentStatCached (tb)->eta);
if (!ret)
ret = compare_by_name (m, a, b, u);
return ret;
}
static int
compare_by_state (GtkTreeModel * m,
GtkTreeIter * a,
GtkTreeIter * b,
gpointer u)
{
int ret = 0;
int sa, sb;
tr_torrent *ta, *tb;
gtk_tree_model_get (m, a, MC_ACTIVITY, &sa, MC_TORRENT, &ta, -1);
gtk_tree_model_get (m, b, MC_ACTIVITY, &sb, MC_TORRENT, &tb, -1);
if (!ret)
ret = compare_int (sa, sb);
if (!ret)
ret = compare_by_queue (m, a, b, u);
return ret;
}
static void
core_set_sort_mode (TrCore * core, const char * mode, gboolean is_reversed)
{
const int col = MC_TORRENT;
GtkTreeIterCompareFunc sort_func;
GtkSortType type = is_reversed ? GTK_SORT_ASCENDING : GTK_SORT_DESCENDING;
GtkTreeSortable * sortable = GTK_TREE_SORTABLE (gtr_core_model (core));
if (!g_strcmp0 (mode, "sort-by-activity"))
sort_func = compare_by_activity;
else if (!g_strcmp0 (mode, "sort-by-age"))
sort_func = compare_by_age;
else if (!g_strcmp0 (mode, "sort-by-progress"))
sort_func = compare_by_progress;
else if (!g_strcmp0 (mode, "sort-by-queue"))
sort_func = compare_by_queue;
else if (!g_strcmp0 (mode, "sort-by-time-left"))
sort_func = compare_by_eta;
else if (!g_strcmp0 (mode, "sort-by-ratio"))
sort_func = compare_by_ratio;
else if (!g_strcmp0 (mode, "sort-by-state"))
sort_func = compare_by_state;
else if (!g_strcmp0 (mode, "sort-by-size"))
sort_func = compare_by_size;
else {
sort_func = compare_by_name;
type = is_reversed ? GTK_SORT_DESCENDING : GTK_SORT_ASCENDING;
}
gtk_tree_sortable_set_sort_func (sortable, col, sort_func, NULL, NULL);
gtk_tree_sortable_set_sort_column_id (sortable, col, type);
}
/***
****
**** WATCHDIR
****
***/
static time_t
get_file_mtime (GFile * file)
{
GFileInfo * info;
time_t mtime = 0;
info = g_file_query_info (file, G_FILE_ATTRIBUTE_TIME_MODIFIED, 0, NULL, NULL);
if (info != NULL)
{
mtime = g_file_info_get_attribute_uint64 (info, G_FILE_ATTRIBUTE_TIME_MODIFIED);
g_object_unref (G_OBJECT (info));
}
return mtime;
}
static void
rename_torrent_and_unref_file (GFile * file)
{
GFileInfo * info;
info = g_file_query_info (file, G_FILE_ATTRIBUTE_STANDARD_EDIT_NAME, 0, NULL, NULL);
if (info != NULL)
{
GError * error = NULL;
const char * old_name;
char * new_name;
GFile * new_file;
old_name = g_file_info_get_attribute_string (info, G_FILE_ATTRIBUTE_STANDARD_EDIT_NAME);
new_name = g_strdup_printf ("%s.added", old_name);
new_file = g_file_set_display_name (file, new_name, NULL, &error);
if (error != NULL)
{
g_message ("Unable to rename \"%s\" as \"%s\": %s", old_name, new_name, error->message);
g_error_free (error);
}
if (new_file != NULL)
g_object_unref (G_OBJECT (new_file));
g_free (new_name);
g_object_unref (G_OBJECT(info));
}
g_object_unref (G_OBJECT(file));
}
static gboolean
core_watchdir_idle (gpointer gcore)
{
GSList * l;
GSList * changing = NULL;
GSList * unchanging = NULL;
TrCore * core = TR_CORE (gcore);
const time_t now = tr_time ();
struct TrCorePrivate * p = core->priv;
/* separate the files into two lists: changing and unchanging */
for (l=p->monitor_files; l!=NULL; l=l->next)
{
GFile * file = l->data;
const time_t mtime = get_file_mtime (file);
if (mtime + 2 >= now)
changing = g_slist_prepend (changing, file);
else
unchanging = g_slist_prepend (unchanging, file);
}
/* add the files that have stopped changing */
if (unchanging != NULL)
{
const gboolean do_start = gtr_pref_flag_get (TR_KEY_start_added_torrents);
const gboolean do_prompt = gtr_pref_flag_get (TR_KEY_show_options_window);
core->priv->adding_from_watch_dir = TRUE;
gtr_core_add_files (core, unchanging, do_start, do_prompt, TRUE);
g_slist_foreach (unchanging, (GFunc)rename_torrent_and_unref_file, NULL);
g_slist_free (unchanging);
core->priv->adding_from_watch_dir = FALSE;
}
/* keep monitoring the ones that are still changing */
g_slist_free (p->monitor_files);
p->monitor_files = changing;
/* if monitor_files is nonempty, keep checking every second */
if (core->priv->monitor_files)
return G_SOURCE_CONTINUE;
core->priv->monitor_idle_tag = 0;
return G_SOURCE_REMOVE;
}
/* If this file is a torrent, add it to our list */
static void
core_watchdir_monitor_file (TrCore * core, GFile * file)
{
char * filename = g_file_get_path (file);
const gboolean is_torrent = g_str_has_suffix (filename, ".torrent");
if (is_torrent)
{
GSList * l;
struct TrCorePrivate * p = core->priv;
/* if we're not already watching this file, start watching it now */
for (l=p->monitor_files; l!=NULL; l=l->next)
if (g_file_equal (file, l->data))
break;
if (l == NULL)
{
g_object_ref (file);
p->monitor_files = g_slist_prepend (p->monitor_files, file);
if (p->monitor_idle_tag == 0)
p->monitor_idle_tag = gdk_threads_add_timeout_seconds (1, core_watchdir_idle, core);
}
}
g_free (filename);
}
/* GFileMonitor noticed a file was created */
static void
on_file_changed_in_watchdir (GFileMonitor * monitor UNUSED,
GFile * file,
GFile * other_type UNUSED,
GFileMonitorEvent event_type,
gpointer core)
{
if (event_type == G_FILE_MONITOR_EVENT_CREATED)
core_watchdir_monitor_file (core, file);
}
/* walk through the pre-existing files in the watchdir */
static void
core_watchdir_scan (TrCore * core)
{
const char * dirname = gtr_pref_string_get (TR_KEY_watch_dir);
GDir * dir = g_dir_open (dirname, 0, NULL);
if (dir != NULL)
{
const char * name;
while ((name = g_dir_read_name (dir)))
{
char * filename = g_build_filename (dirname, name, NULL);
GFile * file = g_file_new_for_path (filename);
core_watchdir_monitor_file (core, file);
g_object_unref (file);
g_free (filename);
}
g_dir_close (dir);
}
}
static void
core_watchdir_update (TrCore * core)
{
const gboolean is_enabled = gtr_pref_flag_get (TR_KEY_watch_dir_enabled);
GFile * dir = g_file_new_for_path (gtr_pref_string_get (TR_KEY_watch_dir));
struct TrCorePrivate * p = core->priv;
if (p->monitor && (!is_enabled || !g_file_equal (dir, p->monitor_dir)))
{
g_signal_handler_disconnect (p->monitor, p->monitor_tag);
g_file_monitor_cancel (p->monitor);
g_object_unref (p->monitor);
g_object_unref (p->monitor_dir);
p->monitor_dir = NULL;
p->monitor = NULL;
p->monitor_tag = 0;
}
if (is_enabled && !p->monitor)
{
GFileMonitor * m = g_file_monitor_directory (dir, 0, NULL, NULL);
core_watchdir_scan (core);
g_object_ref (dir);
p->monitor = m;
p->monitor_dir = dir;
p->monitor_tag = g_signal_connect (m, "changed",
G_CALLBACK (on_file_changed_in_watchdir), core);
}
g_object_unref (dir);
}
/***
****
***/
static void
on_pref_changed (TrCore * core, const tr_quark key, gpointer data UNUSED)
{
switch (key)
{
case TR_KEY_sort_mode:
case TR_KEY_sort_reversed:
{
const char * mode = gtr_pref_string_get (TR_KEY_sort_mode);
const gboolean is_reversed = gtr_pref_flag_get (TR_KEY_sort_reversed);
core_set_sort_mode (core, mode, is_reversed);
break;
}
case TR_KEY_peer_limit_global:
tr_sessionSetPeerLimit (gtr_core_session (core), gtr_pref_int_get (key));
break;
case TR_KEY_peer_limit_per_torrent:
tr_sessionSetPeerLimitPerTorrent (gtr_core_session (core), gtr_pref_int_get (key));
break;
case TR_KEY_inhibit_desktop_hibernation:
core_maybe_inhibit_hibernation (core);
break;
case TR_KEY_watch_dir:
case TR_KEY_watch_dir_enabled:
core_watchdir_update (core);
break;
default:
break;
}
}
/**
***
**/
TrCore *
gtr_core_new (tr_session * session)
{
TrCore * core = TR_CORE (g_object_new (TR_CORE_TYPE, NULL));
core->priv->session = session;
/* init from prefs & listen to pref changes */
on_pref_changed (core, TR_KEY_sort_mode, NULL);
on_pref_changed (core, TR_KEY_sort_reversed, NULL);
on_pref_changed (core, TR_KEY_watch_dir_enabled, NULL);
on_pref_changed (core, TR_KEY_peer_limit_global, NULL);
on_pref_changed (core, TR_KEY_inhibit_desktop_hibernation, NULL);
g_signal_connect (core, "prefs-changed", G_CALLBACK (on_pref_changed), NULL);
return core;
}
tr_session *
gtr_core_close (TrCore * core)
{
tr_session * session = gtr_core_session (core);
if (session)
{
core->priv->session = NULL;
gtr_pref_save (session);
}
return session;
}
/***
**** COMPLETENESS CALLBACK
***/
struct notify_callback_data
{
TrCore * core;
int torrent_id;
};
static gboolean
on_torrent_completeness_changed_idle (gpointer gdata)
{
struct notify_callback_data * data = gdata;
gtr_notify_torrent_completed (data->core, data->torrent_id);
g_object_unref (G_OBJECT (data->core));
g_free (data);
return G_SOURCE_REMOVE;
}
/* this is called in the libtransmission thread, *NOT* the GTK+ thread,
so delegate to the GTK+ thread before calling notify's dbus code... */
static void
on_torrent_completeness_changed (tr_torrent * tor,
tr_completeness completeness,
bool was_running,
void * gcore)
{
if (was_running && (completeness != TR_LEECH) && (tr_torrentStat (tor)->sizeWhenDone != 0))
{
struct notify_callback_data * data = g_new (struct notify_callback_data, 1);
data->core = gcore;
data->torrent_id = tr_torrentId (tor);
g_object_ref (G_OBJECT (data->core));
gdk_threads_add_idle (on_torrent_completeness_changed_idle, data);
}
}
/***
**** METADATA CALLBACK
***/
static const char*
get_collated_name (TrCore * core, const tr_torrent * tor)
{
char buf[2048];
const char * name = tr_torrentName (tor);
char * down = g_utf8_strdown (name ? name : "", -1);
const tr_info * inf = tr_torrentInfo (tor);
g_snprintf (buf, sizeof (buf), "%s\t%s", down, inf->hashString);
g_free (down);
return g_string_chunk_insert_const (core->priv->string_chunk, buf);
}
struct metadata_callback_data
{
TrCore * core;
int torrent_id;
};
static gboolean
find_row_from_torrent_id (GtkTreeModel * model, int id, GtkTreeIter * setme)
{
GtkTreeIter iter;
gboolean match = FALSE;
if (gtk_tree_model_iter_children (model, &iter, NULL)) do
{
int row_id;
gtk_tree_model_get (model, &iter, MC_TORRENT_ID, &row_id, -1);
match = id == row_id;
}
while (!match && gtk_tree_model_iter_next (model, &iter));
if (match)
*setme = iter;
return match;
}
static gboolean
on_torrent_metadata_changed_idle (gpointer gdata)
{
struct notify_callback_data * data = gdata;
tr_session * session = gtr_core_session (data->core);
tr_torrent * tor = tr_torrentFindFromId (session, data->torrent_id);
/* update the torrent's collated name */
if (tor != NULL)
{
GtkTreeIter iter;
GtkTreeModel * model = core_raw_model (data->core);
if (find_row_from_torrent_id (model, data->torrent_id, &iter))
{
const char * collated = get_collated_name (data->core, tor);
GtkListStore * store = GTK_LIST_STORE (model);
gtk_list_store_set (store, &iter, MC_NAME_COLLATED, collated, -1);
}
}
/* cleanup */
g_object_unref (G_OBJECT (data->core));
g_free (data);
return G_SOURCE_REMOVE;
}
/* this is called in the libtransmission thread, *NOT* the GTK+ thread,
so delegate to the GTK+ thread before changing our list store... */
static void
on_torrent_metadata_changed (tr_torrent * tor, void * gcore)
{
struct notify_callback_data * data = g_new (struct notify_callback_data, 1);
data->core = gcore;
data->torrent_id = tr_torrentId (tor);
g_object_ref (G_OBJECT (data->core));
gdk_threads_add_idle (on_torrent_metadata_changed_idle, data);
}
/***
****
**** ADDING TORRENTS
****
***/
static unsigned int
build_torrent_trackers_hash (tr_torrent * tor)
{
unsigned int i;
const char * pch;
uint64_t hash = 0;
const tr_info * const inf = tr_torrentInfo (tor);
for (i=0; i<inf->trackerCount; ++i)
for (pch=inf->trackers[i].announce; *pch; ++pch)
hash = (hash<<4) ^ (hash>>28) ^ *pch;
return hash;
}
static gboolean
is_torrent_active (const tr_stat * st)
{
return (st->peersSendingToUs > 0)
|| (st->peersGettingFromUs > 0)
|| (st->activity == TR_STATUS_CHECK);
}
void
gtr_core_add_torrent (TrCore * core, tr_torrent * tor, gboolean do_notify)
{
if (tor != NULL)
{
GtkTreeIter unused;
const tr_stat * st = tr_torrentStat (tor);
const char * collated = get_collated_name (core, tor);
const unsigned int trackers_hash = build_torrent_trackers_hash (tor);
GtkListStore * store = GTK_LIST_STORE (core_raw_model (core));
gtk_list_store_insert_with_values (store, &unused, 0,
MC_NAME_COLLATED, collated,
MC_TORRENT, tor,
MC_TORRENT_ID, tr_torrentId (tor),
MC_SPEED_UP, st->pieceUploadSpeed_KBps,
MC_SPEED_DOWN, st->pieceDownloadSpeed_KBps,
MC_ACTIVE_PEERS_UP, st->peersGettingFromUs,
MC_ACTIVE_PEERS_DOWN, st->peersSendingToUs + st->webseedsSendingToUs,
MC_RECHECK_PROGRESS, st->recheckProgress,
MC_ACTIVE, is_torrent_active (st),
MC_ACTIVITY, st->activity,
MC_FINISHED, st->finished,
MC_PRIORITY, tr_torrentGetPriority (tor),
MC_QUEUE_POSITION, st->queuePosition,
MC_TRACKERS, trackers_hash,
-1);
if (do_notify)
gtr_notify_torrent_added (tr_torrentName (tor));
tr_torrentSetMetadataCallback (tor, on_torrent_metadata_changed, core);
tr_torrentSetCompletenessCallback (tor, on_torrent_completeness_changed, core);
}
}
static tr_torrent *
core_create_new_torrent (TrCore * core, tr_ctor * ctor)
{
tr_torrent * tor;
bool do_trash = false;
tr_session * session = gtr_core_session (core);
/* let the gtk client handle the removal, since libT
* doesn't have any concept of the glib trash API */
tr_ctorGetDeleteSource (ctor, &do_trash);
tr_ctorSetDeleteSource (ctor, FALSE);
tor = tr_torrentNew (ctor, NULL, NULL);
if (tor && do_trash)
{
const char * config = tr_sessionGetConfigDir (session);
const char * source = tr_ctorGetSourceFile (ctor);
if (source != NULL)
{
/* #1294: don't delete the .torrent file if it's our internal copy */
const int is_internal = (strstr (source, config) == source);
if (!is_internal)
gtr_file_trash_or_remove (source);
}
}
return tor;
}
static int
core_add_ctor (TrCore * core, tr_ctor * ctor,
gboolean do_prompt, gboolean do_notify)
{
tr_info inf;
int err = tr_torrentParse (ctor, &inf);
switch (err)
{
case TR_PARSE_ERR:
break;
case TR_PARSE_DUPLICATE:
/* don't complain about .torrent files in the watch directory
* that have already been added... that gets annoying and we
* don't want to be nagging users to clean up their watch dirs */
if (!tr_ctorGetSourceFile (ctor) || !core->priv->adding_from_watch_dir)
core_emit_err (core, err, inf.name);
tr_metainfoFree (&inf);
tr_ctorFree (ctor);
break;
default:
if (do_prompt)
{
g_signal_emit (core, signals[ADD_PROMPT_SIGNAL], 0, ctor);
}
else
{
gtr_core_add_torrent (core, core_create_new_torrent (core, ctor), do_notify);
tr_ctorFree (ctor);
}
tr_metainfoFree (&inf);
break;
}
return err;
}
static void
core_apply_defaults (tr_ctor * ctor)
{
if (tr_ctorGetPaused (ctor, TR_FORCE, NULL))
tr_ctorSetPaused (ctor, TR_FORCE, !gtr_pref_flag_get (TR_KEY_start_added_torrents));
if (tr_ctorGetDeleteSource (ctor, NULL))
tr_ctorSetDeleteSource (ctor, gtr_pref_flag_get (TR_KEY_trash_original_torrent_files));
if (tr_ctorGetPeerLimit (ctor, TR_FORCE, NULL))
tr_ctorSetPeerLimit (ctor, TR_FORCE, gtr_pref_int_get (TR_KEY_peer_limit_per_torrent));
if (tr_ctorGetDownloadDir (ctor, TR_FORCE, NULL))
tr_ctorSetDownloadDir (ctor, TR_FORCE, gtr_pref_string_get (TR_KEY_download_dir));
}
void
gtr_core_add_ctor (TrCore * core, tr_ctor * ctor)
{
const gboolean do_notify = FALSE;
const gboolean do_prompt = gtr_pref_flag_get (TR_KEY_show_options_window);
core_apply_defaults (ctor);
core_add_ctor (core, ctor, do_prompt, do_notify);
}
/***
****
***/
struct add_from_url_data
{
TrCore * core;
tr_ctor * ctor;
bool do_prompt;
bool do_notify;
};
static void
add_file_async_callback (GObject * file, GAsyncResult * result, gpointer gdata)
{
gsize length;
char * contents;
GError * error = NULL;
struct add_from_url_data * data = gdata;
if (!g_file_load_contents_finish (G_FILE (file), result, &contents, &length, NULL, &error))
{
g_message (_("Couldn't read \"%s\": %s"), g_file_get_parse_name (G_FILE (file)), error->message);
g_error_free (error);
}
else if (!tr_ctorSetMetainfo (data->ctor, (const uint8_t*)contents, length))
{
core_add_ctor (data->core, data->ctor, data->do_prompt, data->do_notify);
}
else
{
tr_ctorFree (data->ctor);
}
core_dec_busy (data->core);
g_free (data);
}
static bool
add_file (TrCore * core,
GFile * file,
gboolean do_start,
gboolean do_prompt,
gboolean do_notify)
{
bool handled = false;
tr_session * session = gtr_core_session (core);
if (session != NULL)
{
tr_ctor * ctor;
bool tried = false;
bool loaded = false;
ctor = tr_ctorNew (session);
core_apply_defaults (ctor);
tr_ctorSetPaused (ctor, TR_FORCE, !do_start);
/* local files... */
if (!tried)
{
char * str = g_file_get_path (file);
if ((tried = g_file_test (str, G_FILE_TEST_EXISTS)))
loaded = !tr_ctorSetMetainfoFromFile (ctor, str);
g_free (str);
}
/* magnet links... */
if (!tried && g_file_has_uri_scheme (file, "magnet"))
{
/* GFile mangles the original string with /// so we have to un-mangle */
char * str = g_file_get_parse_name (file);
char * magnet = g_strdup_printf ("magnet:%s", strchr (str, '?'));
tried = true;
loaded = !tr_ctorSetMetainfoFromMagnetLink (ctor, magnet);
g_free (magnet);
g_free (str);
}
/* hashcodes that we can turn into magnet links... */
if (!tried)
{
char * str = g_file_get_basename (file);
if (gtr_is_hex_hashcode (str))
{
char * magnet = g_strdup_printf ("magnet:?xt=urn:btih:%s", str);
tried = true;
loaded = !tr_ctorSetMetainfoFromMagnetLink (ctor, magnet);
g_free (magnet);
}
g_free (str);
}
/* if we were able to load the metainfo, add the torrent */
if (loaded)
{
handled = true;
core_add_ctor (core, ctor, do_prompt, do_notify);
}
else if (g_file_has_uri_scheme (file, "http") ||
g_file_has_uri_scheme (file, "https") ||
g_file_has_uri_scheme (file, "ftp"))
{
struct add_from_url_data * data;
data = g_new0 (struct add_from_url_data, 1);
data->core = core;
data->ctor = ctor;
data->do_prompt = do_prompt;
data->do_notify = do_notify;
handled = true;
core_inc_busy (core);
g_file_load_contents_async (file, NULL, add_file_async_callback, data);
}
else
{
tr_ctorFree (ctor);
g_message (_("Skipping unknown torrent \"%s\""), g_file_get_parse_name (file));
}
}
return handled;
}
bool
gtr_core_add_from_url (TrCore * core, const char * uri)
{
bool handled;
const bool do_start = gtr_pref_flag_get (TR_KEY_start_added_torrents);
const bool do_prompt = gtr_pref_flag_get (TR_KEY_show_options_window);
const bool do_notify = false;
GFile * file = g_file_new_for_uri (uri);
handled = add_file (core, file, do_start, do_prompt, do_notify);
g_object_unref (file);
gtr_core_torrents_added (core);
return handled;
}
void
gtr_core_add_files (TrCore * core,
GSList * files,
gboolean do_start,
gboolean do_prompt,
gboolean do_notify)
{
GSList * l;
for (l=files; l!=NULL; l=l->next)
add_file (core, l->data, do_start, do_prompt, do_notify);
gtr_core_torrents_added (core);
}
void
gtr_core_torrents_added (TrCore * self)
{
gtr_core_update (self);
core_emit_err (self, TR_CORE_ERR_NO_MORE_TORRENTS, NULL);
}
void
gtr_core_remove_torrent (TrCore * core, int id, gboolean delete_local_data)
{
tr_torrent * tor = gtr_core_find_torrent (core, id);
if (tor != NULL)
{
/* remove from the gui */
GtkTreeIter iter;
GtkTreeModel * model = core_raw_model (core);
if (find_row_from_torrent_id (model, id, &iter))
gtk_list_store_remove (GTK_LIST_STORE (model), &iter);
/* remove the torrent */
tr_torrentRemove (tor, delete_local_data, gtr_file_trash_or_remove);
}
}
void
gtr_core_load (TrCore * self, gboolean forcePaused)
{
int i;
tr_ctor * ctor;
tr_torrent ** torrents;
int count = 0;
ctor = tr_ctorNew (gtr_core_session (self));
if (forcePaused)
tr_ctorSetPaused (ctor, TR_FORCE, TRUE);
tr_ctorSetPeerLimit (ctor, TR_FALLBACK,
gtr_pref_int_get (TR_KEY_peer_limit_per_torrent));
torrents = tr_sessionLoadTorrents (gtr_core_session (self), ctor, &count);
for (i=0; i<count; ++i)
gtr_core_add_torrent (self, torrents[i], FALSE);
tr_free (torrents);
tr_ctorFree (ctor);
}
void
gtr_core_clear (TrCore * self)
{
gtk_list_store_clear (GTK_LIST_STORE (core_raw_model (self)));
}
/***
****
***/
static int
gtr_compare_double (const double a, const double b, int decimal_places)
{
int ret;
const int64_t ia = (int64_t)(a * pow (10, decimal_places));
const int64_t ib = (int64_t)(b * pow (10, decimal_places));
if (ia < ib)
ret = -1;
else if (ia > ib)
ret = 1;
else
ret = 0;
return ret;
}
static void
update_foreach (GtkTreeModel * model, GtkTreeIter * iter)
{
int oldActivity, newActivity;
int oldActivePeerCount, newActivePeerCount;
int oldError, newError;
bool oldFinished, newFinished;
int oldQueuePosition, newQueuePosition;
int oldDownloadPeerCount, newDownloadPeerCount;
int oldUploadPeerCount, newUploadPeerCount;
tr_priority_t oldPriority, newPriority;
unsigned int oldTrackers, newTrackers;
double oldUpSpeed, newUpSpeed;
double oldDownSpeed, newDownSpeed;
double oldRecheckProgress, newRecheckProgress;
gboolean oldActive, newActive;
const tr_stat * st;
tr_torrent * tor;
/* get the old states */
gtk_tree_model_get (model, iter,
MC_TORRENT, &tor,
MC_ACTIVE, &oldActive,
MC_ACTIVE_PEER_COUNT, &oldActivePeerCount,
MC_ACTIVE_PEERS_UP, &oldUploadPeerCount,
MC_ACTIVE_PEERS_DOWN, &oldDownloadPeerCount,
MC_ERROR, &oldError,
MC_ACTIVITY, &oldActivity,
MC_FINISHED, &oldFinished,
MC_PRIORITY, &oldPriority,
MC_QUEUE_POSITION, &oldQueuePosition,
MC_TRACKERS, &oldTrackers,
MC_SPEED_UP, &oldUpSpeed,
MC_RECHECK_PROGRESS, &oldRecheckProgress,
MC_SPEED_DOWN, &oldDownSpeed,
-1);
/* get the new states */
st = tr_torrentStat (tor);
newActive = is_torrent_active (st);
newActivity = st->activity;
newFinished = st->finished;
newPriority = tr_torrentGetPriority (tor);
newQueuePosition = st->queuePosition;
newTrackers = build_torrent_trackers_hash (tor);
newUpSpeed = st->pieceUploadSpeed_KBps;
newDownSpeed = st->pieceDownloadSpeed_KBps;
newRecheckProgress = st->recheckProgress;
newActivePeerCount = st->peersSendingToUs + st->peersGettingFromUs + st->webseedsSendingToUs;
newDownloadPeerCount = st->peersSendingToUs;
newUploadPeerCount = st->peersGettingFromUs + st->webseedsSendingToUs;
newError = st->error;
/* updating the model triggers off resort/refresh,
so don't do it unless something's actually changed... */
if ((newActive != oldActive)
|| (newActivity != oldActivity)
|| (newFinished != oldFinished)
|| (newPriority != oldPriority)
|| (newQueuePosition != oldQueuePosition)
|| (newError != oldError)
|| (newActivePeerCount != oldActivePeerCount)
|| (newDownloadPeerCount != oldDownloadPeerCount)
|| (newUploadPeerCount != oldUploadPeerCount)
|| (newTrackers != oldTrackers)
|| gtr_compare_double (newUpSpeed, oldUpSpeed, 2)
|| gtr_compare_double (newDownSpeed, oldDownSpeed, 2)
|| gtr_compare_double (newRecheckProgress, oldRecheckProgress, 2))
{
gtk_list_store_set (GTK_LIST_STORE (model), iter,
MC_ACTIVE, newActive,
MC_ACTIVE_PEER_COUNT, newActivePeerCount,
MC_ACTIVE_PEERS_UP, newUploadPeerCount,
MC_ACTIVE_PEERS_DOWN, newDownloadPeerCount,
MC_ERROR, newError,
MC_ACTIVITY, newActivity,
MC_FINISHED, newFinished,
MC_PRIORITY, newPriority,
MC_QUEUE_POSITION, newQueuePosition,
MC_TRACKERS, newTrackers,
MC_SPEED_UP, newUpSpeed,
MC_SPEED_DOWN, newDownSpeed,
MC_RECHECK_PROGRESS, newRecheckProgress,
-1);
}
}
void
gtr_core_update (TrCore * core)
{
GtkTreeIter iter;
GtkTreeModel * model;
/* update the model */
model = core_raw_model (core);
if (gtk_tree_model_iter_nth_child (model, &iter, NULL, 0)) do
update_foreach (model, &iter);
while (gtk_tree_model_iter_next (model, &iter));
/* update hibernation */
core_maybe_inhibit_hibernation (core);
}
/**
*** Hibernate
**/
#define SESSION_MANAGER_SERVICE_NAME "org.gnome.SessionManager"
#define SESSION_MANAGER_INTERFACE "org.gnome.SessionManager"
#define SESSION_MANAGER_OBJECT_PATH "/org/gnome/SessionManager"
static gboolean
gtr_inhibit_hibernation (guint * cookie)
{
gboolean success;
GVariant * response;
GDBusConnection * connection;
GError * err = NULL;
const char * application = "Transmission BitTorrent Client";
const char * reason = "BitTorrent Activity";
const int toplevel_xid = 0;
const int flags = 4; /* Inhibit suspending the session or computer */
connection = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &err);
response = g_dbus_connection_call_sync (connection,
SESSION_MANAGER_SERVICE_NAME,
SESSION_MANAGER_OBJECT_PATH,
SESSION_MANAGER_INTERFACE,
"Inhibit",
g_variant_new ("(susu)", application, toplevel_xid, reason, flags),
NULL, G_DBUS_CALL_FLAGS_NONE,
1000, NULL, &err);
if (response != NULL)
*cookie = g_variant_get_uint32 (g_variant_get_child_value (response, 0));
success = (response != NULL) && (err == NULL);
/* logging */
if (success)
{
tr_logAddInfo ("%s", _("Inhibiting desktop hibernation"));
}
else
{
tr_logAddError (_("Couldn't inhibit desktop hibernation: %s"), err->message);
g_error_free (err);
}
/* cleanup */
if (response != NULL)
g_variant_unref (response);
if (connection != NULL)
g_object_unref (connection);
return success;
}
static void
gtr_uninhibit_hibernation (guint inhibit_cookie)
{
GVariant * response;
GDBusConnection * connection;
GError * err = NULL;
connection = g_bus_get_sync (G_BUS_TYPE_SESSION, NULL, &err);
response = g_dbus_connection_call_sync (connection,
SESSION_MANAGER_SERVICE_NAME,
SESSION_MANAGER_OBJECT_PATH,
SESSION_MANAGER_INTERFACE,
"Uninhibit",
g_variant_new ("(u)", inhibit_cookie),
NULL, G_DBUS_CALL_FLAGS_NONE,
1000, NULL, &err);
/* logging */
if (err == NULL)
{
tr_logAddInfo ("%s", _("Allowing desktop hibernation"));
}
else
{
g_warning ("Couldn't uninhibit desktop hibernation: %s.", err->message);
g_error_free (err);
}
/* cleanup */
g_variant_unref (response);
g_object_unref (connection);
}
static void
gtr_core_set_hibernation_allowed (TrCore * core, gboolean allowed)
{
g_return_if_fail (core);
g_return_if_fail (core->priv);
core->priv->inhibit_allowed = allowed != 0;
if (allowed && core->priv->have_inhibit_cookie)
{
gtr_uninhibit_hibernation (core->priv->inhibit_cookie);
core->priv->have_inhibit_cookie = FALSE;
}
if (!allowed && !core->priv->have_inhibit_cookie
&& !core->priv->dbus_error)
{
if (gtr_inhibit_hibernation (&core->priv->inhibit_cookie))
core->priv->have_inhibit_cookie = TRUE;
else
core->priv->dbus_error = TRUE;
}
}
static void
core_maybe_inhibit_hibernation (TrCore * core)
{
/* hibernation is allowed if EITHER
* (a) the "inhibit" pref is turned off OR
* (b) there aren't any active torrents */
const gboolean hibernation_allowed = !gtr_pref_flag_get (TR_KEY_inhibit_desktop_hibernation)
|| !gtr_core_get_active_torrent_count (core);
gtr_core_set_hibernation_allowed (core, hibernation_allowed);
}
/**
*** Prefs
**/
static void
core_commit_prefs_change (TrCore * core, const tr_quark key)
{
gtr_core_pref_changed (core, key);
gtr_pref_save (gtr_core_session (core));
}
void
gtr_core_set_pref (TrCore * self, const tr_quark key, const char * newval)
{
if (g_strcmp0 (newval, gtr_pref_string_get (key)))
{
gtr_pref_string_set (key, newval);
core_commit_prefs_change (self, key);
}
}
void
gtr_core_set_pref_bool (TrCore * self, const tr_quark key, gboolean newval)
{
if (newval != gtr_pref_flag_get (key))
{
gtr_pref_flag_set (key, newval);
core_commit_prefs_change (self, key);
}
}
void
gtr_core_set_pref_int (TrCore * self, const tr_quark key, int newval)
{
if (newval != gtr_pref_int_get (key))
{
gtr_pref_int_set (key, newval);
core_commit_prefs_change (self, key);
}
}
void
gtr_core_set_pref_double (TrCore * self, const tr_quark key, double newval)
{
if (gtr_compare_double (newval, gtr_pref_double_get (key), 4))
{
gtr_pref_double_set (key, newval);
core_commit_prefs_change (self, key);
}
}
/***
****
**** RPC Interface
****
***/
/* #define DEBUG_RPC */
static int nextTag = 1;
typedef void (server_response_func)(TrCore * core, tr_variant * response, gpointer user_data);
struct pending_request_data
{
TrCore * core;
server_response_func * response_func;
gpointer response_func_user_data;
};
static GHashTable * pendingRequests = NULL;
static gboolean
core_read_rpc_response_idle (void * vresponse)
{
tr_variant top;
int64_t intVal;
struct evbuffer * response = vresponse;
tr_variantFromJson (&top, evbuffer_pullup (response, -1), evbuffer_get_length (response));
if (tr_variantDictFindInt (&top, TR_KEY_tag, &intVal))
{
const int tag = (int)intVal;
struct pending_request_data * data = g_hash_table_lookup (pendingRequests, &tag);
if (data)
{
if (data->response_func)
(*data->response_func)(data->core, &top, data->response_func_user_data);
g_hash_table_remove (pendingRequests, &tag);
}
}
tr_variantFree (&top);
evbuffer_free (response);
return G_SOURCE_REMOVE;
}
static void
core_read_rpc_response (tr_session * session UNUSED,
struct evbuffer * response,
void * unused UNUSED)
{
struct evbuffer * buf = evbuffer_new ();
evbuffer_add_buffer (buf, response);
gdk_threads_add_idle (core_read_rpc_response_idle, buf);
}
static void
core_send_rpc_request (TrCore * core, const char * json, int tag,
server_response_func * response_func,
void * response_func_user_data)
{
tr_session * session = gtr_core_session (core);
if (pendingRequests == NULL)
{
pendingRequests = g_hash_table_new_full (g_int_hash, g_int_equal, g_free, g_free);
}
if (session == NULL)
{
g_error ("GTK+ client doesn't support connections to remote servers yet.");
}
else
{
/* remember this request */
struct pending_request_data * data;
data = g_new0 (struct pending_request_data, 1);
data->core = core;
data->response_func = response_func;
data->response_func_user_data = response_func_user_data;
g_hash_table_insert (pendingRequests, g_memdup (&tag, sizeof (int)), data);
/* make the request */
#ifdef DEBUG_RPC
g_message ("request: [%s]", json);
#endif
tr_rpc_request_exec_json (session, json, strlen (json), core_read_rpc_response, GINT_TO_POINTER (tag));
}
}
/***
**** Sending a test-port request via RPC
***/
static void
on_port_test_response (TrCore * core, tr_variant * response, gpointer u UNUSED)
{
tr_variant * args;
bool is_open = FALSE;
if (tr_variantDictFindDict (response, TR_KEY_arguments, &args))
tr_variantDictFindBool (args, TR_KEY_port_is_open, &is_open);
core_emit_port_tested (core, is_open);
}
void
gtr_core_port_test (TrCore * core)
{
char buf[64];
const int tag = nextTag++;
g_snprintf (buf, sizeof (buf), "{ \"method\": \"port-test\", \"tag\": %d }", tag);
core_send_rpc_request (core, buf, tag, on_port_test_response, NULL);
}
/***
**** Updating a blocklist via RPC
***/
static void
on_blocklist_response (TrCore * core, tr_variant * response, gpointer data UNUSED)
{
tr_variant * args;
int64_t ruleCount = -1;
if (tr_variantDictFindDict (response, TR_KEY_arguments, &args))
tr_variantDictFindInt (args, TR_KEY_blocklist_size, &ruleCount);
if (ruleCount > 0)
gtr_pref_int_set (TR_KEY_blocklist_date, tr_time ());
core_emit_blocklist_udpated (core, ruleCount);
}
void
gtr_core_blocklist_update (TrCore * core)
{
char buf[64];
const int tag = nextTag++;
g_snprintf (buf, sizeof (buf), "{ \"method\": \"blocklist-update\", \"tag\": %d }", tag);
core_send_rpc_request (core, buf, tag, on_blocklist_response, NULL);
}
/***
****
***/
void
gtr_core_exec_json (TrCore * core, const char * json)
{
const int tag = nextTag++;
core_send_rpc_request (core, json, tag, NULL, NULL);
}
void
gtr_core_exec (TrCore * core, const tr_variant * top)
{
char * json = tr_variantToStr (top, TR_VARIANT_FMT_JSON_LEAN, NULL);
gtr_core_exec_json (core, json);
tr_free (json);
}
/***
****
***/
size_t
gtr_core_get_torrent_count (TrCore * core)
{
return gtk_tree_model_iter_n_children (core_raw_model (core), NULL);
}
size_t
gtr_core_get_active_torrent_count (TrCore * core)
{
GtkTreeIter iter;
size_t activeCount = 0;
GtkTreeModel * model = core_raw_model (core);
if (gtk_tree_model_iter_nth_child (model, &iter, NULL, 0)) do
{
int activity;
gtk_tree_model_get (model, &iter, MC_ACTIVITY, &activity, -1);
if (activity != TR_STATUS_STOPPED)
++activeCount;
}
while (gtk_tree_model_iter_next (model, &iter));
return activeCount;
}
tr_torrent *
gtr_core_find_torrent (TrCore * core, int id)
{
tr_session * session;
tr_torrent * tor = NULL;
if ((session = gtr_core_session (core)))
tor = tr_torrentFindFromId (session, id);
return tor;
}
void
gtr_core_open_folder (TrCore * core, int torrent_id)
{
const tr_torrent * tor = gtr_core_find_torrent (core, torrent_id);
if (tor != NULL)
{
const gboolean single = tr_torrentInfo (tor)->fileCount == 1;
const char * currentDir = tr_torrentGetCurrentDir (tor);
if (single)
{
gtr_open_file (currentDir);
}
else
{
char * path = g_build_filename (currentDir, tr_torrentName (tor), NULL);
gtr_open_file (path);
g_free (path);
}
}
}
Reference in a new issue View git blame Copy permalink