X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Futils%2Fchannel.c;h=a45247a71060503b55589655e2cd6711dd1bcd7f;hb=2df0d2d5c79582e1dce5cdedede9ac564a6f126b;hp=afce6f5d9cc9aa883afb9569aef9a384c433eb01;hpb=23c8e0f818e5baf91e75993571b67fcb735628f5;p=sysdb.git

diff --git a/src/utils/channel.c b/src/utils/channel.c
index afce6f5..a45247a 100644
--- a/src/utils/channel.c
+++ b/src/utils/channel.c
@@ -57,6 +57,8 @@ struct sdb_channel {
 	size_t head;
 	size_t tail;
 	_Bool full;
+
+	_Bool shutdown;
 };
 
 /*
@@ -79,7 +81,7 @@ channel_write(sdb_channel_t *chan, const void *data)
 {
 	assert(chan);
 
-	if (chan->full)
+	if (chan->full || chan->shutdown)
 		return -1;
 	else if (! data)
 		return 0;
@@ -169,7 +171,6 @@ int
 sdb_channel_select(sdb_channel_t *chan, int *wantread, void *read_data,
 		int *wantwrite, void *write_data, const struct timespec *timeout)
 {
-	struct timespec abstime;
 	int status = 0;
 
 	if (! chan) {
@@ -182,14 +183,6 @@ sdb_channel_select(sdb_channel_t *chan, int *wantread, void *read_data,
 		return -1;
 	}
 
-	if (timeout) {
-		if (clock_gettime(CLOCK_REALTIME, &abstime))
-			return -1;
-
-		abstime.tv_sec += timeout->tv_sec;
-		abstime.tv_nsec += timeout->tv_nsec;
-	}
-
 	pthread_mutex_lock(&chan->lock);
 	while (! status) {
 		int read_status, write_status;
@@ -208,14 +201,36 @@ sdb_channel_select(sdb_channel_t *chan, int *wantread, void *read_data,
 				break;
 		}
 
-		if (timeout)
+		if (chan->shutdown) {
+			if (read_status)
+				status = EBADF;
+			break;
+		}
+
+		if (timeout) {
+			struct timespec abstime;
+
+			if (clock_gettime(CLOCK_REALTIME, &abstime)) {
+				pthread_mutex_unlock(&chan->lock);
+				return -1;
+			}
+
+			abstime.tv_sec += timeout->tv_sec;
+			abstime.tv_nsec += timeout->tv_nsec;
+
+			if (abstime.tv_nsec > 1000000000) {
+				abstime.tv_nsec -= 1000000000;
+				abstime.tv_sec += 1;
+			}
+
 			status = pthread_cond_timedwait(&chan->cond, &chan->lock,
 					&abstime);
+		}
 		else
 			status = pthread_cond_wait(&chan->cond, &chan->lock);
 	}
-
 	pthread_mutex_unlock(&chan->lock);
+
 	if (status) {
 		errno = status;
 		return -1;
@@ -251,5 +266,14 @@ sdb_channel_read(sdb_channel_t *chan, void *data)
 	return status;
 } /* sdb_channel_read */
 
+int
+sdb_channel_shutdown(sdb_channel_t *chan)
+{
+	if (! chan)
+		return -1;
+	chan->shutdown = 1;
+	return 0;
+} /* sdb_channel_shutdown */
+
 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */