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>
5 9df487d7 2003-11-23 devnull static char hstates[] = "nrewE";
6 9df487d7 2003-11-23 devnull static char hxfers[] = " x";
7 f599f28c 2007-04-21 devnull static int _hflush(Hio*, int, int);
10 9df487d7 2003-11-23 devnull hinit(Hio *h, int fd, int mode)
12 9df487d7 2003-11-23 devnull if(fd == -1 || mode != Hread && mode != Hwrite)
13 9df487d7 2003-11-23 devnull return -1;
14 9df487d7 2003-11-23 devnull h->hh = nil;
15 9df487d7 2003-11-23 devnull h->fd = fd;
16 9df487d7 2003-11-23 devnull h->seek = 0;
17 9df487d7 2003-11-23 devnull h->state = mode;
18 9df487d7 2003-11-23 devnull h->start = h->buf + 16; /* leave space for chunk length */
19 9df487d7 2003-11-23 devnull h->stop = h->pos = h->start;
20 9df487d7 2003-11-23 devnull if(mode == Hread){
21 9df487d7 2003-11-23 devnull h->bodylen = ~0UL;
22 9df487d7 2003-11-23 devnull *h->pos = '\0';
24 9df487d7 2003-11-23 devnull h->stop = h->start + Hsize;
25 9df487d7 2003-11-23 devnull return 0;
29 9df487d7 2003-11-23 devnull hiserror(Hio *h)
31 9df487d7 2003-11-23 devnull return h->state == Herr;
35 9df487d7 2003-11-23 devnull hgetc(Hio *h)
37 9df487d7 2003-11-23 devnull uchar *p;
39 9df487d7 2003-11-23 devnull p = h->pos;
40 9df487d7 2003-11-23 devnull if(p < h->stop){
41 9df487d7 2003-11-23 devnull h->pos = p + 1;
42 9df487d7 2003-11-23 devnull return *p;
44 9df487d7 2003-11-23 devnull p -= UTFmax;
45 9df487d7 2003-11-23 devnull if(p < h->start)
46 9df487d7 2003-11-23 devnull p = h->start;
47 9df487d7 2003-11-23 devnull if(!hreadbuf(h, p) || h->pos == h->stop)
48 9df487d7 2003-11-23 devnull return -1;
49 9df487d7 2003-11-23 devnull return *h->pos++;
53 9df487d7 2003-11-23 devnull hungetc(Hio *h)
55 9df487d7 2003-11-23 devnull if(h->state == Hend)
56 9df487d7 2003-11-23 devnull h->state = Hread;
57 9df487d7 2003-11-23 devnull else if(h->state == Hread)
58 9df487d7 2003-11-23 devnull h->pos--;
59 9df487d7 2003-11-23 devnull if(h->pos < h->start || h->state != Hread){
60 9df487d7 2003-11-23 devnull h->state = Herr;
61 9df487d7 2003-11-23 devnull h->pos = h->stop;
62 9df487d7 2003-11-23 devnull return -1;
64 9df487d7 2003-11-23 devnull return 0;
68 9df487d7 2003-11-23 devnull * fill the buffer, saving contents from vsave onwards.
69 9df487d7 2003-11-23 devnull * nothing is saved if vsave is nil.
70 9df487d7 2003-11-23 devnull * returns the beginning of the buffer.
72 9df487d7 2003-11-23 devnull * understands message body sizes and chunked transfer encoding
75 9df487d7 2003-11-23 devnull hreadbuf(Hio *h, void *vsave)
78 9df487d7 2003-11-23 devnull uchar *save;
79 9df487d7 2003-11-23 devnull int c, in, cpy, dpos;
81 9df487d7 2003-11-23 devnull save = vsave;
82 9df487d7 2003-11-23 devnull if(save && (save < h->start || save > h->stop)
83 9df487d7 2003-11-23 devnull || h->state != Hread && h->state != Hend){
84 9df487d7 2003-11-23 devnull h->state = Herr;
85 9df487d7 2003-11-23 devnull h->pos = h->stop;
86 9df487d7 2003-11-23 devnull return nil;
89 9df487d7 2003-11-23 devnull dpos = 0;
90 9df487d7 2003-11-23 devnull if(save && h->pos > save)
91 9df487d7 2003-11-23 devnull dpos = h->pos - save;
93 9df487d7 2003-11-23 devnull if(save){
94 9df487d7 2003-11-23 devnull cpy = h->stop - save;
95 9df487d7 2003-11-23 devnull memmove(h->start, save, cpy);
97 9df487d7 2003-11-23 devnull h->seek += h->stop - h->start - cpy;
98 9df487d7 2003-11-23 devnull h->pos = h->start + dpos;
100 9df487d7 2003-11-23 devnull in = Hsize - cpy;
101 9df487d7 2003-11-23 devnull if(h->state == Hend)
103 9df487d7 2003-11-23 devnull else if(in > h->bodylen)
104 9df487d7 2003-11-23 devnull in = h->bodylen;
107 9df487d7 2003-11-23 devnull * for chunked encoding, fill buffer,
108 9df487d7 2003-11-23 devnull * then read in new chunk length and wipe out that line
110 9df487d7 2003-11-23 devnull hh = h->hh;
111 9df487d7 2003-11-23 devnull if(hh != nil){
112 9df487d7 2003-11-23 devnull if(!in && h->xferenc && h->state != Hend){
113 9df487d7 2003-11-23 devnull if(h->xferenc == 2){
114 9df487d7 2003-11-23 devnull c = hgetc(hh);
115 9df487d7 2003-11-23 devnull if(c == '\r')
116 9df487d7 2003-11-23 devnull c = hgetc(hh);
117 9df487d7 2003-11-23 devnull if(c != '\n'){
118 9df487d7 2003-11-23 devnull h->pos = h->stop;
119 9df487d7 2003-11-23 devnull h->state = Herr;
120 9df487d7 2003-11-23 devnull return nil;
123 9df487d7 2003-11-23 devnull h->xferenc = 2;
125 9df487d7 2003-11-23 devnull while((c = hgetc(hh)) != '\n'){
126 9df487d7 2003-11-23 devnull if(c >= '0' && c <= '9')
127 9df487d7 2003-11-23 devnull c -= '0';
128 9df487d7 2003-11-23 devnull else if(c >= 'a' && c <= 'f')
129 9df487d7 2003-11-23 devnull c -= 'a' - 10;
130 9df487d7 2003-11-23 devnull else if(c >= 'A' && c <= 'F')
131 9df487d7 2003-11-23 devnull c -= 'A' - 10;
134 9df487d7 2003-11-23 devnull in = in * 16 + c;
136 9df487d7 2003-11-23 devnull while(c != '\n'){
137 9df487d7 2003-11-23 devnull if(c < 0){
138 9df487d7 2003-11-23 devnull h->pos = h->stop;
139 9df487d7 2003-11-23 devnull h->state = Herr;
140 9df487d7 2003-11-23 devnull return nil;
142 9df487d7 2003-11-23 devnull c = hgetc(hh);
144 9df487d7 2003-11-23 devnull h->bodylen = in;
146 9df487d7 2003-11-23 devnull in = Hsize - cpy;
147 9df487d7 2003-11-23 devnull if(in > h->bodylen)
148 9df487d7 2003-11-23 devnull in = h->bodylen;
151 9df487d7 2003-11-23 devnull while(hh->pos + in > hh->stop){
152 9df487d7 2003-11-23 devnull if(hreadbuf(hh, hh->pos) == nil){
153 9df487d7 2003-11-23 devnull h->pos = h->stop;
154 9df487d7 2003-11-23 devnull h->state = Herr;
155 9df487d7 2003-11-23 devnull return nil;
158 9df487d7 2003-11-23 devnull memmove(h->start + cpy, hh->pos, in);
159 9df487d7 2003-11-23 devnull hh->pos += in;
161 2277c5d7 2004-03-21 devnull }else if(in){
162 2277c5d7 2004-03-21 devnull if((in = read(h->fd, h->start + cpy, in)) < 0){
163 2277c5d7 2004-03-21 devnull h->state = Herr;
164 2277c5d7 2004-03-21 devnull h->pos = h->stop;
165 2277c5d7 2004-03-21 devnull return nil;
168 9df487d7 2003-11-23 devnull if(in == 0)
169 9df487d7 2003-11-23 devnull h->state = Hend;
171 9df487d7 2003-11-23 devnull h->bodylen -= in;
173 9df487d7 2003-11-23 devnull h->stop = h->start + cpy + in;
174 9df487d7 2003-11-23 devnull *h->stop = '\0';
175 9df487d7 2003-11-23 devnull if(h->pos == h->stop)
176 9df487d7 2003-11-23 devnull return nil;
177 9df487d7 2003-11-23 devnull return h->start;
181 9df487d7 2003-11-23 devnull hbuflen(Hio *h, void *p)
183 9df487d7 2003-11-23 devnull return h->stop - (uchar*)p;
187 9df487d7 2003-11-23 devnull * prepare to receive a message body
188 9df487d7 2003-11-23 devnull * len is the content length (~0 => unspecified)
189 9df487d7 2003-11-23 devnull * te is the transfer encoding
190 9df487d7 2003-11-23 devnull * returns < 0 if setup failed
193 9df487d7 2003-11-23 devnull hbodypush(Hio *hh, ulong len, HFields *te)
198 9df487d7 2003-11-23 devnull if(hh->state != Hread)
199 9df487d7 2003-11-23 devnull return nil;
201 9df487d7 2003-11-23 devnull if(te != nil){
202 9df487d7 2003-11-23 devnull if(te->params != nil || te->next != nil)
203 9df487d7 2003-11-23 devnull return nil;
204 9df487d7 2003-11-23 devnull if(cistrcmp(te->s, "chunked") == 0){
206 9df487d7 2003-11-23 devnull len = 0;
207 9df487d7 2003-11-23 devnull }else if(cistrcmp(te->s, "identity") == 0){
210 9df487d7 2003-11-23 devnull return nil;
213 9df487d7 2003-11-23 devnull h = malloc(sizeof *h);
214 9df487d7 2003-11-23 devnull if(h == nil)
215 9df487d7 2003-11-23 devnull return nil;
217 9df487d7 2003-11-23 devnull h->hh = hh;
218 9df487d7 2003-11-23 devnull h->fd = -1;
219 9df487d7 2003-11-23 devnull h->seek = 0;
220 9df487d7 2003-11-23 devnull h->state = Hread;
221 9df487d7 2003-11-23 devnull h->xferenc = xe;
222 9df487d7 2003-11-23 devnull h->start = h->buf + 16; /* leave space for chunk length */
223 9df487d7 2003-11-23 devnull h->stop = h->pos = h->start;
224 9df487d7 2003-11-23 devnull *h->pos = '\0';
225 9df487d7 2003-11-23 devnull h->bodylen = len;
226 9df487d7 2003-11-23 devnull return h;
230 9df487d7 2003-11-23 devnull * dump the state of the io buffer into a string
233 9df487d7 2003-11-23 devnull hunload(Hio *h)
235 9df487d7 2003-11-23 devnull uchar *p, *t, *stop, *buf;
236 9df487d7 2003-11-23 devnull int ne, n, c;
238 9df487d7 2003-11-23 devnull stop = h->stop;
240 9df487d7 2003-11-23 devnull for(p = h->pos; p < stop; p++){
242 9df487d7 2003-11-23 devnull if(c == 0x80)
245 9df487d7 2003-11-23 devnull p = h->pos;
247 9df487d7 2003-11-23 devnull n = (stop - p) + ne + 3;
248 9df487d7 2003-11-23 devnull buf = mallocz(n, 1);
249 9df487d7 2003-11-23 devnull if(buf == nil)
250 9df487d7 2003-11-23 devnull return nil;
251 9df487d7 2003-11-23 devnull buf[0] = hstates[h->state];
252 9df487d7 2003-11-23 devnull buf[1] = hxfers[h->xferenc];
254 9df487d7 2003-11-23 devnull t = &buf[2];
255 9df487d7 2003-11-23 devnull for(; p < stop; p++){
257 9df487d7 2003-11-23 devnull if(c == 0 || c == 0x80){
258 9df487d7 2003-11-23 devnull *t++ = 0x80;
259 9df487d7 2003-11-23 devnull if(c == 0x80)
260 9df487d7 2003-11-23 devnull *t++ = 0x80;
262 9df487d7 2003-11-23 devnull *t++ = c;
264 9df487d7 2003-11-23 devnull *t++ = '\0';
265 9df487d7 2003-11-23 devnull if(t != buf + n)
266 9df487d7 2003-11-23 devnull return nil;
267 9df487d7 2003-11-23 devnull return (char*)buf;
271 9df487d7 2003-11-23 devnull * read the io buffer state from a string
274 9df487d7 2003-11-23 devnull hload(Hio *h, char *buf)
276 9df487d7 2003-11-23 devnull uchar *p, *t, *stop;
277 9df487d7 2003-11-23 devnull char *s;
280 9df487d7 2003-11-23 devnull s = strchr(hstates, buf[0]);
281 9df487d7 2003-11-23 devnull if(s == nil)
282 2e965b33 2004-05-05 devnull return -1;
283 9df487d7 2003-11-23 devnull h->state = s - hstates;
285 9df487d7 2003-11-23 devnull s = strchr(hxfers, buf[1]);
286 9df487d7 2003-11-23 devnull if(s == nil)
287 4bef0baf 2004-12-27 devnull return -1;
288 9df487d7 2003-11-23 devnull h->xferenc = s - hxfers;
290 9df487d7 2003-11-23 devnull t = h->start;
291 9df487d7 2003-11-23 devnull stop = t + Hsize;
292 9df487d7 2003-11-23 devnull for(p = (uchar*)&buf[2]; c = *p; p++){
293 9df487d7 2003-11-23 devnull if(c == 0x80){
294 9df487d7 2003-11-23 devnull if(p[1] != 0x80)
299 9df487d7 2003-11-23 devnull *t++ = c;
300 9df487d7 2003-11-23 devnull if(t >= stop)
301 2e965b33 2004-05-05 devnull return -1;
303 9df487d7 2003-11-23 devnull *t = '\0';
304 9df487d7 2003-11-23 devnull h->pos = h->start;
305 9df487d7 2003-11-23 devnull h->stop = t;
306 9df487d7 2003-11-23 devnull h->seek = 0;
307 2e965b33 2004-05-05 devnull return 0;
311 9df487d7 2003-11-23 devnull hclose(Hio *h)
313 9df487d7 2003-11-23 devnull if(h->fd >= 0){
314 9df487d7 2003-11-23 devnull if(h->state == Hwrite)
315 9df487d7 2003-11-23 devnull hxferenc(h, 0);
316 9df487d7 2003-11-23 devnull close(h->fd);
318 9df487d7 2003-11-23 devnull h->stop = h->pos = nil;
319 9df487d7 2003-11-23 devnull h->fd = -1;
323 9df487d7 2003-11-23 devnull * flush the buffer and possibly change encoding modes
326 9df487d7 2003-11-23 devnull hxferenc(Hio *h, int on)
328 9df487d7 2003-11-23 devnull if(h->xferenc && !on && h->pos != h->start)
329 9df487d7 2003-11-23 devnull hflush(h);
330 f599f28c 2007-04-21 devnull if(_hflush(h, 1, 0) < 0)
331 9df487d7 2003-11-23 devnull return -1;
332 9df487d7 2003-11-23 devnull h->xferenc = !!on;
333 9df487d7 2003-11-23 devnull return 0;
337 9df487d7 2003-11-23 devnull hputc(Hio *h, int c)
339 9df487d7 2003-11-23 devnull uchar *p;
341 9df487d7 2003-11-23 devnull p = h->pos;
342 9df487d7 2003-11-23 devnull if(p < h->stop){
343 9df487d7 2003-11-23 devnull h->pos = p + 1;
344 9df487d7 2003-11-23 devnull return *p = c;
346 9df487d7 2003-11-23 devnull if(hflush(h) < 0)
347 9df487d7 2003-11-23 devnull return -1;
348 9df487d7 2003-11-23 devnull return *h->pos++ = c;
351 9df487d7 2003-11-23 devnull static int
352 9df487d7 2003-11-23 devnull fmthflush(Fmt *f)
356 9df487d7 2003-11-23 devnull h = f->farg;
357 9df487d7 2003-11-23 devnull h->pos = f->to;
358 9df487d7 2003-11-23 devnull if(hflush(h) < 0)
359 9df487d7 2003-11-23 devnull return 0;
360 9df487d7 2003-11-23 devnull f->stop = h->stop;
361 9df487d7 2003-11-23 devnull f->to = h->pos;
362 9df487d7 2003-11-23 devnull f->start = h->pos;
363 9df487d7 2003-11-23 devnull return 1;
367 9df487d7 2003-11-23 devnull hvprint(Hio *h, char *fmt, va_list args)
372 9df487d7 2003-11-23 devnull f.runes = 0;
373 9df487d7 2003-11-23 devnull f.stop = h->stop;
374 9df487d7 2003-11-23 devnull f.to = h->pos;
375 9df487d7 2003-11-23 devnull f.start = h->pos;
376 9df487d7 2003-11-23 devnull f.flush = fmthflush;
377 9df487d7 2003-11-23 devnull f.farg = h;
378 9df487d7 2003-11-23 devnull f.nfmt = 0;
379 30df15f9 2007-04-21 devnull fmtlocaleinit(&f, nil, nil, nil);
380 18443075 2005-02-08 devnull n = fmtvprint(&f, fmt, args);
381 9df487d7 2003-11-23 devnull h->pos = f.to;
382 9df487d7 2003-11-23 devnull return n;
386 9df487d7 2003-11-23 devnull hprint(Hio *h, char *fmt, ...)
389 9df487d7 2003-11-23 devnull va_list arg;
391 9df487d7 2003-11-23 devnull va_start(arg, fmt);
392 9df487d7 2003-11-23 devnull n = hvprint(h, fmt, arg);
393 9df487d7 2003-11-23 devnull va_end(arg);
394 9df487d7 2003-11-23 devnull return n;
397 4bef0baf 2004-12-27 devnull static int
398 f599f28c 2007-04-21 devnull _hflush(Hio *h, int force, int dolength)
400 9df487d7 2003-11-23 devnull uchar *s;
403 9df487d7 2003-11-23 devnull if(h->state != Hwrite){
404 9df487d7 2003-11-23 devnull h->state = Herr;
405 9df487d7 2003-11-23 devnull h->stop = h->pos;
406 9df487d7 2003-11-23 devnull return -1;
408 9df487d7 2003-11-23 devnull s = h->start;
409 9df487d7 2003-11-23 devnull w = h->pos - s;
410 f599f28c 2007-04-21 devnull if(w == 0 && !force)
411 f599f28c 2007-04-21 devnull return 0;
412 9df487d7 2003-11-23 devnull if(h->xferenc){
413 9df487d7 2003-11-23 devnull *--s = '\n';
414 9df487d7 2003-11-23 devnull *--s = '\r';
416 9df487d7 2003-11-23 devnull *--s = "0123456789abcdef"[w & 0xf];
417 9df487d7 2003-11-23 devnull w >>= 4;
418 9df487d7 2003-11-23 devnull }while(w);
419 9df487d7 2003-11-23 devnull h->pos[0] = '\r';
420 9df487d7 2003-11-23 devnull h->pos[1] = '\n';
421 9df487d7 2003-11-23 devnull w = &h->pos[2] - s;
423 4bef0baf 2004-12-27 devnull if(dolength)
424 4bef0baf 2004-12-27 devnull fprint(h->fd, "Content-Length: %d\r\n\r\n", w);
425 9df487d7 2003-11-23 devnull if(write(h->fd, s, w) != w){
426 9df487d7 2003-11-23 devnull h->state = Herr;
427 9df487d7 2003-11-23 devnull h->stop = h->pos;
428 9df487d7 2003-11-23 devnull return -1;
430 9df487d7 2003-11-23 devnull h->seek += w;
431 9df487d7 2003-11-23 devnull h->pos = h->start;
432 9df487d7 2003-11-23 devnull return 0;
436 4bef0baf 2004-12-27 devnull hflush(Hio *h)
438 f599f28c 2007-04-21 devnull return _hflush(h, 0, 0);
442 4bef0baf 2004-12-27 devnull hlflush(Hio* h)
444 f599f28c 2007-04-21 devnull return _hflush(h, 0, 1);
448 9df487d7 2003-11-23 devnull hwrite(Hio *h, void *vbuf, int len)
450 7eded25c 2005-02-16 devnull uchar *buf;
451 9df487d7 2003-11-23 devnull int n, m;
453 9df487d7 2003-11-23 devnull buf = vbuf;
454 9df487d7 2003-11-23 devnull n = len;
455 9df487d7 2003-11-23 devnull if(n < 0 || h->state != Hwrite){
456 9df487d7 2003-11-23 devnull h->state = Herr;
457 9df487d7 2003-11-23 devnull h->stop = h->pos;
458 9df487d7 2003-11-23 devnull return -1;
460 7eded25c 2005-02-16 devnull if(h->pos + n >= h->stop){
461 7eded25c 2005-02-16 devnull if(h->start != h->pos)
462 7eded25c 2005-02-16 devnull if(hflush(h) < 0)
463 9df487d7 2003-11-23 devnull return -1;
464 7eded25c 2005-02-16 devnull while(h->pos + n >= h->stop){
465 7eded25c 2005-02-16 devnull m = h->stop - h->pos;
466 7eded25c 2005-02-16 devnull if(h->xferenc){
467 7eded25c 2005-02-16 devnull memmove(h->pos, buf, m);
468 7eded25c 2005-02-16 devnull h->pos += m;
469 7eded25c 2005-02-16 devnull if(hflush(h) < 0)
470 7eded25c 2005-02-16 devnull return -1;
472 7eded25c 2005-02-16 devnull if(write(h->fd, buf, m) != m){
473 7eded25c 2005-02-16 devnull h->state = Herr;
474 7eded25c 2005-02-16 devnull h->stop = h->pos;
475 7eded25c 2005-02-16 devnull return -1;
477 7eded25c 2005-02-16 devnull h->seek += m;
480 7eded25c 2005-02-16 devnull buf += m;
483 7eded25c 2005-02-16 devnull memmove(h->pos, buf, n);
484 7eded25c 2005-02-16 devnull h->pos += n;
485 9df487d7 2003-11-23 devnull return len;