added Vector_zeroEnd() function
[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 #include <errno.h>
35
36 #define ENSC_WRAPPERS_PREFIX    "save_ctxinfo: "
37 #define ENSC_WRAPPERS_VSERVER   1
38 #define ENSC_WRAPPERS_UNISTD    1
39 #define ENSC_WRAPPERS_FCNTL     1
40 #include <wrappers.h>
41
42 int     wrapper_exit_code = 255;
43
44 inline static void
45 checkParams(int argc, char UNUSED * argv[])
46 {
47   if (argc<3) {
48     WRITE_MSG(2, "Usage:  save_ctxinfo <VSERVER_DIR> <cmd> <args>*\n");
49     exit(255);
50   }
51 }
52
53 int main(int argc, char *argv[])
54 {
55   char          runfile[(checkParams(argc,argv),strlen(argv[1])) + sizeof("/run.rev/99999")];
56   char          dstfile[PATH_MAX];
57   int           fd;
58   char          buf[sizeof(int)*3+2];
59   xid_t         ctx;
60   ssize_t       len;
61   ssize_t       len1 = strlen(argv[1]);
62
63   strcpy(runfile,      argv[1]);
64   strcpy(runfile+len1, "/run");
65
66   ctx=Evc_get_task_xid(0);
67
68   if (ctx==0) {
69     WRITE_MSG(2, "save_ctxinfo: Can not operate in context 0\n");
70     return 255;
71   }
72
73   if (reinterpret_cast(unsigned int)(ctx)>99999) {
74     WRITE_MSG(2, "save_ctxinfo: unexpected context\n");
75     return 255;
76   }
77
78   EreadlinkD(runfile, dstfile, sizeof(dstfile));
79   len  = utilvserver_fmt_uint(buf, ctx);
80
81   fd = EopenD(dstfile, O_EXCL|O_CREAT|O_WRONLY, 0644);
82   if (write(fd, buf,     len) !=len  ||
83       write(fd, "\n",    1)   !=1) {
84     perror("save_ctxinfo: write()");
85     return -1;
86   }
87   Eclose(fd);
88
89   strcat(runfile, ".rev/");
90   strncat(runfile, buf, len);
91   unlink(runfile);
92   EsymlinkD(argv[1], runfile);
93
94   Eexecv(argv[2], argv+2);
95 }