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