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