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