mirrors/coreutils
0
0
Fork 0
mirror of https://github.com/coreutils/coreutils.git synced 2024-12-03 23:13:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

(defined_S_IFMT): New macro.

Browse Source 
Include stat-macros.h.
Include stdlib.h, for abort().
Don't include stdio.h or assert.h; no longer needed.
(same_file_type): Don't assume S_IFMT is defined, as POSIX
does not require this.  Don't assume S_IFCHR and S_IFBLK have
their usual sort of bit pattern.
(fchmod_new): Open with O_NOCTTY for as well, for minor
improvement on hosts where that matters.  Don't bother to assert,
since the caller (in this source file) checks the same thing.
Discard any errno from a close failure, for consistency with other
code.
This commit is contained in:
Paul Eggert 2005-12-27 07:55:50 +00:00
parent cc4d59e9cd
commit 0d7468cabd
1 changed files with 43 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
lib/chmod-safer.c
View File

@ -23,16 +23,21 @@
#include "chmod-safer.h"
#ifdef S_IFMT
# define defined_S_IFMT true
#else
# define defined_S_IFMT false
#endif
#include "stat-macros.h"
#include <stdbool.h>
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
#include "fcntl--.h" /* for the open->open_safer mapping */
#if !defined O_NOFOLLOW
#ifndef O_NOFOLLOW
# define O_NOFOLLOW 0
#endif
@ -48,12 +53,17 @@ static inline bool
same_file_type (struct stat const *st, dev_t device, mode_t type)
{
/* The types must always match. */
if ( ! (st->st_mode & type))
if (! (defined_S_IFMT ? (st->st_mode & S_IFMT) == type
: type == S_IFDIR ? S_ISDIR (st->st_mode)
: type == S_IFIFO ? S_ISFIFO (st->st_mode)
: type == S_IFBLK ? S_ISBLK (st->st_mode)
: type == S_IFCHR ? S_ISCHR (st->st_mode)
: (abort (), false)))
return false;
/* For character and block devices, the major and minor device
numbers must match, too. */
if (type & (S_IFCHR | S_IFBLK))
if (S_ISBLK (type) || S_ISCHR (type))
return st->st_rdev == device;
return true;
@ -66,15 +76,14 @@ same_file_type (struct stat const *st, dev_t device, mode_t type)
static int
fchmod_new (char const *file, mode_t mode, dev_t device, mode_t file_type)
{
int fail = 1;
int result = 0;
struct stat sb;
int saved_errno = 0;
int fd = open (file, O_NOFOLLOW | O_RDONLY | O_NDELAY);
int fd = open (file, O_RDONLY | O_NOCTTY | O_NOFOLLOW | O_NONBLOCK);
int saved_errno;
assert (O_NOFOLLOW);
if (fd < 0)
return -1;
if (0 <= fd
&& fstat (fd, &sb) == 0
/* Given the entry we've just created, if its link count is
not 1 or its type/device has changed, then someone may be
trying to do something nasty. However, the risk of such an
@ -83,23 +92,23 @@ fchmod_new (char const *file, mode_t mode, dev_t device, mode_t file_type)
code below reports the failure to set permissions.
Note that we don't check the link count if the expected
type is `directory'. */
&& (((sb.st_nlink == 1 || file_type == S_IFDIR)
&& same_file_type (&sb, device, file_type))
|| ((errno = EACCES), 0))
&& fchmod (fd, mode) == 0)
result = fstat (fd, &sb);
if (result == 0)
{
fail = 0;
}
if ((sb.st_nlink == 1 || S_ISDIR (file_type))
&& same_file_type (&sb, device, file_type))
result = fchmod (fd, mode);
else
{
saved_errno = errno;
errno = EACCES;
result = -1;
}
}
if (0 <= fd && close (fd) != 0 && saved_errno == 0)
saved_errno = errno;
close (fd);
errno = saved_errno;
return fail;
return result;
}
/* Use a safer variant of chmod, if the underlying system facilities permit.