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 <sys/socket.h>
21 #include <sys/types.h>
23 #include <arpa/inet.h>
24 #include <netinet/in.h>
41 #define GEMINI_URL_LEN (1024+3) /* URL max len + \r\n + \0 */
44 #define TEMP_REDIRECT 30
45 #define TEMP_FAILURE 40
48 #define PROXY_REFUSED 53
49 #define BAD_REQUEST 59
56 /* maximum hostname and label length, +1 for the NUL-terminator */
57 #define DOMAIN_NAME_LEN (253+1)
58 #define LABEL_LEN (63+1)
60 #define LOGE(c, fmt, ...) logs(LOG_ERR, c, fmt, __VA_ARGS__)
61 #define LOGW(c, fmt, ...) logs(LOG_WARNING, c, fmt, __VA_ARGS__)
62 #define LOGN(c, fmt, ...) logs(LOG_NOTICE, c, fmt, __VA_ARGS__)
63 #define LOGI(c, fmt, ...) logs(LOG_INFO, c, fmt, __VA_ARGS__)
64 #define LOGD(c, fmt, ...) logs(LOG_DEBUG, c, fmt, __VA_ARGS__)
69 const char *default_mime;
71 int auto_index; /* 0 auto, -1 off, 1 on */
73 const char *block_fmt;
83 const char *entrypoint;
86 /* the first location rule is always '*' and holds the default
87 * settings for the vhost, from locations[1] onwards there are
88 * the "real" location rules specified in the configuration. */
89 struct location locations[LOCLEN];
92 extern struct vhost hosts[HOSTSLEN];
94 struct etm { /* extension to mime */
115 extern struct conf conf;
118 extern volatile sig_atomic_t hupped;
138 typedef void (*statefn)(struct pollfd*, struct client*);
141 * DFA: handle_handshake is the initial state, close_conn the final.
142 * Sometimes we have an enter_* function to handle the state switch.
144 * handle_handshake -> handle_open_conn
145 * handle_handshake -> close_conn // on err
147 * handle_open_conn -> handle_cgi_reply // via open_file/dir/...
148 * handle_open_conn -> handle_dirlist // ...same
149 * handle_open_conn -> send_file // ...same
150 * handle_open_conn -> start_reply // on error
152 * handle_cgi_reply -> handle_cgi // after logging the CGI reply
153 * handle_cgi_reply -> start_reply // on error
155 * handle_cgi -> close_conn
157 * handle_dirlist -> send_directory_listing
158 * handle_dirlist -> close_conn // on error
160 * send_directory_listing -> close_conn
162 * send_file -> close_conn
166 char req[GEMINI_URL_LEN];
168 char domain[DOMAIN_NAME_LEN];
172 int fd, waiting_on_child;
173 char sbuf[1024]; /* static buffer */
174 void *buf, *i; /* mmap buffer */
175 ssize_t len, off; /* mmap/static buffer */
177 struct sockaddr_storage addr;
178 struct vhost *host; /* host she's talking to */
189 __attribute__((format (printf, 1, 2)))
190 __attribute__((__noreturn__))
191 void fatal(const char*, ...);
193 __attribute__((format (printf, 3, 4)))
194 void logs(int, struct client*, const char*, ...);
195 void log_request(struct client*, char*, size_t);
197 void sig_handler(int);
198 void gen_certificate(const char*, const char*, const char*);
199 void mkdirs(const char*);
200 char *data_dir(void);
201 void load_local_cert(const char*, const char*);
202 void load_vhosts(void);
203 int make_socket(int, int);
204 void setup_tls(void);
205 void init_config(void);
206 void free_config(void);
207 void drop_priv(void);
208 void usage(const char*);
210 /* provided by lex/yacc */
213 extern int yyparse(void);
214 extern int yylex(void);
216 void yyerror(const char*, ...);
217 int parse_portno(const char*);
218 void parse_conf(const char*);
221 void init_mime(struct mime*);
222 void add_mime(struct mime*, const char*, const char*);
223 void load_default_mime(struct mime*);
224 const char *mime(struct vhost*, const char*);
227 const char *vhost_lang(struct vhost*, const char*);
228 const char *vhost_default_mime(struct vhost*, const char*);
229 const char *vhost_index(struct vhost*, const char*);
230 int vhost_auto_index(struct vhost*, const char*);
231 int vhost_block_return(struct vhost*, const char*, int*, const char**);
232 int vhost_strip(struct vhost*, const char*);
233 void mark_nonblock(int);
234 void loop(struct tls*, int, int);
237 int send_string(int, const char*);
238 int recv_string(int, char**);
239 int send_iri(int, struct iri*);
240 int recv_iri(int, struct iri*);
241 void free_recvd_iri(struct iri*);
242 int send_vhost(int, struct vhost*);
243 int recv_vhost(int, struct vhost**);
244 int send_fd(int, int);
246 int executor_main(void);
252 int valid_multibyte_utf8(struct parser*);
253 char *utf8_nth(char*, size_t);
256 int parse_iri(char*, struct iri*, const char**);
257 int trim_req_iri(char*, const char **);
258 int serialize_iri(struct iri*, char*, size_t);
261 int puny_decode(const char*, char*, size_t, const char**);
264 void block_signals(void);
265 void unblock_signals(void);
266 int starts_with(const char*, const char*);
267 int ends_with(const char*, const char*);
268 ssize_t filesize(int);
269 char *absolutify_path(const char*);
270 char *xstrdup(const char*);