--- libaitio/src/exec.c 2013/12/12 21:59:01 1.1.2.19 +++ libaitio/src/exec.c 2013/12/15 22:57:20 1.2 @@ -173,6 +173,63 @@ io_progCloseAt(prog_t * __restrict prg, u_int idx) } /* + * io_progCloseOf() - Close program at pool with certain handle + * + * @prg = program pool + * @h = handle of program + * return: 0 error, >0 closed programs + */ +int +#ifdef POPEN_STREAM +io_progCloseOf(prog_t * __restrict prg, FILE *h) +#else +io_progCloseOf(prog_t * __restrict prg, int h) +#endif +{ + register int i; + int ret = 0; + struct tagPIOPID *p; +#ifdef POPEN_STREAM + FILE *f; +#else + int f; +#endif + + if (!prg) + return 0; + + pthread_mutex_lock(&prg->prog_mtx); + for (i = 0; i < array_Size(prg->prog_fds); i++) + if (array_Get(prg->prog_fds, i)) { +#ifdef POPEN_STREAM + f = array(prg->prog_fds, i, FILE*); + if (f == h && (p = pio_pgetpid(array(prg->prog_fds, i, FILE*)))) { +#else + f = (int) array(prg->prog_fds, i, intptr_t); + if (f == h && (p = pio_pgetpid((int) array(prg->prog_fds, i, intptr_t)))) { +#endif + kill(p->pid, SIGTERM); + usleep(1000); + if (waitpid(p->pid, &p->stat, WNOHANG) > 0) + kill(p->pid, SIGKILL); +#ifdef POPEN_STREAM + e_pclose(array(prg->prog_fds, i, FILE*)); +#else + e_pclose((int) array(prg->prog_fds, i, intptr_t)); +#endif + array_Del(prg->prog_fds, i, 0); + clrbit(prg->prog_used, i); + prg->prog_cnum--; + ret++; + break; + } + } + pthread_mutex_unlock(&prg->prog_mtx); + + return ret; +} + +/* * io_progOpen2() - Start program from pool on first unused slot * * @prg = program pool @@ -254,7 +311,6 @@ io_progOpen(prog_t * __restrict prg, u_int execNum) pthread_mutex_lock(&prg->prog_mtx); for (i = 0; (execNum ? ret < execNum : 42) && i < array_Size(prg->prog_fds); i++) if (!array_Get(prg->prog_fds, i)) { - printf("%d) %s!!! %d ret=%d\n", i, __func__, execNum, ret); f = e_popen(prg->prog_name, "r+", &pid); #ifdef POPEN_STREAM if (!f) { @@ -380,11 +436,11 @@ io_progCheck(prog_t * __restrict prg, int re) for (i = 0; i < array_Size(prg->prog_fds); i++) if (array_Get(prg->prog_fds, i) && #ifdef POPEN_STREAM - (p = pio_pgetpid(array(prg->prog_fds, i, FILE*)))) + (p = pio_pgetpid(array(prg->prog_fds, i, FILE*)))) { #else - (p = pio_pgetpid((int) array(prg->prog_fds, i, intptr_t)))) + (p = pio_pgetpid((int) array(prg->prog_fds, i, intptr_t)))) { #endif - if (waitpid(p->pid, &p->stat, WNOHANG) > 0) { + if (waitpid(p->pid, &p->stat, WNOHANG)) { clrbit(prg->prog_used, i); #ifdef POPEN_STREAM e_pclose(array(prg->prog_fds, i, FILE*)); @@ -395,6 +451,7 @@ io_progCheck(prog_t * __restrict prg, int re) prg->prog_cnum--; ret++; } + } pthread_mutex_unlock(&prg->prog_mtx); /* resurrect to init number */ @@ -445,18 +502,18 @@ io_progAttach(prog_t * __restrict prg, int newOne) } pthread_mutex_unlock(&prg->prog_mtx); - /* not found free program */ - if (!f && (i = io_progOpen2(prg)) > 0) { + /* execute new one program */ + if (newOne) { + if (f) + io_progOpen(prg, 1); + else if ((i = io_progOpen2(prg)) > -1) + /* not found free program */ #ifdef POPEN_STREAM - f = array(prg->prog_fds, i, FILE*); + f = array(prg->prog_fds, i, FILE*); #else - f = array(prg->prog_fds, i, intptr_t); + f = array(prg->prog_fds, i, intptr_t); #endif } - - /* execute new one program */ - if (newOne && f) - io_progOpen(prg, 1); return f; }