kernel-pfroute: use watcher to receive kernel events

This commit is contained in:
Martin Willi
2013-07-18 16:00:30 +02:00
parent 7f698daef9
commit 46666dd3c1
@@ -866,7 +866,8 @@ static void process_route(private_kernel_pfroute_net_t *this,
/** /**
* Receives PF_ROUTE messages from kernel * Receives PF_ROUTE messages from kernel
*/ */
static job_requeue_t receive_events(private_kernel_pfroute_net_t *this) static bool receive_events(private_kernel_pfroute_net_t *this, int fd,
watcher_event_t event)
{ {
struct { struct {
union { union {
@@ -877,36 +878,32 @@ static job_requeue_t receive_events(private_kernel_pfroute_net_t *this)
char buf[sizeof(struct sockaddr_storage) * RTAX_MAX]; char buf[sizeof(struct sockaddr_storage) * RTAX_MAX];
} msg; } msg;
int len, hdrlen; int len, hdrlen;
bool oldstate;
oldstate = thread_cancelability(TRUE);
len = recv(this->socket, &msg, sizeof(msg), 0);
thread_cancelability(oldstate);
len = recv(this->socket, &msg, sizeof(msg), MSG_DONTWAIT);
if (len < 0) if (len < 0)
{ {
switch (errno) switch (errno)
{ {
case EINTR: case EINTR:
case EAGAIN: case EAGAIN:
return JOB_REQUEUE_DIRECT; return TRUE;
default: default:
DBG1(DBG_KNL, "unable to receive from PF_ROUTE event socket"); DBG1(DBG_KNL, "unable to receive from PF_ROUTE event socket");
sleep(1); sleep(1);
return JOB_REQUEUE_FAIR; return TRUE;
} }
} }
if (len < offsetof(struct rt_msghdr, rtm_flags) || len < msg.rtm.rtm_msglen) if (len < offsetof(struct rt_msghdr, rtm_flags) || len < msg.rtm.rtm_msglen)
{ {
DBG1(DBG_KNL, "received invalid PF_ROUTE message"); DBG1(DBG_KNL, "received invalid PF_ROUTE message");
return JOB_REQUEUE_DIRECT; return TRUE;
} }
if (msg.rtm.rtm_version != RTM_VERSION) if (msg.rtm.rtm_version != RTM_VERSION)
{ {
DBG1(DBG_KNL, "received PF_ROUTE message with unsupported version: %d", DBG1(DBG_KNL, "received PF_ROUTE message with unsupported version: %d",
msg.rtm.rtm_version); msg.rtm.rtm_version);
return JOB_REQUEUE_DIRECT; return TRUE;
} }
switch (msg.rtm.rtm_type) switch (msg.rtm.rtm_type)
{ {
@@ -923,12 +920,12 @@ static job_requeue_t receive_events(private_kernel_pfroute_net_t *this)
hdrlen = sizeof(msg.rtm); hdrlen = sizeof(msg.rtm);
break; break;
default: default:
return JOB_REQUEUE_DIRECT; return TRUE;
} }
if (msg.rtm.rtm_msglen < hdrlen) if (msg.rtm.rtm_msglen < hdrlen)
{ {
DBG1(DBG_KNL, "ignoring short PF_ROUTE message"); DBG1(DBG_KNL, "ignoring short PF_ROUTE message");
return JOB_REQUEUE_DIRECT; return TRUE;
} }
switch (msg.rtm.rtm_type) switch (msg.rtm.rtm_type)
{ {
@@ -958,7 +955,7 @@ static job_requeue_t receive_events(private_kernel_pfroute_net_t *this)
this->condvar->broadcast(this->condvar); this->condvar->broadcast(this->condvar);
this->mutex->unlock(this->mutex); this->mutex->unlock(this->mutex);
return JOB_REQUEUE_DIRECT; return TRUE;
} }
@@ -1699,6 +1696,7 @@ METHOD(kernel_net_t, destroy, void,
if (this->socket != -1) if (this->socket != -1)
{ {
lib->watcher->remove(lib->watcher, this->socket);
close(this->socket); close(this->socket);
} }
@@ -1786,10 +1784,8 @@ kernel_pfroute_net_t *kernel_pfroute_net_create()
} }
else else
{ {
lib->processor->queue_job(lib->processor, lib->watcher->add(lib->watcher, this->socket, WATCHER_READ,
(job_t*)callback_job_create_with_prio( (watcher_cb_t)receive_events, this);
(callback_job_cb_t)receive_events, this, NULL,
(callback_job_cancel_t)return_false, JOB_PRIO_CRITICAL));
} }
if (init_address_list(this) != SUCCESS) if (init_address_list(this) != SUCCESS)
{ {