2016-01-26 16:07:52 +00:00
|
|
|
/*
|
|
|
|
* $Id$
|
|
|
|
*
|
|
|
|
Copyright (c) 2015-2016 Chung, Hyung-Hwan. All rights reserved.
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions
|
|
|
|
are met:
|
|
|
|
1. Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer in the
|
|
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR
|
|
|
|
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WAfRRANTIES
|
|
|
|
OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
|
|
IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
|
|
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
|
|
NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
|
|
THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
#include "mio-prv.h"
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
#if defined(HAVE_SYS_EPOLL_H)
|
|
|
|
# include <sys/epoll.h>
|
|
|
|
# define USE_EPOLL
|
|
|
|
#elif defined(HAVE_SYS_POLL_H)
|
|
|
|
# include <sys/poll.h>
|
|
|
|
# define USE_POLL
|
|
|
|
#else
|
|
|
|
# error NO SUPPORTED MULTIPLEXER
|
|
|
|
#endif
|
2016-01-26 16:02:20 +00:00
|
|
|
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <errno.h>
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
#define DEV_CAPA_ALL_WATCHED (MIO_DEV_CAPA_IN_WATCHED | MIO_DEV_CAPA_OUT_WATCHED | MIO_DEV_CAPA_PRI_WATCHED)
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int schedule_kill_zombie_job (mio_dev_t* dev);
|
|
|
|
static int kill_and_free_device (mio_dev_t* dev, int force);
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2016-04-17 17:00:58 +00:00
|
|
|
#define APPEND_DEVICE_TO_LIST(list,dev) do { \
|
|
|
|
if ((list)->tail) (list)->tail->dev_next = (dev); \
|
|
|
|
else (list)->head = (dev); \
|
|
|
|
(dev)->dev_prev = (list)->tail; \
|
2018-12-12 13:15:54 +00:00
|
|
|
(dev)->dev_next = MIO_NULL; \
|
2016-04-17 17:00:58 +00:00
|
|
|
(list)->tail = (dev); \
|
|
|
|
} while(0)
|
|
|
|
|
|
|
|
#define UNLINK_DEVICE_FROM_LIST(list,dev) do { \
|
|
|
|
if ((dev)->dev_prev) (dev)->dev_prev->dev_next = (dev)->dev_next; \
|
|
|
|
else (list)->head = (dev)->dev_next; \
|
|
|
|
if ((dev)->dev_next) (dev)->dev_next->dev_prev = (dev)->dev_prev; \
|
|
|
|
else (list)->tail = (dev)->dev_prev; \
|
|
|
|
} while (0)
|
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* ========================================================================= */
|
|
|
|
#if defined(USE_POLL)
|
|
|
|
|
|
|
|
#define MUX_CMD_INSERT 1
|
|
|
|
#define MUX_CMD_UPDATE 2
|
|
|
|
#define MUX_CMD_DELETE 3
|
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
#define MUX_INDEX_INVALID MIO_TYPE_MAX(mio_oow_t)
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
struct mio_mux_t
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
|
|
|
struct
|
|
|
|
{
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_oow_t* ptr;
|
|
|
|
mio_oow_t size;
|
|
|
|
mio_oow_t capa;
|
2016-04-24 17:30:43 +00:00
|
|
|
} map; /* handle to index */
|
|
|
|
|
|
|
|
struct
|
|
|
|
{
|
|
|
|
struct pollfd* pfd;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t** dptr;
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_oow_t size;
|
|
|
|
mio_oow_t capa;
|
2016-04-24 17:30:43 +00:00
|
|
|
} pd; /* poll data */
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int mux_open (mio_t* mio)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_mux_t* mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mux = MIO_MMGR_ALLOC (mio->mmgr, MIO_SIZEOF(*mux));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (!mux)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (mux, 0, MIO_SIZEOF(*mux));
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->mux = mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static void mux_close (mio_t* mio)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->mux)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (mio->mmgr, mio->mux);
|
|
|
|
mio->mux = MIO_NULL;
|
2016-04-24 17:30:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int mux_control (mio_dev_t* dev, int cmd, mio_syshnd_t hnd, int dev_capa)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_t* mio;
|
|
|
|
mio_mux_t* mux;
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_oow_t idx;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio = dev->mio;
|
|
|
|
mux = (mio_mux_t*)mio->mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
if (hnd >= mux->map.capa)
|
|
|
|
{
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_oow_t new_capa;
|
|
|
|
mio_oow_t* tmp;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
if (cmd != MUX_CMD_INSERT)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOENT;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
new_capa = MIO_ALIGN_POW2((hnd + 1), 256);
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
tmp = MIO_MMGR_REALLOC(mio->mmgr, mux->map.ptr, new_capa * MIO_SIZEOF(*tmp));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (!tmp)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (idx = mux->map.capa; idx < new_capa; idx++)
|
|
|
|
tmp[idx] = MUX_INDEX_INVALID;
|
|
|
|
|
|
|
|
mux->map.ptr = tmp;
|
|
|
|
mux->map.capa = new_capa;
|
|
|
|
}
|
|
|
|
|
|
|
|
idx = mux->map.ptr[hnd];
|
|
|
|
if (idx != MUX_INDEX_INVALID)
|
|
|
|
{
|
|
|
|
if (cmd == MUX_CMD_INSERT)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_EEXIST;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (cmd != MUX_CMD_INSERT)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOENT;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (cmd)
|
|
|
|
{
|
|
|
|
case MUX_CMD_INSERT:
|
|
|
|
|
|
|
|
if (mux->pd.size >= mux->pd.capa)
|
|
|
|
{
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_oow_t new_capa;
|
2016-04-24 17:30:43 +00:00
|
|
|
struct pollfd* tmp1;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t** tmp2;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
new_capa = MIO_ALIGN_POW2(mux->pd.size + 1, 256);
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
tmp1 = MIO_MMGR_REALLOC(mio->mmgr, mux->pd.pfd, new_capa * MIO_SIZEOF(*tmp1));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (!tmp1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
tmp2 = MIO_MMGR_REALLOC (mio->mmgr, mux->pd.dptr, new_capa * MIO_SIZEOF(*tmp2));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (!tmp2)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (mio->mmgr, tmp1);
|
|
|
|
mio->errnum = MIO_ENOMEM;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
mux->pd.pfd = tmp1;
|
|
|
|
mux->pd.dptr = tmp2;
|
|
|
|
mux->pd.capa = new_capa;
|
|
|
|
}
|
|
|
|
|
|
|
|
idx = mux->pd.size++;
|
|
|
|
|
|
|
|
mux->pd.pfd[idx].fd = hnd;
|
|
|
|
mux->pd.pfd[idx].events = 0;
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_capa & MIO_DEV_CAPA_IN_WATCHED) mux->pd.pfd[idx].events |= POLLIN;
|
|
|
|
if (dev_capa & MIO_DEV_CAPA_OUT_WATCHED) mux->pd.pfd[idx].events |= POLLOUT;
|
2016-04-24 17:30:43 +00:00
|
|
|
mux->pd.pfd[idx].revents = 0;
|
|
|
|
mux->pd.dptr[idx] = dev;
|
|
|
|
|
|
|
|
mux->map.ptr[hnd] = idx;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
case MUX_CMD_UPDATE:
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (mux->pd.dptr[idx] == dev);
|
2016-04-24 17:30:43 +00:00
|
|
|
mux->pd.pfd[idx].events = 0;
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_capa & MIO_DEV_CAPA_IN_WATCHED) mux->pd.pfd[idx].events |= POLLIN;
|
|
|
|
if (dev_capa & MIO_DEV_CAPA_OUT_WATCHED) mux->pd.pfd[idx].events |= POLLOUT;
|
2016-04-24 17:30:43 +00:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
case MUX_CMD_DELETE:
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (mux->pd.dptr[idx] == dev);
|
2016-04-24 17:30:43 +00:00
|
|
|
mux->map.ptr[hnd] = MUX_INDEX_INVALID;
|
|
|
|
|
|
|
|
/* TODO: speed up deletion. allow a hole in the array.
|
|
|
|
* delay array compaction if there is a hole.
|
|
|
|
* set fd for the hole to -1 such that poll()
|
|
|
|
* ignores it. compact the array if another deletion
|
|
|
|
* is requested when there is an existing hole. */
|
|
|
|
idx++;
|
|
|
|
while (idx < mux->pd.size)
|
|
|
|
{
|
|
|
|
int fd;
|
|
|
|
|
|
|
|
mux->pd.pfd[idx - 1] = mux->pd.pfd[idx];
|
|
|
|
mux->pd.dptr[idx - 1] = mux->pd.dptr[idx];
|
|
|
|
|
|
|
|
fd = mux->pd.pfd[idx].fd;
|
|
|
|
mux->map.ptr[fd] = idx - 1;
|
|
|
|
|
|
|
|
idx++;
|
|
|
|
}
|
|
|
|
|
|
|
|
mux->pd.size--;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
default:
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_EINVAL;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#elif defined(USE_EPOLL)
|
|
|
|
|
|
|
|
#define MUX_CMD_INSERT EPOLL_CTL_ADD
|
|
|
|
#define MUX_CMD_UPDATE EPOLL_CTL_MOD
|
|
|
|
#define MUX_CMD_DELETE EPOLL_CTL_DEL
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
struct mio_mux_t
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
|
|
|
int hnd;
|
2016-04-25 16:15:36 +00:00
|
|
|
struct epoll_event revs[100]; /* TODO: is it a good size? */
|
2016-04-24 17:30:43 +00:00
|
|
|
};
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int mux_open (mio_t* mio)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_mux_t* mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mux = MIO_MMGR_ALLOC (mio->mmgr, MIO_SIZEOF(*mux));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (!mux)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (mux, 0, MIO_SIZEOF(*mux));
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
mux->hnd = epoll_create (1000);
|
|
|
|
if (mux->hnd == -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = mio_syserrtoerrnum(errno);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, mux);
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->mux = mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static void mux_close (mio_t* mio)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->mux)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
close (mio->mux->hnd);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, mio->mux);
|
|
|
|
mio->mux = MIO_NULL;
|
2016-04-24 17:30:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static MIO_INLINE int mux_control (mio_dev_t* dev, int cmd, mio_syshnd_t hnd, int dev_capa)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
|
|
|
struct epoll_event ev;
|
|
|
|
|
|
|
|
ev.data.ptr = dev;
|
|
|
|
ev.events = EPOLLHUP | EPOLLERR /*| EPOLLET*/;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_capa & MIO_DEV_CAPA_IN_WATCHED)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
|
|
|
ev.events |= EPOLLIN;
|
|
|
|
#if defined(EPOLLRDHUP)
|
|
|
|
ev.events |= EPOLLRDHUP;
|
|
|
|
#endif
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_capa & MIO_DEV_CAPA_PRI_WATCHED) ev.events |= EPOLLPRI;
|
2016-04-24 17:30:43 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_capa & MIO_DEV_CAPA_OUT_WATCHED) ev.events |= EPOLLOUT;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (epoll_ctl (dev->mio->mux->hnd, cmd, hnd, &ev) == -1)
|
2016-04-24 17:30:43 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = mio_syserrtoerrnum(errno);
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* ========================================================================= */
|
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_t* mio_open (mio_mmgr_t* mmgr, mio_oow_t xtnsize, mio_oow_t tmrcapa, mio_errnum_t* errnum)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_t* mio;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio = MIO_MMGR_ALLOC (mmgr, MIO_SIZEOF(mio_t) + xtnsize);
|
|
|
|
if (mio)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio_init (mio, mmgr, tmrcapa) <= -1)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (errnum) *errnum = mio->errnum;
|
|
|
|
MIO_MMGR_FREE (mmgr, mio);
|
|
|
|
mio = MIO_NULL;
|
2016-01-30 05:24:23 +00:00
|
|
|
}
|
2018-12-12 13:15:54 +00:00
|
|
|
else MIO_MEMSET (mio + 1, 0, xtnsize);
|
2016-01-30 05:24:23 +00:00
|
|
|
}
|
|
|
|
else
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (errnum) *errnum = MIO_ENOMEM;
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
return mio;
|
2016-01-30 05:24:23 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_close (mio_t* mio)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_fini (mio);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, mio);
|
2016-01-30 05:24:23 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
int mio_init (mio_t* mio, mio_mmgr_t* mmgr, mio_oow_t tmrcapa)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (mio, 0, MIO_SIZEOF(*mio));
|
|
|
|
mio->mmgr = mmgr;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-01-30 05:24:23 +00:00
|
|
|
/* intialize the multiplexer object */
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mux_open (mio) <= -1) return -1;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-01-30 05:24:23 +00:00
|
|
|
/* initialize the timer object */
|
|
|
|
if (tmrcapa <= 0) tmrcapa = 1;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->tmr.jobs = MIO_MMGR_ALLOC (mio->mmgr, tmrcapa * MIO_SIZEOF(mio_tmrjob_t));
|
|
|
|
if (!mio->tmr.jobs)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
|
|
|
mux_close (mio);
|
2016-01-30 05:24:23 +00:00
|
|
|
return -1;
|
|
|
|
}
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->tmr.capa = tmrcapa;
|
2016-01-30 05:24:23 +00:00
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
MIO_CWQ_INIT (&mio->cwq);
|
2016-01-30 05:24:23 +00:00
|
|
|
return 0;
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_fini (mio_t* mio)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* dev, * next_dev;
|
2016-04-17 17:00:58 +00:00
|
|
|
struct
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* head;
|
|
|
|
mio_dev_t* tail;
|
2016-04-17 17:00:58 +00:00
|
|
|
} diehard;
|
2019-01-16 04:37:53 +00:00
|
|
|
mio_oow_t i;
|
2016-01-30 05:24:23 +00:00
|
|
|
|
2019-01-16 04:37:53 +00:00
|
|
|
/* clean up free cwq list */
|
|
|
|
for (i = 0; i < MIO_COUNTOF(mio->cwqfl); i++)
|
2019-01-14 09:23:16 +00:00
|
|
|
{
|
2019-01-16 04:37:53 +00:00
|
|
|
mio_cwq_t* cwq;
|
|
|
|
while ((cwq = mio->cwqfl[i]))
|
|
|
|
{
|
|
|
|
mio->cwqfl[i] = cwq->next;
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, cwq);
|
|
|
|
}
|
2019-01-14 09:23:16 +00:00
|
|
|
}
|
|
|
|
|
2016-04-16 16:05:57 +00:00
|
|
|
/* kill all registered devices */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (mio->actdev.head)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_killdev (mio, mio->actdev.head);
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2019-01-16 04:37:53 +00:00
|
|
|
/* kill all halted devices */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (mio->hltdev.head)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_killdev (mio, mio->hltdev.head);
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* clean up all zombie devices */
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (&diehard, 0, MIO_SIZEOF(diehard));
|
|
|
|
for (dev = mio->zmbdev.head; dev; )
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
kill_and_free_device (dev, 1);
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->zmbdev.head == dev)
|
2016-04-17 17:00:58 +00:00
|
|
|
{
|
|
|
|
/* the deive has not been freed. go on to the next one */
|
|
|
|
next_dev = dev->dev_next;
|
|
|
|
|
|
|
|
/* remove the device from the zombie device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
UNLINK_DEVICE_FROM_LIST (&mio->zmbdev, dev);
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_ZOMBIE;
|
2016-04-17 17:00:58 +00:00
|
|
|
|
|
|
|
/* put it to a private list for aborting */
|
|
|
|
APPEND_DEVICE_TO_LIST (&diehard, dev);
|
|
|
|
|
|
|
|
dev = next_dev;
|
|
|
|
}
|
2018-12-12 13:15:54 +00:00
|
|
|
else dev = mio->zmbdev.head;
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2016-04-17 17:00:58 +00:00
|
|
|
while (diehard.head)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* if the kill method returns failure, it can leak some resource
|
|
|
|
* because the device is freed regardless of the failure when 2
|
|
|
|
* is given to kill_and_free_device(). */
|
2016-04-17 17:00:58 +00:00
|
|
|
dev = diehard.head;
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (!(dev->dev_capa & (MIO_DEV_CAPA_ACTIVE | MIO_DEV_CAPA_HALTED | MIO_DEV_CAPA_ZOMBIE)));
|
2016-04-17 17:00:58 +00:00
|
|
|
UNLINK_DEVICE_FROM_LIST (&diehard, dev);
|
|
|
|
kill_and_free_device (dev, 2);
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2016-04-17 17:00:58 +00:00
|
|
|
/* purge scheduled timer jobs and kill the timer */
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_cleartmrjobs (mio);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, mio->tmr.jobs);
|
2016-04-17 17:00:58 +00:00
|
|
|
|
2016-01-30 05:24:23 +00:00
|
|
|
/* close the multiplexer */
|
2018-12-12 13:15:54 +00:00
|
|
|
mux_close (mio);
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_prologue (mio_t* mio)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* TODO: */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_epilogue (mio_t* mio)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* TODO: */
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static MIO_INLINE void unlink_wq (mio_t* mio, mio_wq_t* q)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (q->tmridx != MIO_TMRIDX_INVALID)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_deltmrjob (mio, q->tmridx);
|
|
|
|
MIO_ASSERT (q->tmridx == MIO_TMRIDX_INVALID);
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_WQ_UNLINK (q);
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static MIO_INLINE void handle_event (mio_dev_t* dev, int events, int rdhup)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_t* mio;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio = dev->mio;
|
|
|
|
mio->renew_watch = 0;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (mio == dev->mio);
|
2016-02-05 13:25:59 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
if (dev->dev_evcb->ready)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2016-04-24 17:30:43 +00:00
|
|
|
int x, xevents;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
xevents = events;
|
2018-12-12 13:15:54 +00:00
|
|
|
if (rdhup) xevents |= MIO_DEV_EVENT_HUP;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* return value of ready()
|
|
|
|
* <= -1 - failure. kill the device.
|
|
|
|
* == 0 - ok. but don't invoke recv() or send().
|
|
|
|
* >= 1 - everything is ok. */
|
2019-01-17 08:09:19 +00:00
|
|
|
x = dev->dev_evcb->ready(dev, xevents);
|
2016-02-04 15:06:20 +00:00
|
|
|
if (x <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
2016-02-04 15:06:20 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
else if (x == 0) goto skip_evcb;
|
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev && (events & MIO_DEV_EVENT_PRI))
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2016-02-04 15:06:20 +00:00
|
|
|
/* urgent data */
|
2016-04-25 16:15:36 +00:00
|
|
|
/* TODO: implement urgent data handling */
|
2019-01-17 08:09:19 +00:00
|
|
|
/*x = dev->dev_mth->urgread(dev, mio->bugbuf, &len);*/
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev && (events & MIO_DEV_EVENT_OUT))
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2016-02-04 15:06:20 +00:00
|
|
|
/* write pending requests */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (!MIO_WQ_ISEMPTY(&dev->wq))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_wq_t* q;
|
|
|
|
const mio_uint8_t* uptr;
|
|
|
|
mio_iolen_t urem, ulen;
|
2016-02-04 15:06:20 +00:00
|
|
|
int x;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
q = MIO_WQ_HEAD(&dev->wq);
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
uptr = q->ptr;
|
|
|
|
urem = q->len;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
send_leftover:
|
|
|
|
ulen = urem;
|
2019-01-17 08:09:19 +00:00
|
|
|
x = dev->dev_mth->write(dev, uptr, &ulen, &q->dstaddr);
|
2016-02-04 15:06:20 +00:00
|
|
|
if (x <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (x == 0)
|
|
|
|
{
|
|
|
|
/* keep the left-over */
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMMOVE (q->ptr, uptr, urem);
|
2016-02-04 15:06:20 +00:00
|
|
|
q->len = urem;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
uptr += ulen;
|
|
|
|
urem -= ulen;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
if (urem <= 0)
|
|
|
|
{
|
|
|
|
/* finished writing a single write request */
|
|
|
|
int y, out_closed = 0;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (q->len <= 0 && (dev->dev_capa & MIO_DEV_CAPA_STREAM))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* it was a zero-length write request.
|
|
|
|
* for a stream, it is to close the output. */
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_OUT_CLOSED;
|
|
|
|
mio->renew_watch = 1;
|
2016-02-04 15:06:20 +00:00
|
|
|
out_closed = 1;
|
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
unlink_wq (mio, q);
|
2019-01-17 08:09:19 +00:00
|
|
|
y = dev->dev_evcb->on_write(dev, q->olen, q->ctx, &q->dstaddr);
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (mio->mmgr, q);
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
if (y <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
break;
|
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
if (out_closed)
|
|
|
|
{
|
|
|
|
/* drain all pending requests.
|
|
|
|
* callbacks are skipped for drained requests */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (!MIO_WQ_ISEMPTY(&dev->wq))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
q = MIO_WQ_HEAD(&dev->wq);
|
|
|
|
unlink_wq (mio, q);
|
|
|
|
MIO_MMGR_FREE (dev->mio->mmgr, q);
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else goto send_leftover;
|
|
|
|
}
|
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev && MIO_WQ_ISEMPTY(&dev->wq))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* no pending request to write */
|
2018-12-12 13:15:54 +00:00
|
|
|
if ((dev->dev_capa & MIO_DEV_CAPA_IN_CLOSED) &&
|
|
|
|
(dev->dev_capa & MIO_DEV_CAPA_OUT_CLOSED))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->renew_watch = 1;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
}
|
2016-02-02 07:51:17 +00:00
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev && (events & MIO_DEV_EVENT_IN))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_devaddr_t srcaddr;
|
|
|
|
mio_iolen_t len;
|
2016-02-04 15:06:20 +00:00
|
|
|
int x;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* the devices are all non-blocking. read as much as possible
|
|
|
|
* if on_read callback returns 1 or greater. read only once
|
|
|
|
* if the on_read calllback returns 0. */
|
|
|
|
while (1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
len = MIO_COUNTOF(mio->bigbuf);
|
2019-01-17 08:09:19 +00:00
|
|
|
x = dev->dev_mth->read(dev, mio->bigbuf, &len, &srcaddr);
|
2016-02-04 15:06:20 +00:00
|
|
|
if (x <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (x == 0)
|
|
|
|
{
|
|
|
|
/* no data is available - EWOULDBLOCK or something similar */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (x >= 1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (len <= 0 && (dev->dev_capa & MIO_DEV_CAPA_STREAM))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* EOF received. for a stream device, a zero-length
|
|
|
|
* read is interpreted as EOF. */
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_IN_CLOSED;
|
|
|
|
mio->renew_watch = 1;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
|
|
|
/* call the on_read callback to report EOF */
|
2019-01-17 08:09:19 +00:00
|
|
|
if (dev->dev_evcb->on_read(dev, mio->bigbuf, len, &srcaddr) <= -1 ||
|
2018-12-12 13:15:54 +00:00
|
|
|
(dev->dev_capa & MIO_DEV_CAPA_OUT_CLOSED))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* 1. input ended and its reporting failed or
|
|
|
|
* 2. input ended and no writing is possible */
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* since EOF is received, reading can't be greedy */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
int y;
|
|
|
|
/* TODO: for a stream device, merge received data if bigbuf isn't full and fire the on_read callback
|
|
|
|
* when x == 0 or <= -1. you can */
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* data available */
|
2019-01-17 08:09:19 +00:00
|
|
|
y = dev->dev_evcb->on_read(dev, mio->bigbuf, len, &srcaddr);
|
2016-02-04 15:06:20 +00:00
|
|
|
if (y <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (y == 0)
|
|
|
|
{
|
|
|
|
/* don't be greedy. read only once
|
|
|
|
* for this loop iteration */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-02-02 07:51:17 +00:00
|
|
|
|
2016-02-05 13:25:59 +00:00
|
|
|
if (dev)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (events & (MIO_DEV_EVENT_ERR | MIO_DEV_EVENT_HUP))
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
|
|
|
/* if error or hangup has been reported on the device,
|
|
|
|
* halt the device. this check is performed after
|
|
|
|
* EPOLLIN or EPOLLOUT check because EPOLLERR or EPOLLHUP
|
|
|
|
* can be set together with EPOLLIN or EPOLLOUT. */
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_IN_CLOSED | MIO_DEV_CAPA_OUT_CLOSED;
|
|
|
|
mio->renew_watch = 1;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2016-04-24 17:30:43 +00:00
|
|
|
else if (dev && rdhup)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (events & (MIO_DEV_EVENT_IN | MIO_DEV_EVENT_OUT | MIO_DEV_EVENT_PRI))
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
|
|
|
/* it may be a half-open state. don't do anything here
|
|
|
|
* to let the next read detect EOF */
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_IN_CLOSED | MIO_DEV_CAPA_OUT_CLOSED;
|
|
|
|
mio->renew_watch = 1;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
}
|
2016-02-02 07:51:17 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if ((dev->dev_capa & MIO_DEV_CAPA_IN_CLOSED) &&
|
|
|
|
(dev->dev_capa & MIO_DEV_CAPA_OUT_CLOSED))
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
skip_evcb:
|
2019-01-17 08:09:19 +00:00
|
|
|
if (dev && mio->renew_watch && mio_dev_watch(dev, MIO_DEV_WATCH_RENEW, 0) <= -1)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_halt (dev);
|
|
|
|
dev = MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static MIO_INLINE int __exec (mio_t* mio)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_ntime_t tmout;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
|
|
|
#if defined(_WIN32)
|
|
|
|
ULONG nentries, i;
|
|
|
|
#else
|
|
|
|
int nentries, i;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_mux_t* mux;
|
2016-01-26 16:02:20 +00:00
|
|
|
#endif
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
/*if (!mio->actdev.head) return 0;*/
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
/* execute callbacks for completed write operations */
|
|
|
|
while (!MIO_CWQ_ISEMPTY(&mio->cwq))
|
|
|
|
{
|
|
|
|
mio_cwq_t* cwq;
|
2019-01-16 04:37:53 +00:00
|
|
|
mio_oow_t cwqfl_index;
|
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
cwq = MIO_CWQ_HEAD(&mio->cwq);
|
|
|
|
if (cwq->dev->dev_evcb->on_write(cwq->dev, cwq->olen, cwq->ctx, &cwq->dstaddr) <= -1) return -1;
|
2019-01-14 08:45:56 +00:00
|
|
|
cwq->dev->cw_count--;
|
2019-01-14 08:33:14 +00:00
|
|
|
MIO_CWQ_UNLINK (cwq);
|
2019-01-14 09:23:16 +00:00
|
|
|
|
2019-01-16 04:37:53 +00:00
|
|
|
cwqfl_index = MIO_ALIGN_POW2(cwq->dstaddr.len, MIO_CWQFL_ALIGN) / MIO_CWQFL_SIZE;
|
|
|
|
if (cwqfl_index < MIO_COUNTOF(mio->cwqfl))
|
2019-01-14 09:23:16 +00:00
|
|
|
{
|
|
|
|
/* reuse the cwq object if dstaddr is 0 in size. chain it to the free list */
|
2019-01-16 04:37:53 +00:00
|
|
|
cwq->next = mio->cwqfl[cwqfl_index];
|
|
|
|
mio->cwqfl[cwqfl_index] = cwq;
|
2019-01-14 09:23:16 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* TODO: more reuse of objects of different size? */
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, cwq);
|
|
|
|
}
|
2019-01-14 08:33:14 +00:00
|
|
|
}
|
|
|
|
|
2016-01-30 19:08:28 +00:00
|
|
|
/* execute the scheduled jobs before checking devices with the
|
|
|
|
* multiplexer. the scheduled jobs can safely destroy the devices */
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_firetmrjobs (mio, MIO_NULL, MIO_NULL);
|
2016-01-30 19:08:28 +00:00
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
if (mio_gettmrtmout(mio, MIO_NULL, &tmout) <= -1)
|
2016-01-30 05:24:23 +00:00
|
|
|
{
|
|
|
|
/* defaults to 1 second if timeout can't be acquired */
|
2016-01-30 19:08:28 +00:00
|
|
|
tmout.sec = 1; /* TODO: make the default timeout configurable */
|
2016-01-30 05:24:23 +00:00
|
|
|
tmout.nsec = 0;
|
|
|
|
}
|
|
|
|
|
2016-01-26 16:02:20 +00:00
|
|
|
#if defined(_WIN32)
|
|
|
|
/*
|
2018-12-12 13:15:54 +00:00
|
|
|
if (GetQueuedCompletionStatusEx (mio->iocp, mio->ovls, MIO_COUNTOF(mio->ovls), &nentries, timeout, FALSE) == FALSE)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
|
|
|
// TODO: set errnum
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; i < nentries; i++)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
*/
|
2016-04-24 17:30:43 +00:00
|
|
|
#elif defined(USE_POLL)
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mux = (mio_mux_t*)mio->mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
nentries = poll(mux->pd.pfd, mux->pd.size, MIO_SECNSEC_TO_MSEC(tmout.sec, tmout.nsec));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (nentries == -1)
|
|
|
|
{
|
|
|
|
if (errno == EINTR) return 0;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = mio_syserrtoerrnum(errno);
|
2016-04-24 17:30:43 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; i < mux->pd.size; i++)
|
|
|
|
{
|
|
|
|
if (mux->pd.pfd[i].fd >= 0 && mux->pd.pfd[i].revents)
|
|
|
|
{
|
|
|
|
int events = 0;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* dev;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
dev = mux->pd.dptr[i];
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (!(mux->pd.pfd[i].revents & POLLNVAL));
|
|
|
|
if (mux->pd.pfd[i].revents & POLLIN) events |= MIO_DEV_EVENT_IN;
|
|
|
|
if (mux->pd.pfd[i].revents & POLLOUT) events |= MIO_DEV_EVENT_OUT;
|
|
|
|
if (mux->pd.pfd[i].revents & POLLPRI) events |= MIO_DEV_EVENT_PRI;
|
|
|
|
if (mux->pd.pfd[i].revents & POLLERR) events |= MIO_DEV_EVENT_ERR;
|
|
|
|
if (mux->pd.pfd[i].revents & POLLHUP) events |= MIO_DEV_EVENT_HUP;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
handle_event (dev, events, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#elif defined(USE_EPOLL)
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mux = (mio_mux_t*)mio->mux;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
nentries = epoll_wait(mux->hnd, mux->revs, MIO_COUNTOF(mux->revs), MIO_SECNSEC_TO_MSEC(tmout.sec, tmout.nsec));
|
2016-04-24 17:30:43 +00:00
|
|
|
if (nentries == -1)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2016-01-30 05:24:23 +00:00
|
|
|
if (errno == EINTR) return 0; /* it's actually ok */
|
|
|
|
/* other errors are critical - EBADF, EFAULT, EINVAL */
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = mio_syserrtoerrnum(errno);
|
2016-01-26 16:02:20 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2016-01-30 05:24:23 +00:00
|
|
|
/* TODO: merge events??? for the same descriptor */
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2016-01-26 16:02:20 +00:00
|
|
|
for (i = 0; i < nentries; i++)
|
|
|
|
{
|
2016-04-24 17:30:43 +00:00
|
|
|
int events = 0, rdhup = 0;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* dev;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
|
|
|
dev = mux->revs[i].data.ptr;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mux->revs[i].events & EPOLLIN) events |= MIO_DEV_EVENT_IN;
|
|
|
|
if (mux->revs[i].events & EPOLLOUT) events |= MIO_DEV_EVENT_OUT;
|
|
|
|
if (mux->revs[i].events & EPOLLPRI) events |= MIO_DEV_EVENT_PRI;
|
|
|
|
if (mux->revs[i].events & EPOLLERR) events |= MIO_DEV_EVENT_ERR;
|
|
|
|
if (mux->revs[i].events & EPOLLHUP) events |= MIO_DEV_EVENT_HUP;
|
2016-04-24 17:30:43 +00:00
|
|
|
#if defined(EPOLLRDHUP)
|
|
|
|
else if (mux->revs[i].events & EPOLLRDHUP) rdhup = 1;
|
|
|
|
#endif
|
|
|
|
handle_event (dev, events, rdhup);
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
#else
|
|
|
|
|
|
|
|
# error NO SUPPORTED MULTIPLEXER
|
|
|
|
#endif
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* kill all halted devices */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (mio->hltdev.head)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
printf (">>>>>>>>>>>>>> KILLING HALTED DEVICE %p\n", mio->hltdev.head);
|
|
|
|
mio_killdev (mio, mio->hltdev.head);
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (mio->hltdev.tail == MIO_NULL);
|
2016-02-02 07:51:17 +00:00
|
|
|
|
2016-01-26 16:02:20 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_exec (mio_t* mio)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
|
|
|
int n;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->in_exec = 1;
|
|
|
|
n = __exec (mio);
|
|
|
|
mio->in_exec = 0;
|
2016-02-05 13:25:59 +00:00
|
|
|
|
|
|
|
return n;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_stop (mio_t* mio, mio_stopreq_t stopreq)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->stopreq = stopreq;
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_loop (mio_t* mio)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (!mio->actdev.head) return 0;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->stopreq = MIO_STOPREQ_NONE;
|
|
|
|
mio->renew_watch = 0;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
if (mio_prologue(mio) <= -1) return -1;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
while (mio->stopreq == MIO_STOPREQ_NONE && mio->actdev.head)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2019-01-11 07:35:43 +00:00
|
|
|
if (mio_exec(mio) <= -1) break;
|
2016-01-26 16:02:20 +00:00
|
|
|
/* you can do other things here */
|
|
|
|
}
|
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
mio_epilogue(mio);
|
2016-01-26 16:02:20 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 14:16:49 +00:00
|
|
|
mio_dev_t* mio_makedev (mio_t* mio, mio_oow_t dev_size, mio_dev_mth_t* dev_mth, mio_dev_evcb_t* dev_evcb, void* make_ctx)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* dev;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev_size < MIO_SIZEOF(mio_dev_t))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_EINVAL;
|
|
|
|
return MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
dev = MIO_MMGR_ALLOC(mio->mmgr, dev_size);
|
2016-02-04 15:06:20 +00:00
|
|
|
if (!dev)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOMEM;
|
|
|
|
return MIO_NULL;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (dev, 0, dev_size);
|
|
|
|
dev->mio = mio;
|
2016-02-04 15:06:20 +00:00
|
|
|
dev->dev_size = dev_size;
|
|
|
|
/* default capability. dev->dev_mth->make() can change this.
|
2018-12-12 13:15:54 +00:00
|
|
|
* mio_dev_watch() is affected by the capability change. */
|
|
|
|
dev->dev_capa = MIO_DEV_CAPA_IN | MIO_DEV_CAPA_OUT;
|
2016-02-04 15:06:20 +00:00
|
|
|
dev->dev_mth = dev_mth;
|
|
|
|
dev->dev_evcb = dev_evcb;
|
2019-01-17 08:09:19 +00:00
|
|
|
dev->rtmridx = MIO_TMRIDX_INVALID;
|
2019-01-14 08:33:14 +00:00
|
|
|
MIO_WQ_INIT (&dev->wq);
|
2019-01-14 08:45:56 +00:00
|
|
|
dev->cw_count = 0;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
|
|
|
/* call the callback function first */
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOERR;
|
2019-01-11 07:35:43 +00:00
|
|
|
if (dev->dev_mth->make(dev, make_ctx) <= -1)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->errnum == MIO_ENOERR) mio->errnum = MIO_EDEVMAKE;
|
2016-02-04 15:06:20 +00:00
|
|
|
goto oops;
|
|
|
|
}
|
|
|
|
|
2016-04-02 15:25:35 +00:00
|
|
|
/* the make callback must not change these fields */
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (dev->dev_mth == dev_mth);
|
|
|
|
MIO_ASSERT (dev->dev_evcb == dev_evcb);
|
|
|
|
MIO_ASSERT (dev->dev_prev == MIO_NULL);
|
|
|
|
MIO_ASSERT (dev->dev_next == MIO_NULL);
|
2016-04-02 15:25:35 +00:00
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* set some internal capability bits according to the capabilities
|
|
|
|
* removed by the device making callback for convenience sake. */
|
2018-12-12 13:15:54 +00:00
|
|
|
if (!(dev->dev_capa & MIO_DEV_CAPA_IN)) dev->dev_capa |= MIO_DEV_CAPA_IN_CLOSED;
|
|
|
|
if (!(dev->dev_capa & MIO_DEV_CAPA_OUT)) dev->dev_capa |= MIO_DEV_CAPA_OUT_CLOSED;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
|
|
|
#if defined(_WIN32)
|
2019-01-14 08:33:14 +00:00
|
|
|
if (CreateIoCompletionPort((HANDLE)dev->dev_mth->getsyshnd(dev), mio->iocp, MIO_IOCP_KEY, 0) == NULL)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* TODO: set errnum from GetLastError()... */
|
|
|
|
goto oops_after_make;
|
|
|
|
}
|
|
|
|
#else
|
2019-01-11 07:35:43 +00:00
|
|
|
if (mio_dev_watch(dev, MIO_DEV_WATCH_START, 0) <= -1) goto oops_after_make;
|
2016-02-04 15:06:20 +00:00
|
|
|
#endif
|
|
|
|
|
2016-04-17 17:00:58 +00:00
|
|
|
/* and place the new device object at the back of the active device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
APPEND_DEVICE_TO_LIST (&mio->actdev, dev);
|
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_ACTIVE;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
|
|
|
return dev;
|
|
|
|
|
|
|
|
oops_after_make:
|
2019-01-11 07:35:43 +00:00
|
|
|
if (kill_and_free_device(dev, 0) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* schedule a timer job that reattempts to destroy the device */
|
2019-01-11 07:35:43 +00:00
|
|
|
if (schedule_kill_zombie_job(dev) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* job scheduling failed. i have no choice but to
|
|
|
|
* destroy the device now.
|
|
|
|
*
|
|
|
|
* NOTE: this while loop can block the process
|
|
|
|
* if the kill method keep returning failure */
|
2019-01-11 07:35:43 +00:00
|
|
|
while (kill_and_free_device(dev, 1) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->stopreq != MIO_STOPREQ_NONE)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* i can't wait until destruction attempt gets
|
|
|
|
* fully successful. there is a chance that some
|
|
|
|
* resources can leak inside the device */
|
|
|
|
kill_and_free_device (dev, 2);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_NULL;
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
oops:
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (mio->mmgr, dev);
|
|
|
|
return MIO_NULL;
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int kill_and_free_device (mio_dev_t* dev, int force)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_t* mio;
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (!(dev->dev_capa & MIO_DEV_CAPA_ACTIVE));
|
|
|
|
MIO_ASSERT (!(dev->dev_capa & MIO_DEV_CAPA_HALTED));
|
2016-04-17 17:00:58 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio = dev->mio;
|
2016-04-16 16:05:57 +00:00
|
|
|
|
|
|
|
if (dev->dev_mth->kill(dev, force) <= -1)
|
|
|
|
{
|
|
|
|
if (force >= 2) goto free_device;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (!(dev->dev_capa & MIO_DEV_CAPA_ZOMBIE))
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
APPEND_DEVICE_TO_LIST (&mio->zmbdev, dev);
|
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_ZOMBIE;
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
free_device:
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_ZOMBIE)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* detach it from the zombie device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
UNLINK_DEVICE_FROM_LIST (&mio->zmbdev, dev);
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_ZOMBIE;
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (mio->mmgr, dev);
|
2016-04-16 16:05:57 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static void kill_zombie_job_handler (mio_t* mio, const mio_ntime_t* now, mio_tmrjob_t* job)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_t* dev = (mio_dev_t*)job->ctx;
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (dev->dev_capa & MIO_DEV_CAPA_ZOMBIE);
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
if (kill_and_free_device(dev, 0) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2019-01-11 07:35:43 +00:00
|
|
|
if (schedule_kill_zombie_job(dev) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* i have to choice but to free up the devide by force */
|
2019-01-11 07:35:43 +00:00
|
|
|
while (kill_and_free_device(dev, 1) <= -1)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2019-01-11 07:35:43 +00:00
|
|
|
if (mio->stopreq != MIO_STOPREQ_NONE)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* i can't wait until destruction attempt gets
|
|
|
|
* fully successful. there is a chance that some
|
|
|
|
* resources can leak inside the device */
|
|
|
|
kill_and_free_device (dev, 2);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int schedule_kill_zombie_job (mio_dev_t* dev)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_tmrjob_t kill_zombie_job;
|
|
|
|
mio_ntime_t tmout;
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_inittime (&tmout, 3, 0); /* TODO: take it from configuration */
|
2016-04-16 16:05:57 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (&kill_zombie_job, 0, MIO_SIZEOF(kill_zombie_job));
|
2016-04-16 16:05:57 +00:00
|
|
|
kill_zombie_job.ctx = dev;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_gettime (&kill_zombie_job.when);
|
|
|
|
mio_addtime (&kill_zombie_job.when, &tmout, &kill_zombie_job.when);
|
2016-04-16 16:05:57 +00:00
|
|
|
kill_zombie_job.handler = kill_zombie_job_handler;
|
|
|
|
/*kill_zombie_job.idxptr = &rdev->tmridx_kill_zombie;*/
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
return mio_instmrjob (dev->mio, &kill_zombie_job) == MIO_TMRIDX_INVALID? -1: 0;
|
2016-04-16 16:05:57 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_killdev (mio_t* mio, mio_dev_t* dev)
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (mio == dev->mio);
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_ZOMBIE)
|
2016-04-12 13:56:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (MIO_WQ_ISEMPTY(&dev->wq));
|
2019-01-17 08:09:19 +00:00
|
|
|
MIO_ASSERT (dev->cw_count == 0);
|
|
|
|
MIO_ASSERT (dev->rtmridx == MIO_TMRIDX_INVALID);
|
2016-04-12 13:56:59 +00:00
|
|
|
goto kill_device;
|
|
|
|
}
|
|
|
|
|
2019-01-17 08:09:19 +00:00
|
|
|
if (dev->rtmridx != MIO_TMRIDX_INVALID)
|
|
|
|
{
|
|
|
|
printf ("REMOVING.... TIMER FOR DEV\n");
|
|
|
|
mio_deltmrjob (mio, dev->rtmridx);
|
|
|
|
dev->rtmridx = MIO_TMRIDX_INVALID;
|
|
|
|
}
|
|
|
|
|
2019-01-14 08:33:14 +00:00
|
|
|
/* clear completed write event queues - TODO: do i need to fire these? */
|
2019-01-14 08:45:56 +00:00
|
|
|
if (dev->cw_count > 0)
|
2019-01-14 08:33:14 +00:00
|
|
|
{
|
2019-01-14 08:45:56 +00:00
|
|
|
mio_cwq_t* cwq, * next;
|
|
|
|
cwq = MIO_CWQ_HEAD(&mio->cwq);
|
|
|
|
while (cwq != &mio->cwq)
|
|
|
|
{
|
|
|
|
next = MIO_CWQ_NEXT(cwq);
|
|
|
|
if (cwq->dev == dev)
|
|
|
|
{
|
|
|
|
MIO_CWQ_UNLINK (cwq);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, cwq);
|
|
|
|
}
|
|
|
|
cwq = next;
|
|
|
|
}
|
2019-01-14 08:33:14 +00:00
|
|
|
}
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
/* clear pending send requests */
|
2018-12-12 13:15:54 +00:00
|
|
|
while (!MIO_WQ_ISEMPTY(&dev->wq))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_wq_t* q;
|
|
|
|
q = MIO_WQ_HEAD(&dev->wq);
|
|
|
|
unlink_wq (mio, q);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, q);
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_HALTED)
|
2016-04-12 13:56:59 +00:00
|
|
|
{
|
2016-04-16 16:05:57 +00:00
|
|
|
/* this device is in the halted state.
|
|
|
|
* unlink it from the halted device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
UNLINK_DEVICE_FROM_LIST (&mio->hltdev, dev);
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_HALTED;
|
2016-04-12 13:56:59 +00:00
|
|
|
}
|
|
|
|
else
|
2016-02-02 07:51:17 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (dev->dev_capa & MIO_DEV_CAPA_ACTIVE);
|
|
|
|
UNLINK_DEVICE_FROM_LIST (&mio->actdev, dev);
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_ACTIVE;
|
2016-02-02 07:51:17 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_dev_watch (dev, MIO_DEV_WATCH_STOP, 0);
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2016-04-12 13:56:59 +00:00
|
|
|
kill_device:
|
2016-04-16 16:05:57 +00:00
|
|
|
if (kill_and_free_device(dev, 0) <= -1)
|
2016-04-12 13:56:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (dev->dev_capa & MIO_DEV_CAPA_ZOMBIE);
|
2016-04-16 16:05:57 +00:00
|
|
|
if (schedule_kill_zombie_job (dev) <= -1)
|
2016-04-12 13:56:59 +00:00
|
|
|
{
|
2016-04-16 16:05:57 +00:00
|
|
|
/* i have to choice but to free up the devide by force */
|
|
|
|
while (kill_and_free_device (dev, 1) <= -1)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (mio->stopreq != MIO_STOPREQ_NONE)
|
2016-04-16 16:05:57 +00:00
|
|
|
{
|
|
|
|
/* i can't wait until destruction attempt gets
|
|
|
|
* fully successful. there is a chance that some
|
|
|
|
* resources can leak inside the device */
|
|
|
|
kill_and_free_device (dev, 2);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2016-04-12 13:56:59 +00:00
|
|
|
}
|
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
void mio_dev_halt (mio_dev_t* dev)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_ACTIVE)
|
2016-02-02 07:51:17 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_t* mio;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio = dev->mio;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2016-04-17 17:00:58 +00:00
|
|
|
/* delink the device object from the active device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
UNLINK_DEVICE_FROM_LIST (&mio->actdev, dev);
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_ACTIVE;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2016-04-12 13:56:59 +00:00
|
|
|
/* place it at the back of the halted device list */
|
2018-12-12 13:15:54 +00:00
|
|
|
APPEND_DEVICE_TO_LIST (&mio->hltdev, dev);
|
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_HALTED;
|
2016-02-02 07:51:17 +00:00
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_dev_ioctl (mio_dev_t* dev, int cmd, void* arg)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
if (dev->dev_mth->ioctl) return dev->dev_mth->ioctl (dev, cmd, arg);
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ENOSUP; /* TODO: different error code ? */
|
2016-02-04 15:06:20 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_dev_watch (mio_dev_t* dev, mio_dev_watch_cmd_t cmd, int events)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2016-04-24 17:30:43 +00:00
|
|
|
int mux_cmd;
|
2016-02-04 15:06:20 +00:00
|
|
|
int dev_capa;
|
2016-01-26 16:02:20 +00:00
|
|
|
|
2016-04-12 13:56:59 +00:00
|
|
|
/* the virtual device doesn't perform actual I/O.
|
2018-12-12 13:15:54 +00:00
|
|
|
* it's different from not hanving MIO_DEV_CAPA_IN and MIO_DEV_CAPA_OUT.
|
2016-04-12 13:56:59 +00:00
|
|
|
* a non-virtual device without the capabilities still gets attention
|
|
|
|
* of the system multiplexer for hangup and error. */
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_VIRTUAL) return 0;
|
2016-04-12 13:56:59 +00:00
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
/*ev.data.ptr = dev;*/
|
2016-01-26 16:02:20 +00:00
|
|
|
switch (cmd)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
case MIO_DEV_WATCH_START:
|
2016-02-04 15:06:20 +00:00
|
|
|
/* upon start, only input watching is requested */
|
2018-12-12 13:15:54 +00:00
|
|
|
events = MIO_DEV_EVENT_IN;
|
2016-04-24 17:30:43 +00:00
|
|
|
mux_cmd = MUX_CMD_INSERT;
|
2016-01-26 16:02:20 +00:00
|
|
|
break;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
case MIO_DEV_WATCH_RENEW:
|
2016-02-04 15:06:20 +00:00
|
|
|
/* auto-renwal mode. input watching is requested all the time.
|
|
|
|
* output watching is requested only if there're enqueued
|
|
|
|
* data for writing. */
|
2018-12-12 13:15:54 +00:00
|
|
|
events = MIO_DEV_EVENT_IN;
|
|
|
|
if (!MIO_WQ_ISEMPTY(&dev->wq)) events |= MIO_DEV_EVENT_OUT;
|
2016-04-17 17:00:58 +00:00
|
|
|
/* fall through */
|
2018-12-12 13:15:54 +00:00
|
|
|
case MIO_DEV_WATCH_UPDATE:
|
2016-02-04 15:06:20 +00:00
|
|
|
/* honor event watching requests as given by the caller */
|
2016-04-24 17:30:43 +00:00
|
|
|
mux_cmd = MUX_CMD_UPDATE;
|
2016-01-26 16:02:20 +00:00
|
|
|
break;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
case MIO_DEV_WATCH_STOP:
|
2016-02-04 15:06:20 +00:00
|
|
|
events = 0; /* override events */
|
2016-04-24 17:30:43 +00:00
|
|
|
mux_cmd = MUX_CMD_DELETE;
|
2016-01-26 16:02:20 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_EINVAL;
|
2016-01-26 16:02:20 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
dev_capa = dev->dev_capa;
|
|
|
|
dev_capa &= ~(DEV_CAPA_ALL_WATCHED);
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
/* this function honors MIO_DEV_EVENT_IN and MIO_DEV_EVENT_OUT only
|
2016-02-04 15:06:20 +00:00
|
|
|
* as valid input event bits. it intends to provide simple abstraction
|
|
|
|
* by reducing the variety of event bits that the caller has to handle. */
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if ((events & MIO_DEV_EVENT_IN) && !(dev->dev_capa & (MIO_DEV_CAPA_IN_CLOSED | MIO_DEV_CAPA_IN_DISABLED)))
|
2016-01-26 16:02:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_IN)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_PRI) dev_capa |= MIO_DEV_CAPA_PRI_WATCHED;
|
|
|
|
dev_capa |= MIO_DEV_CAPA_IN_WATCHED;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-01-26 16:02:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if ((events & MIO_DEV_EVENT_OUT) && !(dev->dev_capa & MIO_DEV_CAPA_OUT_CLOSED))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_OUT) dev_capa |= MIO_DEV_CAPA_OUT_WATCHED;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
if (mux_cmd == MUX_CMD_UPDATE && (dev_capa & DEV_CAPA_ALL_WATCHED) == (dev->dev_capa & DEV_CAPA_ALL_WATCHED))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2016-04-18 14:21:23 +00:00
|
|
|
/* no change in the device capacity. skip calling epoll_ctl */
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-01-14 08:33:14 +00:00
|
|
|
if (mux_control(dev, mux_cmd, dev->dev_mth->getsyshnd(dev), dev_capa) <= -1) return -1;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
dev->dev_capa = dev_capa;
|
2016-01-26 16:02:20 +00:00
|
|
|
return 0;
|
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2019-01-17 08:09:19 +00:00
|
|
|
static void on_read_timeout (mio_t* mio, const mio_ntime_t* now, mio_tmrjob_t* job)
|
|
|
|
{
|
|
|
|
mio_dev_t* dev;
|
|
|
|
int x;
|
|
|
|
|
|
|
|
dev = (mio_dev_t*)job->ctx;
|
|
|
|
|
|
|
|
dev->mio->errnum = MIO_ETMOUT;
|
|
|
|
x = dev->dev_evcb->on_read(dev, MIO_NULL, -1, MIO_NULL);
|
|
|
|
|
|
|
|
MIO_ASSERT (dev->rtmridx == MIO_TMRIDX_INVALID);
|
|
|
|
|
|
|
|
if (x <= -1) mio_dev_halt (dev);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int __dev_read (mio_dev_t* dev, int enabled, const mio_ntime_t* tmout, void* rdctx)
|
|
|
|
{
|
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_IN_CLOSED)
|
|
|
|
{
|
|
|
|
dev->mio->errnum = MIO_ENOCAPA;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (enabled)
|
|
|
|
{
|
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_IN_DISABLED;
|
|
|
|
if (!dev->mio->in_exec && (dev->dev_capa & MIO_DEV_CAPA_IN_WATCHED)) goto renew_watch_now;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_IN_DISABLED;
|
|
|
|
if (!dev->mio->in_exec && !(dev->dev_capa & MIO_DEV_CAPA_IN_WATCHED)) goto renew_watch_now;
|
|
|
|
}
|
|
|
|
|
|
|
|
dev->mio->renew_watch = 1;
|
|
|
|
goto update_timer;
|
|
|
|
|
|
|
|
renew_watch_now:
|
|
|
|
if (mio_dev_watch(dev, MIO_DEV_WATCH_RENEW, 0) <= -1) return -1;
|
|
|
|
goto update_timer;
|
|
|
|
|
|
|
|
update_timer:
|
|
|
|
if (dev->rtmridx != MIO_TMRIDX_INVALID)
|
|
|
|
{
|
|
|
|
/* read timeout already on the socket. remove it first */
|
|
|
|
mio_deltmrjob (dev->mio, dev->rtmridx);
|
|
|
|
dev->rtmridx = MIO_TMRIDX_INVALID;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (tmout && mio_ispostime(tmout))
|
|
|
|
{
|
|
|
|
mio_tmrjob_t tmrjob;
|
|
|
|
|
|
|
|
MIO_MEMSET (&tmrjob, 0, MIO_SIZEOF(tmrjob));
|
|
|
|
tmrjob.ctx = dev;
|
|
|
|
mio_gettime (&tmrjob.when);
|
|
|
|
mio_addtime (&tmrjob.when, tmout, &tmrjob.when);
|
|
|
|
tmrjob.handler = on_read_timeout;
|
|
|
|
tmrjob.idxptr = &dev->rtmridx;
|
|
|
|
|
|
|
|
dev->rtmridx = mio_instmrjob(dev->mio, &tmrjob);
|
|
|
|
if (dev->rtmridx == MIO_TMRIDX_INVALID)
|
|
|
|
{
|
|
|
|
/* if timer registration fails, timeout will never be triggered */
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_dev_read (mio_dev_t* dev, int enabled)
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2019-01-17 08:09:19 +00:00
|
|
|
return __dev_read(dev, enabled, MIO_NULL, MIO_NULL);
|
|
|
|
|
|
|
|
#if 0
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_IN_CLOSED)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ENOCAPA;
|
2016-02-05 13:25:59 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2016-02-04 15:06:20 +00:00
|
|
|
if (enabled)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa &= ~MIO_DEV_CAPA_IN_DISABLED;
|
|
|
|
if (!dev->mio->in_exec && (dev->dev_capa & MIO_DEV_CAPA_IN_WATCHED)) goto renew_watch_now;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
else
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_IN_DISABLED;
|
|
|
|
if (!dev->mio->in_exec && !(dev->dev_capa & MIO_DEV_CAPA_IN_WATCHED)) goto renew_watch_now;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->renew_watch = 1;
|
2016-02-05 13:25:59 +00:00
|
|
|
return 0;
|
|
|
|
|
|
|
|
renew_watch_now:
|
2019-01-14 08:33:14 +00:00
|
|
|
if (mio_dev_watch(dev, MIO_DEV_WATCH_RENEW, 0) <= -1) return -1;
|
2016-02-05 13:25:59 +00:00
|
|
|
return 0;
|
2019-01-17 08:09:19 +00:00
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int mio_dev_timedread (mio_dev_t* dev, int enabled, const mio_ntime_t* tmout)
|
|
|
|
{
|
|
|
|
return __dev_read(dev, enabled, tmout, MIO_NULL);
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static void on_write_timeout (mio_t* mio, const mio_ntime_t* now, mio_tmrjob_t* job)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_wq_t* q;
|
|
|
|
mio_dev_t* dev;
|
2016-02-05 13:25:59 +00:00
|
|
|
int x;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
q = (mio_wq_t*)job->ctx;
|
2016-02-05 13:25:59 +00:00
|
|
|
dev = q->dev;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ETMOUT;
|
2019-01-14 08:33:14 +00:00
|
|
|
x = dev->dev_evcb->on_write(dev, -1, q->ctx, &q->dstaddr);
|
2016-02-05 13:25:59 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_ASSERT (q->tmridx == MIO_TMRIDX_INVALID);
|
|
|
|
MIO_WQ_UNLINK(q);
|
|
|
|
MIO_MMGR_FREE (mio->mmgr, q);
|
2016-02-05 13:25:59 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (x <= -1) mio_dev_halt (dev);
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
static int __dev_write (mio_dev_t* dev, const void* data, mio_iolen_t len, const mio_ntime_t* tmout, void* wrctx, const mio_devaddr_t* dstaddr)
|
2016-01-28 16:44:47 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
const mio_uint8_t* uptr;
|
|
|
|
mio_iolen_t urem, ulen;
|
|
|
|
mio_wq_t* q;
|
2019-01-14 08:33:14 +00:00
|
|
|
mio_cwq_t* cwq;
|
2019-01-16 04:37:53 +00:00
|
|
|
mio_oow_t cwq_extra_aligned, cwqfl_index;
|
2016-02-04 15:06:20 +00:00
|
|
|
int x;
|
2016-02-02 07:51:17 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_OUT_CLOSED)
|
2016-02-02 07:51:17 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ENOCAPA;
|
2016-02-02 07:51:17 +00:00
|
|
|
return -1;
|
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
|
|
|
uptr = data;
|
|
|
|
urem = len;
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (!MIO_WQ_ISEMPTY(&dev->wq))
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
|
|
|
/* the writing queue is not empty.
|
|
|
|
* enqueue this request immediately */
|
|
|
|
goto enqueue_data;
|
|
|
|
}
|
2016-02-02 07:51:17 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (dev->dev_capa & MIO_DEV_CAPA_STREAM)
|
2016-01-28 16:44:47 +00:00
|
|
|
{
|
2016-02-05 13:25:59 +00:00
|
|
|
/* use the do..while() loop to be able to send a zero-length data */
|
|
|
|
do
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2016-02-05 13:25:59 +00:00
|
|
|
ulen = urem;
|
2019-01-14 08:33:14 +00:00
|
|
|
x = dev->dev_mth->write(dev, data, &ulen, dstaddr);
|
2016-02-05 13:25:59 +00:00
|
|
|
if (x <= -1) return -1;
|
|
|
|
else if (x == 0)
|
|
|
|
{
|
|
|
|
/* [NOTE]
|
|
|
|
* the write queue is empty at this moment. a zero-length
|
2019-01-11 07:35:43 +00:00
|
|
|
* request for a stream device can still get enqueued if the
|
2016-02-05 13:25:59 +00:00
|
|
|
* write callback returns 0 though i can't figure out if there
|
|
|
|
* is a compelling reason to do so
|
|
|
|
*/
|
|
|
|
goto enqueue_data; /* enqueue remaining data */
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
urem -= ulen;
|
|
|
|
uptr += ulen;
|
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-02-05 13:25:59 +00:00
|
|
|
while (urem > 0);
|
|
|
|
|
|
|
|
if (len <= 0) /* original length */
|
2016-01-28 16:44:47 +00:00
|
|
|
{
|
2016-02-05 13:25:59 +00:00
|
|
|
/* a zero-length writing request is to close the writing end */
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->dev_capa |= MIO_DEV_CAPA_OUT_CLOSED;
|
2016-01-28 16:44:47 +00:00
|
|
|
}
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2019-01-16 04:37:53 +00:00
|
|
|
/* if i trigger the write completion callback here, the performance
|
|
|
|
* may increase, but there can be annoying recursion issues if the
|
|
|
|
* callback requests another writing operation. it's imperative to
|
|
|
|
* delay the callback until this write function is finished.
|
|
|
|
* ---> if (dev->dev_evcb->on_write(dev, len, wrctx, dstaddr) <= -1) return -1; */
|
2019-01-14 08:33:14 +00:00
|
|
|
goto enqueue_completed_write;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
else
|
2016-02-04 15:06:20 +00:00
|
|
|
{
|
2016-02-05 13:25:59 +00:00
|
|
|
ulen = urem;
|
2016-04-02 15:25:35 +00:00
|
|
|
|
2016-04-18 14:21:23 +00:00
|
|
|
x = dev->dev_mth->write (dev, data, &ulen, dstaddr);
|
2016-02-05 13:25:59 +00:00
|
|
|
if (x <= -1) return -1;
|
|
|
|
else if (x == 0) goto enqueue_data;
|
|
|
|
|
2019-01-16 04:37:53 +00:00
|
|
|
/* partial writing is still considered ok for a non-stream device. */
|
|
|
|
|
|
|
|
/* read the comment in the 'if' block above for why i enqueue the write completion event
|
|
|
|
* instead of calling the event callback here...
|
|
|
|
* --> if (dev->dev_evcb->on_write(dev, ulen, wrctx, dstaddr) <= -1) return -1; */
|
2019-01-14 08:33:14 +00:00
|
|
|
goto enqueue_completed_write;
|
2016-02-04 15:06:20 +00:00
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2016-02-05 13:25:59 +00:00
|
|
|
return 1; /* written immediately and called on_write callback */
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2016-02-02 07:51:17 +00:00
|
|
|
enqueue_data:
|
2018-12-12 13:15:54 +00:00
|
|
|
if (!(dev->dev_capa & MIO_DEV_CAPA_OUT_QUEUED))
|
2016-03-29 15:02:01 +00:00
|
|
|
{
|
|
|
|
/* writing queuing is not requested. so return failure */
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ENOCAPA;
|
2016-03-29 15:02:01 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2016-02-02 07:51:17 +00:00
|
|
|
/* queue the remaining data*/
|
2019-01-14 08:33:14 +00:00
|
|
|
q = (mio_wq_t*)MIO_MMGR_ALLOC(dev->mio->mmgr, MIO_SIZEOF(*q) + (dstaddr? dstaddr->len: 0) + urem);
|
2016-02-02 07:51:17 +00:00
|
|
|
if (!q)
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->errnum = MIO_ENOMEM;
|
2016-02-02 07:51:17 +00:00
|
|
|
return -1;
|
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
q->tmridx = MIO_TMRIDX_INVALID;
|
2016-02-05 13:25:59 +00:00
|
|
|
q->dev = dev;
|
2016-02-02 15:40:09 +00:00
|
|
|
q->ctx = wrctx;
|
2016-03-29 15:02:01 +00:00
|
|
|
|
2016-04-18 14:21:23 +00:00
|
|
|
if (dstaddr)
|
2016-03-29 15:02:01 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
q->dstaddr.ptr = (mio_uint8_t*)(q + 1);
|
2016-04-18 14:21:23 +00:00
|
|
|
q->dstaddr.len = dstaddr->len;
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMCPY (q->dstaddr.ptr, dstaddr->ptr, dstaddr->len);
|
2016-03-29 15:02:01 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2016-04-18 14:21:23 +00:00
|
|
|
q->dstaddr.len = 0;
|
2016-03-29 15:02:01 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
q->ptr = (mio_uint8_t*)(q + 1) + q->dstaddr.len;
|
2016-02-02 07:51:17 +00:00
|
|
|
q->len = urem;
|
2016-02-05 13:25:59 +00:00
|
|
|
q->olen = len;
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMCPY (q->ptr, uptr, urem);
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
if (tmout && mio_ispostime(tmout))
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_tmrjob_t tmrjob;
|
2016-02-04 15:06:20 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MEMSET (&tmrjob, 0, MIO_SIZEOF(tmrjob));
|
2016-02-05 13:25:59 +00:00
|
|
|
tmrjob.ctx = q;
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_gettime (&tmrjob.when);
|
|
|
|
mio_addtime (&tmrjob.when, tmout, &tmrjob.when);
|
2016-02-05 13:25:59 +00:00
|
|
|
tmrjob.handler = on_write_timeout;
|
|
|
|
tmrjob.idxptr = &q->tmridx;
|
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
q->tmridx = mio_instmrjob(dev->mio, &tmrjob);
|
2018-12-12 13:15:54 +00:00
|
|
|
if (q->tmridx == MIO_TMRIDX_INVALID)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_MMGR_FREE (dev->mio->mmgr, q);
|
2016-02-05 13:25:59 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
MIO_WQ_ENQ (&dev->wq, q);
|
|
|
|
if (!dev->mio->in_exec && !(dev->dev_capa & MIO_DEV_CAPA_OUT_WATCHED))
|
2016-02-02 07:51:17 +00:00
|
|
|
{
|
2016-02-04 15:06:20 +00:00
|
|
|
/* if output is not being watched, arrange to do so */
|
2019-01-11 07:35:43 +00:00
|
|
|
if (mio_dev_watch(dev, MIO_DEV_WATCH_RENEW, 0) <= -1)
|
2016-01-28 16:44:47 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
unlink_wq (dev->mio, q);
|
|
|
|
MIO_MMGR_FREE (dev->mio->mmgr, q);
|
2016-02-02 07:51:17 +00:00
|
|
|
return -1;
|
2016-01-28 16:44:47 +00:00
|
|
|
}
|
|
|
|
}
|
2016-02-05 13:25:59 +00:00
|
|
|
else
|
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
dev->mio->renew_watch = 1;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
2016-01-28 16:44:47 +00:00
|
|
|
|
2016-02-05 13:25:59 +00:00
|
|
|
return 0; /* request pused to a write queue. */
|
2019-01-14 08:33:14 +00:00
|
|
|
|
|
|
|
enqueue_completed_write:
|
|
|
|
/* queue the remaining data*/
|
2019-01-16 04:37:53 +00:00
|
|
|
cwq_extra_aligned = (dstaddr? dstaddr->len: 0);
|
|
|
|
cwq_extra_aligned = MIO_ALIGN_POW2(cwq_extra_aligned, MIO_CWQFL_ALIGN);
|
|
|
|
cwqfl_index = cwq_extra_aligned / MIO_CWQFL_SIZE;
|
|
|
|
|
|
|
|
if (cwqfl_index < MIO_COUNTOF(dev->mio->cwqfl) && dev->mio->cwqfl[cwqfl_index])
|
2019-01-14 08:33:14 +00:00
|
|
|
{
|
2019-01-16 04:37:53 +00:00
|
|
|
/* take an available cwq object from the free cwq list */
|
|
|
|
cwq = dev->mio->cwqfl[cwqfl_index];
|
|
|
|
dev->mio->cwqfl[cwqfl_index] = cwq->next;
|
2019-01-14 09:23:16 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-01-16 04:37:53 +00:00
|
|
|
cwq = (mio_cwq_t*)MIO_MMGR_ALLOC(dev->mio->mmgr, MIO_SIZEOF(*cwq) + cwq_extra_aligned);
|
2019-01-14 09:23:16 +00:00
|
|
|
if (!cwq)
|
|
|
|
{
|
|
|
|
dev->mio->errnum = MIO_ENOMEM;
|
|
|
|
return -1;
|
|
|
|
}
|
2019-01-14 08:33:14 +00:00
|
|
|
}
|
|
|
|
|
2019-01-14 09:23:16 +00:00
|
|
|
MIO_MEMSET (cwq, 0, MIO_SIZEOF(*cwq));
|
2019-01-14 08:33:14 +00:00
|
|
|
cwq->dev = dev;
|
|
|
|
cwq->ctx = wrctx;
|
|
|
|
if (dstaddr)
|
|
|
|
{
|
|
|
|
cwq->dstaddr.ptr = (mio_uint8_t*)(cwq + 1);
|
|
|
|
cwq->dstaddr.len = dstaddr->len;
|
|
|
|
MIO_MEMCPY (cwq->dstaddr.ptr, dstaddr->ptr, dstaddr->len);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
cwq->dstaddr.len = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
cwq->olen = len;
|
|
|
|
|
|
|
|
MIO_CWQ_ENQ (&dev->mio->cwq, cwq);
|
2019-01-14 08:45:56 +00:00
|
|
|
dev->cw_count++; /* increment the number of complete write operations */
|
2019-01-14 08:33:14 +00:00
|
|
|
return 0;
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_dev_write (mio_dev_t* dev, const void* data, mio_iolen_t len, void* wrctx, const mio_devaddr_t* dstaddr)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2019-01-11 07:35:43 +00:00
|
|
|
return __dev_write(dev, data, len, MIO_NULL, wrctx, dstaddr);
|
2016-02-05 13:25:59 +00:00
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_dev_timedwrite (mio_dev_t* dev, const void* data, mio_iolen_t len, const mio_ntime_t* tmout, void* wrctx, const mio_devaddr_t* dstaddr)
|
2016-02-05 13:25:59 +00:00
|
|
|
{
|
2019-01-11 07:35:43 +00:00
|
|
|
return __dev_write(dev, data, len, tmout, wrctx, dstaddr);
|
2016-01-28 16:44:47 +00:00
|
|
|
}
|
2016-01-31 02:05:39 +00:00
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
int mio_makesyshndasync (mio_t* mio, mio_syshnd_t hnd)
|
2016-02-02 15:40:09 +00:00
|
|
|
{
|
|
|
|
#if defined(F_GETFL) && defined(F_SETFL) && defined(O_NONBLOCK)
|
|
|
|
int flags;
|
|
|
|
|
2019-01-11 07:35:43 +00:00
|
|
|
if ((flags = fcntl(hnd, F_GETFL)) <= -1 ||
|
|
|
|
(flags = fcntl(hnd, F_SETFL, flags | O_NONBLOCK)) <= -1)
|
2016-02-02 15:40:09 +00:00
|
|
|
{
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = mio_syserrtoerrnum (errno);
|
2016-02-02 15:40:09 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
#else
|
2018-12-12 13:15:54 +00:00
|
|
|
mio->errnum = MIO_ENOSUP;
|
2016-02-02 15:40:09 +00:00
|
|
|
return -1;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2018-12-12 13:15:54 +00:00
|
|
|
mio_errnum_t mio_syserrtoerrnum (int no)
|
2016-01-31 02:05:39 +00:00
|
|
|
{
|
|
|
|
switch (no)
|
|
|
|
{
|
|
|
|
case ENOMEM:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ENOMEM;
|
2016-01-31 02:05:39 +00:00
|
|
|
|
|
|
|
case EINVAL:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EINVAL;
|
2016-01-31 02:05:39 +00:00
|
|
|
|
2016-04-24 17:30:43 +00:00
|
|
|
case EEXIST:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EEXIST;
|
2016-04-24 17:30:43 +00:00
|
|
|
|
2016-01-31 02:05:39 +00:00
|
|
|
case ENOENT:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ENOENT;
|
2016-01-31 02:05:39 +00:00
|
|
|
|
|
|
|
case EMFILE:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EMFILE;
|
2016-01-31 02:05:39 +00:00
|
|
|
|
|
|
|
#if defined(ENFILE)
|
|
|
|
case ENFILE:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ENFILE;
|
2016-01-31 02:05:39 +00:00
|
|
|
#endif
|
|
|
|
|
2016-04-18 14:21:23 +00:00
|
|
|
#if defined(EWOULDBLOCK) && defined(EAGAIN) && (EWOULDBLOCK != EAGAIN)
|
2016-04-16 16:05:57 +00:00
|
|
|
case EAGAIN:
|
2016-04-18 14:21:23 +00:00
|
|
|
case EWOULDBLOCK:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EAGAIN;
|
2016-04-18 14:21:23 +00:00
|
|
|
#elif defined(EAGAIN)
|
|
|
|
case EAGAIN:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EAGAIN;
|
2016-04-18 14:21:23 +00:00
|
|
|
#elif defined(EWOULDBLOCK)
|
|
|
|
case EWOULDBLOCK:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EAGAIN;
|
2016-04-16 16:05:57 +00:00
|
|
|
#endif
|
|
|
|
|
2016-02-01 07:42:49 +00:00
|
|
|
#if defined(ECONNREFUSED)
|
|
|
|
case ECONNREFUSED:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ECONRF;
|
2016-02-01 07:42:49 +00:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#if defined(ECONNRESETD)
|
|
|
|
case ECONNRESET:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ECONRS;
|
2016-02-01 07:42:49 +00:00
|
|
|
#endif
|
|
|
|
|
2016-04-21 15:07:58 +00:00
|
|
|
#if defined(EPERM)
|
|
|
|
case EPERM:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_EPERM;
|
2016-04-21 15:07:58 +00:00
|
|
|
#endif
|
|
|
|
|
2016-01-31 02:05:39 +00:00
|
|
|
default:
|
2018-12-12 13:15:54 +00:00
|
|
|
return MIO_ESYSERR;
|
2016-01-31 02:05:39 +00:00
|
|
|
}
|
|
|
|
}
|