add tools to log restart and stop events

This should probably be used by init instead of
logging the event from CK.

3 files changed, 368 insertions, 0 deletions

diff --git a/tools/Makefile.am b/tools/Makefile.am
index bf6a4ae..1fe0f6b 100644
--- a/tools/Makefile.am
+++ b/tools/Makefile.am
@@ -50,6 +50,8 @@ bin_PROGRAMS = \
 
 sbin_PROGRAMS = \
 	ck-log-system-start		\
+	ck-log-system-restart		\
+	ck-log-system-stop		\
 	$(NULL)
 
 ck_list_sessions_SOURCES =		\
@@ -79,6 +81,24 @@ ck_log_system_start_LDADD =		\
 	$(top_builddir)/src/libck-event-log.la	\
 	$(NULL)
 
+ck_log_system_restart_SOURCES =		\
+	ck-log-system-restart.c		\
+	$(NULL)
+
+ck_log_system_restart_LDADD =		\
+	$(HISTORY_LIBS)			\
+	$(top_builddir)/src/libck-event-log.la	\
+	$(NULL)
+
+ck_log_system_stop_SOURCES =		\
+	ck-log-system-stop.c		\
+	$(NULL)
+
+ck_log_system_stop_LDADD =		\
+	$(HISTORY_LIBS)			\
+	$(top_builddir)/src/libck-event-log.la	\
+	$(NULL)
+
 libexec_PROGRAMS = 			\
 	ck-collect-session-info		\
 	ck-get-x11-server-pid		\
