compat: add ida_simple_{get,remove}
[~emulex/for-vlad/old/compat.git] / compat / compat-3.1.c
1 /*
2  * Copyright 2012    Hauke Mehrtens <hauke@hauke-m.de>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License version 2 as
6  * published by the Free Software Foundation.
7  *
8  * Compatibility file for Linux wireless for kernels 3.1.
9  */
10
11 #include <linux/idr.h>
12
13 static DEFINE_SPINLOCK(simple_ida_lock);
14
15 /**
16  * ida_simple_get - get a new id.
17  * @ida: the (initialized) ida.
18  * @start: the minimum id (inclusive, < 0x8000000)
19  * @end: the maximum id (exclusive, < 0x8000000 or 0)
20  * @gfp_mask: memory allocation flags
21  *
22  * Allocates an id in the range start <= id < end, or returns -ENOSPC.
23  * On memory allocation failure, returns -ENOMEM.
24  *
25  * Use ida_simple_remove() to get rid of an id.
26  */
27 int ida_simple_get(struct ida *ida, unsigned int start, unsigned int end,
28                    gfp_t gfp_mask)
29 {
30         int ret, id;
31         unsigned int max;
32         unsigned long flags;
33
34         BUG_ON((int)start < 0);
35         BUG_ON((int)end < 0);
36
37         if (end == 0)
38                 max = 0x80000000;
39         else {
40                 BUG_ON(end < start);
41                 max = end - 1;
42         }
43
44 again:
45         if (!ida_pre_get(ida, gfp_mask))
46                 return -ENOMEM;
47
48         spin_lock_irqsave(&simple_ida_lock, flags);
49         ret = ida_get_new_above(ida, start, &id);
50         if (!ret) {
51                 if (id > max) {
52                         ida_remove(ida, id);
53                         ret = -ENOSPC;
54                 } else {
55                         ret = id;
56                 }
57         }
58         spin_unlock_irqrestore(&simple_ida_lock, flags);
59
60         if (unlikely(ret == -EAGAIN))
61                 goto again;
62
63         return ret;
64 }
65 EXPORT_SYMBOL(ida_simple_get);
66
67 /**
68  * ida_simple_remove - remove an allocated id.
69  * @ida: the (initialized) ida.
70  * @id: the id returned by ida_simple_get.
71  */
72 void ida_simple_remove(struct ida *ida, unsigned int id)
73 {
74         unsigned long flags;
75
76         BUG_ON((int)id < 0);
77         spin_lock_irqsave(&simple_ida_lock, flags);
78         ida_remove(ida, id);
79         spin_unlock_irqrestore(&simple_ida_lock, flags);
80 }
81 EXPORT_SYMBOL(ida_simple_remove);
82 /* source lib/idr.c */
83