Prex Home / Browse Source - Prex Version: 0.9.0

root/usr/test/shutdown/shutdown.c

DEFINITIONS

1. main
2. usage
3. shutdown

   1 /*
   2  * Copyright (c) 2006-2009, Kohsuke Ohtani
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the author nor the names of any co-contributors
  14  *    may be used to endorse or promote products derived from this software
  15  *    without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  30 /*
  31  * shutdown.c - test shutdown function.
  32  *
  33  * NOTE: This program requires CAP_PWRMGMT capability.
  34  */
  36 #include <sys/prex.h>
  37 #include <sys/ioctl.h>
  38 #include <ipc/pow.h>
  39 #include <ipc/ipc.h>
  41 #include <unistd.h>
  42 #include <err.h>
  43 #include <errno.h>
  44 #include <stdlib.h>
  45 #include <stdio.h>
  46 #include <string.h>
  47 #include <signal.h>
  49 static void usage(void);
  50 static void shutdown(int);
  52 int
main(int argc, char *argv[])
  54 {
  55         int ch, checkch, rflag;
rflag = 0;
  58         if (argc > 1) {
  59                 if (!strcmp(argv[1], "-r"))
rflag = 1;      /* reboot */
  61                 else {
usage();
exit(1);
  64                 }
  65         }
  66         /*
  67          * User confirmation is required for security.
  68          */
printf("Do you want to shutdown the system now? (y/n) ");
checkch = ch = getchar();
  71         while (ch != '\n' && ch != EOF)
ch = getchar();
  73         if (checkch != 'y')
exit(0);
shutdown(rflag);
fprintf(stderr, "Shutdown failed!\n");
exit(1);
  80 }
  82 static void
usage(void)
  84 {
fprintf(stderr, "usage: shutdown [-r]\n");
exit(1);
  88         /* NOTREACHED */
  89 }
  91 static void
shutdown(int reboot)
  93 {
object_t powobj;
  95         struct msg m;
  96         int error;
  98         if ((error = object_lookup("!pow", &powobj)) != 0)
  99                 return;
m.hdr.code = POW_SET_POWER;
m.data[0] = reboot ? PWR_REBOOT : PWR_OFF;
error = msg_send(powobj, &m, sizeof(m));
 104 }