ref.h 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. /*
  2. * net/tipc/ref.h: Include file for TIPC object registry code
  3. *
  4. * Copyright (c) 2003-2005, Ericsson Research Canada
  5. * Copyright (c) 2005, Wind River Systems
  6. * Copyright (c) 2005-2006, Ericsson AB
  7. * All rights reserved.
  8. *
  9. * Redistribution and use in source and binary forms, with or without
  10. * modification, are permitted provided that the following conditions are met:
  11. *
  12. * Redistributions of source code must retain the above copyright notice, this
  13. * list of conditions and the following disclaimer.
  14. * Redistributions in binary form must reproduce the above copyright notice,
  15. * this list of conditions and the following disclaimer in the documentation
  16. * and/or other materials provided with the distribution.
  17. * Neither the names of the copyright holders nor the names of its
  18. * contributors may be used to endorse or promote products derived from this
  19. * software without specific prior written permission.
  20. *
  21. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  22. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  25. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  26. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  27. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  28. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  29. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  31. * POSSIBILITY OF SUCH DAMAGE.
  32. */
  33. #ifndef _TIPC_REF_H
  34. #define _TIPC_REF_H
  35. /**
  36. * struct reference - TIPC object reference entry
  37. * @object: pointer to object associated with reference entry
  38. * @lock: spinlock controlling access to object
  39. * @data: reference value associated with object (or link to next unused entry)
  40. */
  41. struct reference {
  42. void *object;
  43. spinlock_t lock;
  44. union {
  45. u32 next_plus_upper;
  46. u32 reference;
  47. } data;
  48. };
  49. /**
  50. * struct ref_table - table of TIPC object reference entries
  51. * @entries: pointer to array of reference entries
  52. * @index_mask: bitmask for array index portion of reference values
  53. * @first_free: array index of first unused object reference entry
  54. * @last_free: array index of last unused object reference entry
  55. */
  56. struct ref_table {
  57. struct reference *entries;
  58. u32 index_mask;
  59. u32 first_free;
  60. u32 last_free;
  61. };
  62. extern struct ref_table ref_table;
  63. int ref_table_init(u32 requested_size, u32 start);
  64. void ref_table_stop(void);
  65. u32 ref_acquire(void *object, spinlock_t **lock);
  66. void ref_discard(u32 ref);
  67. /**
  68. * ref_lock - lock referenced object and return pointer to it
  69. */
  70. static inline void *ref_lock(u32 ref)
  71. {
  72. if (likely(ref_table.entries)) {
  73. struct reference *r =
  74. &ref_table.entries[ref & ref_table.index_mask];
  75. spin_lock_bh(&r->lock);
  76. if (likely(r->data.reference == ref))
  77. return r->object;
  78. spin_unlock_bh(&r->lock);
  79. }
  80. return 0;
  81. }
  82. /**
  83. * ref_unlock - unlock referenced object
  84. */
  85. static inline void ref_unlock(u32 ref)
  86. {
  87. if (likely(ref_table.entries)) {
  88. struct reference *r =
  89. &ref_table.entries[ref & ref_table.index_mask];
  90. if (likely(r->data.reference == ref))
  91. spin_unlock_bh(&r->lock);
  92. else
  93. err("ref_unlock() invoked using obsolete reference\n");
  94. }
  95. }
  96. /**
  97. * ref_deref - return pointer referenced object (without locking it)
  98. */
  99. static inline void *ref_deref(u32 ref)
  100. {
  101. if (likely(ref_table.entries)) {
  102. struct reference *r =
  103. &ref_table.entries[ref & ref_table.index_mask];
  104. if (likely(r->data.reference == ref))
  105. return r->object;
  106. }
  107. return 0;
  108. }
  109. #endif