kernel-pfroute: use watcher to receive kernel events
This commit is contained in:
@@ -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)
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user