2 * Copyright 2014 Red Hat Inc.
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
22 * Authors: Ben Skeggs <bskeggs@redhat.com>
25 #include <nvif/client.h>
26 #include <nvif/driver.h>
27 #include <nvif/notify.h>
28 #include <nvif/object.h>
29 #include <nvif/ioctl.h>
30 #include <nvif/event.h>
33 nvif_notify_put_(struct nvif_notify *notify)
35 struct nvif_object *object = notify->object;
37 struct nvif_ioctl_v0 ioctl;
38 struct nvif_ioctl_ntfy_put_v0 ntfy;
40 .ioctl.type = NVIF_IOCTL_V0_NTFY_PUT,
41 .ntfy.index = notify->index,
44 if (atomic_inc_return(¬ify->putcnt) != 1)
47 return nvif_object_ioctl(object, &args, sizeof(args), NULL);
51 nvif_notify_put(struct nvif_notify *notify)
53 if (likely(notify->object) &&
54 test_and_clear_bit(NVIF_NOTIFY_USER, ¬ify->flags)) {
55 int ret = nvif_notify_put_(notify);
56 if (test_bit(NVIF_NOTIFY_WORK, ¬ify->flags))
57 flush_work(¬ify->work);
64 nvif_notify_get_(struct nvif_notify *notify)
66 struct nvif_object *object = notify->object;
68 struct nvif_ioctl_v0 ioctl;
69 struct nvif_ioctl_ntfy_get_v0 ntfy;
71 .ioctl.type = NVIF_IOCTL_V0_NTFY_GET,
72 .ntfy.index = notify->index,
75 if (atomic_dec_return(¬ify->putcnt) != 0)
78 return nvif_object_ioctl(object, &args, sizeof(args), NULL);
82 nvif_notify_get(struct nvif_notify *notify)
84 if (likely(notify->object) &&
85 !test_and_set_bit(NVIF_NOTIFY_USER, ¬ify->flags))
86 return nvif_notify_get_(notify);
91 nvif_notify_work(struct work_struct *work)
93 struct nvif_notify *notify = container_of(work, typeof(*notify), work);
94 if (notify->func(notify) == NVIF_NOTIFY_KEEP)
95 nvif_notify_get_(notify);
99 nvif_notify(const void *header, u32 length, const void *data, u32 size)
101 struct nvif_notify *notify = NULL;
103 struct nvif_notify_rep_v0 v0;
105 int ret = NVIF_NOTIFY_DROP;
107 if (length == sizeof(args->v0) && args->v0.version == 0) {
108 if (WARN_ON(args->v0.route))
109 return NVIF_NOTIFY_DROP;
110 notify = (void *)(unsigned long)args->v0.token;
113 if (!WARN_ON(notify == NULL)) {
114 struct nvif_client *client = nvif_client(notify->object);
115 if (!WARN_ON(notify->size != size)) {
116 if (test_bit(NVIF_NOTIFY_WORK, ¬ify->flags)) {
117 atomic_inc(¬ify->putcnt);
118 memcpy((void *)notify->data, data, size);
119 schedule_work(¬ify->work);
120 return NVIF_NOTIFY_DROP;
123 ret = notify->func(notify);
125 if (ret != NVIF_NOTIFY_DROP && client->driver->keep) {
126 atomic_inc(¬ify->putcnt);
127 nvif_notify_get_(notify);
136 nvif_notify_fini(struct nvif_notify *notify)
138 struct nvif_object *object = notify->object;
140 struct nvif_ioctl_v0 ioctl;
141 struct nvif_ioctl_ntfy_del_v0 ntfy;
143 .ioctl.type = NVIF_IOCTL_V0_NTFY_DEL,
144 .ntfy.index = notify->index,
146 int ret = nvif_notify_put(notify);
147 if (ret >= 0 && object) {
148 ret = nvif_object_ioctl(object, &args, sizeof(args), NULL);
150 nvif_object_ref(NULL, ¬ify->object);
151 kfree((void *)notify->data);
158 nvif_notify_init(struct nvif_object *object, void (*dtor)(struct nvif_notify *),
159 int (*func)(struct nvif_notify *), bool work, u8 event,
160 void *data, u32 size, u32 reply, struct nvif_notify *notify)
163 struct nvif_ioctl_v0 ioctl;
164 struct nvif_ioctl_ntfy_new_v0 ntfy;
165 struct nvif_notify_req_v0 req;
169 notify->object = NULL;
170 nvif_object_ref(object, ¬ify->object);
172 atomic_set(¬ify->putcnt, 1);
176 notify->size = reply;
178 INIT_WORK(¬ify->work, nvif_notify_work);
179 set_bit(NVIF_NOTIFY_WORK, ¬ify->flags);
180 notify->data = kmalloc(notify->size, GFP_KERNEL);
185 if (!(args = kmalloc(sizeof(*args) + size, GFP_KERNEL)))
187 args->ioctl.version = 0;
188 args->ioctl.type = NVIF_IOCTL_V0_NTFY_NEW;
189 args->ntfy.version = 0;
190 args->ntfy.event = event;
191 args->req.version = 0;
192 args->req.reply = notify->size;
194 args->req.token = (unsigned long)(void *)notify;
196 memcpy(args->req.data, data, size);
197 ret = nvif_object_ioctl(object, args, sizeof(*args) + size, NULL);
198 notify->index = args->ntfy.index;
202 nvif_notify_fini(notify);
207 nvif_notify_del(struct nvif_notify *notify)
209 nvif_notify_fini(notify);
214 nvif_notify_ref(struct nvif_notify *notify, struct nvif_notify **pnotify)
216 BUG_ON(notify != NULL);
218 (*pnotify)->dtor(*pnotify);
223 nvif_notify_new(struct nvif_object *object, int (*func)(struct nvif_notify *),
224 bool work, u8 type, void *data, u32 size, u32 reply,
225 struct nvif_notify **pnotify)
227 struct nvif_notify *notify = kzalloc(sizeof(*notify), GFP_KERNEL);
229 int ret = nvif_notify_init(object, nvif_notify_del, func, work,
230 type, data, size, reply, notify);