File:  [ELWIX - Embedded LightWeight unIX -] / libaitsched / inc / aitsched.h
Revision 1.5: download - view: text, annotated - select for diffs - revision graph
Tue Jan 24 21:59:46 2012 UTC (12 years, 5 months ago) by misho
Branches: MAIN
CVS tags: sched1_4, SCHED1_3, HEAD
version 1.3
- add scheduler timespec macros
- change schedTimer task argument from timeval to timespec structure
- refactoring timer structure in entire scheduler to use timespec
- add schedPolling API for idle timeout control
- add idle polling support!

    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.5 2012/01/24 21:59:46 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
   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: 	}	hook_exec;
  105: 	struct {
  106: 		/* init(sched_root_task_t *root, void *data) -> int */
  107: 		sched_hook_func_t	init;
  108: 		/* fini(sched_root_task_t *root, NULL) -> int */
  109: 		sched_hook_func_t	fini;
  110: 		/* error(sched_root_task_t *root, int errno) -> int */
  111: 		sched_hook_func_t	error;
  112: 	}	hook_root;
  113: };
  114: typedef struct sched_HooksTask hooks_task_t;
  115: 
  116: /* task callback, like pthread callback! */
  117: typedef void *(*sched_task_func_t)(sched_task_t * /* current task data*/);
  118: 
  119: /* task lock helpers */
  120: #define TASK_LOCK(x)		((x)->task_lock++)
  121: #define TASK_UNLOCK(x)		((x)->task_lock ^= (x)->task_lock)
  122: #define TASK_ISLOCKED(x)	((x)->task_lock)
  123: 
  124: /* task & queue */
  125: struct sched_Task {
  126: 	volatile int			task_lock;
  127: 	unsigned int			task_id;
  128: 	sched_task_type_t		task_type;
  129: #define TASK_TYPE(x)	(x)->task_type
  130: 
  131: 	sched_root_task_t		*task_root;
  132: #define TASK_ROOT(x)	(x)->task_root
  133: 	sched_task_func_t		task_func;
  134: 
  135: 	void				*task_arg;
  136: 	union {
  137: 		unsigned long	v;
  138: 		intptr_t	fd;
  139: 		struct timespec	ts;
  140: 	}				task_val;
  141: #define TASK_ARG(x)	(x)->task_arg
  142: #define TASK_VAL(x)	(x)->task_val.v
  143: #define TASK_FD(x)	(x)->task_val.fd
  144: #define TASK_TS(x)	(x)->task_val.ts
  145: 
  146: 	struct iovec			task_data;
  147: #define TASK_DATA(x)	(x)->task_data.iov_base
  148: #define TASK_DATLEN(x)	(x)->task_data.iov_len
  149: 
  150: 	TAILQ_ENTRY(sched_Task)		task_node;
  151: };
  152: typedef TAILQ_HEAD(, sched_Task) sched_queue_t;
  153: #define TASK_DATA_SET(x, _dp, _dl)	do { \
  154: 						if ((x)) { \
  155: 							(x)->task_data.iov_base = (_dp); \
  156: 							(x)->task_data.iov_len = _dl; \
  157: 						} \
  158: 					while (0)
  159: 
  160: /* root task */
  161: struct sched_RootTask {
  162: 	int		root_kq;
  163: 	struct timespec	root_wait;
  164: 	struct timespec	root_poll;
  165: 	pthread_mutex_t	root_mtx[taskMAX];
  166: 
  167: 	sched_queue_t	root_read;
  168: 	sched_queue_t	root_write;
  169: 	sched_queue_t	root_timer;
  170: 	sched_queue_t	root_event;
  171: 	sched_queue_t	root_ready;
  172: 	sched_queue_t	root_unuse;
  173: 	sched_queue_t	root_eventlo;
  174: 	int		root_eventlo_miss;
  175: 
  176: 	hooks_task_t	root_hooks;
  177: 	struct iovec	root_data;
  178: #define ROOT_DATA(x)	(x)->root_data.iov_base
  179: #define ROOT_DATLEN(x)	(x)->root_data.iov_len
  180: };
  181: 
  182: 
  183: inline int sched_GetErrno();
  184: inline const char *sched_GetError();
  185: 
  186: 
  187: /*
  188:  * schedInit() - Init scheduler
  189:  * @data = optional data if !=NULL
  190:  * @datlen = data len if data is set
  191:  * return: allocated root task if ok or NULL error
  192:  */
  193: sched_root_task_t *schedInit(void ** __restrict data, size_t datlen);
  194: #define schedBegin()	schedInit((void**) &schedRegisterHooks, 0)
  195: /*
  196:  * schedEnd() - End scheduler & free all resources
  197:  * @root = root task
  198:  * return: -1 error or 0 ok
  199:  */
  200: int schedEnd(sched_root_task_t ** __restrict root);
  201: /*
  202:  * schedRegisterHooks() - Register IO handles and bind tasks to it
  203:  * @root = root task
  204:  * return: -1 error or 0 ok
  205:  */
  206: int schedRegisterHooks(sched_root_task_t * __restrict root);
  207: /*
  208:  * schedPolling() - Polling timeout period if no timer task is present
  209:  * @root = root task
  210:  * @ts = timeout polling period, if ==NULL INFINIT timeout
  211:  * @tsold = old timeout polling if !=NULL
  212:  * return: -1 error or 0 ok
  213:  */
  214: inline int schedPolling(sched_root_task_t * __restrict root, 
  215: 		struct timespec * __restrict ts, struct timespec * __restrict tsold);
  216: /*
  217:  * schedCall() - Call task execution function
  218:  * @task = current task
  219:  * return: !=NULL error or =NULL ok
  220:  */
  221: inline void *schedCall(sched_task_t * __restrict task);
  222: /*
  223:  * schedFetch() - Fetch ready task
  224:  * @root = root task
  225:  * return: =NULL error or !=NULL ready task
  226:  */
  227: inline void *schedFetch(sched_root_task_t * __restrict root);
  228: /*
  229:  * schedRun() - Scheduler *run loop*
  230:  * @root = root task
  231:  * @killState = kill condition variable, if !=0 stop scheduler loop
  232:  * return: -1 error or 0 ok
  233:  */
  234: int schedRun(sched_root_task_t * __restrict root, volatile intptr_t * __restrict killState);
  235: /*
  236:  * schedCancel() - Cancel task from scheduler
  237:  * @task = task
  238:  * return: -1 error or 0 ok
  239:  */
  240: int schedCancel(sched_task_t * __restrict task);
  241: /*
  242:  * schedCancelby() - Cancel task from scheduler by criteria
  243:  * @root = root task
  244:  * @type = cancel from queue type, if =taskMAX cancel same task from all queues
  245:  * @criteria = find task by criteria [CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_TV]
  246:  * @param = search parameter
  247:  * @hook = custom cleanup hook function, may be NULL
  248:  * return: -1 error, -2 error in sub-stage cancel execution, -3 error from custom hook or 0 ok
  249:  */
  250: int schedCancelby(sched_root_task_t * __restrict root, sched_task_type_t type, 
  251: 		u_char criteria, void *param, sched_hook_func_t hook);
  252: 
  253: 
  254: /*
  255:  * schedRead() - Add READ I/O task to scheduler queue
  256:  * @root = root task
  257:  * @func = task execution function
  258:  * @arg = 1st func argument
  259:  * @fd = fd handle
  260:  * @opt_data = Optional data
  261:  * @opt_dlen = Optional data length
  262:  * return: NULL error or !=NULL new queued task
  263:  */
  264: sched_task_t *schedRead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  265: 		int fd, void *opt_data, size_t opt_dlen);
  266: /*
  267:  * schedWrite() - Add WRITE I/O task to scheduler queue
  268:  * @root = root task
  269:  * @func = task execution function
  270:  * @arg = 1st func argument
  271:  * @fd = fd handle
  272:  * @opt_data = Optional data
  273:  * @opt_dlen = Optional data length
  274:  * return: NULL error or !=NULL new queued task
  275:  */
  276: sched_task_t *schedWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  277: 		int fd, void *opt_data, size_t opt_dlen);
  278: /*
  279:  * schedTimer() - Add TIMER task to scheduler queue
  280:  * @root = root task
  281:  * @func = task execution function
  282:  * @arg = 1st func argument
  283:  * @ts = timeout argument structure
  284:  * @opt_data = Optional data
  285:  * @opt_dlen = Optional data length
  286:  * return: NULL error or !=NULL new queued task
  287:  */
  288: sched_task_t *schedTimer(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  289: 		struct timespec ts, void *opt_data, size_t opt_dlen);
  290: /*
  291:  * schedEvent() - Add EVENT task to scheduler queue
  292:  * @root = root task
  293:  * @func = task execution function
  294:  * @arg = 1st func argument
  295:  * @val = additional func argument
  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 *schedEvent(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  301: 		unsigned long val, void *opt_data, size_t opt_dlen);
  302: /*
  303:  * schedEventLo() - Add EVENT_Lo task to scheduler queue
  304:  * @root = root task
  305:  * @func = task execution function
  306:  * @arg = 1st func argument
  307:  * @val = additional func argument
  308:  * @opt_data = Optional data
  309:  * @opt_dlen = Optional data length
  310:  * return: NULL error or !=NULL new queued task
  311:  */
  312: sched_task_t *schedEventLo(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  313: 		unsigned long val, void *opt_data, size_t opt_dlen);
  314: /*
  315:  * schedCallOnce() - Call once from scheduler
  316:  * @root = root task
  317:  * @func = task execution function
  318:  * @arg = 1st func argument
  319:  * @val = additional func argument
  320:  * @opt_data = Optional data
  321:  * @opt_dlen = Optional data length
  322:  * return: return value from called func
  323:  */
  324: sched_task_t *schedCallOnce(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, 
  325: 		unsigned long val, void *opt_data, size_t opt_dlen);
  326: 
  327: 
  328: #endif

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