source: trunk/daemon/daemon.c @ 14532

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

Add more booleans to the picture

  • Property svn:keywords set to Date Rev Author Id
File size: 23.8 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 14532 2015-05-31 22:13:31Z 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    /* setup event state */
498    ev_base = event_base_new();
499    if (ev_base == NULL)
500    {
501        char buf[256];
502        tr_snprintf(buf, sizeof(buf), "Failed to init daemon event state: %s", tr_strerror(errno));
503        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
504        return 1;
505    }
506
507    /* start the session */
508    tr_formatter_mem_init (MEM_K, MEM_K_STR, MEM_M_STR, MEM_G_STR, MEM_T_STR);
509    tr_formatter_size_init (DISK_K, DISK_K_STR, DISK_M_STR, DISK_G_STR, DISK_T_STR);
510    tr_formatter_speed_init (SPEED_K, SPEED_K_STR, SPEED_M_STR, SPEED_G_STR, SPEED_T_STR);
511    session = tr_sessionInit ("daemon", configDir, true, settings);
512    tr_sessionSetRPCCallback (session, on_rpc_callback, NULL);
513    tr_logAddNamedInfo (NULL, "Using settings from \"%s\"", configDir);
514    tr_sessionSaveSettings (session, configDir, settings);
515
516    pid_filename = NULL;
517    tr_variantDictFindStr (settings, key_pidfile, &pid_filename, NULL);
518    if (pid_filename && *pid_filename)
519    {
520        tr_error * error = NULL;
521        tr_sys_file_t fp = tr_sys_file_open (pid_filename,
522                                             TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE,
523                                             0666, &error);
524        if (fp != TR_BAD_SYS_FILE)
525        {
526            tr_sys_file_write_fmt (fp, "%d", NULL, (int)getpid ());
527            tr_sys_file_close (fp, NULL);
528            tr_logAddInfo ("Saved pidfile \"%s\"", pid_filename);
529            pidfile_created = true;
530        }
531        else
532        {
533            tr_logAddError ("Unable to save pidfile \"%s\": %s", pid_filename, error->message);
534            tr_error_free (error);
535        }
536    }
537
538    if (tr_variantDictFindBool (settings, TR_KEY_rpc_authentication_required, &boolVal) && boolVal)
539        tr_logAddNamedInfo (MY_NAME, "requiring authentication");
540
541    mySession = session;
542
543    /* If we got a SIGHUP during startup, process that now. */
544    if (seenHUP)
545        daemon_reconfigure (arg);
546
547    /* maybe add a watchdir */
548    {
549        const char * dir;
550
551        if (tr_variantDictFindBool (settings, TR_KEY_watch_dir_enabled, &boolVal)
552            && boolVal
553            && tr_variantDictFindStr (settings, TR_KEY_watch_dir, &dir, NULL)
554            && dir
555            && *dir)
556        {
557            tr_logAddInfo ("Watching \"%s\" for new .torrent files", dir);
558            watchdir = dtr_watchdir_new (mySession, dir, onFileAdded);
559        }
560    }
561
562    /* load the torrents */
563    {
564        tr_torrent ** torrents;
565        tr_ctor * ctor = tr_ctorNew (mySession);
566        if (arg->paused)
567            tr_ctorSetPaused (ctor, TR_FORCE, true);
568        torrents = tr_sessionLoadTorrents (mySession, ctor, NULL);
569        tr_free (torrents);
570        tr_ctorFree (ctor);
571    }
572
573#ifdef HAVE_SYSLOG
574    if (!foreground)
575        openlog (MY_NAME, LOG_CONS|LOG_PID, LOG_DAEMON);
576#endif
577
578    /* Create new timer event to report daemon status */
579    {
580        struct timeval one_sec = { 1, 0 };
581        status_ev = event_new(ev_base, -1, EV_PERSIST, &periodicUpdate, watchdir);
582        if (status_ev == NULL)
583        {
584            tr_logAddError("Failed to create status event %s", tr_strerror(errno));
585            goto cleanup;
586        }
587        if (event_add(status_ev, &one_sec) == -1)
588        {
589            tr_logAddError("Failed to add status event %s", tr_strerror(errno));
590            goto cleanup;
591        }
592    }
593
594    sd_notify( 0, "READY=1\n" );
595
596    /* Run daemon event loop */
597    if (event_base_dispatch(ev_base) == -1)
598    {
599        tr_logAddError("Failed to launch daemon event loop: %s", tr_strerror(errno));
600        goto cleanup;
601    }
602
603cleanup:
604    sd_notify( 0, "STATUS=Closing transmission session...\n" );
605    printf ("Closing transmission session...");
606
607    if (status_ev)
608    {
609        event_del(status_ev);
610        event_free(status_ev);
611    }
612    event_base_free(ev_base);
613
614    tr_sessionSaveSettings (mySession, configDir, settings);
615    dtr_watchdir_free (watchdir);
616    tr_sessionClose (mySession);
617    pumpLogMessages (logfile);
618    printf (" done.\n");
619
620    /* shutdown */
621#ifdef HAVE_SYSLOG
622    if (!foreground)
623    {
624        syslog (LOG_INFO, "%s", "Closing session");
625        closelog ();
626    }
627#endif
628
629    /* cleanup */
630    if (pidfile_created)
631        tr_sys_path_remove (pid_filename, NULL);
632
633    sd_notify (0, "STATUS=\n");
634
635    return 0;
636}
637
638int
639tr_main (int    argc,
640         char * argv[])
641{
642    const dtr_callbacks cb =
643    {
644        .on_start       = &daemon_start,
645        .on_stop        = &daemon_stop,
646        .on_reconfigure = &daemon_reconfigure,
647    };
648
649    int ret;
650    bool loaded, dumpSettings, foreground;
651    tr_error * error = NULL;
652
653    struct daemon_data arg;
654    tr_variant * const settings = &arg.settings;
655    const char ** const configDir = &arg.configDir;
656
657    key_pidfile = tr_quark_new ("pidfile",  7);
658
659    /* load settings from defaults + config file */
660    tr_variantInitDict (settings, 0);
661    tr_variantDictAddBool (settings, TR_KEY_rpc_enabled, true);
662    *configDir = getConfigDir (argc, (const char**)argv);
663    loaded = tr_sessionLoadSettings (settings, *configDir, MY_NAME);
664
665    /* overwrite settings from the comamndline */
666    if (!parse_args (argc, (const char**) argv, settings, &arg.paused, &dumpSettings, &foreground, &ret))
667        goto cleanup;
668
669    if (foreground && logfile == TR_BAD_SYS_FILE)
670        logfile = tr_sys_file_get_std (TR_STD_SYS_FILE_ERR, NULL);
671
672    if (!loaded)
673    {
674        printMessage (logfile, TR_LOG_ERROR, MY_NAME, "Error loading config file -- exiting.", __FILE__, __LINE__);
675        ret = 1;
676        goto cleanup;
677    }
678
679    if (dumpSettings)
680    {
681        char * str = tr_variantToStr (settings, TR_VARIANT_FMT_JSON, NULL);
682        fprintf (stderr, "%s", str);
683        tr_free (str);
684        goto cleanup;
685    }
686
687    if (!dtr_daemon (&cb, &arg, foreground, &ret, &error))
688    {
689        char buf[256];
690        tr_snprintf (buf, sizeof (buf), "Failed to daemonize: %s", error->message);
691        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
692        tr_error_free (error);
693    }
694
695cleanup:
696    tr_variantFree (settings);
697
698    return ret;
699}
Note: See TracBrowser for help on using the repository browser.