1 /*
2  * Copyright (c) 2022, sakumisu
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  */
6 #include "usbh_core.h"
7 #include "usbh_cdc_acm.h"
8 
9 #undef USB_DBG_TAG
10 #define USB_DBG_TAG "usbh_cdc_acm"
11 #include "usb_log.h"
12 
13 #define DEV_FORMAT "/dev/ttyACM%d"
14 
15 USB_NOCACHE_RAM_SECTION USB_MEM_ALIGNX uint8_t g_cdc_acm_buf[CONFIG_USBHOST_MAX_CDC_ACM_CLASS][USB_ALIGN_UP(64, CONFIG_USB_ALIGN_SIZE)];
16 
17 static struct usbh_cdc_acm g_cdc_acm_class[CONFIG_USBHOST_MAX_CDC_ACM_CLASS];
18 static uint32_t g_devinuse = 0;
19 
usbh_cdc_acm_class_alloc(void)20 static struct usbh_cdc_acm *usbh_cdc_acm_class_alloc(void)
21 {
22     uint8_t devno;
23 
24     for (devno = 0; devno < CONFIG_USBHOST_MAX_CDC_ACM_CLASS; devno++) {
25         if ((g_devinuse & (1U << devno)) == 0) {
26             g_devinuse |= (1U << devno);
27             memset(&g_cdc_acm_class[devno], 0, sizeof(struct usbh_cdc_acm));
28             g_cdc_acm_class[devno].minor = devno;
29             return &g_cdc_acm_class[devno];
30         }
31     }
32     return NULL;
33 }
34 
usbh_cdc_acm_class_free(struct usbh_cdc_acm * cdc_acm_class)35 static void usbh_cdc_acm_class_free(struct usbh_cdc_acm *cdc_acm_class)
36 {
37     uint8_t devno = cdc_acm_class->minor;
38 
39     if (devno < 32) {
40         g_devinuse &= ~(1U << devno);
41     }
42     memset(cdc_acm_class, 0, sizeof(struct usbh_cdc_acm));
43 }
44 
usbh_cdc_acm_set_line_coding(struct usbh_cdc_acm * cdc_acm_class,struct cdc_line_coding * line_coding)45 int usbh_cdc_acm_set_line_coding(struct usbh_cdc_acm *cdc_acm_class, struct cdc_line_coding *line_coding)
46 {
47     struct usb_setup_packet *setup;
48 
49     if (!cdc_acm_class || !cdc_acm_class->hport) {
50         return -USB_ERR_INVAL;
51     }
52     setup = cdc_acm_class->hport->setup;
53 
54     setup->bmRequestType = USB_REQUEST_DIR_OUT | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
55     setup->bRequest = CDC_REQUEST_SET_LINE_CODING;
56     setup->wValue = 0;
57     setup->wIndex = cdc_acm_class->intf;
58     setup->wLength = 7;
59 
60     memcpy(g_cdc_acm_buf[cdc_acm_class->minor], line_coding, sizeof(struct cdc_line_coding));
61 
62     return usbh_control_transfer(cdc_acm_class->hport, setup, g_cdc_acm_buf[cdc_acm_class->minor]);
63 }
64 
usbh_cdc_acm_get_line_coding(struct usbh_cdc_acm * cdc_acm_class,struct cdc_line_coding * line_coding)65 int usbh_cdc_acm_get_line_coding(struct usbh_cdc_acm *cdc_acm_class, struct cdc_line_coding *line_coding)
66 {
67     struct usb_setup_packet *setup;
68     int ret;
69 
70     if (!cdc_acm_class || !cdc_acm_class->hport) {
71         return -USB_ERR_INVAL;
72     }
73     setup = cdc_acm_class->hport->setup;
74 
75     setup->bmRequestType = USB_REQUEST_DIR_IN | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
76     setup->bRequest = CDC_REQUEST_GET_LINE_CODING;
77     setup->wValue = 0;
78     setup->wIndex = cdc_acm_class->intf;
79     setup->wLength = 7;
80 
81     ret = usbh_control_transfer(cdc_acm_class->hport, setup, g_cdc_acm_buf[cdc_acm_class->minor]);
82     if (ret < 0) {
83         return ret;
84     }
85     memcpy(line_coding, g_cdc_acm_buf[cdc_acm_class->minor], sizeof(struct cdc_line_coding));
86     return ret;
87 }
88 
usbh_cdc_acm_set_line_state(struct usbh_cdc_acm * cdc_acm_class,bool dtr,bool rts)89 int usbh_cdc_acm_set_line_state(struct usbh_cdc_acm *cdc_acm_class, bool dtr, bool rts)
90 {
91     struct usb_setup_packet *setup;
92 
93     if (!cdc_acm_class || !cdc_acm_class->hport) {
94         return -USB_ERR_INVAL;
95     }
96     setup = cdc_acm_class->hport->setup;
97 
98     setup->bmRequestType = USB_REQUEST_DIR_OUT | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
99     setup->bRequest = CDC_REQUEST_SET_CONTROL_LINE_STATE;
100     setup->wValue = (dtr << 0) | (rts << 1);
101     setup->wIndex = cdc_acm_class->intf;
102     setup->wLength = 0;
103 
104     return usbh_control_transfer(cdc_acm_class->hport, setup, NULL);
105 }
106 
usbh_cdc_acm_connect(struct usbh_hubport * hport,uint8_t intf)107 static int usbh_cdc_acm_connect(struct usbh_hubport *hport, uint8_t intf)
108 {
109     struct usb_endpoint_descriptor *ep_desc;
110     int ret = 0;
111 
112     struct usbh_cdc_acm *cdc_acm_class = usbh_cdc_acm_class_alloc();
113     if (cdc_acm_class == NULL) {
114         USB_LOG_ERR("Fail to alloc cdc_acm_class\r\n");
115         return -USB_ERR_NOMEM;
116     }
117 
118     cdc_acm_class->hport = hport;
119     cdc_acm_class->intf = intf;
120 
121     hport->config.intf[intf].priv = cdc_acm_class;
122     hport->config.intf[intf + 1].priv = NULL;
123 
124 #ifdef CONFIG_USBHOST_CDC_ACM_NOTIFY
125     ep_desc = &hport->config.intf[intf].altsetting[0].ep[0].ep_desc;
126     USBH_EP_INIT(cdc_acm_class->intin, ep_desc);
127 #endif
128     for (uint8_t i = 0; i < hport->config.intf[intf + 1].altsetting[0].intf_desc.bNumEndpoints; i++) {
129         ep_desc = &hport->config.intf[intf + 1].altsetting[0].ep[i].ep_desc;
130 
131         if (ep_desc->bEndpointAddress & 0x80) {
132             USBH_EP_INIT(cdc_acm_class->bulkin, ep_desc);
133         } else {
134             USBH_EP_INIT(cdc_acm_class->bulkout, ep_desc);
135         }
136     }
137 
138     snprintf(hport->config.intf[intf].devname, CONFIG_USBHOST_DEV_NAMELEN, DEV_FORMAT, cdc_acm_class->minor);
139 
140     USB_LOG_INFO("Register CDC ACM Class:%s\r\n", hport->config.intf[intf].devname);
141 
142 #if 0
143     USB_LOG_INFO("Test cdc acm rx and tx and rx for 5 times, baudrate is 115200\r\n");
144 
145     struct cdc_line_coding linecoding;
146     uint8_t count = 5;
147 
148     linecoding.dwDTERate = 115200;
149     linecoding.bDataBits = 8;
150     linecoding.bParityType = 0;
151     linecoding.bCharFormat = 0;
152     usbh_cdc_acm_set_line_coding(cdc_acm_class, &linecoding);
153     usbh_cdc_acm_set_line_state(cdc_acm_class, true, false);
154 
155     memset(g_cdc_acm_buf, 'a', sizeof(g_cdc_acm_buf));
156     ret = usbh_cdc_acm_bulk_out_transfer(cdc_acm_class, g_cdc_acm_buf, sizeof(g_cdc_acm_buf), 0xfffffff);
157     USB_LOG_RAW("out ret:%d\r\n", ret);
158     while (count--) {
159         ret = usbh_cdc_acm_bulk_in_transfer(cdc_acm_class, g_cdc_acm_buf, sizeof(g_cdc_acm_buf), 0xfffffff);
160         USB_LOG_RAW("in ret:%d\r\n", ret);
161         if (ret > 0) {
162             for (uint32_t i = 0; i < ret; i++) {
163                 USB_LOG_RAW("%02x ", g_cdc_acm_buf[i]);
164             }
165         }
166         USB_LOG_RAW("\r\n");
167     }
168 #endif
169 
170     usbh_cdc_acm_run(cdc_acm_class);
171     return ret;
172 }
173 
usbh_cdc_acm_disconnect(struct usbh_hubport * hport,uint8_t intf)174 static int usbh_cdc_acm_disconnect(struct usbh_hubport *hport, uint8_t intf)
175 {
176     int ret = 0;
177 
178     struct usbh_cdc_acm *cdc_acm_class = (struct usbh_cdc_acm *)hport->config.intf[intf].priv;
179 
180     if (cdc_acm_class) {
181         if (cdc_acm_class->bulkin) {
182             usbh_kill_urb(&cdc_acm_class->bulkin_urb);
183         }
184 
185         if (cdc_acm_class->bulkout) {
186             usbh_kill_urb(&cdc_acm_class->bulkout_urb);
187         }
188 
189 #ifdef CONFIG_USBHOST_CDC_ACM_NOTIFY
190         if (cdc_acm_class->intin) {
191             usbh_kill_urb(&cdc_acm_class->intin_urb);
192         }
193 #endif
194 
195         if (hport->config.intf[intf].devname[0] != '\0') {
196             usb_osal_thread_schedule_other();
197             USB_LOG_INFO("Unregister CDC ACM Class:%s\r\n", hport->config.intf[intf].devname);
198             usbh_cdc_acm_stop(cdc_acm_class);
199         }
200 
201         usbh_cdc_acm_class_free(cdc_acm_class);
202     }
203 
204     return ret;
205 }
206 
usbh_cdc_acm_bulk_in_transfer(struct usbh_cdc_acm * cdc_acm_class,uint8_t * buffer,uint32_t buflen,uint32_t timeout)207 int usbh_cdc_acm_bulk_in_transfer(struct usbh_cdc_acm *cdc_acm_class, uint8_t *buffer, uint32_t buflen, uint32_t timeout)
208 {
209     int ret;
210     struct usbh_urb *urb = &cdc_acm_class->bulkin_urb;
211 
212     usbh_bulk_urb_fill(urb, cdc_acm_class->hport, cdc_acm_class->bulkin, buffer, buflen, timeout, NULL, NULL);
213     ret = usbh_submit_urb(urb);
214     if (ret == 0) {
215         ret = urb->actual_length;
216     }
217     return ret;
218 }
219 
usbh_cdc_acm_bulk_out_transfer(struct usbh_cdc_acm * cdc_acm_class,uint8_t * buffer,uint32_t buflen,uint32_t timeout)220 int usbh_cdc_acm_bulk_out_transfer(struct usbh_cdc_acm *cdc_acm_class, uint8_t *buffer, uint32_t buflen, uint32_t timeout)
221 {
222     int ret;
223     struct usbh_urb *urb = &cdc_acm_class->bulkout_urb;
224 
225     usbh_bulk_urb_fill(urb, cdc_acm_class->hport, cdc_acm_class->bulkout, buffer, buflen, timeout, NULL, NULL);
226     ret = usbh_submit_urb(urb);
227     if (ret == 0) {
228         ret = urb->actual_length;
229     }
230     return ret;
231 }
232 
usbh_cdc_data_connect(struct usbh_hubport * hport,uint8_t intf)233 static int usbh_cdc_data_connect(struct usbh_hubport *hport, uint8_t intf)
234 {
235     (void)hport;
236     (void)intf;
237     return 0;
238 }
239 
usbh_cdc_data_disconnect(struct usbh_hubport * hport,uint8_t intf)240 static int usbh_cdc_data_disconnect(struct usbh_hubport *hport, uint8_t intf)
241 {
242     (void)hport;
243     (void)intf;
244     return 0;
245 }
246 
usbh_cdc_acm_run(struct usbh_cdc_acm * cdc_acm_class)247 __WEAK void usbh_cdc_acm_run(struct usbh_cdc_acm *cdc_acm_class)
248 {
249     (void)cdc_acm_class;
250 }
251 
usbh_cdc_acm_stop(struct usbh_cdc_acm * cdc_acm_class)252 __WEAK void usbh_cdc_acm_stop(struct usbh_cdc_acm *cdc_acm_class)
253 {
254     (void)cdc_acm_class;
255 }
256 
257 const struct usbh_class_driver cdc_acm_class_driver = {
258     .driver_name = "cdc_acm",
259     .connect = usbh_cdc_acm_connect,
260     .disconnect = usbh_cdc_acm_disconnect
261 };
262 
263 const struct usbh_class_driver cdc_data_class_driver = {
264     .driver_name = "cdc_data",
265     .connect = usbh_cdc_data_connect,
266     .disconnect = usbh_cdc_data_disconnect
267 };
268 
269 CLASS_INFO_DEFINE const struct usbh_class_info cdc_acm_class_info = {
270     .match_flags = USB_CLASS_MATCH_INTF_CLASS | USB_CLASS_MATCH_INTF_SUBCLASS,
271     .bInterfaceClass = USB_DEVICE_CLASS_CDC,
272     .bInterfaceSubClass = CDC_ABSTRACT_CONTROL_MODEL,
273     .bInterfaceProtocol = 0x00,
274     .id_table = NULL,
275     .class_driver = &cdc_acm_class_driver
276 };
277 
278 CLASS_INFO_DEFINE const struct usbh_class_info cdc_data_class_info = {
279     .match_flags = USB_CLASS_MATCH_INTF_CLASS,
280     .bInterfaceClass = USB_DEVICE_CLASS_CDC_DATA,
281     .bInterfaceSubClass = 0x00,
282     .bInterfaceProtocol = 0x00,
283     .id_table = NULL,
284     .class_driver = &cdc_data_class_driver
285 };
286