vreboot.c (2628B)
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