commit 1e1d29133f12031bd432b6818143c309f18de417
author Michael Wright <michaelwr@google.com> Mon Aug 19 20:08:04 2013 -0700
committer Michael Wright <michaelwr@google.com> Thu Aug 22 16:38:08 2013 -0700
tree f0b65f01260dac1e4d10cf4cb0133cf36e967719
parent 18860c524915bc991a9015bdbab32e918f5298d7

Add standard options to readlink

Change-Id: I4c2624d22ce879a3578d5b17440a9895b19e2f1f

toolbox/readlink.c

diff --git a/toolbox/readlink.c b/toolbox/readlink.c
index 9c1ab95..d114e20 100644
--- a/toolbox/readlink.c
+++ b/toolbox/readlink.c
@@ -34,26 +34,62 @@
 #include <stdlib.h>
 #include <unistd.h>
 
+static int skip_newline, quiet_errors, canonicalize;
+
 static void usage(char* name) {
-    fprintf(stderr, "Usage: %s FILE\n", name);
+    fprintf(stderr, "Usage: %s [OPTION]... FILE\n", name);
 }
 
 int readlink_main(int argc, char* argv[]) {
-    if (argc != 2) {
+    int c;
+    while ((c = getopt(argc, argv, "nfqs")) != -1) {
+        switch (c) {
+        case 'n':
+            skip_newline = 1;
+            break;
+        case 'f':
+            canonicalize = 1;
+            break;
+        case 'q':
+        case 's':
+            quiet_errors = 1;
+            break;
+        case '?':
+        default:
+            usage(argv[0]);
+            return EXIT_FAILURE;
+        }
+    }
+    int index = optind;
+    if (argc - index != 1) {
         usage(argv[0]);
         return EXIT_FAILURE;
     }
 
     char name[PATH_MAX+1];
-    ssize_t len = readlink(argv[1], name, PATH_MAX);
+    if (canonicalize) {
+        if(!realpath(argv[optind], name)) {
+            if (!quiet_errors) {
+                perror("readlink");
+            }
+            return EXIT_FAILURE;
+        }
+    } else {
+        ssize_t len = readlink(argv[1], name, PATH_MAX);
 
-    if (len < 0) {
-        perror("readlink");
-        return EXIT_FAILURE;
+        if (len < 0) {
+            if (!quiet_errors) {
+                perror("readlink");
+            }
+            return EXIT_FAILURE;
+        }
+        name[len] = '\0';
     }
 
-    name[len] = '\0';
-    printf("%s\n", name);
+    fputs(name, stdout);
+    if (!skip_newline) {
+        fputs("\n", stdout);
+    }
 
     return EXIT_SUCCESS;
 }