Commit cfeed4da authored by Wayne Davison's avatar Wayne Davison

- If we return an error because of dry_run being set, we now set errno.

- The do_mkstemp() routine now preserves the errno value from fchmod()
  if that call fails.
- Improved the CHECK_RO macro stuff.
parent 4068d861
...@@ -31,19 +31,27 @@ extern int read_only; ...@@ -31,19 +31,27 @@ extern int read_only;
extern int list_only; extern int list_only;
extern int preserve_perms; extern int preserve_perms;
#define CHECK_RO if (read_only || list_only) {errno = EROFS; return -1;} #define RETURN_ERROR_IF(x,e) \
do { \
if (x) { \
errno = (e); \
return -1; \
} \
} while (0)
#define RETURN_ERROR_IF_RO_OR_LO RETURN_ERROR_IF(read_only || list_only, EROFS)
int do_unlink(char *fname) int do_unlink(char *fname)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return unlink(fname); return unlink(fname);
} }
int do_symlink(char *fname1, char *fname2) int do_symlink(char *fname1, char *fname2)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return symlink(fname1, fname2); return symlink(fname1, fname2);
} }
...@@ -51,7 +59,7 @@ int do_symlink(char *fname1, char *fname2) ...@@ -51,7 +59,7 @@ int do_symlink(char *fname1, char *fname2)
int do_link(char *fname1, char *fname2) int do_link(char *fname1, char *fname2)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return link(fname1, fname2); return link(fname1, fname2);
} }
#endif #endif
...@@ -59,7 +67,7 @@ int do_link(char *fname1, char *fname2) ...@@ -59,7 +67,7 @@ int do_link(char *fname1, char *fname2)
int do_lchown(const char *path, uid_t owner, gid_t group) int do_lchown(const char *path, uid_t owner, gid_t group)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return lchown(path, owner, group); return lchown(path, owner, group);
} }
...@@ -67,7 +75,7 @@ int do_lchown(const char *path, uid_t owner, gid_t group) ...@@ -67,7 +75,7 @@ int do_lchown(const char *path, uid_t owner, gid_t group)
int do_mknod(char *pathname, mode_t mode, dev_t dev) int do_mknod(char *pathname, mode_t mode, dev_t dev)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return mknod(pathname, mode, dev); return mknod(pathname, mode, dev);
} }
#endif #endif
...@@ -75,15 +83,15 @@ int do_mknod(char *pathname, mode_t mode, dev_t dev) ...@@ -75,15 +83,15 @@ int do_mknod(char *pathname, mode_t mode, dev_t dev)
int do_rmdir(char *pathname) int do_rmdir(char *pathname)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return rmdir(pathname); return rmdir(pathname);
} }
int do_open(char *pathname, int flags, mode_t mode) int do_open(char *pathname, int flags, mode_t mode)
{ {
if (flags != O_RDONLY) { if (flags != O_RDONLY) {
if (dry_run) return -1; RETURN_ERROR_IF(dry_run, ENOMSG);
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
} }
return open(pathname, flags | O_BINARY, mode); return open(pathname, flags | O_BINARY, mode);
...@@ -94,9 +102,9 @@ int do_chmod(const char *path, mode_t mode) ...@@ -94,9 +102,9 @@ int do_chmod(const char *path, mode_t mode)
{ {
int code; int code;
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
code = chmod(path, mode); code = chmod(path, mode);
if ((code != 0) && preserve_perms) if (code != 0 && preserve_perms)
return code; return code;
return 0; return 0;
} }
...@@ -105,7 +113,7 @@ int do_chmod(const char *path, mode_t mode) ...@@ -105,7 +113,7 @@ int do_chmod(const char *path, mode_t mode)
int do_rename(char *fname1, char *fname2) int do_rename(char *fname1, char *fname2)
{ {
if (dry_run) return 0; if (dry_run) return 0;
CHECK_RO RETURN_ERROR_IF_RO_OR_LO;
return rename(fname1, fname2); return rename(fname1, fname2);
} }
...@@ -131,9 +139,8 @@ void trim_trailing_slashes(char *name) ...@@ -131,9 +139,8 @@ void trim_trailing_slashes(char *name)
int do_mkdir(char *fname, mode_t mode) int do_mkdir(char *fname, mode_t mode)
{ {
if (dry_run) if (dry_run) return 0;
return 0; RETURN_ERROR_IF_RO_OR_LO;
CHECK_RO;
trim_trailing_slashes(fname); trim_trailing_slashes(fname);
return mkdir(fname, mode); return mkdir(fname, mode);
} }
...@@ -142,22 +149,26 @@ int do_mkdir(char *fname, mode_t mode) ...@@ -142,22 +149,26 @@ int do_mkdir(char *fname, mode_t mode)
/* like mkstemp but forces permissions */ /* like mkstemp but forces permissions */
int do_mkstemp(char *template, mode_t perms) int do_mkstemp(char *template, mode_t perms)
{ {
if (dry_run) return -1; RETURN_ERROR_IF(dry_run, ENOMSG);
if (read_only) {errno = EROFS; return -1;} RETURN_ERROR_IF(read_only, EROFS);
#if defined(HAVE_SECURE_MKSTEMP) && defined(HAVE_FCHMOD) #if defined(HAVE_SECURE_MKSTEMP) && defined(HAVE_FCHMOD)
{ {
int fd = mkstemp(template); int fd = mkstemp(template);
if (fd == -1) return -1; if (fd == -1)
if ((fchmod(fd, perms) != 0) && preserve_perms) { return -1;
if (fchmod(fd, perms) != 0 && preserve_perms) {
int errno_save = errno;
close(fd); close(fd);
unlink(template); unlink(template);
errno = errno_save;
return -1; return -1;
} }
return fd; return fd;
} }
#else #else
if (!mktemp(template)) return -1; if (!mktemp(template))
return -1;
return do_open(template, O_RDWR|O_EXCL|O_CREAT, perms); return do_open(template, O_RDWR|O_EXCL|O_CREAT, perms);
#endif #endif
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment