Diff for /libaitsched/src/aitsched.c between versions 1.6.2.4 and 1.9.2.2

version 1.6.2.4, 2012/05/03 15:05:09 version 1.9.2.2, 2012/05/31 21:36:40
Line 101  schedRegisterHooks(sched_root_task_t * __restrict root Line 101  schedRegisterHooks(sched_root_task_t * __restrict root
   
         root->root_hooks.hook_add.read = sched_hook_read;          root->root_hooks.hook_add.read = sched_hook_read;
         root->root_hooks.hook_add.write = sched_hook_write;          root->root_hooks.hook_add.write = sched_hook_write;
           root->root_hooks.hook_add.alarm = sched_hook_alarm;
           root->root_hooks.hook_add.node = sched_hook_node;
           root->root_hooks.hook_add.proc = sched_hook_proc;
           root->root_hooks.hook_add.user = sched_hook_user;
           root->root_hooks.hook_add.signal = sched_hook_signal;
   
         root->root_hooks.hook_exec.cancel = sched_hook_cancel;          root->root_hooks.hook_exec.cancel = sched_hook_cancel;
         root->root_hooks.hook_exec.fetch = sched_hook_fetch;          root->root_hooks.hook_exec.fetch = sched_hook_fetch;
Line 153  schedInit(void ** __restrict data, size_t datlen) Line 158  schedInit(void ** __restrict data, size_t datlen)
                 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);
                   TAILQ_INIT(&root->root_alarm);
                   TAILQ_INIT(&root->root_node);
                   TAILQ_INIT(&root->root_proc);
                   TAILQ_INIT(&root->root_user);
                   TAILQ_INIT(&root->root_signal);
                 TAILQ_INIT(&root->root_event);                  TAILQ_INIT(&root->root_event);
                 TAILQ_INIT(&root->root_eventlo);                  TAILQ_INIT(&root->root_eventlo);
                 TAILQ_INIT(&root->root_ready);                  TAILQ_INIT(&root->root_ready);
Line 190  schedInit(void ** __restrict data, size_t datlen) Line 200  schedInit(void ** __restrict data, size_t datlen)
 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 198  schedEnd(sched_root_task_t ** __restrict root) Line 208  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_SAFE(task, &(*root)->root_write, task_node, tmp)
        TAILQ_FOREACH(task, &(*root)->root_write, task_node) { 
                 schedCancel(task);                  schedCancel(task);
        }        TAILQ_FOREACH_SAFE(task, &(*root)->root_timer, task_node, tmp)
        TAILQ_FOREACH(task, &(*root)->root_timer, task_node) { 
                 schedCancel(task);                  schedCancel(task);
        }        TAILQ_FOREACH_SAFE(task, &(*root)->root_alarm, task_node, tmp)
        TAILQ_FOREACH(task, &(*root)->root_event, task_node) { 
                 schedCancel(task);                  schedCancel(task);
        }        TAILQ_FOREACH_SAFE(task, &(*root)->root_node, task_node, tmp)
        TAILQ_FOREACH(task, &(*root)->root_eventlo, task_node) { 
                 schedCancel(task);                  schedCancel(task);
        }        TAILQ_FOREACH_SAFE(task, &(*root)->root_proc, task_node, tmp)
        TAILQ_FOREACH(task, &(*root)->root_ready, task_node) { 
                 schedCancel(task);                  schedCancel(task);
        }        TAILQ_FOREACH_SAFE(task, &(*root)->root_user, task_node, tmp)
                 schedCancel(task);
         TAILQ_FOREACH_SAFE(task, &(*root)->root_signal, task_node, tmp)
                 schedCancel(task);
         TAILQ_FOREACH_SAFE(task, &(*root)->root_event, task_node, tmp)
                 schedCancel(task);
         TAILQ_FOREACH_SAFE(task, &(*root)->root_eventlo, task_node, tmp)
                 schedCancel(task);
         TAILQ_FOREACH_SAFE(task, &(*root)->root_ready, task_node, tmp)
                 schedCancel(task);
   
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
         pthread_mutex_lock(&(*root)->root_mtx[taskUNUSE]);          pthread_mutex_lock(&(*root)->root_mtx[taskUNUSE]);
 #endif  #endif
        while ((task = TAILQ_FIRST(&(*root)->root_unuse))) {        TAILQ_FOREACH_SAFE(task, &(*root)->root_unuse, task_node, tmp) {
                 TAILQ_REMOVE(&(*root)->root_unuse, task, task_node);                  TAILQ_REMOVE(&(*root)->root_unuse, task, task_node);
                 free(task);                  free(task);
         }          }
Line 258  schedCall(sched_task_t * __restrict task) Line 272  schedCall(sched_task_t * __restrict task)
         if (!TASK_ISLOCKED(task))          if (!TASK_ISLOCKED(task))
                 TASK_LOCK(task);                  TASK_LOCK(task);
   
         task->task_id++;  
         ptr = task->task_func(task);          ptr = task->task_func(task);
   
         TASK_UNLOCK(task);          TASK_UNLOCK(task);
Line 288  schedFetch(sched_root_task_t * __restrict root) Line 301  schedFetch(sched_root_task_t * __restrict root)
 }  }
   
 /*  /*
    * schedTrigger() - Triggering USER task
    *
    * @task = task
    * return: -1 error or 0 ok
    */
   int
   schedTrigger(sched_task_t * __restrict task)
   {
           struct kevent chg[1];
           struct timespec timeout = { 0, 0 };
   
           if (!task || !TASK_ROOT(task))
                   return -1;
   
   #ifdef __NetBSD__
           EV_SET(chg, TASK_VAL(task), EVFILT_USER, 0, NOTE_TRIGGER, 0, (intptr_t) TASK_VAL(task));
   #else
           EV_SET(chg, TASK_VAL(task), EVFILT_USER, 0, NOTE_TRIGGER, 0, (void*) TASK_VAL(task));
   #endif
           if (kevent(TASK_ROOT(task)->root_kq, chg, 1, NULL, 0, &timeout) == -1) {
                   LOGERR;
                   return -1;
           }
   
           return 0;
   }
   
   /*
  * schedCancel() - Cancel task from scheduler   * schedCancel() - Cancel task from scheduler
  *   *
  * @task = task   * @task = task
Line 315  schedCancel(sched_task_t * __restrict task) Line 356  schedCancel(sched_task_t * __restrict task)
                 case taskTIMER:                  case taskTIMER:
                         queue = &TASK_ROOT(task)->root_timer;                          queue = &TASK_ROOT(task)->root_timer;
                         break;                          break;
                   case taskALARM:
                           queue = &TASK_ROOT(task)->root_alarm;
                           break;
                   case taskNODE:
                           queue = &TASK_ROOT(task)->root_node;
                           break;
                   case taskPROC:
                           queue = &TASK_ROOT(task)->root_proc;
                           break;
                   case taskUSER:
                           queue = &TASK_ROOT(task)->root_user;
                           break;
                   case taskSIGNAL:
                           queue = &TASK_ROOT(task)->root_signal;
                           break;
                 case taskEVENT:                  case taskEVENT:
                         queue = &TASK_ROOT(task)->root_event;                          queue = &TASK_ROOT(task)->root_event;
                         break;                          break;
Line 331  schedCancel(sched_task_t * __restrict task) Line 387  schedCancel(sched_task_t * __restrict task)
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
                 pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);                  pthread_mutex_lock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);
 #endif  #endif
                TAILQ_REMOVE(queue, task, task_node);                TAILQ_REMOVE(queue, TASK_ID(task), task_node);
 #ifdef HAVE_LIBPTHREAD  #ifdef HAVE_LIBPTHREAD
                 pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);                  pthread_mutex_unlock(&TASK_ROOT(task)->root_mtx[TASK_TYPE(task)]);
 #endif  #endif
Line 347  schedCancel(sched_task_t * __restrict task) Line 403  schedCancel(sched_task_t * __restrict task)
  *   *
  * @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_ANY|CRITERIA_CALL|CRITERIA_ARG|CRITERIA_FD|CRITERIA_VAL|CRITERIA_TS|CRITERIA_DATA]
  * @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 356  int Line 413  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;
           /* if type == taskMAX check in all queues */
         if (type == taskMAX) {          if (type == taskMAX) {
                 if (schedCancelby(root, taskREAD, criteria, param, hook))                  if (schedCancelby(root, taskREAD, criteria, param, hook))
                         return -2;                          return -2;
Line 369  schedCancelby(sched_root_task_t * __restrict root, sch Line 427  schedCancelby(sched_root_task_t * __restrict root, sch
                         return -2;                          return -2;
                 if (schedCancelby(root, taskTIMER, criteria, param, hook))                  if (schedCancelby(root, taskTIMER, criteria, param, hook))
                         return -2;                          return -2;
                   if (schedCancelby(root, taskALARM, criteria, param, hook))
                           return -2;
                   if (schedCancelby(root, taskNODE, criteria, param, hook))
                           return -2;
                   if (schedCancelby(root, taskPROC, criteria, param, hook))
                           return -2;
                   if (schedCancelby(root, taskUSER, criteria, param, hook))
                           return -2;
                   if (schedCancelby(root, taskSIGNAL, criteria, param, hook))
                           return -2;
                 if (schedCancelby(root, taskEVENT, criteria, param, hook))                  if (schedCancelby(root, taskEVENT, criteria, param, hook))
                         return -2;                          return -2;
                 if (schedCancelby(root, taskEVENTLO, criteria, param, hook))                  if (schedCancelby(root, taskEVENTLO, criteria, param, hook))
Line 377  schedCancelby(sched_root_task_t * __restrict root, sch Line 445  schedCancelby(sched_root_task_t * __restrict root, sch
                         return -2;                          return -2;
                 return 0;                  return 0;
         }          }
           /* choosen queue */
         switch (type) {          switch (type) {
                 case taskREAD:                  case taskREAD:
                         queue = &root->root_read;                          queue = &root->root_read;
Line 387  schedCancelby(sched_root_task_t * __restrict root, sch Line 456  schedCancelby(sched_root_task_t * __restrict root, sch
                 case taskTIMER:                  case taskTIMER:
                         queue = &root->root_timer;                          queue = &root->root_timer;
                         break;                          break;
                   case taskALARM:
                           queue = &root->root_alarm;
                           break;
                   case taskNODE:
                           queue = &root->root_node;
                           break;
                   case taskPROC:
                           queue = &root->root_proc;
                           break;
                   case taskUSER:
                           queue = &root->root_user;
                           break;
                   case taskSIGNAL:
                           queue = &root->root_signal;
                           break;
                 case taskEVENT:                  case taskEVENT:
                         queue = &root->root_event;                          queue = &root->root_event;
                         break;                          break;
Line 403  schedCancelby(sched_root_task_t * __restrict root, sch Line 487  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_ANY:
                                 flg = 1;
                                 break;                                  break;
                        }                        case CRITERIA_CALL:
                } else if (criteria == CRITERIA_ARG) {                                if (TASK_FUNC(task) == (sched_task_func_t) param)
                        if (task->task_arg == param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_ARG:
                } else if (criteria == CRITERIA_FD) {                                if (TASK_ARG(task) == param)
                        if (TASK_FD(task) == (intptr_t) param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_FD:
                } else if (criteria == CRITERIA_VAL) {                                if (TASK_FD(task) == (intptr_t) param)
                        if (TASK_VAL(task) == (u_long) param) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_VAL:
                } else if (criteria == CRITERIA_TV) {                                if (TASK_VAL(task) == (u_long) param)
                        if (!sched_timespeccmp(&TASK_TS(task), (struct timespec*) param, -)) {                                        flg = 1;
                                flg++; 
                                 break;                                  break;
                        }                        case CRITERIA_TS:
                } else {                                if (!sched_timespeccmp(&TASK_TS(task), (struct timespec*) param, -))
#ifdef HAVE_LIBPTHREAD                                        flg = 1;
                        pthread_mutex_unlock(&root->root_mtx[type]);                                break;
#endif                        case CRITERIA_DATA:
                        sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria);                                if (TASK_DATA(task) == param)
                        return -1;                                        flg = 1;
                                 break;
                         default:
                                 sched_SetErr(EINVAL, "Invalid parameter criteria %d", criteria);
                                 flg = -1;
                 }                  }
#ifdef HAVE_LIBPTHREAD                if (flg < 0)            /* error */
        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; 
 }  }
   
 /*  /*

Removed from v.1.6.2.4  
changed lines
  Added in v.1.9.2.2


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