driver.c 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244
  1. /*
  2. * driver.c - driver support
  3. *
  4. * (C) 2006-2007 Venkatesh Pallipadi <venkatesh.pallipadi@intel.com>
  5. * Shaohua Li <shaohua.li@intel.com>
  6. * Adam Belay <abelay@novell.com>
  7. *
  8. * This code is licenced under the GPL.
  9. */
  10. #include <linux/mutex.h>
  11. #include <linux/module.h>
  12. #include <linux/cpuidle.h>
  13. #include "cpuidle.h"
  14. DEFINE_SPINLOCK(cpuidle_driver_lock);
  15. static void __cpuidle_set_cpu_driver(struct cpuidle_driver *drv, int cpu);
  16. static struct cpuidle_driver * __cpuidle_get_cpu_driver(int cpu);
  17. static void __cpuidle_driver_init(struct cpuidle_driver *drv)
  18. {
  19. drv->refcnt = 0;
  20. }
  21. static int __cpuidle_register_driver(struct cpuidle_driver *drv, int cpu)
  22. {
  23. if (!drv || !drv->state_count)
  24. return -EINVAL;
  25. if (cpuidle_disabled())
  26. return -ENODEV;
  27. if (__cpuidle_get_cpu_driver(cpu))
  28. return -EBUSY;
  29. __cpuidle_driver_init(drv);
  30. __cpuidle_set_cpu_driver(drv, cpu);
  31. return 0;
  32. }
  33. static void __cpuidle_unregister_driver(struct cpuidle_driver *drv, int cpu)
  34. {
  35. if (drv != __cpuidle_get_cpu_driver(cpu))
  36. return;
  37. if (!WARN_ON(drv->refcnt > 0))
  38. __cpuidle_set_cpu_driver(NULL, cpu);
  39. }
  40. #ifdef CONFIG_CPU_IDLE_MULTIPLE_DRIVERS
  41. static DEFINE_PER_CPU(struct cpuidle_driver *, cpuidle_drivers);
  42. static void __cpuidle_set_cpu_driver(struct cpuidle_driver *drv, int cpu)
  43. {
  44. per_cpu(cpuidle_drivers, cpu) = drv;
  45. }
  46. static struct cpuidle_driver *__cpuidle_get_cpu_driver(int cpu)
  47. {
  48. return per_cpu(cpuidle_drivers, cpu);
  49. }
  50. static void __cpuidle_unregister_all_cpu_driver(struct cpuidle_driver *drv)
  51. {
  52. int cpu;
  53. for_each_present_cpu(cpu)
  54. __cpuidle_unregister_driver(drv, cpu);
  55. }
  56. static int __cpuidle_register_all_cpu_driver(struct cpuidle_driver *drv)
  57. {
  58. int ret = 0;
  59. int i, cpu;
  60. for_each_present_cpu(cpu) {
  61. ret = __cpuidle_register_driver(drv, cpu);
  62. if (ret)
  63. break;
  64. }
  65. if (ret)
  66. for_each_present_cpu(i) {
  67. if (i == cpu)
  68. break;
  69. __cpuidle_unregister_driver(drv, i);
  70. }
  71. return ret;
  72. }
  73. int cpuidle_register_cpu_driver(struct cpuidle_driver *drv, int cpu)
  74. {
  75. int ret;
  76. spin_lock(&cpuidle_driver_lock);
  77. ret = __cpuidle_register_driver(drv, cpu);
  78. spin_unlock(&cpuidle_driver_lock);
  79. return ret;
  80. }
  81. void cpuidle_unregister_cpu_driver(struct cpuidle_driver *drv, int cpu)
  82. {
  83. spin_lock(&cpuidle_driver_lock);
  84. __cpuidle_unregister_driver(drv, cpu);
  85. spin_unlock(&cpuidle_driver_lock);
  86. }
  87. /**
  88. * cpuidle_register_driver - registers a driver
  89. * @drv: the driver
  90. */
  91. int cpuidle_register_driver(struct cpuidle_driver *drv)
  92. {
  93. int ret;
  94. spin_lock(&cpuidle_driver_lock);
  95. ret = __cpuidle_register_all_cpu_driver(drv);
  96. spin_unlock(&cpuidle_driver_lock);
  97. return ret;
  98. }
  99. EXPORT_SYMBOL_GPL(cpuidle_register_driver);
  100. /**
  101. * cpuidle_unregister_driver - unregisters a driver
  102. * @drv: the driver
  103. */
  104. void cpuidle_unregister_driver(struct cpuidle_driver *drv)
  105. {
  106. spin_lock(&cpuidle_driver_lock);
  107. __cpuidle_unregister_all_cpu_driver(drv);
  108. spin_unlock(&cpuidle_driver_lock);
  109. }
  110. EXPORT_SYMBOL_GPL(cpuidle_unregister_driver);
  111. #else
  112. static struct cpuidle_driver *cpuidle_curr_driver;
  113. static inline void __cpuidle_set_cpu_driver(struct cpuidle_driver *drv, int cpu)
  114. {
  115. cpuidle_curr_driver = drv;
  116. }
  117. static inline struct cpuidle_driver *__cpuidle_get_cpu_driver(int cpu)
  118. {
  119. return cpuidle_curr_driver;
  120. }
  121. /**
  122. * cpuidle_register_driver - registers a driver
  123. * @drv: the driver
  124. */
  125. int cpuidle_register_driver(struct cpuidle_driver *drv)
  126. {
  127. int ret, cpu;
  128. cpu = get_cpu();
  129. spin_lock(&cpuidle_driver_lock);
  130. ret = __cpuidle_register_driver(drv, cpu);
  131. spin_unlock(&cpuidle_driver_lock);
  132. put_cpu();
  133. return ret;
  134. }
  135. EXPORT_SYMBOL_GPL(cpuidle_register_driver);
  136. /**
  137. * cpuidle_unregister_driver - unregisters a driver
  138. * @drv: the driver
  139. */
  140. void cpuidle_unregister_driver(struct cpuidle_driver *drv)
  141. {
  142. int cpu;
  143. cpu = get_cpu();
  144. spin_lock(&cpuidle_driver_lock);
  145. __cpuidle_unregister_driver(drv, cpu);
  146. spin_unlock(&cpuidle_driver_lock);
  147. put_cpu();
  148. }
  149. EXPORT_SYMBOL_GPL(cpuidle_unregister_driver);
  150. #endif
  151. /**
  152. * cpuidle_get_driver - return the current driver
  153. */
  154. struct cpuidle_driver *cpuidle_get_driver(void)
  155. {
  156. struct cpuidle_driver *drv;
  157. int cpu;
  158. cpu = get_cpu();
  159. drv = __cpuidle_get_cpu_driver(cpu);
  160. put_cpu();
  161. return drv;
  162. }
  163. EXPORT_SYMBOL_GPL(cpuidle_get_driver);
  164. /**
  165. * cpuidle_get_cpu_driver - return the driver tied with a cpu
  166. */
  167. struct cpuidle_driver *cpuidle_get_cpu_driver(struct cpuidle_device *dev)
  168. {
  169. if (!dev)
  170. return NULL;
  171. return __cpuidle_get_cpu_driver(dev->cpu);
  172. }
  173. EXPORT_SYMBOL_GPL(cpuidle_get_cpu_driver);
  174. struct cpuidle_driver *cpuidle_driver_ref(void)
  175. {
  176. struct cpuidle_driver *drv;
  177. spin_lock(&cpuidle_driver_lock);
  178. drv = cpuidle_get_driver();
  179. drv->refcnt++;
  180. spin_unlock(&cpuidle_driver_lock);
  181. return drv;
  182. }
  183. void cpuidle_driver_unref(void)
  184. {
  185. struct cpuidle_driver *drv = cpuidle_get_driver();
  186. spin_lock(&cpuidle_driver_lock);
  187. if (drv && !WARN_ON(drv->refcnt <= 0))
  188. drv->refcnt--;
  189. spin_unlock(&cpuidle_driver_lock);
  190. }