fix null pointer out of memory

diff --git a/src/rename.c b/src/rename.c
index 385927f5bd24a63667d5c05be6633c4d53f8d356..f73d7e9a06d116f2583250bee6a481d162c9a737 100644 (file)
--- a/src/rename.c
+++ b/src/rename.c
@@ -86,6 +86,9 @@ char* rename_string(const char* src)
        int dest_count = 0;
        char* dest = malloc(sizeof(char) * STR_MAX_LENGTH);
 
+       if (dest == NULL)
+               return NULL;
+
        for (size_t i = 0; i < length; i++) {
                if (src[i] == SPACE) {
                        dest[dest_count++] = '_';
@@ -142,6 +145,9 @@ char* rename_revert(const char* src)
        int dest_count = 0;
        char* dest = malloc(sizeof(char) * STR_MAX_LENGTH);
 
+       if (dest == NULL)
+               return NULL;
+
        for (size_t i = 0; i < length; i++) {
                if (src[i] == '_') {
                        dest[dest_count++] = SPACE;