4b8c80
From 8e155b3dad29a96cd2586185c63604c42bc5690e Mon Sep 17 00:00:00 2001
4b8c80
From: =?UTF-8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= <zbyszek@in.waw.pl>
4b8c80
Date: Wed, 8 May 2019 14:39:57 +0200
4b8c80
Subject: [PATCH] sd-event: add sd_event_source_disable_unref() helper
4b8c80
4b8c80
(cherry picked from commit afd15bbb4b6414b9356799c63029e36642dae8e4)
4b8c80
Related: CVE-2020-1712
4b8c80
---
4b8c80
 man/rules/meson.build              |  4 +++-
4b8c80
 man/sd_event_source_unref.xml      | 30 +++++++++++++++++++-----------
4b8c80
 src/libsystemd/libsystemd.sym      |  1 +
4b8c80
 src/libsystemd/sd-event/sd-event.c |  6 ++++++
4b8c80
 src/systemd/sd-event.h             |  1 +
4b8c80
 5 files changed, 30 insertions(+), 12 deletions(-)
4b8c80
4b8c80
diff --git a/man/rules/meson.build b/man/rules/meson.build
4b8c80
index 989d11c9b9..7ae94ea265 100644
4b8c80
--- a/man/rules/meson.build
4b8c80
+++ b/man/rules/meson.build
4b8c80
@@ -340,7 +340,9 @@ manpages = [
4b8c80
  ['sd_event_source_set_userdata', '3', ['sd_event_source_get_userdata'], ''],
4b8c80
  ['sd_event_source_unref',
4b8c80
   '3',
4b8c80
-  ['sd_event_source_ref', 'sd_event_source_unrefp'],
4b8c80
+  ['sd_event_source_disable_unref',
4b8c80
+   'sd_event_source_ref',
4b8c80
+   'sd_event_source_unrefp'],
4b8c80
   ''],
4b8c80
  ['sd_event_wait',
4b8c80
   '3',
4b8c80
diff --git a/man/sd_event_source_unref.xml b/man/sd_event_source_unref.xml
4b8c80
index d1b83c57aa..af8fed33f2 100644
4b8c80
--- a/man/sd_event_source_unref.xml
4b8c80
+++ b/man/sd_event_source_unref.xml
4b8c80
@@ -22,6 +22,7 @@
4b8c80
     <refname>sd_event_source_unref</refname>
4b8c80
     <refname>sd_event_source_unrefp</refname>
4b8c80
     <refname>sd_event_source_ref</refname>
4b8c80
+    <refname>sd_event_source_disable_unref</refname>
4b8c80
 
4b8c80
     <refpurpose>Increase or decrease event source reference counters</refpurpose>
4b8c80
   </refnamediv>
4b8c80
@@ -45,6 +46,10 @@
4b8c80
         <paramdef>sd_event_source *<parameter>source</parameter></paramdef>
4b8c80
       </funcprototype>
4b8c80
 
4b8c80
+      <funcprototype>
4b8c80
+        <funcdef>sd_event_source* <function>sd_event_source_disable_unref</function></funcdef>
4b8c80
+        <paramdef>sd_event_source *<parameter>source</parameter></paramdef>
4b8c80
+      </funcprototype>
4b8c80
     </funcsynopsis>
4b8c80
   </refsynopsisdiv>
4b8c80
 
4b8c80
@@ -80,23 +85,26 @@
4b8c80
     the passed event source object is
4b8c80
     <constant>NULL</constant>.</para>
4b8c80
 
4b8c80
-    <para>Note that event source objects stay alive and may be
4b8c80
-    dispatched as long as they have a reference counter greater than
4b8c80
-    zero. In order to drop a reference of an event source and make
4b8c80
-    sure the associated event source handler function is not called
4b8c80
-    anymore it is recommended to combine a call of
4b8c80
+    <para>Note that event source objects stay alive and may be dispatched as long as they have a reference
4b8c80
+    counter greater than zero. In order to drop a reference of an event source and make sure the associated
4b8c80
+    event source handler function is not called anymore it is recommended to combine a call of
4b8c80
     <function>sd_event_source_unref()</function> with a prior call to
4b8c80
-    <function>sd_event_source_set_enabled()</function> with
4b8c80
-    <constant>SD_EVENT_OFF</constant>.</para>
4b8c80
+    <function>sd_event_source_set_enabled()</function> with <constant>SD_EVENT_OFF</constant> or call
4b8c80
+    <function>sd_event_source_disable_unref()</function>, see below.</para>
4b8c80
+
4b8c80
+    <para><function>sd_event_source_disable_unref()</function> combines a call to
4b8c80
+    <function>sd_event_source_set_enabled()</function> with <constant>SD_EVENT_OFF</constant> with
4b8c80
+    <function>sd_event_source_unref()</function>. This ensures that the source is disabled before the local
4b8c80
+    reference to it is lost. The <parameter>source</parameter> parameter is allowed to be
4b8c80
+    <constant>NULL</constant>.</para>
4b8c80
   </refsect1>
4b8c80
 
4b8c80
   <refsect1>
4b8c80
     <title>Return Value</title>
4b8c80
 
4b8c80
-    <para><function>sd_event_source_unref()</function> always returns
4b8c80
-    <constant>NULL</constant>.
4b8c80
-    <function>sd_event_source_ref()</function> always returns the
4b8c80
-    event source object passed in.</para>
4b8c80
+    <para><function>sd_event_source_unref()</function> and
4b8c80
+    <function>sd_event_source_disable_unref()</function> always return <constant>NULL</constant>.
4b8c80
+    <function>sd_event_source_ref()</function> always returns the event source object passed in.</para>
4b8c80
   </refsect1>
4b8c80
 
4b8c80
   <xi:include href="libsystemd-pkgconfig.xml" />
4b8c80
diff --git a/src/libsystemd/libsystemd.sym b/src/libsystemd/libsystemd.sym
4b8c80
index e9972593a6..778e88a16c 100644
4b8c80
--- a/src/libsystemd/libsystemd.sym
4b8c80
+++ b/src/libsystemd/libsystemd.sym
4b8c80
@@ -570,4 +570,5 @@ global:
4b8c80
         sd_event_source_set_destroy_callback;
4b8c80
         sd_event_source_get_destroy_callback;
4b8c80
         sd_bus_enqueue_for_read;
4b8c80
+        sd_event_source_disable_unref;
4b8c80
 } LIBSYSTEMD_238;
4b8c80
diff --git a/src/libsystemd/sd-event/sd-event.c b/src/libsystemd/sd-event/sd-event.c
4b8c80
index d53b9a7026..0d3bf5cbb6 100644
4b8c80
--- a/src/libsystemd/sd-event/sd-event.c
4b8c80
+++ b/src/libsystemd/sd-event/sd-event.c
4b8c80
@@ -580,6 +580,12 @@ _public_ sd_event* sd_event_unref(sd_event *e) {
4b8c80
         return NULL;
4b8c80
 }
4b8c80
 
4b8c80
+_public_ sd_event_source* sd_event_source_disable_unref(sd_event_source *s) {
4b8c80
+        if (s)
4b8c80
+                (void) sd_event_source_set_enabled(s, SD_EVENT_OFF);
4b8c80
+        return sd_event_source_unref(s);
4b8c80
+}
4b8c80
+
4b8c80
 static bool event_pid_changed(sd_event *e) {
4b8c80
         assert(e);
4b8c80
 
4b8c80
diff --git a/src/systemd/sd-event.h b/src/systemd/sd-event.h
4b8c80
index 7fcae4ac49..9876be01c6 100644
4b8c80
--- a/src/systemd/sd-event.h
4b8c80
+++ b/src/systemd/sd-event.h
4b8c80
@@ -113,6 +113,7 @@ int sd_event_get_iteration(sd_event *e, uint64_t *ret);
4b8c80
 
4b8c80
 sd_event_source* sd_event_source_ref(sd_event_source *s);
4b8c80
 sd_event_source* sd_event_source_unref(sd_event_source *s);
4b8c80
+sd_event_source* sd_event_source_disable_unref(sd_event_source *s);
4b8c80
 
4b8c80
 sd_event *sd_event_source_get_event(sd_event_source *s);
4b8c80
 void* sd_event_source_get_userdata(sd_event_source *s);