1 files changed, 8 insertions, 3 deletions

diff --git a/src/encryption.c b/src/encryption.c
index 4327506..0bc3d50 100644
--- a/src/encryption.c
+++ b/src/encryption.c
@@ -5,7 +5,6 @@
 */
 
 #define _GNU_SOURCE
-#define TESTING
 
 #include "encryption.h"
 #include "ll.h"
@@ -237,9 +236,15 @@ int ENorDE_cryptvxgg(const struct nodelist *list, const char *passphrase, int fl
         if(flag == VXGG_DECRYPT) {
             char *newname = NULL;
             #ifndef TESTING
-                asprintf(&newname, "%%0.%lds", strlen(p->fullpath) - strlen(".vxgg"));
+                if(asprintf(&newname, "%%0.%lds", strlen(p->fullpath) - strlen(".vxgg")) < 0) {
+                    endwin();
+                    error(1, errno, "[VX-GAMBLEGROUND] Couldn't get shortened name");
+                }
             #else
-                asprintf(&newname, "%%0.%lds", strlen(p->fullpath) - strlen(".vxgg") - PHRASESIZE - 1);
+                if(asprintf(&newname, "%%0.%lds", strlen(p->fullpath) - strlen(".vxgg") - PHRASESIZE - 1) < 0) {
+                    endwin();
+                    error(1, errno, "[VX-GAMBLEGROUND] Couldn't get shortened name");
+                }
             #endif
             rename_format(p->fullpath, newname, p->fullpath);
             free(newname);