diff --git a/tools/ck-log-system-restart.c b/tools/ck-log-system-restart.c
new file mode 100644
index 0000000..e53670f
--- /dev/null
+++ b/tools/ck-log-system-restart.c
@@ -0,0 +1,174 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2008 William Jon McCann <jmccann@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
+ * 02111-1307, USA.
+ *
+ */
+
+#include "config.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <string.h>
+
+#include <locale.h>
+
+#include <glib.h>
+#include <glib/gi18n.h>
+#include <glib/gstdio.h>
+
+#include "ck-log-event.h"
+
+#define DEFAULT_LOG_FILENAME LOCALSTATEDIR "/log/ConsoleKit/history"
+
+
+/* Adapted from auditd auditd-event.c */
+static gboolean
+open_log_file (const char *filename,
+               int        *fdp,
+               FILE      **filep)
+{
+        int      flags;
+        int      fd;
+        FILE    *file;
+        gboolean ret;
+
+        ret = FALSE;
+
+        /*
+         * Likely errors on rotate: ENFILE, ENOMEM, ENOSPC
+         */
+        flags = O_WRONLY | O_APPEND;
+#ifdef O_NOFOLLOW
+        flags |= O_NOFOLLOW;
+#endif
+
+retry:
+        fd = g_open (filename, flags, 0600);
+        if (fd < 0) {
+                if (errno == ENOENT) {
+                        /* FIXME: should we just skip if file doesn't exist? */
+                        fd = g_open (filename,
+                                     O_CREAT | O_EXCL | O_APPEND,
+                                     S_IRUSR | S_IWUSR | S_IRGRP);
+                        if (fd < 0) {
+                                g_warning ("Couldn't create log file %s (%s)",
+                                           filename,
+                                           g_strerror (errno));
+                                goto out;
+                        }
+
+                        close (fd);
+                        fd = g_open (filename, flags, 0600);
+                } else if (errno == ENFILE) {
+                        /* All system descriptors used, try again... */
+                        goto retry;
+                }
+                if (fd < 0) {
+                        g_warning ("Couldn't open log file %s (%s)",
+                                   filename,
+                                   g_strerror (errno));
+                        goto out;
+                }
+        }
+
+        if (fcntl (fd, F_SETFD, FD_CLOEXEC) == -1) {
+                close (fd);
+                g_warning ("Error setting log file CLOEXEC flag (%s)",
+                           g_strerror (errno));
+                goto out;
+        }
+
+        fchown (fd, 0, 0);
+
+        file = fdopen (fd, "a");
+        if (file == NULL) {
+                g_warning ("Error setting up log descriptor (%s)",
+                           g_strerror (errno));
+                close (fd);
+                goto out;
+        }
+
+        /* Set it to line buffering */
+        setlinebuf (file);
+
+        ret = TRUE;
+
+        if (fdp != NULL) {
+                *fdp = fd;
+        }
+        if (filep != NULL) {
+                *filep = file;
+        }
+
+ out:
+
+        return ret;
+}
+
+static gboolean
+write_log_for_event (CkLogEvent *event)
+{
+        GString *str;
+        FILE    *file;
+        int      fd;
+
+        str = g_string_new (NULL);
+
+        ck_log_event_to_string (event, str);
+
+        if (! open_log_file (DEFAULT_LOG_FILENAME, &fd, &file)) {
+                exit (1);
+        }
+
+        if (file != NULL) {
+                int rc;
+
+                rc = fprintf (file, "%s\n", str->str);
+                if (rc <= 0) {
+                        g_warning ("Record was not written to disk (%s)",
+                                   g_strerror (errno));
+                }
+        } else {
+                g_warning ("Log file not open for writing");
+        }
+
+        g_string_free (str, TRUE);
+
+        return TRUE;
+}
+
+int
+main (int    argc,
+      char **argv)
+{
+        CkLogEvent event;
+
+        memset (&event, 0, sizeof (CkLogEvent));
+
+        event.type = CK_LOG_EVENT_SYSTEM_RESTART;
+        g_get_current_time (&event.timestamp);
+
+        write_log_for_event (&event);
+
+        return 0;
+}
diff --git a/tools/ck-log-system-stop.c b/tools/ck-log-system-stop.c
new file mode 100644
index 0000000..a59a1e9
--- /dev/null
+++ b/tools/ck-log-system-stop.c
@@ -0,0 +1,174 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2008 William Jon McCann <jmccann@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
+ * 02111-1307, USA.
+ *
+ */
+
+#include "config.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <string.h>
+
+#include <locale.h>
+
+#include <glib.h>
+#include <glib/gi18n.h>
+#include <glib/gstdio.h>
+
+#include "ck-log-event.h"
+
+#define DEFAULT_LOG_FILENAME LOCALSTATEDIR "/log/ConsoleKit/history"
+
+
+/* Adapted from auditd auditd-event.c */
+static gboolean
+open_log_file (const char *filename,
+               int        *fdp,
+               FILE      **filep)
+{
+        int      flags;
+        int      fd;
+        FILE    *file;
+        gboolean ret;
+
+        ret = FALSE;
+
+        /*
+         * Likely errors on rotate: ENFILE, ENOMEM, ENOSPC
+         */
+        flags = O_WRONLY | O_APPEND;
+#ifdef O_NOFOLLOW
+        flags |= O_NOFOLLOW;
+#endif
+
+retry:
+        fd = g_open (filename, flags, 0600);
+        if (fd < 0) {
+                if (errno == ENOENT) {
+                        /* FIXME: should we just skip if file doesn't exist? */
+                        fd = g_open (filename,
+                                     O_CREAT | O_EXCL | O_APPEND,
+                                     S_IRUSR | S_IWUSR | S_IRGRP);
+                        if (fd < 0) {
+                                g_warning ("Couldn't create log file %s (%s)",
+                                           filename,
+                                           g_strerror (errno));
+                                goto out;
+                        }
+
+                        close (fd);
+                        fd = g_open (filename, flags, 0600);
+                } else if (errno == ENFILE) {
+                        /* All system descriptors used, try again... */
+                        goto retry;
+                }
+                if (fd < 0) {
+                        g_warning ("Couldn't open log file %s (%s)",
+                                   filename,
+                                   g_strerror (errno));
+                        goto out;
+                }
+        }
+
+        if (fcntl (fd, F_SETFD, FD_CLOEXEC) == -1) {
+                close (fd);
+                g_warning ("Error setting log file CLOEXEC flag (%s)",
+                           g_strerror (errno));
+                goto out;
+        }
+
+        fchown (fd, 0, 0);
+
+        file = fdopen (fd, "a");
+        if (file == NULL) {
+                g_warning ("Error setting up log descriptor (%s)",
+                           g_strerror (errno));
+                close (fd);
+                goto out;
+        }
+
+        /* Set it to line buffering */
+        setlinebuf (file);
+
+        ret = TRUE;
+
+        if (fdp != NULL) {
+                *fdp = fd;
+        }
+        if (filep != NULL) {
+                *filep = file;
+        }
+
+ out:
+
+        return ret;
+}
+
+static gboolean
+write_log_for_event (CkLogEvent *event)
+{
+        GString *str;
+        FILE    *file;
+        int      fd;
+
+        str = g_string_new (NULL);
+
+        ck_log_event_to_string (event, str);
+
+        if (! open_log_file (DEFAULT_LOG_FILENAME, &fd, &file)) {
+                exit (1);
+        }
+
+        if (file != NULL) {
+                int rc;
+
+                rc = fprintf (file, "%s\n", str->str);
+                if (rc <= 0) {
+                        g_warning ("Record was not written to disk (%s)",
+                                   g_strerror (errno));
+                }
+        } else {
+                g_warning ("Log file not open for writing");
+        }
+
+        g_string_free (str, TRUE);
+
+        return TRUE;
+}
+
+int
+main (int    argc,
+      char **argv)
+{
+        CkLogEvent event;
+
+        memset (&event, 0, sizeof (CkLogEvent));
+
+        event.type = CK_LOG_EVENT_SYSTEM_STOP;
+        g_get_current_time (&event.timestamp);
+
+        write_log_for_event (&event);
+
+        return 0;
+}