--- libaitsched/inc/aitsched.h 2012/05/15 15:47:12 1.7.2.2 +++ libaitsched/inc/aitsched.h 2012/08/02 13:56:19 1.12 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: aitsched.h,v 1.7.2.2 2012/05/15 15:47:12 misho Exp $ +* $Id: aitsched.h,v 1.12 2012/08/02 13:56:19 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -49,17 +49,24 @@ SUCH DAMAGE. #include #include +#include #include #include #include +#ifdef EVFILT_LIO +#include +#endif /* criteria type */ -#define CRITERIA_CALL 0 -#define CRITERIA_ARG 1 -#define CRITERIA_FD 2 -#define CRITERIA_VAL 3 -#define CRITERIA_TS 4 +#define CRITERIA_ANY 0 +#define CRITERIA_CALL 1 +#define CRITERIA_ARG 2 +#define CRITERIA_FD 3 +#define CRITERIA_VAL 4 +#define CRITERIA_TS 5 +#define CRITERIA_DATA 6 +#define CRITERIA_ID 7 /* early declaration for root & task */ @@ -70,8 +77,16 @@ typedef enum { taskREAD = 0, taskWRITE, taskTIMER, - taskEVENT, - taskEVENTLO, + taskALARM, + taskNODE, + taskPROC, + taskSIGNAL, + taskAIO, + taskLIO, + taskUSER, + taskEVENT, + taskEVENTLO, + taskSUSPEND, taskREADY, taskUNUSE, taskMAX @@ -85,16 +100,34 @@ struct sched_HooksTask { sched_hook_func_t read; /* write(sched_task_t *task, NULL) -> int */ sched_hook_func_t write; + /* timer(sched_task_t *task, struct timespec *ts) -> int */ + sched_hook_func_t timer; + /* alarm(sched_task_t *task, NULL) -> int */ + sched_hook_func_t alarm; + /* node(sched_task_t *task, NULL) -> int */ + sched_hook_func_t node; + /* proc(sched_task_t *task, NULL) -> int */ + sched_hook_func_t proc; + /* signal(sched_task_t *task, NULL) -> int */ + sched_hook_func_t signal; + /* aio(sched_task_t *task, NULL) -> int */ + sched_hook_func_t aio; + /* lio(sched_task_t *task, NULL) -> int */ + sched_hook_func_t lio; + /* user(sched_task_t *task, NULL) -> int */ + sched_hook_func_t user; /* event(sched_task_t *task, NULL) -> int */ sched_hook_func_t event; /* eventlo(sched_task_t *task, NULL) -> int */ sched_hook_func_t eventlo; - /* timer(sched_task_t *task, struct timespec *ts) -> int */ - sched_hook_func_t timer; + /* suspend(sched_task_t *task, NULL) -> int */ + sched_hook_func_t suspend; } hook_add; struct { /* cancel(sched_task_t *task, NULL) -> int */ sched_hook_func_t cancel; + /* resume(sched_task_t *task, NULL) -> int */ + sched_hook_func_t resume; /* run(sched_root_task_t *root, NULL) -> int */ sched_hook_func_t run; /* fetch(sched_root_task_t *root, NULL) -> sched_task_t* */ @@ -119,14 +152,15 @@ typedef struct sched_HooksTask hooks_task_t; typedef void *(*sched_task_func_t)(sched_task_t * /* current task data*/); /* task lock helpers */ -#define TASK_LOCK(x) ((x)->task_lock++) +#define TASK_LOCK(x) ((x)->task_lock = 42) #define TASK_UNLOCK(x) ((x)->task_lock ^= (x)->task_lock) #define TASK_ISLOCKED(x) ((x)->task_lock) /* task & queue */ struct sched_Task { volatile int task_lock; - unsigned int task_id; + uintptr_t task_id; +#define TASK_ID(x) ((struct sched_Task*) (x)->task_id) sched_task_type_t task_type; #define TASK_TYPE(x) (x)->task_type @@ -166,16 +200,25 @@ struct sched_RootTask { struct timespec root_wait; struct timespec root_poll; intptr_t root_cond; + void *root_ret; pthread_mutex_t root_mtx[taskMAX]; sched_queue_t root_read; sched_queue_t root_write; sched_queue_t root_timer; + sched_queue_t root_alarm; + sched_queue_t root_node; + sched_queue_t root_proc; + sched_queue_t root_signal; + sched_queue_t root_aio; + sched_queue_t root_lio; + sched_queue_t root_user; sched_queue_t root_event; + sched_queue_t root_eventlo; + sched_queue_t root_suspend; sched_queue_t root_ready; sched_queue_t root_unuse; - sched_queue_t root_eventlo; int root_eventlo_miss; hooks_task_t root_hooks; @@ -184,6 +227,7 @@ struct sched_RootTask { #define ROOT_DATLEN(x) (x)->root_data.iov_len }; #define ROOT_QUEUE_EMPTY(x, _q) TAILQ_EMPTY(&((x)->root_##_q)) +#define ROOT_RETURN(x) (x)->root_ret inline int sched_GetErrno(); @@ -265,7 +309,8 @@ int schedCancel(sched_task_t * __restrict task); * * @root = root task * @type = cancel from queue type, if =taskMAX cancel same task from all queues - * @criteria = find task by criteria [CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_TS] + * @criteria = find task by criteria + * [CRITERIA_ANY|CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_ID|CRITERIA_TS|CRITERIA_DATA] * @param = search parameter * @hook = custom cleanup hook function, may be NULL * return: -1 error, -2 error in sub-stage cancel execution, -3 error from custom hook or 0 ok @@ -305,6 +350,176 @@ sched_task_t *schedWrite(sched_root_task_t * __restric #define schedWriteSelf(x) schedWrite(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ TASK_FD((x)), TASK_DATA((x)), TASK_DATLEN((x))) /* + * schedAlarm() - Add ALARM task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @ts = timeout argument structure, minimum alarm timer resolution is 1msec! + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedAlarm(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + struct timespec ts, void *opt_data, size_t opt_dlen); +#define schedAlarmSelf(x) schedAlarm(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_TS((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedNode() - Add NODE task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @fd = fd handle + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedNode(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + int fd, void *opt_data, size_t opt_dlen); +#define schedNodeSelf(x) schedNode(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_FD((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedProc() - Add PROC task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @pid = PID + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedProc(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + unsigned long pid, void *opt_data, size_t opt_dlen); +#define schedProcSelf(x) schedProc(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_VAL((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedSignal() - Add SIGNAL task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @sig = Signal + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedSignal(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + unsigned long sig, void *opt_data, size_t opt_dlen); +#define schedSignalSelf(x) schedSignal(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_VAL((x)), TASK_DATA((x)), TASK_DATLEN((x))) + +#ifdef EVFILT_LIO +/* + * schedAIO() - Add AIO task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @acb = AIO cb structure address + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +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); +/* + * schedAIORead() - Add AIO read task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @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, off_t offset); +/* + * schedAIOWrite() - Add AIO write task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @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, off_t offset); + +/* + * schedLIO() - Add AIO bulk tasks to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @acbs = AIO cb structure addresses + * @opt_data = Optional data + * @opt_dlen = Optional data length + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedLIO(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + struct aiocb ** __restrict acbs, void *opt_data, size_t opt_dlen); +/* + * 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 + */ +inline 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); +/* + * 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); +#endif /* EVFILT_LIO */ + +/* + * schedUser() - Add trigger USER task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @id = Trigger ID + * @opt_data = Optional data + * @opt_dlen = Optional user's trigger flags + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedUser(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + unsigned long id, void *opt_data, size_t opt_dlen); +#define schedUserSelf(x) schedUser(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_VAL((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedTrigger() - Triggering USER task + * + * @task = task + * return: -1 error or 0 ok + */ +int schedTrigger(sched_task_t * __restrict task); + +/* * schedTimer() - Add TIMER task to scheduler queue * * @root = root task @@ -349,6 +564,32 @@ sched_task_t *schedEventLo(sched_root_task_t * __restr unsigned long val, void *opt_data, size_t opt_dlen); #define schedEventLoSelf(x) schedEventLo(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ TASK_VAL((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedSuspend() - Add Suspended task to scheduler queue + * + * @root = root task + * @func = task execution function + * @arg = 1st func argument + * @id = Trigger ID + * @opt_data = Optional data + * @opt_dlen = Optional user's trigger flags + * return: NULL error or !=NULL new queued task + */ +sched_task_t *schedSuspend(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, + unsigned long id, void *opt_data, size_t opt_dlen); +#define schedSuspendSelf(x) schedUser(TASK_ROOT((x)), TASK_FUNC((x)), TASK_ARG((x)), \ + TASK_VAL((x)), TASK_DATA((x)), TASK_DATLEN((x))) +/* + * schedResumeby() - Resume suspended task + * + * @root = root task + * @criteria = find task by criteria + * [CRITERIA_ANY|CRITERIA_ID|CRITERIA_DATA] + * @param = search parameter (sched_task_t *task| u_long id) + * return: -1 error or 0 resumed ok + */ +int schedResumeby(sched_root_task_t * __restrict root, unsigned char criteria, void *param); + /* * schedCallOnce() - Call once from scheduler *