diff options
| -rw-r--r-- | nvif/notify.c | 4 | ||||
| -rw-r--r-- | nvif/object.c | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/nvif/notify.c b/nvif/notify.c index 7e03cdd17..0898c3155 100644 --- a/nvif/notify.c +++ b/nvif/notify.c @@ -237,8 +237,10 @@ nvif_notify_new(struct nvif_object *object, int (*func)(struct nvif_notify *), if (notify) { int ret = nvif_notify_init(object, nvif_notify_del, func, work, type, data, size, reply, notify); - if (ret) + if (ret) { kfree(notify); + notify = NULL; + } *pnotify = notify; return ret; } diff --git a/nvif/object.c b/nvif/object.c index b0c82206e..dd85b56f6 100644 --- a/nvif/object.c +++ b/nvif/object.c @@ -275,8 +275,10 @@ nvif_object_new(struct nvif_object *parent, u32 handle, u32 oclass, if (object) { int ret = nvif_object_init(parent, nvif_object_del, handle, oclass, data, size, object); - if (ret) + if (ret) { kfree(object); + object = NULL; + } *pobject = object; return ret; } |