chiark / gitweb /
util: make asynchronous_close() really work like an asynchronous version of safe_close()
[elogind.git] / 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 void help(void) {
138         printf("%s COMMAND\n\n"
139                "Suspend the system, hibernate the system, or both.\n\n"
140                "Commands:\n"
141                "  -h --help            Show this help and exit\n"
142                "  --version            Print version string and exit\n"
143                "  suspend              Suspend the system\n"
144                "  hibernate            Hibernate the system\n"
145                "  hybrid-sleep         Both hibernate and suspend the system\n"
146                , program_invocation_short_name);
147 }
148
149 static int parse_argv(int argc, char *argv[]) {
150         enum {
151                 ARG_VERSION = 0x100,
152         };
153
154         static const struct option options[] = {
155                 { "help",         no_argument,       NULL, 'h'           },
156                 { "version",      no_argument,       NULL, ARG_VERSION   },
157                 {}
158         };
159
160         int c;
161
162         assert(argc >= 0);
163         assert(argv);
164
165         while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)
166                 switch(c) {
167                 case 'h':
168                         help();
169                         return 0; /* done */
170
171                 case ARG_VERSION:
172                         puts(PACKAGE_STRING);
173                         puts(SYSTEMD_FEATURES);
174                         return 0 /* done */;
175
176                 case '?':
177                         return -EINVAL;
178
179                 default:
180                         assert_not_reached("Unhandled option");
181                 }
182
183         if (argc - optind != 1) {
184                 log_error("Usage: %s COMMAND",
185                           program_invocation_short_name);
186                 return -EINVAL;
187         }
188
189         arg_verb = argv[optind];
190
191         if (!streq(arg_verb, "suspend") &&
192             !streq(arg_verb, "hibernate") &&
193             !streq(arg_verb, "hybrid-sleep")) {
194                 log_error("Unknown command '%s'.", arg_verb);
195                 return -EINVAL;
196         }
197
198         return 1 /* work to do */;
199 }
200
201 int main(int argc, char *argv[]) {
202         _cleanup_strv_free_ char **modes = NULL, **states = NULL;
203         int r;
204
205         log_set_target(LOG_TARGET_AUTO);
206         log_parse_environment();
207         log_open();
208
209         r = parse_argv(argc, argv);
210         if (r <= 0)
211                 goto finish;
212
213         r = parse_sleep_config(arg_verb, &modes, &states);
214         if (r < 0)
215                 goto finish;
216
217         r = execute(modes, states);
218
219 finish:
220         return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
221 }