1 files changed, 10 insertions, 5 deletions

diff --git a/drivers/staging/gasket/gasket_interrupt.c b/drivers/staging/gasket/gasket_interrupt.c
index 2d6195f7300e..864342acfd86 100644
--- a/drivers/staging/gasket/gasket_interrupt.c
+++ b/drivers/staging/gasket/gasket_interrupt.c
@@ -487,14 +487,16 @@ int gasket_interrupt_system_status(struct gasket_dev *gasket_dev)
 int gasket_interrupt_set_eventfd(struct gasket_interrupt_data *interrupt_data,
                                  int interrupt, int event_fd)
 {
-        struct eventfd_ctx *ctx = eventfd_ctx_fdget(event_fd);
-
-        if (IS_ERR(ctx))
-                return PTR_ERR(ctx);
+        struct eventfd_ctx *ctx;
 
         if (interrupt < 0 || interrupt >= interrupt_data->num_interrupts)
                 return -EINVAL;
 
+        ctx = eventfd_ctx_fdget(event_fd);
+
+        if (IS_ERR(ctx))
+                return PTR_ERR(ctx);
+
         interrupt_data->eventfd_ctxs[interrupt] = ctx;
         return 0;
 }
@@ -505,6 +507,9 @@ int gasket_interrupt_clear_eventfd(struct gasket_interrupt_data *interrupt_data,
         if (interrupt < 0 || interrupt >= interrupt_data->num_interrupts)
                 return -EINVAL;
 
-        interrupt_data->eventfd_ctxs[interrupt] = NULL;
+        if (interrupt_data->eventfd_ctxs[interrupt]) {
+                eventfd_ctx_put(interrupt_data->eventfd_ctxs[interrupt]);
+                interrupt_data->eventfd_ctxs[interrupt] = NULL;
+        }
         return 0;
 }