initial checkin
[util-vserver.git] / util-vserver / src / save_ctxinfo.c
1 // $Id$    --*- c++ -*--
2
3 // Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
4 //  
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; version 2 of the License.
8 //  
9 // This program is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 // GNU General Public License for more details.
13 //  
14 // You should have received a copy of the GNU General Public License
15 // along with this program; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17
18 // Saves current ctx + vserver-info into 'argv[1] + /run' which must be a dead
19 // symlink
20
21 #ifdef HAVE_CONFIG_H
22 #  include <config.h>
23 #endif
24
25 #include "vserver.h"
26 #include "internal.h"
27 #include "util.h"
28
29 #include <sys/types.h>
30 #include <sys/stat.h>
31 #include <fcntl.h>
32 #include <stdio.h>
33 #include <limits.h>
34
35 inline static void
36 checkParams(int argc, char *argv[])
37 {
38   if (argc<3) {
39     WRITE_MSG(2, "Usage:  save_ctxinfo <VSERVER_DIR> <cmd> <args>*\n");
40     exit(255);
41   }
42 }
43
44 int main(int argc, char *argv[])
45 {
46   char          runfile[(checkParams(argc,argv),strlen(argv[1])) + sizeof("/run")];
47   char          dstfile[PATH_MAX];
48   int           fd;
49   char          buf[32];
50   ctx_t         ctx;
51   size_t        len;
52   size_t        len1 = strlen(argv[1]);
53
54   strcpy(runfile,      argv[1]);
55   strcpy(runfile+len1, "/run");
56
57   ctx=getcctx();
58   if (ctx==-1) {
59     perror("getctx()");
60     return -1;
61   }
62
63   if (readlink(runfile, dstfile, sizeof(dstfile))==-1) {
64     perror("readlink()");
65     return -1;
66   }
67
68   fd = open(dstfile, O_EXCL|O_CREAT|O_WRONLY, 0644);
69   if (fd==-1) {
70     perror("open()");
71     return -1;
72   }
73
74   len  = utilvserver_uint2str(buf, sizeof(buf), ctx, 10);
75
76   if (write(fd, buf,     len) !=len  ||
77       write(fd, "\n",    1)   !=1    ||
78       write(fd, argv[1], len1)!=len1 ||
79       write(fd, "\n",    1)   !=1) {
80     perror("write()");
81     return -1;
82   }
83
84   if (close(fd)==-1) {
85     perror("close()");
86     return -1;
87   }
88
89   execv(argv[2], argv+2);
90   perror("execv()");
91   return -1;
92 }