X-Git-Url: https://git.danieliu.xyz/?a=blobdiff_plain;f=src%2Fcstack.c;h=df27ff436fde1aa29f05425fb0ad1be39e766202;hb=4b8ad1b3b464836eca2e876dd418a8e8d87bd721;hp=a5ecc0695fd770bfc3d246dccf8320d0fdf6499d;hpb=43aa3385fb0e78b302a9191789d59115402035f8;p=smdp.git

diff --git a/src/cstack.c b/src/cstack.c
index a5ecc06..df27ff4 100644
--- a/src/cstack.c
+++ b/src/cstack.c
@@ -19,44 +19,52 @@
  *
  */
 
+#include <wchar.h>
+#include <stdio.h> // fprintf
 #include <stdlib.h> // malloc, realloc
 
 #include "cstack.h"
 
 cstack_t *cstack_init() {
-    cstack_t *stack = malloc(sizeof(cstack_t));
-    stack->content = NULL;
-    stack->alloc = stack->size = 0;
-    stack->head = -1;
-    stack->push = cstack_push;
-    stack->pop = cstack_pop;
-    stack->top = cstack_top;
-    stack->empty = cstack_empty;
-    stack->delete = cstack_delete;
+    cstack_t *stack = NULL;
+    if((stack = malloc(sizeof(cstack_t))) != NULL) {
+        stack->content = NULL;
+        stack->alloc = stack->size = 0;
+        stack->head = -1;
+        stack->push = cstack_push;
+        stack->pop = cstack_pop;
+        stack->top = cstack_top;
+        stack->empty = cstack_empty;
+        stack->delete = cstack_delete;
+    } else {
+        fprintf(stderr, "%s\n", "cstack_init() failed to allocate memory.");
+        exit(EXIT_FAILURE);
+    }
     return stack;
 }
 
-void cstack_push(cstack_t *self, char c) {
+void cstack_push(cstack_t *self, wchar_t c) {
     if(self->size + sizeof(c) > self->alloc) {
-        self->alloc += (sizeof(char));
-        self->content = realloc(self->content, self->alloc);
+        self->alloc += (sizeof(wchar_t));
+        if((self->content = realloc(self->content, self->alloc)) == NULL) {
+            fprintf(stderr, "%s\n", "cstack_push() failed to reallocate memory.");
+            exit(EXIT_FAILURE);
+        }
     }
     self->content[++self->head] = c;
-    self->size += (sizeof(char));
+    self->size += (sizeof(wchar_t));
 }
 
-char cstack_pop(cstack_t *self) {
-    self->size -= (sizeof(char));
+wchar_t cstack_pop(cstack_t *self) {
+    self->size -= (sizeof(wchar_t));
     return self->content[self->head--];
 }
 
-int cstack_top(cstack_t *self, char c) {
-    if(self->head >= 0 && self->content[self->head] == c)
-        return 1;
-    return 0;
+bool cstack_top(cstack_t *self, wchar_t c) {
+    return self->head >= 0 && self->content[self->head] == c;
 }
 
-int cstack_empty(cstack_t *self) {
+bool cstack_empty(cstack_t *self) {
     return self->head == -1;
 }