diff options
Diffstat (limited to 'libiberty/dyn-string.c')
| -rw-r--r-- | libiberty/dyn-string.c | 198 |
1 files changed, 137 insertions, 61 deletions
diff --git a/libiberty/dyn-string.c b/libiberty/dyn-string.c index f4a9d9a..69897f8 100644 --- a/libiberty/dyn-string.c +++ b/libiberty/dyn-string.c @@ -36,14 +36,26 @@ Boston, MA 02111-1307, USA. */ #include "libiberty.h" #include "dyn-string.h" +/* If this file is being compiled for inclusion in the C++ runtime + library, as part of the demangler implementation, we don't want to + abort if an allocation fails. Instead, percolate an error code up + through the call chain. */ + +#ifdef IN_LIBGCC2 +#define RETURN_ON_ALLOCATION_FAILURE +#endif + /* Performs in-place initialization of a dyn_string struct. This function can be used with a dyn_string struct on the stack or embedded in another object. The contents of of the string itself are still dynamically allocated. The string initially is capable of holding at least SPACE characeters, including the terminating - NUL. If SPACE is 0, it will silently be increated to 1. */ + NUL. If SPACE is 0, it will silently be increated to 1. -void + If RETURN_ON_ALLOCATION_FAILURE is defined and memory allocation + fails, returns 0. Otherwise returns 1. */ + +int dyn_string_init (ds_struct_ptr, space) struct dyn_string *ds_struct_ptr; int space; @@ -52,22 +64,44 @@ dyn_string_init (ds_struct_ptr, space) if (space == 0) space = 1; - ds_struct_ptr->allocated = space; +#ifdef RETURN_ON_ALLOCATION_FAILURE + ds_struct_ptr->s = (char *) malloc (space); + if (ds_struct_ptr->s == NULL) + return 0; +#else ds_struct_ptr->s = (char *) xmalloc (space); +#endif + ds_struct_ptr->allocated = space; ds_struct_ptr->length = 0; ds_struct_ptr->s[0] = '\0'; -} -/* Create a new dynamic string capable of holding at least SPACE characters, - including the terminating NUL. If SPACE is 0, it will be silently - increased to 1. */ + return 1; +} + +/* Create a new dynamic string capable of holding at least SPACE + characters, including the terminating NUL. If SPACE is 0, it will + be silently increased to 1. If RETURN_ON_ALLOCATION_FAILURE is + defined and memory allocation fails, returns NULL. Otherwise + returns the newly allocated string. */ dyn_string_t dyn_string_new (space) int space; { - dyn_string_t result = (dyn_string_t) xmalloc (sizeof (struct dyn_string)); + dyn_string_t result; +#ifdef RETURN_ON_ALLOCATION_FAILURE + result = (dyn_string_t) malloc (sizeof (struct dyn_string)); + if (result == NULL) + return NULL; + if (!dyn_string_init (result, space)) + { + free (result); + return NULL; + } +#else + result = (dyn_string_t) xmalloc (sizeof (struct dyn_string)); dyn_string_init (result, space); +#endif return result; } @@ -83,7 +117,7 @@ dyn_string_delete (ds) /* Returns the contents of DS in a buffer allocated with malloc. It is the caller's responsibility to deallocate the buffer using free. - DS is then set to the empty string. */ + DS is then set to the empty string. Deletes DS itself. */ char* dyn_string_release (ds) @@ -93,15 +127,18 @@ dyn_string_release (ds) char* result = ds->s; /* The buffer is no longer owned by DS. */ ds->s = NULL; - /* Reinitialize DS to the empty string. */ - dyn_string_init (ds, 0); + /* Delete DS. */ + free (ds); /* Return the old buffer. */ return result; } /* Increase the capacity of DS so it can hold at least SPACE characters, plus the terminating NUL. This function will not (at - present) reduce the capacity of DS. */ + present) reduce the capacity of DS. Returns DS on success. + + If RETURN_ON_ALLOCATION_FAILURE is defined and a memory allocation + operation fails, deletes DS and returns NULL. */ dyn_string_t dyn_string_resize (ds, space) @@ -113,14 +150,24 @@ dyn_string_resize (ds, space) /* Increase SPACE to hold the NUL termination. */ ++space; + /* Increase allocation by factors of two. */ while (space > new_allocated) new_allocated *= 2; if (new_allocated != ds->allocated) { - /* We actually need more space. */ ds->allocated = new_allocated; + /* We actually need more space. */ +#ifdef RETURN_ON_ALLOCATION_FAILURE + ds->s = (char *) realloc (ds->s, ds->allocated); + if (ds->s == NULL) + { + free (ds); + return NULL; + } +#else ds->s = (char *) xrealloc (ds->s, ds->allocated); +#endif } return ds; @@ -138,9 +185,10 @@ dyn_string_clear (ds) } /* Makes the contents of DEST the same as the contents of SRC. DEST - and SRC must be distinct. */ + and SRC must be distinct. Returns 1 on success. On failure, if + RETURN_ON_ALLOCATION_FAILURE, deletes DEST and returns 0. */ -void +int dyn_string_copy (dest, src) dyn_string_t dest; dyn_string_t src; @@ -149,55 +197,66 @@ dyn_string_copy (dest, src) abort (); /* Make room in DEST. */ - dyn_string_resize (dest, src->length); + if (dyn_string_resize (dest, src->length) == NULL) + return 0; /* Copy DEST into SRC. */ strcpy (dest->s, src->s); /* Update the size of DEST. */ dest->length = src->length; + return 1; } -/* Copies SRC, a NUL-terminated string, into DEST. */ +/* Copies SRC, a NUL-terminated string, into DEST. Returns 1 on + success. On failure, if RETURN_ON_ALLOCATION_FAILURE, deletes DEST + and returns 0. */ -void +int dyn_string_copy_cstr (dest, src) dyn_string_t dest; const char *src; { int length = strlen (src); /* Make room in DEST. */ - dyn_string_resize (dest, length); + if (dyn_string_resize (dest, length) == NULL) + return 0; /* Copy DEST into SRC. */ strcpy (dest->s, src); /* Update the size of DEST. */ dest->length = length; + return 1; } /* Inserts SRC at the beginning of DEST. DEST is expanded as - necessary. SRC and DEST must be distinct. */ + necessary. SRC and DEST must be distinct. Returns 1 on success. + On failure, if RETURN_ON_ALLOCATION_FAILURE, deletes DEST and + returns 0. */ -void +int dyn_string_prepend (dest, src) dyn_string_t dest; dyn_string_t src; { - dyn_string_insert (dest, 0, src); + return dyn_string_insert (dest, 0, src); } /* Inserts SRC, a NUL-terminated string, at the beginning of DEST. - DEST is expanded as necessary. */ + DEST is expanded as necessary. Returns 1 on success. On failure, + if RETURN_ON_ALLOCATION_FAILURE, deletes DEST and returns 0. */ -void +int dyn_string_prepend_cstr (dest, src) dyn_string_t dest; const char *src; { - dyn_string_insert_cstr (dest, 0, src); + return dyn_string_insert_cstr (dest, 0, src); } -/* Inserts SRC into DEST starting at position POS. DEST is expanded as - necessary. SRC and DEST must be distinct. */ +/* Inserts SRC into DEST starting at position POS. DEST is expanded + as necessary. SRC and DEST must be distinct. Returns 1 on + success. On failure, if RETURN_ON_ALLOCATION_FAILURE, deletes DEST + and returns 0. */ -void +int dyn_string_insert (dest, pos, src) dyn_string_t dest; int pos; @@ -208,7 +267,8 @@ dyn_string_insert (dest, pos, src) if (src == dest) abort (); - dyn_string_resize (dest, dest->length + src->length); + if (dyn_string_resize (dest, dest->length + src->length) == NULL) + return 0; /* Make room for the insertion. Be sure to copy the NUL. */ for (i = dest->length; i >= pos; --i) dest->s[i + src->length] = dest->s[i]; @@ -216,12 +276,15 @@ dyn_string_insert (dest, pos, src) strncpy (dest->s + pos, src->s, src->length); /* Compute the new length. */ dest->length += src->length; + return 1; } /* Inserts SRC, a NUL-terminated string, into DEST starting at - position POS. DEST is expanded as necessary. */ + position POS. DEST is expanded as necessary. Returns 1 on + success. On failure, RETURN_ON_ALLOCATION_FAILURE, deletes DEST + and returns 0. */ -void +int dyn_string_insert_cstr (dest, pos, src) dyn_string_t dest; int pos; @@ -230,7 +293,8 @@ dyn_string_insert_cstr (dest, pos, src) int i; int length = strlen (src); - dyn_string_resize (dest, dest->length + length); + if (dyn_string_resize (dest, dest->length + length) == NULL) + return 0; /* Make room for the insertion. Be sure to copy the NUL. */ for (i = dest->length; i >= pos; --i) dest->s[i + length] = dest->s[i]; @@ -238,63 +302,72 @@ dyn_string_insert_cstr (dest, pos, src) strncpy (dest->s + pos, src, length); /* Compute the new length. */ dest->length += length; + return 1; } -/* Append S to DS, resizing DS if necessary. Returns DS. */ +/* Append S to DS, resizing DS if necessary. Returns 1 on success. + On failure, if RETURN_ON_ALLOCATION_FAILURE, deletes DEST and + returns 0. */ -dyn_string_t -dyn_string_append (ds, s) - dyn_string_t ds; +int +dyn_string_append (dest, s) + dyn_string_t dest; dyn_string_t s; { - dyn_string_resize (ds, ds->length + s->length); - strcpy (ds->s + ds->length, s->s); - ds->length += s->length; - return ds; + if (dyn_string_resize (dest, dest->length + s->length) == 0) + return 0; + strcpy (dest->s + dest->length, s->s); + dest->length += s->length; + return 1; } /* Append the NUL-terminated string S to DS, resizing DS if necessary. - Returns DS. */ + Returns 1 on success. On failure, if RETURN_ON_ALLOCATION_FAILURE, + deletes DEST and returns 0. */ -dyn_string_t -dyn_string_append_cstr (ds, s) - dyn_string_t ds; +int +dyn_string_append_cstr (dest, s) + dyn_string_t dest; const char *s; { int len = strlen (s); /* The new length is the old length plus the size of our string, plus one for the null at the end. */ - dyn_string_resize (ds, ds->length + len); - strcpy (ds->s + ds->length, s); - ds->length += len; - - return ds; + if (dyn_string_resize (dest, dest->length + len) == NULL) + return 0; + strcpy (dest->s + dest->length, s); + dest->length += len; + return 1; } -/* Appends C to the end of DS. */ +/* Appends C to the end of DEST. Returns 1 on success. On failiure, + if RETURN_ON_ALLOCATION_FAILURE, deletes DEST and returns 0. */ -dyn_string_t -dyn_string_append_char (ds, c) - dyn_string_t ds; +int +dyn_string_append_char (dest, c) + dyn_string_t dest; int c; { /* Make room for the extra character. */ - dyn_string_resize (ds, ds->length + 1); + if (dyn_string_resize (dest, dest->length + 1) == NULL) + return 0; /* Append the character; it will overwrite the old NUL. */ - ds->s[ds->length] = c; + dest->s[dest->length] = c; /* Add a new NUL at the end. */ - ds->s[ds->length + 1] = '\0'; + dest->s[dest->length + 1] = '\0'; /* Update the length. */ - ++(ds->length); - return ds; + ++(dest->length); + return 1; } /* Sets the contents of DEST to the substring of SRC starting at START and ending before END. START must be less than or equal to END, - and both must be between zero and the length of SRC, inclusive. */ + and both must be between zero and the length of SRC, inclusive. + Returns 1 on success. On failure, if RETURN_ON_ALLOCATION_FAILURE, + deletes DEST and returns 0. */ -void +int dyn_string_substring (dest, src, start, end) dyn_string_t dest; dyn_string_t src; @@ -308,7 +381,8 @@ dyn_string_substring (dest, src, start, end) abort (); /* Make room for the substring. */ - dyn_string_resize (dest, length); + if (dyn_string_resize (dest, length) == NULL) + return 0; /* Copy the characters in the substring, */ for (i = length; --i >= 0; ) dest->s[i] = src->s[start + i]; @@ -316,6 +390,8 @@ dyn_string_substring (dest, src, start, end) dest->s[length] = '\0'; /* Record the length of the substring. */ dest->length = length; + + return 1; } /* Returns non-zero if DS1 and DS2 have the same contents. */ |