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>