relay: fix close handling
[project/uhttpd.git] / uhttpd.h
1 /*
2 * uhttpd - Tiny single-threaded httpd
3 *
4 * Copyright (C) 2010-2013 Jo-Philipp Wich <xm@subsignal.org>
5 * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
6 *
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.
10 *
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.
18 */
19
20 #ifndef __UHTTPD_H
21 #define __UHTTPD_H
22
23 #include <netinet/in.h>
24 #include <limits.h>
25 #include <dirent.h>
26
27 #include <libubox/list.h>
28 #include <libubox/uloop.h>
29 #include <libubox/ustream.h>
30 #include <libubox/blob.h>
31 #include <libubox/utils.h>
32 #ifdef HAVE_UBUS
33 #include <libubus.h>
34 #include <json/json.h>
35 #endif
36 #ifdef HAVE_TLS
37 #include <libubox/ustream-ssl.h>
38 #endif
39
40 #include "utils.h"
41
42 #define UH_LIMIT_CLIENTS 64
43
44 #define __enum_header(_name, _val) HDR_##_name,
45 #define __blobmsg_header(_name, _val) [HDR_##_name] = { .name = #_val, .type = BLOBMSG_TYPE_STRING },
46
47 struct client;
48
49 struct config {
50 const char *docroot;
51 const char *realm;
52 const char *file;
53 const char *error_handler;
54 const char *cgi_prefix;
55 const char *cgi_path;
56 const char *lua_handler;
57 const char *lua_prefix;
58 const char *ubus_prefix;
59 const char *ubus_socket;
60 int no_symlinks;
61 int no_dirlists;
62 int network_timeout;
63 int rfc1918_filter;
64 int tcp_keepalive;
65 int max_script_requests;
66 int max_connections;
67 int http_keepalive;
68 int script_timeout;
69 int ubus_noauth;
70 };
71
72 struct auth_realm {
73 struct list_head list;
74 const char *path;
75 const char *user;
76 const char *pass;
77 };
78
79 enum http_method {
80 UH_HTTP_MSG_GET,
81 UH_HTTP_MSG_POST,
82 UH_HTTP_MSG_HEAD,
83 };
84
85 enum http_version {
86 UH_HTTP_VER_0_9,
87 UH_HTTP_VER_1_0,
88 UH_HTTP_VER_1_1,
89 };
90
91 enum http_user_agent {
92 UH_UA_UNKNOWN,
93 UH_UA_GECKO,
94 UH_UA_CHROME,
95 UH_UA_SAFARI,
96 UH_UA_MSIE,
97 UH_UA_KONQUEROR,
98 UH_UA_OPERA,
99 UH_UA_MSIE_OLD,
100 UH_UA_MSIE_NEW,
101 };
102
103 struct http_request {
104 enum http_method method;
105 enum http_version version;
106 enum http_user_agent ua;
107 int redirect_status;
108 int content_length;
109 bool expect_cont;
110 bool connection_close;
111 uint8_t transfer_chunked;
112 const struct auth_realm *realm;
113 };
114
115 enum client_state {
116 CLIENT_STATE_INIT,
117 CLIENT_STATE_HEADER,
118 CLIENT_STATE_DATA,
119 CLIENT_STATE_DONE,
120 CLIENT_STATE_CLOSE,
121 };
122
123 struct interpreter {
124 struct list_head list;
125 const char *path;
126 const char *ext;
127 };
128
129 struct path_info {
130 const char *root;
131 const char *phys;
132 const char *name;
133 const char *info;
134 const char *query;
135 const char *auth;
136 bool redirected;
137 struct stat stat;
138 const struct interpreter *ip;
139 };
140
141 struct env_var {
142 const char *name;
143 const char *value;
144 };
145
146 struct relay {
147 struct ustream_fd sfd;
148 struct uloop_process proc;
149 struct uloop_timeout timeout;
150 struct client *cl;
151
152 bool process_done;
153 int ret;
154 int header_ofs;
155
156 void (*header_cb)(struct relay *r, const char *name, const char *value);
157 void (*header_end)(struct relay *r);
158 void (*close)(struct relay *r, int ret);
159 };
160
161 struct dispatch_proc {
162 struct uloop_timeout timeout;
163 struct blob_buf hdr;
164 struct uloop_fd wrfd;
165 struct relay r;
166 int status_code;
167 char *status_msg;
168 };
169
170 struct dispatch_handler {
171 struct list_head list;
172 bool script;
173
174 bool (*check_url)(const char *url);
175 bool (*check_path)(struct path_info *pi, const char *url);
176 void (*handle_request)(struct client *cl, char *url, struct path_info *pi);
177 };
178
179 #ifdef HAVE_UBUS
180 struct dispatch_ubus {
181 struct ubus_request req;
182
183 struct uloop_timeout timeout;
184 struct json_tokener *jstok;
185 struct json_object *jsobj;
186 struct json_object *jsobj_cur;
187 int post_len;
188
189 const char *sid;
190 uint32_t obj;
191 const char *func;
192
193 struct blob_buf buf;
194 bool req_pending;
195 bool array;
196 int array_idx;
197 };
198 #endif
199
200 struct dispatch {
201 int (*data_send)(struct client *cl, const char *data, int len);
202 void (*data_done)(struct client *cl);
203 void (*write_cb)(struct client *cl);
204 void (*close_fds)(struct client *cl);
205 void (*free)(struct client *cl);
206
207 void *req_data;
208 void (*req_free)(struct client *cl);
209
210 bool data_blocked;
211
212 union {
213 struct {
214 struct blob_attr **hdr;
215 int fd;
216 } file;
217 struct dispatch_proc proc;
218 #ifdef HAVE_UBUS
219 struct dispatch_ubus ubus;
220 #endif
221 };
222 };
223
224 struct client {
225 struct list_head list;
226 int id;
227
228 struct ustream *us;
229 struct ustream_fd sfd;
230 #ifdef HAVE_TLS
231 struct ustream_ssl ssl;
232 #endif
233 struct uloop_timeout timeout;
234 int requests;
235
236 enum client_state state;
237 bool tls;
238
239 struct http_request request;
240 struct uh_addr srv_addr, peer_addr;
241
242 struct blob_buf hdr;
243 struct dispatch dispatch;
244 };
245
246 extern char uh_buf[4096];
247 extern int n_clients;
248 extern struct config conf;
249 extern const char * const http_versions[];
250 extern const char * const http_methods[];
251 extern struct dispatch_handler cgi_dispatch;
252
253 void uh_index_add(const char *filename);
254
255 bool uh_accept_client(int fd, bool tls);
256
257 void uh_unblock_listeners(void);
258 void uh_setup_listeners(void);
259 int uh_socket_bind(const char *host, const char *port, bool tls);
260
261 bool uh_use_chunked(struct client *cl);
262 void uh_chunk_write(struct client *cl, const void *data, int len);
263 void uh_chunk_vprintf(struct client *cl, const char *format, va_list arg);
264
265 void __printf(2, 3)
266 uh_chunk_printf(struct client *cl, const char *format, ...);
267
268 void uh_chunk_eof(struct client *cl);
269 void uh_request_done(struct client *cl);
270
271 void uh_http_header(struct client *cl, int code, const char *summary);
272 void __printf(4, 5)
273 uh_client_error(struct client *cl, int code, const char *summary, const char *fmt, ...);
274
275 void uh_handle_request(struct client *cl);
276 void client_poll_post_data(struct client *cl);
277 void uh_client_read_cb(struct client *cl);
278 void uh_client_notify_state(struct client *cl);
279
280 void uh_auth_add(const char *path, const char *user, const char *pass);
281 bool uh_auth_check(struct client *cl, struct path_info *pi);
282
283 void uh_close_listen_fds(void);
284 void uh_close_fds(void);
285
286 void uh_interpreter_add(const char *ext, const char *path);
287 void uh_dispatch_add(struct dispatch_handler *d);
288
289 void uh_relay_open(struct client *cl, struct relay *r, int fd, int pid);
290 void uh_relay_close(struct relay *r, int ret);
291 void uh_relay_free(struct relay *r);
292 void uh_relay_kill(struct client *cl, struct relay *r);
293
294 struct env_var *uh_get_process_vars(struct client *cl, struct path_info *pi);
295 bool uh_create_process(struct client *cl, struct path_info *pi, char *url,
296 void (*cb)(struct client *cl, struct path_info *pi, char *url));
297
298 int uh_plugin_init(const char *name);
299 void uh_plugin_post_init(void);
300
301 #endif