7 #if defined(__LCLINT__)
8 #define _BITS_SIGTHREAD_H /* XXX avoid __sigset_t heartburn. */
10 /*@-incondefs -protoparammatch@*/
12 /*@constant int SA_SIGINFO@*/
13 extern int sighold(int sig)
14 /*@globals errno, systemState @*/;
15 extern int sigignore(int sig)
16 /*@globals errno, systemState @*/;
17 extern int sigpause(int sig)
18 /*@globals errno, systemState @*/;
19 extern int sigrelse(int sig)
20 /*@globals errno, systemState @*/;
21 extern void (*sigset(int sig, void (*disp)(int)))(int)
22 /*@globals errno, systemState @*/;
25 extern void insque(struct qelem * __elem, struct qelem * __prev)
26 /*@modifies __elem, __prev @*/;
27 extern void remque(struct qelem * __elem)
28 /*@modifies __elem @*/;
30 extern pthread_t pthread_self(void)
32 extern int pthread_equal(pthread_t t1, pthread_t t2)
35 extern int pthread_create(/*@out@*/ pthread_t *restrict thread,
36 const pthread_attr_t *restrict attr,
37 void *(*start_routine)(void*), void *restrict arg)
38 /*@modifies *thread @*/;
39 extern int pthread_join(pthread_t thread, /*@out@*/ void **value_ptr)
40 /*@modifies *value_ptr @*/;
42 extern int pthread_setcancelstate(int state, /*@out@*/ int *oldstate)
43 /*@globals internalState @*/
44 /*@modifies *oldstate, internalState @*/;
45 extern int pthread_setcanceltype(int type, /*@out@*/ int *oldtype)
46 /*@globals internalState @*/
47 /*@modifies *oldtype, internalState @*/;
48 extern void pthread_testcancel(void)
49 /*@globals internalState @*/
50 /*@modifies internalState @*/;
51 extern void pthread_cleanup_pop(int execute)
52 /*@globals internalState @*/
53 /*@modifies internalState @*/;
54 extern void pthread_cleanup_push(void (*routine)(void*), void *arg)
55 /*@globals internalState @*/
56 /*@modifies internalState @*/;
57 extern void _pthread_cleanup_pop(/*@out@*/ struct _pthread_cleanup_buffer *__buffer, int execute)
58 /*@globals internalState @*/
59 /*@modifies internalState @*/;
60 extern void _pthread_cleanup_push(/*@out@*/ struct _pthread_cleanup_buffer *__buffer, void (*routine)(void*), /*@out@*/ void *arg)
61 /*@globals internalState @*/
62 /*@modifies internalState @*/;
64 extern int pthread_mutexattr_destroy(pthread_mutexattr_t *attr)
65 /*@globals errno, internalState @*/
66 /*@modifies *attr, errno, internalState @*/;
67 extern int pthread_mutexattr_init(/*@out@*/ pthread_mutexattr_t *attr)
68 /*@globals errno, internalState @*/
69 /*@modifies *attr, errno, internalState @*/;
71 int pthread_mutexattr_gettype(const pthread_mutexattr_t *restrict attr,
72 /*@out@*/ int *restrict type)
73 /*@modifies *type @*/;
74 int pthread_mutexattr_settype(pthread_mutexattr_t *attr, int type)
75 /*@globals errno, internalState @*/
76 /*@modifies *attr, errno, internalState @*/;
78 extern int pthread_mutex_destroy(pthread_mutex_t *mutex)
79 /*@modifies *mutex @*/;
80 extern int pthread_mutex_init(/*@out@*/ pthread_mutex_t *restrict mutex,
81 /*@null@*/ const pthread_mutexattr_t *restrict attr)
82 /*@globals errno, internalState @*/
83 /*@modifies *mutex, errno, internalState @*/;
85 extern int pthread_mutex_lock(pthread_mutex_t *mutex)
87 /*@modifies *mutex, errno @*/;
88 extern int pthread_mutex_trylock(pthread_mutex_t *mutex)
90 /*@modifies *mutex, errno @*/;
91 extern int pthread_mutex_unlock(pthread_mutex_t *mutex)
93 /*@modifies *mutex, errno @*/;
95 extern int pthread_cond_destroy(pthread_cond_t *cond)
96 /*@modifies *cond @*/;
97 extern int pthread_cond_init(/*@out@*/ pthread_cond_t *restrict cond,
98 const pthread_condattr_t *restrict attr)
99 /*@globals errno, internalState @*/
100 /*@modifies *cond, errno, internalState @*/;
102 extern int pthread_cond_timedwait(pthread_cond_t *restrict cond,
103 pthread_mutex_t *restrict mutex,
104 const struct timespec *restrict abstime)
105 /*@modifies *cond, *mutex @*/;
106 extern int pthread_cond_wait(pthread_cond_t *restrict cond,
107 pthread_mutex_t *restrict mutex)
108 /*@modifies *cond, *mutex @*/;
109 extern int pthread_cond_broadcast(pthread_cond_t *cond)
110 /*@globals errno, internalState @*/
111 /*@modifies *cond, errno, internalState @*/;
112 extern int pthread_cond_signal(pthread_cond_t *cond)
113 /*@globals errno, internalState @*/
114 /*@modifies *cond, errno, internalState @*/;
117 /*@=incondefs =protoparammatch@*/
121 #include <sys/signal.h>
122 #include <sys/wait.h>
125 #if defined(HAVE_PTHREAD_H)
129 /* XXX suggested in bugzilla #159024 */
130 #if PTHREAD_MUTEX_DEFAULT != PTHREAD_MUTEX_NORMAL
131 #error RPM expects PTHREAD_MUTEX_DEFAULT == PTHREAD_MUTEX_NORMAL
134 #ifndef PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
136 static pthread_mutex_t rpmsigTbl_lock = PTHREAD_MUTEX_INITIALIZER;
140 static pthread_mutex_t rpmsigTbl_lock = PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
144 #define DO_LOCK() pthread_mutex_lock(&rpmsigTbl_lock);
145 #define DO_UNLOCK() pthread_mutex_unlock(&rpmsigTbl_lock);
146 #define INIT_LOCK() \
147 { pthread_mutexattr_t attr; \
148 (void) pthread_mutexattr_init(&attr); \
149 (void) pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE); \
150 (void) pthread_mutex_init (&rpmsigTbl_lock, &attr); \
151 (void) pthread_mutexattr_destroy(&attr); \
152 rpmsigTbl_sigchld->active = 0; \
154 #define ADD_REF(__tbl) (__tbl)->active++
155 #define SUB_REF(__tbl) --(__tbl)->active
156 #define CLEANUP_HANDLER(__handler, __arg, __oldtypeptr) \
157 (void) pthread_setcanceltype (PTHREAD_CANCEL_ASYNCHRONOUS, (__oldtypeptr));\
158 pthread_cleanup_push((__handler), (__arg));
159 #define CLEANUP_RESET(__execute, __oldtype) \
160 pthread_cleanup_pop(__execute); \
161 (void) pthread_setcanceltype ((__oldtype), &(__oldtype));
163 #define SAME_THREAD(_a, _b) pthread_equal(((pthread_t)_a), ((pthread_t)_b))
165 #define ME() ((void *)pthread_self())
172 #define ADD_REF(__tbl) /*@-noeffect@*/ (0) /*@=noeffect@*/
173 #define SUB_REF(__tbl) /*@-noeffect@*/ (0) /*@=noeffect@*/
174 #define CLEANUP_HANDLER(__handler, __arg, __oldtypeptr)
175 #define CLEANUP_RESET(__execute, __oldtype)
177 #define SAME_THREAD(_a, _b) (42)
179 #define ME() (((void *)getpid()))
181 #endif /* HAVE_PTHREAD_H */
187 #define _RPMSQ_DEBUG 0
189 int _rpmsq_debug = _RPMSQ_DEBUG;
192 static struct rpmsqElem rpmsqRock;
196 rpmsq rpmsqQueue = &rpmsqRock;
199 int rpmsqInsert(void * elem, void * prev)
201 rpmsq sq = (rpmsq) elem;
207 fprintf(stderr, " Insert(%p): %p\n", ME(), sq);
209 ret = sighold(SIGCHLD);
216 sq->pipes[0] = sq->pipes[1] = -1;
220 ret = pthread_mutex_init(&sq->mutex, NULL);
221 insque(elem, (prev != NULL ? prev : rpmsqQueue));
222 ret = sigrelse(SIGCHLD);
228 int rpmsqRemove(void * elem)
230 rpmsq sq = (rpmsq) elem;
237 fprintf(stderr, " Remove(%p): %p\n", ME(), sq);
239 ret = sighold (SIGCHLD);
243 /* Unlock the mutex and then destroy it */
244 if((ret = pthread_mutex_unlock(&sq->mutex)) == 0)
245 ret = pthread_mutex_destroy(&sq->mutex);
249 if (sq->pipes[1]) ret = close(sq->pipes[1]);
250 if (sq->pipes[0]) ret = close(sq->pipes[0]);
251 sq->pipes[0] = sq->pipes[1] = -1;
253 #ifdef NOTYET /* rpmpsmWait debugging message needs */
259 ret = sigrelse(SIGCHLD);
266 sigset_t rpmsqCaught;
270 static struct rpmsig_s {
272 void (*handler) (int signum, void * info, void * context);
274 struct sigaction oact;
276 { SIGINT, rpmsqAction },
277 #define rpmsigTbl_sigint (&rpmsigTbl[0])
278 { SIGQUIT, rpmsqAction },
279 #define rpmsigTbl_sigquit (&rpmsigTbl[1])
280 { SIGCHLD, rpmsqAction },
281 #define rpmsigTbl_sigchld (&rpmsigTbl[2])
282 { SIGHUP, rpmsqAction },
283 #define rpmsigTbl_sighup (&rpmsigTbl[3])
284 { SIGTERM, rpmsqAction },
285 #define rpmsigTbl_sigterm (&rpmsigTbl[4])
286 { SIGPIPE, rpmsqAction },
287 #define rpmsigTbl_sigpipe (&rpmsigTbl[5])
292 void rpmsqAction(int signum,
293 /*@unused@*/ void * info, /*@unused@*/ void * context)
298 for (tbl = rpmsigTbl; tbl->signum >= 0; tbl++) {
299 if (tbl->signum != signum)
302 (void) sigaddset(&rpmsqCaught, signum);
309 pid_t reaped = waitpid(0, &status, WNOHANG);
311 /* XXX errno set to ECHILD/EINVAL/EINTR. */
313 /*@innerbreak@*/ break;
315 /* XXX insque(3)/remque(3) are dequeue, not ring. */
316 for (sq = rpmsqQueue->q_forw;
317 sq != NULL && sq != rpmsqQueue;
322 if (sq->child != reaped)
323 /*@innercontinue@*/ continue;
327 /* Unlock the mutex. The waiter will then be able to
330 * XXX: jbj, wtd, if this fails?
332 ret = pthread_mutex_unlock(&sq->mutex);
334 /*@innerbreak@*/ break;
337 /*@switchbreak@*/ break;
339 /*@switchbreak@*/ break;
346 int rpmsqEnable(int signum, /*@null@*/ rpmsqAction_t handler)
347 /*@globals rpmsigTbl @*/
348 /*@modifies rpmsigTbl @*/
350 int tblsignum = (signum >= 0 ? signum : -signum);
356 if (rpmsqQueue->id == NULL)
357 rpmsqQueue->id = ME();
358 for (tbl = rpmsigTbl; tbl->signum >= 0; tbl++) {
359 if (tblsignum != tbl->signum)
362 if (signum >= 0) { /* Enable. */
363 if (ADD_REF(tbl) <= 0) {
364 (void) sigdelset(&rpmsqCaught, tbl->signum);
366 /* XXX Don't set a signal handler if already SIG_IGN */
367 (void) sigaction(tbl->signum, NULL, &tbl->oact);
368 if (tbl->oact.sa_handler == SIG_IGN)
371 (void) sigemptyset (&sa.sa_mask);
372 sa.sa_flags = SA_SIGINFO;
373 #if defined(__LCLINT__) /* XXX glibc has union to track handler prototype. */
374 sa.sa_handler = (handler != NULL ? handler : tbl->handler);
376 sa.sa_sigaction = (handler != NULL ? handler : tbl->handler);
378 if (sigaction(tbl->signum, &sa, &tbl->oact) < 0) {
382 tbl->active = 1; /* XXX just in case */
384 tbl->handler = handler;
386 } else { /* Disable. */
387 if (SUB_REF(tbl) <= 0) {
388 if (sigaction(tbl->signum, &tbl->oact, NULL) < 0)
390 tbl->active = 0; /* XXX just in case */
391 tbl->handler = (handler != NULL ? handler : rpmsqAction);
401 pid_t rpmsqFork(rpmsq sq)
405 int nothreads = 0; /* XXX: Shouldn't this be a global? */
408 xx = rpmsqInsert(sq, NULL);
411 fprintf(stderr, " Enable(%p): %p\n", ME(), sq);
413 xx = rpmsqEnable(SIGCHLD, NULL);
416 xx = pipe(sq->pipes);
418 xx = sighold(SIGCHLD);
421 * Initialize the cond var mutex. We have to aquire the lock we
422 * use for the condition before we fork. Otherwise it is possible for
423 * the child to exit, we get sigchild and the sig handler to send
424 * the condition signal before we are waiting on the condition.
427 if(pthread_mutex_lock(&sq->mutex)) {
428 /* Yack we did not get the lock, lets just give up */
430 xx = close(sq->pipes[0]);
431 xx = close(sq->pipes[1]);
432 sq->pipes[0] = sq->pipes[1] = -1;
439 if (pid < (pid_t) 0) { /* fork failed. */
440 sq->child = (pid_t)-1;
442 xx = close(sq->pipes[0]);
443 xx = close(sq->pipes[1]);
444 sq->pipes[0] = sq->pipes[1] = -1;
447 } else if (pid == (pid_t) 0) { /* Child. */
450 /* Block to permit parent time to wait. */
452 xx = close(sq->pipes[1]);
453 xx = read(sq->pipes[0], &yy, sizeof(yy));
454 xx = close(sq->pipes[0]);
455 sq->pipes[0] = sq->pipes[1] = -1;
460 fprintf(stderr, " Child(%p): %p child %d\n", ME(), sq, getpid());
463 } else { /* Parent. */
469 fprintf(stderr, " Parent(%p): %p child %d\n", ME(), sq, sq->child);
475 xx = sigrelse(SIGCHLD);
480 * Wait for child process to be reaped, and unregister SIGCHLD handler.
481 * @todo Rewrite to use waitpid on helper thread.
482 * @param sq scriptlet queue element
483 * @return 0 on success
485 static int rpmsqWaitUnregister(rpmsq sq)
486 /*@globals fileSystem, internalState @*/
487 /*@modifies sq, fileSystem, internalState @*/
493 /* Protect sq->reaped from handler changes. */
494 ret = sighold(SIGCHLD);
496 /* Start the child, linux often runs child before parent. */
498 if (sq->pipes[0] >= 0)
499 xx = close(sq->pipes[0]);
500 if (sq->pipes[1] >= 0)
501 xx = close(sq->pipes[1]);
502 sq->pipes[0] = sq->pipes[1] = -1;
505 /* Put a stopwatch on the time spent waiting to measure performance gain. */
506 (void) rpmswEnter(&sq->op, -1);
508 /* Wait for handler to receive SIGCHLD. */
510 while (ret == 0 && sq->reaped != sq->child) {
512 /* Note that sigpause re-enables SIGCHLD. */
513 ret = sigpause(SIGCHLD);
515 xx = sigrelse(SIGCHLD);
518 * We start before the fork with this mutex locked;
519 * The only one that unlocks this the signal handler.
520 * So if we get the lock the child has been reaped.
522 ret = pthread_mutex_lock(&sq->mutex);
523 xx = sighold(SIGCHLD);
528 /* Accumulate stopwatch time spent waiting, potential performance gain. */
529 sq->ms_scriptlets += rpmswExit(&sq->op, -1)/1000;
531 xx = sigrelse(SIGCHLD);
535 fprintf(stderr, " Wake(%p): %p child %d reaper %d ret %d\n", ME(), sq, sq->child, sq->reaper, ret);
538 /* Remove processed SIGCHLD item from queue. */
539 xx = rpmsqRemove(sq);
541 /* Disable SIGCHLD handler on refcount == 0. */
542 xx = rpmsqEnable(-SIGCHLD, NULL);
545 fprintf(stderr, " Disable(%p): %p\n", ME(), sq);
551 pid_t rpmsqWait(rpmsq sq)
556 fprintf(stderr, " Wait(%p): %p child %d reaper %d\n", ME(), sq, sq->child, sq->reaper);
560 (void) rpmsqWaitUnregister(sq);
565 reaped = waitpid(sq->child, &status, 0);
566 } while (reaped >= 0 && reaped != sq->child);
571 fprintf(stderr, " Waitpid(%p): %p child %d reaped %d\n", ME(), sq, sq->child, sq->reaped);
577 fprintf(stderr, " Fini(%p): %p child %d status 0x%x\n", ME(), sq, sq->child, sq->status);
583 void * rpmsqThread(void * (*start) (void * arg), void * arg)
588 ret = pthread_create(&pth, NULL, start, arg);
589 return (ret == 0 ? (void *)pth : NULL);
592 int rpmsqJoin(void * thread)
594 pthread_t pth = (pthread_t) thread;
597 return pthread_join(pth, NULL);
600 int rpmsqThreadEqual(void * thread)
602 pthread_t t1 = (pthread_t) thread;
603 pthread_t t2 = pthread_self();
604 return pthread_equal(t1, t2);
608 * SIGCHLD cancellation handler.
611 sigchld_cancel (void *arg)
612 /*@globals rpmsigTbl, fileSystem, internalState @*/
613 /*@modifies rpmsigTbl, fileSystem, internalState @*/
615 pid_t child = *(pid_t *) arg;
618 (void) kill(child, SIGKILL);
621 result = waitpid(child, NULL, 0);
622 } while (result == (pid_t)-1 && errno == EINTR);
625 if (SUB_REF (rpmsigTbl_sigchld) == 0) {
626 (void) rpmsqEnable(-SIGQUIT, NULL);
627 (void) rpmsqEnable(-SIGINT, NULL);
633 * Execute a command, returning its status.
636 rpmsqExecve (const char ** argv)
637 /*@globals rpmsigTbl @*/
638 /*@modifies rpmsigTbl @*/
644 sigset_t newMask, oldMask;
645 rpmsq sq = memset(alloca(sizeof(*sq)), 0, sizeof(*sq));
647 #ifndef PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
652 if (ADD_REF (rpmsigTbl_sigchld) == 0) {
653 if (rpmsqEnable(SIGINT, NULL) < 0) {
654 SUB_REF (rpmsigTbl_sigchld);
657 if (rpmsqEnable(SIGQUIT, NULL) < 0) {
658 SUB_REF (rpmsigTbl_sigchld);
659 goto out_restore_sigint;
664 (void) sigemptyset (&newMask);
665 (void) sigaddset (&newMask, SIGCHLD);
666 if (sigprocmask (SIG_BLOCK, &newMask, &oldMask) < 0) {
668 if (SUB_REF (rpmsigTbl_sigchld) == 0)
669 goto out_restore_sigquit_and_sigint;
673 CLEANUP_HANDLER(sigchld_cancel, &pid, &oldtype);
676 if (pid < (pid_t) 0) { /* fork failed. */
678 } else if (pid == (pid_t) 0) { /* Child. */
680 /* Restore the signals. */
681 (void) sigaction (SIGINT, &rpmsigTbl_sigint->oact, NULL);
682 (void) sigaction (SIGQUIT, &rpmsigTbl_sigquit->oact, NULL);
683 (void) sigprocmask (SIG_SETMASK, &oldMask, NULL);
685 /* Reset rpmsigTbl lock and refcnt. */
688 (void) execve (argv[0], (char *const *) argv, environ);
690 } else { /* Parent. */
692 result = waitpid(pid, &status, 0);
693 } while (result == (pid_t)-1 && errno == EINTR);
698 CLEANUP_RESET(0, oldtype);
701 if ((SUB_REF (rpmsigTbl_sigchld) == 0 &&
702 (rpmsqEnable(-SIGINT, NULL) < 0 || rpmsqEnable (-SIGQUIT, NULL) < 0))
703 || sigprocmask (SIG_SETMASK, &oldMask, NULL) != 0)
709 out_restore_sigquit_and_sigint:
710 (void) rpmsqEnable(-SIGQUIT, NULL);
712 (void) rpmsqEnable(-SIGINT, NULL);