fix use-after-realloc issue with the request url
[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 union {
148 struct {
149 struct blob_attr **hdr;
150 int fd;
151 } file;
152 struct {
153 struct blob_buf hdr;
154 struct relay r;
155 int status_code;
156 char *status_msg;
157 } proc;
158 };
159 };
160
161 struct client {
162 struct list_head list;
163 int id;
164
165 struct ustream *us;
166 struct ustream_fd sfd;
167 #ifdef HAVE_TLS
168 struct ustream_ssl stream_ssl;
169 #endif
170 struct uloop_timeout timeout;
171
172 enum client_state state;
173
174 struct http_request request;
175 struct uh_addr srv_addr, peer_addr;
176
177 struct blob_buf hdr;
178 struct dispatch dispatch;
179 };
180
181 extern char uh_buf[4096];
182 extern int n_clients;
183 extern struct config conf;
184 extern const char * const http_versions[];
185 extern const char * const http_methods[];
186 extern struct dispatch_handler cgi_dispatch;
187
188 void uh_index_add(const char *filename);
189
190 void uh_accept_client(int fd);
191
192 void uh_unblock_listeners(void);
193 void uh_setup_listeners(void);
194 int uh_socket_bind(const char *host, const char *port, bool tls);
195
196 bool uh_use_chunked(struct client *cl);
197 void uh_chunk_write(struct client *cl, const void *data, int len);
198 void uh_chunk_vprintf(struct client *cl, const char *format, va_list arg);
199
200 void __printf(2, 3)
201 uh_chunk_printf(struct client *cl, const char *format, ...);
202
203 void uh_chunk_eof(struct client *cl);
204 void uh_request_done(struct client *cl);
205
206 void uh_http_header(struct client *cl, int code, const char *summary);
207 void __printf(4, 5)
208 uh_client_error(struct client *cl, int code, const char *summary, const char *fmt, ...);
209
210 void uh_handle_request(struct client *cl);
211
212 void uh_auth_add(const char *path, const char *user, const char *pass);
213 bool uh_auth_check(struct client *cl, struct path_info *pi);
214
215 void uh_close_listen_fds(void);
216 void uh_close_fds(void);
217
218 void uh_interpreter_add(const char *ext, const char *path);
219 void uh_dispatch_add(struct dispatch_handler *d);
220
221 void uh_relay_open(struct client *cl, struct relay *r, int fd, int pid);
222 void uh_relay_close(struct relay *r, int ret);
223 void uh_relay_free(struct relay *r);
224
225 struct env_var *uh_get_process_vars(struct client *cl, struct path_info *pi);
226 bool uh_create_process(struct client *cl, struct path_info *pi,
227 void (*cb)(struct client *cl, struct path_info *pi, int fd));
228
229 #endif