|
1 /* |
|
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
3 * |
|
4 * Copyright 2018 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 <stdio.h> |
|
30 #include <stdlib.h> |
|
31 #include <string.h> |
|
32 |
|
33 #include "writer.h" |
|
34 |
|
35 void writer_init(Writer *w, SYS_NETFD fd, char *buf, size_t len) { |
|
36 w->fd = fd; |
|
37 w->buffer = buf; |
|
38 w->size = len; |
|
39 w->pos = 0; |
|
40 w->error = 0; |
|
41 } |
|
42 |
|
43 int writer_flush(Writer *w) { |
|
44 if(w->error) { |
|
45 return w->error; |
|
46 } |
|
47 |
|
48 size_t pos = 0; |
|
49 size_t len = w->pos; |
|
50 |
|
51 while(len > 0) { |
|
52 ssize_t r = net_write(w->fd, w->buffer + pos, len); |
|
53 if(r <= 0) { |
|
54 break; |
|
55 } |
|
56 len -= r; |
|
57 pos += r; |
|
58 } |
|
59 |
|
60 if(pos != w->pos) { |
|
61 w->error = 1; |
|
62 return 1; |
|
63 } |
|
64 |
|
65 w->pos = 0; |
|
66 return 0; |
|
67 } |
|
68 |
|
69 int writer_put(Writer *w, const char *s, size_t len) { |
|
70 if(w->error) { |
|
71 return w->error; |
|
72 } |
|
73 |
|
74 size_t a = w->size - w->pos; |
|
75 if(a == 0) { |
|
76 if(writer_flush(w)) { |
|
77 return 1; |
|
78 } |
|
79 } |
|
80 |
|
81 size_t cplen = len > a ? a : len; |
|
82 memcpy(w->buffer, s, cplen); |
|
83 w->pos += cplen; |
|
84 |
|
85 if(cplen < len) { |
|
86 return writer_put(w, s + cplen, len - cplen); |
|
87 } else { |
|
88 return 0; |
|
89 } |
|
90 } |
|
91 |
|
92 int writer_puts(Writer *w, sstr_t s) { |
|
93 return writer_put(w, s.ptr, s.length); |
|
94 } |
|
95 |
|
96 int writer_putc(Writer *w, char c) { |
|
97 if(w->pos == w->size) { |
|
98 if(writer_flush(w)) { |
|
99 return 1; |
|
100 } |
|
101 } |
|
102 w->buffer[w->pos++] = c; |
|
103 return 0; |
|
104 } |