From 9e857b957545c1a59073badb3d05bd62da4fd076 Mon Sep 17 00:00:00 2001
From: Jonathan Dieter <jdieter@gmail.com>
Date: Thu, 26 Jul 2018 14:54:58 +0100
Subject: [PATCH] Use _GNU_SOURCE so we can use easy basename() function

Signed-off-by: Jonathan Dieter <jdieter@gmail.com>
---
 src/unzck.c  | 8 +++++---
 src/zck.c    | 9 +++++----
 src/zck_dl.c | 9 +++------
 3 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/src/unzck.c b/src/unzck.c
index ee71f5e..efafb53 100644
--- a/src/unzck.c
+++ b/src/unzck.c
@@ -24,6 +24,8 @@
  * POSSIBILITY OF SUCH DAMAGE.
  */
 
+#define _GNU_SOURCE
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
@@ -111,9 +113,9 @@ int main (int argc, char *argv[]) {
         perror("");
         exit(1);
     }
-
-    char *out_name = malloc(strlen(arguments.args[0]) - 3);
-    snprintf(out_name, strlen(arguments.args[0]) - 3, "%s", arguments.args[0]);
+    char *base_name = basename(arguments.args[0]);
+    char *out_name = malloc(strlen(base_name) - 3);
+    snprintf(out_name, strlen(base_name) - 3, "%s", base_name);
 
     int dst_fd = STDOUT_FILENO;
     if(!arguments.stdout) {
diff --git a/src/zck.c b/src/zck.c
index cf48ab9..92a0e2a 100644
--- a/src/zck.c
+++ b/src/zck.c
@@ -24,6 +24,8 @@
  * POSSIBILITY OF SUCH DAMAGE.
  */
 
+#define _GNU_SOURCE
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
@@ -118,10 +120,9 @@ int main (int argc, char *argv[]) {
 
     zck_set_log_level(arguments.log_level);
 
-    char *out_name;
-    out_name = malloc(strlen(arguments.args[0]) + 5);
-    snprintf(out_name, strlen(arguments.args[0]) + 5, "%s.zck",
-             arguments.args[0]);
+    char *base_name = basename(arguments.args[0]);
+    char *out_name = malloc(strlen(base_name) + 5);
+    snprintf(out_name, strlen(base_name) + 5, "%s.zck", base_name);
 
     /* Set dictionary if available */
     char *dict = NULL;
diff --git a/src/zck_dl.c b/src/zck_dl.c
index d2bc20f..93bdc5a 100644
--- a/src/zck_dl.c
+++ b/src/zck_dl.c
@@ -24,6 +24,8 @@
  * POSSIBILITY OF SUCH DAMAGE.
  */
 
+#define _GNU_SOURCE
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
@@ -32,7 +34,6 @@
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <unistd.h>
-#include <libgen.h>
 #include <errno.h>
 #include <argp.h>
 #include <zck.h>
@@ -300,13 +301,10 @@ int main (int argc, char *argv[]) {
         exit(10);
     }
 
-    char *outname_full = calloc(1, strlen(arguments.args[0])+1);
-    memcpy(outname_full, arguments.args[0], strlen(arguments.args[0]));
-    char *outname = basename(outname_full);
+    char *outname = basename(arguments.args[0]);
     int dst_fd = open(outname, O_RDWR | O_CREAT, 0644);
     if(dst_fd < 0) {
         printf("Unable to open %s: %s\n", outname, strerror(errno));
-        free(outname_full);
         exit(10);
     }
     zckCtx *zck_tgt = zck_create();
@@ -447,7 +445,6 @@ int main (int argc, char *argv[]) {
             break;
     }
 out:
-    free(outname_full);
     zck_dl_free(&dl);
     zck_free(&zck_tgt);
     zck_free(&zck_src);
-- 
2.30.2