source: trunk/daemon/daemon.c @ 14675

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

Add hidden "watch-dir-force-generic" setting handling to daemon

  • Property svn:keywords set to Date Rev Author Id
File size: 24.6 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 14675 2016-01-25 22:09:49Z 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/event.h>
25
26#include <libtransmission/transmission.h>
27#include <libtransmission/error.h>
28#include <libtransmission/file.h>
29#include <libtransmission/tr-getopt.h>
30#include <libtransmission/log.h>
31#include <libtransmission/utils.h>
32#include <libtransmission/variant.h>
33#include <libtransmission/version.h>
34#include <libtransmission/watchdir.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
45#define MY_NAME "transmission-daemon"
46
47#define MEM_K 1024
48#define MEM_K_STR "KiB"
49#define MEM_M_STR "MiB"
50#define MEM_G_STR "GiB"
51#define MEM_T_STR "TiB"
52
53#define DISK_K 1000
54#define DISK_B_STR  "B"
55#define DISK_K_STR "kB"
56#define DISK_M_STR "MB"
57#define DISK_G_STR "GB"
58#define DISK_T_STR "TB"
59
60#define SPEED_K 1000
61#define SPEED_B_STR  "B/s"
62#define SPEED_K_STR "kB/s"
63#define SPEED_M_STR "MB/s"
64#define SPEED_G_STR "GB/s"
65#define SPEED_T_STR "TB/s"
66
67static bool seenHUP = false;
68static const char *logfileName = NULL;
69static tr_sys_file_t logfile = TR_BAD_SYS_FILE;
70static tr_session * mySession = NULL;
71static tr_quark key_pidfile = 0;
72static tr_quark key_watch_dir_force_generic = 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 * const * 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 tr_watchdir_status
189onFileAdded (tr_watchdir_t   dir,
190             const char    * name,
191             void          * context)
192{
193    tr_session * session = context;
194
195    if (!tr_str_has_suffix (name, ".torrent"))
196        return TR_WATCHDIR_IGNORE;
197
198    char * filename = tr_buildPath (tr_watchdir_get_path (dir), name, NULL);
199    tr_ctor * ctor = tr_ctorNew (session);
200    int err = tr_ctorSetMetainfoFromFile (ctor, filename);
201
202    if (!err)
203    {
204        tr_torrentNew (ctor, &err, NULL);
205
206        if (err == TR_PARSE_ERR)
207            tr_logAddError ("Error parsing .torrent file \"%s\"", name);
208        else
209        {
210            bool trash = false;
211            const bool test = tr_ctorGetDeleteSource (ctor, &trash);
212
213            tr_logAddInfo ("Parsing .torrent file successful \"%s\"", name);
214
215            if (test && trash)
216            {
217                tr_error * error = NULL;
218
219                tr_logAddInfo ("Deleting input .torrent file \"%s\"", name);
220                if (!tr_sys_path_remove (filename, &error))
221                {
222                    tr_logAddError ("Error deleting .torrent file: %s", error->message);
223                    tr_error_free (error);
224                }
225            }
226            else
227            {
228                char * new_filename = tr_strdup_printf ("%s.added", filename);
229                tr_sys_path_rename (filename, new_filename, NULL);
230                tr_free (new_filename);
231            }
232        }
233    }
234    else
235    {
236        err = TR_PARSE_ERR;
237    }
238
239    tr_ctorFree (ctor);
240    tr_free (filename);
241
242    return err == TR_PARSE_ERR ? TR_WATCHDIR_RETRY : TR_WATCHDIR_ACCEPT;
243}
244
245static void
246printMessage (tr_sys_file_t logfile, int level, const char * name, const char * message, const char * file, int line)
247{
248    if (logfile != TR_BAD_SYS_FILE)
249    {
250        char timestr[64];
251        tr_logGetTimeStr (timestr, sizeof (timestr));
252        if (name)
253            tr_sys_file_write_fmt (logfile, "[%s] %s %s (%s:%d)" TR_NATIVE_EOL_STR,
254                                   NULL, timestr, name, message, file, line);
255        else
256            tr_sys_file_write_fmt (logfile, "[%s] %s (%s:%d)" TR_NATIVE_EOL_STR,
257                                   NULL, timestr, message, file, line);
258    }
259#ifdef HAVE_SYSLOG
260    else /* daemon... write to syslog */
261    {
262        int priority;
263
264        /* figure out the syslog priority */
265        switch (level) {
266            case TR_LOG_ERROR: priority = LOG_ERR; break;
267            case TR_LOG_DEBUG: priority = LOG_DEBUG; break;
268            default:           priority = LOG_INFO; break;
269        }
270
271        if (name)
272            syslog (priority, "%s %s (%s:%d)", name, message, file, line);
273        else
274            syslog (priority, "%s (%s:%d)", message, file, line);
275    }
276#else
277    (void) level;
278#endif
279}
280
281static void
282pumpLogMessages (tr_sys_file_t logfile)
283{
284    const tr_log_message * l;
285    tr_log_message * list = tr_logGetQueue ();
286
287    for (l=list; l!=NULL; l=l->next)
288        printMessage (logfile, l->level, l->name, l->message, l->file, l->line);
289
290    if (logfile != TR_BAD_SYS_FILE)
291        tr_sys_file_flush (logfile, NULL);
292
293    tr_logFreeQueue (list);
294}
295
296static void
297reportStatus (void)
298{
299    const double up = tr_sessionGetRawSpeed_KBps (mySession, TR_UP);
300    const double dn = tr_sessionGetRawSpeed_KBps (mySession, TR_DOWN);
301
302    if (up>0 || dn>0)
303        sd_notifyf (0, "STATUS=Uploading %.2f KBps, Downloading %.2f KBps.\n", up, dn);
304    else
305        sd_notify (0, "STATUS=Idle.\n");
306}
307
308static void
309periodicUpdate (evutil_socket_t   fd UNUSED,
310                short             what UNUSED,
311                void            * context UNUSED)
312{
313    pumpLogMessages (logfile);
314    reportStatus ();
315}
316
317static tr_rpc_callback_status
318on_rpc_callback (tr_session            * session UNUSED,
319                 tr_rpc_callback_type    type,
320                 struct tr_torrent     * tor UNUSED,
321                 void                  * user_data UNUSED)
322{
323    if (type == TR_RPC_SESSION_CLOSE)
324        event_base_loopexit(ev_base, NULL);
325    return TR_RPC_OK;
326}
327
328static bool
329parse_args (int           argc,
330            const char ** argv,
331            tr_variant  * settings,
332            bool        * paused,
333            bool        * dump_settings,
334            bool        * foreground,
335            int         * exit_code)
336{
337    int c;
338    const char * optarg;
339
340    *paused = false;
341    *dump_settings = false;
342    *foreground = false;
343
344    tr_optind = 1;
345    while ((c = tr_getopt (getUsage (), argc, argv, options, &optarg))) {
346        switch (c) {
347            case 'a': tr_variantDictAddStr  (settings, TR_KEY_rpc_whitelist, optarg);
348                      tr_variantDictAddBool (settings, TR_KEY_rpc_whitelist_enabled, true);
349                      break;
350            case 'b': tr_variantDictAddBool (settings, TR_KEY_blocklist_enabled, true);
351                      break;
352            case 'B': tr_variantDictAddBool (settings, TR_KEY_blocklist_enabled, false);
353                      break;
354            case 'c': tr_variantDictAddStr  (settings, TR_KEY_watch_dir, optarg);
355                      tr_variantDictAddBool (settings, TR_KEY_watch_dir_enabled, true);
356                      break;
357            case 'C': tr_variantDictAddBool (settings, TR_KEY_watch_dir_enabled, false);
358                      break;
359            case 941: tr_variantDictAddStr  (settings, TR_KEY_incomplete_dir, optarg);
360                      tr_variantDictAddBool (settings, TR_KEY_incomplete_dir_enabled, true);
361                      break;
362            case 942: tr_variantDictAddBool (settings, TR_KEY_incomplete_dir_enabled, false);
363                      break;
364            case 'd': *dump_settings = true;
365                      break;
366            case 'e': if (reopen_log_file (optarg))
367                          logfileName = optarg;
368                      break;
369            case 'f': *foreground = true;
370                      break;
371            case 'g': /* handled above */
372                      break;
373            case 'V': /* version */
374                      fprintf (stderr, "%s %s\n", MY_NAME, LONG_VERSION_STRING);
375                      *exit_code = 0;
376                      return false;
377            case 'o': tr_variantDictAddBool (settings, TR_KEY_dht_enabled, true);
378                      break;
379            case 'O': tr_variantDictAddBool (settings, TR_KEY_dht_enabled, false);
380                      break;
381            case 'p': tr_variantDictAddInt (settings, TR_KEY_rpc_port, atoi (optarg));
382                      break;
383            case 't': tr_variantDictAddBool (settings, TR_KEY_rpc_authentication_required, true);
384                      break;
385            case 'T': tr_variantDictAddBool (settings, TR_KEY_rpc_authentication_required, false);
386                      break;
387            case 'u': tr_variantDictAddStr (settings, TR_KEY_rpc_username, optarg);
388                      break;
389            case 'v': tr_variantDictAddStr (settings, TR_KEY_rpc_password, optarg);
390                      break;
391            case 'w': tr_variantDictAddStr (settings, TR_KEY_download_dir, optarg);
392                      break;
393            case 'P': tr_variantDictAddInt (settings, TR_KEY_peer_port, atoi (optarg));
394                      break;
395            case 'm': tr_variantDictAddBool (settings, TR_KEY_port_forwarding_enabled, true);
396                      break;
397            case 'M': tr_variantDictAddBool (settings, TR_KEY_port_forwarding_enabled, false);
398                      break;
399            case 'L': tr_variantDictAddInt (settings, TR_KEY_peer_limit_global, atoi (optarg));
400                      break;
401            case 'l': tr_variantDictAddInt (settings, TR_KEY_peer_limit_per_torrent, atoi (optarg));
402                      break;
403            case 800: *paused = true;
404                      break;
405            case 910: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_ENCRYPTION_REQUIRED);
406                      break;
407            case 911: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_ENCRYPTION_PREFERRED);
408                      break;
409            case 912: tr_variantDictAddInt (settings, TR_KEY_encryption, TR_CLEAR_PREFERRED);
410                      break;
411            case 'i': tr_variantDictAddStr (settings, TR_KEY_bind_address_ipv4, optarg);
412                      break;
413            case 'I': tr_variantDictAddStr (settings, TR_KEY_bind_address_ipv6, optarg);
414                      break;
415            case 'r': tr_variantDictAddStr (settings, TR_KEY_rpc_bind_address, optarg);
416                      break;
417            case 953: tr_variantDictAddReal (settings, TR_KEY_ratio_limit, atof (optarg));
418                      tr_variantDictAddBool (settings, TR_KEY_ratio_limit_enabled, true);
419                      break;
420            case 954: tr_variantDictAddBool (settings, TR_KEY_ratio_limit_enabled, false);
421                      break;
422            case 'x': tr_variantDictAddStr (settings, key_pidfile, optarg);
423                      break;
424            case 'y': tr_variantDictAddBool (settings, TR_KEY_lpd_enabled, true);
425                      break;
426            case 'Y': tr_variantDictAddBool (settings, TR_KEY_lpd_enabled, false);
427                      break;
428            case 810: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_ERROR);
429                      break;
430            case 811: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_INFO);
431                      break;
432            case 812: tr_variantDictAddInt (settings,  TR_KEY_message_level, TR_LOG_DEBUG);
433                      break;
434            case 830: tr_variantDictAddBool (settings, TR_KEY_utp_enabled, true);
435                      break;
436            case 831: tr_variantDictAddBool (settings, TR_KEY_utp_enabled, false);
437                      break;
438            default:  tr_getopt_usage (MY_NAME, getUsage (), options);
439                      *exit_code = 0;
440                      return false;
441        }
442    }
443
444    return true;
445}
446
447struct daemon_data
448{
449  tr_variant   settings;
450  const char * configDir;
451  bool         paused;
452};
453
454static void
455daemon_reconfigure (void * arg UNUSED)
456{
457    if (!mySession)
458    {
459        tr_logAddInfo ("Deferring reload until session is fully started.");
460        seenHUP = true;
461    }
462    else
463    {
464        tr_variant settings;
465        const char * configDir;
466
467        /* reopen the logfile to allow for log rotation */
468        if (logfileName != NULL)
469            reopen_log_file (logfileName);
470
471        configDir = tr_sessionGetConfigDir (mySession);
472        tr_logAddInfo ("Reloading settings from \"%s\"", configDir);
473        tr_variantInitDict (&settings, 0);
474        tr_variantDictAddBool (&settings, TR_KEY_rpc_enabled, true);
475        tr_sessionLoadSettings (&settings, configDir, MY_NAME);
476        tr_sessionSet (mySession, &settings);
477        tr_variantFree (&settings);
478        tr_sessionReloadBlocklists (mySession);
479    }
480}
481
482static void
483daemon_stop (void * arg UNUSED)
484{
485    event_base_loopexit (ev_base, NULL);
486}
487
488static int
489daemon_start (void * raw_arg,
490              bool   foreground)
491{
492    bool boolVal;
493    const char * pid_filename;
494    bool pidfile_created = false;
495    tr_session * session = NULL;
496    struct event * status_ev = NULL;
497    tr_watchdir_t watchdir = NULL;
498
499    struct daemon_data * const arg = raw_arg;
500    tr_variant * const settings = &arg->settings;
501    const char * const configDir = arg->configDir;
502
503#ifndef HAVE_SYSLOG
504    (void) foreground;
505#endif
506
507    sd_notifyf (0, "MAINPID=%d\n", (int)getpid());
508
509    /* should go before libevent calls */
510    tr_net_init ();
511
512    /* setup event state */
513    ev_base = event_base_new ();
514    if (ev_base == NULL)
515    {
516        char buf[256];
517        tr_snprintf (buf, sizeof (buf), "Failed to init daemon event state: %s", tr_strerror (errno));
518        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
519        return 1;
520    }
521
522    /* start the session */
523    tr_formatter_mem_init (MEM_K, MEM_K_STR, MEM_M_STR, MEM_G_STR, MEM_T_STR);
524    tr_formatter_size_init (DISK_K, DISK_K_STR, DISK_M_STR, DISK_G_STR, DISK_T_STR);
525    tr_formatter_speed_init (SPEED_K, SPEED_K_STR, SPEED_M_STR, SPEED_G_STR, SPEED_T_STR);
526    session = tr_sessionInit (configDir, true, settings);
527    tr_sessionSetRPCCallback (session, on_rpc_callback, NULL);
528    tr_logAddNamedInfo (NULL, "Using settings from \"%s\"", configDir);
529    tr_sessionSaveSettings (session, configDir, settings);
530
531    pid_filename = NULL;
532    tr_variantDictFindStr (settings, key_pidfile, &pid_filename, NULL);
533    if (pid_filename && *pid_filename)
534    {
535        tr_error * error = NULL;
536        tr_sys_file_t fp = tr_sys_file_open (pid_filename,
537                                             TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE,
538                                             0666, &error);
539        if (fp != TR_BAD_SYS_FILE)
540        {
541            tr_sys_file_write_fmt (fp, "%d", NULL, (int)getpid ());
542            tr_sys_file_close (fp, NULL);
543            tr_logAddInfo ("Saved pidfile \"%s\"", pid_filename);
544            pidfile_created = true;
545        }
546        else
547        {
548            tr_logAddError ("Unable to save pidfile \"%s\": %s", pid_filename, error->message);
549            tr_error_free (error);
550        }
551    }
552
553    if (tr_variantDictFindBool (settings, TR_KEY_rpc_authentication_required, &boolVal) && boolVal)
554        tr_logAddNamedInfo (MY_NAME, "requiring authentication");
555
556    mySession = session;
557
558    /* If we got a SIGHUP during startup, process that now. */
559    if (seenHUP)
560        daemon_reconfigure (arg);
561
562    /* maybe add a watchdir */
563    if (tr_variantDictFindBool (settings, TR_KEY_watch_dir_enabled, &boolVal) && boolVal)
564    {
565        const char * dir;
566        bool force_generic;
567
568        if (!tr_variantDictFindBool (settings, key_watch_dir_force_generic, &force_generic))
569          force_generic = false;
570
571        if (tr_variantDictFindStr (settings, TR_KEY_watch_dir, &dir, NULL) && dir != NULL && *dir != '\0')
572        {
573            tr_logAddInfo ("Watching \"%s\" for new .torrent files", dir);
574            if ((watchdir = tr_watchdir_new (dir, &onFileAdded, mySession, ev_base, force_generic)) == NULL)
575                goto cleanup;
576        }
577    }
578
579    /* load the torrents */
580    {
581        tr_torrent ** torrents;
582        tr_ctor * ctor = tr_ctorNew (mySession);
583        if (arg->paused)
584            tr_ctorSetPaused (ctor, TR_FORCE, true);
585        torrents = tr_sessionLoadTorrents (mySession, ctor, NULL);
586        tr_free (torrents);
587        tr_ctorFree (ctor);
588    }
589
590#ifdef HAVE_SYSLOG
591    if (!foreground)
592        openlog (MY_NAME, LOG_CONS|LOG_PID, LOG_DAEMON);
593#endif
594
595    /* Create new timer event to report daemon status */
596    {
597        struct timeval one_sec = { 1, 0 };
598        status_ev = event_new(ev_base, -1, EV_PERSIST, &periodicUpdate, NULL);
599        if (status_ev == NULL)
600        {
601            tr_logAddError("Failed to create status event %s", tr_strerror(errno));
602            goto cleanup;
603        }
604        if (event_add(status_ev, &one_sec) == -1)
605        {
606            tr_logAddError("Failed to add status event %s", tr_strerror(errno));
607            goto cleanup;
608        }
609    }
610
611    sd_notify( 0, "READY=1\n" );
612
613    /* Run daemon event loop */
614    if (event_base_dispatch(ev_base) == -1)
615    {
616        tr_logAddError("Failed to launch daemon event loop: %s", tr_strerror(errno));
617        goto cleanup;
618    }
619
620cleanup:
621    sd_notify( 0, "STATUS=Closing transmission session...\n" );
622    printf ("Closing transmission session...");
623
624    tr_watchdir_free (watchdir);
625
626    if (status_ev)
627    {
628        event_del(status_ev);
629        event_free(status_ev);
630    }
631    event_base_free(ev_base);
632
633    tr_sessionSaveSettings (mySession, configDir, settings);
634    tr_sessionClose (mySession);
635    pumpLogMessages (logfile);
636    printf (" done.\n");
637
638    /* shutdown */
639#ifdef HAVE_SYSLOG
640    if (!foreground)
641    {
642        syslog (LOG_INFO, "%s", "Closing session");
643        closelog ();
644    }
645#endif
646
647    /* cleanup */
648    if (pidfile_created)
649        tr_sys_path_remove (pid_filename, NULL);
650
651    sd_notify (0, "STATUS=\n");
652
653    return 0;
654}
655
656int
657tr_main (int    argc,
658         char * argv[])
659{
660    const dtr_callbacks cb =
661    {
662        .on_start       = &daemon_start,
663        .on_stop        = &daemon_stop,
664        .on_reconfigure = &daemon_reconfigure,
665    };
666
667    int ret;
668    bool loaded, dumpSettings, foreground;
669    tr_error * error = NULL;
670
671    struct daemon_data arg;
672    tr_variant * const settings = &arg.settings;
673    const char ** const configDir = &arg.configDir;
674
675    key_pidfile = tr_quark_new ("pidfile", 7);
676    key_watch_dir_force_generic = tr_quark_new ("watch-dir-force-generic", 23);
677
678    /* load settings from defaults + config file */
679    tr_variantInitDict (settings, 0);
680    tr_variantDictAddBool (settings, TR_KEY_rpc_enabled, true);
681    *configDir = getConfigDir (argc, (const char* const *)argv);
682    loaded = tr_sessionLoadSettings (settings, *configDir, MY_NAME);
683
684    /* overwrite settings from the comamndline */
685    if (!parse_args (argc, (const char**) argv, settings, &arg.paused, &dumpSettings, &foreground, &ret))
686        goto cleanup;
687
688    if (foreground && logfile == TR_BAD_SYS_FILE)
689        logfile = tr_sys_file_get_std (TR_STD_SYS_FILE_ERR, NULL);
690
691    if (!loaded)
692    {
693        printMessage (logfile, TR_LOG_ERROR, MY_NAME, "Error loading config file -- exiting.", __FILE__, __LINE__);
694        ret = 1;
695        goto cleanup;
696    }
697
698    if (dumpSettings)
699    {
700        char * str = tr_variantToStr (settings, TR_VARIANT_FMT_JSON, NULL);
701        fprintf (stderr, "%s", str);
702        tr_free (str);
703        goto cleanup;
704    }
705
706    if (!dtr_daemon (&cb, &arg, foreground, &ret, &error))
707    {
708        char buf[256];
709        tr_snprintf (buf, sizeof (buf), "Failed to daemonize: %s", error->message);
710        printMessage (logfile, TR_LOG_ERROR, MY_NAME, buf, __FILE__, __LINE__);
711        tr_error_free (error);
712    }
713
714cleanup:
715    tr_variantFree (settings);
716
717    return ret;
718}
Note: See TracBrowser for help on using the repository browser.