2013-06-17 23:39:17 +02:00
|
|
|
/*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <sys/param.h>
|
|
|
|
#include <sys/event.h>
|
2013-07-06 21:05:17 +02:00
|
|
|
|
2013-07-28 19:21:49 +02:00
|
|
|
#ifdef __MACH__
|
|
|
|
#include <sys/sysctl.h>
|
|
|
|
#endif
|
|
|
|
|
2013-06-17 23:39:17 +02:00
|
|
|
#include "kore.h"
|
|
|
|
|
2014-04-02 00:06:24 +02:00
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
|
|
#include "contrib/postgres/kore_pgsql.h"
|
|
|
|
#endif
|
|
|
|
|
2013-06-17 23:39:17 +02:00
|
|
|
static int kfd = -1;
|
|
|
|
static struct kevent *events;
|
2013-07-06 21:05:17 +02:00
|
|
|
static u_int32_t nchanges;
|
2013-06-17 23:39:17 +02:00
|
|
|
static struct kevent *changelist;
|
2013-06-27 12:37:14 +02:00
|
|
|
static u_int32_t event_count = 0;
|
2013-06-17 23:39:17 +02:00
|
|
|
|
|
|
|
void
|
2013-06-24 09:36:40 +02:00
|
|
|
kore_platform_init(void)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
2013-07-28 19:21:49 +02:00
|
|
|
#ifndef __MACH__
|
2013-06-17 23:39:17 +02:00
|
|
|
cpu_count = 0;
|
2013-07-28 19:21:49 +02:00
|
|
|
#else
|
|
|
|
long n;
|
|
|
|
size_t len = sizeof(n);
|
|
|
|
int mib[] = { CTL_HW, HW_AVAILCPU };
|
|
|
|
|
|
|
|
sysctl(mib, 2, &n, &len, NULL, 0);
|
|
|
|
if (n < 1) {
|
|
|
|
mib[1] = HW_NCPU;
|
|
|
|
sysctl(mib, 2, &n, &len, NULL, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (n >= 1)
|
|
|
|
cpu_count = (u_int16_t)n;
|
|
|
|
#endif /* !__MACH__ */
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_platform_worker_setcpu(struct kore_worker *kw)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_platform_event_init(void)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
2013-07-27 20:56:15 +02:00
|
|
|
struct listener *l;
|
|
|
|
|
2013-06-17 23:39:17 +02:00
|
|
|
if ((kfd = kqueue()) == -1)
|
|
|
|
fatal("kqueue(): %s", errno_s);
|
|
|
|
|
|
|
|
nchanges = 0;
|
2013-07-27 20:56:15 +02:00
|
|
|
event_count = worker_max_connections + nlisteners;
|
2013-06-27 12:37:14 +02:00
|
|
|
events = kore_calloc(event_count, sizeof(struct kevent));
|
|
|
|
changelist = kore_calloc(event_count, sizeof(struct kevent));
|
2013-06-26 16:37:22 +02:00
|
|
|
|
2013-07-27 20:56:15 +02:00
|
|
|
LIST_FOREACH(l, &listeners, list) {
|
|
|
|
kore_platform_event_schedule(l->fd,
|
|
|
|
EVFILT_READ, EV_ADD | EV_DISABLE, l);
|
|
|
|
}
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|
|
|
|
|
2013-07-06 21:05:17 +02:00
|
|
|
void
|
2013-06-26 16:37:22 +02:00
|
|
|
kore_platform_event_wait(void)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
2013-07-27 20:56:15 +02:00
|
|
|
struct listener *l;
|
2013-06-17 23:39:17 +02:00
|
|
|
struct connection *c;
|
2013-07-27 20:56:15 +02:00
|
|
|
u_int8_t type;
|
2013-06-26 16:37:22 +02:00
|
|
|
struct timespec timeo;
|
2013-07-27 20:56:15 +02:00
|
|
|
int n, i;
|
2013-06-17 23:39:17 +02:00
|
|
|
|
2013-06-26 16:37:22 +02:00
|
|
|
timeo.tv_sec = 0;
|
|
|
|
timeo.tv_nsec = 100000000;
|
2013-06-27 12:37:14 +02:00
|
|
|
n = kevent(kfd, changelist, nchanges, events, event_count, &timeo);
|
2013-06-17 23:39:17 +02:00
|
|
|
if (n == -1) {
|
|
|
|
if (errno == EINTR)
|
2013-07-06 21:05:17 +02:00
|
|
|
return;
|
2013-06-17 23:39:17 +02:00
|
|
|
fatal("kevent(): %s", errno_s);
|
|
|
|
}
|
|
|
|
|
|
|
|
nchanges = 0;
|
|
|
|
if (n > 0)
|
|
|
|
kore_debug("main(): %d sockets available", n);
|
|
|
|
|
|
|
|
for (i = 0; i < n; i++) {
|
2013-07-27 20:56:15 +02:00
|
|
|
if (events[i].udata == NULL)
|
|
|
|
fatal("events[%d].udata == NULL", i);
|
|
|
|
|
|
|
|
type = *(u_int8_t *)events[i].udata;
|
2013-06-17 23:39:17 +02:00
|
|
|
|
|
|
|
if (events[i].flags & EV_EOF ||
|
|
|
|
events[i].flags & EV_ERROR) {
|
2013-07-27 20:56:15 +02:00
|
|
|
if (type == KORE_TYPE_LISTENER)
|
2013-06-17 23:39:17 +02:00
|
|
|
fatal("error on server socket");
|
|
|
|
|
2014-04-02 00:06:24 +02:00
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
|
|
if (type == KORE_TYPE_PGSQL_CONN) {
|
|
|
|
kore_pgsql_handle(events[i].udata, 1);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2013-06-17 23:39:17 +02:00
|
|
|
c = (struct connection *)events[i].udata;
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_connection_disconnect(c);
|
2013-06-17 23:39:17 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2014-04-02 00:06:24 +02:00
|
|
|
switch (type) {
|
|
|
|
case KORE_TYPE_LISTENER:
|
2013-07-27 20:56:15 +02:00
|
|
|
l = (struct listener *)events[i].udata;
|
|
|
|
|
|
|
|
while ((worker->accepted < worker->accept_treshold) &&
|
|
|
|
(worker_active_connections <
|
|
|
|
worker_max_connections)) {
|
|
|
|
kore_connection_accept(l, &c);
|
2013-06-27 00:22:48 +02:00
|
|
|
if (c == NULL)
|
2013-07-13 22:24:00 +02:00
|
|
|
break;
|
2013-06-27 00:22:48 +02:00
|
|
|
|
2013-06-27 12:37:14 +02:00
|
|
|
worker->accepted++;
|
2013-06-27 00:22:48 +02:00
|
|
|
kore_platform_event_schedule(c->fd,
|
|
|
|
EVFILT_READ, EV_ADD, c);
|
|
|
|
kore_platform_event_schedule(c->fd,
|
|
|
|
EVFILT_WRITE, EV_ADD | EV_ONESHOT, c);
|
|
|
|
}
|
2014-04-02 00:06:24 +02:00
|
|
|
break;
|
|
|
|
case KORE_TYPE_CONNECTION:
|
2013-06-17 23:39:17 +02:00
|
|
|
c = (struct connection *)events[i].udata;
|
2013-07-13 19:56:38 +02:00
|
|
|
if (events[i].filter == EVFILT_READ &&
|
|
|
|
!(c->flags & CONN_READ_BLOCK))
|
2013-06-17 23:39:17 +02:00
|
|
|
c->flags |= CONN_READ_POSSIBLE;
|
2013-07-13 19:56:38 +02:00
|
|
|
if (events[i].filter == EVFILT_WRITE &&
|
|
|
|
!(c->flags & CONN_WRITE_BLOCK))
|
2013-06-17 23:39:17 +02:00
|
|
|
c->flags |= CONN_WRITE_POSSIBLE;
|
|
|
|
|
|
|
|
if (!kore_connection_handle(c)) {
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_connection_disconnect(c);
|
2013-06-17 23:39:17 +02:00
|
|
|
} else {
|
|
|
|
if (!TAILQ_EMPTY(&(c->send_queue))) {
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_platform_event_schedule(c->fd,
|
|
|
|
EVFILT_WRITE, EV_ADD | EV_ONESHOT,
|
|
|
|
c);
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|
|
|
|
}
|
2014-04-02 00:06:24 +02:00
|
|
|
break;
|
|
|
|
#if defined(KORE_USE_PGSQL)
|
|
|
|
case KORE_TYPE_PGSQL_CONN:
|
|
|
|
kore_pgsql_handle(events[i].udata, 0);
|
|
|
|
break;
|
|
|
|
#endif
|
|
|
|
default:
|
|
|
|
fatal("wrong type in event %d", type);
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_platform_event_schedule(int fd, int type, int flags, void *data)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
2013-07-06 21:05:17 +02:00
|
|
|
if (nchanges >= event_count) {
|
2013-06-20 23:29:26 +02:00
|
|
|
kore_log(LOG_WARNING, "cannot schedule %d (%d) on %d",
|
|
|
|
type, flags, fd);
|
|
|
|
} else {
|
|
|
|
EV_SET(&changelist[nchanges], fd, type, flags, 0, 0, data);
|
|
|
|
nchanges++;
|
|
|
|
}
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|
|
|
|
|
2013-06-26 16:37:22 +02:00
|
|
|
void
|
|
|
|
kore_platform_enable_accept(void)
|
|
|
|
{
|
2013-07-27 20:56:15 +02:00
|
|
|
struct listener *l;
|
|
|
|
|
|
|
|
LIST_FOREACH(l, &listeners, list)
|
|
|
|
kore_platform_event_schedule(l->fd, EVFILT_READ, EV_ENABLE, l);
|
2013-06-26 16:37:22 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
kore_platform_disable_accept(void)
|
|
|
|
{
|
2013-07-27 20:56:15 +02:00
|
|
|
struct listener *l;
|
|
|
|
|
|
|
|
LIST_FOREACH(l, &listeners, list)
|
|
|
|
kore_platform_event_schedule(l->fd, EVFILT_READ, EV_DISABLE, l);
|
2013-06-26 16:37:22 +02:00
|
|
|
}
|
|
|
|
|
2014-04-02 00:06:24 +02:00
|
|
|
void
|
|
|
|
kore_platform_schedule_read(int fd, void *data)
|
|
|
|
{
|
|
|
|
kore_platform_event_schedule(fd, EVFILT_READ, EV_ADD, data);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
kore_platform_disable_read(int fd)
|
|
|
|
{
|
|
|
|
kore_platform_event_schedule(fd, EVFILT_READ, EV_DELETE, NULL);
|
|
|
|
}
|
|
|
|
|
2013-06-17 23:39:17 +02:00
|
|
|
void
|
2013-06-26 11:18:32 +02:00
|
|
|
kore_platform_proctitle(char *title)
|
2013-06-17 23:39:17 +02:00
|
|
|
{
|
2013-07-17 20:17:00 +02:00
|
|
|
#ifndef __MACH__
|
2013-06-17 23:39:17 +02:00
|
|
|
setproctitle("%s", title);
|
2013-07-17 20:17:00 +02:00
|
|
|
#endif
|
2013-06-17 23:39:17 +02:00
|
|
|
}
|