1 /* 2 * Copyright 2012 Red Hat Inc. 3 * 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: 10 * 11 * The above copyright notice and this permission notice shall be included in 12 * all copies or substantial portions of the Software. 13 * 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. 21 * 22 * Authors: Ben Skeggs 23 */ 24 #include <core/client.h> 25 #include <core/device.h> 26 #include <core/notify.h> 27 #include <core/option.h> 28 29 #include <nvif/class.h> 30 #include <nvif/event.h> 31 #include <nvif/unpack.h> 32 33 struct nvkm_client_notify { 34 struct nvkm_client *client; 35 struct nvkm_notify n; 36 u8 version; 37 u8 size; 38 union { 39 struct nvif_notify_rep_v0 v0; 40 } rep; 41 }; 42 43 static int 44 nvkm_client_notify(struct nvkm_notify *n) 45 { 46 struct nvkm_client_notify *notify = container_of(n, typeof(*notify), n); 47 struct nvkm_client *client = notify->client; 48 return client->ntfy(¬ify->rep, notify->size, n->data, n->size); 49 } 50 51 int 52 nvkm_client_notify_put(struct nvkm_client *client, int index) 53 { 54 if (index < ARRAY_SIZE(client->notify)) { 55 if (client->notify[index]) { 56 nvkm_notify_put(&client->notify[index]->n); 57 return 0; 58 } 59 } 60 return -ENOENT; 61 } 62 63 int 64 nvkm_client_notify_get(struct nvkm_client *client, int index) 65 { 66 if (index < ARRAY_SIZE(client->notify)) { 67 if (client->notify[index]) { 68 nvkm_notify_get(&client->notify[index]->n); 69 return 0; 70 } 71 } 72 return -ENOENT; 73 } 74 75 int 76 nvkm_client_notify_del(struct nvkm_client *client, int index) 77 { 78 if (index < ARRAY_SIZE(client->notify)) { 79 if (client->notify[index]) { 80 nvkm_notify_fini(&client->notify[index]->n); 81 kfree(client->notify[index]); 82 client->notify[index] = NULL; 83 return 0; 84 } 85 } 86 return -ENOENT; 87 } 88 89 int 90 nvkm_client_notify_new(struct nvkm_object *object, 91 struct nvkm_event *event, void *data, u32 size) 92 { 93 struct nvkm_client *client = object->client; 94 struct nvkm_client_notify *notify; 95 union { 96 struct nvif_notify_req_v0 v0; 97 } *req = data; 98 u8 index, reply; 99 int ret; 100 101 for (index = 0; index < ARRAY_SIZE(client->notify); index++) { 102 if (!client->notify[index]) 103 break; 104 } 105 106 if (index == ARRAY_SIZE(client->notify)) 107 return -ENOSPC; 108 109 notify = kzalloc(sizeof(*notify), GFP_KERNEL); 110 if (!notify) 111 return -ENOMEM; 112 113 nvif_ioctl(object, "notify new size %d\n", size); 114 if (nvif_unpack(req->v0, 0, 0, true)) { 115 nvif_ioctl(object, "notify new vers %d reply %d route %02x " 116 "token %llx\n", req->v0.version, 117 req->v0.reply, req->v0.route, req->v0.token); 118 notify->version = req->v0.version; 119 notify->size = sizeof(notify->rep.v0); 120 notify->rep.v0.version = req->v0.version; 121 notify->rep.v0.route = req->v0.route; 122 notify->rep.v0.token = req->v0.token; 123 reply = req->v0.reply; 124 } 125 126 if (ret == 0) { 127 ret = nvkm_notify_init(object, event, nvkm_client_notify, 128 false, data, size, reply, ¬ify->n); 129 if (ret == 0) { 130 client->notify[index] = notify; 131 notify->client = client; 132 return index; 133 } 134 } 135 136 kfree(notify); 137 return ret; 138 } 139 140 static int 141 nvkm_client_mthd_devlist(struct nvkm_object *object, void *data, u32 size) 142 { 143 union { 144 struct nv_client_devlist_v0 v0; 145 } *args = data; 146 int ret; 147 148 nvif_ioctl(object, "client devlist size %d\n", size); 149 if (nvif_unpack(args->v0, 0, 0, true)) { 150 nvif_ioctl(object, "client devlist vers %d count %d\n", 151 args->v0.version, args->v0.count); 152 if (size == sizeof(args->v0.device[0]) * args->v0.count) { 153 ret = nvkm_device_list(args->v0.device, args->v0.count); 154 if (ret >= 0) { 155 args->v0.count = ret; 156 ret = 0; 157 } 158 } else { 159 ret = -EINVAL; 160 } 161 } 162 163 return ret; 164 } 165 166 static int 167 nvkm_client_mthd(struct nvkm_object *object, u32 mthd, void *data, u32 size) 168 { 169 switch (mthd) { 170 case NV_CLIENT_DEVLIST: 171 return nvkm_client_mthd_devlist(object, data, size); 172 default: 173 break; 174 } 175 return -EINVAL; 176 } 177 178 static int 179 nvkm_client_child_new(const struct nvkm_oclass *oclass, 180 void *data, u32 size, struct nvkm_object **pobject) 181 { 182 return oclass->base.ctor(oclass, data, size, pobject); 183 } 184 185 static int 186 nvkm_client_child_get(struct nvkm_object *object, int index, 187 struct nvkm_oclass *oclass) 188 { 189 const struct nvkm_sclass *sclass; 190 191 switch (index) { 192 case 0: sclass = &nvkm_udevice_sclass; break; 193 default: 194 return -EINVAL; 195 } 196 197 oclass->ctor = nvkm_client_child_new; 198 oclass->base = *sclass; 199 return 0; 200 } 201 202 static const struct nvkm_object_func 203 nvkm_client_object_func = { 204 .mthd = nvkm_client_mthd, 205 .sclass = nvkm_client_child_get, 206 }; 207 208 void 209 nvkm_client_remove(struct nvkm_client *client, struct nvkm_object *object) 210 { 211 if (!RB_EMPTY_NODE(&object->node)) 212 rb_erase(&object->node, &client->objroot); 213 } 214 215 bool 216 nvkm_client_insert(struct nvkm_client *client, struct nvkm_object *object) 217 { 218 struct rb_node **ptr = &client->objroot.rb_node; 219 struct rb_node *parent = NULL; 220 221 while (*ptr) { 222 struct nvkm_object *this = 223 container_of(*ptr, typeof(*this), node); 224 parent = *ptr; 225 if (object->object < this->object) 226 ptr = &parent->rb_left; 227 else 228 if (object->object > this->object) 229 ptr = &parent->rb_right; 230 else 231 return false; 232 } 233 234 rb_link_node(&object->node, parent, ptr); 235 rb_insert_color(&object->node, &client->objroot); 236 return true; 237 } 238 239 struct nvkm_object * 240 nvkm_client_search(struct nvkm_client *client, u64 handle) 241 { 242 struct rb_node *node = client->objroot.rb_node; 243 while (node) { 244 struct nvkm_object *object = 245 container_of(node, typeof(*object), node); 246 if (handle < object->object) 247 node = node->rb_left; 248 else 249 if (handle > object->object) 250 node = node->rb_right; 251 else 252 return object; 253 } 254 return NULL; 255 } 256 257 int 258 nvkm_client_fini(struct nvkm_client *client, bool suspend) 259 { 260 struct nvkm_object *object = &client->object; 261 const char *name[2] = { "fini", "suspend" }; 262 int i; 263 nvif_debug(object, "%s notify\n", name[suspend]); 264 for (i = 0; i < ARRAY_SIZE(client->notify); i++) 265 nvkm_client_notify_put(client, i); 266 return nvkm_object_fini(&client->object, suspend); 267 } 268 269 int 270 nvkm_client_init(struct nvkm_client *client) 271 { 272 return nvkm_object_init(&client->object); 273 } 274 275 void 276 nvkm_client_del(struct nvkm_client **pclient) 277 { 278 struct nvkm_client *client = *pclient; 279 int i; 280 if (client) { 281 nvkm_client_fini(client, false); 282 for (i = 0; i < ARRAY_SIZE(client->notify); i++) 283 nvkm_client_notify_del(client, i); 284 nvkm_object_dtor(&client->object); 285 kfree(*pclient); 286 *pclient = NULL; 287 } 288 } 289 290 int 291 nvkm_client_new(const char *name, u64 device, const char *cfg, 292 const char *dbg, struct nvkm_client **pclient) 293 { 294 struct nvkm_oclass oclass = {}; 295 struct nvkm_client *client; 296 297 if (!(client = *pclient = kzalloc(sizeof(*client), GFP_KERNEL))) 298 return -ENOMEM; 299 oclass.client = client; 300 301 nvkm_object_ctor(&nvkm_client_object_func, &oclass, &client->object); 302 snprintf(client->name, sizeof(client->name), "%s", name); 303 client->device = device; 304 client->debug = nvkm_dbgopt(dbg, "CLIENT"); 305 client->objroot = RB_ROOT; 306 client->dmaroot = RB_ROOT; 307 return 0; 308 } 309