broadcaster: check payload size for stdin, fix off-by-one

1 files changed, 8 insertions, 3 deletions

diff --git a/src/broadcaster.c b/src/broadcaster.c
index bd80a5c..66d6b58 100644
--- a/src/broadcaster.c
+++ b/src/broadcaster.c
@@ -26,10 +26,14 @@ int main(int argc, char **argv)
     if (strcmp(message, "-") == 0) {
       /* read payload from stdin */
       int i;
-      while ((MAXBUFLEN-total_len > 0)
-	     && ((i = read(0, &(buf[total_len]), MAXBUFLEN-total_len)) > 0))
+      while ((MAXBUFLEN - total_len - 1 > 0)
+	     && ((i = read(0, &(buf[total_len]), MAXBUFLEN - total_len - 1)) > 0))
 	total_len += i;
       ENP(i, "read(stdin)");
+      if (read(0, &i, 1) != 0) {
+	fprintf(stderr, "payload to long: max %d bytes\n", MAXBUFLEN - 1);
+	exit(1);
+      }
     }else{
       /* use cmd line for payload */
       total_len += strlen(message);
@@ -37,7 +41,8 @@ int main(int argc, char **argv)
 	strcpy(buf + 1 + task_len, message);
     }
     if (total_len > MAXBUFLEN) {
-      fprintf(stderr,"payload to long: max %d bytes, was %d\n", MAXBUFLEN, total_len);
+      fprintf(stderr,"payload to long: max %d bytes, was %d\n",
+	      MAXBUFLEN - 1, total_len);
       exit(1);
     }