further locking fixes
[util-vserver.git] / util-vserver / src / vshelper-sync.c
1 // $Id$    --*- 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
79   for (;;) {
80     char        buf[512];
81     ssize_t     len;
82     fd_set      fds;
83
84     FD_ZERO(&fds);
85     FD_SET(fd, &fds);
86
87 #ifndef __linux
88 #  error vshelper relies on the Linux select() behavior (timeout holds remaining time)
89 #endif
90
91     switch (select(fd+1, &fds, 0,0, &timeout)) {
92       case 0    :  return EXIT_FAILURE; // timeout
93       case -1   :
94         perror("vshelper: select()");
95         return EXIT_FAILURE;
96       default   :  break;
97     }
98
99     assert(FD_ISSET(fd, &fds));
100
101     len = read(fd,buf,sizeof buf);
102     if (len==0) break;
103     if (len==-1) {
104       perror("vshelper-sync: read()");
105       return EXIT_FAILURE;
106     }
107   }
108
109   return EXIT_SUCCESS;
110 }