Merge pull request #48 from ryanpcmcquen/patch-1
[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, memmove
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->strip = cstring_strip;
36         x->reset = cstring_reset;
37         x->delete = cstring_delete;
38     } else {
39         fprintf(stderr, "%s\n", "cstring_init() failed to allocate memory.");
40         exit(EXIT_FAILURE);
41     }
42     return x;
43 }
44
45 void cstring_expand(cstring_t *self, char x) {
46     if(self->size + sizeof(x) + sizeof(char) > self->alloc) {
47         self->alloc += (REALLOC_ADD * sizeof(char));
48         if((self->text = realloc(self->text, self->alloc)) == NULL) {
49             fprintf(stderr, "%s\n", "cstring_expand() failed to reallocate memory.");
50             exit(EXIT_FAILURE);
51         }
52     }
53     self->text[self->size] = x;
54     self->text[self->size+1] = '\0';
55     self->size = strlen(self->text);
56 }
57
58 void cstring_expand_arr(cstring_t *self, char *x) {
59     if(self->size + strlen(x) + sizeof(char) > self->alloc) {
60         self->alloc = ((strlen(x) + self->size + 1) * sizeof(char));
61         if((self->text = realloc(self->text, self->alloc)) == NULL) {
62             fprintf(stderr, "%s\n", "cstring_expand() failed to reallocate memory.");
63             exit(EXIT_FAILURE);
64         }
65     }
66     self->text = strcat(self->text, x);
67     self->size = strlen(self->text);
68     self->text[self->size+1] = '\0';
69 }
70
71 void cstring_strip(cstring_t *self, int pos, int len) {
72     if(pos + len >= self->size) {
73         if(pos <= self->size) {
74             self->text[pos] = '\0';
75             self->size = pos;
76         }
77         return;
78     }
79     memmove(&self->text[pos], &self->text[pos+len], self->size - pos);
80     self->size -= len;
81 }
82
83 void cstring_reset(cstring_t *self) {
84     free(self->text);
85     self->text = NULL;
86     self->size = self->alloc = 0;
87 }
88
89 void cstring_delete(cstring_t *self) {
90     free(self->text);
91     free(self);
92 }