Commit | Line | Data |
---|---|---|
83723896 FT |
1 | /* |
2 | ashd - A Sane HTTP Daemon | |
3 | Copyright (C) 2008 Fredrik Tolf <fredrik@dolda2000.com> | |
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, either version 3 of the License, or | |
8 | (at your option) any later version. | |
9 | ||
10 | This program is distributed in the hope that it will be useful, | |
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 | GNU General Public License for more details. | |
14 | ||
15 | You should have received a copy of the GNU General Public License | |
16 | along with this program. If not, see <http://www.gnu.org/licenses/>. | |
17 | */ | |
18 | ||
19 | #include <stdlib.h> | |
e1cdf02e | 20 | #include <stdio.h> |
0b7964e9 | 21 | #include <unistd.h> |
e1cdf02e | 22 | #include <fcntl.h> |
83723896 | 23 | #include <string.h> |
83723896 | 24 | #include <errno.h> |
e1cdf02e | 25 | #include <sys/socket.h> |
83723896 FT |
26 | |
27 | #ifdef HAVE_CONFIG_H | |
28 | #include <config.h> | |
29 | #endif | |
30 | #include <log.h> | |
31 | #include <utils.h> | |
32 | #include <mt.h> | |
33 | #include <mtio.h> | |
34 | ||
e1cdf02e FT |
35 | struct stdiofd { |
36 | int fd; | |
37 | int sock; | |
38 | int timeout; | |
39 | }; | |
40 | ||
41 | static ssize_t mtread(void *cookie, char *buf, size_t len) | |
42 | { | |
43 | struct stdiofd *d = cookie; | |
44 | int ev; | |
45 | ssize_t ret; | |
46 | ||
47 | while(1) { | |
48 | ret = read(d->fd, buf, len); | |
49 | if((ret < 0) && (errno == EAGAIN)) { | |
50 | ev = block(d->fd, EV_READ, d->timeout); | |
51 | if(ev < 0) { | |
52 | /* If we just go on, we should get the real error. */ | |
53 | continue; | |
54 | } else if(ev == 0) { | |
55 | errno = ETIMEDOUT; | |
56 | return(-1); | |
57 | } else { | |
58 | continue; | |
59 | } | |
60 | } else { | |
61 | return(ret); | |
62 | } | |
63 | } | |
64 | } | |
65 | ||
66 | static ssize_t mtwrite(void *cookie, const char *buf, size_t len) | |
67 | { | |
68 | struct stdiofd *d = cookie; | |
69 | int ev; | |
9dc25d3d | 70 | size_t off; |
e1cdf02e FT |
71 | ssize_t ret; |
72 | ||
9dc25d3d FT |
73 | off = 0; |
74 | while(off < len) { | |
e1cdf02e | 75 | if(d->sock) |
9dc25d3d | 76 | ret = send(d->fd, buf + off, len - off, MSG_DONTWAIT | MSG_NOSIGNAL); |
e1cdf02e | 77 | else |
9dc25d3d FT |
78 | ret = write(d->fd, buf + off, len - off); |
79 | if(ret < 0) { | |
80 | if(errno == EAGAIN) { | |
81 | ev = block(d->fd, EV_WRITE, d->timeout); | |
82 | if(ev < 0) { | |
83 | /* If we just go on, we should get the real error. */ | |
84 | continue; | |
85 | } else if(ev == 0) { | |
86 | errno = ETIMEDOUT; | |
87 | return(off); | |
88 | } else { | |
89 | continue; | |
90 | } | |
e1cdf02e | 91 | } else { |
9dc25d3d | 92 | return(off); |
e1cdf02e FT |
93 | } |
94 | } else { | |
9dc25d3d | 95 | off += ret; |
e1cdf02e FT |
96 | } |
97 | } | |
9dc25d3d | 98 | return(off); |
e1cdf02e FT |
99 | } |
100 | ||
101 | static int mtclose(void *cookie) | |
102 | { | |
103 | struct stdiofd *d = cookie; | |
104 | ||
105 | close(d->fd); | |
106 | free(d); | |
107 | return(0); | |
108 | } | |
109 | ||
110 | static cookie_io_functions_t iofuns = { | |
111 | .read = mtread, | |
112 | .write = mtwrite, | |
113 | .close = mtclose, | |
114 | }; | |
115 | ||
116 | FILE *mtstdopen(int fd, int issock, int timeout, char *mode) | |
117 | { | |
118 | struct stdiofd *d; | |
119 | FILE *ret; | |
120 | ||
121 | omalloc(d); | |
122 | d->fd = fd; | |
123 | d->sock = issock; | |
124 | d->timeout = timeout; | |
125 | ret = fopencookie(d, mode, iofuns); | |
126 | if(!ret) | |
127 | free(d); | |
128 | else | |
129 | fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK); | |
130 | return(ret); | |
131 | } |