ctrlaltdel: use err() instead of fprintf() and exit()

[kzak@redhat.com: - remove unnecessary program name from err(),
                  - use program_invocation_short_name]

Signed-off-by: Marek Polacek <mmpolacek@gmail.com>
Signed-off-by: Karel Zak <kzak@redhat.com>
This commit is contained in:
Marek Polacek 2010-10-17 11:01:22 +02:00 committed by Karel Zak
parent 026ed2cedd
commit 0c3e520240
1 changed files with 18 additions and 29 deletions

View File

@ -1,48 +1,37 @@
/*
* ctrlaltdel.c - Set the function of the Ctrl-Alt-Del combination
* Created 4-Jul-92 by Peter Orbaek <poe@daimi.aau.dk>
* ftp://ftp.daimi.aau.dk/pub/linux/poe/
* 1999-02-22 Arkadiusz Mikiewicz <misiek@pld.ORG.PL>
* - added Native Language Support
*
*/
#include <err.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include "linux_reboot.h"
#include "nls.h"
int
main(int argc, char *argv[]) {
int main(int argc, char *argv[])
{
setlocale(LC_ALL, "");
bindtextdomain(PACKAGE, LOCALEDIR);
textdomain(PACKAGE);
if(geteuid()) {
fprintf(stderr,
_("You must be root to set the Ctrl-Alt-Del behaviour.\n"));
exit(1);
}
if (geteuid())
errx(EXIT_FAILURE,
_("You must be root to set the Ctrl-Alt-Del behaviour"));
if(argc == 2 && !strcmp("hard", argv[1])) {
if(my_reboot(LINUX_REBOOT_CMD_CAD_ON) < 0) {
perror("ctrlaltdel: reboot");
exit(1);
}
} else if(argc == 2 && !strcmp("soft", argv[1])) {
if(my_reboot(LINUX_REBOOT_CMD_CAD_OFF) < 0) {
perror("ctrlaltdel: reboot");
exit(1);
}
} else {
fprintf(stderr, _("Usage: ctrlaltdel hard|soft\n"));
exit(1);
}
exit(0);
if (argc == 2 && !strcmp("hard", argv[1])) {
if (my_reboot(LINUX_REBOOT_CMD_CAD_ON) < 0)
err(EXIT_FAILURE, "reboot");
} else if (argc == 2 && !strcmp("soft", argv[1])) {
if (my_reboot(LINUX_REBOOT_CMD_CAD_OFF) < 0)
err(EXIT_FAILURE, "reboot");
} else
errx(EXIT_FAILURE, _("Usage: %s hard|soft"),
program_invocation_short_name);
return EXIT_SUCCESS;
}