File:  [ELWIX - Embedded LightWeight unIX -] / libaitsched / inc / aitsched.h
Revision 1.6.2.2: download - view: text, annotated - select for diffs - revision graph
Mon May 14 12:07:08 2012 UTC (12 years, 1 month ago) by misho
Branches: sched1_5
Diff to: branchpoint 1.6: preferred, unified
add ROOT_QUEUE_EMPTY test macro for scheduler

    1: /*************************************************************************
    2: * (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
    3: *  by Michael Pounov <misho@openbsd-bg.org>
    4: *
    5: * $Author: misho $
    6: * $Id: aitsched.h,v 1.6.2.2 2012/05/14 12:07:08 misho Exp $
    7: *
    8: **************************************************************************
    9: The ELWIX and AITNET software is distributed under the following
   10: terms:
   11: 
   12: All of the documentation and software included in the ELWIX and AITNET
   13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   14: 
   15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
   16: 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
   17: 
   18: Redistribution and use in source and binary forms, with or without
   19: modification, are permitted provided that the following conditions
   20: are met:
   21: 1. Redistributions of source code must retain the above copyright
   22:    notice, this list of conditions and the following disclaimer.
   23: 2. Redistributions in binary form must reproduce the above copyright
   24:    notice, this list of conditions and the following disclaimer in the
   25:    documentation and/or other materials provided with the distribution.
   26: 3. All advertising materials mentioning features or use of this software
   27:    must display the following acknowledgement:
   28: This product includes software developed by Michael Pounov <misho@elwix.org>
   29: ELWIX - Embedded LightWeight unIX and its contributors.
   30: 4. Neither the name of AITNET nor the names of its contributors
   31:    may be used to endorse or promote products derived from this software
   32:    without specific prior written permission.
   33: 
   34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
   35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   44: SUCH DAMAGE.
   45: */
   46: #ifndef __AITSCHED_H
   47: #define __AITSCHED_H
   48: 
   49: 
   50: #include <sys/types.h>
   51: #include <sys/queue.h>
   52: #include <sys/uio.h>
   53: #include <stdint.h>
   54: #include <pthread.h>
   55: 
   56: 
   57: /* criteria type */
   58: #define CRITERIA_CALL	0
   59: #define CRITERIA_ARG	1
   60: #define CRITERIA_FD	2
   61: #define CRITERIA_VAL	3
   62: #define CRITERIA_TV	4
   63: 
   64: 
   65: /* early declaration for root & task */
   66: typedef struct sched_Task	sched_task_t;
   67: typedef struct sched_RootTask	sched_root_task_t;
   68: 
   69: typedef enum {
   70: 	taskREAD = 0,
   71: 	taskWRITE,
   72: 	taskTIMER,
   73: 	taskEVENT, 
   74: 	taskEVENTLO, 
   75: 	taskREADY,
   76: 	taskUNUSE,
   77: 	taskMAX
   78: } sched_task_type_t;
   79: 
   80: /* hooks */
   81: typedef void *(*sched_hook_func_t)(void *, void *);
   82: struct sched_HooksTask {
   83: 	struct {
   84: 		/* read(sched_task_t *task, NULL) -> int */
   85: 		sched_hook_func_t	read;
   86: 		/* write(sched_task_t *task, NULL) -> int */
   87: 		sched_hook_func_t	write;
   88: 		/* event(sched_task_t *task, NULL) -> int */
   89: 		sched_hook_func_t	event;
   90: 		/* eventlo(sched_task_t *task, NULL) -> int */
   91: 		sched_hook_func_t	eventlo;
   92: 		/* timer(sched_task_t *task, struct timespec *ts) -> int */
   93: 		sched_hook_func_t	timer;
   94: 	}	hook_add;
   95: 	struct {
   96: 		/* cancel(sched_task_t *task, NULL) -> int */
   97: 		sched_hook_func_t	cancel;
   98: 		/* run(sched_root_task_t *root, NULL) -> int */
   99: 		sched_hook_func_t	run;
  100: 		/* fetch(sched_root_task_t *root, NULL) -> sched_task_t* */
  101: 		sched_hook_func_t	fetch;
  102: 		/* exception(sched_root_task_t *root, NULL) -> int */
  103: 		sched_hook_func_t	exception;
  104: 		/* condition(sched_root_task_t *root, intptr_t *stopValue) -> int */
  105: 		sched_hook_func_t	condition;
  106: 	}	hook_exec;
  107: 	struct {
  108: 		/* init(sched_root_task_t *root, void *data) -> int */
  109: 		sched_hook_func_t	init;
  110: 		/* fini(sched_root_task_t *root, NULL) -> int */
  111: 		sched_hook_func_t	fini;
  112: 		/* error(sched_root_task_t *root, int errno) -> int */
  113: 		sched_hook_func_t	error;
  114: 	}	hook_root;
  115: };
  116: typedef struct sched_HooksTask hooks_task_t;
  117: 
  118: /* task callback, like pthread callback! */
  119: typedef void *(*sched_task_func_t)(sched_task_t * /* current task data*/);
  120: 
  121: /* task lock helpers */
  122: #define TASK_LOCK(x)		((x)->task_lock++)
  123: #define TASK_UNLOCK(x)		((x)->task_lock ^= (x)->task_lock)
  124: #define TASK_ISLOCKED(x)	((x)->task_lock)
  125: 
  126: /* task & queue */
  127: struct sched_Task {
  128: 	volatile int			task_lock;
  129: 	unsigned int			task_id;
  130: 	sched_task_type_t		task_type;
  131: #define TASK_TYPE(x)	(x)->task_type
  132: 
  133: 	sched_root_task_t		*task_root;
  134: #define TASK_ROOT(x)	(x)->task_root
  135: 	sched_task_func_t		task_func;
  136: 
  137: 	void				*task_arg;
  138: 	union {
  139: 		unsigned long	v;
  140: 		intptr_t	fd;
  141: 		struct timespec	ts;
  142: 	}				task_val;
  143: #define TASK_ARG(x)	(x)->task_arg
  144: #define TASK_VAL(x)	(x)->task_val.v
  145: #define TASK_FD(x)	(x)->task_val.fd
  146: #define TASK_TS(x)	(x)->task_val.ts
  147: 
  148: 	struct iovec			task_data;
  149: #define TASK_DATA(x)	(x)->task_data.iov_base
  150: #define TASK_DATLEN(x)	(x)->task_data.iov_len
  151: 
  152: 	TAILQ_ENTRY(sched_Task)		task_node;
  153: };
  154: typedef TAILQ_HEAD(, sched_Task) sched_queue_t;
  155: #define TASK_DATA_SET(x, _dp, _dl)	do { \
  156: 						if ((x)) { \
  157: 							(x)->task_data.iov_base = (_dp); \
  158: 							(x)->task_data.iov_len = _dl; \
  159: 						} \
  160: 					while (0)
  161: 
  162: /* root task */
  163: struct sched_RootTask {
  164: 	int		root_kq;
  165: 	struct timespec	root_wait;
  166: 	struct timespec	root_poll;
  167: 	intptr_t	root_cond;
  168: 
  169: 	pthread_mutex_t	root_mtx[taskMAX];
  170: 
  171: 	sched_queue_t	root_read;
  172: 	sched_queue_t	root_write;
  173: 	sched_queue_t	root_timer;
  174: 	sched_queue_t	root_event;
  175: 	sched_queue_t	root_ready;
  176: 	sched_queue_t	root_unuse;
  177: 	sched_queue_t	root_eventlo;
  178: 	int		root_eventlo_miss;
  179: 
  180: 	hooks_task_t	root_hooks;
  181: 	struct iovec	root_data;
  182: #define ROOT_DATA(x)	(x)->root_data.iov_base
  183: #define ROOT_DATLEN(x)	(x)->root_data.iov_len
  184: };
  185: #define ROOT_QUEUE_EMPTY(x, _q)	TAILQ_EMPTY(&((x)->root_##_q))
  186: 
  187: 
  188: inline int sched_GetErrno();
  189: inline const char *sched_GetError();
  190: 
  191: 
  192: /*
  193:  * schedInit() - Init scheduler
  194:  *
  195:  * @data = optional data if !=NULL
  196:  * @datlen = data len if data is set
  197:  * return: allocated root task if ok or NULL error
  198:  */
  199: sched_root_task_t *schedInit(void ** __restrict data, size_t datlen);
  200: #define schedBegin()	schedInit((void**) &schedRegisterHooks, 0)
  201: /*
  202:  * schedEnd() - End scheduler & free all resources
  203:  *
  204:  * @root = root task
  205:  * return: -1 error or 0 ok
  206:  */
  207: int schedEnd(sched_root_task_t ** __restrict root);
  208: /*
  209:  * schedRegisterHooks() - Register IO handles and bind tasks to it
  210:  *
  211:  * @root = root task
  212:  * return: -1 error or 0 ok
  213:  */
  214: int schedRegisterHooks(sched_root_task_t * __restrict root);
  215: /*
  216:  * schedPolling() - Polling timeout period if no timer task is present
  217:  *
  218:  * @root = root task
  219:  * @ts = timeout polling period, if ==NULL INFINIT timeout
  220:  * @tsold = old timeout polling if !=NULL
  221:  * return: -1 error or 0 ok
  222:  */
  223: inline int schedPolling(sched_root_task_t * __restrict root, 
  224: 		struct timespec * __restrict ts, struct timespec * __restrict tsold);
  225: /*
  226:  * schedTermCondition() - Activate hook for scheduler condition kill
  227:  *
  228:  * @root = root task
  229:  * @condValue = condition value, kill schedRun() if condValue == killState
  230:  * return: -1 error ok 0 ok
  231:  */
  232: inline int schedTermCondition(sched_root_task_t * __restrict root, intptr_t condValue);
  233: /*
  234:  * schedCall() - Call task execution function
  235:  *
  236:  * @task = current task
  237:  * return: !=NULL error or =NULL ok
  238:  */
  239: inline void *schedCall(sched_task_t * __restrict task);
  240: /*
  241:  * schedFetch() - Fetch ready task
  242:  *
  243:  * @root = root task
  244:  * return: =NULL error or !=NULL ready task
  245:  */
  246: inline void *schedFetch(sched_root_task_t * __restrict root);
  247: /*
  248:  * schedRun() - Scheduler *run loop*
  249:  *
  250:  * @root = root task
  251:  * @killState = kill condition variable, if !=0 stop scheduler loop
  252:  * return: -1 error or 0 ok
  253:  */
  254: int schedRun(sched_root_task_t *root, volatile intptr_t * __restrict killState);
  255: /*
  256:  * schedCancel() - Cancel task from scheduler
  257:  *
  258:  * @task = task
  259:  * return: -1 error or 0 ok
  260:  */
  261: int schedCancel(sched_task_t * __restrict task);
  262: /*
  263:  * schedCancelby() - Cancel task from scheduler by criteria
  264:  *
  265:  * @root = root task
  266:  * @type = cancel from queue type, if =taskMAX cancel same task from all queues
  267:  * @criteria = find task by criteria [CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_TV]
  268:  * @param = search parameter
  269:  * @hook = custom cleanup hook function, may be NULL
  270:  * return: -1 error, -2 error in sub-stage cancel execution, -3 error from custom hook or 0 ok
  271:  */
  272: int schedCancelby(sched_root_task_t * __restrict root, sched_task_type_t type, 
  273: 		u_char criteria, void *param, sched_hook_func_t hook);
  274: 
  275: 
  276: /*
  277:  * schedRead() - Add READ I/O task to scheduler queue
  278:  *
  279:  * @root = root task
  280:  * @func = task execution function
  281:  * @arg = 1st func argument
  282:  * @fd = fd handle
  283:  * @opt_data = Optional data
  284:  * @opt_dlen = Optional data length
  285:  * return: NULL error or !=NULL new queued task
  286:  */
  287: sched_task_t *schedRead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  288: 		int fd, void *opt_data, size_t opt_dlen);
  289: /*
  290:  * schedWrite() - Add WRITE I/O task to scheduler queue
  291:  *
  292:  * @root = root task
  293:  * @func = task execution function
  294:  * @arg = 1st func argument
  295:  * @fd = fd handle
  296:  * @opt_data = Optional data
  297:  * @opt_dlen = Optional data length
  298:  * return: NULL error or !=NULL new queued task
  299:  */
  300: sched_task_t *schedWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  301: 		int fd, void *opt_data, size_t opt_dlen);
  302: /*
  303:  * schedTimer() - Add TIMER task to scheduler queue
  304:  *
  305:  * @root = root task
  306:  * @func = task execution function
  307:  * @arg = 1st func argument
  308:  * @ts = timeout argument structure
  309:  * @opt_data = Optional data
  310:  * @opt_dlen = Optional data length
  311:  * return: NULL error or !=NULL new queued task
  312:  */
  313: sched_task_t *schedTimer(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  314: 		struct timespec ts, void *opt_data, size_t opt_dlen);
  315: /*
  316:  * schedEvent() - Add EVENT task to scheduler queue
  317:  *
  318:  * @root = root task
  319:  * @func = task execution function
  320:  * @arg = 1st func argument
  321:  * @val = additional func argument
  322:  * @opt_data = Optional data
  323:  * @opt_dlen = Optional data length
  324:  * return: NULL error or !=NULL new queued task
  325:  */
  326: sched_task_t *schedEvent(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  327: 		unsigned long val, void *opt_data, size_t opt_dlen);
  328: /*
  329:  * schedEventLo() - Add EVENT_Lo task to scheduler queue
  330:  *
  331:  * @root = root task
  332:  * @func = task execution function
  333:  * @arg = 1st func argument
  334:  * @val = additional func argument
  335:  * @opt_data = Optional data
  336:  * @opt_dlen = Optional data length
  337:  * return: NULL error or !=NULL new queued task
  338:  */
  339: sched_task_t *schedEventLo(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  340: 		unsigned long val, void *opt_data, size_t opt_dlen);
  341: /*
  342:  * schedCallOnce() - Call once from scheduler
  343:  *
  344:  * @root = root task
  345:  * @func = task execution function
  346:  * @arg = 1st func argument
  347:  * @val = additional func argument
  348:  * @opt_data = Optional data
  349:  * @opt_dlen = Optional data length
  350:  * return: return value from called func
  351:  */
  352: sched_task_t *schedCallOnce(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  353: 		unsigned long val, void *opt_data, size_t opt_dlen);
  354: 
  355: 
  356: #endif

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>