9251c7d0c2094a9a981c63f41f6293b58f12a745
[compat-rdma/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 #if ! defined(RHEL_MINOR) || (RHEL_MINOR < 3)
12 #include <linux/idr.h>
13
14 static DEFINE_SPINLOCK(simple_ida_lock);
15
16 /**
17  * ida_simple_get - get a new id.
18  * @ida: the (initialized) ida.
19  * @start: the minimum id (inclusive, < 0x8000000)
20  * @end: the maximum id (exclusive, < 0x8000000 or 0)
21  * @gfp_mask: memory allocation flags
22  *
23  * Allocates an id in the range start <= id < end, or returns -ENOSPC.
24  * On memory allocation failure, returns -ENOMEM.
25  *
26  * Use ida_simple_remove() to get rid of an id.
27  */
28 int ida_simple_get(struct ida *ida, unsigned int start, unsigned int end,
29                    gfp_t gfp_mask)
30 {
31         int ret, id;
32         unsigned int max;
33         unsigned long flags;
34
35         BUG_ON((int)start < 0);
36         BUG_ON((int)end < 0);
37
38         if (end == 0)
39                 max = 0x80000000;
40         else {
41                 BUG_ON(end < start);
42                 max = end - 1;
43         }
44
45 again:
46         if (!ida_pre_get(ida, gfp_mask))
47                 return -ENOMEM;
48
49         spin_lock_irqsave(&simple_ida_lock, flags);
50         ret = ida_get_new_above(ida, start, &id);
51         if (!ret) {
52                 if (id > max) {
53                         ida_remove(ida, id);
54                         ret = -ENOSPC;
55                 } else {
56                         ret = id;
57                 }
58         }
59         spin_unlock_irqrestore(&simple_ida_lock, flags);
60
61         if (unlikely(ret == -EAGAIN))
62                 goto again;
63
64         return ret;
65 }
66 EXPORT_SYMBOL(ida_simple_get);
67
68 /**
69  * ida_simple_remove - remove an allocated id.
70  * @ida: the (initialized) ida.
71  * @id: the id returned by ida_simple_get.
72  */
73 void ida_simple_remove(struct ida *ida, unsigned int id)
74 {
75         unsigned long flags;
76
77         BUG_ON((int)id < 0);
78         spin_lock_irqsave(&simple_ida_lock, flags);
79         ida_remove(ida, id);
80         spin_unlock_irqrestore(&simple_ida_lock, flags);
81 }
82 EXPORT_SYMBOL(ida_simple_remove);
83 /* source lib/idr.c */
84 #endif
85