Add bounds check for read command

hydrabus · Feb 23, 2024 · 34e3d63 · 34e3d63
1 parent 09ac850
commit 34e3d63
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/hydrabus/hydrabus_mode.c b/src/hydrabus/hydrabus_mode.c
@@ -445,6 +445,11 @@ static int hydrabus_mode_read(t_hydra_console *con, t_tokenline_parsed *p,
 		count = 1;
 	}
 
+	if(count > UINT8_MAX) {
+		cprintf(con, "ERROR: read is limited to %d\r\n", UINT8_MAX);
+		return t - token_pos;
+	}
+
 	mode_status = !HYDRABUS_MODE_STATUS_OK;
 	if(con->mode->exec->read != NULL) {
 		mode_status = con->mode->exec->read(con, p_proto->buffer_rx, count);