util: add a bit of syntactic sugar to run short code fragments with a different umask

[elogind.git] / src / shared / util.h

diff --git a/src/shared/util.h b/src/shared/util.h
index 7f050758a0088b9ebdb87d11c91557b1b5b5abde..69a47653aa4507023e45e5694b0a1b63c2589b4d 100644 (file)
--- a/src/shared/util.h
+++ b/src/shared/util.h
@@ -616,8 +616,22 @@ char *strrep(const char *s, unsigned n);
 
 void* greedy_realloc(void **p, size_t *allocated, size_t need);
 
 
 void* greedy_realloc(void **p, size_t *allocated, size_t need);
 

-static inline void reset_errno(int *saved_errno) {
+static inline void _reset_errno_(int *saved_errno) {

         errno = *saved_errno;
 }
 
         errno = *saved_errno;
 }
 

-#define PROTECT_ERRNO __attribute__((cleanup(reset_errno))) int _saved_errno_ = errno
+#define PROTECT_ERRNO __attribute__((cleanup(_reset_errno_))) int _saved_errno_ = errno
+
+struct umask_struct {
+        mode_t mask;
+        bool quit;
+};
+
+static inline void _reset_umask_(struct umask_struct *s) {
+        umask(s->mask);
+};
+
+#define RUN_WITH_UMASK(mask)                                            \
+        for (__attribute__((cleanup(_reset_umask_))) struct umask_struct _saved_umask_ = { umask(mask), false }; \
+             !_saved_umask_.quit ;                                      \
+             _saved_umask_.quit = true)