error handing for malloc/fgetc - #25
[smdp.git] / src / cstring.c
1 /*
2  * An implementation of expandable c strings in heap memory.
3  * Copyright (C) 2014 Michael Goehler
4  *
5  * This file is part of mdp.
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program. If not, see <http://www.gnu.org/licenses/>.
19  *
20  */
21
22 #include <string.h> // strlen
23 #include <stdio.h> // fprintf
24 #include <stdlib.h> // malloc, realloc
25
26 #include "cstring.h"
27
28 cstring_t *cstring_init() {
29     cstring_t *x = NULL;
30     if((x = malloc(sizeof(cstring_t))) != NULL) {
31         x->text = NULL;
32         x->size = x->alloc = 0;
33         x->expand = cstring_expand;
34         x->expand_arr = cstring_expand_arr;
35         x->reset = cstring_reset;
36         x->delete = cstring_delete;
37     } else {
38         fprintf(stderr, "%s\n", "cstring_init() failed to allocate memory.");
39         exit(EXIT_FAILURE);
40     }
41     return x;
42 }
43
44 void cstring_expand(cstring_t *self, char x) {
45     if(self->size + sizeof(x) + sizeof(char) > self->alloc) {
46         self->alloc += (REALLOC_ADD * sizeof(char));
47         if((self->text = realloc(self->text, self->alloc)) == NULL) {
48             fprintf(stderr, "%s\n", "cstring_expand() failed to reallocate memory.");
49             exit(EXIT_FAILURE);
50         }
51     }
52     self->text[self->size] = x;
53     self->text[self->size+1] = '\0';
54     self->size = strlen(self->text);
55 }
56
57 void cstring_expand_arr(cstring_t *self, char *x) {
58     if(self->size + strlen(x) + sizeof(char) > self->alloc) {
59         self->alloc = ((strlen(x) + self->size + 1) * sizeof(char));
60         if((self->text = realloc(self->text, self->alloc)) == NULL) {
61             fprintf(stderr, "%s\n", "cstring_expand() failed to reallocate memory.");
62             exit(EXIT_FAILURE);
63         }
64     }
65     self->text = strcat(self->text, x);
66     self->size = strlen(self->text);
67     self->text[self->size+1] = '\0';
68 }
69
70 void cstring_reset(cstring_t *self) {
71     free(self->text);
72     self->text = NULL;
73     self->size = self->alloc = 0;
74 }
75
76 void cstring_delete(cstring_t *self) {
77     free(self->text);
78     free(self);
79 }