Merge remote-tracking branch 'remotes/kraxel/tags/pull-ipxe-20150903-1' into staging

[qemu.git] / tests / tcg / testthread.c

diff --git a/tests/tcg/testthread.c b/tests/tcg/testthread.c
index 27e4825bc6d60fb1d9ddc620b4f9a58ca5d35216..810ba5de67b3b1c762112517aa3697ee69b56ff9 100644 (file)
--- a/tests/tcg/testthread.c
+++ b/tests/tcg/testthread.c
@@ -1,13 +1,19 @@
+#include <assert.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
-#include <signal.h>
 #include <unistd.h>
 #include <inttypes.h>
 #include <pthread.h>
 #include <sys/wait.h>
 #include <sched.h>
 
+void checked_write(int fd, const void *buf, size_t count)
+{
+    ssize_t rc = write(fd, buf, count);
+    assert(rc == count);
+}
+
 void *thread1_func(void *arg)
 {
     int i;
@@ -15,7 +21,7 @@ void *thread1_func(void *arg)
 
     for(i=0;i<10;i++) {
         snprintf(buf, sizeof(buf), "thread1: %d %s\n", i, (char *)arg);
-        write(1, buf, strlen(buf));
+        checked_write(1, buf, strlen(buf));
         usleep(100 * 1000);
     }
     return NULL;
@@ -27,7 +33,7 @@ void *thread2_func(void *arg)
     char buf[512];
     for(i=0;i<20;i++) {
         snprintf(buf, sizeof(buf), "thread2: %d %s\n", i, (char *)arg);
-        write(1, buf, strlen(buf));
+        checked_write(1, buf, strlen(buf));
         usleep(150 * 1000);
     }
     return NULL;