xstring: add char* concatenation, use func for realloc memory

1 files changed, 14 insertions, 11 deletions

diff --git a/xstring.c b/xstring.c
index d540828..17804f3 100644
--- a/xstring.c
+++ b/xstring.c
@@ -58,19 +58,22 @@ static void string_realloc(string *str, int cap)
 

 void string_addch(string *str, int ch)

 {

-	if(str->capacity <= (str->len + 1)) {

-		int i;

-		int updated_cap = str->capacity + INIT_LEN_STRING;

-		char *tmp_s = str->s;

-		str->s = malloc(sizeof(char) * (updated_cap + 1));

-		for(i = 0; i < updated_cap; i++)

-			str->s[i] = (i >= str->capacity) ? '\0' : tmp_s[i];

-

-		str->capacity = updated_cap;

-		free(tmp_s);

-	}

+	if((str->len + 1) >= str->capacity)

+		string_realloc(str, str->capacity + INIT_LEN_STRING);

 

 	str->s[str->len] = ch;

 	str->len++;

 	str->s[str->len] = '\0';

 }

+

+void string_addstr(string *str, const char *src, size_t src_len)

+{

+	if((str->len + src_len) >= str->capacity)

+		string_realloc(str, str->capacity + src_len);

+

+	while(src_len > 0 && (str->s[str->len] = *src++)) {

+		src_len--;

+		str->len++;

+	}

+	str->s[str->len] = '\0';

+}