Diff for /libaitsched/src/aitsched.c between versions 1.4.2.6 and 1.7.2.1

version 1.4.2.6, 2012/01/24 14:51:03 version 1.7.2.1, 2012/05/15 15:47:12
Line 12  terms: Line 12  terms:
 All of the documentation and software included in the ELWIX and AITNET  All of the documentation and software included in the ELWIX and AITNET
 Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>  Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   
Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
         by Michael Pounov <misho@elwix.org>.  All rights reserved.          by Michael Pounov <misho@elwix.org>.  All rights reserved.
   
 Redistribution and use in source and binary forms, with or without  Redistribution and use in source and binary forms, with or without
Line 86  sched_SetErr(int eno, char *estr, ...) Line 86  sched_SetErr(int eno, char *estr, ...)
   
 /*  /*
  * schedRegisterHooks() - Register IO handles and bind tasks to it   * schedRegisterHooks() - Register IO handles and bind tasks to it
    *
  * @root = root task   * @root = root task
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
  */   */
 int  int
 schedRegisterHooks(sched_root_task_t * __restrict root)  schedRegisterHooks(sched_root_task_t * __restrict root)
 {  {
        if (!root || (root->root_data.iov_base && root->root_data.iov_len))        assert(root);
                return -1; 
   
         if (root->root_hooks.hook_root.fini)          if (root->root_hooks.hook_root.fini)
                 root->root_hooks.hook_root.fini(root, NULL);                  root->root_hooks.hook_root.fini(root, NULL);
Line 113  schedRegisterHooks(sched_root_task_t * __restrict root Line 113  schedRegisterHooks(sched_root_task_t * __restrict root
   
 /*  /*
  * schedInit() - Init scheduler   * schedInit() - Init scheduler
    *
  * @data = optional data if !=NULL   * @data = optional data if !=NULL
  * @datlen = data len if data is set   * @datlen = data len if data is set
  * return: allocated root task if ok or NULL error   * return: allocated root task if ok or NULL error
Line 133  schedInit(void ** __restrict data, size_t datlen) Line 134  schedInit(void ** __restrict data, size_t datlen)
                 memset(root, 0, sizeof(sched_root_task_t));                  memset(root, 0, sizeof(sched_root_task_t));
   
                 /* INFINIT polling period by default */                  /* INFINIT polling period by default */
                root->root_poll.tv_sec = root->root_poll.tv_nsec = -1;                sched_timespecinf(&root->root_poll);
   
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
                 for (i = 0; i < taskMAX; i++)                  for (i = 0; i < taskMAX; i++)
Line 182  schedInit(void ** __restrict data, size_t datlen) Line 183  schedInit(void ** __restrict data, size_t datlen)
   
 /*  /*
  * schedEnd() - End scheduler & free all resources   * schedEnd() - End scheduler & free all resources
    *
  * @root = root task   * @root = root task
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
  */   */
 int  int
 schedEnd(sched_root_task_t ** __restrict root)  schedEnd(sched_root_task_t ** __restrict root)
 {  {
        sched_task_t *task;        sched_task_t *task, *tmp;
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
         register int i;          register int i;
 #endif  #endif
Line 196  schedEnd(sched_root_task_t ** __restrict root) Line 198  schedEnd(sched_root_task_t ** __restrict root)
         if (!root || !*root)          if (!root || !*root)
                 return -1;                  return -1;
   
        TAILQ_FOREACH(task, &(*root)->root_read, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_read, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
        TAILQ_FOREACH(task, &(*root)->root_write, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_write, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
        TAILQ_FOREACH(task, &(*root)->root_timer, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_timer, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
        TAILQ_FOREACH(task, &(*root)->root_event, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_event, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
        TAILQ_FOREACH(task, &(*root)->root_eventlo, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_eventlo, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
        TAILQ_FOREACH(task, &(*root)->root_ready, task_node) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_ready, task_node, tmp) {
                 schedCancel(task);                  schedCancel(task);
         }          }
   
Line 241  schedEnd(sched_root_task_t ** __restrict root) Line 243  schedEnd(sched_root_task_t ** __restrict root)
   
 /*  /*
  * schedCall() - Call task execution function   * schedCall() - Call task execution function
    *
  * @task = current task   * @task = current task
  * return: !=NULL error or =NULL ok   * return: !=NULL error or =NULL ok
  */   */
Line 264  schedCall(sched_task_t * __restrict task) Line 267  schedCall(sched_task_t * __restrict task)
   
 /*  /*
  * schedFetch() - Fetch ready task   * schedFetch() - Fetch ready task
    *
  * @root = root task   * @root = root task
  * return: =NULL error or !=NULL ready task   * return: =NULL error or !=NULL ready task
  */   */
Line 285  schedFetch(sched_root_task_t * __restrict root) Line 289  schedFetch(sched_root_task_t * __restrict root)
   
 /*  /*
  * schedCancel() - Cancel task from scheduler   * schedCancel() - Cancel task from scheduler
    *
  * @task = task   * @task = task
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
  */   */
Line 339  schedCancel(sched_task_t * __restrict task) Line 344  schedCancel(sched_task_t * __restrict task)
   
 /*  /*
  * schedCancelby() - Cancel task from scheduler by criteria   * schedCancelby() - Cancel task from scheduler by criteria
    *
  * @root = root task   * @root = root task
  * @type = cancel from queue type, if =taskMAX cancel same task from all queues   * @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_TV] * @criteria = find task by criteria [CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_TS]
  * @param = search parameter   * @param = search parameter
  * @hook = custom cleanup hook function, may be NULL   * @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   * return: -1 error, -2 error in sub-stage cancel execution, -3 error from custom hook or 0 ok
Line 350  int Line 356  int
 schedCancelby(sched_root_task_t * __restrict root, sched_task_type_t type,   schedCancelby(sched_root_task_t * __restrict root, sched_task_type_t type, 
                 u_char criteria, void *param, sched_hook_func_t hook)                  u_char criteria, void *param, sched_hook_func_t hook)
 {  {
        sched_task_t *task;        sched_task_t *task, *tmp;
         sched_queue_t *queue;          sched_queue_t *queue;
        int flg = 0;        register int flg = 0;
   
         if (!root)          if (!root)
                 return -1;                  return -1;
Line 397  schedCancelby(sched_root_task_t * __restrict root, sch Line 403  schedCancelby(sched_root_task_t * __restrict root, sch
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
         pthread_mutex_lock(&root->root_mtx[type]);          pthread_mutex_lock(&root->root_mtx[type]);
 #endif  #endif
        TAILQ_FOREACH(task, queue, task_node)        TAILQ_FOREACH_SAFE(task, queue, task_node, tmp) {
                if (criteria == CRITERIA_CALL) {                flg ^= flg;
                        if (task->task_func == (sched_task_func_t) param) {                switch (criteria) {
                                flg++;                        case CRITERIA_CALL:
                                 if (TASK_FUNC(task) == (sched_task_func_t) param)
                                         flg = 1;
                                 break;                                  break;
                        }                        case CRITERIA_ARG:
                } else if (criteria == CRITERIA_ARG) {                                if (TASK_ARG(task) == param)
                        if (task->task_arg == param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_FD:
                } else if (criteria == CRITERIA_FD) {                                if (TASK_FD(task) == (intptr_t) param)
                        if (TASK_FD(task) == (intptr_t) param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_VAL:
                } else if (criteria == CRITERIA_VAL) {                                if (TASK_VAL(task) == (u_long) param)
                        if (TASK_VAL(task) == (u_long) param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_TS:
                } else if (criteria == CRITERIA_TV) {                                if (!sched_timespeccmp(&TASK_TS(task), (struct timespec*) param, -))
                        if (!timespeccmp(&TASK_TS(task), (struct timespec*) param, -)) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        default:
                } else {                                sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria);
#ifdef HAVE_LIBPTHREAD                                flg = -1;
                        pthread_mutex_unlock(&root->root_mtx[type]); 
#endif 
                        sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria); 
                        return -1; 
                 }                  }
#ifdef HAVE_LIBPTHREAD                if (flg < 0)
        pthread_mutex_unlock(&root->root_mtx[type]);                        break;
#endif                /* cancel choosen task */
        if (!flg || !task)   /* task not found */                if (flg > 0) {
                return 0;                        if (TASK_ROOT(task)->root_hooks.hook_exec.cancel)
                                 if (TASK_ROOT(task)->root_hooks.hook_exec.cancel(task, NULL)) {
                                         flg = -1;
                                         break;
                                 }
                         /* custom hook */
                         if (hook)
                                 if (hook(task, NULL)) {
                                         flg = -3;
                                         break;
                                 }
   
        if (TASK_ROOT(task)->root_hooks.hook_exec.cancel)                        TAILQ_REMOVE(queue, task, task_node);
                if (TASK_ROOT(task)->root_hooks.hook_exec.cancel(task, NULL))                        if (TASK_TYPE(task) != taskUNUSE)
                        return -1;                                _sched_unuseTask(task);
        if (hook) 
                if (hook(task, NULL)) 
                        return -3; 
   
                           flg ^= flg;     /* ok */
                   }
           }
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
        pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[type]);        pthread_mutex_unlock(&root->root_mtx[type]);
 #endif  #endif
        TAILQ_REMOVE(queue, task, task_node);        return flg;
#ifdef HAVE_LIBPTHREAD 
        pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[type]); 
#endif 
 
        if (TASK_TYPE(task) != taskUNUSE) 
                _sched_unuseTask(task); 
        return 0; 
 }  }
   
 /*  /*
  * schedRun() - Scheduler *run loop*   * schedRun() - Scheduler *run loop*
    *
  * @root = root task   * @root = root task
  * @killState = kill condition variable, if !=0 stop scheduler loop   * @killState = kill condition variable, if !=0 stop scheduler loop
  * return: -1 error or 0 ok   * return: -1 error or 0 ok
  */   */
 int  int
schedRun(sched_root_task_t * __restrict root, volatile intptr_t * __restrict killState)schedRun(sched_root_task_t *root, volatile intptr_t * __restrict killState)
 {  {
         sched_task_t *task;          sched_task_t *task;
   
Line 473  schedRun(sched_root_task_t * __restrict root, volatile Line 477  schedRun(sched_root_task_t * __restrict root, volatile
         if (root->root_hooks.hook_exec.run)          if (root->root_hooks.hook_exec.run)
                 if (root->root_hooks.hook_exec.run(root, NULL))                  if (root->root_hooks.hook_exec.run(root, NULL))
                         return -1;                          return -1;
        if (root->root_hooks.hook_exec.fetch) {
                if (killState)        if (killState) {
                        while (!*killState) {                if (root->root_hooks.hook_exec.condition)
                         /* condition scheduler loop */
                         while (root && root->root_hooks.hook_exec.fetch && 
                                         root->root_hooks.hook_exec.condition && 
                                         root->root_hooks.hook_exec.condition(root, (void*) killState)) {
                                 if ((task = root->root_hooks.hook_exec.fetch(root, NULL)))                                  if ((task = root->root_hooks.hook_exec.fetch(root, NULL)))
                                         schedCall(task);                                          schedCall(task);
                         }                          }
                 else                  else
                        while ((task = root->root_hooks.hook_exec.fetch(root, NULL)))                        /* trigger scheduler loop */
                         while (!*killState && root && root->root_hooks.hook_exec.fetch) {
                                 if ((task = root->root_hooks.hook_exec.fetch(root, NULL)))
                                         schedCall(task);
                         }
         } else
                 /* infinite scheduler loop */
                 while (root && root->root_hooks.hook_exec.fetch)
                         if ((task = root->root_hooks.hook_exec.fetch(root, NULL)))
                                 schedCall(task);                                  schedCall(task);
         }  
   
         return 0;          return 0;
 }  }
   
 /*  /*
  * schedPolling() - Polling timeout period if no timer task is present   * schedPolling() - Polling timeout period if no timer task is present
    *
  * @root = root task   * @root = root task
  * @ts = timeout polling period, if ==NULL INFINIT timeout   * @ts = timeout polling period, if ==NULL INFINIT timeout
  * @tsold = old timeout polling if !=NULL   * @tsold = old timeout polling if !=NULL
Line 505  schedPolling(sched_root_task_t * __restrict root, stru Line 521  schedPolling(sched_root_task_t * __restrict root, stru
                 *tsold = root->root_poll;                  *tsold = root->root_poll;
   
         if (!ts)          if (!ts)
                root->root_poll.tv_sec = root->root_poll.tv_nsec = -1;                sched_timespecinf(&root->root_poll);
         else          else
                 root->root_poll = *ts;                  root->root_poll = *ts;
   
           return 0;
   }
   
   /*
    * schedTermCondition() - Activate hook for scheduler condition kill
    *
    * @root = root task
    * @condValue = condition value, kill schedRun() if condValue == killState
    * return: -1 error ok 0 ok
    */
   inline int
   schedTermCondition(sched_root_task_t * __restrict root, intptr_t condValue)
   {
           if (!root)
                   return -1;
   
           root->root_cond = condValue;
           root->root_hooks.hook_exec.condition = sched_hook_condition;
         return 0;          return 0;
 }  }

Removed from v.1.4.2.6  
changed lines
  Added in v.1.7.2.1


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