Fix the signal behaviour for internal states which set the thread

state to running despite the SA_RESTART flag which is really just for
syscalls.

1 files changed, 51 insertions, 6 deletions

diff --git a/lib/libkse/thread/thr_sig.c b/lib/libkse/thread/thr_sig.c
index 3fa3c56..58d8e9d 100644
--- a/lib/libkse/thread/thr_sig.c
+++ b/lib/libkse/thread/thr_sig.c
@@ -242,14 +242,59 @@ _thread_signal(pthread_t pthread, int sig)
 	 */
 	sigaddset(&pthread->sigpend,sig);
 
-	/* Check if the thread is waiting on a child process: */
-	if (sig == SIGCHLD && pthread->state == PS_WAIT_WAIT) {
-		/* Change the state of the thread to run: */
-		PTHREAD_NEW_STATE(pthread,PS_RUNNING);
-
 	/* Check if system calls are not restarted: */
-	} else if ((_thread_sigact[sig - 1].sa_flags & SA_RESTART) == 0) {
+	if ((_thread_sigact[sig - 1].sa_flags & SA_RESTART) == 0) {
+		/*
+		 * Process according to thread state:
+		 */
+		switch (pthread->state) {
+		/*
+		 * States which do not change when a signal is trapped:
+		 */
+		case PS_COND_WAIT:
+		case PS_DEAD:
+		case PS_FDLR_WAIT:
+		case PS_FDLW_WAIT:
+		case PS_FILE_WAIT:
+		case PS_JOIN:
+		case PS_MUTEX_WAIT:
+		case PS_RUNNING:
+		case PS_STATE_MAX:
+		case PS_SIGTHREAD:
+		case PS_SUSPENDED:
+			/* Nothing to do here. */
+			break;
+
+		/*
+		 * System calls that are restarted when interrupted by
+		 * a signal:
+		 */
+		case PS_FDR_WAIT:
+		case PS_FDW_WAIT:
+		case PS_SELECT_WAIT:
+			break;
+
+		/*
+		 * States that are interrupted by the occurrence of a signal
+		 * other than the scheduling alarm: 
+		 */
+		case PS_SLEEP_WAIT:
+		case PS_SIGWAIT:
+		case PS_WAIT_WAIT:
+			/* Flag the operation as interrupted: */
+			pthread->interrupted = 1;
+
+			/* Change the state of the thread to run: */
+			PTHREAD_NEW_STATE(pthread,PS_RUNNING);
+
+			/* Return the signal number: */
+			pthread->signo = sig;
+			break;
+		}
+	} else {
 		/*
+		 * System calls are flagged for restart.
+		 *
 		 * Process according to thread state:
 		 */
 		switch (pthread->state) {