2 * Copyright (c) 2004 Topspin Communications. All rights reserved.
3 * Copyright (c) 2005 Intel Corporation. All rights reserved.
4 * Copyright (c) 2005 Sun Microsystems, Inc. All rights reserved.
5 * Copyright (c) 2005 Voltaire, Inc. All rights reserved.
7 * This software is available to you under a choice of one of two
8 * licenses. You may choose to be licensed under the terms of the GNU
9 * General Public License (GPL) Version 2, available from the file
10 * COPYING in the main directory of this source tree, or the
11 * OpenIB.org BSD license below:
13 * Redistribution and use in source and binary forms, with or
14 * without modification, are permitted provided that the following
17 * - Redistributions of source code must retain the above
18 * copyright notice, this list of conditions and the following
21 * - Redistributions in binary form must reproduce the above
22 * copyright notice, this list of conditions and the following
23 * disclaimer in the documentation and/or other materials
24 * provided with the distribution.
26 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
27 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
28 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
29 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
30 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
31 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
32 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
35 * $Id: cache.c 1349 2004-12-16 21:09:43Z roland $
38 #include <linux/module.h>
39 #include <linux/errno.h>
40 #include <linux/slab.h>
41 #include <linux/sched.h> /* INIT_WORK, schedule_work(), flush_scheduled_work() */
43 #include <rdma/ib_cache.h>
45 #include "core_priv.h"
47 struct ib_pkey_cache {
54 union ib_gid table[0];
57 struct ib_update_work {
58 struct work_struct work;
59 struct ib_device *device;
63 static inline int start_port(struct ib_device *device)
65 return device->node_type == IB_NODE_SWITCH ? 0 : 1;
68 static inline int end_port(struct ib_device *device)
70 return device->node_type == IB_NODE_SWITCH ? 0 : device->phys_port_cnt;
73 int ib_get_cached_gid(struct ib_device *device,
78 struct ib_gid_cache *cache;
82 if (port_num < start_port(device) || port_num > end_port(device))
85 read_lock_irqsave(&device->cache.lock, flags);
87 cache = device->cache.gid_cache[port_num - start_port(device)];
89 if (index < 0 || index >= cache->table_len)
92 *gid = cache->table[index];
94 read_unlock_irqrestore(&device->cache.lock, flags);
98 EXPORT_SYMBOL(ib_get_cached_gid);
100 int ib_find_cached_gid(struct ib_device *device,
105 struct ib_gid_cache *cache;
114 read_lock_irqsave(&device->cache.lock, flags);
116 for (p = 0; p <= end_port(device) - start_port(device); ++p) {
117 cache = device->cache.gid_cache[p];
118 for (i = 0; i < cache->table_len; ++i) {
119 if (!memcmp(gid, &cache->table[i], sizeof *gid)) {
120 *port_num = p + start_port(device);
129 read_unlock_irqrestore(&device->cache.lock, flags);
133 EXPORT_SYMBOL(ib_find_cached_gid);
135 int ib_get_cached_pkey(struct ib_device *device,
140 struct ib_pkey_cache *cache;
144 if (port_num < start_port(device) || port_num > end_port(device))
147 read_lock_irqsave(&device->cache.lock, flags);
149 cache = device->cache.pkey_cache[port_num - start_port(device)];
151 if (index < 0 || index >= cache->table_len)
154 *pkey = cache->table[index];
156 read_unlock_irqrestore(&device->cache.lock, flags);
160 EXPORT_SYMBOL(ib_get_cached_pkey);
162 int ib_find_cached_pkey(struct ib_device *device,
167 struct ib_pkey_cache *cache;
172 if (port_num < start_port(device) || port_num > end_port(device))
175 read_lock_irqsave(&device->cache.lock, flags);
177 cache = device->cache.pkey_cache[port_num - start_port(device)];
181 for (i = 0; i < cache->table_len; ++i)
182 if ((cache->table[i] & 0x7fff) == (pkey & 0x7fff)) {
188 read_unlock_irqrestore(&device->cache.lock, flags);
192 EXPORT_SYMBOL(ib_find_cached_pkey);
194 int ib_get_cached_lmc(struct ib_device *device,
201 if (port_num < start_port(device) || port_num > end_port(device))
204 read_lock_irqsave(&device->cache.lock, flags);
205 *lmc = device->cache.lmc_cache[port_num - start_port(device)];
206 read_unlock_irqrestore(&device->cache.lock, flags);
210 EXPORT_SYMBOL(ib_get_cached_lmc);
212 static void ib_cache_update(struct ib_device *device,
215 struct ib_port_attr *tprops = NULL;
216 struct ib_pkey_cache *pkey_cache = NULL, *old_pkey_cache;
217 struct ib_gid_cache *gid_cache = NULL, *old_gid_cache;
221 tprops = kmalloc(sizeof *tprops, GFP_KERNEL);
225 ret = ib_query_port(device, port, tprops);
227 printk(KERN_WARNING "ib_query_port failed (%d) for %s\n",
232 pkey_cache = kmalloc(sizeof *pkey_cache + tprops->pkey_tbl_len *
233 sizeof *pkey_cache->table, GFP_KERNEL);
237 pkey_cache->table_len = tprops->pkey_tbl_len;
239 gid_cache = kmalloc(sizeof *gid_cache + tprops->gid_tbl_len *
240 sizeof *gid_cache->table, GFP_KERNEL);
244 gid_cache->table_len = tprops->gid_tbl_len;
246 for (i = 0; i < pkey_cache->table_len; ++i) {
247 ret = ib_query_pkey(device, port, i, pkey_cache->table + i);
249 printk(KERN_WARNING "ib_query_pkey failed (%d) for %s (index %d)\n",
250 ret, device->name, i);
255 for (i = 0; i < gid_cache->table_len; ++i) {
256 ret = ib_query_gid(device, port, i, gid_cache->table + i);
258 printk(KERN_WARNING "ib_query_gid failed (%d) for %s (index %d)\n",
259 ret, device->name, i);
264 write_lock_irq(&device->cache.lock);
266 old_pkey_cache = device->cache.pkey_cache[port - start_port(device)];
267 old_gid_cache = device->cache.gid_cache [port - start_port(device)];
269 device->cache.pkey_cache[port - start_port(device)] = pkey_cache;
270 device->cache.gid_cache [port - start_port(device)] = gid_cache;
272 device->cache.lmc_cache[port - start_port(device)] = tprops->lmc;
274 write_unlock_irq(&device->cache.lock);
276 kfree(old_pkey_cache);
277 kfree(old_gid_cache);
287 static void ib_cache_task(void *work_ptr)
289 struct ib_update_work *work = work_ptr;
291 ib_cache_update(work->device, work->port_num);
295 static void ib_cache_event(struct ib_event_handler *handler,
296 struct ib_event *event)
298 struct ib_update_work *work;
300 if (event->event == IB_EVENT_PORT_ERR ||
301 event->event == IB_EVENT_PORT_ACTIVE ||
302 event->event == IB_EVENT_LID_CHANGE ||
303 event->event == IB_EVENT_PKEY_CHANGE ||
304 event->event == IB_EVENT_SM_CHANGE) {
305 work = kmalloc(sizeof *work, GFP_ATOMIC);
307 INIT_WORK(&work->work, ib_cache_task, work);
308 work->device = event->device;
309 work->port_num = event->element.port_num;
310 schedule_work(&work->work);
315 static void ib_cache_setup_one(struct ib_device *device)
319 rwlock_init(&device->cache.lock);
321 device->cache.pkey_cache =
322 kmalloc(sizeof *device->cache.pkey_cache *
323 (end_port(device) - start_port(device) + 1), GFP_KERNEL);
324 device->cache.gid_cache =
325 kmalloc(sizeof *device->cache.gid_cache *
326 (end_port(device) - start_port(device) + 1), GFP_KERNEL);
328 device->cache.lmc_cache = kmalloc(sizeof *device->cache.lmc_cache *
330 start_port(device) + 1),
333 if (!device->cache.pkey_cache || !device->cache.gid_cache ||
334 !device->cache.lmc_cache) {
335 printk(KERN_WARNING "Couldn't allocate cache "
336 "for %s\n", device->name);
340 for (p = 0; p <= end_port(device) - start_port(device); ++p) {
341 device->cache.pkey_cache[p] = NULL;
342 device->cache.gid_cache [p] = NULL;
343 ib_cache_update(device, p + start_port(device));
346 INIT_IB_EVENT_HANDLER(&device->cache.event_handler,
347 device, ib_cache_event);
348 if (ib_register_event_handler(&device->cache.event_handler))
354 for (p = 0; p <= end_port(device) - start_port(device); ++p) {
355 kfree(device->cache.pkey_cache[p]);
356 kfree(device->cache.gid_cache[p]);
360 kfree(device->cache.pkey_cache);
361 kfree(device->cache.gid_cache);
362 kfree(device->cache.lmc_cache);
365 static void ib_cache_cleanup_one(struct ib_device *device)
369 ib_unregister_event_handler(&device->cache.event_handler);
370 flush_scheduled_work();
372 for (p = 0; p <= end_port(device) - start_port(device); ++p) {
373 kfree(device->cache.pkey_cache[p]);
374 kfree(device->cache.gid_cache[p]);
377 kfree(device->cache.pkey_cache);
378 kfree(device->cache.gid_cache);
379 kfree(device->cache.lmc_cache);
382 static struct ib_client cache_client = {
384 .add = ib_cache_setup_one,
385 .remove = ib_cache_cleanup_one
388 int __init ib_cache_setup(void)
390 return ib_register_client(&cache_client);
393 void __exit ib_cache_cleanup(void)
395 ib_unregister_client(&cache_client);