Diff for /libaitsched/src/aitsched.c between versions 1.4 and 1.4.2.4

version 1.4, 2012/01/08 00:51:17 version 1.4.2.4, 2012/01/24 14:04:58
Line 122  schedInit(void ** __restrict data, size_t datlen) Line 122  schedInit(void ** __restrict data, size_t datlen)
 {  {
         sched_root_task_t *root = NULL;          sched_root_task_t *root = NULL;
         int (*func)(sched_root_task_t *);          int (*func)(sched_root_task_t *);
   #ifdef HAVE_LIBPTHREAD
           register int i;
   #endif
   
         root = malloc(sizeof(sched_root_task_t));          root = malloc(sizeof(sched_root_task_t));
         if (!root) {          if (!root) {
                 LOGERR;                  LOGERR;
         } else {          } else {
                 memset(root, 0, sizeof(sched_root_task_t));                  memset(root, 0, sizeof(sched_root_task_t));
   
   #ifdef HAVE_LIBPTHREAD
                   for (i = 0; i < taskMAX; i++)
                           if (pthread_mutex_init(&root->root_mtx[i], NULL)) {
                                   LOGERR;
                                   while (i)
                                           pthread_mutex_destroy(&root->root_mtx[--i]);
                                   free(root);
                                   return NULL;
                           }
   
                   for (i = 0; i < taskMAX; i++)
                           pthread_mutex_lock(&root->root_mtx[i]);
   #endif
   
                 TAILQ_INIT(&root->root_read);                  TAILQ_INIT(&root->root_read);
                 TAILQ_INIT(&root->root_write);                  TAILQ_INIT(&root->root_write);
                 TAILQ_INIT(&root->root_timer);                  TAILQ_INIT(&root->root_timer);
Line 136  schedInit(void ** __restrict data, size_t datlen) Line 154  schedInit(void ** __restrict data, size_t datlen)
                 TAILQ_INIT(&root->root_ready);                  TAILQ_INIT(&root->root_ready);
                 TAILQ_INIT(&root->root_unuse);                  TAILQ_INIT(&root->root_unuse);
   
   #ifdef HAVE_LIBPTHREAD
                   for (i = 0; i < taskMAX; i++)
                           pthread_mutex_unlock(&root->root_mtx[i]);
   #endif
   
                 if (data && *data) {                  if (data && *data) {
                         if (datlen) {                          if (datlen) {
                                 root->root_data.iov_base = *data;                                  root->root_data.iov_base = *data;
Line 163  int Line 186  int
 schedEnd(sched_root_task_t ** __restrict root)  schedEnd(sched_root_task_t ** __restrict root)
 {  {
         sched_task_t *task;          sched_task_t *task;
   #ifdef HAVE_LIBPTHREAD
           register int i;
   #endif
   
         if (!root || !*root)          if (!root || !*root)
                 return -1;                  return -1;
Line 179  schedEnd(sched_root_task_t ** __restrict root) Line 205  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
           pthread_mutex_unlock(&(*root)->root_mtx[taskUNUSE]);
   #endif
   
         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);
   
   #ifdef HAVE_LIBPTHREAD
           for (i = 0; i < taskMAX; i++)
                   pthread_mutex_destroy(&(*root)->root_mtx[i]);
   #endif
   
         free(*root);          free(*root);
         *root = NULL;          *root = NULL;
         return 0;          return 0;
Line 250  schedCancel(sched_task_t * __restrict task) Line 290  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 287  schedCancel(sched_task_t * __restrict task) Line 337  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 340  schedCancelby(sched_root_task_t * __restrict root, sch Line 416  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;
 }  }

Removed from v.1.4  
changed lines
  Added in v.1.4.2.4


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