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

version 1.4.2.1, 2012/01/08 02:01:41 version 1.4.2.6, 2012/01/24 14:51:03
Line 132  schedInit(void ** __restrict data, size_t datlen) Line 132  schedInit(void ** __restrict data, size_t datlen)
         } else {          } else {
                 memset(root, 0, sizeof(sched_root_task_t));                  memset(root, 0, sizeof(sched_root_task_t));
   
                   /* INFINIT polling period by default */
                   root->root_poll.tv_sec = root->root_poll.tv_nsec = -1;
   
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
                 for (i = 0; i < taskMAX; i++)                  for (i = 0; i < taskMAX; i++)
                         if (pthread_mutex_init(&root->root_mtx[i], NULL)) {                          if (pthread_mutex_init(&root->root_mtx[i], NULL)) {
Line 143  schedInit(void ** __restrict data, size_t datlen) Line 146  schedInit(void ** __restrict data, size_t datlen)
                         }                          }
   
                 for (i = 0; i < taskMAX; i++)                  for (i = 0; i < taskMAX; i++)
                        ROOT_QLOCK(root, i);                        pthread_mutex_lock(&root->root_mtx[i]);
 #endif  #endif
   
                 TAILQ_INIT(&root->root_read);                  TAILQ_INIT(&root->root_read);
Line 156  schedInit(void ** __restrict data, size_t datlen) Line 159  schedInit(void ** __restrict data, size_t datlen)
   
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
                 for (i = 0; i < taskMAX; i++)                  for (i = 0; i < taskMAX; i++)
                        ROOT_QUNLOCK(root, i);                        pthread_mutex_unlock(&root->root_mtx[i]);
 #endif  #endif
   
                 if (data && *data) {                  if (data && *data) {
Line 193  schedEnd(sched_root_task_t ** __restrict root) Line 196  schedEnd(sched_root_task_t ** __restrict root)
         if (!root || !*root)          if (!root || !*root)
                 return -1;                  return -1;
   
 #ifdef HAVE_LIBPTHREAD  
         for (i = 0; i < taskMAX; i++)  
                 ROOT_QLOCK(*root, i);  
 #endif  
         TAILQ_FOREACH(task, &(*root)->root_read, task_node) {          TAILQ_FOREACH(task, &(*root)->root_read, task_node) {
                 schedCancel(task);                  schedCancel(task);
         }          }
Line 209  schedEnd(sched_root_task_t ** __restrict root) Line 208  schedEnd(sched_root_task_t ** __restrict root)
         TAILQ_FOREACH(task, &(*root)->root_event, task_node) {          TAILQ_FOREACH(task, &(*root)->root_event, task_node) {
                 schedCancel(task);                  schedCancel(task);
         }          }
           TAILQ_FOREACH(task, &(*root)->root_eventlo, task_node) {
                   schedCancel(task);
           }
         TAILQ_FOREACH(task, &(*root)->root_ready, task_node) {          TAILQ_FOREACH(task, &(*root)->root_ready, task_node) {
                 schedCancel(task);                  schedCancel(task);
         }          }
   
   #ifdef HAVE_LIBPTHREAD
           pthread_mutex_lock(&(*root)->root_mtx[taskUNUSE]);
   #endif
         while ((task = TAILQ_FIRST(&(*root)->root_unuse))) {          while ((task = TAILQ_FIRST(&(*root)->root_unuse))) {
                 TAILQ_REMOVE(&(*root)->root_unuse, task, task_node);                  TAILQ_REMOVE(&(*root)->root_unuse, task, task_node);
                 free(task);                  free(task);
         }          }
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
        for (i = 0; i < taskMAX; i++)        pthread_mutex_unlock(&(*root)->root_mtx[taskUNUSE]);
                ROOT_QUNLOCK(*root, i); 
 #endif  #endif
   
         if ((*root)->root_hooks.hook_root.fini)          if ((*root)->root_hooks.hook_root.fini)
Line 289  schedCancel(sched_task_t * __restrict task) Line 293  schedCancel(sched_task_t * __restrict task)
 {  {
         sched_queue_t *queue;          sched_queue_t *queue;
   
        if (!task || !task->task_root)        if (!task || !TASK_ROOT(task))
                 return -1;                  return -1;
   
        if (task->task_root->root_hooks.hook_exec.cancel)        if (TASK_ROOT(task)->root_hooks.hook_exec.cancel)
                if (task->task_root->root_hooks.hook_exec.cancel(task, NULL))                if (TASK_ROOT(task)->root_hooks.hook_exec.cancel(task, NULL))
                         return -1;                          return -1;
   
        switch (task->task_type) {        switch (TASK_TYPE(task)) {
                 case taskREAD:                  case taskREAD:
                        queue = &task->task_root->root_read;                        queue = &TASK_ROOT(task)->root_read;
                         break;                          break;
                 case taskWRITE:                  case taskWRITE:
                        queue = &task->task_root->root_write;                        queue = &TASK_ROOT(task)->root_write;
                         break;                          break;
                 case taskTIMER:                  case taskTIMER:
                        queue = &task->task_root->root_timer;                        queue = &TASK_ROOT(task)->root_timer;
                         break;                          break;
                 case taskEVENT:                  case taskEVENT:
                        queue = &task->task_root->root_event;                        queue = &TASK_ROOT(task)->root_event;
                         break;                          break;
                   case taskEVENTLO:
                           queue = &TASK_ROOT(task)->root_eventlo;
                           break;
                 case taskREADY:                  case taskREADY:
                        queue = &task->task_root->root_ready;                        queue = &TASK_ROOT(task)->root_ready;
                         break;                          break;
                 default:                  default:
                         queue = NULL;                          queue = NULL;
         }          }
        if (queue)        if (queue) {
 #ifdef HAVE_LIBPTHREAD
                 pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);
 #endif
                 TAILQ_REMOVE(queue, task, task_node);                  TAILQ_REMOVE(queue, task, task_node);
        if (task->task_type != taskUNUSE)#ifdef HAVE_LIBPTHREAD
                 pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);
 #endif
         }
         if (TASK_TYPE(task) != taskUNUSE)
                 _sched_unuseTask(task);                  _sched_unuseTask(task);
   
         return 0;          return 0;
Line 326  schedCancel(sched_task_t * __restrict task) Line 340  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
 * @queue = cancel from queue, if =NULL 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_TV]
  * @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
  */   */
 int  int
schedCancelby(sched_root_task_t * __restrict root, sched_queue_t * __restrict queue, 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;
           sched_queue_t *queue;
         int flg = 0;          int flg = 0;
   
         if (!root)          if (!root)
                 return -1;                  return -1;
        if (!queue) {        if (type == taskMAX) {
                if (schedCancelby(root, &root->root_read, criteria, param, hook))                if (schedCancelby(root, taskREAD, criteria, param, hook))
                         return -2;                          return -2;
                if (schedCancelby(root, &root->root_write, criteria, param, hook))                if (schedCancelby(root, taskWRITE, criteria, param, hook))
                         return -2;                          return -2;
                if (schedCancelby(root, &root->root_timer, criteria, param, hook))                if (schedCancelby(root, taskTIMER, criteria, param, hook))
                         return -2;                          return -2;
                if (schedCancelby(root, &root->root_event, criteria, param, hook))                if (schedCancelby(root, taskEVENT, criteria, param, hook))
                         return -2;                          return -2;
                if (schedCancelby(root, &root->root_ready, criteria, param, hook))                if (schedCancelby(root, taskEVENTLO, criteria, param, hook))
                         return -2;                          return -2;
                if (schedCancelby(root, &root->root_read, criteria, param, hook))                if (schedCancelby(root, taskREADY, criteria, param, hook))
                         return -2;                          return -2;
                 return 0;                  return 0;
         }          }
           switch (type) {
                   case taskREAD:
                           queue = &root->root_read;
                           break;
                   case taskWRITE:
                           queue = &root->root_write;
                           break;
                   case taskTIMER:
                           queue = &root->root_timer;
                           break;
                   case taskEVENT:
                           queue = &root->root_event;
                           break;
                   case taskEVENTLO:
                           queue = &root->root_eventlo;
                           break;
                   case taskREADY:
                           queue = &root->root_ready;
                           break;
                   default:
                           return 0;
           }
   
   #ifdef HAVE_LIBPTHREAD
           pthread_mutex_lock(&root->root_mtx[type]);
   #endif
         TAILQ_FOREACH(task, queue, task_node)          TAILQ_FOREACH(task, queue, task_node)
                 if (criteria == CRITERIA_CALL) {                  if (criteria == CRITERIA_CALL) {
                         if (task->task_func == (sched_task_func_t) param) {                          if (task->task_func == (sched_task_func_t) param) {
Line 379  schedCancelby(sched_root_task_t * __restrict root, sch Line 419  schedCancelby(sched_root_task_t * __restrict root, sch
                                 break;                                  break;
                         }                          }
                 } else if (criteria == CRITERIA_TV) {                  } else if (criteria == CRITERIA_TV) {
                        if (!timercmp(&TASK_TV(task), (struct timeval*) param, -)) {                        if (!timespeccmp(&TASK_TS(task), (struct timespec*) param, -)) {
                                 flg++;                                  flg++;
                                 break;                                  break;
                         }                          }
                 } else {                  } else {
   #ifdef HAVE_LIBPTHREAD
                           pthread_mutex_unlock(&root->root_mtx[type]);
   #endif
                         sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria);                          sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria);
                         return -1;                          return -1;
                 }                  }
   #ifdef HAVE_LIBPTHREAD
           pthread_mutex_unlock(&root->root_mtx[type]);
   #endif
         if (!flg || !task)      /* task not found */          if (!flg || !task)      /* task not found */
                 return 0;                  return 0;
   
        if (task->task_root->root_hooks.hook_exec.cancel)        if (TASK_ROOT(task)->root_hooks.hook_exec.cancel)
                if (task->task_root->root_hooks.hook_exec.cancel(task, NULL))                if (TASK_ROOT(task)->root_hooks.hook_exec.cancel(task, NULL))
                         return -1;                          return -1;
         if (hook)          if (hook)
                 if (hook(task, NULL))                  if (hook(task, NULL))
                         return -3;                          return -3;
   
   #ifdef HAVE_LIBPTHREAD
           pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[type]);
   #endif
         TAILQ_REMOVE(queue, task, task_node);          TAILQ_REMOVE(queue, task, task_node);
   #ifdef HAVE_LIBPTHREAD
           pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[type]);
   #endif
   
        if (task->task_type != taskUNUSE)        if (TASK_TYPE(task) != taskUNUSE)
                 _sched_unuseTask(task);                  _sched_unuseTask(task);
         return 0;          return 0;
 }  }
Line 431  schedRun(sched_root_task_t * __restrict root, volatile Line 483  schedRun(sched_root_task_t * __restrict root, volatile
                         while ((task = root->root_hooks.hook_exec.fetch(root, NULL)))                          while ((task = root->root_hooks.hook_exec.fetch(root, NULL)))
                                 schedCall(task);                                  schedCall(task);
         }          }
   
           return 0;
   }
   
   /*
    * schedPolling() - Polling timeout period if no timer task is present
    * @root = root task
    * @ts = timeout polling period, if ==NULL INFINIT timeout
    * @tsold = old timeout polling if !=NULL
    * return: -1 error or 0 ok
    */
   inline int
   schedPolling(sched_root_task_t * __restrict root, struct timespec * __restrict ts, 
                   struct timespec * __restrict tsold)
   {
           if (!root)
                   return -1;
   
           if (tsold)
                   *tsold = root->root_poll;
   
           if (!ts)
                   root->root_poll.tv_sec = root->root_poll.tv_nsec = -1;
           else
                   root->root_poll = *ts;
   
         return 0;          return 0;
 }  }

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


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