2 * Copyright (c) 2020 Omar Polo <op@omarpolo.com>
4 * Permission to use, copy, modify, and distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
20 #include <arpa/inet.h>
21 #include <netinet/in.h>
22 #include <sys/socket.h>
37 #define GEMINI_URL_LEN (1024+3) /* URL max len + \r\n + \0 */
39 /* large enough to hold a copy of a gemini URL and still have extra room */
43 #define TEMP_REDIRECT 30
44 #define TEMP_FAILURE 40
46 #define PROXY_REFUSED 53
47 #define BAD_REQUEST 59
54 #define LOGE(c, fmt, ...) logs(LOG_ERR, c, fmt, __VA_ARGS__)
55 #define LOGW(c, fmt, ...) logs(LOG_WARNING, c, fmt, __VA_ARGS__)
56 #define LOGN(c, fmt, ...) logs(LOG_NOTICE, c, fmt, __VA_ARGS__)
57 #define LOGI(c, fmt, ...) logs(LOG_INFO, c, fmt, __VA_ARGS__)
58 #define LOGD(c, fmt, ...) logs(LOG_DEBUG, c, fmt, __VA_ARGS__)
74 struct location locations[LOCLEN];
77 extern struct vhost hosts[HOSTSLEN];
79 struct etm { /* extension to mime */
98 extern struct conf conf;
128 char req[GEMINI_URL_LEN];
133 int fd, waiting_on_child;
134 char sbuf[1024]; /* static buffer */
135 void *buf, *i; /* mmap buffer */
136 ssize_t len, off; /* mmap/static buffer */
137 struct sockaddr_storage addr;
138 struct vhost *host; /* host she's talking to */
150 __attribute__((format (printf, 1, 2)))
151 __attribute__((__noreturn__))
152 void fatal(const char*, ...);
154 __attribute__((format (printf, 3, 4)))
155 void logs(int, struct client*, const char*, ...);
156 void log_request(struct client*, char*, size_t);
158 void sig_handler(int);
159 int starts_with(const char*, const char*);
160 int ends_with(const char*, const char*);
161 ssize_t filesize(int);
162 char *absolutify_path(const char*);
163 void yyerror(const char*);
164 int parse_portno(const char*);
165 void parse_conf(const char*);
166 void load_vhosts(struct tls_config*);
167 int make_socket(int, int);
168 int listener_main(void);
169 void init_config(void);
170 void usage(const char*);
172 /* provided by lex/yacc */
175 extern int yyparse(void);
176 extern int yylex(void);
179 void init_mime(struct mime*);
180 void add_mime(struct mime*, const char*, const char*);
181 void load_default_mime(struct mime*);
182 const char *mime(struct vhost*, const char*);
185 const char *vhost_lang(struct vhost*, const char*);
186 const char *vhost_default_mime(struct vhost*, const char*);
187 const char *vhost_index(struct vhost*, const char*);
188 int check_path(struct client*, const char*, int*);
189 void open_file(struct pollfd*, struct client*);
190 void check_for_cgi(char *, char*, struct pollfd*, struct client*);
191 void mark_nonblock(int);
192 void handle_handshake(struct pollfd*, struct client*);
193 void handle_open_conn(struct pollfd*, struct client*);
194 void start_reply(struct pollfd*, struct client*, int, const char*);
195 void start_cgi(const char*, const char*, const char*, struct pollfd*, struct client*);
196 void send_file(struct pollfd*, struct client*);
197 void send_dir(struct pollfd*, struct client*);
198 void cgi_poll_on_child(struct pollfd*, struct client*);
199 void cgi_poll_on_client(struct pollfd*, struct client*);
200 void handle_cgi(struct pollfd*, struct client*);
201 void close_conn(struct pollfd*, struct client*);
202 void do_accept(int, struct tls*, struct pollfd*, struct client*);
203 void handle(struct pollfd*, struct client*);
204 void loop(struct tls*, int, int);
207 int send_string(int, const char*);
208 int recv_string(int, char**);
209 int send_vhost(int, struct vhost*);
210 int recv_vhost(int, struct vhost**);
211 int send_fd(int, int);
213 int executor_main(int);
219 int valid_multibyte_utf8(struct parser*);
222 int parse_iri(char*, struct iri*, const char**);
223 int trim_req_iri(char*);