daandemeyer / rpms / systemd

Forked from rpms/systemd 2 years ago
Clone
bd1529
From a11334f0eae67b5159a416193e2e37634281000a Mon Sep 17 00:00:00 2001
bd1529
From: Lennart Poettering <lennart@poettering.net>
bd1529
Date: Thu, 8 Nov 2018 09:33:31 +0100
bd1529
Subject: [PATCH] unit: make UNIT() cast function deal with NULL pointers
bd1529
bd1529
Fixes: #10681
bd1529
(cherry picked from commit bbf11206230d1b089118971f98a047151cb5c4fa)
bd1529
bd1529
Related: #1871827
bd1529
---
bd1529
 src/core/unit.h | 7 ++++++-
bd1529
 1 file changed, 6 insertions(+), 1 deletion(-)
bd1529
bd1529
diff --git a/src/core/unit.h b/src/core/unit.h
bd1529
index 6e37fd6f5a..ec45b5fb48 100644
bd1529
--- a/src/core/unit.h
bd1529
+++ b/src/core/unit.h
bd1529
@@ -597,7 +597,12 @@ extern const UnitVTable * const unit_vtable[_UNIT_TYPE_MAX];
bd1529
         }
bd1529
 
bd1529
 /* For casting the various unit types into a unit */
bd1529
-#define UNIT(u) (&(u)->meta)
bd1529
+#define UNIT(u)                                         \
bd1529
+        ({                                              \
bd1529
+                typeof(u) _u_ = (u);                    \
bd1529
+                Unit *_w_ = _u_ ? &(_u_)->meta : NULL;  \
bd1529
+                _w_;                                    \
bd1529
+        })
bd1529
 
bd1529
 #define UNIT_HAS_EXEC_CONTEXT(u) (UNIT_VTABLE(u)->exec_context_offset > 0)
bd1529
 #define UNIT_HAS_CGROUP_CONTEXT(u) (UNIT_VTABLE(u)->cgroup_context_offset > 0)