Fixed output.c

Handle non 0 terminated strings
jroeger23 · Mar 15, 2019 · e18f268 · e18f268
1 parent bb695a7
commit e18f268
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/output.c b/output.c
@@ -5,6 +5,8 @@
 #include <string.h>
 #include <stdarg.h>
 
+#define STRING_MAX 4096
+
 output_t output = { .level = verbose };
 
 #define PREFIX(FD, L) write(FD, prefixes[L], prefixes_len[L]);
@@ -28,15 +30,15 @@ void sprint(char *str, output_level_t l)
         if(output.level < l) return;
 
         PREFIX(STDOUT_FILENO, l);
-        write(STDOUT_FILENO, str, strlen(str));
+        write(STDOUT_FILENO, str, strnlen(str, STRING_MAX));
 }
 
 void eprint(char *str, output_level_t l)
 {
         if(output.level < l) return;
 
         PREFIX(STDERR_FILENO, l);
-        write(STDERR_FILENO, str, strlen(str));
+        write(STDERR_FILENO, str, strnlen(str, STRING_MAX));
 }
 
 void sprintln(char *str, output_level_t l)