2 * uhttpd - Tiny single-threaded httpd
4 * Copyright (C) 2010-2013 Jo-Philipp Wich <xm@subsignal.org>
5 * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
7 * Permission to use, copy, modify, and/or distribute this software for any
8 * purpose with or without fee is hereby granted, provided that the above
9 * copyright notice and this permission notice appear in all copies.
11 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
23 void uh_relay_free(struct relay *r)
29 kill(r->proc.pid, SIGKILL);
31 uloop_process_delete(&r->proc);
32 ustream_free(&r->sfd.stream);
38 void uh_relay_close(struct relay *r, int ret)
40 struct ustream *us = &r->sfd.stream;
45 us->notify_read = NULL;
46 us->notify_write = NULL;
47 us->notify_state = NULL;
53 static void relay_error(struct relay *r)
55 struct ustream *s = &r->sfd.stream;
59 ustream_get_read_buf(s, &len);
61 ustream_consume(s, len);
62 ustream_state_change(s);
65 static void relay_process_headers(struct relay *r)
67 struct ustream *s = &r->sfd.stream;
74 while (r->header_cb) {
78 buf = ustream_get_read_buf(s, &len);
82 newline = strchr(buf, '\n');
86 line_len = newline + 1 - buf;
87 if (newline > buf && newline[-1] == '\r')
95 ustream_consume(s, line_len);
99 val = uh_split_header(buf);
105 r->header_cb(r, buf, val);
106 ustream_consume(s, line_len);
110 static void relay_read_cb(struct ustream *s, int bytes)
112 struct relay *r = container_of(s, struct relay, sfd.stream);
113 struct client *cl = r->cl;
114 struct ustream *us = cl->us;
118 relay_process_headers(r);
122 * if eof, ensure that remaining data is discarded, so the
123 * state change cb will tear down the stream
130 if (!s->eof && ustream_pending_data(us, true)) {
131 ustream_set_read_blocked(s, true);
135 buf = ustream_get_read_buf(s, &len);
139 uh_chunk_write(cl, buf, len);
140 ustream_consume(s, len);
143 static void relay_close_if_done(struct relay *r)
145 struct ustream *s = &r->sfd.stream;
147 if (!s->eof || ustream_pending_data(s, false))
150 uh_relay_close(r, r->ret);
153 static void relay_state_cb(struct ustream *s)
155 struct relay *r = container_of(s, struct relay, sfd.stream);
158 relay_close_if_done(r);
161 static void relay_proc_cb(struct uloop_process *proc, int ret)
163 struct relay *r = container_of(proc, struct relay, proc);
165 ustream_poll(&r->sfd.stream);
166 r->process_done = true;
168 relay_close_if_done(r);
171 void uh_relay_kill(struct client *cl, struct relay *r)
173 struct ustream *us = &r->sfd.stream;
175 kill(r->proc.pid, SIGKILL);
177 ustream_state_change(us);
180 void uh_relay_open(struct client *cl, struct relay *r, int fd, int pid)
182 struct ustream *us = &r->sfd.stream;
185 us->notify_read = relay_read_cb;
186 us->notify_state = relay_state_cb;
187 us->string_data = true;
188 ustream_fd_init(&r->sfd, fd);
191 r->proc.cb = relay_proc_cb;
192 uloop_process_add(&r->proc);