Imported Upstream version 1.4
[debian/gzip] / lib / unlinkat.c
1 /* Work around unlinkat bugs on Solaris 9.
2
3    Copyright (C) 2009-2010 Free Software Foundation, Inc.
4
5    This program is free software: you can redistribute it and/or modify
6    it under the terms of the GNU General Public License as published by
7    the Free Software Foundation; either version 3 of the License, or
8    (at your option) any later version.
9
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU General Public License for more details.
14
15    You should have received a copy of the GNU General Public License
16    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
17
18 /* Written by Eric Blake.  */
19
20 #include <config.h>
21
22 #include <unistd.h>
23
24 #include <errno.h>
25 #include <fcntl.h>
26 #include <string.h>
27 #include <sys/stat.h>
28
29 #include "openat.h"
30
31 #if HAVE_UNLINKAT
32
33 # undef unlinkat
34
35 /* unlinkat without AT_REMOVEDIR does not honor trailing / on Solaris
36    9.  Solve it in a similar manner to unlink.  */
37
38 int
39 rpl_unlinkat (int fd, char const *name, int flag)
40 {
41   size_t len;
42   int result = 0;
43   /* rmdir behavior has no problems with trailing slash.  */
44   if (flag & AT_REMOVEDIR)
45     return unlinkat (fd, name, flag);
46
47   len = strlen (name);
48   if (len && ISSLASH (name[len - 1]))
49     {
50       /* See the lengthy comment in unlink.c why we disobey the POSIX
51          rule of letting unlink("link-to-dir/") attempt to unlink a
52          directory.  */
53       struct stat st;
54       result = lstatat (fd, name, &st);
55       if (result == 0)
56         {
57           /* Trailing NUL will overwrite the trailing slash.  */
58           char *short_name = malloc (len);
59           if (!short_name)
60             {
61               errno = EPERM;
62               return -1;
63             }
64           memcpy (short_name, name, len);
65           while (len && ISSLASH (short_name[len - 1]))
66             short_name[--len] = '\0';
67           if (len && (lstatat (fd, short_name, &st) || S_ISLNK (st.st_mode)))
68             {
69               free (short_name);
70               errno = EPERM;
71               return -1;
72             }
73           free (short_name);
74         }
75     }
76   if (!result)
77     result = unlinkat (fd, name, flag);
78   return result;
79 }
80
81 #else /* !HAVE_UNLINKAT */
82
83 /* Replacement for Solaris' function by the same name.
84    <http://www.google.com/search?q=unlinkat+site:docs.sun.com>
85    First, try to simulate it via (unlink|rmdir) ("/proc/self/fd/FD/FILE").
86    Failing that, simulate it via save_cwd/fchdir/(unlink|rmdir)/restore_cwd.
87    If either the save_cwd or the restore_cwd fails (relatively unlikely),
88    then give a diagnostic and exit nonzero.
89    Otherwise, this function works just like Solaris' unlinkat.  */
90
91 # define AT_FUNC_NAME unlinkat
92 # define AT_FUNC_F1 rmdir
93 # define AT_FUNC_F2 unlink
94 # define AT_FUNC_USE_F1_COND AT_REMOVEDIR
95 # define AT_FUNC_POST_FILE_PARAM_DECLS , int flag
96 # define AT_FUNC_POST_FILE_ARGS        /* empty */
97 # include "at-func.c"
98 # undef AT_FUNC_NAME
99 # undef AT_FUNC_F1
100 # undef AT_FUNC_F2
101 # undef AT_FUNC_USE_F1_COND
102 # undef AT_FUNC_POST_FILE_PARAM_DECLS
103 # undef AT_FUNC_POST_FILE_ARGS
104
105 #endif /* !HAVE_UNLINKAT */