Improved handling if realloc fails.

ape_buffer.c

@@ -69,16 +69,16 @@ static void zbuffer_prepapre_buf(buffer *b, size_t input_size)
         bufsize <<= 1;
     }
 
-    printf("zbuf prepared for size %ld\n", bufsize);
-
     b->zbuf->buf_size = bufsize;
 
     if (b->zbuf->buf) {
         b->zbuf->buf = realloc(b->zbuf->buf, bufsize);
-        printf("Realloc input buffer for size %ld\n", bufsize);
+        if (b->zbuf->buf == NULL) {
+            printf("Could not reallocate memory for zbuffer\n");
+            exit(1);
+        }
     } else {
         b->zbuf->buf = malloc(bufsize);
-        printf("Alloc input buffer for size %ld\n", bufsize);
     }
 
     b->zbuf->zstream.avail_out = bufsize - b->zbuf->zstream.avail_in;
@@ -229,6 +229,10 @@ void buffer_prepare(buffer *b, size_t size)
         }
         b->size += size;
         b->data = realloc(b->data, sizeof(char) * b->size);
+        if (b->data == NULL) {
+            printf("Could not reallocate memory for buffer\n");
+            exit(1);
+        }
     }
 #if APE_USE_ZLIB
     if (b->zbuf) {
@@ -249,6 +253,10 @@ static void buffer_prepare_for(buffer *b, size_t size, size_t forsize)
         }
         b->size += size;
         b->data = realloc(b->data, sizeof(char) * b->size);
+        if (b->data == NULL) {
+            printf("Could not reallocate memory for buffer\n");
+            exit(1);
+        }
     }
 #if APE_USE_ZLIB
     if (b->zbuf) {
@@ -399,6 +407,10 @@ buffer *buffer_to_buffer_utf8(buffer *b)
     if (newb->size > newb->used+1) {
         newb->size = newb->used+1;
         newb->data = realloc(newb->data, newb->size);
+        if (newb->data == NULL) {
+            printf("Could not reallocate memory for utf8 buffer\n");
+            exit(1);
+        }
     }
 
     return newb;
@@ -453,6 +465,10 @@ buffer *buffer_utf8_to_buffer(buffer *b)
     if (newb->size > newb->used+1) {
         newb->size = newb->used+1;
         newb->data = realloc(newb->data, newb->size);
+        if (newb->data == NULL) {
+            printf("Could not reallocate memory for utf8 buffer\n");
+            exit(1);
+        }
     }
 
     return newb;

ape_event_epoll.c

@@ -101,6 +101,10 @@ static void event_epoll_setsize(struct _fdevent *ev, int size)
 {
     ev->events = realloc(ev->events,
             sizeof(struct epoll_event) * (size));
+    if (ev->events == NULL) {
+        printf("Could not reallocate memory for epoll\n");
+        exit(1);
+    }
 }
 
 static int event_epoll_revent(struct _fdevent *ev, int i)

ape_event_kqueue.c

@@ -105,6 +105,10 @@ static ape_event_descriptor *event_kqueue_get_evd(struct _fdevent *ev, int i)
 static void event_kqueue_setsize(struct _fdevent *ev, int size)
 {
 	ev->events = realloc(ev->events, sizeof(struct kevent) * (size * 2));
+	if (ev->events == NULL) {
+		printf("Could not reallocate memory for kqueue\n");
+		exit(1);
+	}
 }
 
 static int event_kqueue_revent(struct _fdevent *ev, int i)

ape_events.h

@@ -92,7 +92,7 @@ struct _fdevent {
     struct epoll_event *events;
     int epoll_fd;
 #elif defined USE_SELECT_HANDLER
-	struct select_fdinfo_t **events;       /* Pointers into fds */
+    struct select_fdinfo_t **events;       /* Pointers into fds */
     ape_htable_t *fdhash;
 #endif
     int basemem;                /* Number of elements in events */