Implement memory management TODOs

[centaur.git] / src / elfucli.c

diff --git a/src/elfucli.c b/src/elfucli.c
index b3ea29f2440381c68f685c051215cdc3da8cdc27..b01c91a1c6992aecac7fbce4e5ec890eca20912c 100644 (file)
--- a/src/elfucli.c
+++ b/src/elfucli.c
@@ -149,9 +149,18 @@ int main(int argc, char **argv)
             goto ERR;
           } else {
             printf("--reladd: Injecting %s...\n", optarg);
-            elfu_mCheck(mrel);
-            elfu_mReladd(me, mrel);
+            if (elfu_mCheck(mrel)) {
+              printf("--reladd: Check for input file failed.\n");
+              elfu_mElfDestroy(mrel);
+              goto ERR;
+            }
+            if (elfu_mReladd(me, mrel)) {
+              printf("--reladd: Failed.\n");
+              elfu_mElfDestroy(mrel);
+              goto ERR;
+            }
             printf("--reladd: Injected %s.\n", optarg);
+            elfu_mElfDestroy(mrel);
           }
         }
         break;