usr: nerve: Ensure nerve packet does not overflow

Signed-off-by: Ian Moffett <ian@osmora.org>

1 files changed, 4 insertions, 0 deletions

diff --git a/usr.bin/nerve/nerve.c b/usr.bin/nerve/nerve.c
index 21deb09..a7d39d8 100644
--- a/usr.bin/nerve/nerve.c
+++ b/usr.bin/nerve/nerve.c
@@ -139,6 +139,10 @@ get_nerve_payload(int argc, char *argv[], struct nerve_payload *res)
 
     /* Start grabbing bytes */
     for (int i = 3; i < argc; ++i) {
+        if (res->len >= NERVE_PACKET_LEN) {
+            printf("[*] truncated packet\n");
+            break;
+        }
         payload_str = argv[i];
         datum = atoi(payload_str);
         res->packet[res->len++] = datum;