1 9df487d7 2003-11-23 devnull #include <u.h>
  2 9df487d7 2003-11-23 devnull #include <libc.h>
  3 9df487d7 2003-11-23 devnull #include <httpd.h>
  4 9df487d7 2003-11-23 devnull 
  5 9df487d7 2003-11-23 devnull static	char	hstates[] = "nrewE";
  6 9df487d7 2003-11-23 devnull static	char	hxfers[] = " x";
  7 9df487d7 2003-11-23 devnull 
  8 9df487d7 2003-11-23 devnull int
  9 9df487d7 2003-11-23 devnull hinit(Hio *h, int fd, int mode)
 10 9df487d7 2003-11-23 devnull {
 11 9df487d7 2003-11-23 devnull 	if(fd == -1 || mode != Hread && mode != Hwrite)
 12 9df487d7 2003-11-23 devnull 		return -1;
 13 9df487d7 2003-11-23 devnull 	h->hh = nil;
 14 9df487d7 2003-11-23 devnull 	h->fd = fd;
 15 9df487d7 2003-11-23 devnull 	h->seek = 0;
 16 9df487d7 2003-11-23 devnull 	h->state = mode;
 17 9df487d7 2003-11-23 devnull 	h->start = h->buf + 16;		/* leave space for chunk length */
 18 9df487d7 2003-11-23 devnull 	h->stop = h->pos = h->start;
 19 9df487d7 2003-11-23 devnull 	if(mode == Hread){
 20 9df487d7 2003-11-23 devnull 		h->bodylen = ~0UL;
 21 9df487d7 2003-11-23 devnull 		*h->pos = '\0';
 22 9df487d7 2003-11-23 devnull 	}else
 23 9df487d7 2003-11-23 devnull 		h->stop = h->start + Hsize;
 24 9df487d7 2003-11-23 devnull 	return 0;
 25 9df487d7 2003-11-23 devnull }
 26 9df487d7 2003-11-23 devnull 
 27 9df487d7 2003-11-23 devnull int
 28 9df487d7 2003-11-23 devnull hiserror(Hio *h)
 29 9df487d7 2003-11-23 devnull {
 30 9df487d7 2003-11-23 devnull 	return h->state == Herr;
 31 9df487d7 2003-11-23 devnull }
 32 9df487d7 2003-11-23 devnull 
 33 9df487d7 2003-11-23 devnull int
 34 9df487d7 2003-11-23 devnull hgetc(Hio *h)
 35 9df487d7 2003-11-23 devnull {
 36 9df487d7 2003-11-23 devnull 	uchar *p;
 37 9df487d7 2003-11-23 devnull 
 38 9df487d7 2003-11-23 devnull 	p = h->pos;
 39 9df487d7 2003-11-23 devnull 	if(p < h->stop){
 40 9df487d7 2003-11-23 devnull 		h->pos = p + 1;
 41 9df487d7 2003-11-23 devnull 		return *p;
 42 9df487d7 2003-11-23 devnull 	}
 43 9df487d7 2003-11-23 devnull 	p -= UTFmax;
 44 9df487d7 2003-11-23 devnull 	if(p < h->start)
 45 9df487d7 2003-11-23 devnull 		p = h->start;
 46 9df487d7 2003-11-23 devnull 	if(!hreadbuf(h, p) || h->pos == h->stop)
 47 9df487d7 2003-11-23 devnull 		return -1;
 48 9df487d7 2003-11-23 devnull 	return *h->pos++;
 49 9df487d7 2003-11-23 devnull }
 50 9df487d7 2003-11-23 devnull 
 51 9df487d7 2003-11-23 devnull int
 52 9df487d7 2003-11-23 devnull hungetc(Hio *h)
 53 9df487d7 2003-11-23 devnull {
 54 9df487d7 2003-11-23 devnull 	if(h->state == Hend)
 55 9df487d7 2003-11-23 devnull 		h->state = Hread;
 56 9df487d7 2003-11-23 devnull 	else if(h->state == Hread)
 57 9df487d7 2003-11-23 devnull 		h->pos--;
 58 9df487d7 2003-11-23 devnull 	if(h->pos < h->start || h->state != Hread){
 59 9df487d7 2003-11-23 devnull 		h->state = Herr;
 60 9df487d7 2003-11-23 devnull 		h->pos = h->stop;
 61 9df487d7 2003-11-23 devnull 		return -1;
 62 9df487d7 2003-11-23 devnull 	}
 63 9df487d7 2003-11-23 devnull 	return 0;
 64 9df487d7 2003-11-23 devnull }
 65 9df487d7 2003-11-23 devnull 
 66 9df487d7 2003-11-23 devnull /*
 67 9df487d7 2003-11-23 devnull  * fill the buffer, saving contents from vsave onwards.
 68 9df487d7 2003-11-23 devnull  * nothing is saved if vsave is nil.
 69 9df487d7 2003-11-23 devnull  * returns the beginning of the buffer.
 70 9df487d7 2003-11-23 devnull  *
 71 9df487d7 2003-11-23 devnull  * understands message body sizes and chunked transfer encoding
 72 9df487d7 2003-11-23 devnull  */
 73 9df487d7 2003-11-23 devnull void *
 74 9df487d7 2003-11-23 devnull hreadbuf(Hio *h, void *vsave)
 75 9df487d7 2003-11-23 devnull {
 76 9df487d7 2003-11-23 devnull 	Hio *hh;
 77 9df487d7 2003-11-23 devnull 	uchar *save;
 78 9df487d7 2003-11-23 devnull 	int c, in, cpy, dpos;
 79 9df487d7 2003-11-23 devnull 
 80 9df487d7 2003-11-23 devnull 	save = vsave;
 81 9df487d7 2003-11-23 devnull 	if(save && (save < h->start || save > h->stop)
 82 9df487d7 2003-11-23 devnull 	|| h->state != Hread && h->state != Hend){
 83 9df487d7 2003-11-23 devnull 		h->state = Herr;
 84 9df487d7 2003-11-23 devnull 		h->pos = h->stop;
 85 9df487d7 2003-11-23 devnull 		return nil;
 86 9df487d7 2003-11-23 devnull 	}
 87 9df487d7 2003-11-23 devnull 
 88 9df487d7 2003-11-23 devnull 	dpos = 0;
 89 9df487d7 2003-11-23 devnull 	if(save && h->pos > save)
 90 9df487d7 2003-11-23 devnull 		dpos = h->pos - save;
 91 9df487d7 2003-11-23 devnull 	cpy = 0;
 92 9df487d7 2003-11-23 devnull 	if(save){
 93 9df487d7 2003-11-23 devnull 		cpy = h->stop - save;
 94 9df487d7 2003-11-23 devnull 		memmove(h->start, save, cpy);
 95 9df487d7 2003-11-23 devnull 	}
 96 9df487d7 2003-11-23 devnull 	h->seek += h->stop - h->start - cpy;
 97 9df487d7 2003-11-23 devnull 	h->pos = h->start + dpos;
 98 9df487d7 2003-11-23 devnull 
 99 9df487d7 2003-11-23 devnull 	in = Hsize - cpy;
100 9df487d7 2003-11-23 devnull 	if(h->state == Hend)
101 9df487d7 2003-11-23 devnull 		in = 0;
102 9df487d7 2003-11-23 devnull 	else if(in > h->bodylen)
103 9df487d7 2003-11-23 devnull 		in = h->bodylen;
104 9df487d7 2003-11-23 devnull 
105 9df487d7 2003-11-23 devnull 	/*
106 9df487d7 2003-11-23 devnull 	 * for chunked encoding, fill buffer,
107 9df487d7 2003-11-23 devnull 	 * then read in new chunk length and wipe out that line
108 9df487d7 2003-11-23 devnull 	 */
109 9df487d7 2003-11-23 devnull 	hh = h->hh;
110 9df487d7 2003-11-23 devnull 	if(hh != nil){
111 9df487d7 2003-11-23 devnull 		if(!in && h->xferenc && h->state != Hend){
112 9df487d7 2003-11-23 devnull 			if(h->xferenc == 2){
113 9df487d7 2003-11-23 devnull 				c = hgetc(hh);
114 9df487d7 2003-11-23 devnull 				if(c == '\r')
115 9df487d7 2003-11-23 devnull 					c = hgetc(hh);
116 9df487d7 2003-11-23 devnull 				if(c != '\n'){
117 9df487d7 2003-11-23 devnull 					h->pos = h->stop;
118 9df487d7 2003-11-23 devnull 					h->state = Herr;
119 9df487d7 2003-11-23 devnull 					return nil;
120 9df487d7 2003-11-23 devnull 				}
121 9df487d7 2003-11-23 devnull 			}
122 9df487d7 2003-11-23 devnull 			h->xferenc = 2;
123 9df487d7 2003-11-23 devnull 			in = 0;
124 9df487d7 2003-11-23 devnull 			while((c = hgetc(hh)) != '\n'){
125 9df487d7 2003-11-23 devnull 				if(c >= '0' && c <= '9')
126 9df487d7 2003-11-23 devnull 					c -= '0';
127 9df487d7 2003-11-23 devnull 				else if(c >= 'a' && c <= 'f')
128 9df487d7 2003-11-23 devnull 					c -= 'a' - 10;
129 9df487d7 2003-11-23 devnull 				else if(c >= 'A' && c <= 'F')
130 9df487d7 2003-11-23 devnull 					c -= 'A' - 10;
131 9df487d7 2003-11-23 devnull 				else
132 9df487d7 2003-11-23 devnull 					break;
133 9df487d7 2003-11-23 devnull 				in = in * 16 + c;
134 9df487d7 2003-11-23 devnull 			}
135 9df487d7 2003-11-23 devnull 			while(c != '\n'){
136 9df487d7 2003-11-23 devnull 				if(c < 0){
137 9df487d7 2003-11-23 devnull 					h->pos = h->stop;
138 9df487d7 2003-11-23 devnull 					h->state = Herr;
139 9df487d7 2003-11-23 devnull 					return nil;
140 9df487d7 2003-11-23 devnull 				}
141 9df487d7 2003-11-23 devnull 				c = hgetc(hh);
142 9df487d7 2003-11-23 devnull 			}
143 9df487d7 2003-11-23 devnull 			h->bodylen = in;
144 9df487d7 2003-11-23 devnull 
145 9df487d7 2003-11-23 devnull 			in = Hsize - cpy;
146 9df487d7 2003-11-23 devnull 			if(in > h->bodylen)
147 9df487d7 2003-11-23 devnull 				in = h->bodylen;
148 9df487d7 2003-11-23 devnull 		}
149 9df487d7 2003-11-23 devnull 		if(in){
150 9df487d7 2003-11-23 devnull 			while(hh->pos + in > hh->stop){
151 9df487d7 2003-11-23 devnull 				if(hreadbuf(hh, hh->pos) == nil){
152 9df487d7 2003-11-23 devnull 					h->pos = h->stop;
153 9df487d7 2003-11-23 devnull 					h->state = Herr;
154 9df487d7 2003-11-23 devnull 					return nil;
155 9df487d7 2003-11-23 devnull 				}
156 9df487d7 2003-11-23 devnull 			}
157 9df487d7 2003-11-23 devnull 			memmove(h->start + cpy, hh->pos, in);
158 9df487d7 2003-11-23 devnull 			hh->pos += in;
159 9df487d7 2003-11-23 devnull 		}
160 9df487d7 2003-11-23 devnull 	}else if(in && (in = read(h->fd, h->start + cpy, in)) < 0){
161 9df487d7 2003-11-23 devnull 		h->state = Herr;
162 9df487d7 2003-11-23 devnull 		h->pos = h->stop;
163 9df487d7 2003-11-23 devnull 		return nil;
164 9df487d7 2003-11-23 devnull 	}
165 9df487d7 2003-11-23 devnull 	if(in == 0)
166 9df487d7 2003-11-23 devnull 		h->state = Hend;
167 9df487d7 2003-11-23 devnull 
168 9df487d7 2003-11-23 devnull 	h->bodylen -= in;
169 9df487d7 2003-11-23 devnull 
170 9df487d7 2003-11-23 devnull 	h->stop = h->start + cpy + in;
171 9df487d7 2003-11-23 devnull 	*h->stop = '\0';
172 9df487d7 2003-11-23 devnull 	if(h->pos == h->stop)
173 9df487d7 2003-11-23 devnull 		return nil;
174 9df487d7 2003-11-23 devnull 	return h->start;
175 9df487d7 2003-11-23 devnull }
176 9df487d7 2003-11-23 devnull 
177 9df487d7 2003-11-23 devnull int
178 9df487d7 2003-11-23 devnull hbuflen(Hio *h, void *p)
179 9df487d7 2003-11-23 devnull {
180 9df487d7 2003-11-23 devnull 	return h->stop - (uchar*)p;
181 9df487d7 2003-11-23 devnull }
182 9df487d7 2003-11-23 devnull 
183 9df487d7 2003-11-23 devnull /*
184 9df487d7 2003-11-23 devnull  * prepare to receive a message body
185 9df487d7 2003-11-23 devnull  * len is the content length (~0 => unspecified)
186 9df487d7 2003-11-23 devnull  * te is the transfer encoding
187 9df487d7 2003-11-23 devnull  * returns < 0 if setup failed
188 9df487d7 2003-11-23 devnull  */
189 9df487d7 2003-11-23 devnull Hio*
190 9df487d7 2003-11-23 devnull hbodypush(Hio *hh, ulong len, HFields *te)
191 9df487d7 2003-11-23 devnull {
192 9df487d7 2003-11-23 devnull 	Hio *h;
193 9df487d7 2003-11-23 devnull 	int xe;
194 9df487d7 2003-11-23 devnull 
195 9df487d7 2003-11-23 devnull 	if(hh->state != Hread)
196 9df487d7 2003-11-23 devnull 		return nil;
197 9df487d7 2003-11-23 devnull 	xe = 0;
198 9df487d7 2003-11-23 devnull 	if(te != nil){
199 9df487d7 2003-11-23 devnull 		if(te->params != nil || te->next != nil)
200 9df487d7 2003-11-23 devnull 			return nil;
201 9df487d7 2003-11-23 devnull 		if(cistrcmp(te->s, "chunked") == 0){
202 9df487d7 2003-11-23 devnull 			xe = 1;
203 9df487d7 2003-11-23 devnull 			len = 0;
204 9df487d7 2003-11-23 devnull 		}else if(cistrcmp(te->s, "identity") == 0){
205 9df487d7 2003-11-23 devnull 			;
206 9df487d7 2003-11-23 devnull 		}else
207 9df487d7 2003-11-23 devnull 			return nil;
208 9df487d7 2003-11-23 devnull 	}
209 9df487d7 2003-11-23 devnull 
210 9df487d7 2003-11-23 devnull 	h = malloc(sizeof *h);
211 9df487d7 2003-11-23 devnull 	if(h == nil)
212 9df487d7 2003-11-23 devnull 		return nil;
213 9df487d7 2003-11-23 devnull 
214 9df487d7 2003-11-23 devnull 	h->hh = hh;
215 9df487d7 2003-11-23 devnull 	h->fd = -1;
216 9df487d7 2003-11-23 devnull 	h->seek = 0;
217 9df487d7 2003-11-23 devnull 	h->state = Hread;
218 9df487d7 2003-11-23 devnull 	h->xferenc = xe;
219 9df487d7 2003-11-23 devnull 	h->start = h->buf + 16;		/* leave space for chunk length */
220 9df487d7 2003-11-23 devnull 	h->stop = h->pos = h->start;
221 9df487d7 2003-11-23 devnull 	*h->pos = '\0';
222 9df487d7 2003-11-23 devnull 	h->bodylen = len;
223 9df487d7 2003-11-23 devnull 	return h;
224 9df487d7 2003-11-23 devnull }
225 9df487d7 2003-11-23 devnull 
226 9df487d7 2003-11-23 devnull /*
227 9df487d7 2003-11-23 devnull  * dump the state of the io buffer into a string
228 9df487d7 2003-11-23 devnull  */
229 9df487d7 2003-11-23 devnull char *
230 9df487d7 2003-11-23 devnull hunload(Hio *h)
231 9df487d7 2003-11-23 devnull {
232 9df487d7 2003-11-23 devnull 	uchar *p, *t, *stop, *buf;
233 9df487d7 2003-11-23 devnull 	int ne, n, c;
234 9df487d7 2003-11-23 devnull 
235 9df487d7 2003-11-23 devnull 	stop = h->stop;
236 9df487d7 2003-11-23 devnull 	ne = 0;
237 9df487d7 2003-11-23 devnull 	for(p = h->pos; p < stop; p++){
238 9df487d7 2003-11-23 devnull 		c = *p;
239 9df487d7 2003-11-23 devnull 		if(c == 0x80)
240 9df487d7 2003-11-23 devnull 			ne++;
241 9df487d7 2003-11-23 devnull 	}
242 9df487d7 2003-11-23 devnull 	p = h->pos;
243 9df487d7 2003-11-23 devnull 
244 9df487d7 2003-11-23 devnull 	n = (stop - p) + ne + 3;
245 9df487d7 2003-11-23 devnull 	buf = mallocz(n, 1);
246 9df487d7 2003-11-23 devnull 	if(buf == nil)
247 9df487d7 2003-11-23 devnull 		return nil;
248 9df487d7 2003-11-23 devnull 	buf[0] = hstates[h->state];
249 9df487d7 2003-11-23 devnull 	buf[1] = hxfers[h->xferenc];
250 9df487d7 2003-11-23 devnull 
251 9df487d7 2003-11-23 devnull 	t = &buf[2];
252 9df487d7 2003-11-23 devnull 	for(; p < stop; p++){
253 9df487d7 2003-11-23 devnull 		c = *p;
254 9df487d7 2003-11-23 devnull 		if(c == 0 || c == 0x80){
255 9df487d7 2003-11-23 devnull 			*t++ = 0x80;
256 9df487d7 2003-11-23 devnull 			if(c == 0x80)
257 9df487d7 2003-11-23 devnull 				*t++ = 0x80;
258 9df487d7 2003-11-23 devnull 		}else
259 9df487d7 2003-11-23 devnull 			*t++ = c;
260 9df487d7 2003-11-23 devnull 	}
261 9df487d7 2003-11-23 devnull 	*t++ = '\0';
262 9df487d7 2003-11-23 devnull 	if(t != buf + n)
263 9df487d7 2003-11-23 devnull 		return nil;
264 9df487d7 2003-11-23 devnull 	return (char*)buf;
265 9df487d7 2003-11-23 devnull }
266 9df487d7 2003-11-23 devnull 
267 9df487d7 2003-11-23 devnull /*
268 9df487d7 2003-11-23 devnull  * read the io buffer state from a string
269 9df487d7 2003-11-23 devnull  */
270 9df487d7 2003-11-23 devnull int
271 9df487d7 2003-11-23 devnull hload(Hio *h, char *buf)
272 9df487d7 2003-11-23 devnull {
273 9df487d7 2003-11-23 devnull 	uchar *p, *t, *stop;
274 9df487d7 2003-11-23 devnull 	char *s;
275 9df487d7 2003-11-23 devnull 	int c;
276 9df487d7 2003-11-23 devnull 
277 9df487d7 2003-11-23 devnull 	s = strchr(hstates, buf[0]);
278 9df487d7 2003-11-23 devnull 	if(s == nil)
279 9df487d7 2003-11-23 devnull 		return 0;
280 9df487d7 2003-11-23 devnull 	h->state = s - hstates;
281 9df487d7 2003-11-23 devnull 
282 9df487d7 2003-11-23 devnull 	s = strchr(hxfers, buf[1]);
283 9df487d7 2003-11-23 devnull 	if(s == nil)
284 9df487d7 2003-11-23 devnull 		return 0;
285 9df487d7 2003-11-23 devnull 	h->xferenc = s - hxfers;
286 9df487d7 2003-11-23 devnull 
287 9df487d7 2003-11-23 devnull 	t = h->start;
288 9df487d7 2003-11-23 devnull 	stop = t + Hsize;
289 9df487d7 2003-11-23 devnull 	for(p = (uchar*)&buf[2]; c = *p; p++){
290 9df487d7 2003-11-23 devnull 		if(c == 0x80){
291 9df487d7 2003-11-23 devnull 			if(p[1] != 0x80)
292 9df487d7 2003-11-23 devnull 				c = 0;
293 9df487d7 2003-11-23 devnull 			else
294 9df487d7 2003-11-23 devnull 				p++;
295 9df487d7 2003-11-23 devnull 		}
296 9df487d7 2003-11-23 devnull 		*t++ = c;
297 9df487d7 2003-11-23 devnull 		if(t >= stop)
298 9df487d7 2003-11-23 devnull 			return 0;
299 9df487d7 2003-11-23 devnull 	}
300 9df487d7 2003-11-23 devnull 	*t = '\0';
301 9df487d7 2003-11-23 devnull 	h->pos = h->start;
302 9df487d7 2003-11-23 devnull 	h->stop = t;
303 9df487d7 2003-11-23 devnull 	h->seek = 0;
304 9df487d7 2003-11-23 devnull 	return 1;
305 9df487d7 2003-11-23 devnull }
306 9df487d7 2003-11-23 devnull 
307 9df487d7 2003-11-23 devnull void
308 9df487d7 2003-11-23 devnull hclose(Hio *h)
309 9df487d7 2003-11-23 devnull {
310 9df487d7 2003-11-23 devnull 	if(h->fd >= 0){
311 9df487d7 2003-11-23 devnull 		if(h->state == Hwrite)
312 9df487d7 2003-11-23 devnull 			hxferenc(h, 0);
313 9df487d7 2003-11-23 devnull 		close(h->fd);
314 9df487d7 2003-11-23 devnull 	}
315 9df487d7 2003-11-23 devnull 	h->stop = h->pos = nil;
316 9df487d7 2003-11-23 devnull 	h->fd = -1;
317 9df487d7 2003-11-23 devnull }
318 9df487d7 2003-11-23 devnull 
319 9df487d7 2003-11-23 devnull /*
320 9df487d7 2003-11-23 devnull  * flush the buffer and possibly change encoding modes
321 9df487d7 2003-11-23 devnull  */
322 9df487d7 2003-11-23 devnull int
323 9df487d7 2003-11-23 devnull hxferenc(Hio *h, int on)
324 9df487d7 2003-11-23 devnull {
325 9df487d7 2003-11-23 devnull 	if(h->xferenc && !on && h->pos != h->start)
326 9df487d7 2003-11-23 devnull 		hflush(h);
327 9df487d7 2003-11-23 devnull 	if(hflush(h) < 0)
328 9df487d7 2003-11-23 devnull 		return -1;
329 9df487d7 2003-11-23 devnull 	h->xferenc = !!on;
330 9df487d7 2003-11-23 devnull 	return 0;
331 9df487d7 2003-11-23 devnull }
332 9df487d7 2003-11-23 devnull 
333 9df487d7 2003-11-23 devnull int
334 9df487d7 2003-11-23 devnull hputc(Hio *h, int c)
335 9df487d7 2003-11-23 devnull {
336 9df487d7 2003-11-23 devnull 	uchar *p;
337 9df487d7 2003-11-23 devnull 
338 9df487d7 2003-11-23 devnull 	p = h->pos;
339 9df487d7 2003-11-23 devnull 	if(p < h->stop){
340 9df487d7 2003-11-23 devnull 		h->pos = p + 1;
341 9df487d7 2003-11-23 devnull 		return *p = c;
342 9df487d7 2003-11-23 devnull 	}
343 9df487d7 2003-11-23 devnull 	if(hflush(h) < 0)
344 9df487d7 2003-11-23 devnull 		return -1;
345 9df487d7 2003-11-23 devnull 	return *h->pos++ = c;
346 9df487d7 2003-11-23 devnull }
347 9df487d7 2003-11-23 devnull 
348 9df487d7 2003-11-23 devnull static int
349 9df487d7 2003-11-23 devnull fmthflush(Fmt *f)
350 9df487d7 2003-11-23 devnull {
351 9df487d7 2003-11-23 devnull 	Hio *h;
352 9df487d7 2003-11-23 devnull 
353 9df487d7 2003-11-23 devnull 	h = f->farg;
354 9df487d7 2003-11-23 devnull 	h->pos = f->to;
355 9df487d7 2003-11-23 devnull 	if(hflush(h) < 0)
356 9df487d7 2003-11-23 devnull 		return 0;
357 9df487d7 2003-11-23 devnull 	f->stop = h->stop;
358 9df487d7 2003-11-23 devnull 	f->to = h->pos;
359 9df487d7 2003-11-23 devnull 	f->start = h->pos;
360 9df487d7 2003-11-23 devnull 	return 1;
361 9df487d7 2003-11-23 devnull }
362 9df487d7 2003-11-23 devnull 
363 9df487d7 2003-11-23 devnull int
364 9df487d7 2003-11-23 devnull hvprint(Hio *h, char *fmt, va_list args)
365 9df487d7 2003-11-23 devnull {
366 9df487d7 2003-11-23 devnull 	int n;
367 9df487d7 2003-11-23 devnull 	Fmt f;
368 9df487d7 2003-11-23 devnull 
369 9df487d7 2003-11-23 devnull 	f.runes = 0;
370 9df487d7 2003-11-23 devnull 	f.stop = h->stop;
371 9df487d7 2003-11-23 devnull 	f.to = h->pos;
372 9df487d7 2003-11-23 devnull 	f.start = h->pos;
373 9df487d7 2003-11-23 devnull 	f.flush = fmthflush;
374 9df487d7 2003-11-23 devnull 	f.farg = h;
375 9df487d7 2003-11-23 devnull 	f.nfmt = 0;
376 9df487d7 2003-11-23 devnull 	f.args = args;
377 9df487d7 2003-11-23 devnull 	n = dofmt(&f, fmt);
378 9df487d7 2003-11-23 devnull 	h->pos = f.to;
379 9df487d7 2003-11-23 devnull 	return n;
380 9df487d7 2003-11-23 devnull }
381 9df487d7 2003-11-23 devnull 
382 9df487d7 2003-11-23 devnull int
383 9df487d7 2003-11-23 devnull hprint(Hio *h, char *fmt, ...)
384 9df487d7 2003-11-23 devnull {
385 9df487d7 2003-11-23 devnull 	int n;
386 9df487d7 2003-11-23 devnull 	va_list arg;
387 9df487d7 2003-11-23 devnull 
388 9df487d7 2003-11-23 devnull 	va_start(arg, fmt);
389 9df487d7 2003-11-23 devnull 	n = hvprint(h, fmt, arg);
390 9df487d7 2003-11-23 devnull 	va_end(arg);
391 9df487d7 2003-11-23 devnull 	return n;
392 9df487d7 2003-11-23 devnull }
393 9df487d7 2003-11-23 devnull 
394 9df487d7 2003-11-23 devnull int
395 9df487d7 2003-11-23 devnull hflush(Hio *h)
396 9df487d7 2003-11-23 devnull {
397 9df487d7 2003-11-23 devnull 	uchar *s;
398 9df487d7 2003-11-23 devnull 	int w;
399 9df487d7 2003-11-23 devnull 
400 9df487d7 2003-11-23 devnull 	if(h->state != Hwrite){
401 9df487d7 2003-11-23 devnull 		h->state = Herr;
402 9df487d7 2003-11-23 devnull 		h->stop = h->pos;
403 9df487d7 2003-11-23 devnull 		return -1;
404 9df487d7 2003-11-23 devnull 	}
405 9df487d7 2003-11-23 devnull 	s = h->start;
406 9df487d7 2003-11-23 devnull 	w = h->pos - s;
407 9df487d7 2003-11-23 devnull 	if(h->xferenc){
408 9df487d7 2003-11-23 devnull 		*--s = '\n';
409 9df487d7 2003-11-23 devnull 		*--s = '\r';
410 9df487d7 2003-11-23 devnull 		do{
411 9df487d7 2003-11-23 devnull 			*--s = "0123456789abcdef"[w & 0xf];
412 9df487d7 2003-11-23 devnull 			w >>= 4;
413 9df487d7 2003-11-23 devnull 		}while(w);
414 9df487d7 2003-11-23 devnull 		h->pos[0] = '\r';
415 9df487d7 2003-11-23 devnull 		h->pos[1] = '\n';
416 9df487d7 2003-11-23 devnull 		w = &h->pos[2] - s;
417 9df487d7 2003-11-23 devnull 	}
418 9df487d7 2003-11-23 devnull 	if(write(h->fd, s, w) != w){
419 9df487d7 2003-11-23 devnull 		h->state = Herr;
420 9df487d7 2003-11-23 devnull 		h->stop = h->pos;
421 9df487d7 2003-11-23 devnull 		return -1;
422 9df487d7 2003-11-23 devnull 	}
423 9df487d7 2003-11-23 devnull 	h->seek += w;
424 9df487d7 2003-11-23 devnull 	h->pos = h->start;
425 9df487d7 2003-11-23 devnull 	return 0;
426 9df487d7 2003-11-23 devnull }
427 9df487d7 2003-11-23 devnull 
428 9df487d7 2003-11-23 devnull int
429 9df487d7 2003-11-23 devnull hwrite(Hio *h, void *vbuf, int len)
430 9df487d7 2003-11-23 devnull {
431 9df487d7 2003-11-23 devnull 	uchar *pos, *buf;
432 9df487d7 2003-11-23 devnull 	int n, m;
433 9df487d7 2003-11-23 devnull 
434 9df487d7 2003-11-23 devnull 	buf = vbuf;
435 9df487d7 2003-11-23 devnull 	n = len;
436 9df487d7 2003-11-23 devnull 	if(n < 0 || h->state != Hwrite){
437 9df487d7 2003-11-23 devnull 		h->state = Herr;
438 9df487d7 2003-11-23 devnull 		h->stop = h->pos;
439 9df487d7 2003-11-23 devnull 		return -1;
440 9df487d7 2003-11-23 devnull 	}
441 9df487d7 2003-11-23 devnull 	pos = h->pos;
442 9df487d7 2003-11-23 devnull 	if(pos + n >= h->stop){
443 9df487d7 2003-11-23 devnull 		m = pos - h->start;
444 9df487d7 2003-11-23 devnull 		if(m){
445 9df487d7 2003-11-23 devnull 			m = Hsize - m;
446 9df487d7 2003-11-23 devnull 			if(m){
447 9df487d7 2003-11-23 devnull 				memmove(pos, buf, m);
448 9df487d7 2003-11-23 devnull 				buf += m;
449 9df487d7 2003-11-23 devnull 				n -= m;
450 9df487d7 2003-11-23 devnull 			}
451 9df487d7 2003-11-23 devnull 			if(write(h->fd, h->start, Hsize) != Hsize){
452 9df487d7 2003-11-23 devnull 				h->state = Herr;
453 9df487d7 2003-11-23 devnull 				h->stop = h->pos;
454 9df487d7 2003-11-23 devnull 				return -1;
455 9df487d7 2003-11-23 devnull 			}
456 9df487d7 2003-11-23 devnull 			h->seek += Hsize;
457 9df487d7 2003-11-23 devnull 		}
458 9df487d7 2003-11-23 devnull 		m = n % Hsize;
459 9df487d7 2003-11-23 devnull 		n -= m;
460 9df487d7 2003-11-23 devnull 		if(n != 0 && write(h->fd, buf, n) != n){
461 9df487d7 2003-11-23 devnull 			h->state = Herr;
462 9df487d7 2003-11-23 devnull 			h->stop = h->pos;
463 9df487d7 2003-11-23 devnull 			return -1;
464 9df487d7 2003-11-23 devnull 		}
465 9df487d7 2003-11-23 devnull 		h->seek += n;
466 9df487d7 2003-11-23 devnull 		buf += n;
467 9df487d7 2003-11-23 devnull 		pos = h->pos = h->start;
468 9df487d7 2003-11-23 devnull 		n = m;
469 9df487d7 2003-11-23 devnull 	}
470 9df487d7 2003-11-23 devnull 	memmove(pos, buf, n);
471 9df487d7 2003-11-23 devnull 	h->pos = pos + n;
472 9df487d7 2003-11-23 devnull 	return len;
473 9df487d7 2003-11-23 devnull }