From: Christian Heller <c.heller@plomlompom.de>
Date: Fri, 14 Nov 2014 03:10:25 +0000 (+0100)
Subject: To read_file_into_queue(), add queue size test.
X-Git-Tag: tce~600
X-Git-Url: https://plomlompom.com/repos/%7B%7B%20web_path%20%7D%7D/decks/%7B%7Bprefix%7D%7D/edit?a=commitdiff_plain;h=9b95b591441e81cde337024591473efa943c2245;p=plomrogue

To read_file_into_queue(), add queue size test.
---

diff --git a/src/common/readwrite.c b/src/common/readwrite.c
index 43c83e9..77d83d9 100644
--- a/src/common/readwrite.c
+++ b/src/common/readwrite.c
@@ -189,6 +189,7 @@ extern uint8_t read_file_into_queue(FILE * file, char ** queue,
     int test = try_fgetc(file, __func__);
     if (EOF != test)
     {
+        char * err_size = "Queue growing too large.";
         do
         {
             char c = (char) test;
@@ -200,6 +201,7 @@ extern uint8_t read_file_into_queue(FILE * file, char ** queue,
             memcpy(new_queue, *queue, *queue_size);
             char * new_pos = new_queue + *queue_size;
             * new_pos = c;
+            exit_err(*queue_size == UINT32_MAX, err_size);
             *queue_size = *queue_size + 1;
             free(*queue);
             *queue = new_queue;
@@ -210,6 +212,7 @@ extern uint8_t read_file_into_queue(FILE * file, char ** queue,
             char * new_queue = try_malloc(*queue_size + 1, __func__);
             memcpy(new_queue, *queue, *queue_size);
             new_queue[*queue_size] = '\0';
+            exit_err(*queue_size == UINT32_MAX, err_size);
             *queue_size = *queue_size + 1;
             free(*queue);
             *queue = new_queue;