util-vserver 0.30.214
[util-vserver.git] / src / fakerunlevel.c
1 // $Id: fakerunlevel.c 2562 2007-07-12 21:59:06Z dhozac $
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 #ifdef HAVE_CONFIG_H
26 #  include <config.h>
27 #endif
28
29 #include "util.h"
30
31 #include <utmp.h>
32 #include <stdio.h>
33 #include <string.h>
34 #include <stdlib.h>
35 #include <errno.h>
36 #include <fcntl.h>
37 #include <grp.h>
38 #include <sys/types.h>
39 #include <sys/stat.h>
40
41 #define ENSC_WRAPPERS_PREFIX    "fakerunlevel: "
42 #define ENSC_WRAPPERS_UNISTD    1
43 #define ENSC_WRAPPERS_FCNTL     1
44 #include <wrappers.h>
45
46 int     wrapper_exit_code = 1;
47
48 static void
49 showHelp(int fd, int exit_code)
50 {
51   WRITE_MSG(fd,
52             "Usage: fakerunlevel <runlevel> <utmp_file>\n\n"
53             "Put a runlevel record in file <utmp_file>\n\n"
54             "Please report bugs to " PACKAGE_BUGREPORT "\n");
55
56   exit(exit_code);
57 }
58
59 static void
60 showVersion()
61 {
62   WRITE_MSG(1,
63             "fakerunlevel " VERSION "\n"
64             "This program is part of " PACKAGE_STRING "\n\n"
65             "Copyright (C) 2003 Enrico Scholz\n"
66             VERSION_COPYRIGHT_DISCLAIMER);
67   exit(0);
68 }
69
70 int main (int argc, char *argv[])
71 {
72   if (argc==1) showHelp(2,1);
73   if (strcmp(argv[1], "--help")==0)    showHelp(1,0);
74   if (strcmp(argv[1], "--version")==0) showVersion();
75   if (argc!=3) showHelp(2,1);
76
77   {
78     int  const          runlevel = atoi(argv[1]);
79     char const * const  fname    = argv[2];
80     int                 fd;
81     struct utmp         ut;
82     
83     gid_t               gid;
84     char                *gid_str = getenv("UTMP_GID");
85     
86     if (runlevel<0 || runlevel>9) showHelp(2,1);
87
88     Echroot(".");
89     Echdir("/");
90
91       // Real NSS is too expensive/insecure in this state; therefore, use the
92       // value detected at ./configure stage or overridden by $UTMP_GID
93       // env-variable
94     gid = gid_str ? atoi(gid_str) : UTMP_GID;
95     Esetgroups(1, &gid);
96     Esetgid(gid);
97
98     if (getgid()!=gid) {
99       WRITE_MSG(2, "fakerunlevel: Failed to set group\n");
100       return EXIT_FAILURE;
101     }
102
103     umask(002);
104     fd = EopenD(fname, O_WRONLY|O_CREAT|O_APPEND, 0664);
105     Eclose(fd);
106
107     utmpname (fname);
108     setutent();
109     memset (&ut,0,sizeof(ut));
110     ut.ut_type = RUN_LVL;
111     ut.ut_pid  = ('#' << 8) + runlevel+'0';
112     pututline (&ut);
113     endutent();
114   }
115
116   return EXIT_SUCCESS;
117 }