From ac013c6f539401d08cecf9351c74f382f5e435d6 Mon Sep 17 00:00:00 2001
From: Tom Ryder <tom@sanctum.geek.nz>
Date: Thu, 17 Mar 2016 23:43:37 +1300
Subject: Handle path being too long

---
 pwd.c | 8 ++++++--
 pwd.h | 2 ++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/pwd.c b/pwd.c
index 8416ed4..0c0c695 100644
--- a/pwd.c
+++ b/pwd.c
@@ -2,8 +2,12 @@
 
 int main(void)
 {
-    char buf[1024];
-    fprintf(stdout, "%s\n", getcwd(buf, 1024));
+    char buf[PATH_MAX];
+    if (getcwd(buf, PATH_MAX) == NULL) {
+        perror("getcwd");
+        exit(EXIT_FAILURE);
+    }
+    fprintf(stdout, "%s\n", buf);
     exit(EXIT_SUCCESS);
 }
 
diff --git a/pwd.h b/pwd.h
index 7ea4bed..a14c34b 100644
--- a/pwd.h
+++ b/pwd.h
@@ -5,4 +5,6 @@
 #include <stdlib.h>
 #include <unistd.h>
 
+#define PATH_MAX 512
+
 #endif
-- 
cgit v1.2.3