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