kore/src/bsd.c

277 lines
5.8 KiB
C
Raw Normal View History

/*
2016-01-04 12:58:51 +01:00
* Copyright (c) 2013-2016 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>
#include <sys/sysctl.h>
#if defined(__FreeBSD_version)
#include <sys/cpuset.h>
#endif
2016-01-22 12:08:13 +01:00
#include <errno.h>
#include <string.h>
#include "kore.h"
2014-04-02 00:06:24 +02:00
#if defined(KORE_USE_PGSQL)
2014-07-03 22:14:46 +02:00
#include "pgsql.h"
2014-04-02 00:06:24 +02:00
#endif
2014-06-30 14:43:34 +02:00
#if defined(KORE_USE_TASKS)
2014-07-03 22:14:46 +02:00
#include "tasks.h"
2014-06-30 14:43:34 +02:00
#endif
static int kfd = -1;
static struct kevent *events;
2013-06-27 12:37:14 +02:00
static u_int32_t event_count = 0;
void
2013-06-24 09:36:40 +02:00
kore_platform_init(void)
{
#if defined(__MACH__) || defined(__FreeBSD_version)
long n;
size_t len = sizeof(n);
int mib[] = { CTL_HW, HW_NCPU };
if (sysctl(mib, 2, &n, &len, NULL, 0) == -1) {
kore_debug("kore_platform_init(): sysctl %s", errno_s);
cpu_count = 1;
} else {
cpu_count = (u_int16_t)n;
}
#else
cpu_count = 0;
#endif /* __MACH__ || __FreeBSD_version */
}
void
2013-06-26 11:18:32 +02:00
kore_platform_worker_setcpu(struct kore_worker *kw)
{
#if defined(__FreeBSD_version)
cpuset_t cpuset;
CPU_ZERO(&cpuset);
CPU_SET(kw->cpu, &cpuset);
if (cpuset_setaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID,
-1, sizeof(cpuset), &cpuset) == -1) {
fatal("failed: %s", errno_s);
}
#endif /* __FreeBSD_version */
}
void
2013-06-26 11:18:32 +02:00
kore_platform_event_init(void)
{
struct listener *l;
if ((kfd = kqueue()) == -1)
fatal("kqueue(): %s", errno_s);
event_count = (worker_max_connections * 2) + nlisteners;
2013-06-27 12:37:14 +02:00
events = kore_calloc(event_count, sizeof(struct kevent));
/* Hack to check if we're running under the parent or not. */
if (worker != NULL) {
LIST_FOREACH(l, &listeners, list) {
kore_platform_event_schedule(l->fd,
EVFILT_READ, EV_ADD | EV_DISABLE, l);
}
}
}
void
kore_platform_event_cleanup(void)
{
if (kfd >= 0) {
close(kfd);
kfd = -1;
}
if (events != NULL) {
kore_mem_free(events);
events = NULL;
}
}
int
2014-07-28 23:35:12 +02:00
kore_platform_event_wait(u_int64_t timer)
{
u_int32_t r;
struct listener *l;
struct connection *c;
u_int8_t type;
struct timespec timeo;
int n, i;
timeo.tv_sec = timer / 1000;
timeo.tv_nsec = (timer % 1000) * 1000000;
n = kevent(kfd, NULL, 0, events, event_count, &timeo);
if (n == -1) {
if (errno == EINTR)
return (0);
fatal("kevent(): %s", errno_s);
}
if (n > 0)
kore_debug("main(): %d sockets available", n);
r = 0;
for (i = 0; i < n; i++) {
if (events[i].udata == NULL)
fatal("events[%d].udata == NULL", i);
type = *(u_int8_t *)events[i].udata;
if (events[i].flags & EV_EOF ||
events[i].flags & EV_ERROR) {
2014-06-30 14:43:34 +02:00
switch (type) {
case KORE_TYPE_LISTENER:
fatal("error on server socket");
2014-06-30 14:43:34 +02:00
/* NOTREACHED */
2014-04-02 00:06:24 +02:00
#if defined(KORE_USE_PGSQL)
2014-06-30 14:43:34 +02:00
case KORE_TYPE_PGSQL_CONN:
2014-04-02 00:06:24 +02:00
kore_pgsql_handle(events[i].udata, 1);
2014-06-30 14:43:34 +02:00
break;
#endif
#if defined(KORE_USE_TASKS)
case KORE_TYPE_TASK:
kore_task_handle(events[i].udata, 1);
break;
2014-04-02 00:06:24 +02:00
#endif
2014-06-30 14:43:34 +02:00
default:
c = (struct connection *)events[i].udata;
kore_connection_disconnect(c);
break;
}
2014-04-02 00:06:24 +02:00
continue;
}
2014-04-02 00:06:24 +02:00
switch (type) {
case KORE_TYPE_LISTENER:
l = (struct listener *)events[i].udata;
while (worker_active_connections <
worker_max_connections) {
if (worker_accept_threshold != 0 &&
r >= worker_accept_threshold)
break;
if (!kore_connection_accept(l, &c)) {
r = 1;
break;
}
if (c == NULL)
break;
r++;
kore_platform_event_all(c->fd, c);
}
2014-04-02 00:06:24 +02:00
break;
case KORE_TYPE_CONNECTION:
c = (struct connection *)events[i].udata;
if (events[i].filter == EVFILT_READ &&
!(c->flags & CONN_READ_BLOCK))
c->flags |= CONN_READ_POSSIBLE;
if (events[i].filter == EVFILT_WRITE &&
!(c->flags & CONN_WRITE_BLOCK))
c->flags |= CONN_WRITE_POSSIBLE;
if (c->handle != NULL && !c->handle(c))
2013-06-26 11:18:32 +02:00
kore_connection_disconnect(c);
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;
2014-06-30 14:43:34 +02:00
#endif
#if defined(KORE_USE_TASKS)
case KORE_TYPE_TASK:
kore_task_handle(events[i].udata, 0);
break;
2014-04-02 00:06:24 +02:00
#endif
default:
fatal("wrong type in event %d", type);
}
}
return (r);
}
void
kore_platform_event_all(int fd, void *c)
{
kore_platform_event_schedule(fd, EVFILT_READ, EV_ADD | EV_CLEAR, c);
kore_platform_event_schedule(fd, EVFILT_WRITE, EV_ADD | EV_CLEAR, c);
}
void
2013-06-26 11:18:32 +02:00
kore_platform_event_schedule(int fd, int type, int flags, void *data)
{
struct kevent event[1];
EV_SET(&event[0], fd, type, flags, 0, 0, data);
if (kevent(kfd, event, 1, NULL, 0, NULL) == -1)
fatal("kevent: %s", errno_s);
}
void
kore_platform_enable_accept(void)
{
struct listener *l;
LIST_FOREACH(l, &listeners, list)
kore_platform_event_schedule(l->fd, EVFILT_READ, EV_ENABLE, l);
}
void
kore_platform_disable_accept(void)
{
struct listener *l;
LIST_FOREACH(l, &listeners, list)
kore_platform_event_schedule(l->fd, EVFILT_READ, EV_DISABLE, l);
}
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);
}
2015-12-09 21:29:44 +01:00
void
kore_platform_schedule_write(int fd, void *data)
{
kore_platform_event_schedule(fd, EVFILT_WRITE, EV_ADD, data);
}
2014-04-02 00:06:24 +02:00
void
kore_platform_disable_read(int fd)
{
kore_platform_event_schedule(fd, EVFILT_READ, EV_DELETE, NULL);
}
void
2013-06-26 11:18:32 +02:00
kore_platform_proctitle(char *title)
{
#ifndef __MACH__
setproctitle("%s", title);
#endif
}