diff options
| -rw-r--r-- | dimension/lexer.l | 3 | ||||
| -rw-r--r-- | dimension/parse.c | 2 |
2 files changed, 5 insertions, 0 deletions
diff --git a/dimension/lexer.l b/dimension/lexer.l index 4beeee7..d198b62 100644 --- a/dimension/lexer.l +++ b/dimension/lexer.l @@ -97,6 +97,9 @@ if (string_length + len + 1 >= string_length) { \ string_extent = 2*(string_length + len + 1); \ lvalp->value = realloc(lvalp->value, string_extent); \ + if (!lvalp->value) \ + dmnsn_error(DMNSN_SEVERITY_HIGH, \ + "Couldn't allocate space for token value."); \ } \ \ strncpy(lvalp->value + string_length, str, len + 1); \ diff --git a/dimension/parse.c b/dimension/parse.c index 7897b86..9f4a783 100644 --- a/dimension/parse.c +++ b/dimension/parse.c @@ -134,6 +134,8 @@ dmnsn_patricia_insert(dmnsn_patricia_trie *trie, /* Split the tree */ dmnsn_patricia_trie *copy = dmnsn_new_patricia_trie(); copy->prefix = realloc(copy->prefix, strlen(prefix) + 1); + if (!trie->prefix) + dmnsn_error(DMNSN_SEVERITY_HIGH, "Couldn't allocate room for prefix."); strcpy(copy->prefix, prefix); *prefix = '\0'; |