src/fakerunlevel.c

   1 // $Id: fakerunlevel.c,v 1.1 2003/09/29 22:01:57 ensc Exp $
   2
   3 // Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
   4 // based on fakerunlevel.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         This program add a RUNLEVEL record in a utmp file.
  22         This is used when a vserver lack a private init process
  23         so runlevel properly report the fake runlevel.
  24 */
  25 #include <utmp.h>
  26 #include <stdio.h>
  27 #include <string.h>
  28 #include <stdlib.h>
  29 #include <errno.h>
  30
  31 static void usage()
  32 {
  33         fprintf (stderr,"fakerunlevel version %s\n",VERSION);
  34         fprintf (stderr
  35                 ,"\n"
  36                  "fakerunlevel runlevel utmp_file\n"
  37                  "\n"
  38                  "Put a runlevel record in file utmp_file\n");
  39 }
  40
  41 int main (int argc, char *argv[])
  42 {
  43         if (argc != 3){
  44                 usage();
  45         }else{
  46                 int runlevel = atoi(argv[1]);
  47                 const char *fname = argv[2];
  48                 if (runlevel < 1 || runlevel > 5){
  49                         usage();
  50                 }else{
  51                         // Make sure the file exist
  52                         FILE *fout = fopen (fname,"a");
  53                         if (fout == NULL){
  54                                 fprintf (stderr,"Can't open file %s (%s)\n",fname
  55                                         ,strerror(errno));
  56                         }else{
  57                                 struct utmp ut;
  58
  59                                 fclose (fout);
  60                                 utmpname (fname);
  61                                 setutent();
  62                                 memset (&ut,0,sizeof(ut));
  63                                 ut.ut_type = RUN_LVL;
  64                                 ut.ut_pid = ('#' << 8) + runlevel+'0';
  65                                 pututline (&ut);
  66                                 endutent();
  67                         }
  68                 }
  69         }
  70
  71         return 0;
  72 }
  73