#include <errno.h>

#include <getopt.h>

#include <stdio.h>

#include "sd-messages.h"

#include "def.h"

#include "fd-util.h"

#include "fileio.h"

#include "log.h"

#include "sleep-config.h"

#include "string-util.h"

#include "strv.h"

#include "util.h"

static char* arg_verb = NULL;

static int write_mode(char **modes) {

int r = 0;

char **mode;

STRV_FOREACH(mode, modes) {

int k;

k = write_string_file("/sys/power/disk", *mode, 0);

if (k == 0)

return 0;

log_debug_errno(k, "Failed to write '%s' to /sys/power/disk: %m",

*mode);

if (r == 0)

r = k;

}

if (r < 0)

log_error_errno(r, "Failed to write mode to /sys/power/disk: %m");

return r;

}

static int write_state(FILE **f, char **states) {

char **state;

int r = 0;

STRV_FOREACH(state, states) {

int k;

k = write_string_stream(*f, *state, true);

if (k == 0)

return 0;

log_debug_errno(k, "Failed to write '%s' to /sys/power/state: %m",

*state);

if (r == 0)

r = k;

fclose(*f);

*f = fopen("/sys/power/state", "we");

if (!*f)

return log_error_errno(errno, "Failed to open /sys/power/state: %m");

}

return r;

}

static int execute(char **modes, char **states) {

char *arguments[] = {

NULL,

(char*) "pre",

arg_verb,

NULL

};

static const char* const dirs[] = {SYSTEM_SLEEP_PATH, NULL};

int r;

_cleanup_fclose_ FILE *f = NULL;

/* This file is opened first, so that if we hit an error,

f = fopen("/sys/power/state", "we");

if (!f)

return log_error_errno(errno, "Failed to open /sys/power/state: %m");

/* Configure the hibernation mode */

r = write_mode(modes);

if (r < 0)

return r;

execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);

log_struct(LOG_INFO,

LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_START),

LOG_MESSAGE("Suspending system..."),

"SLEEP=%s", arg_verb,

NULL);

r = write_state(&f, states);

if (r < 0)

return r;

log_struct(LOG_INFO,

LOG_MESSAGE_ID(SD_MESSAGE_SLEEP_STOP),

LOG_MESSAGE("System resumed."),

"SLEEP=%s", arg_verb,

NULL);

arguments[1] = (char*) "post";

execute_directories(dirs, DEFAULT_TIMEOUT_USEC, arguments);

return r;

}

static void help(void) {

printf("%s COMMAND\n\n"

"Suspend the system, hibernate the system, or both.\n\n"

"Commands:\n"

" -h --help Show this help and exit\n"

" --version Print version string and exit\n"

" suspend Suspend the system\n"

" hibernate Hibernate the system\n"

" hybrid-sleep Both hibernate and suspend the system\n"

, program_invocation_short_name);

}

static int parse_argv(int argc, char *argv[]) {

enum {

ARG_VERSION = 0x100,

};

static const struct option options[] = {

{ "help", no_argument, NULL, 'h' },

{ "version", no_argument, NULL, ARG_VERSION },

{}

};

int c;

assert(argc >= 0);

assert(argv);

while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0)

switch(c) {

case 'h':

help();

return 0; /* done */

case ARG_VERSION:

return version();

case '?':

return -EINVAL;

default:

assert_not_reached("Unhandled option");

}

if (argc - optind != 1) {

log_error("Usage: %s COMMAND",

program_invocation_short_name);

return -EINVAL;

}

arg_verb = argv[optind];

if (!streq(arg_verb, "suspend") &&

!streq(arg_verb, "hibernate") &&

!streq(arg_verb, "hybrid-sleep")) {

log_error("Unknown command '%s'.", arg_verb);

return -EINVAL;

}

return 1 /* work to do */;

}

int main(int argc, char *argv[]) {

_cleanup_strv_free_ char **modes = NULL, **states = NULL;

int r;

log_set_target(LOG_TARGET_AUTO);

log_parse_environment();

log_open();

r = parse_argv(argc, argv);

if (r <= 0)

goto finish;

r = parse_sleep_config(arg_verb, &modes, &states);

if (r < 0)

goto finish;

r = execute(modes, states);

finish:

return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;

}