src/sessiond/main.c

   1 /* vi: set et sw=4 ts=4 cino=t0,(0: */
   2 /* -*- Mode: C; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   3 /*
   4  * This file is part of tlm
   5  *
   6  * Copyright (C) 2014-2015 Intel Corporation.
   7  *
   8  * Contact: Imran Zaman <imran.zaman@intel.com>
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful, but
  16  * WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with this library; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  23  * 02110-1301 USA
  24  */
  25
  26 #include <config.h>
  27 #include <errno.h>
  28 #include <signal.h>
  29 #include <string.h>
  30 #include <stdio.h>
  31 #include <errno.h>
  32 #include <glib-unix.h>
  33 #include <glib.h>
  34 #include <gio/gio.h>
  35 #include <sys/prctl.h>
  36
  37 #include "common/tlm-log.h"
  38 #include "tlm-session-daemon.h"
  39
  40 static TlmSessionDaemon *_daemon = NULL;
  41 static guint _sig_source_id[2];
  42
  43 static void
  44 _on_daemon_closed (gpointer data, GObject *server)
  45 {
  46     _daemon = NULL;
  47     DBG ("Daemon closed");
  48     if (data) g_main_loop_quit ((GMainLoop *)data);
  49 }
  50
  51 static gboolean
  52 _handle_quit_signal (gpointer user_data)
  53 {
  54     GMainLoop *ml = (GMainLoop *) user_data;
  55
  56     g_return_val_if_fail (ml != NULL, FALSE);
  57     DBG ("Received quit signal");
  58     if (ml) g_main_loop_quit (ml);
  59
  60     return FALSE;
  61 }
  62
  63 static void
  64 _install_sighandlers (GMainLoop *main_loop)
  65 {
  66     GSource *source = NULL;
  67     GMainContext *ctx = g_main_loop_get_context (main_loop);
  68
  69     if (signal (SIGINT, SIG_IGN) == SIG_ERR)
  70         WARN ("failed to ignore SIGINT: %s", strerror(errno));
  71
  72     source = g_unix_signal_source_new (SIGTERM);
  73     g_source_set_callback (source,
  74                            _handle_quit_signal,
  75                            main_loop,
  76                            NULL);
  77     _sig_source_id[0] = g_source_attach (source, ctx);
  78
  79     source = g_unix_signal_source_new (SIGHUP);
  80     g_source_set_callback (source,
  81                            _handle_quit_signal,
  82                            main_loop,
  83                            NULL);
  84     _sig_source_id[1] = g_source_attach (source, ctx);
  85
  86     if (prctl(PR_SET_PDEATHSIG, SIGHUP))
  87         WARN ("failed to set parent death signal");
  88 }
  89
  90 int main (int argc, char **argv)
  91 {
  92     GMainLoop *main_loop = NULL;
  93     gint in_fd = 0, out_fd = 1;
  94
  95     /* Duplicates stdin and stdout descriptors and point the descriptors
  96      * to /dev/null to avoid anyone writing to descriptors
  97      * */
  98     in_fd = dup(0);
  99     if (in_fd == -1) {
 100         WARN ("Failed to dup stdin : %s(%d)", strerror(errno), errno);
 101         in_fd = 0;
 102     }
 103     if (!freopen("/dev/null", "r+", stdin)) {
 104         WARN ("Unable to redirect stdin to /dev/null");
 105     }
 106
 107     out_fd = dup(1);
 108     if (out_fd == -1) {
 109         WARN ("Failed to dup stdout : %s(%d)", strerror(errno), errno);
 110         out_fd = 1;
 111     }
 112
 113     if (!freopen("/dev/null", "w+", stdout)) {
 114         WARN ("Unable to redirect stdout to /dev/null");
 115     }
 116
 117     /* Reattach stdout to stderr */
 118     //dup2 (2, 1);
 119
 120 #if !GLIB_CHECK_VERSION (2, 36, 0)
 121     g_type_init ();
 122 #endif
 123
 124     DBG ("old pgid=%u", getpgrp ());
 125
 126     _daemon = tlm_session_daemon_new (in_fd, out_fd);
 127     if (_daemon == NULL) {
 128         return -1;
 129     }
 130
 131     main_loop = g_main_loop_new (NULL, FALSE);
 132     g_object_weak_ref (G_OBJECT (_daemon), _on_daemon_closed, main_loop);
 133     _install_sighandlers (main_loop);
 134
 135     tlm_log_init(G_LOG_DOMAIN);
 136
 137     DBG ("Entering main event loop");
 138
 139     g_main_loop_run (main_loop);
 140
 141     if(_daemon) {
 142         g_object_unref (_daemon);
 143     }
 144
 145     if (main_loop) {
 146         g_main_loop_unref (main_loop);
 147     }
 148     tlm_log_close (NULL);
 149     return 0;
 150 }