src/vreboot.c

   1 // $Id$
   2
   3 // Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
   4 // based on vreboot.cc by Jacques Gelinas
   5 //
   6 // This program is free software; you can redistribute it and/or modify
   7 // it under the terms of the GNU General Public License as published by
   8 // the Free Software Foundation; either version 2, or (at your option)
   9 // any later version.
  10 //
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19
  20 /*
  21         Used to send a reboot message to the reboot manager. It opens /dev/reboot
  22         and write "reboot\n".
  23 */
  24 #ifdef HAVE_CONFIG_H
  25 #  include <config.h>
  26 #endif
  27
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <unistd.h>
  31 #include <sys/types.h>
  32 #include <errno.h>
  33 #include <sys/stat.h>
  34 #include <sys/time.h>
  35 #include <sys/socket.h>
  36 #include <sys/un.h>
  37 #include <stdbool.h>
  38
  39 /*
  40         Connect to a unix domain socket
  41 */
  42 static int vreboot_connect (const char *sockpath, bool showerror)
  43 {
  44         int ret = -1;
  45         int fd =  socket (AF_UNIX,SOCK_STREAM,0);
  46         if (fd == -1){
  47                 if (showerror) perror("socket client");
  48         }else{
  49                 struct sockaddr_un un;
  50                 int s;
  51
  52                 un.sun_family = AF_UNIX;
  53                 strcpy (un.sun_path,sockpath);
  54                 s = connect(fd,(struct sockaddr*)&un,sizeof(un));
  55                 if (s == -1){
  56                         if (showerror) fprintf (stderr,"connect %s (%s)\n"
  57                                 ,sockpath,strerror(errno));
  58                 }else{
  59                         ret = fd;
  60                 }
  61         }
  62         return ret;
  63 }
  64
  65 static void usage()
  66 {
  67         fprintf (stderr,"vreboot version %s\n",VERSION);
  68         fprintf (stderr,"\n");
  69         fprintf (stderr,"vreboot [ --socket path ]\n");
  70         fprintf (stderr,"vhalt   [ --socket path ]\n");
  71         fprintf (stderr,"vreboot request a reboot or a halt of a virtual server\n");
  72 }
  73
  74 int main (int argc, char *argv[])
  75 {
  76         int ret = -1;
  77         int i;
  78         const char *sockpath = "/dev/reboot";
  79         for (i=1; i<argc; i++){
  80                 const char *arg = argv[i];
  81                 const char *opt = argv[i+1];
  82                 if (strcmp(arg,"--socket")==0){
  83                         sockpath = opt;
  84                         i++;
  85                 }else if (strcmp(arg,"--help")==0){
  86                         break;
  87
  88                 }else{
  89                         fprintf (stderr,"Invalid option %s\n",arg);
  90                         break;
  91                 }
  92         }
  93         if (argc != i){
  94                 usage();
  95         }else{
  96                 int fd = vreboot_connect (sockpath,true);
  97                 if (fd != -1){
  98                         if (strstr(argv[0],"halt")!=NULL){
  99                                 write (fd,"halt\n",5);
 100                         }else{
 101                                 write (fd,"reboot\n",7);
 102                         }
 103                         close (fd);
 104                         ret = 0;
 105                 }
 106         }
 107         return ret;
 108 }
 109