pxrc.c 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * Driver for Phoenix RC Flight Controller Adapter
  4. *
  5. * Copyright (C) 2018 Marcus Folkesson <marcus.folkesson@gmail.com>
  6. */
  7. #include <linux/kernel.h>
  8. #include <linux/errno.h>
  9. #include <linux/slab.h>
  10. #include <linux/module.h>
  11. #include <linux/uaccess.h>
  12. #include <linux/usb.h>
  13. #include <linux/usb/input.h>
  14. #include <linux/mutex.h>
  15. #include <linux/input.h>
  16. #define PXRC_VENDOR_ID 0x1781
  17. #define PXRC_PRODUCT_ID 0x0898
  18. struct pxrc {
  19. struct input_dev *input;
  20. struct usb_interface *intf;
  21. struct urb *urb;
  22. struct mutex pm_mutex;
  23. bool is_open;
  24. char phys[64];
  25. };
  26. static void pxrc_usb_irq(struct urb *urb)
  27. {
  28. struct pxrc *pxrc = urb->context;
  29. u8 *data = urb->transfer_buffer;
  30. int error;
  31. switch (urb->status) {
  32. case 0:
  33. /* success */
  34. break;
  35. case -ETIME:
  36. /* this urb is timing out */
  37. dev_dbg(&pxrc->intf->dev,
  38. "%s - urb timed out - was the device unplugged?\n",
  39. __func__);
  40. return;
  41. case -ECONNRESET:
  42. case -ENOENT:
  43. case -ESHUTDOWN:
  44. case -EPIPE:
  45. /* this urb is terminated, clean up */
  46. dev_dbg(&pxrc->intf->dev, "%s - urb shutting down with status: %d\n",
  47. __func__, urb->status);
  48. return;
  49. default:
  50. dev_dbg(&pxrc->intf->dev, "%s - nonzero urb status received: %d\n",
  51. __func__, urb->status);
  52. goto exit;
  53. }
  54. if (urb->actual_length == 8) {
  55. input_report_abs(pxrc->input, ABS_X, data[0]);
  56. input_report_abs(pxrc->input, ABS_Y, data[2]);
  57. input_report_abs(pxrc->input, ABS_RX, data[3]);
  58. input_report_abs(pxrc->input, ABS_RY, data[4]);
  59. input_report_abs(pxrc->input, ABS_RUDDER, data[5]);
  60. input_report_abs(pxrc->input, ABS_THROTTLE, data[6]);
  61. input_report_abs(pxrc->input, ABS_MISC, data[7]);
  62. input_report_key(pxrc->input, BTN_A, data[1]);
  63. }
  64. exit:
  65. /* Resubmit to fetch new fresh URBs */
  66. error = usb_submit_urb(urb, GFP_ATOMIC);
  67. if (error && error != -EPERM)
  68. dev_err(&pxrc->intf->dev,
  69. "%s - usb_submit_urb failed with result: %d",
  70. __func__, error);
  71. }
  72. static int pxrc_open(struct input_dev *input)
  73. {
  74. struct pxrc *pxrc = input_get_drvdata(input);
  75. int retval;
  76. mutex_lock(&pxrc->pm_mutex);
  77. retval = usb_submit_urb(pxrc->urb, GFP_KERNEL);
  78. if (retval) {
  79. dev_err(&pxrc->intf->dev,
  80. "%s - usb_submit_urb failed, error: %d\n",
  81. __func__, retval);
  82. retval = -EIO;
  83. goto out;
  84. }
  85. pxrc->is_open = true;
  86. out:
  87. mutex_unlock(&pxrc->pm_mutex);
  88. return retval;
  89. }
  90. static void pxrc_close(struct input_dev *input)
  91. {
  92. struct pxrc *pxrc = input_get_drvdata(input);
  93. mutex_lock(&pxrc->pm_mutex);
  94. usb_kill_urb(pxrc->urb);
  95. pxrc->is_open = false;
  96. mutex_unlock(&pxrc->pm_mutex);
  97. }
  98. static void pxrc_free_urb(void *_pxrc)
  99. {
  100. struct pxrc *pxrc = _pxrc;
  101. usb_free_urb(pxrc->urb);
  102. }
  103. static int pxrc_probe(struct usb_interface *intf,
  104. const struct usb_device_id *id)
  105. {
  106. struct usb_device *udev = interface_to_usbdev(intf);
  107. struct pxrc *pxrc;
  108. struct usb_endpoint_descriptor *epirq;
  109. size_t xfer_size;
  110. void *xfer_buf;
  111. int error;
  112. /*
  113. * Locate the endpoint information. This device only has an
  114. * interrupt endpoint.
  115. */
  116. error = usb_find_common_endpoints(intf->cur_altsetting,
  117. NULL, NULL, &epirq, NULL);
  118. if (error) {
  119. dev_err(&intf->dev, "Could not find endpoint\n");
  120. return error;
  121. }
  122. pxrc = devm_kzalloc(&intf->dev, sizeof(*pxrc), GFP_KERNEL);
  123. if (!pxrc)
  124. return -ENOMEM;
  125. mutex_init(&pxrc->pm_mutex);
  126. pxrc->intf = intf;
  127. usb_set_intfdata(pxrc->intf, pxrc);
  128. xfer_size = usb_endpoint_maxp(epirq);
  129. xfer_buf = devm_kmalloc(&intf->dev, xfer_size, GFP_KERNEL);
  130. if (!xfer_buf)
  131. return -ENOMEM;
  132. pxrc->urb = usb_alloc_urb(0, GFP_KERNEL);
  133. if (!pxrc->urb)
  134. return -ENOMEM;
  135. error = devm_add_action_or_reset(&intf->dev, pxrc_free_urb, pxrc);
  136. if (error)
  137. return error;
  138. usb_fill_int_urb(pxrc->urb, udev,
  139. usb_rcvintpipe(udev, epirq->bEndpointAddress),
  140. xfer_buf, xfer_size, pxrc_usb_irq, pxrc, 1);
  141. pxrc->input = devm_input_allocate_device(&intf->dev);
  142. if (!pxrc->input) {
  143. dev_err(&intf->dev, "couldn't allocate input device\n");
  144. return -ENOMEM;
  145. }
  146. pxrc->input->name = "PXRC Flight Controller Adapter";
  147. usb_make_path(udev, pxrc->phys, sizeof(pxrc->phys));
  148. strlcat(pxrc->phys, "/input0", sizeof(pxrc->phys));
  149. pxrc->input->phys = pxrc->phys;
  150. usb_to_input_id(udev, &pxrc->input->id);
  151. pxrc->input->open = pxrc_open;
  152. pxrc->input->close = pxrc_close;
  153. input_set_capability(pxrc->input, EV_KEY, BTN_A);
  154. input_set_abs_params(pxrc->input, ABS_X, 0, 255, 0, 0);
  155. input_set_abs_params(pxrc->input, ABS_Y, 0, 255, 0, 0);
  156. input_set_abs_params(pxrc->input, ABS_RX, 0, 255, 0, 0);
  157. input_set_abs_params(pxrc->input, ABS_RY, 0, 255, 0, 0);
  158. input_set_abs_params(pxrc->input, ABS_RUDDER, 0, 255, 0, 0);
  159. input_set_abs_params(pxrc->input, ABS_THROTTLE, 0, 255, 0, 0);
  160. input_set_abs_params(pxrc->input, ABS_MISC, 0, 255, 0, 0);
  161. input_set_drvdata(pxrc->input, pxrc);
  162. error = input_register_device(pxrc->input);
  163. if (error)
  164. return error;
  165. return 0;
  166. }
  167. static void pxrc_disconnect(struct usb_interface *intf)
  168. {
  169. /* All driver resources are devm-managed. */
  170. }
  171. static int pxrc_suspend(struct usb_interface *intf, pm_message_t message)
  172. {
  173. struct pxrc *pxrc = usb_get_intfdata(intf);
  174. mutex_lock(&pxrc->pm_mutex);
  175. if (pxrc->is_open)
  176. usb_kill_urb(pxrc->urb);
  177. mutex_unlock(&pxrc->pm_mutex);
  178. return 0;
  179. }
  180. static int pxrc_resume(struct usb_interface *intf)
  181. {
  182. struct pxrc *pxrc = usb_get_intfdata(intf);
  183. int retval = 0;
  184. mutex_lock(&pxrc->pm_mutex);
  185. if (pxrc->is_open && usb_submit_urb(pxrc->urb, GFP_KERNEL) < 0)
  186. retval = -EIO;
  187. mutex_unlock(&pxrc->pm_mutex);
  188. return retval;
  189. }
  190. static int pxrc_pre_reset(struct usb_interface *intf)
  191. {
  192. struct pxrc *pxrc = usb_get_intfdata(intf);
  193. mutex_lock(&pxrc->pm_mutex);
  194. usb_kill_urb(pxrc->urb);
  195. return 0;
  196. }
  197. static int pxrc_post_reset(struct usb_interface *intf)
  198. {
  199. struct pxrc *pxrc = usb_get_intfdata(intf);
  200. int retval = 0;
  201. if (pxrc->is_open && usb_submit_urb(pxrc->urb, GFP_KERNEL) < 0)
  202. retval = -EIO;
  203. mutex_unlock(&pxrc->pm_mutex);
  204. return retval;
  205. }
  206. static int pxrc_reset_resume(struct usb_interface *intf)
  207. {
  208. return pxrc_resume(intf);
  209. }
  210. static const struct usb_device_id pxrc_table[] = {
  211. { USB_DEVICE(PXRC_VENDOR_ID, PXRC_PRODUCT_ID) },
  212. { }
  213. };
  214. MODULE_DEVICE_TABLE(usb, pxrc_table);
  215. static struct usb_driver pxrc_driver = {
  216. .name = "pxrc",
  217. .probe = pxrc_probe,
  218. .disconnect = pxrc_disconnect,
  219. .id_table = pxrc_table,
  220. .suspend = pxrc_suspend,
  221. .resume = pxrc_resume,
  222. .pre_reset = pxrc_pre_reset,
  223. .post_reset = pxrc_post_reset,
  224. .reset_resume = pxrc_reset_resume,
  225. };
  226. module_usb_driver(pxrc_driver);
  227. MODULE_AUTHOR("Marcus Folkesson <marcus.folkesson@gmail.com>");
  228. MODULE_DESCRIPTION("PhoenixRC Flight Controller Adapter");
  229. MODULE_LICENSE("GPL v2");