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