void
tr_dhtUninit (tr_session *ss)
{
if (session != ss)
return;
tr_logAddNamedDbg ("DHT", "Uninitializing DHT");
if (dht_timer != NULL) {
event_free (dht_timer);
dht_timer = NULL;
}
/* Since we only save known good nodes, avoid erasing older data if we
don't know enough nodes. */
if ((tr_dhtStatus (ss, AF_INET, NULL) < TR_DHT_FIREWALLED) &&
(tr_dhtStatus (ss, AF_INET6, NULL) < TR_DHT_FIREWALLED)) {
tr_logAddNamedInfo ("DHT", "Not saving nodes, DHT not ready");
} else {
tr_variant benc;
struct sockaddr_in sins[300];
struct sockaddr_in6 sins6[300];
char compact[300 * 6], compact6[300 * 18];
char *dat_file;
int i, j, num = 300, num6 = 300;
int n = dht_get_nodes (sins, &num, sins6, &num6);
tr_logAddNamedInfo ("DHT", "Saving %d (%d + %d) nodes", n, num, num6);
j = 0;
for (i=0; i<num; ++i) {
memcpy (compact + j, &sins[i].sin_addr, 4);
memcpy (compact + j + 4, &sins[i].sin_port, 2);
j += 6;
}
j = 0;
for (i=0; i<num6; ++i) {
memcpy (compact6 + j, &sins6[i].sin6_addr, 16);
memcpy (compact6 + j + 16, &sins6[i].sin6_port, 2);
j += 18;
}
tr_variantInitDict (&benc, 3);
tr_variantDictAddRaw (&benc, TR_KEY_id, myid, 20);
if (num > 0)
tr_variantDictAddRaw (&benc, TR_KEY_nodes, compact, num * 6);
if (num6 > 0)
tr_variantDictAddRaw (&benc, TR_KEY_nodes6, compact6, num6 * 18);
dat_file = tr_buildPath (ss->configDir, "dht.dat", NULL);
tr_variantToFile (&benc, TR_VARIANT_FMT_BENC, dat_file);
tr_variantFree (&benc);
tr_free (dat_file);
}
dht_uninit ();
tr_logAddNamedDbg ("DHT", "Done uninitializing DHT");
session = NULL;
}
void tr_torrentSaveResume(tr_torrent* tor)
{
int err;
tr_variant top;
char* filename;
if (!tr_isTorrent(tor))
{
return;
}
tr_variantInitDict(&top, 50); /* arbitrary "big enough" number */
tr_variantDictAddInt(&top, TR_KEY_seeding_time_seconds, tor->secondsSeeding);
tr_variantDictAddInt(&top, TR_KEY_downloading_time_seconds, tor->secondsDownloading);
tr_variantDictAddInt(&top, TR_KEY_activity_date, tor->activityDate);
tr_variantDictAddInt(&top, TR_KEY_added_date, tor->addedDate);
tr_variantDictAddInt(&top, TR_KEY_corrupt, tor->corruptPrev + tor->corruptCur);
tr_variantDictAddInt(&top, TR_KEY_done_date, tor->doneDate);
tr_variantDictAddStr(&top, TR_KEY_destination, tor->downloadDir);
if (tor->incompleteDir != NULL)
{
tr_variantDictAddStr(&top, TR_KEY_incomplete_dir, tor->incompleteDir);
}
tr_variantDictAddInt(&top, TR_KEY_downloaded, tor->downloadedPrev + tor->downloadedCur);
tr_variantDictAddInt(&top, TR_KEY_uploaded, tor->uploadedPrev + tor->uploadedCur);
tr_variantDictAddInt(&top, TR_KEY_max_peers, tor->maxConnectedPeers);
tr_variantDictAddInt(&top, TR_KEY_bandwidth_priority, tr_torrentGetPriority(tor));
tr_variantDictAddBool(&top, TR_KEY_paused, !tor->isRunning && !tor->isQueued);
savePeers(&top, tor);
if (tr_torrentHasMetadata(tor))
{
saveFilePriorities(&top, tor);
saveDND(&top, tor);
saveProgress(&top, tor);
}
saveSpeedLimits(&top, tor);
saveRatioLimits(&top, tor);
saveIdleLimits(&top, tor);
saveFilenames(&top, tor);
saveName(&top, tor);
filename = getResumeFilename(tor);
if ((err = tr_variantToFile(&top, TR_VARIANT_FMT_BENC, filename)) != 0)
{
tr_torrentSetLocalError(tor, "Unable to save resume file: %s", tr_strerror(err));
}
tr_free(filename);
tr_variantFree(&top);
}
static void saveCumulativeStats(tr_session const* session, tr_session_stats const* s)
{
char* filename;
tr_variant top;
tr_variantInitDict(&top, 5);
tr_variantDictAddInt(&top, TR_KEY_downloaded_bytes, s->downloadedBytes);
tr_variantDictAddInt(&top, TR_KEY_files_added, s->filesAdded);
tr_variantDictAddInt(&top, TR_KEY_seconds_active, s->secondsActive);
tr_variantDictAddInt(&top, TR_KEY_session_count, s->sessionCount);
tr_variantDictAddInt(&top, TR_KEY_uploaded_bytes, s->uploadedBytes);
filename = getFilename(session);
tr_logAddDeep(__FILE__, __LINE__, NULL, "Saving stats to \"%s\"", filename);
tr_variantToFile(&top, TR_VARIANT_FMT_JSON, filename);
tr_free(filename);
tr_variantFree(&top);
}
int
main (int argc, char * argv[])
{
int i;
int changedCount = 0;
#ifdef _WIN32
tr_win32_make_args_utf8 (&argc, &argv);
#endif
files = tr_new0 (const char*, argc);
tr_logSetLevel (TR_LOG_ERROR);
if (parseCommandLine (argc, (const char**)argv))
return EXIT_FAILURE;
if (showVersion)
{
fprintf (stderr, MY_NAME" "LONG_VERSION_STRING"\n");
return EXIT_SUCCESS;
}
if (fileCount < 1)
{
fprintf (stderr, "ERROR: No torrent files specified.\n");
tr_getopt_usage (MY_NAME, getUsage (), options);
fprintf (stderr, "\n");
return EXIT_FAILURE;
}
if (!add && !deleteme && !replace[0])
{
fprintf (stderr, "ERROR: Must specify -a, -d or -r\n");
tr_getopt_usage (MY_NAME, getUsage (), options);
fprintf (stderr, "\n");
return EXIT_FAILURE;
}
for (i=0; i<fileCount; ++i)
{
tr_variant top;
bool changed = false;
const char * filename = files[i];
printf ("%s\n", filename);
if (tr_variantFromFile (&top, TR_VARIANT_FMT_BENC, filename))
{
printf ("\tError reading file\n");
continue;
}
if (deleteme != NULL)
changed |= removeURL (&top, deleteme);
if (add != NULL)
changed = addURL (&top, add);
if (replace[0] && replace[1])
changed |= replaceURL (&top, replace[0], replace[1]);
if (changed)
{
++changedCount;
tr_variantToFile (&top, TR_VARIANT_FMT_BENC, filename);
}
tr_variantFree (&top);
}
printf ("Changed %d files\n", changedCount);
tr_free (files);
return EXIT_SUCCESS;
}
