ncb: Check return value of write()

[karo-tx-uboot.git] / tools / ncb.c

diff --git a/tools/ncb.c b/tools/ncb.c
index 30acbead5f3bfb7ec0e27719a388608b0a92e33c..ec8d8a7435208e432b81b8daf29be868ec408921 100644 (file)
--- a/tools/ncb.c
+++ b/tools/ncb.c
@@ -1,3 +1,4 @@
+#include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <sys/socket.h>
@@ -29,7 +30,8 @@ int main (int argc, char *argv[])
                len = recvfrom (s, buf, sizeof buf, 0, (struct sockaddr *) &addr, &addr_len);
                if (len < 0)
                        break;
-               write (1, buf, len);
+               if (write (1, buf, len) != len)
+                       fprintf(stderr, "WARNING: serial characters dropped\n");
        }
 
        return 0;