--- libaitsched/src/tasks.c 2012/08/01 14:11:43 1.10.2.3 +++ libaitsched/src/tasks.c 2012/08/02 12:58:02 1.10.2.10 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: tasks.c,v 1.10.2.3 2012/08/01 14:11:43 misho Exp $ +* $Id: tasks.c,v 1.10.2.10 2012/08/02 12:58:02 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -482,6 +482,7 @@ schedAlarm(sched_root_task_t * __restrict root, sched_ return task; } +#ifdef AIO_SUPPORT /* * schedAIO() - Add AIO task to scheduler queue * @@ -497,14 +498,10 @@ sched_task_t * schedAIO(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, struct aiocb * __restrict acb, void *opt_data, size_t opt_dlen) { -#ifndef EVFILT_AIO - sched_SetErr(ENOTSUP, "Not supported kevent() filter"); - return NULL; -#else sched_task_t *task; void *ptr; - if (!root || !func) + if (!root || !func || !acb || !opt_dlen) return NULL; /* get new task */ @@ -538,7 +535,6 @@ schedAIO(sched_root_task_t * __restrict root, sched_ta task = _sched_unuseTask(task); return task; -#endif } /* @@ -550,24 +546,28 @@ schedAIO(sched_root_task_t * __restrict root, sched_ta * @fd = file descriptor * @buffer = Buffer * @buflen = Buffer length + * @offset = Offset from start of file, if =-1 from current position * return: NULL error or !=NULL new queued task */ inline sched_task_t * schedAIORead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd, - void *buffer, size_t buflen) + void *buffer, size_t buflen, off_t offset) { -#ifndef EVFILT_AIO - sched_SetErr(ENOTSUP, "Not supported kevent() filter"); - return NULL; -#else struct aiocb *acb; - off_t off = 0; + off_t off; - if (!root || !func) + if (!root || !func || !buffer || !buflen) return NULL; - else - memset(buffer, 0, buflen); + if (offset == (off_t) -1) { + off = lseek(fd, 0, SEEK_CUR); + if (off == -1) { + LOGERR; + return NULL; + } + } else + off = offset; + if (!(acb = malloc(sizeof(struct aiocb)))) { LOGERR; return NULL; @@ -577,17 +577,10 @@ schedAIORead(sched_root_task_t * __restrict root, sche acb->aio_fildes = fd; acb->aio_nbytes = buflen; acb->aio_buf = buffer; - off = lseek(fd, 0, SEEK_CUR); - if (off == -1) { - LOGERR; - free(acb); - return NULL; - } else - acb->aio_offset = off; + acb->aio_offset = off; acb->aio_sigevent.sigev_notify = SIGEV_KEVENT; - acb->aio_sigevent.sigev_notify_kevent_flags = EV_CLEAR | EV_DISPATCH; acb->aio_sigevent.sigev_notify_kqueue = root->root_kq; - acb->aio_sigevent.sigev_value.sival_ptr = (void*) O_RDONLY; + acb->aio_sigevent.sigev_value.sival_ptr = acb; if (aio_read(acb)) { LOGERR; @@ -596,7 +589,6 @@ schedAIORead(sched_root_task_t * __restrict root, sche } return schedAIO(root, func, arg, acb, buffer, buflen); -#endif } /* @@ -608,24 +600,28 @@ schedAIORead(sched_root_task_t * __restrict root, sche * @fd = file descriptor * @buffer = Buffer * @buflen = Buffer length + * @offset = Offset from start of file, if =-1 from current position * return: NULL error or !=NULL new queued task */ inline sched_task_t * schedAIOWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd, - void *buffer, size_t buflen) + void *buffer, size_t buflen, off_t offset) { -#ifndef EVFILT_AIO - sched_SetErr(ENOTSUP, "Not supported kevent() filter"); - return NULL; -#else struct aiocb *acb; - off_t off = 0; + off_t off; - if (!root || !func) + if (!root || !func || !buffer || !buflen) return NULL; - else - memset(buffer, 0, buflen); + if (offset == (off_t) -1) { + off = lseek(fd, 0, SEEK_CUR); + if (off == -1) { + LOGERR; + return NULL; + } + } else + off = offset; + if (!(acb = malloc(sizeof(struct aiocb)))) { LOGERR; return NULL; @@ -635,17 +631,10 @@ schedAIOWrite(sched_root_task_t * __restrict root, sch acb->aio_fildes = fd; acb->aio_nbytes = buflen; acb->aio_buf = buffer; - off = lseek(fd, 0, SEEK_CUR); - if (off == -1) { - LOGERR; - free(acb); - return NULL; - } else - acb->aio_offset = off; + acb->aio_offset = off; acb->aio_sigevent.sigev_notify = SIGEV_KEVENT; - acb->aio_sigevent.sigev_notify_kevent_flags = EV_CLEAR | EV_DISPATCH; acb->aio_sigevent.sigev_notify_kqueue = root->root_kq; - acb->aio_sigevent.sigev_value.sival_ptr = (void*) O_WRONLY; + acb->aio_sigevent.sigev_value.sival_ptr = acb; if (aio_write(acb)) { LOGERR; @@ -654,8 +643,146 @@ schedAIOWrite(sched_root_task_t * __restrict root, sch } return schedAIO(root, func, arg, acb, buffer, buflen); -#endif } + +#ifdef EVFILT_LIO +/* + * schedLIORead() - Add list of AIO read tasks to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @fd = file descriptor + * @bufs = Buffer's list + * @nbufs = Number of Buffers + * @offset = Offset from start of file, if =-1 from current position + * return: NULL error or !=NULL new queued task + */ +sched_task_t * +schedLIORead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd, + struct iovec *bufs, size_t nbufs, off_t offset) +{ + struct sigevent sig; + struct aiocb **acb; + off_t off; + register int i; + + if (!root || !func || !bufs || !nbufs) + return NULL; + + if (offset == (off_t) -1) { + off = lseek(fd, 0, SEEK_CUR); + if (off == -1) { + LOGERR; + return NULL; + } + } else + off = offset; + + if (!(acb = calloc(sizeof(void*), nbufs))) { + LOGERR; + return NULL; + } else + memset(acb, 0, sizeof(void*) * nbufs); + for (i = 0; i < nbufs; off += bufs[i++].iov_len) { + acb[i] = malloc(sizeof(struct aiocb)); + if (!acb[i]) { + LOGERR; + for (i = 0; i < nbufs; i++) + if (acb[i]) + free(acb[i]); + free(acb); + return NULL; + } else + memset(acb[i], 0, sizeof(struct aiocb)); + acb[i]->aio_fildes = fd; + acb[i]->aio_nbytes = bufs[i].iov_len; + acb[i]->aio_buf = bufs[i].iov_base; + acb[i]->aio_offset = off; + acb[i]->aio_lio_opcode = LIO_READ; + } + memset(&sig, 0, sizeof sig); + sig.sigev_notify = SIGEV_KEVENT; + sig.sigev_notify_kqueue = root->root_kq; + sig.sigev_value.sival_ptr = acb; + + if (lio_listio(LIO_NOWAIT, acb, nbufs, &sig)) { + LOGERR; + return NULL; + } + + return schedAIO(root, func, arg, (void*) acb, bufs, nbufs); +} + +/* + * schedLIOWrite() - Add list of AIO write tasks to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @fd = file descriptor + * @bufs = Buffer's list + * @nbufs = Number of Buffers + * @offset = Offset from start of file, if =-1 from current position + * return: NULL error or !=NULL new queued task + */ +inline sched_task_t * +schedLIOWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd, + struct iovec *bufs, size_t nbufs, off_t offset) +{ + struct sigevent sig; + struct aiocb **acb; + off_t off; + register int i; + + if (!root || !func || !bufs || !nbufs) + return NULL; + + if (offset == (off_t) -1) { + off = lseek(fd, 0, SEEK_CUR); + if (off == -1) { + LOGERR; + return NULL; + } + } else + off = offset; + + if (!(acb = calloc(sizeof(void*), nbufs))) { + LOGERR; + return NULL; + } else + memset(acb, 0, sizeof(void*) * nbufs); + for (i = 0; i < nbufs; off += bufs[i++].iov_len) { + acb[i] = malloc(sizeof(struct aiocb)); + if (!acb[i]) { + LOGERR; + for (i = 0; i < nbufs; i++) + if (acb[i]) + free(acb[i]); + free(acb); + return NULL; + } else + memset(acb[i], 0, sizeof(struct aiocb)); + acb[i]->aio_fildes = fd; + acb[i]->aio_nbytes = bufs[i].iov_len; + acb[i]->aio_buf = bufs[i].iov_base; + acb[i]->aio_offset = off; + acb[i]->aio_lio_opcode = LIO_WRITE; + } + memset(&sig, 0, sizeof sig); + sig.sigev_notify = SIGEV_KEVENT; + sig.sigev_notify_kqueue = root->root_kq; + sig.sigev_value.sival_ptr = acb; + + if (lio_listio(LIO_NOWAIT, acb, nbufs, &sig)) { + LOGERR; + return NULL; + } + + return schedAIO(root, func, arg, (void*) acb, bufs, nbufs); +} +#endif /* EVFILT_LIO */ +#endif /* AIO_SUPPORT */ /* * schedTimer() - Add TIMER task to scheduler queue