src/journal/cat.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2012 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <stdio.h>
  23 #include <getopt.h>
  24 #include <assert.h>
  25 #include <unistd.h>
  26 #include <stdlib.h>
  27 #include <errno.h>
  28 #include <sys/fcntl.h>
  29
  30 #include <systemd/sd-journal.h>
  31
  32 #include "util.h"
  33 #include "build.h"
  34
  35 static char *arg_identifier = NULL;
  36 static int arg_priority = LOG_INFO;
  37 static bool arg_level_prefix = true;
  38
  39 static int help(void) {
  40
  41         printf("%s [OPTIONS...] {COMMAND} ...\n\n"
  42                "Execute process with stdout/stderr connected to the journal.\n\n"
  43                "  -h --help               Show this help\n"
  44                "     --version            Show package version\n"
  45                "  -t --identifier=STRING  Set syslog identifier\n"
  46                "  -p --priority=PRIORITY  Set priority value (0..7)\n"
  47                "     --level-prefix=BOOL  Control whether level prefix shall be parsed\n",
  48                program_invocation_short_name);
  49
  50         return 0;
  51 }
  52
  53 static int parse_argv(int argc, char *argv[]) {
  54
  55         enum {
  56                 ARG_VERSION = 0x100,
  57                 ARG_LEVEL_PREFIX
  58         };
  59
  60         static const struct option options[] = {
  61                 { "help",         no_argument,       NULL, 'h'              },
  62                 { "version",      no_argument,       NULL, ARG_VERSION      },
  63                 { "identifier",   required_argument, NULL, 't'              },
  64                 { "priority",     required_argument, NULL, 'p'              },
  65                 { "level-prefix", required_argument, NULL, ARG_LEVEL_PREFIX },
  66                 { NULL,           0,                 NULL, 0                }
  67         };
  68
  69         int c;
  70
  71         assert(argc >= 0);
  72         assert(argv);
  73
  74         while ((c = getopt_long(argc, argv, "+ht:p:", options, NULL)) >= 0) {
  75
  76                 switch (c) {
  77
  78                 case 'h':
  79                         help();
  80                         return 0;
  81
  82                 case ARG_VERSION:
  83                         puts(PACKAGE_STRING);
  84                         puts(DISTRIBUTION);
  85                         puts(SYSTEMD_FEATURES);
  86                         return 0;
  87
  88                 case 't':
  89                         free(arg_identifier);
  90                         if (isempty(optarg))
  91                                 arg_identifier = NULL;
  92                         else {
  93                                 arg_identifier = strdup(optarg);
  94                                 if (!arg_identifier) {
  95                                         log_error("Out of memory.");
  96                                         return -ENOMEM;
  97                                 }
  98                         }
  99                         break;
 100
 101                 case 'p':
 102                         arg_priority = log_level_from_string(optarg);
 103                         if (arg_priority < 0) {
 104                                 log_error("Failed to parse priority value.");
 105                                 return arg_priority;
 106                         }
 107                         break;
 108
 109                 case ARG_LEVEL_PREFIX: {
 110                         int k;
 111
 112                         k = parse_boolean(optarg);
 113                         if (k < 0) {
 114                                 log_error("Failed to parse level prefix value.");
 115                                 return k;
 116                         }
 117                         arg_level_prefix = k;
 118                         break;
 119                 }
 120
 121                 default:
 122                         log_error("Unknown option code %c", c);
 123                         return -EINVAL;
 124                 }
 125         }
 126
 127         return 1;
 128 }
 129
 130 int main(int argc, char *argv[]) {
 131         int r, fd = -1, saved_stderr = -1;
 132
 133         log_parse_environment();
 134         log_open();
 135
 136         r = parse_argv(argc, argv);
 137         if (r <= 0)
 138                 goto finish;
 139
 140         fd = sd_journal_stream_fd(arg_identifier, arg_priority, arg_level_prefix);
 141         if (fd < 0) {
 142                 log_error("Failed to create stream fd: %s", strerror(-fd));
 143                 r = fd;
 144                 goto finish;
 145         }
 146
 147         saved_stderr = fcntl(STDERR_FILENO, F_DUPFD_CLOEXEC, 3);
 148
 149         if (dup3(fd, STDOUT_FILENO, 0) < 0 ||
 150             dup3(fd, STDERR_FILENO, 0) < 0) {
 151                 log_error("Failed to duplicate fd: %m");
 152                 r = -errno;
 153                 goto finish;
 154         }
 155
 156         if (fd >= 3)
 157                 close_nointr_nofail(fd);
 158
 159         fd = -1;
 160
 161         if (argc <= optind)
 162                 execl("/bin/cat", "/bin/cat", NULL);
 163         else
 164                 execvp(argv[optind], argv + optind);
 165
 166         r = -errno;
 167
 168         /* Let's try to restore a working stderr, so we can print the error message */
 169         if (saved_stderr >= 0)
 170                 dup3(saved_stderr, STDERR_FILENO, 0);
 171
 172         log_error("Failed to execute process: %s", strerror(-r));
 173
 174 finish:
 175         if (fd >= 0)
 176                 close_nointr_nofail(fd);
 177
 178         if (saved_stderr >= 0)
 179                 close_nointr_nofail(saved_stderr);
 180
 181         return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
 182 }