implement proper flow control for relaying postdata
[project/uhttpd.git] / uhttpd.h
1 /*
2  * uhttpd - Tiny single-threaded httpd - Main header
3  *
4  *   Copyright (C) 2010-2012 Jo-Philipp Wich <xm@subsignal.org>
5  *   Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
6  *
7  *  Licensed under the Apache License, Version 2.0 (the "License");
8  *  you may not use this file except in compliance with the License.
9  *  You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  *  Unless required by applicable law or agreed to in writing, software
14  *  distributed under the License is distributed on an "AS IS" BASIS,
15  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  *  See the License for the specific language governing permissions and
17  *  limitations under the License.
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
33 #include "utils.h"
34
35 #define UH_LIMIT_CLIENTS        64
36
37 #define __enum_header(_name) HDR_##_name,
38 #define __blobmsg_header(_name) [HDR_##_name] = { .name = #_name, .type = BLOBMSG_TYPE_STRING },
39
40 struct client;
41
42 struct config {
43         const char *docroot;
44         const char *realm;
45         const char *file;
46         const char *error_handler;
47         const char *cgi_prefix;
48         const char *cgi_path;
49         int no_symlinks;
50         int no_dirlists;
51         int network_timeout;
52         int rfc1918_filter;
53         int tcp_keepalive;
54         int max_requests;
55         int http_keepalive;
56         int script_timeout;
57 };
58
59 struct auth_realm {
60         struct list_head list;
61         const char *path;
62         const char *user;
63         const char *pass;
64 };
65
66 enum http_method {
67         UH_HTTP_MSG_GET,
68         UH_HTTP_MSG_POST,
69         UH_HTTP_MSG_HEAD,
70 };
71
72 enum http_version {
73         UH_HTTP_VER_0_9,
74         UH_HTTP_VER_1_0,
75         UH_HTTP_VER_1_1,
76 };
77
78 struct http_request {
79         enum http_method method;
80         enum http_version version;
81         int redirect_status;
82         int content_length;
83         bool expect_cont;
84         uint8_t transfer_chunked;
85         const struct auth_realm *realm;
86 };
87
88 enum client_state {
89         CLIENT_STATE_INIT,
90         CLIENT_STATE_HEADER,
91         CLIENT_STATE_DATA,
92         CLIENT_STATE_DONE,
93         CLIENT_STATE_CLOSE,
94 };
95
96 struct interpreter {
97         struct list_head list;
98         const char *path;
99         const char *ext;
100 };
101
102 struct path_info {
103         const char *root;
104         const char *phys;
105         const char *name;
106         const char *info;
107         const char *query;
108         const char *auth;
109         bool redirected;
110         struct stat stat;
111         const struct interpreter *ip;
112 };
113
114 struct env_var {
115         const char *name;
116         const char *value;
117 };
118
119 struct relay {
120         struct ustream_fd sfd;
121         struct uloop_process proc;
122         struct client *cl;
123
124         bool process_done;
125         int ret;
126         int header_ofs;
127
128         void (*header_cb)(struct relay *r, const char *name, const char *value);
129         void (*header_end)(struct relay *r);
130         void (*close)(struct relay *r, int ret);
131 };
132
133 struct dispatch_handler {
134         struct list_head list;
135
136         bool (*check_url)(const char *url);
137         bool (*check_path)(struct path_info *pi, const char *url);
138         void (*handle_request)(struct client *cl, const char *url, struct path_info *pi);
139 };
140
141 struct dispatch {
142         void (*data_send)(struct client *cl, const char *data, int len);
143         void (*data_done)(struct client *cl);
144         void (*write_cb)(struct client *cl);
145         void (*close_fds)(struct client *cl);
146         void (*free)(struct client *cl);
147         bool data_blocked;
148
149         union {
150                 struct {
151                         struct blob_attr **hdr;
152                         int fd;
153                 } file;
154                 struct {
155                         struct blob_buf hdr;
156                         struct relay r;
157                         int status_code;
158                         char *status_msg;
159                 } proc;
160         };
161 };
162
163 struct client {
164         struct list_head list;
165         int id;
166
167         struct ustream *us;
168         struct ustream_fd sfd;
169 #ifdef HAVE_TLS
170         struct ustream_ssl stream_ssl;
171 #endif
172         struct uloop_timeout timeout;
173
174         enum client_state state;
175
176         struct http_request request;
177         struct uh_addr srv_addr, peer_addr;
178
179         struct blob_buf hdr;
180         struct dispatch dispatch;
181 };
182
183 extern char uh_buf[4096];
184 extern int n_clients;
185 extern struct config conf;
186 extern const char * const http_versions[];
187 extern const char * const http_methods[];
188 extern struct dispatch_handler cgi_dispatch;
189
190 void uh_index_add(const char *filename);
191
192 void uh_accept_client(int fd);
193
194 void uh_unblock_listeners(void);
195 void uh_setup_listeners(void);
196 int uh_socket_bind(const char *host, const char *port, bool tls);
197
198 bool uh_use_chunked(struct client *cl);
199 void uh_chunk_write(struct client *cl, const void *data, int len);
200 void uh_chunk_vprintf(struct client *cl, const char *format, va_list arg);
201
202 void __printf(2, 3)
203 uh_chunk_printf(struct client *cl, const char *format, ...);
204
205 void uh_chunk_eof(struct client *cl);
206 void uh_request_done(struct client *cl);
207
208 void uh_http_header(struct client *cl, int code, const char *summary);
209 void __printf(4, 5)
210 uh_client_error(struct client *cl, int code, const char *summary, const char *fmt, ...);
211
212 void uh_handle_request(struct client *cl);
213 void client_poll_post_data(struct client *cl);
214
215 void uh_auth_add(const char *path, const char *user, const char *pass);
216 bool uh_auth_check(struct client *cl, struct path_info *pi);
217
218 void uh_close_listen_fds(void);
219 void uh_close_fds(void);
220
221 void uh_interpreter_add(const char *ext, const char *path);
222 void uh_dispatch_add(struct dispatch_handler *d);
223
224 void uh_relay_open(struct client *cl, struct relay *r, int fd, int pid);
225 void uh_relay_close(struct relay *r, int ret);
226 void uh_relay_free(struct relay *r);
227
228 struct env_var *uh_get_process_vars(struct client *cl, struct path_info *pi);
229 bool uh_create_process(struct client *cl, struct path_info *pi,
230                        void (*cb)(struct client *cl, struct path_info *pi, int fd));
231
232 #endif