You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
223 lines
6.9 KiB
223 lines
6.9 KiB
9 years ago
|
/*
|
||
|
|
Copyright (c) 2013 Martin Sustrik All rights reserved.
|
||
|
|
Copyright (c) 2013 GoPivotal, Inc. All rights reserved.
|
||
|
|
|
||
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
of this software and associated documentation files (the "Software"),
|
||
|
|
to deal in the Software without restriction, including without limitation
|
||
|
|
the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||
|
|
and/or sell copies of the Software, and to permit persons to whom
|
||
|
|
the Software is furnished to do so, subject to the following conditions:
|
||
|
|
|
||
|
|
The above copyright notice and this permission notice shall be included
|
||
|
|
in all copies or substantial portions of the Software.
|
||
|
|
|
||
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||
|
|
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||
|
|
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
||
|
|
IN THE SOFTWARE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "ctx.h"
|
||
|
|
#include "usock.h"
|
||
|
|
|
||
|
|
#include "../utils/err.h"
|
||
|
|
#include "../utils/cont.h"
|
||
|
|
#include "../utils/fast.h"
|
||
|
|
|
||
|
|
#define NN_WORKER_MAX_EVENTS 32
|
||
|
|
|
||
|
|
#define NN_WORKER_OP_STATE_IDLE 1
|
||
|
|
#define NN_WORKER_OP_STATE_ACTIVE 2
|
||
|
|
#define NN_WORKER_OP_STATE_ACTIVE_ZEROISERROR 3
|
||
|
|
|
||
|
|
/* The value of this variable is irrelevant. It's used only as a placeholder
|
||
|
|
for the address that is used as the 'stop' event ID. */
|
||
|
|
const int nn_worker_stop = 0;
|
||
|
|
|
||
|
|
/* Private functions. */
|
||
|
|
static void nn_worker_routine (void *arg);
|
||
|
|
|
||
|
|
void nn_worker_task_init (struct nn_worker_task *self, int src,
|
||
|
|
struct nn_fsm *owner)
|
||
|
|
{
|
||
|
|
self->src = src;
|
||
|
|
self->owner = owner;
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_task_term (struct nn_worker_task *self)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_op_init (struct nn_worker_op *self, int src,
|
||
|
|
struct nn_fsm *owner)
|
||
|
|
{
|
||
|
|
self->src = src;
|
||
|
|
self->owner = owner;
|
||
|
|
self->state = NN_WORKER_OP_STATE_IDLE;
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_op_term (struct nn_worker_op *self)
|
||
|
|
{
|
||
|
|
nn_assert_state (self, NN_WORKER_OP_STATE_IDLE);
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_op_start (struct nn_worker_op *self, int zeroiserror)
|
||
|
|
{
|
||
|
|
nn_assert_state (self, NN_WORKER_OP_STATE_IDLE);
|
||
|
|
self->state = zeroiserror ? NN_WORKER_OP_STATE_ACTIVE_ZEROISERROR :
|
||
|
|
NN_WORKER_OP_STATE_ACTIVE;
|
||
|
|
}
|
||
|
|
|
||
|
|
int nn_worker_op_isidle (struct nn_worker_op *self)
|
||
|
|
{
|
||
|
|
return self->state == NN_WORKER_OP_STATE_IDLE ? 1 : 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int nn_worker_init (struct nn_worker *self)
|
||
|
|
{
|
||
|
|
self->cp = CreateIoCompletionPort (INVALID_HANDLE_VALUE, NULL, 0, 0);
|
||
|
|
win_assert (self->cp);
|
||
|
|
nn_timerset_init (&self->timerset);
|
||
|
|
nn_thread_init (&self->thread, nn_worker_routine, self);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_term (struct nn_worker *self)
|
||
|
|
{
|
||
|
|
BOOL brc;
|
||
|
|
|
||
|
|
/* Ask worker thread to terminate. */
|
||
|
|
brc = PostQueuedCompletionStatus (self->cp, 0,
|
||
|
|
(ULONG_PTR) &nn_worker_stop, NULL);
|
||
|
|
win_assert (brc);
|
||
|
|
|
||
|
|
/* Wait till worker thread terminates. */
|
||
|
|
nn_thread_term (&self->thread);
|
||
|
|
|
||
|
|
nn_timerset_term (&self->timerset);
|
||
|
|
brc = CloseHandle (self->cp);
|
||
|
|
win_assert (brc);
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_execute (struct nn_worker *self, struct nn_worker_task *task)
|
||
|
|
{
|
||
|
|
BOOL brc;
|
||
|
|
|
||
|
|
brc = PostQueuedCompletionStatus (self->cp, 0, (ULONG_PTR) task, NULL);
|
||
|
|
win_assert (brc);
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_add_timer (struct nn_worker *self, int timeout,
|
||
|
|
struct nn_worker_timer *timer)
|
||
|
|
{
|
||
|
|
nn_timerset_add (&((struct nn_worker*) self)->timerset, timeout,
|
||
|
|
&timer->hndl);
|
||
|
|
}
|
||
|
|
|
||
|
|
void nn_worker_rm_timer (struct nn_worker *self, struct nn_worker_timer *timer)
|
||
|
|
{
|
||
|
|
nn_timerset_rm (&((struct nn_worker*) self)->timerset, &timer->hndl);
|
||
|
|
}
|
||
|
|
|
||
|
|
HANDLE nn_worker_getcp (struct nn_worker *self)
|
||
|
|
{
|
||
|
|
return self->cp;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void nn_worker_routine (void *arg)
|
||
|
|
{
|
||
|
|
int rc;
|
||
|
|
BOOL brc;
|
||
|
|
struct nn_worker *self;
|
||
|
|
int timeout;
|
||
|
|
ULONG count;
|
||
|
|
ULONG i;
|
||
|
|
struct nn_timerset_hndl *thndl;
|
||
|
|
struct nn_worker_timer *timer;
|
||
|
|
struct nn_worker_task *task;
|
||
|
|
struct nn_worker_op *op;
|
||
|
|
OVERLAPPED_ENTRY entries [NN_WORKER_MAX_EVENTS];
|
||
|
|
|
||
|
|
self = (struct nn_worker*) arg;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
|
||
|
|
/* Process all expired timers. */
|
||
|
|
while (1) {
|
||
|
|
rc = nn_timerset_event (&self->timerset, &thndl);
|
||
|
|
if (nn_fast (rc == -EAGAIN))
|
||
|
|
break;
|
||
|
|
errnum_assert (rc == 0, -rc);
|
||
|
|
timer = nn_cont (thndl, struct nn_worker_timer, hndl);
|
||
|
|
nn_ctx_enter (timer->owner->ctx);
|
||
|
|
nn_fsm_feed (timer->owner, -1, NN_WORKER_TIMER_TIMEOUT, timer);
|
||
|
|
nn_ctx_leave (timer->owner->ctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Compute the time interval till next timer expiration. */
|
||
|
|
timeout = nn_timerset_timeout (&self->timerset);
|
||
|
|
|
||
|
|
/* Wait for new events and/or timeouts. */
|
||
|
|
brc = GetQueuedCompletionStatusEx (self->cp, entries,
|
||
|
|
NN_WORKER_MAX_EVENTS, &count, timeout < 0 ? INFINITE : timeout,
|
||
|
|
FALSE);
|
||
|
|
if (nn_slow (!brc && GetLastError () == WAIT_TIMEOUT))
|
||
|
|
continue;
|
||
|
|
win_assert (brc);
|
||
|
|
|
||
|
|
for (i = 0; i != count; ++i) {
|
||
|
|
|
||
|
|
/* Process I/O completion events. */
|
||
|
|
if (nn_fast (entries [i].lpOverlapped)) {
|
||
|
|
op = nn_cont (entries [i].lpOverlapped,
|
||
|
|
struct nn_worker_op, olpd);
|
||
|
|
|
||
|
|
/* The 'Internal' field is actually an NTSTATUS. Report
|
||
|
|
success and error. Ignore warnings and informational
|
||
|
|
messages.*/
|
||
|
|
rc = entries [i].Internal & 0xc0000000;
|
||
|
|
switch (rc) {
|
||
|
|
case 0x00000000:
|
||
|
|
rc = NN_WORKER_OP_DONE;
|
||
|
|
break;
|
||
|
|
case 0xc0000000:
|
||
|
|
rc = NN_WORKER_OP_ERROR;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Raise the completion event. */
|
||
|
|
nn_ctx_enter (op->owner->ctx);
|
||
|
|
nn_assert (op->state != NN_WORKER_OP_STATE_IDLE);
|
||
|
|
if (rc != NN_WORKER_OP_ERROR &&
|
||
|
|
op->state == NN_WORKER_OP_STATE_ACTIVE_ZEROISERROR &&
|
||
|
|
entries [i].dwNumberOfBytesTransferred == 0)
|
||
|
|
rc = NN_WORKER_OP_ERROR;
|
||
|
|
op->state = NN_WORKER_OP_STATE_IDLE;
|
||
|
|
nn_fsm_feed (op->owner, op->src, rc, op);
|
||
|
|
nn_ctx_leave (op->owner->ctx);
|
||
|
|
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Worker thread shutdown is requested. */
|
||
|
|
if (nn_slow (entries [i].lpCompletionKey ==
|
||
|
|
(ULONG_PTR) &nn_worker_stop))
|
||
|
|
return;
|
||
|
|
|
||
|
|
/* Process tasks. */
|
||
|
|
task = (struct nn_worker_task*) entries [i].lpCompletionKey;
|
||
|
|
nn_ctx_enter (task->owner->ctx);
|
||
|
|
nn_fsm_feed (task->owner, task->src,
|
||
|
|
NN_WORKER_TASK_EXECUTE, task);
|
||
|
|
nn_ctx_leave (task->owner->ctx);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|