reap off gpg processes
[gpgme.git] / gpgme / posix-io.c
1 /* posix-io.c - Posix I/O functions
2  *      Copyright (C) 2000 Werner Koch (dd9jn)
3  *
4  * This file is part of GPGME.
5  *
6  * GPGME is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GPGME is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #ifndef HAVE_DOSISH_SYSTEM
23
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <assert.h>
28 #include <errno.h>
29 #include <sys/time.h>
30 #include <sys/types.h>
31 #include <sys/wait.h>
32 #include <signal.h>
33 #include <fcntl.h>
34 #include "syshdr.h"
35
36 #include "util.h"
37 #include "io.h"
38
39 int
40 _gpgme_io_read ( int fd, void *buffer, size_t count )
41 {
42     int nread;
43
44     DEBUG2 ("fd %d: about to read %d bytes\n", fd, (int)count );
45     do {
46         nread = read (fd, buffer, count);
47     } while (nread == -1 && errno == EINTR );
48     DEBUG2 ("fd %d: got %d bytes\n", fd, nread );
49     return nread;
50 }
51
52
53 int
54 _gpgme_io_write ( int fd, const void *buffer, size_t count )
55 {
56     int nwritten;
57
58     DEBUG2 ("fd %d: about to write %d bytes\n", fd, (int)count );
59     do {
60         nwritten = write (fd, buffer, count);
61     } while (nwritten == -1 && errno == EINTR );
62     DEBUG2 ("fd %d:          wrote %d bytes\n", fd, (int)nwritten );
63     return nwritten;
64 }
65
66 int
67 _gpgme_io_pipe ( int filedes[2], int inherit_idx )
68 {
69     /* we don't need inherit_idx in this implementation */
70     return pipe ( filedes );
71 }
72
73 int
74 _gpgme_io_close ( int fd )
75 {
76     if ( fd == -1 )
77         return -1;
78     return close (fd);
79 }
80
81 int
82 _gpgme_io_set_nonblocking ( int fd )
83 {
84     int flags;
85
86     flags = fcntl (fd, F_GETFL, 0);
87     if (flags == -1)
88         return -1;
89     flags |= O_NONBLOCK;
90     return fcntl (fd, F_SETFL, flags);
91 }
92
93
94 int
95 _gpgme_io_spawn ( const char *path, char **argv,
96                   struct spawn_fd_item_s *fd_child_list,
97                   struct spawn_fd_item_s *fd_parent_list )
98 {
99     static volatile int fixed_signals;
100     pid_t pid;
101     int i;
102
103     if ( !fixed_signals ) { 
104         struct sigaction act;
105         
106         sigaction( SIGPIPE, NULL, &act );
107         if( act.sa_handler == SIG_DFL ) {
108             act.sa_handler = SIG_IGN;
109             sigemptyset( &act.sa_mask );
110             act.sa_flags = 0;
111             sigaction( SIGPIPE, &act, NULL);
112         }
113         fixed_signals = 1;
114         /* fixme: This is not really MT safe */
115     }
116
117     
118     pid = fork ();
119     if (pid == -1) 
120         return -1;
121
122     if ( !pid ) { /* child */
123         int duped_stdin = 0;
124         int duped_stderr = 0;
125
126         /* first close all fds which will not be duped */
127         for (i=0; fd_child_list[i].fd != -1; i++ ) {
128             if (fd_child_list[i].dup_to == -1 )
129                 close (fd_child_list[i].fd);
130         }
131         /* and now dup and close the rest */
132         for (i=0; fd_child_list[i].fd != -1; i++ ) {
133             if (fd_child_list[i].dup_to != -1 ) {
134                 if ( dup2 (fd_child_list[i].fd,
135                            fd_child_list[i].dup_to ) == -1 ) {
136                     DEBUG1 ("dup2 failed in child: %s\n", strerror (errno));
137                     _exit (8);
138                 }
139                 if ( fd_child_list[i].dup_to == 0 )
140                     duped_stdin=1;
141                 if ( fd_child_list[i].dup_to == 2 )
142                     duped_stderr=1;
143                 close (fd_child_list[i].fd);
144             }
145         }
146
147         if( !duped_stdin || !duped_stderr ) {
148             int fd = open ( "/dev/null", O_RDWR );
149             if ( fd == -1 ) {
150                 DEBUG1 ("can't open `/dev/null': %s\n", strerror (errno) );
151                 _exit (8);
152             }
153             /* Make sure that the process has a connected stdin */
154             if ( !duped_stdin ) {
155                 if ( dup2 ( fd, 0 ) == -1 ) {
156                     DEBUG1("dup2(/dev/null, 0) failed: %s\n",
157                              strerror (errno) );
158                     _exit (8);
159                 }
160             }
161             if ( !duped_stderr ) {
162                 if ( dup2 ( fd, 2 ) == -1 ) {
163                     DEBUG1 ("dup2(dev/null, 2) failed: %s\n", strerror (errno));
164                     _exit (8);
165                 }
166             }
167             close (fd);
168         }
169
170         execv ( path, argv );
171         /* Hmm: in that case we could write a special status code to the
172          * status-pipe */
173         DEBUG1 ("exec of `%s' failed\n", path );
174         _exit (8);
175     } /* end child */
176     
177     /* .dup_to is not used in the parent list */
178     for (i=0; fd_parent_list[i].fd != -1; i++ ) {
179         close (fd_parent_list[i].fd);
180     }
181
182     return (int)pid;
183 }
184
185
186 int
187 _gpgme_io_waitpid ( int pid, int hang, int *r_status, int *r_signal )
188 {
189     int status;
190
191     *r_status = 0;
192     *r_signal = 0;
193     if ( waitpid ( pid, &status, hang? 0 : WNOHANG ) == pid ) {
194         if ( WIFSIGNALED (status) ) {
195             *r_status = 4; /* Need some value here */
196             *r_signal = WTERMSIG (status);
197         }
198         else if ( WIFEXITED (status) ) {
199             *r_status = WEXITSTATUS (status);
200         }
201         else {
202             *r_status = 4; /* oops */
203         }
204         return 1;
205     }
206     return 0;
207 }
208
209 int
210 _gpgme_io_kill ( int pid, int hard )
211 {
212     return kill ( pid, hard? SIGKILL : SIGTERM );
213 }
214
215
216 /*
217  * Select on the list of fds.
218  * Returns: -1 = error
219  *           0 = timeout or nothing to select
220  *          >0 = number of signaled fds
221  */
222 int
223 _gpgme_io_select ( struct io_select_fd_s *fds, size_t nfds )
224 {
225     static fd_set readfds;
226     static fd_set writefds;
227     int any, i, max_fd, n, count;
228     struct timeval timeout = { 0, 200 }; /* Use a 200ms timeout */
229     void *dbg_help;
230     
231     FD_ZERO ( &readfds );
232     FD_ZERO ( &writefds );
233     max_fd = 0;
234
235     DEBUG_BEGIN (dbg_help, "gpgme:select on [ ");
236     any = 0;
237     for ( i=0; i < nfds; i++ ) {
238         if ( fds[i].fd == -1 ) 
239             continue;
240         if ( fds[i].for_read ) {
241             assert ( !FD_ISSET ( fds[i].fd, &readfds ) );
242             FD_SET ( fds[i].fd, &readfds );
243             if ( fds[i].fd > max_fd )
244                 max_fd = fds[i].fd;
245             DEBUG_ADD1 (dbg_help, "r%d ", fds[i].fd );
246             any = 1;
247         }
248         else if ( fds[i].for_write ) {
249             assert ( !FD_ISSET ( fds[i].fd, &writefds ) );
250             FD_SET ( fds[i].fd, &writefds );
251             if ( fds[i].fd > max_fd )
252                 max_fd = fds[i].fd;
253             DEBUG_ADD1 (dbg_help, "w%d ", fds[i].fd );
254             any = 1;
255         }
256         fds[i].signaled = 0;
257     }
258     DEBUG_END (dbg_help, "]" );
259     if ( !any )
260         return 0;
261
262     do {
263         count = select ( max_fd+1, &readfds, &writefds, NULL, &timeout );
264     } while ( count < 0 && errno == EINTR);
265     if ( count < 0 ) {
266         DEBUG1 ("_gpgme_io_select failed: %s\n", strerror (errno) );
267         return -1; /* error */
268     }
269
270     DEBUG_BEGIN (dbg_help, "select OK [ " );
271     if (DEBUG_ENABLED(dbg_help)) {
272         for (i=0; i <= max_fd; i++ ) {
273             if (FD_ISSET (i, &readfds) )
274                 DEBUG_ADD1 (dbg_help, "r%d ", i );
275             if (FD_ISSET (i, &writefds) )
276                 DEBUG_ADD1 (dbg_help, "w%d ", i );
277         }
278         DEBUG_END (dbg_help, "]" );
279     }
280     
281     /* n is used to optimize it a little bit */
282     for ( n=count, i=0; i < nfds && n ; i++ ) {
283         if ( fds[i].fd == -1 ) 
284             ;
285         else if ( fds[i].for_read ) {
286             if ( FD_ISSET ( fds[i].fd, &readfds ) ) {
287                 fds[i].signaled = 1;
288                 n--;
289             }
290         }
291         else if ( fds[i].for_write ) {
292             if ( FD_ISSET ( fds[i].fd, &writefds ) ) {
293                 fds[i].signaled = 1;
294                 n--;
295             }
296         }
297     }
298     return count;
299 }
300
301
302 #endif /*!HAVE_DOSISH_SYSTEM*/
303
304
305