1 /* |
|
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
3 * |
|
4 * Copyright 2011 Olaf Wintermann. All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions are met: |
|
8 * |
|
9 * 1. Redistributions of source code must retain the above copyright |
|
10 * notice, this list of conditions and the following disclaimer. |
|
11 * |
|
12 * 2. Redistributions in binary form must reproduce the above copyright |
|
13 * notice, this list of conditions and the following disclaimer in the |
|
14 * documentation and/or other materials provided with the distribution. |
|
15 * |
|
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
26 * POSSIBILITY OF SUCH DAMAGE. |
|
27 */ |
|
28 |
|
29 #include <unistd.h> |
|
30 #include <stdlib.h> |
|
31 #include <sys/uio.h> |
|
32 |
|
33 #include "io.h" |
|
34 #include "pool.h" |
|
35 |
|
36 IOStream native_io_funcs = { |
|
37 system_write, |
|
38 system_read |
|
39 }; |
|
40 |
|
41 IOStream net_io_funcs = { |
|
42 net_stream_write, |
|
43 net_stream_read |
|
44 }; |
|
45 |
|
46 |
|
47 IOStream* stream_new_from_fd(int fd) { |
|
48 SystemIOStream *st = malloc(sizeof(SystemIOStream)); |
|
49 st->st = native_io_funcs; |
|
50 st->fd = fd; |
|
51 return (IOStream*)st; |
|
52 } |
|
53 |
|
54 ssize_t system_write(IOStream *st, void *buf, size_t nbytes) { |
|
55 return write(((SystemIOStream*)st)->fd, buf, nbytes); |
|
56 } |
|
57 |
|
58 ssize_t system_read(IOStream *st, void *buf, size_t nbytes) { |
|
59 return read(((SystemIOStream*)st)->fd, buf, nbytes); |
|
60 } |
|
61 |
|
62 |
|
63 IOStream* net_stream_from_fd(int fd) { |
|
64 NetIOStream *st = malloc(sizeof(NetIOStream)); |
|
65 st->st = net_io_funcs; |
|
66 st->fd = fd; |
|
67 st->max_read = 0; |
|
68 st->rd = 0; |
|
69 } |
|
70 |
|
71 ssize_t net_stream_write(IOStream *st, void *buf, size_t nbytes) { |
|
72 // TODO: implement |
|
73 } |
|
74 |
|
75 ssize_t net_stream_read(IOStream *st, void *buf, size_t nbytes) { |
|
76 NetIOStream *n = (NetIOStream*)st; |
|
77 if(n->max_read != 0 && n->rd >= n->max_read) { |
|
78 return 0; |
|
79 } |
|
80 ssize_t r = read(n->fd, buf, nbytes); |
|
81 n->rd += r; |
|
82 return r; |
|
83 } |
|
84 |
|
85 |
|
86 ssize_t net_read(SYS_NETFD fd, void *buf, size_t nbytes) { |
|
87 ssize_t r = ((IOStream*)fd)->read(fd, buf, nbytes); |
|
88 if(r == 0) { |
|
89 return IO_EOF; |
|
90 } |
|
91 return r; |
|
92 } |
|
93 |
|
94 ssize_t net_write(SYS_NETFD fd, void *buf, size_t nbytes) { |
|
95 ssize_t r = ((IOStream*)fd)->write(fd, buf, nbytes); |
|
96 if(r < 0) { |
|
97 return IO_ERROR; |
|
98 } |
|
99 return r; |
|
100 } |
|
101 |
|
102 |
|
103 /* iovec buffer */ |
|
104 iovec_buf_t *iovec_buf_create(pool_handle_t *pool) { |
|
105 iovec_buf_t *buf = pool_malloc(pool, sizeof(iovec_buf_t)); |
|
106 |
|
107 buf->pool = pool; |
|
108 buf->iov = pool_calloc(pool, 32, sizeof(struct iovec)); |
|
109 buf->maxiovec = 32; |
|
110 buf->iovctn = 0; |
|
111 |
|
112 return buf; |
|
113 } |
|
114 |
|
115 void iovec_buf_write(iovec_buf_t *io, void *buf, size_t nbyte) { |
|
116 if(io->iovctn >= io->maxiovec) { |
|
117 io->iov = pool_realloc( |
|
118 io->pool, |
|
119 io->iov, |
|
120 (io->maxiovec + 16) * sizeof(struct iovec)); |
|
121 } |
|
122 |
|
123 io->iov[io->iovctn].iov_base = buf; |
|
124 io->iov[io->iovctn].iov_len = nbyte; |
|
125 io->iovctn++; |
|
126 } |
|
127 |
|
128 ssize_t iovec_buf_flush(iovec_buf_t *io, int fd) { |
|
129 return writev(fd, io->iov, io->iovctn); |
|
130 } |
|