921f12d4f3d6c6fbef2c025df94fa16d6a503f51
[util-vserver.git] / src / vshelper-sync.c
1 // $Id: vshelper-sync.c,v 1.5 2005/05/05 09:21:21 ensc Exp $    --*- c -*--
2
3 // Copyright (C) 2004 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
19 #ifdef HAVE_CONFIG_H
20 #  include <config.h>
21 #endif
22
23 #include "util.h"
24
25 #include <stdlib.h>
26 #include <fcntl.h>
27 #include <unistd.h>
28 #include <signal.h>
29 #include <stdio.h>
30 #include <assert.h>
31
32 static void
33 showHelp(int fd, char const *cmd, int res)
34 {
35   WRITE_MSG(fd, "Usage:  ");
36   WRITE_STR(fd, cmd);
37   WRITE_MSG(fd,
38             " [--] <pipe> <timeout>\n\n"
39             "Please report bugs to " PACKAGE_BUGREPORT "\n");
40   exit(res);
41 }
42
43 static void
44 showVersion()
45 {
46   WRITE_MSG(1,
47             "vshelper-sync " VERSION " -- waits for data from a pipe"
48             "This program is part of " PACKAGE_STRING "\n\n"
49             "Copyright (C) 2004 Enrico Scholz\n"
50             VERSION_COPYRIGHT_DISCLAIMER);
51   exit(0);
52 }
53
54 int main(int argc, char *argv[])
55 {
56   int                   fd;
57   int                   idx = 1;
58   struct timeval        timeout;
59
60   if (argc>=2) {
61     if (strcmp(argv[1], "--help")   ==0) showHelp(1, argv[0], 0);
62     if (strcmp(argv[1], "--version")==0) showVersion();
63     if (strcmp(argv[1], "--")       ==0) ++idx;
64   }
65
66   if (argc<idx+2) {
67     WRITE_MSG(2, "Not enough parameters; use '--help' for more information\n");
68     return EXIT_FAILURE;
69   }
70
71   fd = open(argv[idx], O_RDONLY|O_NONBLOCK, 0);
72   if (fd==-1) {
73     perror("vshelper-sync: open()");
74     return EXIT_FAILURE;
75   }
76
77   timeout.tv_sec  = atoi(argv[idx+1]);
78   timeout.tv_usec = 0;
79
80   for (;;) {
81     char        buf[512];
82     ssize_t     len;
83     fd_set      fds;
84
85     FD_ZERO(&fds);
86     FD_SET(fd, &fds);
87
88 #ifndef __linux__
89 #  error vshelper relies on the Linux select() behavior (timeout holds remaining time)
90 #endif
91
92     switch (select(fd+1, &fds, 0,0, &timeout)) {
93       case 0    :  return EXIT_FAILURE; // timeout
94       case -1   :
95         perror("vshelper: select()");
96         return EXIT_FAILURE;
97       default   :  break;
98     }
99
100     assert(FD_ISSET(fd, &fds));
101
102     len = read(fd,buf,sizeof buf);
103     if (len==0) break;
104     if (len==-1) {
105       perror("vshelper-sync: read()");
106       return EXIT_FAILURE;
107     }
108   }
109
110   return EXIT_SUCCESS;
111 }