connection: Clear fds we shouldn't close to -1
This initializes all the fd arguments in closures to -1 and clears
them back to -1 when they've been dispatched or serialized.
This means that any valid fd in a closure is currently libwayland's
responsibility to close in the case of an error.
Signed-off-by: Derek Foreman <derekf@osg.samsung.com>
Reviewed-by: Daniel Stone <daniels@collabora.com>
diff --git a/src/connection.c b/src/connection.c
index 8d8eb60..29f565b 100644
--- a/src/connection.c
+++ b/src/connection.c
@@ -524,6 +524,17 @@
}
}
+static void
+wl_closure_clear_fds(struct wl_closure *closure)
+{
+ int i;
+
+ for (i = 0; closure->message->signature[i]; i++) {
+ if (closure->message->signature[i] == 'h')
+ closure->args[i].h = -1;
+ }
+}
+
static struct wl_closure *
wl_closure_init(const struct wl_message *message, uint32_t size,
int *num_arrays, union wl_argument *args)
@@ -557,6 +568,14 @@
closure->message = message;
closure->count = count;
+ /* Set these all to -1 so we can close any that have been
+ * set to a real value during wl_closure_destroy().
+ * We may have copied a bunch of fds into the closure with
+ * memcpy previously, but those are undup()d client fds
+ * that we would have replaced anyway.
+ */
+ wl_closure_clear_fds(closure);
+
return closure;
}
@@ -948,6 +967,8 @@
opcode, target->interface->name);
}
ffi_call(&cif, implementation[opcode], NULL, ffi_args);
+
+ wl_closure_clear_fds(closure);
}
void
@@ -956,6 +977,8 @@
{
dispatcher(target->implementation, target, opcode, closure->message,
closure->args);
+
+ wl_closure_clear_fds(closure);
}
static int
@@ -980,6 +1003,7 @@
"can't send file descriptor");
return -1;
}
+ closure->args[i].h = -1;
}
return 0;