1 /*
2 * Copyright (c) 2022, sakumisu
3 *
4 * SPDX-License-Identifier: Apache-2.0
5 */
6 #include "usbh_core.h"
7 #include "usbh_printer.h"
8
9 #define DEV_FORMAT "/dev/printer"
10
11 static struct usbh_printer g_printer_class[CONFIG_USBHOST_MAX_PRINTER_CLASS];
12 static uint32_t g_devinuse = 0;
13
usbh_printer_class_alloc(void)14 static struct usbh_printer *usbh_printer_class_alloc(void)
15 {
16 int devno;
17
18 for (devno = 0; devno < CONFIG_USBHOST_MAX_PRINTER_CLASS; devno++) {
19 if ((g_devinuse & (1 << devno)) == 0) {
20 g_devinuse |= (1 << devno);
21 memset(&g_printer_class[devno], 0, sizeof(struct usbh_printer));
22 g_printer_class[devno].minor = devno;
23 return &g_printer_class[devno];
24 }
25 }
26 return NULL;
27 }
28
usbh_printer_class_free(struct usbh_printer * printer_class)29 static void usbh_printer_class_free(struct usbh_printer *printer_class)
30 {
31 int devno = printer_class->minor;
32
33 if (devno >= 0 && devno < 32) {
34 g_devinuse &= ~(1 << devno);
35 }
36 memset(printer_class, 0, sizeof(struct usbh_printer));
37 }
38
usbh_printer_get_device_id(struct usbh_printer * printer_class,uint8_t * buffer)39 static int usbh_printer_get_device_id(struct usbh_printer *printer_class, uint8_t *buffer)
40 {
41 struct usb_setup_packet *setup = &printer_class->hport->setup;
42 int ret;
43
44 setup->bmRequestType = USB_REQUEST_DIR_IN | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
45 setup->bRequest = PRINTER_REQUEST_GET_DEVICE_ID;
46 setup->wValue = 0;
47 setup->wIndex = printer_class->intf;
48 setup->wLength = 256;
49
50 return usbh_control_transfer(printer_class->hport->ep0, setup, buffer);
51 }
52
usbh_printer_get_port_status(struct usbh_printer * printer_class,uint8_t * buffer)53 static int usbh_printer_get_port_status(struct usbh_printer *printer_class, uint8_t *buffer)
54 {
55 struct usb_setup_packet *setup = &printer_class->hport->setup;
56 int ret;
57
58 setup->bmRequestType = USB_REQUEST_DIR_IN | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
59 setup->bRequest = PRINTER_REQUEST_GET_PORT_SATTUS;
60 setup->wValue = 0;
61 setup->wIndex = printer_class->intf;
62 setup->wLength = 1;
63
64 return usbh_control_transfer(printer_class->hport->ep0, setup, buffer);
65 }
66
usbh_printer_soft_reset(struct usbh_printer * printer_class)67 static int usbh_printer_soft_reset(struct usbh_printer *printer_class)
68 {
69 struct usb_setup_packet *setup = &printer_class->hport->setup;
70 int ret;
71
72 setup->bmRequestType = USB_REQUEST_DIR_OUT | USB_REQUEST_CLASS | USB_REQUEST_RECIPIENT_INTERFACE;
73 setup->bRequest = PRINTER_REQUEST_SOFT_RESET;
74 setup->wValue = 0;
75 setup->wIndex = printer_class->intf;
76 setup->wLength = 0;
77
78 return usbh_control_transfer(printer_class->hport->ep0, setup, NULL);
79 }
80
usbh_printer_connect(struct usbh_hubport * hport,uint8_t intf)81 static int usbh_printer_connect(struct usbh_hubport *hport, uint8_t intf)
82 {
83 struct usbh_endpoint_cfg ep_cfg = { 0 };
84 struct usb_endpoint_descriptor *ep_desc;
85 int ret;
86
87 struct usbh_printer *printer_class = usbh_printer_class_alloc();
88 if (printer_class == NULL) {
89 USB_LOG_ERR("Fail to alloc printer_class\r\n");
90 return -ENOMEM;
91 }
92
93 printer_class->hport = hport;
94 printer_class->intf = intf;
95
96 hport->config.intf[intf].priv = printer_class;
97
98 for (uint8_t i = 0; i < hport->config.intf[intf + 1].altsetting[0].intf_desc.bNumEndpoints; i++) {
99 ep_desc = &hport->config.intf[intf + 1].altsetting[0].ep[i].ep_desc;
100
101 if (ep_desc->bEndpointAddress & 0x80) {
102 usbh_hport_activate_epx(&printer_class->bulkin, hport, ep_desc);
103 } else {
104 usbh_hport_activate_epx(&printer_class->bulkout, hport, ep_desc);
105 }
106 }
107
108 // uint8_t *device_id = usb_iomalloc(256);
109 // ret = usbh_printer_get_device_id(printer_class, device_id);
110 strncpy(hport->config.intf[intf].devname, DEV_FORMAT, CONFIG_USBHOST_DEV_NAMELEN);
111
112 USB_LOG_INFO("Register Printer Class:%s\r\n", hport->config.intf[intf].devname);
113
114 return 0;
115 }
116
usbh_printer_disconnect(struct usbh_hubport * hport,uint8_t intf)117 static int usbh_printer_disconnect(struct usbh_hubport *hport, uint8_t intf)
118 {
119 int ret = 0;
120
121 struct usbh_printer *printer_class = (struct usbh_printer *)hport->config.intf[intf].priv;
122
123 if (printer_class) {
124 if (printer_class->bulkin) {
125 usbh_pipe_free(printer_class->bulkin);
126 }
127
128 if (printer_class->bulkout) {
129 usbh_pipe_free(printer_class->bulkout);
130 }
131
132 if (hport->config.intf[intf].devname[0] != '\0') {
133 USB_LOG_INFO("Unregister Printer Class:%s\r\n", hport->config.intf[intf].devname);
134 }
135
136 usbh_printer_class_free(printer_class);
137 }
138
139 return ret;
140 }
141
142 static const struct usbh_class_driver printer_class_driver = {
143 .driver_name = "printer",
144 .connect = usbh_printer_connect,
145 .disconnect = usbh_printer_disconnect
146 };
147
148 CLASS_INFO_DEFINE const struct usbh_class_info printer_class_info = {
149 .match_flags = USB_CLASS_MATCH_INTF_CLASS | USB_CLASS_MATCH_INTF_SUBCLASS | USB_CLASS_MATCH_INTF_PROTOCOL,
150 .class = USB_DEVICE_CLASS_PRINTER,
151 .subclass = PRINTER_SUBCLASS,
152 .protocol = 0x00,
153 .vid = 0x00,
154 .pid = 0x00,
155 .class_driver = &printer_class_driver
156 };
157