mirror of
https://github.com/jorisvink/kore
synced 2025-03-12 05:49:02 -04:00
0dda6f996f
new connections and which ones will not be notified for it. Fixes the thundering herd problem, and nicely spreads out load between all the workers equally. A configuration option (workers_max_connections) is available to tweak how many connections a worker will have before giving up the accept lock. Two ways are added to this commit for access locking: - Locking via semaphores. - Locking via GCC's builtin atomic methods. The default is running with semaphores disabled (OpenBSD cannot do sem_init() with pshared set to 1, which is required). If you want to use semaphores add KORE_USE_SEMAPHORES to CFLAGS, and -lpthread to LDFLAGS in the Makefile. Other fixes: - BSD: add a timeout to kevent(). - Merge kore_worker_wait together, linux knows waitpid() as well. - Send the correct SIGQUIT signal to workers instead of SIGINT. - Fix kore_time_ms(). - Log fatal worker messages in syslog. - Refactor code even more. - Do not free our own kore_worker structure.
258 lines
7.3 KiB
C
258 lines
7.3 KiB
C
/*
|
|
* Copyright (c) 2013 Joris Vink <joris@coders.se>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
#ifndef __H_KORE_H
|
|
#define __H_KORE_H
|
|
|
|
#define KORE_RESULT_ERROR 0
|
|
#define KORE_RESULT_OK 1
|
|
#define KORE_RESULT_RETRY 2
|
|
|
|
#define errno_s strerror(errno)
|
|
#define ssl_errno_s ERR_error_string(ERR_get_error(), NULL)
|
|
|
|
#define KORE_DOMAINNAME_LEN 254
|
|
#define KORE_PIDFILE_DEFAULT "/var/run/kore.pid"
|
|
|
|
#define kore_debug(fmt, ...) \
|
|
if (kore_debug) \
|
|
kore_debug_internal(__FILE__, __LINE__, fmt, ##__VA_ARGS__)
|
|
|
|
#define NETBUF_RECV 0
|
|
#define NETBUF_SEND 1
|
|
|
|
#define NETBUF_CALL_CB_ALWAYS 0x01
|
|
#define NETBUF_FORCE_REMOVE 0x02
|
|
#define NETBUF_RETAIN 0x04
|
|
|
|
struct netbuf {
|
|
u_int8_t *buf;
|
|
u_int32_t offset;
|
|
u_int32_t len;
|
|
u_int8_t type;
|
|
u_int8_t flags;
|
|
|
|
void *owner;
|
|
void *extra;
|
|
int (*cb)(struct netbuf *);
|
|
|
|
TAILQ_ENTRY(netbuf) list;
|
|
};
|
|
|
|
struct listener {
|
|
int fd;
|
|
struct sockaddr_in sin;
|
|
};
|
|
|
|
#define CONN_STATE_UNKNOWN 0
|
|
#define CONN_STATE_SSL_SHAKE 1
|
|
#define CONN_STATE_ESTABLISHED 2
|
|
#define CONN_STATE_DISCONNECTING 3
|
|
|
|
#define CONN_PROTO_UNKNOWN 0
|
|
#define CONN_PROTO_SPDY 1
|
|
#define CONN_PROTO_HTTP 2
|
|
|
|
#define CONN_READ_POSSIBLE 0x01
|
|
#define CONN_WRITE_POSSIBLE 0x02
|
|
|
|
struct connection {
|
|
int fd;
|
|
int state;
|
|
int proto;
|
|
struct sockaddr_in sin;
|
|
void *owner;
|
|
SSL *ssl;
|
|
int flags;
|
|
|
|
u_int8_t inflate_started;
|
|
z_stream z_inflate;
|
|
u_int8_t deflate_started;
|
|
z_stream z_deflate;
|
|
|
|
TAILQ_HEAD(, netbuf) send_queue;
|
|
TAILQ_HEAD(, netbuf) recv_queue;
|
|
|
|
u_int32_t client_stream_id;
|
|
TAILQ_HEAD(, spdy_stream) spdy_streams;
|
|
|
|
TAILQ_ENTRY(connection) list;
|
|
};
|
|
|
|
#define HANDLER_TYPE_STATIC 1
|
|
#define HANDLER_TYPE_DYNAMIC 2
|
|
|
|
struct kore_module_handle {
|
|
char *path;
|
|
char *func;
|
|
void *addr;
|
|
int type;
|
|
regex_t rctx;
|
|
|
|
TAILQ_ENTRY(kore_module_handle) list;
|
|
};
|
|
|
|
struct kore_worker {
|
|
u_int16_t id;
|
|
u_int16_t cpu;
|
|
pid_t pid;
|
|
TAILQ_ENTRY(kore_worker) list;
|
|
};
|
|
|
|
struct kore_domain {
|
|
char *domain;
|
|
char *certfile;
|
|
char *certkey;
|
|
int accesslog;
|
|
SSL_CTX *ssl_ctx;
|
|
TAILQ_HEAD(, kore_module_handle) handlers;
|
|
TAILQ_ENTRY(kore_domain) list;
|
|
};
|
|
|
|
TAILQ_HEAD(kore_domain_h, kore_domain);
|
|
|
|
#define KORE_BUF_INITIAL 128
|
|
#define KORE_BUF_INCREMENT KORE_BUF_INITIAL
|
|
|
|
struct kore_buf {
|
|
u_int8_t *data;
|
|
u_int32_t length;
|
|
u_int32_t offset;
|
|
};
|
|
|
|
struct buf_vec {
|
|
u_int8_t *data;
|
|
u_int32_t length;
|
|
};
|
|
|
|
extern pid_t kore_pid;
|
|
extern int kore_debug;
|
|
extern int server_port;
|
|
extern char *server_ip;
|
|
extern char *chroot_path;
|
|
extern char *runas_user;
|
|
extern char *kore_module_onload;
|
|
extern char *kore_pidfile;
|
|
extern char *config_file;
|
|
|
|
extern u_int16_t cpu_count;
|
|
extern u_int8_t worker_count;
|
|
extern u_int32_t worker_max_connections;
|
|
|
|
extern struct listener server;
|
|
extern struct kore_worker *worker;
|
|
extern struct kore_domain_h domains;
|
|
extern struct kore_domain *primary_dom;
|
|
extern struct passwd *pw;
|
|
|
|
void kore_signal(int);
|
|
void kore_worker_wait(int);
|
|
void kore_worker_init(void);
|
|
void kore_worker_shutdown(void);
|
|
void kore_worker_dispatch_signal(int);
|
|
void kore_worker_connection_add(struct connection *);
|
|
void kore_worker_connection_move(struct connection *);
|
|
void kore_worker_connection_remove(struct connection *);
|
|
|
|
void kore_platform_event_init(void);
|
|
void kore_platform_event_wait(void);
|
|
void kore_platform_proctitle(char *);
|
|
void kore_platform_enable_accept(void);
|
|
void kore_platform_disable_accept(void);
|
|
void kore_platform_event_schedule(int, int, int, void *);
|
|
void kore_platform_worker_setcpu(struct kore_worker *);
|
|
|
|
void kore_platform_init(void);
|
|
void kore_accesslog_init(void);
|
|
int kore_accesslog_wait(void);
|
|
void kore_worker_spawn(u_int16_t);
|
|
void kore_accesslog_worker_init(void);
|
|
void kore_worker_entry(struct kore_worker *);
|
|
int kore_ssl_sni_cb(SSL *, int *, void *);
|
|
int kore_ssl_npn_cb(SSL *, const u_char **, unsigned int *, void *);
|
|
|
|
int kore_connection_nonblock(int);
|
|
int kore_connection_handle(struct connection *);
|
|
void kore_connection_remove(struct connection *);
|
|
void kore_connection_disconnect(struct connection *);
|
|
int kore_connection_accept(struct listener *, struct connection **);
|
|
|
|
u_int64_t kore_time_ms(void);
|
|
void kore_log_init(void);
|
|
void *kore_malloc(size_t);
|
|
void kore_parse_config(void);
|
|
void *kore_calloc(size_t, size_t);
|
|
void *kore_realloc(void *, size_t);
|
|
time_t kore_date_to_time(char *);
|
|
char *kore_time_to_date(time_t);
|
|
char *kore_strdup(const char *);
|
|
void kore_log(int, const char *, ...);
|
|
void kore_strlcpy(char *, const char *, size_t);
|
|
void kore_server_disconnect(struct connection *);
|
|
int kore_split_string(char *, char *, char **, size_t);
|
|
long long kore_strtonum(const char *, long long, long long, int *);
|
|
|
|
void kore_domain_init(void);
|
|
int kore_domain_new(char *);
|
|
void kore_module_load(char *);
|
|
void kore_module_reload(void);
|
|
int kore_module_loaded(void);
|
|
void kore_domain_closelogs(void);
|
|
void *kore_module_handler_find(char *, char *);
|
|
void kore_domain_sslstart(struct kore_domain *);
|
|
int kore_module_handler_new(char *, char *, char *, int);
|
|
struct kore_domain *kore_domain_lookup(const char *);
|
|
|
|
void fatal(const char *, ...);
|
|
void kore_debug_internal(char *, int, const char *, ...);
|
|
|
|
u_int16_t net_read16(u_int8_t *);
|
|
u_int32_t net_read32(u_int8_t *);
|
|
void net_write16(u_int8_t *, u_int16_t);
|
|
void net_write32(u_int8_t *, u_int32_t);
|
|
int net_recv(struct connection *);
|
|
int net_send(struct connection *);
|
|
int net_send_flush(struct connection *);
|
|
int net_recv_flush(struct connection *);
|
|
void net_recv_queue(struct connection *, size_t, int,
|
|
struct netbuf **, int (*cb)(struct netbuf *));
|
|
int net_recv_expand(struct connection *c, struct netbuf *, size_t,
|
|
int (*cb)(struct netbuf *));
|
|
void net_send_queue(struct connection *, u_int8_t *, size_t, int,
|
|
struct netbuf **, int (*cb)(struct netbuf *));
|
|
|
|
struct kore_buf *kore_buf_create(u_int32_t);
|
|
void kore_buf_append(struct kore_buf *, u_int8_t *, u_int32_t);
|
|
u_int8_t *kore_buf_release(struct kore_buf *, u_int32_t *);
|
|
void kore_buf_appendf(struct kore_buf *, const char *, ...);
|
|
void kore_buf_appendv(struct kore_buf *, struct buf_vec *, u_int16_t);
|
|
void kore_buf_appendb(struct kore_buf *, struct kore_buf *);
|
|
|
|
struct spdy_header_block *spdy_header_block_create(int);
|
|
struct spdy_stream *spdy_stream_lookup(struct connection *, u_int32_t);
|
|
int spdy_stream_get_header(struct spdy_header_block *,
|
|
char *, char **);
|
|
|
|
int spdy_frame_recv(struct netbuf *);
|
|
void spdy_frame_send(struct connection *, u_int16_t,
|
|
u_int8_t, u_int32_t, struct spdy_stream *, u_int32_t);
|
|
void spdy_header_block_add(struct spdy_header_block *,
|
|
char *, char *);
|
|
u_int8_t *spdy_header_block_release(struct connection *,
|
|
struct spdy_header_block *, u_int32_t *);
|
|
|
|
#endif /* !__H_KORE_H */
|