source: trunk/daemon/daemon.c @ 14581

Last change on this file since 14581 was 14581, checked in by mikedld, 6 years ago

Remove unused session tag

  • Property svn:keywords set to Date Rev Author Id
File size: 23.9 KB
Line 
1/*
2 * This file Copyright (C) 2008-2014 Mnemosyne LLC
3 *
4 * It may be used under the GNU GPL versions 2 or 3
5 * or any future license endorsed by Mnemosyne LLC.
6 *
7 * $Id: daemon.c 14581 2015-10-18 18:39:14Z mikedld $
8 */
9
10#include <errno.h>
11#include <stdio.h> /* printf */
12#include <stdlib.h> /* atoi */
13
14#ifdef HAVE_SYSLOG
15#include <syslog.h>
16#endif
17
18#ifdef _WIN32
19 #include <process.h> /* getpid */
20#else
21 #include <unistd.h> /* getpid */
22#endif
23
24#include <event2/buffer.h>
25#include <event2/event.h>
26
27#include <libtransmission/transmission.h>
28#include <libtransmission/error.h>
29#include <libtransmission/file.h>
30#include <libtransmission/tr-getopt.h>
31#include <libtransmission/log.h>
32#include <libtransmission/utils.h>
33#include <libtransmission/variant.h>
34#include <libtransmission/version.h>
35
36#ifdef USE_SYSTEMD_DAEMON
37 #include <systemd/sd-daemon.h>
38#else
39 static void sd_notify (int status UNUSED, const char * str UNUSED) { }
40 static void sd_notifyf (int status UNUSED, const char * fmt UNUSED, ...) { }
41#endif
42
43#include "daemon.h"
44#include "watch.h"
45
46#define MY_NAME "transmission-daemon"
47
48#define MEM_K 1024
49#define MEM_K_STR "KiB"
50#define MEM_M_STR "MiB"
51#define MEM_G_STR "GiB"
52#define MEM_T_STR "TiB"
53
54#define DISK_K 1000
55#define DISK_B_STR  "B"
56#define DISK_K_STR "kB"
57#define DISK_M_STR "MB"
58#define DISK_G_STR "GB"
59#define DISK_T_STR "TB"
60
61#define SPEED_K 1000
62#define SPEED_B_STR  "B/s"
63#define SPEED_K_STR "kB/s"
64#define SPEED_M_STR "MB/s"
65#define SPEED_G_STR "GB/s"
66#define SPEED_T_STR "TB/s"
67
68static bool seenHUP = false;
69static const char *logfileName = NULL;
70static tr_sys_file_t logfile = TR_BAD_SYS_FILE;
71static tr_session * mySession = NULL;
72static tr_quark key_pidfile = 0;
73static struct event_base *ev_base = NULL;
74
75/***
76****  Config File
77***/
78
79static const char *
80getUsage (void)
81{
82    return "Transmission " LONG_VERSION_STRING
83           "  http://www.transmissionbt.com/\n"
84           "A fast and easy BitTorrent client\n"
85           "\n"
86           MY_NAME " is a headless Transmission session\n"
87           "that can be controlled via transmission-remote\n"
88           "or the web interface.\n"
89           "\n"
90           "Usage: " MY_NAME " [options]";
91}
92
93static const struct tr_option options[] =
94{
95
96    { 'a', "allowed", "Allowed IP addresses. (Default: " TR_DEFAULT_RPC_WHITELIST ")", "a", 1, "<list>" },
97    { 'b', "blocklist", "Enable peer blocklists", "b", 0, NULL },
98    { 'B', "no-blocklist", "Disable peer blocklists", "B", 0, NULL },
99    { 'c', "watch-dir", "Where to watch for new .torrent files", "c", 1, "<directory>" },
100    { 'C', "no-watch-dir", "Disable the watch-dir", "C", 0, NULL },
101    { 941, "incomplete-dir", "Where to store new torrents until they're complete", NULL, 1, "<directory>" },
102    { 942, "no-incomplete-dir", "Don't store incomplete torrents in a different location", NULL, 0, NULL },
103    { 'd', "dump-settings", "Dump the settings and exit", "d", 0, NULL },
104    { 'e', "logfile", "Dump the log messages to this filename", "e", 1, "<filename>" },
105    { 'f', "foreground", "Run in the foreground instead of daemonizing", "f", 0, NULL },
106    { 'g', "config-dir", "Where to look for configuration files", "g", 1, "<path>" },
107    { 'p', "port", "RPC port (Default: " TR_DEFAULT_RPC_PORT_STR ")", "p", 1, "<port>" },
108    { 't', "auth", "Require authentication", "t", 0, NULL },
109    { 'T', "no-auth", "Don't require authentication", "T", 0, NULL },
110    { 'u', "username", "Set username for authentication", "u", 1, "<username>" },
111    { 'v', "password", "Set password for authentication", "v", 1, "<password>" },
112    { 'V', "version", "Show version number and exit", "V", 0, NULL },
113    { 810, "log-error", "Show error messages", NULL, 0, NULL },
114    { 811, "log-info", "Show error and info messages", NULL, 0, NULL },
115    { 812, "log-debug", "Show error, info, and debug messages", NULL, 0, NULL },
116    { 'w', "download-dir", "Where to save downloaded data", "w", 1, "<path>" },
117    { 800, "paused", "Pause all torrents on startup", NULL, 0, NULL },
118    { 'o', "dht", "Enable distributed hash tables (DHT)", "o", 0, NULL },
119    { 'O', "no-dht", "Disable distributed hash tables (DHT)", "O", 0, NULL },
120    { 'y', "lpd", "Enable local peer discovery (LPD)", "y", 0, NULL },
121    { 'Y', "no-lpd", "Disable local peer discovery (LPD)", "Y", 0, NULL },
122    { 830, "utp", "Enable uTP for peer connections", NULL, 0, NULL },
123    { 831, "no-utp", "Disable uTP for peer connections", NULL, 0, NULL },
124    { 'P', "peerport", "Port for incoming peers (Default: " TR_DEFAULT_PEER_PORT_STR ")", "P", 1, "<port>" },
125    { 'm', "portmap", "Enable portmapping via NAT-PMP or UPnP", "m", 0, NULL },
126    { 'M', "no-portmap", "Disable portmapping", "M", 0, NULL },
127    { 'L', "peerlimit-global", "Maximum overall number of peers (Default: " TR_DEFAULT_PEER_LIMIT_GLOBAL_STR ")", "L", 1, "<limit>" },
128    { 'l', "peerlimit-torrent", "Maximum number of peers per torrent (Default: " TR_DEFAULT_PEER_LIMIT_TORRENT_STR ")", "l", 1, "<limit>" },
129    { 910, "encryption-required",  "Encrypt all peer connections", "er", 0, NULL },
130    { 911, "encryption-preferred", "Prefer encrypted peer connections", "ep", 0, NULL },
131    { 912, "encryption-tolerated", "Prefer unencrypted peer connections", "et", 0, NULL },
132    { 'i', "bind-address-ipv4", "Where to listen for peer connections", "i", 1, "<ipv4 addr>" },
133    { 'I', "bind-address-ipv6", "Where to listen for peer connections", "I", 1, "<ipv6 addr>" },
134    { 'r', "rpc-bind-address", "Where to listen for RPC connections", "r", 1, "<ipv4 addr>" },
135    { 953, "global-seedratio", "All torrents, unless overridden by a per-torrent setting, should seed until a specific ratio", "gsr", 1, "ratio" },
136    { 954, "no-global-seedratio", "All torrents, unless overridden by a per-torrent setting, should seed regardless of ratio", "GSR", 0, NULL },
137    { 'x', "pid-file", "Enable PID file", "x", 1, "<pid-file>" },
138    { 0, NULL, NULL, NULL, 0, NULL }
139};
140
141static bool
142reopen_log_file (const char *filename)
143{
144    tr_error * error = NULL;
145    const tr_sys_file_t old_log_file = logfile;
146    const tr_sys_file_t new_log_file = tr_sys_file_open (filename,
147                                                         TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_APPEND,
148                                                         0666, &error);
149
150    if (new_log_file == TR_BAD_SYS_FILE)
151    {
152        fprintf (stderr, "Couldn't (re)open log file \"%s\": %s\n", filename, error->message);
153        tr_error_free (error);
154        return false;
155    }
156
157    logfile = new_log_file;
158
159    if (old_log_file != TR_BAD_SYS_FILE)
160        tr_sys_file_close (old_log_file, NULL);
161
162    return true;
163}
164
165static const char*
166getConfigDir (int argc, const char ** argv)
167{
168    int c;
169    const char * configDir = NULL;
170    const char * optarg;
171    const int ind = tr_optind;
172
173    while ((c = tr_getopt (getUsage (), argc, argv, options, &optarg))) {
174        if (c == 'g') {
175            configDir = optarg;
176            break;
177        }
178    }
179
180    tr_optind = ind;
181
182    if (configDir == NULL)
183        configDir = tr_getDefaultConfigDir (MY_NAME);
184
185    return configDir;
186}
187
188static void
189onFileAdded (tr_session * session, const char * dir, const char * file)
190{
191    char * filename = tr_buildPath (dir, file, NULL);
192    tr_ctor * ctor = tr_ctorNew (session);
193    int err = tr_ctorSetMetainfoFromFile (ctor, filename);
194
195    if (!err)
196    {
197        tr_torrentNew (ctor, &err, NULL);
198
199        if (err == TR_PARSE_ERR)
200            tr_logAddError ("Error parsing .torrent file \"%s\"", file);
201        else
202        {
203            bool trash = false;
204            const bool test = tr_ctorGetDeleteSource (ctor, &trash);
205
206            tr_logAddInfo ("Parsing .torrent file successful \"%s\"", file);
207
208            if (test && trash)
209            {
210                tr_error * error = NULL;
211
212                tr_logAddInfo ("Deleting input .torrent file \"%s\"", file);
213                if (!tr_sys_path_remove (filename, &error))
214                {
215                    tr_logAddError ("Error deleting .torrent file: %s", error->message);
216                    tr_error_free (error);
217                }
218            }
219            else
220            {
221                char * new_filename = tr_strdup_printf ("%s.added", filename);
222                tr_sys_path_rename (filename, new_filename, NULL);
223                tr_free (new_filename);
224            }
225        }
226    }
227
228    tr_ctorFree (ctor);
229    tr_free (filename);
230}
231
232static void
233printMessage (tr_sys_file_t logfile, int level, const char * name, const char * message, const char * file, int line)
234{
235    if (logfile != TR_BAD_SYS_FILE)
236    {
237        char timestr[64];
238        tr_logGetTimeStr (timestr, sizeof (timestr));
239        if (name)
240            tr_sys_file_write_fmt (logfile, "[%s] %s %s (%s:%d)" TR_NATIVE_EOL_STR,
241                                   NULL, timestr, name, message, file, line);
242        else
243            tr_sys_file_write_fmt (logfile, "[%s] %s (%s:%d)" TR_NATIVE_EOL_STR,
244                                   NULL, timestr, message, file, line);
245    }
246#ifdef HAVE_SYSLOG
247    else /* daemon... write to syslog */
248    {
249        int priority;
250
251        /* figure out the syslog priority */
252        switch (level) {
253            case TR_LOG_ERROR: priority = LOG_ERR; break;
254            case TR_LOG_DEBUG: priority = LOG_DEBUG; break;
255            default:           priority = LOG_INFO; break;
256        }
257
258        if (name)
259            syslog (priority, "%s %s (%s:%d)", name, message, file, line);
260        else
261            syslog (priority, "%s (%s:%d)", message, file, line);
262    }
263#else
264    (void) level;
265#endif
266}
267
268static void
269pumpLogMessages (tr_sys_file_t logfile)
270{
271    const tr_log_message * l;
272    tr_log_message * list = tr_logGetQueue ();
273
274    for (l=list; l!=NULL; l=l->next)
275        printMessage (logfile, l->level, l->name, l->message, l->file, l->line);
276
277    if (logfile != TR_BAD_SYS_FILE)
278        tr_sys_file_flush (logfile, NULL);
279
280    tr_logFreeQueue (list);
281}
282
283static void
284reportStatus (void)
285{
286    const double up = tr_sessionGetRawSpeed_KBps (mySession, TR_UP);
287    const double dn = tr_sessionGetRawSpeed_KBps (mySession, TR_DOWN);
288
289    if (up>0 || dn>0)
290        sd_notifyf (0, "STATUS=Uploading %.2f KBps, Downloading %.2f KBps.\n", up, dn);
291    else
292        sd_notify (0, "STATUS=Idle.\n");
293}
294
295static void
296periodicUpdate (evutil_socket_t fd UNUSED, short what UNUSED, void *watchdir)
297{
298    dtr_watchdir_update (watchdir);
299
300    pumpLogMessages (logfile);
301
302    reportStatus ();
303}
304
305static tr_rpc_callback_status
306on_rpc_callback (tr_session            * session UNUSED,
307                 tr_rpc_callback_type    type,
308                 struct tr_torrent     * tor UNUSED,
309                 void                  * user_data UNUSED)
310{
311    if (type == TR_RPC_SESSION_CLOSE)
312        event_base_loopexit(ev_base, NULL);
313    return TR_RPC_OK;
314}
315
316static bool
317parse_args (int           argc,
318            const char ** argv,
319            tr_variant  * settings,
320            bool        * paused,
321            bool        * dump_settings,
322            bool        * foreground,
323            int         * exit_code)
324{
325    int c;
326    const char * optarg;
327
328    *paused = false;
329    *dump_settings = false;
330    *foreground = false;
331
332    tr_optind = 1;
333    while ((c = tr_getopt (getUsage (), argc, argv, options, &optarg))) {
334        switch (c) {
335            case 'a': tr_variantDictAddStr  (settings, TR_KEY_rpc_whitelist, optarg);
336                      tr_variantDictAddBool (settings, TR_KEY_rpc_whitelist_enabled, true);
337                      break;
338            case 'b': tr_variantDictAddBool (settings, TR_KEY_blocklist_enabled, true);
339                      break;
340            case 'B': tr_variantDictAddBool (settings, TR_KEY_blocklist_enabled, false);
341                      break;
342            case 'c': tr_variantDictAddStr  (settings, TR_KEY_watch_dir, optarg);
343                      tr_variantDictAddBool (settings, TR_KEY_watch_dir_enabled, true);
344                      break;
345            case 'C': tr_variantDictAddBool (settings, TR_KEY_watch_dir_enabled, false);
346                      break;
347            case 941: tr_variantDictAddStr  (settings, TR_KEY_incomplete_dir, optarg);
348                      tr_variantDictAddBool (settings, TR_KEY_incomplete_dir_enabled, true);
349                      break;
350            case 942: tr_variantDictAddBool (settings, TR_KEY_incomplete_dir_enabled, false);
351                      break;
352            case 'd': *dump_settings = true;
353                      break;
354            case 'e': if (reopen_log_file (optarg))
355                          logfileName = optarg;
356                      break;
357            case 'f': *foreground = true;
358                      break;
359            case 'g': /* handled above */
360                      break;
361            case 'V': /* version */
362                      fprintf (stderr, "%s %s\n", MY_NAME, LONG_VERSION_STRING);
363                      *exit_code = 0;
364                      return false;
365            case 'o': tr_variantDictAddBool (settings, TR_KEY_dht_enabled, true);
366                      break;
367            case 'O': tr_variantDictAddBool (settings, TR_KEY_dht_enabled, false);
368                      break;
369            case 'p': tr_variantDictAddInt (settings, TR_KEY_rpc_port, atoi (optarg));
370                      break;
371            case 't': tr_variantDictAddBool (settings, TR_KEY_rpc_authentication_required, true);
372                      break;
373            case 'T': tr_variantDictAddBool (settings, TR_KEY_rpc_authentication_required, false);
374                      break;
375            case 'u': tr_variantDictAddStr (settings, TR_KEY_rpc_username, optarg);
376                      break;
377            case 'v': tr_variantDictAddStr (settings, TR_KEY_rpc_password, optarg);
378                      break;
379            case 'w': tr_variantDictAddStr (settings, TR_KEY_download_dir, optarg);
380                      break;
381            case 'P': tr_variantDictAddInt (settings, TR_KEY_peer_port, atoi (optarg));
382                      break;
383            case 'm': tr_variantDictAddBool (settings, TR_KEY_port_forwarding_enabled, true);
384                      break;
385            case 'M': tr_variantDictAddBool (settings, TR_KEY_port_forwarding_enabled, false);
386                      break;
387            case 'L': tr_variantDictAddInt (settings, TR_KEY_peer_limit_global, atoi (optarg));
388                      break;
389            case 'l': tr_variantDictAddInt (settings, TR_KEY_peer_limit_per_torrent, atoi (optarg));
390                      break;
391            case 800: *paused = true;
392                      break;
393            case 910: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_ENCRYPTION_REQUIRED);
394                      break;
395            case 911: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_ENCRYPTION_PREFERRED);
396                      break;
397            case 912: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_CLEAR_PREFERRED);
398                      break;
399            case 'i': tr_variantDictAddStr (settings, TR_KEY_bind_address_ipv4, optarg);
400                      break;
401            case 'I': tr_variantDictAddStr (settings, TR_KEY_bind_address_ipv6, optarg);
402                      break;
403            case 'r': tr_variantDictAddStr (settings, TR_KEY_rpc_bind_address, optarg);
404                      break;
405            case 953: tr_variantDictAddReal (settings, TR_KEY_ratio_limit, atof (optarg));
406                      tr_variantDictAddBool (settings, TR_KEY_ratio_limit_enabled, true);
407                      break;
408            case 954: tr_variantDictAddBool (settings, TR_KEY_ratio_limit_enabled, false);
409                      break;
410            case 'x': tr_variantDictAddStr (settings, key_pidfile, optarg);
411                      break;
412            case 'y': tr_variantDictAddBool (settings, TR_KEY_lpd_enabled, true);
413                      break;
414            case 'Y': tr_variantDictAddBool (settings, TR_KEY_lpd_enabled, false);
415                      break;
416            case 810: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_ERROR);
417                      break;
418            case 811: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_INFO);
419                      break;
420            case 812: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_DEBUG);
421                      break;
422            case 830: tr_variantDictAddBool (settings, TR_KEY_utp_enabled, true);
423                      break;
424            case 831: tr_variantDictAddBool (settings, TR_KEY_utp_enabled, false);
425                      break;
426            default:  tr_getopt_usage (MY_NAME, getUsage (), options);
427                      *exit_code = 0;
428                      return false;
429        }
430    }
431
432    return true;
433}
434
435struct daemon_data
436{
437  tr_variant   settings;
438  const char * configDir;
439  bool         paused;
440};
441
442static void
443daemon_reconfigure (void * arg UNUSED)
444{
445    if (!mySession)
446    {
447        tr_logAddInfo ("Deferring reload until session is fully started.");
448        seenHUP = true;
449    }
450    else
451    {
452        tr_variant settings;
453        const char * configDir;
454
455        /* reopen the logfile to allow for log rotation */
456        if (logfileName != NULL)
457            reopen_log_file (logfileName);
458
459        configDir = tr_sessionGetConfigDir (mySession);
460        tr_logAddInfo ("Reloading settings from \"%s\"", configDir);
461        tr_variantInitDict (&settings, 0);
462        tr_variantDictAddBool (&settings, TR_KEY_rpc_enabled, true);
463        tr_sessionLoadSettings (&settings, configDir, MY_NAME);
464        tr_sessionSet (mySession, &settings);
465        tr_variantFree (&settings);
466        tr_sessionReloadBlocklists (mySession);
467    }
468}
469
470static void
471daemon_stop (void * arg UNUSED)
472{
473    event_base_loopexit (ev_base, NULL);
474}
475
476static int
477daemon_start (void * raw_arg,
478              bool   foreground)
479{
480    bool boolVal;
481    const char * pid_filename;
482    dtr_watchdir * watchdir = NULL;
483    bool pidfile_created = false;
484    tr_session * session = NULL;
485    struct event *status_ev;
486
487    struct daemon_data * const arg = raw_arg;
488    tr_variant * const settings = &arg->settings;
489    const char * const configDir = arg->configDir;
490
491#ifndef HAVE_SYSLOG
492    (void) foreground;
493#endif
494
495    sd_notifyf (0, "MAINPID=%d\n", (int)getpid());
496
497    /* should go before libevent calls */
498    tr_net_init ();
499
500    /* setup event state */
501    ev_base = event_base_new ();
502    if (ev_base == NULL)
503    {
504        char buf[256];
505        tr_snprintf (buf, sizeof (buf), "Failed to init daemon event state: %s", tr_strerror (errno));
506        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
507        return 1;
508    }
509
510    /* start the session */
511    tr_formatter_mem_init (MEM_K, MEM_K_STR, MEM_M_STR, MEM_G_STR, MEM_T_STR);
512    tr_formatter_size_init (DISK_K, DISK_K_STR, DISK_M_STR, DISK_G_STR, DISK_T_STR);
513    tr_formatter_speed_init (SPEED_K, SPEED_K_STR, SPEED_M_STR, SPEED_G_STR, SPEED_T_STR);
514    session = tr_sessionInit (configDir, true, settings);
515    tr_sessionSetRPCCallback (session, on_rpc_callback, NULL);
516    tr_logAddNamedInfo (NULL, "Using settings from \"%s\"", configDir);
517    tr_sessionSaveSettings (session, configDir, settings);
518
519    pid_filename = NULL;
520    tr_variantDictFindStr (settings, key_pidfile, &pid_filename, NULL);
521    if (pid_filename && *pid_filename)
522    {
523        tr_error * error = NULL;
524        tr_sys_file_t fp = tr_sys_file_open (pid_filename,
525                                             TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE,
526                                             0666, &error);
527        if (fp != TR_BAD_SYS_FILE)
528        {
529            tr_sys_file_write_fmt (fp, "%d", NULL, (int)getpid ());
530            tr_sys_file_close (fp, NULL);
531            tr_logAddInfo ("Saved pidfile \"%s\"", pid_filename);
532            pidfile_created = true;
533        }
534        else
535        {
536            tr_logAddError ("Unable to save pidfile \"%s\": %s", pid_filename, error->message);
537            tr_error_free (error);
538        }
539    }
540
541    if (tr_variantDictFindBool (settings, TR_KEY_rpc_authentication_required, &boolVal) && boolVal)
542        tr_logAddNamedInfo (MY_NAME, "requiring authentication");
543
544    mySession = session;
545
546    /* If we got a SIGHUP during startup, process that now. */
547    if (seenHUP)
548        daemon_reconfigure (arg);
549
550    /* maybe add a watchdir */
551    {
552        const char * dir;
553
554        if (tr_variantDictFindBool (settings, TR_KEY_watch_dir_enabled, &boolVal)
555            && boolVal
556            && tr_variantDictFindStr (settings, TR_KEY_watch_dir, &dir, NULL)
557            && dir
558            && *dir)
559        {
560            tr_logAddInfo ("Watching \"%s\" for new .torrent files", dir);
561            watchdir = dtr_watchdir_new (mySession, dir, onFileAdded);
562        }
563    }
564
565    /* load the torrents */
566    {
567        tr_torrent ** torrents;
568        tr_ctor * ctor = tr_ctorNew (mySession);
569        if (arg->paused)
570            tr_ctorSetPaused (ctor, TR_FORCE, true);
571        torrents = tr_sessionLoadTorrents (mySession, ctor, NULL);
572        tr_free (torrents);
573        tr_ctorFree (ctor);
574    }
575
576#ifdef HAVE_SYSLOG
577    if (!foreground)
578        openlog (MY_NAME, LOG_CONS|LOG_PID, LOG_DAEMON);
579#endif
580
581    /* Create new timer event to report daemon status */
582    {
583        struct timeval one_sec = { 1, 0 };
584        status_ev = event_new(ev_base, -1, EV_PERSIST, &periodicUpdate, watchdir);
585        if (status_ev == NULL)
586        {
587            tr_logAddError("Failed to create status event %s", tr_strerror(errno));
588            goto cleanup;
589        }
590        if (event_add(status_ev, &one_sec) == -1)
591        {
592            tr_logAddError("Failed to add status event %s", tr_strerror(errno));
593            goto cleanup;
594        }
595    }
596
597    sd_notify( 0, "READY=1\n" );
598
599    /* Run daemon event loop */
600    if (event_base_dispatch(ev_base) == -1)
601    {
602        tr_logAddError("Failed to launch daemon event loop: %s", tr_strerror(errno));
603        goto cleanup;
604    }
605
606cleanup:
607    sd_notify( 0, "STATUS=Closing transmission session...\n" );
608    printf ("Closing transmission session...");
609
610    if (status_ev)
611    {
612        event_del(status_ev);
613        event_free(status_ev);
614    }
615    event_base_free(ev_base);
616
617    tr_sessionSaveSettings (mySession, configDir, settings);
618    dtr_watchdir_free (watchdir);
619    tr_sessionClose (mySession);
620    pumpLogMessages (logfile);
621    printf (" done.\n");
622
623    /* shutdown */
624#ifdef HAVE_SYSLOG
625    if (!foreground)
626    {
627        syslog (LOG_INFO, "%s", "Closing session");
628        closelog ();
629    }
630#endif
631
632    /* cleanup */
633    if (pidfile_created)
634        tr_sys_path_remove (pid_filename, NULL);
635
636    sd_notify (0, "STATUS=\n");
637
638    return 0;
639}
640
641int
642tr_main (int    argc,
643         char * argv[])
644{
645    const dtr_callbacks cb =
646    {
647        .on_start       = &daemon_start,
648        .on_stop        = &daemon_stop,
649        .on_reconfigure = &daemon_reconfigure,
650    };
651
652    int ret;
653    bool loaded, dumpSettings, foreground;
654    tr_error * error = NULL;
655
656    struct daemon_data arg;
657    tr_variant * const settings = &arg.settings;
658    const char ** const configDir = &arg.configDir;
659
660    key_pidfile = tr_quark_new ("pidfile",  7);
661
662    /* load settings from defaults + config file */
663    tr_variantInitDict (settings, 0);
664    tr_variantDictAddBool (settings, TR_KEY_rpc_enabled, true);
665    *configDir = getConfigDir (argc, (const char**)argv);
666    loaded = tr_sessionLoadSettings (settings, *configDir, MY_NAME);
667
668    /* overwrite settings from the comamndline */
669    if (!parse_args (argc, (const char**) argv, settings, &arg.paused, &dumpSettings, &foreground, &ret))
670        goto cleanup;
671
672    if (foreground && logfile == TR_BAD_SYS_FILE)
673        logfile = tr_sys_file_get_std (TR_STD_SYS_FILE_ERR, NULL);
674
675    if (!loaded)
676    {
677        printMessage (logfile, TR_LOG_ERROR, MY_NAME, "Error loading config file -- exiting.", __FILE__, __LINE__);
678        ret = 1;
679        goto cleanup;
680    }
681
682    if (dumpSettings)
683    {
684        char * str = tr_variantToStr (settings, TR_VARIANT_FMT_JSON, NULL);
685        fprintf (stderr, "%s", str);
686        tr_free (str);
687        goto cleanup;
688    }
689
690    if (!dtr_daemon (&cb, &arg, foreground, &ret, &error))
691    {
692        char buf[256];
693        tr_snprintf (buf, sizeof (buf), "Failed to daemonize: %s", error->message);
694        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
695        tr_error_free (error);
696    }
697
698cleanup:
699    tr_variantFree (settings);
700
701    return ret;
702}
Note: See TracBrowser for help on using the repository browser.