Rev 4537 | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 4537 | Rev 4668 | ||
---|---|---|---|
1 | /* |
1 | /* |
2 | * Copyright (c) 2005 Martin Decky |
2 | * Copyright (c) 2005 Martin Decky |
3 | * All rights reserved. |
3 | * All rights reserved. |
4 | * |
4 | * |
5 | * Redistribution and use in source and binary forms, with or without |
5 | * Redistribution and use in source and binary forms, with or without |
6 | * modification, are permitted provided that the following conditions |
6 | * modification, are permitted provided that the following conditions |
7 | * are met: |
7 | * are met: |
8 | * |
8 | * |
9 | * - Redistributions of source code must retain the above copyright |
9 | * - Redistributions of source code must retain the above copyright |
10 | * notice, this list of conditions and the following disclaimer. |
10 | * notice, this list of conditions and the following disclaimer. |
11 | * - Redistributions in binary form must reproduce the above copyright |
11 | * - Redistributions in binary form must reproduce the above copyright |
12 | * notice, this list of conditions and the following disclaimer in the |
12 | * notice, this list of conditions and the following disclaimer in the |
13 | * documentation and/or other materials provided with the distribution. |
13 | * documentation and/or other materials provided with the distribution. |
14 | * - The name of the author may not be used to endorse or promote products |
14 | * - The name of the author may not be used to endorse or promote products |
15 | * derived from this software without specific prior written permission. |
15 | * derived from this software without specific prior written permission. |
16 | * |
16 | * |
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
27 | */ |
27 | */ |
28 | 28 | ||
29 | /** @addtogroup libc |
29 | /** @addtogroup libc |
30 | * @{ |
30 | * @{ |
31 | */ |
31 | */ |
32 | /** @file |
32 | /** @file |
33 | */ |
33 | */ |
34 | 34 | ||
35 | #include <stdio.h> |
35 | #include <stdio.h> |
36 | #include <unistd.h> |
36 | #include <unistd.h> |
37 | #include <fcntl.h> |
37 | #include <fcntl.h> |
- | 38 | #include <assert.h> |
|
38 | #include <string.h> |
39 | #include <string.h> |
39 | #include <errno.h> |
40 | #include <errno.h> |
40 | #include <bool.h> |
41 | #include <bool.h> |
41 | #include <malloc.h> |
42 | #include <malloc.h> |
42 | #include <io/klog.h> |
43 | #include <io/klog.h> |
43 | #include <vfs/vfs.h> |
44 | #include <vfs/vfs.h> |
44 | #include <ipc/devmap.h> |
45 | #include <ipc/devmap.h> |
45 | #include <adt/list.h> |
46 | #include <adt/list.h> |
46 | 47 | ||
- | 48 | static void _fflushbuf(FILE *stream); |
|
- | 49 | ||
47 | static FILE stdin_null = { |
50 | static FILE stdin_null = { |
48 | .fd = -1, |
51 | .fd = -1, |
49 | .error = true, |
52 | .error = true, |
50 | .eof = true, |
53 | .eof = true, |
51 | .klog = false, |
54 | .klog = false, |
52 | .phone = -1 |
55 | .phone = -1, |
- | 56 | .btype = _IONBF, |
|
- | 57 | .buf = NULL, |
|
- | 58 | .buf_size = 0, |
|
- | 59 | .buf_head = NULL |
|
53 | }; |
60 | }; |
54 | 61 | ||
55 | static FILE stdout_klog = { |
62 | static FILE stdout_klog = { |
56 | .fd = -1, |
63 | .fd = -1, |
57 | .error = false, |
64 | .error = false, |
58 | .eof = false, |
65 | .eof = false, |
59 | .klog = true, |
66 | .klog = true, |
60 | .phone = -1 |
67 | .phone = -1, |
- | 68 | .btype = _IOLBF, |
|
- | 69 | .buf = NULL, |
|
- | 70 | .buf_size = BUFSIZ, |
|
- | 71 | .buf_head = NULL |
|
61 | }; |
72 | }; |
62 | 73 | ||
63 | static FILE stderr_klog = { |
74 | static FILE stderr_klog = { |
64 | .fd = -1, |
75 | .fd = -1, |
65 | .error = false, |
76 | .error = false, |
66 | .eof = false, |
77 | .eof = false, |
67 | .klog = true, |
78 | .klog = true, |
68 | .phone = -1 |
79 | .phone = -1, |
- | 80 | .btype = _IONBF, |
|
- | 81 | .buf = NULL, |
|
- | 82 | .buf_size = 0, |
|
- | 83 | .buf_head = NULL |
|
69 | }; |
84 | }; |
70 | 85 | ||
71 | FILE *stdin = NULL; |
86 | FILE *stdin = NULL; |
72 | FILE *stdout = NULL; |
87 | FILE *stdout = NULL; |
73 | FILE *stderr = NULL; |
88 | FILE *stderr = NULL; |
74 | 89 | ||
75 | static LIST_INITIALIZE(files); |
90 | static LIST_INITIALIZE(files); |
76 | 91 | ||
77 | void stdio_init(int filc, fdi_node_t *filv[]) |
92 | void __stdio_init(int filc, fdi_node_t *filv[]) |
78 | { |
93 | { |
79 | if (filc > 0) { |
94 | if (filc > 0) { |
80 | stdin = fopen_node(filv[0], "r"); |
95 | stdin = fopen_node(filv[0], "r"); |
81 | } else { |
96 | } else { |
82 | stdin = &stdin_null; |
97 | stdin = &stdin_null; |
83 | list_append(&stdin->link, &files); |
98 | list_append(&stdin->link, &files); |
84 | } |
99 | } |
85 | 100 | ||
86 | if (filc > 1) { |
101 | if (filc > 1) { |
87 | stdout = fopen_node(filv[1], "w"); |
102 | stdout = fopen_node(filv[1], "w"); |
88 | } else { |
103 | } else { |
89 | stdout = &stdout_klog; |
104 | stdout = &stdout_klog; |
90 | list_append(&stdout->link, &files); |
105 | list_append(&stdout->link, &files); |
91 | } |
106 | } |
92 | 107 | ||
93 | if (filc > 2) { |
108 | if (filc > 2) { |
94 | stderr = fopen_node(filv[2], "w"); |
109 | stderr = fopen_node(filv[2], "w"); |
95 | } else { |
110 | } else { |
96 | stderr = &stderr_klog; |
111 | stderr = &stderr_klog; |
97 | list_append(&stderr->link, &files); |
112 | list_append(&stderr->link, &files); |
98 | } |
113 | } |
99 | } |
114 | } |
100 | 115 | ||
101 | void stdio_done(void) |
116 | void __stdio_done(void) |
102 | { |
117 | { |
103 | link_t *link = files.next; |
118 | link_t *link = files.next; |
104 | 119 | ||
105 | while (link != &files) { |
120 | while (link != &files) { |
106 | FILE *file = list_get_instance(link, FILE, link); |
121 | FILE *file = list_get_instance(link, FILE, link); |
107 | fclose(file); |
122 | fclose(file); |
108 | link = files.next; |
123 | link = files.next; |
109 | } |
124 | } |
110 | } |
125 | } |
111 | 126 | ||
112 | static bool parse_mode(const char *mode, int *flags) |
127 | static bool parse_mode(const char *mode, int *flags) |
113 | { |
128 | { |
114 | /* Parse mode except first character. */ |
129 | /* Parse mode except first character. */ |
115 | const char *mp = mode; |
130 | const char *mp = mode; |
116 | if (*mp++ == 0) { |
131 | if (*mp++ == 0) { |
117 | errno = EINVAL; |
132 | errno = EINVAL; |
118 | return false; |
133 | return false; |
119 | } |
134 | } |
120 | 135 | ||
121 | if ((*mp == 'b') || (*mp == 't')) |
136 | if ((*mp == 'b') || (*mp == 't')) |
122 | mp++; |
137 | mp++; |
123 | 138 | ||
124 | bool plus; |
139 | bool plus; |
125 | if (*mp == '+') { |
140 | if (*mp == '+') { |
126 | mp++; |
141 | mp++; |
127 | plus = true; |
142 | plus = true; |
128 | } else |
143 | } else |
129 | plus = false; |
144 | plus = false; |
130 | 145 | ||
131 | if (*mp != 0) { |
146 | if (*mp != 0) { |
132 | errno = EINVAL; |
147 | errno = EINVAL; |
133 | return false; |
148 | return false; |
134 | } |
149 | } |
135 | 150 | ||
136 | /* Parse first character of mode and determine flags for open(). */ |
151 | /* Parse first character of mode and determine flags for open(). */ |
137 | switch (mode[0]) { |
152 | switch (mode[0]) { |
138 | case 'r': |
153 | case 'r': |
139 | *flags = plus ? O_RDWR : O_RDONLY; |
154 | *flags = plus ? O_RDWR : O_RDONLY; |
140 | break; |
155 | break; |
141 | case 'w': |
156 | case 'w': |
142 | *flags = (O_TRUNC | O_CREAT) | (plus ? O_RDWR : O_WRONLY); |
157 | *flags = (O_TRUNC | O_CREAT) | (plus ? O_RDWR : O_WRONLY); |
143 | break; |
158 | break; |
144 | case 'a': |
159 | case 'a': |
145 | /* TODO: a+ must read from beginning, append to the end. */ |
160 | /* TODO: a+ must read from beginning, append to the end. */ |
146 | if (plus) { |
161 | if (plus) { |
147 | errno = ENOTSUP; |
162 | errno = ENOTSUP; |
148 | return false; |
163 | return false; |
149 | } |
164 | } |
150 | *flags = (O_APPEND | O_CREAT) | (plus ? O_RDWR : O_WRONLY); |
165 | *flags = (O_APPEND | O_CREAT) | (plus ? O_RDWR : O_WRONLY); |
151 | default: |
166 | default: |
152 | errno = EINVAL; |
167 | errno = EINVAL; |
153 | return false; |
168 | return false; |
154 | } |
169 | } |
155 | 170 | ||
156 | return true; |
171 | return true; |
157 | } |
172 | } |
158 | 173 | ||
- | 174 | /** Set stream buffer. */ |
|
- | 175 | void setvbuf(FILE *stream, void *buf, int mode, size_t size) |
|
- | 176 | { |
|
- | 177 | stream->btype = mode; |
|
- | 178 | stream->buf = buf; |
|
- | 179 | stream->buf_size = size; |
|
- | 180 | stream->buf_head = stream->buf; |
|
- | 181 | } |
|
- | 182 | ||
- | 183 | static void _setvbuf(FILE *stream) |
|
- | 184 | { |
|
- | 185 | /* FIXME: Use more complex rules for setting buffering options. */ |
|
- | 186 | ||
- | 187 | switch (stream->fd) { |
|
- | 188 | case 1: |
|
- | 189 | setvbuf(stream, NULL, _IOLBF, BUFSIZ); |
|
- | 190 | break; |
|
- | 191 | case 0: |
|
- | 192 | case 2: |
|
- | 193 | setvbuf(stream, NULL, _IONBF, 0); |
|
- | 194 | break; |
|
- | 195 | default: |
|
- | 196 | setvbuf(stream, NULL, _IOFBF, BUFSIZ); |
|
- | 197 | } |
|
- | 198 | } |
|
- | 199 | ||
- | 200 | /** Allocate stream buffer. */ |
|
- | 201 | static int _fallocbuf(FILE *stream) |
|
- | 202 | { |
|
- | 203 | assert(stream->buf == NULL); |
|
- | 204 | ||
- | 205 | stream->buf = malloc(stream->buf_size); |
|
- | 206 | if (stream->buf == NULL) { |
|
- | 207 | errno = ENOMEM; |
|
- | 208 | return -1; |
|
- | 209 | } |
|
- | 210 | ||
- | 211 | stream->buf_head = stream->buf; |
|
- | 212 | return 0; |
|
- | 213 | } |
|
- | 214 | ||
159 | /** Open a stream. |
215 | /** Open a stream. |
160 | * |
216 | * |
161 | * @param path Path of the file to open. |
217 | * @param path Path of the file to open. |
162 | * @param mode Mode string, (r|w|a)[b|t][+]. |
218 | * @param mode Mode string, (r|w|a)[b|t][+]. |
163 | * |
219 | * |
164 | */ |
220 | */ |
165 | FILE *fopen(const char *path, const char *mode) |
221 | FILE *fopen(const char *path, const char *mode) |
166 | { |
222 | { |
167 | int flags; |
223 | int flags; |
168 | if (!parse_mode(mode, &flags)) |
224 | if (!parse_mode(mode, &flags)) |
169 | return NULL; |
225 | return NULL; |
170 | 226 | ||
171 | /* Open file. */ |
227 | /* Open file. */ |
172 | FILE *stream = malloc(sizeof(FILE)); |
228 | FILE *stream = malloc(sizeof(FILE)); |
173 | if (stream == NULL) { |
229 | if (stream == NULL) { |
174 | errno = ENOMEM; |
230 | errno = ENOMEM; |
175 | return NULL; |
231 | return NULL; |
176 | } |
232 | } |
177 | 233 | ||
178 | stream->fd = open(path, flags, 0666); |
234 | stream->fd = open(path, flags, 0666); |
179 | if (stream->fd < 0) { |
235 | if (stream->fd < 0) { |
180 | /* errno was set by open() */ |
236 | /* errno was set by open() */ |
181 | free(stream); |
237 | free(stream); |
182 | return NULL; |
238 | return NULL; |
183 | } |
239 | } |
184 | 240 | ||
185 | stream->error = false; |
241 | stream->error = false; |
186 | stream->eof = false; |
242 | stream->eof = false; |
187 | stream->klog = false; |
243 | stream->klog = false; |
188 | stream->phone = -1; |
244 | stream->phone = -1; |
- | 245 | _setvbuf(stream); |
|
189 | 246 | ||
190 | list_append(&stream->link, &files); |
247 | list_append(&stream->link, &files); |
191 | 248 | ||
192 | return stream; |
249 | return stream; |
193 | } |
250 | } |
194 | 251 | ||
195 | FILE *fdopen(int fd, const char *mode) |
252 | FILE *fdopen(int fd, const char *mode) |
196 | { |
253 | { |
197 | /* Open file. */ |
254 | /* Open file. */ |
198 | FILE *stream = malloc(sizeof(FILE)); |
255 | FILE *stream = malloc(sizeof(FILE)); |
199 | if (stream == NULL) { |
256 | if (stream == NULL) { |
200 | errno = ENOMEM; |
257 | errno = ENOMEM; |
201 | return NULL; |
258 | return NULL; |
202 | } |
259 | } |
203 | 260 | ||
204 | stream->fd = fd; |
261 | stream->fd = fd; |
205 | stream->error = false; |
262 | stream->error = false; |
206 | stream->eof = false; |
263 | stream->eof = false; |
207 | stream->klog = false; |
264 | stream->klog = false; |
208 | stream->phone = -1; |
265 | stream->phone = -1; |
- | 266 | _setvbuf(stream); |
|
209 | 267 | ||
210 | list_append(&stream->link, &files); |
268 | list_append(&stream->link, &files); |
211 | 269 | ||
212 | return stream; |
270 | return stream; |
213 | } |
271 | } |
214 | 272 | ||
215 | FILE *fopen_node(fdi_node_t *node, const char *mode) |
273 | FILE *fopen_node(fdi_node_t *node, const char *mode) |
216 | { |
274 | { |
217 | int flags; |
275 | int flags; |
218 | if (!parse_mode(mode, &flags)) |
276 | if (!parse_mode(mode, &flags)) |
219 | return NULL; |
277 | return NULL; |
220 | 278 | ||
221 | /* Open file. */ |
279 | /* Open file. */ |
222 | FILE *stream = malloc(sizeof(FILE)); |
280 | FILE *stream = malloc(sizeof(FILE)); |
223 | if (stream == NULL) { |
281 | if (stream == NULL) { |
224 | errno = ENOMEM; |
282 | errno = ENOMEM; |
225 | return NULL; |
283 | return NULL; |
226 | } |
284 | } |
227 | 285 | ||
228 | stream->fd = open_node(node, flags); |
286 | stream->fd = open_node(node, flags); |
229 | if (stream->fd < 0) { |
287 | if (stream->fd < 0) { |
230 | /* errno was set by open_node() */ |
288 | /* errno was set by open_node() */ |
231 | free(stream); |
289 | free(stream); |
232 | return NULL; |
290 | return NULL; |
233 | } |
291 | } |
234 | 292 | ||
235 | stream->error = false; |
293 | stream->error = false; |
236 | stream->eof = false; |
294 | stream->eof = false; |
237 | stream->klog = false; |
295 | stream->klog = false; |
238 | stream->phone = -1; |
296 | stream->phone = -1; |
- | 297 | _setvbuf(stream); |
|
239 | 298 | ||
240 | list_append(&stream->link, &files); |
299 | list_append(&stream->link, &files); |
241 | 300 | ||
242 | return stream; |
301 | return stream; |
243 | } |
302 | } |
244 | 303 | ||
245 | int fclose(FILE *stream) |
304 | int fclose(FILE *stream) |
246 | { |
305 | { |
247 | int rc = 0; |
306 | int rc = 0; |
248 | 307 | ||
249 | fflush(stream); |
308 | fflush(stream); |
250 | 309 | ||
251 | if (stream->phone >= 0) |
310 | if (stream->phone >= 0) |
252 | ipc_hangup(stream->phone); |
311 | ipc_hangup(stream->phone); |
253 | 312 | ||
254 | if (stream->fd >= 0) |
313 | if (stream->fd >= 0) |
255 | rc = close(stream->fd); |
314 | rc = close(stream->fd); |
256 | 315 | ||
257 | list_remove(&stream->link); |
316 | list_remove(&stream->link); |
258 | 317 | ||
259 | if ((stream != &stdin_null) |
318 | if ((stream != &stdin_null) |
260 | && (stream != &stdout_klog) |
319 | && (stream != &stdout_klog) |
261 | && (stream != &stderr_klog)) |
320 | && (stream != &stderr_klog)) |
262 | free(stream); |
321 | free(stream); |
263 | 322 | ||
264 | stream = NULL; |
323 | stream = NULL; |
265 | 324 | ||
266 | if (rc != 0) { |
325 | if (rc != 0) { |
267 | /* errno was set by close() */ |
326 | /* errno was set by close() */ |
268 | return EOF; |
327 | return EOF; |
269 | } |
328 | } |
270 | 329 | ||
271 | return 0; |
330 | return 0; |
272 | } |
331 | } |
273 | 332 | ||
274 | /** Read from a stream. |
333 | /** Read from a stream. |
275 | * |
334 | * |
276 | * @param buf Destination buffer. |
335 | * @param buf Destination buffer. |
277 | * @param size Size of each record. |
336 | * @param size Size of each record. |
278 | * @param nmemb Number of records to read. |
337 | * @param nmemb Number of records to read. |
279 | * @param stream Pointer to the stream. |
338 | * @param stream Pointer to the stream. |
280 | * |
339 | * |
281 | */ |
340 | */ |
282 | size_t fread(void *buf, size_t size, size_t nmemb, FILE *stream) |
341 | size_t fread(void *buf, size_t size, size_t nmemb, FILE *stream) |
283 | { |
342 | { |
284 | size_t left = size * nmemb; |
343 | size_t left = size * nmemb; |
285 | size_t done = 0; |
344 | size_t done = 0; |
286 | 345 | ||
- | 346 | /* Make sure no data is pending write. */ |
|
- | 347 | _fflushbuf(stream); |
|
- | 348 | ||
287 | while ((left > 0) && (!stream->error) && (!stream->eof)) { |
349 | while ((left > 0) && (!stream->error) && (!stream->eof)) { |
288 | ssize_t rd = read(stream->fd, buf + done, left); |
350 | ssize_t rd = read(stream->fd, buf + done, left); |
289 | 351 | ||
290 | if (rd < 0) |
352 | if (rd < 0) |
291 | stream->error = true; |
353 | stream->error = true; |
292 | else if (rd == 0) |
354 | else if (rd == 0) |
293 | stream->eof = true; |
355 | stream->eof = true; |
294 | else { |
356 | else { |
295 | left -= rd; |
357 | left -= rd; |
296 | done += rd; |
358 | done += rd; |
297 | } |
359 | } |
298 | } |
360 | } |
299 | 361 | ||
300 | return (done / size); |
362 | return (done / size); |
301 | } |
363 | } |
302 | 364 | ||
303 | /** Write to a stream. |
- | |
304 | * |
- | |
305 | * @param buf Source buffer. |
- | |
306 | * @param size Size of each record. |
- | |
307 | * @param nmemb Number of records to write. |
- | |
308 | * @param stream Pointer to the stream. |
- | |
309 | * |
- | |
310 | */ |
- | |
311 | size_t fwrite(const void *buf, size_t size, size_t nmemb, FILE *stream) |
365 | static size_t _fwrite(const void *buf, size_t size, size_t nmemb, FILE *stream) |
312 | { |
366 | { |
313 | size_t left = size * nmemb; |
367 | size_t left = size * nmemb; |
314 | size_t done = 0; |
368 | size_t done = 0; |
315 | 369 | ||
316 | while ((left > 0) && (!stream->error)) { |
370 | while ((left > 0) && (!stream->error)) { |
317 | ssize_t wr; |
371 | ssize_t wr; |
318 | 372 | ||
319 | if (stream->klog) |
373 | if (stream->klog) |
320 | wr = klog_write(buf + done, left); |
374 | wr = klog_write(buf + done, left); |
321 | else |
375 | else |
322 | wr = write(stream->fd, buf + done, left); |
376 | wr = write(stream->fd, buf + done, left); |
323 | 377 | ||
324 | if (wr <= 0) |
378 | if (wr <= 0) |
325 | stream->error = true; |
379 | stream->error = true; |
326 | else { |
380 | else { |
327 | left -= wr; |
381 | left -= wr; |
328 | done += wr; |
382 | done += wr; |
329 | } |
383 | } |
330 | } |
384 | } |
331 | 385 | ||
332 | return (done / size); |
386 | return (done / size); |
333 | } |
387 | } |
334 | 388 | ||
- | 389 | /** Drain stream buffer, do not sync stream. */ |
|
- | 390 | static void _fflushbuf(FILE *stream) |
|
- | 391 | { |
|
- | 392 | size_t bytes_used; |
|
- | 393 | ||
- | 394 | if ((!stream->buf) || (stream->btype == _IONBF) || (stream->error)) |
|
- | 395 | return; |
|
- | 396 | ||
- | 397 | bytes_used = stream->buf_head - stream->buf; |
|
- | 398 | if (bytes_used == 0) |
|
- | 399 | return; |
|
- | 400 | ||
- | 401 | (void) _fwrite(stream->buf, 1, bytes_used, stream); |
|
- | 402 | stream->buf_head = stream->buf; |
|
- | 403 | } |
|
- | 404 | ||
- | 405 | /** Write to a stream. |
|
- | 406 | * |
|
- | 407 | * @param buf Source buffer. |
|
- | 408 | * @param size Size of each record. |
|
- | 409 | * @param nmemb Number of records to write. |
|
- | 410 | * @param stream Pointer to the stream. |
|
- | 411 | * |
|
- | 412 | */ |
|
- | 413 | size_t fwrite(const void *buf, size_t size, size_t nmemb, FILE *stream) |
|
- | 414 | { |
|
- | 415 | uint8_t *data; |
|
- | 416 | size_t bytes_left; |
|
- | 417 | size_t now; |
|
- | 418 | size_t buf_free; |
|
- | 419 | size_t total_written; |
|
- | 420 | size_t i; |
|
- | 421 | uint8_t b; |
|
- | 422 | bool need_flush; |
|
- | 423 | ||
- | 424 | /* If not buffered stream, write out directly. */ |
|
- | 425 | if (stream->btype == _IONBF) { |
|
- | 426 | now = _fwrite(buf, size, nmemb, stream); |
|
- | 427 | fflush(stream); |
|
- | 428 | return now; |
|
- | 429 | } |
|
- | 430 | ||
- | 431 | /* Perform lazy allocation of stream buffer. */ |
|
- | 432 | if (stream->buf == NULL) { |
|
- | 433 | if (_fallocbuf(stream) != 0) |
|
- | 434 | return 0; /* Errno set by _fallocbuf(). */ |
|
- | 435 | } |
|
- | 436 | ||
- | 437 | data = (uint8_t *) buf; |
|
- | 438 | bytes_left = size * nmemb; |
|
- | 439 | total_written = 0; |
|
- | 440 | need_flush = false; |
|
- | 441 | ||
- | 442 | while ((!stream->error) && (bytes_left > 0)) { |
|
- | 443 | buf_free = stream->buf_size - (stream->buf_head - stream->buf); |
|
- | 444 | if (bytes_left > buf_free) |
|
- | 445 | now = buf_free; |
|
- | 446 | else |
|
- | 447 | now = bytes_left; |
|
- | 448 | ||
- | 449 | for (i = 0; i < now; i++) { |
|
- | 450 | b = data[i]; |
|
- | 451 | stream->buf_head[i] = b; |
|
- | 452 | ||
- | 453 | if ((b == '\n') && (stream->btype == _IOLBF)) |
|
- | 454 | need_flush = true; |
|
- | 455 | } |
|
- | 456 | ||
- | 457 | buf += now; |
|
- | 458 | stream->buf_head += now; |
|
- | 459 | buf_free -= now; |
|
- | 460 | bytes_left -= now; |
|
- | 461 | total_written += now; |
|
- | 462 | ||
- | 463 | if (buf_free == 0) { |
|
- | 464 | /* Only need to drain buffer. */ |
|
- | 465 | _fflushbuf(stream); |
|
- | 466 | need_flush = false; |
|
- | 467 | } |
|
- | 468 | } |
|
- | 469 | ||
- | 470 | if (need_flush) |
|
- | 471 | fflush(stream); |
|
- | 472 | ||
- | 473 | return (total_written / size); |
|
- | 474 | } |
|
- | 475 | ||
335 | int fputc(wchar_t c, FILE *stream) |
476 | int fputc(wchar_t c, FILE *stream) |
336 | { |
477 | { |
337 | char buf[STR_BOUNDS(1)]; |
478 | char buf[STR_BOUNDS(1)]; |
338 | size_t sz = 0; |
479 | size_t sz = 0; |
339 | 480 | ||
340 | if (chr_encode(c, buf, &sz, STR_BOUNDS(1)) == EOK) { |
481 | if (chr_encode(c, buf, &sz, STR_BOUNDS(1)) == EOK) { |
341 | size_t wr = fwrite(buf, sz, 1, stream); |
482 | size_t wr = fwrite(buf, sz, 1, stream); |
342 | 483 | ||
343 | if (wr < sz) |
484 | if (wr < sz) |
344 | return EOF; |
485 | return EOF; |
345 | 486 | ||
346 | return (int) c; |
487 | return (int) c; |
347 | } |
488 | } |
348 | 489 | ||
349 | return EOF; |
490 | return EOF; |
350 | } |
491 | } |
351 | 492 | ||
352 | int putchar(wchar_t c) |
493 | int putchar(wchar_t c) |
353 | { |
494 | { |
354 | return fputc(c, stdout); |
495 | return fputc(c, stdout); |
355 | } |
496 | } |
356 | 497 | ||
357 | int fputs(const char *str, FILE *stream) |
498 | int fputs(const char *str, FILE *stream) |
358 | { |
499 | { |
359 | return fwrite(str, str_size(str), 1, stream); |
500 | return fwrite(str, str_size(str), 1, stream); |
360 | } |
501 | } |
361 | 502 | ||
362 | int puts(const char *str) |
503 | int puts(const char *str) |
363 | { |
504 | { |
364 | return fputs(str, stdout); |
505 | return fputs(str, stdout); |
365 | } |
506 | } |
366 | 507 | ||
367 | int fgetc(FILE *stream) |
508 | int fgetc(FILE *stream) |
368 | { |
509 | { |
369 | char c; |
510 | char c; |
370 | 511 | ||
371 | /* This could be made faster by only flushing when needed. */ |
512 | /* This could be made faster by only flushing when needed. */ |
372 | if (stdout) |
513 | if (stdout) |
373 | fflush(stdout); |
514 | fflush(stdout); |
374 | if (stderr) |
515 | if (stderr) |
375 | fflush(stderr); |
516 | fflush(stderr); |
376 | 517 | ||
377 | if (fread(&c, sizeof(char), 1, stream) < sizeof(char)) |
518 | if (fread(&c, sizeof(char), 1, stream) < sizeof(char)) |
378 | return EOF; |
519 | return EOF; |
379 | 520 | ||
380 | return (int) c; |
521 | return (int) c; |
381 | } |
522 | } |
382 | 523 | ||
383 | int getchar(void) |
524 | int getchar(void) |
384 | { |
525 | { |
385 | return fgetc(stdin); |
526 | return fgetc(stdin); |
386 | } |
527 | } |
387 | 528 | ||
388 | int fseek(FILE *stream, long offset, int origin) |
529 | int fseek(FILE *stream, long offset, int origin) |
389 | { |
530 | { |
390 | off_t rc = lseek(stream->fd, offset, origin); |
531 | off_t rc = lseek(stream->fd, offset, origin); |
391 | if (rc == (off_t) (-1)) { |
532 | if (rc == (off_t) (-1)) { |
392 | /* errno has been set by lseek. */ |
533 | /* errno has been set by lseek. */ |
393 | return -1; |
534 | return -1; |
394 | } |
535 | } |
395 | 536 | ||
396 | stream->eof = false; |
537 | stream->eof = false; |
397 | 538 | ||
398 | return 0; |
539 | return 0; |
399 | } |
540 | } |
400 | 541 | ||
401 | void rewind(FILE *stream) |
542 | void rewind(FILE *stream) |
402 | { |
543 | { |
403 | (void) fseek(stream, 0, SEEK_SET); |
544 | (void) fseek(stream, 0, SEEK_SET); |
404 | } |
545 | } |
405 | 546 | ||
406 | int fflush(FILE *stream) |
547 | int fflush(FILE *stream) |
407 | { |
548 | { |
- | 549 | _fflushbuf(stream); |
|
- | 550 | ||
408 | if (stream->klog) { |
551 | if (stream->klog) { |
409 | klog_update(); |
552 | klog_update(); |
410 | return EOK; |
553 | return EOK; |
411 | } |
554 | } |
412 | 555 | ||
413 | if (stream->fd >= 0) |
556 | if (stream->fd >= 0) |
414 | return fsync(stream->fd); |
557 | return fsync(stream->fd); |
415 | 558 | ||
416 | return ENOENT; |
559 | return ENOENT; |
417 | } |
560 | } |
418 | 561 | ||
419 | int feof(FILE *stream) |
562 | int feof(FILE *stream) |
420 | { |
563 | { |
421 | return stream->eof; |
564 | return stream->eof; |
422 | } |
565 | } |
423 | 566 | ||
424 | int ferror(FILE *stream) |
567 | int ferror(FILE *stream) |
425 | { |
568 | { |
426 | return stream->error; |
569 | return stream->error; |
427 | } |
570 | } |
428 | 571 | ||
429 | int fphone(FILE *stream) |
572 | int fphone(FILE *stream) |
430 | { |
573 | { |
431 | if (stream->fd >= 0) { |
574 | if (stream->fd >= 0) { |
432 | if (stream->phone < 0) |
575 | if (stream->phone < 0) |
433 | stream->phone = fd_phone(stream->fd); |
576 | stream->phone = fd_phone(stream->fd); |
434 | 577 | ||
435 | return stream->phone; |
578 | return stream->phone; |
436 | } |
579 | } |
437 | 580 | ||
438 | return -1; |
581 | return -1; |
439 | } |
582 | } |
440 | 583 | ||
441 | int fnode(FILE *stream, fdi_node_t *node) |
584 | int fnode(FILE *stream, fdi_node_t *node) |
442 | { |
585 | { |
443 | if (stream->fd >= 0) |
586 | if (stream->fd >= 0) |
444 | return fd_node(stream->fd, node); |
587 | return fd_node(stream->fd, node); |
445 | 588 | ||
446 | return ENOENT; |
589 | return ENOENT; |
447 | } |
590 | } |
448 | 591 | ||
449 | /** @} |
592 | /** @} |
450 | */ |
593 | */ |
451 | 594 |