new xmalloc_readlink_follow() routine to fully expand trailing symlinks
[people/mcb30/busybox.git] / libbb / xreadlink.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  *  xreadlink.c - safe implementation of readlink.
4  *  Returns a NULL on failure...
5  */
6
7 #include "libbb.h"
8
9 /*
10  * NOTE: This function returns a malloced char* that you will have to free
11  * yourself. You have been warned.
12  */
13 char *xmalloc_readlink(const char *path)
14 {
15         enum { GROWBY = 80 }; /* how large we will grow strings by */
16
17         char *buf = NULL;
18         int bufsize = 0, readsize = 0;
19
20         do {
21                 bufsize += GROWBY;
22                 buf = xrealloc(buf, bufsize);
23                 readsize = readlink(path, buf, bufsize);
24                 if (readsize == -1) {
25                         free(buf);
26                         return NULL;
27                 }
28         } while (bufsize < readsize + 1);
29
30         buf[readsize] = '\0';
31
32         return buf;
33 }
34
35 /*
36  * this routine is not the same as realpath(), which canonicalizes
37  * the given path completely.  this routine only follows trailing
38  * symlinks until a real file is reached, and returns its name. 
39  * intermediate symlinks are not expanded.  as above, a malloced
40  * char* is returned, which must be freed.
41  */
42 char *xmalloc_readlink_follow(const char *path)
43 {
44         char *buf = NULL, *lpc, *linkpath;
45         int bufsize;
46         smallint looping = 0;
47
48         buf = strdup(path);
49         bufsize = strlen(path) + 1;
50
51         while(1) {
52                 linkpath = xmalloc_readlink(buf);
53                 if (!linkpath) {
54                         if (errno == EINVAL) /* not a symlink */
55                                 return buf;
56                         free(buf);
57                         return NULL;
58                 } 
59
60                 if (*linkpath == '/') {
61                         free(buf);
62                         buf = linkpath;
63                         bufsize = strlen(linkpath) + 1;
64                 } else {
65                         bufsize += strlen(linkpath);
66                         if (looping++ > MAXSYMLINKS) {
67                                 free(linkpath);
68                                 free(buf);
69                                 return NULL;
70                         }
71                         buf = xrealloc(buf, bufsize);
72                         lpc = bb_get_last_path_component_strip(buf);
73                         strcpy(lpc, linkpath);
74                         free(linkpath);
75                 }
76         }
77
78 }
79
80 char *xmalloc_readlink_or_warn(const char *path)
81 {
82         char *buf = xmalloc_readlink(path);
83         if (!buf) {
84                 /* EINVAL => "file: Invalid argument" => puzzled user */
85                 bb_error_msg("%s: cannot read link (not a symlink?)", path);
86         }
87         return buf;
88 }
89
90 /* UNUSED */
91 #if 0
92 char *xmalloc_realpath(const char *path)
93 {
94 #if defined(__GLIBC__) && !defined(__UCLIBC__)
95         /* glibc provides a non-standard extension */
96         return realpath(path, NULL);
97 #else
98         char buf[PATH_MAX+1];
99
100         /* on error returns NULL (xstrdup(NULL) ==NULL) */
101         return xstrdup(realpath(path, buf));
102 #endif
103 }
104 #endif