Annotation of libaitsched/src/tasks.c, revision 1.10.2.3
1.1 misho 1: /*************************************************************************
2: * (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
1.10.2.3! misho 6: * $Id: tasks.c,v 1.10.2.2 2012/08/01 13:43:14 misho Exp $
1.1 misho 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:
1.6 misho 15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
1.1 misho 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: #include "global.h"
47:
48:
1.4 misho 49: #pragma GCC visibility push(hidden)
50:
51: inline sched_task_t *
52: _sched_useTask(sched_root_task_t * __restrict root)
53: {
1.7 misho 54: sched_task_t *task, *tmp;
1.4 misho 55:
1.7 misho 56: TAILQ_FOREACH_SAFE(task, &root->root_unuse, task_node, tmp) {
1.4 misho 57: if (!TASK_ISLOCKED(task)) {
1.5 misho 58: #ifdef HAVE_LIBPTHREAD
59: pthread_mutex_lock(&root->root_mtx[taskUNUSE]);
60: #endif
1.4 misho 61: TAILQ_REMOVE(&root->root_unuse, task, task_node);
1.5 misho 62: #ifdef HAVE_LIBPTHREAD
63: pthread_mutex_unlock(&root->root_mtx[taskUNUSE]);
64: #endif
1.4 misho 65: break;
66: }
67: }
68:
69: if (!task) {
70: task = malloc(sizeof(sched_task_t));
71: if (!task) {
72: LOGERR;
73: return NULL;
74: }
75: }
76:
1.9 misho 77: memset(task, 0, sizeof(sched_task_t));
78: task->task_id = (uintptr_t) task;
1.4 misho 79: return task;
80: }
81:
82: inline sched_task_t *
83: _sched_unuseTask(sched_task_t * __restrict task)
84: {
85: TASK_UNLOCK(task);
1.5 misho 86: TASK_TYPE(task) = taskUNUSE;
87: #ifdef HAVE_LIBPTHREAD
88: pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[taskUNUSE]);
89: #endif
1.9 misho 90: TAILQ_INSERT_TAIL(&TASK_ROOT(task)->root_unuse, TASK_ID(task), task_node);
1.5 misho 91: #ifdef HAVE_LIBPTHREAD
92: pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[taskUNUSE]);
93: #endif
1.4 misho 94: task = NULL;
95:
96: return task;
97: }
98:
99: #pragma GCC visibility pop
100:
101:
1.1 misho 102: /*
1.2 misho 103: * schedRead() - Add READ I/O task to scheduler queue
1.6 misho 104: *
1.1 misho 105: * @root = root task
106: * @func = task execution function
107: * @arg = 1st func argument
1.2 misho 108: * @fd = fd handle
1.5 misho 109: * @opt_data = Optional data
110: * @opt_dlen = Optional data length
1.1 misho 111: * return: NULL error or !=NULL new queued task
112: */
113: sched_task_t *
1.5 misho 114: schedRead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd,
115: void *opt_data, size_t opt_dlen)
1.1 misho 116: {
117: sched_task_t *task;
118: void *ptr;
119:
120: if (!root || !func)
121: return NULL;
122:
123: /* get new task */
1.4 misho 124: if (!(task = _sched_useTask(root)))
125: return NULL;
1.1 misho 126:
127: task->task_func = func;
1.5 misho 128: TASK_TYPE(task) = taskREAD;
129: TASK_ROOT(task) = root;
1.1 misho 130:
131: TASK_ARG(task) = arg;
1.2 misho 132: TASK_FD(task) = fd;
1.1 misho 133:
1.5 misho 134: TASK_DATA(task) = opt_data;
135: TASK_DATLEN(task) = opt_dlen;
136:
1.1 misho 137: if (root->root_hooks.hook_add.read)
138: ptr = root->root_hooks.hook_add.read(task, NULL);
139: else
140: ptr = NULL;
141:
1.5 misho 142: if (!ptr) {
143: #ifdef HAVE_LIBPTHREAD
144: pthread_mutex_lock(&root->root_mtx[taskREAD]);
145: #endif
1.9 misho 146: TAILQ_INSERT_TAIL(&root->root_read, TASK_ID(task), task_node);
1.5 misho 147: #ifdef HAVE_LIBPTHREAD
148: pthread_mutex_unlock(&root->root_mtx[taskREAD]);
149: #endif
150: } else
1.4 misho 151: task = _sched_unuseTask(task);
1.1 misho 152:
153: return task;
154: }
155:
156: /*
1.2 misho 157: * schedWrite() - Add WRITE I/O task to scheduler queue
1.6 misho 158: *
1.1 misho 159: * @root = root task
160: * @func = task execution function
161: * @arg = 1st func argument
1.2 misho 162: * @fd = fd handle
1.5 misho 163: * @opt_data = Optional data
164: * @opt_dlen = Optional data length
1.1 misho 165: * return: NULL error or !=NULL new queued task
166: */
167: sched_task_t *
1.5 misho 168: schedWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd,
169: void *opt_data, size_t opt_dlen)
1.1 misho 170: {
171: sched_task_t *task;
172: void *ptr;
173:
174: if (!root || !func)
175: return NULL;
176:
177: /* get new task */
1.4 misho 178: if (!(task = _sched_useTask(root)))
179: return NULL;
1.1 misho 180:
181: task->task_func = func;
1.5 misho 182: TASK_TYPE(task) = taskWRITE;
183: TASK_ROOT(task) = root;
1.1 misho 184:
185: TASK_ARG(task) = arg;
1.2 misho 186: TASK_FD(task) = fd;
1.1 misho 187:
1.5 misho 188: TASK_DATA(task) = opt_data;
189: TASK_DATLEN(task) = opt_dlen;
190:
1.1 misho 191: if (root->root_hooks.hook_add.write)
192: ptr = root->root_hooks.hook_add.write(task, NULL);
193: else
194: ptr = NULL;
195:
1.5 misho 196: if (!ptr) {
197: #ifdef HAVE_LIBPTHREAD
198: pthread_mutex_lock(&root->root_mtx[taskWRITE]);
199: #endif
1.9 misho 200: TAILQ_INSERT_TAIL(&root->root_write, TASK_ID(task), task_node);
1.5 misho 201: #ifdef HAVE_LIBPTHREAD
202: pthread_mutex_unlock(&root->root_mtx[taskWRITE]);
203: #endif
204: } else
1.4 misho 205: task = _sched_unuseTask(task);
1.1 misho 206:
207: return task;
208: }
209:
210: /*
1.9 misho 211: * schedNode() - Add NODE task to scheduler queue
212: *
213: * @root = root task
214: * @func = task execution function
215: * @arg = 1st func argument
216: * @fd = fd handle
217: * @opt_data = Optional data
218: * @opt_dlen = Optional data length
219: * return: NULL error or !=NULL new queued task
220: */
221: sched_task_t *
222: schedNode(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd,
223: void *opt_data, size_t opt_dlen)
224: {
225: sched_task_t *task;
226: void *ptr;
227:
228: if (!root || !func)
229: return NULL;
230:
231: /* get new task */
232: if (!(task = _sched_useTask(root)))
233: return NULL;
234:
235: task->task_func = func;
236: TASK_TYPE(task) = taskNODE;
237: TASK_ROOT(task) = root;
238:
239: TASK_ARG(task) = arg;
240: TASK_FD(task) = fd;
241:
242: TASK_DATA(task) = opt_data;
243: TASK_DATLEN(task) = opt_dlen;
244:
245: if (root->root_hooks.hook_add.node)
246: ptr = root->root_hooks.hook_add.node(task, NULL);
247: else
248: ptr = NULL;
249:
250: if (!ptr) {
251: #ifdef HAVE_LIBPTHREAD
252: pthread_mutex_lock(&root->root_mtx[taskNODE]);
253: #endif
254: TAILQ_INSERT_TAIL(&root->root_node, TASK_ID(task), task_node);
255: #ifdef HAVE_LIBPTHREAD
256: pthread_mutex_unlock(&root->root_mtx[taskNODE]);
257: #endif
258: } else
259: task = _sched_unuseTask(task);
260:
261: return task;
262: }
263:
264: /*
265: * schedProc() - Add PROC task to scheduler queue
266: *
267: * @root = root task
268: * @func = task execution function
269: * @arg = 1st func argument
270: * @pid = PID
271: * @opt_data = Optional data
272: * @opt_dlen = Optional data length
273: * return: NULL error or !=NULL new queued task
274: */
275: sched_task_t *
276: schedProc(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long pid,
277: void *opt_data, size_t opt_dlen)
278: {
279: sched_task_t *task;
280: void *ptr;
281:
282: if (!root || !func)
283: return NULL;
284:
285: /* get new task */
286: if (!(task = _sched_useTask(root)))
287: return NULL;
288:
289: task->task_func = func;
290: TASK_TYPE(task) = taskPROC;
291: TASK_ROOT(task) = root;
292:
293: TASK_ARG(task) = arg;
294: TASK_VAL(task) = pid;
295:
296: TASK_DATA(task) = opt_data;
297: TASK_DATLEN(task) = opt_dlen;
298:
299: if (root->root_hooks.hook_add.proc)
300: ptr = root->root_hooks.hook_add.proc(task, NULL);
301: else
302: ptr = NULL;
303:
304: if (!ptr) {
305: #ifdef HAVE_LIBPTHREAD
306: pthread_mutex_lock(&root->root_mtx[taskPROC]);
307: #endif
308: TAILQ_INSERT_TAIL(&root->root_proc, TASK_ID(task), task_node);
309: #ifdef HAVE_LIBPTHREAD
310: pthread_mutex_unlock(&root->root_mtx[taskPROC]);
311: #endif
312: } else
313: task = _sched_unuseTask(task);
314:
315: return task;
316: }
317:
318: /*
319: * schedUser() - Add trigger USER task to scheduler queue
320: *
321: * @root = root task
322: * @func = task execution function
323: * @arg = 1st func argument
324: * @id = Trigger ID
325: * @opt_data = Optional data
326: * @opt_dlen = Optional user's trigger flags
327: * return: NULL error or !=NULL new queued task
328: */
329: sched_task_t *
330: schedUser(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long id,
331: void *opt_data, size_t opt_dlen)
332: {
333: #ifndef EVFILT_USER
334: sched_SetErr(ENOTSUP, "Not supported kevent() filter");
335: return NULL;
336: #else
337: sched_task_t *task;
338: void *ptr;
339:
340: if (!root || !func)
341: return NULL;
342:
343: /* get new task */
344: if (!(task = _sched_useTask(root)))
345: return NULL;
346:
347: task->task_func = func;
348: TASK_TYPE(task) = taskUSER;
349: TASK_ROOT(task) = root;
350:
351: TASK_ARG(task) = arg;
352: TASK_VAL(task) = id;
353:
354: TASK_DATA(task) = opt_data;
355: TASK_DATLEN(task) = opt_dlen;
356:
357: if (root->root_hooks.hook_add.user)
358: ptr = root->root_hooks.hook_add.user(task, NULL);
359: else
360: ptr = NULL;
361:
362: if (!ptr) {
363: #ifdef HAVE_LIBPTHREAD
364: pthread_mutex_lock(&root->root_mtx[taskUSER]);
365: #endif
366: TAILQ_INSERT_TAIL(&root->root_user, TASK_ID(task), task_node);
367: #ifdef HAVE_LIBPTHREAD
368: pthread_mutex_unlock(&root->root_mtx[taskUSER]);
369: #endif
370: } else
371: task = _sched_unuseTask(task);
372:
373: return task;
374: #endif
375: }
376:
377: /*
378: * schedSignal() - Add SIGNAL task to scheduler queue
379: *
380: * @root = root task
381: * @func = task execution function
382: * @arg = 1st func argument
383: * @sig = Signal
384: * @opt_data = Optional data
385: * @opt_dlen = Optional data length
386: * return: NULL error or !=NULL new queued task
387: */
388: sched_task_t *
389: schedSignal(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long sig,
390: void *opt_data, size_t opt_dlen)
391: {
392: sched_task_t *task;
393: void *ptr;
394:
395: if (!root || !func)
396: return NULL;
397:
398: /* get new task */
399: if (!(task = _sched_useTask(root)))
400: return NULL;
401:
402: task->task_func = func;
403: TASK_TYPE(task) = taskSIGNAL;
404: TASK_ROOT(task) = root;
405:
406: TASK_ARG(task) = arg;
407: TASK_VAL(task) = sig;
408:
409: TASK_DATA(task) = opt_data;
410: TASK_DATLEN(task) = opt_dlen;
411:
412: if (root->root_hooks.hook_add.signal)
413: ptr = root->root_hooks.hook_add.signal(task, NULL);
414: else
415: ptr = NULL;
416:
417: if (!ptr) {
418: #ifdef HAVE_LIBPTHREAD
419: pthread_mutex_lock(&root->root_mtx[taskSIGNAL]);
420: #endif
421: TAILQ_INSERT_TAIL(&root->root_signal, TASK_ID(task), task_node);
422: #ifdef HAVE_LIBPTHREAD
423: pthread_mutex_unlock(&root->root_mtx[taskSIGNAL]);
424: #endif
425: } else
426: task = _sched_unuseTask(task);
427:
428: return task;
429: }
430:
431: /*
1.8 misho 432: * schedAlarm() - Add ALARM task to scheduler queue
433: *
434: * @root = root task
435: * @func = task execution function
436: * @arg = 1st func argument
437: * @ts = timeout argument structure, minimum alarm timer resolution is 1msec!
438: * @opt_data = Optional data
439: * @opt_dlen = Optional data length
440: * return: NULL error or !=NULL new queued task
441: */
442: sched_task_t *
443: schedAlarm(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, struct timespec ts,
444: void *opt_data, size_t opt_dlen)
445: {
446: sched_task_t *task;
447: void *ptr;
448:
449: if (!root || !func)
450: return NULL;
451:
452: /* get new task */
453: if (!(task = _sched_useTask(root)))
454: return NULL;
455:
456: task->task_func = func;
457: TASK_TYPE(task) = taskALARM;
458: TASK_ROOT(task) = root;
459:
460: TASK_ARG(task) = arg;
461: TASK_TS(task) = ts;
462:
463: TASK_DATA(task) = opt_data;
464: TASK_DATLEN(task) = opt_dlen;
465:
466: if (root->root_hooks.hook_add.alarm)
467: ptr = root->root_hooks.hook_add.alarm(task, NULL);
468: else
469: ptr = NULL;
470:
471: if (!ptr) {
472: #ifdef HAVE_LIBPTHREAD
473: pthread_mutex_lock(&root->root_mtx[taskALARM]);
474: #endif
1.9 misho 475: TAILQ_INSERT_TAIL(&root->root_alarm, TASK_ID(task), task_node);
1.8 misho 476: #ifdef HAVE_LIBPTHREAD
477: pthread_mutex_unlock(&root->root_mtx[taskALARM]);
478: #endif
479: } else
480: task = _sched_unuseTask(task);
481:
482: return task;
483: }
484:
485: /*
1.10.2.1 misho 486: * schedAIO() - Add AIO task to scheduler queue
487: *
488: * @root = root task
489: * @func = task execution function
490: * @arg = 1st func argument
491: * @acb = AIO cb structure address
492: * @opt_data = Optional data
493: * @opt_dlen = Optional data length
494: * return: NULL error or !=NULL new queued task
495: */
496: sched_task_t *
1.10.2.2 misho 497: schedAIO(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg,
498: struct aiocb * __restrict acb, void *opt_data, size_t opt_dlen)
1.10.2.1 misho 499: {
500: #ifndef EVFILT_AIO
501: sched_SetErr(ENOTSUP, "Not supported kevent() filter");
502: return NULL;
503: #else
504: sched_task_t *task;
505: void *ptr;
506:
507: if (!root || !func)
508: return NULL;
509:
510: /* get new task */
511: if (!(task = _sched_useTask(root)))
512: return NULL;
513:
514: task->task_func = func;
515: TASK_TYPE(task) = taskAIO;
516: TASK_ROOT(task) = root;
517:
518: TASK_ARG(task) = arg;
1.10.2.2 misho 519: TASK_VAL(task) = (u_long) acb;
1.10.2.1 misho 520:
521: TASK_DATA(task) = opt_data;
522: TASK_DATLEN(task) = opt_dlen;
523:
524: if (root->root_hooks.hook_add.aio)
525: ptr = root->root_hooks.hook_add.aio(task, NULL);
526: else
527: ptr = NULL;
528:
529: if (!ptr) {
530: #ifdef HAVE_LIBPTHREAD
531: pthread_mutex_lock(&root->root_mtx[taskAIO]);
532: #endif
533: TAILQ_INSERT_TAIL(&root->root_aio, TASK_ID(task), task_node);
534: #ifdef HAVE_LIBPTHREAD
535: pthread_mutex_unlock(&root->root_mtx[taskAIO]);
536: #endif
537: } else
538: task = _sched_unuseTask(task);
539:
540: return task;
541: #endif
542: }
543:
544: /*
1.10.2.2 misho 545: * schedAIORead() - Add AIO read task to scheduler queue
546: *
547: * @root = root task
548: * @func = task execution function
549: * @arg = 1st func argument
550: * @fd = file descriptor
551: * @buffer = Buffer
552: * @buflen = Buffer length
553: * return: NULL error or !=NULL new queued task
554: */
555: inline sched_task_t *
556: schedAIORead(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd,
557: void *buffer, size_t buflen)
558: {
559: #ifndef EVFILT_AIO
560: sched_SetErr(ENOTSUP, "Not supported kevent() filter");
561: return NULL;
562: #else
563: struct aiocb *acb;
564: off_t off = 0;
565:
566: if (!root || !func)
567: return NULL;
568: else
569: memset(buffer, 0, buflen);
570:
571: if (!(acb = malloc(sizeof(struct aiocb)))) {
572: LOGERR;
573: return NULL;
574: } else
575: memset(acb, 0, sizeof(struct aiocb));
576:
577: acb->aio_fildes = fd;
578: acb->aio_nbytes = buflen;
579: acb->aio_buf = buffer;
580: off = lseek(fd, 0, SEEK_CUR);
581: if (off == -1) {
582: LOGERR;
583: free(acb);
584: return NULL;
585: } else
586: acb->aio_offset = off;
587: acb->aio_sigevent.sigev_notify = SIGEV_KEVENT;
1.10.2.3! misho 588: acb->aio_sigevent.sigev_notify_kevent_flags = EV_CLEAR | EV_DISPATCH;
1.10.2.2 misho 589: acb->aio_sigevent.sigev_notify_kqueue = root->root_kq;
590: acb->aio_sigevent.sigev_value.sival_ptr = (void*) O_RDONLY;
591:
592: if (aio_read(acb)) {
593: LOGERR;
594: free(acb);
595: return NULL;
596: }
597:
598: return schedAIO(root, func, arg, acb, buffer, buflen);
599: #endif
600: }
601:
602: /*
603: * schedAIOWrite() - Add AIO write task to scheduler queue
604: *
605: * @root = root task
606: * @func = task execution function
607: * @arg = 1st func argument
608: * @fd = file descriptor
609: * @buffer = Buffer
610: * @buflen = Buffer length
611: * return: NULL error or !=NULL new queued task
612: */
613: inline sched_task_t *
614: schedAIOWrite(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, int fd,
615: void *buffer, size_t buflen)
616: {
617: #ifndef EVFILT_AIO
618: sched_SetErr(ENOTSUP, "Not supported kevent() filter");
619: return NULL;
620: #else
621: struct aiocb *acb;
622: off_t off = 0;
623:
624: if (!root || !func)
625: return NULL;
626: else
627: memset(buffer, 0, buflen);
628:
629: if (!(acb = malloc(sizeof(struct aiocb)))) {
630: LOGERR;
631: return NULL;
632: } else
633: memset(acb, 0, sizeof(struct aiocb));
634:
635: acb->aio_fildes = fd;
636: acb->aio_nbytes = buflen;
637: acb->aio_buf = buffer;
638: off = lseek(fd, 0, SEEK_CUR);
639: if (off == -1) {
640: LOGERR;
641: free(acb);
642: return NULL;
643: } else
644: acb->aio_offset = off;
645: acb->aio_sigevent.sigev_notify = SIGEV_KEVENT;
1.10.2.3! misho 646: acb->aio_sigevent.sigev_notify_kevent_flags = EV_CLEAR | EV_DISPATCH;
1.10.2.2 misho 647: acb->aio_sigevent.sigev_notify_kqueue = root->root_kq;
648: acb->aio_sigevent.sigev_value.sival_ptr = (void*) O_WRONLY;
649:
650: if (aio_write(acb)) {
651: LOGERR;
652: free(acb);
653: return NULL;
654: }
655:
656: return schedAIO(root, func, arg, acb, buffer, buflen);
657: #endif
658: }
659:
660: /*
1.1 misho 661: * schedTimer() - Add TIMER task to scheduler queue
1.6 misho 662: *
1.1 misho 663: * @root = root task
664: * @func = task execution function
665: * @arg = 1st func argument
1.5 misho 666: * @ts = timeout argument structure
667: * @opt_data = Optional data
668: * @opt_dlen = Optional data length
1.1 misho 669: * return: NULL error or !=NULL new queued task
670: */
671: sched_task_t *
1.5 misho 672: schedTimer(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, struct timespec ts,
673: void *opt_data, size_t opt_dlen)
1.1 misho 674: {
1.9 misho 675: sched_task_t *task, *tmp, *t = NULL;
1.1 misho 676: void *ptr;
1.5 misho 677: struct timespec now;
1.1 misho 678:
679: if (!root || !func)
680: return NULL;
681:
682: /* get new task */
1.4 misho 683: if (!(task = _sched_useTask(root)))
684: return NULL;
1.1 misho 685:
686: task->task_func = func;
1.5 misho 687: TASK_TYPE(task) = taskTIMER;
688: TASK_ROOT(task) = root;
1.1 misho 689:
690: TASK_ARG(task) = arg;
691:
1.5 misho 692: TASK_DATA(task) = opt_data;
693: TASK_DATLEN(task) = opt_dlen;
694:
1.1 misho 695: /* calculate timeval structure */
1.5 misho 696: clock_gettime(CLOCK_MONOTONIC, &now);
697: now.tv_sec += ts.tv_sec;
698: now.tv_nsec += ts.tv_nsec;
699: if (now.tv_nsec >= 1000000000L) {
1.1 misho 700: now.tv_sec++;
1.5 misho 701: now.tv_nsec -= 1000000000L;
702: } else if (now.tv_nsec < 0) {
1.1 misho 703: now.tv_sec--;
1.5 misho 704: now.tv_nsec += 1000000000L;
1.1 misho 705: }
1.5 misho 706: TASK_TS(task) = now;
1.1 misho 707:
708: if (root->root_hooks.hook_add.timer)
709: ptr = root->root_hooks.hook_add.timer(task, NULL);
710: else
711: ptr = NULL;
712:
713: if (!ptr) {
1.5 misho 714: #ifdef HAVE_LIBPTHREAD
715: pthread_mutex_lock(&root->root_mtx[taskTIMER]);
716: #endif
1.1 misho 717: #ifdef TIMER_WITHOUT_SORT
1.9 misho 718: TAILQ_INSERT_TAIL(&root->root_timer, TASK_ID(task), task_node);
1.1 misho 719: #else
1.9 misho 720: TAILQ_FOREACH_SAFE(t, &root->root_timer, task_node, tmp)
1.5 misho 721: if (sched_timespeccmp(&TASK_TS(task), &TASK_TS(t), -) < 1)
1.1 misho 722: break;
723: if (!t)
1.9 misho 724: TAILQ_INSERT_TAIL(&root->root_timer, TASK_ID(task), task_node);
1.1 misho 725: else
1.9 misho 726: TAILQ_INSERT_BEFORE(t, TASK_ID(task), task_node);
1.1 misho 727: #endif
1.5 misho 728: #ifdef HAVE_LIBPTHREAD
729: pthread_mutex_unlock(&root->root_mtx[taskTIMER]);
730: #endif
1.4 misho 731: } else
732: task = _sched_unuseTask(task);
1.1 misho 733:
734: return task;
735: }
736:
737: /*
738: * schedEvent() - Add EVENT task to scheduler queue
1.6 misho 739: *
1.1 misho 740: * @root = root task
741: * @func = task execution function
742: * @arg = 1st func argument
1.2 misho 743: * @val = additional func argument
1.5 misho 744: * @opt_data = Optional data
745: * @opt_dlen = Optional data length
1.1 misho 746: * return: NULL error or !=NULL new queued task
747: */
748: sched_task_t *
1.5 misho 749: schedEvent(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long val,
750: void *opt_data, size_t opt_dlen)
1.1 misho 751: {
752: sched_task_t *task;
753: void *ptr;
754:
755: if (!root || !func)
756: return NULL;
757:
758: /* get new task */
1.4 misho 759: if (!(task = _sched_useTask(root)))
760: return NULL;
1.1 misho 761:
762: task->task_func = func;
1.5 misho 763: TASK_TYPE(task) = taskEVENT;
764: TASK_ROOT(task) = root;
1.1 misho 765:
766: TASK_ARG(task) = arg;
1.2 misho 767: TASK_VAL(task) = val;
1.1 misho 768:
1.5 misho 769: TASK_DATA(task) = opt_data;
770: TASK_DATLEN(task) = opt_dlen;
771:
1.1 misho 772: if (root->root_hooks.hook_add.event)
773: ptr = root->root_hooks.hook_add.event(task, NULL);
774: else
775: ptr = NULL;
776:
1.5 misho 777: if (!ptr) {
778: #ifdef HAVE_LIBPTHREAD
779: pthread_mutex_lock(&root->root_mtx[taskEVENT]);
780: #endif
1.9 misho 781: TAILQ_INSERT_TAIL(&root->root_event, TASK_ID(task), task_node);
1.5 misho 782: #ifdef HAVE_LIBPTHREAD
783: pthread_mutex_unlock(&root->root_mtx[taskEVENT]);
784: #endif
785: } else
1.4 misho 786: task = _sched_unuseTask(task);
1.1 misho 787:
788: return task;
789: }
790:
791:
792: /*
793: * schedEventLo() - Add EVENT_Lo task to scheduler queue
1.6 misho 794: *
1.1 misho 795: * @root = root task
796: * @func = task execution function
797: * @arg = 1st func argument
1.2 misho 798: * @val = additional func argument
1.5 misho 799: * @opt_data = Optional data
800: * @opt_dlen = Optional data length
1.1 misho 801: * return: NULL error or !=NULL new queued task
802: */
803: sched_task_t *
1.5 misho 804: schedEventLo(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long val,
805: void *opt_data, size_t opt_dlen)
1.1 misho 806: {
807: sched_task_t *task;
808: void *ptr;
809:
810: if (!root || !func)
811: return NULL;
812:
813: /* get new task */
1.4 misho 814: if (!(task = _sched_useTask(root)))
815: return NULL;
1.1 misho 816:
817: task->task_func = func;
1.5 misho 818: TASK_TYPE(task) = taskEVENT;
819: TASK_ROOT(task) = root;
1.1 misho 820:
821: TASK_ARG(task) = arg;
1.2 misho 822: TASK_VAL(task) = val;
1.1 misho 823:
1.5 misho 824: TASK_DATA(task) = opt_data;
825: TASK_DATLEN(task) = opt_dlen;
826:
1.1 misho 827: if (root->root_hooks.hook_add.eventlo)
828: ptr = root->root_hooks.hook_add.eventlo(task, NULL);
829: else
830: ptr = NULL;
831:
1.5 misho 832: if (!ptr) {
833: #ifdef HAVE_LIBPTHREAD
834: pthread_mutex_lock(&root->root_mtx[taskEVENTLO]);
835: #endif
1.9 misho 836: TAILQ_INSERT_TAIL(&root->root_eventlo, TASK_ID(task), task_node);
1.5 misho 837: #ifdef HAVE_LIBPTHREAD
838: pthread_mutex_unlock(&root->root_mtx[taskEVENTLO]);
839: #endif
840: } else
1.4 misho 841: task = _sched_unuseTask(task);
1.1 misho 842:
843: return task;
844: }
845:
846: /*
1.10 misho 847: * schedSuspend() - Add Suspended task to scheduler queue
848: *
849: * @root = root task
850: * @func = task execution function
851: * @arg = 1st func argument
852: * @id = Trigger ID
853: * @opt_data = Optional data
854: * @opt_dlen = Optional data length
855: * return: NULL error or !=NULL new queued task
856: */
857: sched_task_t *
858: schedSuspend(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long id,
859: void *opt_data, size_t opt_dlen)
860: {
861: sched_task_t *task;
862: void *ptr;
863:
864: if (!root || !func)
865: return NULL;
866:
867: /* get new task */
868: if (!(task = _sched_useTask(root)))
869: return NULL;
870:
871: task->task_func = func;
872: TASK_TYPE(task) = taskSUSPEND;
873: TASK_ROOT(task) = root;
874:
875: TASK_ARG(task) = arg;
876: TASK_VAL(task) = id;
877:
878: TASK_DATA(task) = opt_data;
879: TASK_DATLEN(task) = opt_dlen;
880:
881: if (root->root_hooks.hook_add.suspend)
882: ptr = root->root_hooks.hook_add.suspend(task, NULL);
883: else
884: ptr = NULL;
885:
886: if (!ptr) {
887: #ifdef HAVE_LIBPTHREAD
888: pthread_mutex_lock(&root->root_mtx[taskSUSPEND]);
889: #endif
890: TAILQ_INSERT_TAIL(&root->root_suspend, TASK_ID(task), task_node);
891: #ifdef HAVE_LIBPTHREAD
892: pthread_mutex_unlock(&root->root_mtx[taskSUSPEND]);
893: #endif
894: } else
895: task = _sched_unuseTask(task);
896:
897: return task;
898: }
899:
900: /*
1.1 misho 901: * schedCallOnce() - Call once from scheduler
1.6 misho 902: *
1.1 misho 903: * @root = root task
904: * @func = task execution function
905: * @arg = 1st func argument
1.2 misho 906: * @val = additional func argument
1.5 misho 907: * @opt_data = Optional data
908: * @opt_dlen = Optional data length
1.2 misho 909: * return: return value from called func
1.1 misho 910: */
911: sched_task_t *
1.5 misho 912: schedCallOnce(sched_root_task_t * __restrict root, sched_task_func_t func, void *arg, u_long val,
913: void *opt_data, size_t opt_dlen)
1.1 misho 914: {
915: sched_task_t *task;
1.2 misho 916: void *ret;
1.1 misho 917:
918: if (!root || !func)
919: return NULL;
920:
921: /* get new task */
1.4 misho 922: if (!(task = _sched_useTask(root)))
923: return NULL;
1.1 misho 924:
925: task->task_func = func;
1.5 misho 926: TASK_TYPE(task) = taskEVENT;
927: TASK_ROOT(task) = root;
1.1 misho 928:
929: TASK_ARG(task) = arg;
1.2 misho 930: TASK_VAL(task) = val;
1.1 misho 931:
1.5 misho 932: TASK_DATA(task) = opt_data;
933: TASK_DATLEN(task) = opt_dlen;
934:
1.2 misho 935: ret = schedCall(task);
1.1 misho 936:
1.4 misho 937: _sched_unuseTask(task);
1.2 misho 938: return ret;
1.1 misho 939: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>