src/sleep/sleep.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2012 Lennart Poettering
   5   Copyright 2013 Zbigniew Jędrzejewski-Szmek
   6
   7   systemd is free software; you can redistribute it and/or modify it
   8   under the terms of the GNU Lesser General Public License as published by
   9   the Free Software Foundation; either version 2.1 of the License, or
  10   (at your option) any later version.
  11
  12   systemd is distributed in the hope that it will be useful, but
  13   WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15   Lesser General Public License for more details.
  16
  17   You should have received a copy of the GNU Lesser General Public License
  18   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19 ***/
  20
  21 #include <errno.h>
  22 #include <getopt.h>
  23 #include <stdio.h>
  24
  25 #include "sd-messages.h"
  26
  27 #include "def.h"
  28 #include "exec-util.h"
  29 #include "fd-util.h"
  30 #include "fileio.h"
  31 #include "log.h"
  32 #include "sleep-config.h"
  33 #include "string-util.h"
  34 #include "strv.h"
  35 #include "util.h"
  36
  37 static char* arg_verb = NULL;
  38
  39 static int write_mode(char **modes) {
  40         int r = 0;
  41         char **mode;
  42
  43         STRV_FOREACH(mode, modes) {
  44                 int k;
  45
  46                 k = write_string_file("/sys/power/disk", *mode, 0);
  47                 if (k == 0)
  48                         return 0;
  49
  50                 log_debug_errno(k, "Failed to write '%s' to /sys/power/disk: %m",
  51                                 *mode);
  52                 if (r == 0)
  53                         r = k;
  54         }
  55
  56         if (r < 0)
  57                 log_error_errno(r, "Failed to write mode to /sys/power/disk: %m");
  58
  59         return r;
  60 }
  61
  62 static int write_state(FILE **f, char **states) {
  63         char **state;
  64         int r = 0;
  65
  66         STRV_FOREACH(state, states) {
  67                 int k;
  68
  69                 k = write_string_stream(*f, *state, true);
  70                 if (k == 0)
  71                         return 0;
  72                 log_debug_errno(k, "Failed to write '%s' to /sys/power/state: %m",
  73                                 *state);
  74                 if (r == 0)
  75                         r = k;
  76
  77                 fclose(*f);
  78                 *f = fopen("/sys/power/state", "we");
  79                 if (!*f)
  80                         return log_error_errno(errno, "Failed to open /sys/power/state: %m");
  81         }
  82
  83         return r;
  84 }
  85
  86 static int execute(char **modes, char **states) {
  87
  88         char *arguments[] = {
  89                 NULL,
  90                 (char*) "pre",
  91                 arg_verb,
  92                 NULL
  93         };
  94         static const char* const dirs[] = {SYSTEM_SLEEP_PATH, NULL};
  95
  96         int r;
  97         _cleanup_fclose_ FILE *f = NULL;
  98
  99         /* This file is opened first, so that if we hit an error,
 100          * we can abort before modifying any state. */
 101         f = fopen("/sys/power/state", "we");
 102         if (!f)
 103                 return log_error_errno(errno, "Failed to open /sys/power/state: %m");
 104
 105         /* Configure the hibernation mode */
 106         r = write_mode(modes);
 107         if (r < 0)
 108                 return r;
 109
 110         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, NULL, NULL, arguments);
 111
 112         log_struct(LOG_INFO,
 113                    "MESSAGE_ID=" SD_MESSAGE_SLEEP_START_STR,
 114                    LOG_MESSAGE("Suspending system..."),
 115                    "SLEEP=%s", arg_verb,
 116                    NULL);
 117
 118         r = write_state(&f, states);
 119         if (r < 0)
 120                 return r;
 121
 122         log_struct(LOG_INFO,
 123                    "MESSAGE_ID=" SD_MESSAGE_SLEEP_STOP_STR,
 124                    LOG_MESSAGE("System resumed."),
 125                    "SLEEP=%s", arg_verb,
 126                    NULL);
 127
 128         arguments[1] = (char*) "post";
 129         execute_directories(dirs, DEFAULT_TIMEOUT_USEC, NULL, NULL, arguments);
 130
 131         return r;
 132 }
 133
 134 static void help(void) {
 135         printf("%s COMMAND\n\n"
 136                "Suspend the system, hibernate the system, or both.\n\n"
 137                "Commands:\n"
 138                "  -h --help            Show this help and exit\n"
 139                "  --version            Print version string and exit\n"
 140                "  suspend              Suspend the system\n"
 141                "  hibernate            Hibernate the system\n"
 142                "  hybrid-sleep         Both hibernate and suspend the system\n"
 143                , program_invocation_short_name);
 144 }
 145
 146 static int parse_argv(int argc, char *argv[]) {
 147         enum {
 148                 ARG_VERSION = 0x100,
 149         };
 150
 151         static const struct option options[] = {
 152                 { "help",         no_argument,       NULL, 'h'           },
 153                 { "version",      no_argument,       NULL, ARG_VERSION   },
 154                 {}
 155         };
 156
 157         int c;
 158
 159         assert(argc >= 0);
 160         assert(argv);
 161
 162         while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
 163                 switch(c) {
 164                 case 'h':
 165                         help();
 166                         return 0; /* done */
 167
 168                 case ARG_VERSION:
 169                         return version();
 170
 171                 case '?':
 172                         return -EINVAL;
 173
 174                 default:
 175                         assert_not_reached("Unhandled option");
 176                 }
 177
 178         if (argc - optind != 1) {
 179                 log_error("Usage: %s COMMAND",
 180                           program_invocation_short_name);
 181                 return -EINVAL;
 182         }
 183
 184         arg_verb = argv[optind];
 185
 186         if (!streq(arg_verb, "suspend") &&
 187             !streq(arg_verb, "hibernate") &&
 188             !streq(arg_verb, "hybrid-sleep")) {
 189                 log_error("Unknown command '%s'.", arg_verb);
 190                 return -EINVAL;
 191         }
 192
 193         return 1 /* work to do */;
 194 }
 195
 196 int main(int argc, char *argv[]) {
 197         _cleanup_strv_free_ char **modes = NULL, **states = NULL;
 198         int r;
 199
 200         log_set_target(LOG_TARGET_AUTO);
 201         log_parse_environment();
 202         log_open();
 203
 204         r = parse_argv(argc, argv);
 205         if (r <= 0)
 206                 goto finish;
 207
 208         r = parse_sleep_config(arg_verb, &modes, &states);
 209         if (r < 0)
 210                 goto finish;
 211
 212         r = execute(modes, states);
 213
 214 finish:
 215         return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
 216 }