1 /* 2 * Copyright 2007, Frank A Kingswood <frank@kingswood-consulting.co.uk> 3 * Copyright 2007, Werner Cornelius <werner@cornelius-consult.de> 4 * Copyright 2009, Boris Hajduk <boris@hajduk.org> 5 * 6 * ch341.c implements a serial port driver for the Winchiphead CH341. 7 * 8 * The CH341 device can be used to implement an RS232 asynchronous 9 * serial port, an IEEE-1284 parallel printer port or a memory-like 10 * interface. In all cases the CH341 supports an I2C interface as well. 11 * This driver only supports the asynchronous serial interface. 12 * 13 * This program is free software; you can redistribute it and/or 14 * modify it under the terms of the GNU General Public License version 15 * 2 as published by the Free Software Foundation. 16 */ 17 18 #include <linux/kernel.h> 19 #include <linux/init.h> 20 #include <linux/tty.h> 21 #include <linux/module.h> 22 #include <linux/slab.h> 23 #include <linux/usb.h> 24 #include <linux/usb/serial.h> 25 #include <linux/serial.h> 26 #include <asm/unaligned.h> 27 28 #define DEFAULT_BAUD_RATE 9600 29 #define DEFAULT_TIMEOUT 1000 30 31 /* flags for IO-Bits */ 32 #define CH341_BIT_RTS (1 << 6) 33 #define CH341_BIT_DTR (1 << 5) 34 35 /******************************/ 36 /* interrupt pipe definitions */ 37 /******************************/ 38 /* always 4 interrupt bytes */ 39 /* first irq byte normally 0x08 */ 40 /* second irq byte base 0x7d + below */ 41 /* third irq byte base 0x94 + below */ 42 /* fourth irq byte normally 0xee */ 43 44 /* second interrupt byte */ 45 #define CH341_MULT_STAT 0x04 /* multiple status since last interrupt event */ 46 47 /* status returned in third interrupt answer byte, inverted in data 48 from irq */ 49 #define CH341_BIT_CTS 0x01 50 #define CH341_BIT_DSR 0x02 51 #define CH341_BIT_RI 0x04 52 #define CH341_BIT_DCD 0x08 53 #define CH341_BITS_MODEM_STAT 0x0f /* all bits */ 54 55 /*******************************/ 56 /* baudrate calculation factor */ 57 /*******************************/ 58 #define CH341_BAUDBASE_FACTOR 1532620800 59 #define CH341_BAUDBASE_DIVMAX 3 60 61 /* Break support - the information used to implement this was gleaned from 62 * the Net/FreeBSD uchcom.c driver by Takanori Watanabe. Domo arigato. 63 */ 64 65 #define CH341_REQ_WRITE_REG 0x9A 66 #define CH341_REQ_READ_REG 0x95 67 #define CH341_REG_BREAK1 0x05 68 #define CH341_REG_BREAK2 0x18 69 #define CH341_NBREAK_BITS_REG1 0x01 70 #define CH341_NBREAK_BITS_REG2 0x40 71 72 73 static const struct usb_device_id id_table[] = { 74 { USB_DEVICE(0x4348, 0x5523) }, 75 { USB_DEVICE(0x1a86, 0x7523) }, 76 { USB_DEVICE(0x1a86, 0x5523) }, 77 { }, 78 }; 79 MODULE_DEVICE_TABLE(usb, id_table); 80 81 struct ch341_private { 82 spinlock_t lock; /* access lock */ 83 unsigned baud_rate; /* set baud rate */ 84 u8 line_control; /* set line control value RTS/DTR */ 85 u8 line_status; /* active status of modem control inputs */ 86 u8 multi_status_change; /* status changed multiple since last call */ 87 }; 88 89 static int ch341_control_out(struct usb_device *dev, u8 request, 90 u16 value, u16 index) 91 { 92 int r; 93 94 dev_dbg(&dev->dev, "ch341_control_out(%02x,%02x,%04x,%04x)\n", 95 USB_DIR_OUT|0x40, (int)request, (int)value, (int)index); 96 97 r = usb_control_msg(dev, usb_sndctrlpipe(dev, 0), request, 98 USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_OUT, 99 value, index, NULL, 0, DEFAULT_TIMEOUT); 100 101 return r; 102 } 103 104 static int ch341_control_in(struct usb_device *dev, 105 u8 request, u16 value, u16 index, 106 char *buf, unsigned bufsize) 107 { 108 int r; 109 110 dev_dbg(&dev->dev, "ch341_control_in(%02x,%02x,%04x,%04x,%p,%u)\n", 111 USB_DIR_IN|0x40, (int)request, (int)value, (int)index, buf, 112 (int)bufsize); 113 114 r = usb_control_msg(dev, usb_rcvctrlpipe(dev, 0), request, 115 USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_DIR_IN, 116 value, index, buf, bufsize, DEFAULT_TIMEOUT); 117 return r; 118 } 119 120 static int ch341_set_baudrate(struct usb_device *dev, 121 struct ch341_private *priv) 122 { 123 short a, b; 124 int r; 125 unsigned long factor; 126 short divisor; 127 128 if (!priv->baud_rate) 129 return -EINVAL; 130 factor = (CH341_BAUDBASE_FACTOR / priv->baud_rate); 131 divisor = CH341_BAUDBASE_DIVMAX; 132 133 while ((factor > 0xfff0) && divisor) { 134 factor >>= 3; 135 divisor--; 136 } 137 138 if (factor > 0xfff0) 139 return -EINVAL; 140 141 factor = 0x10000 - factor; 142 a = (factor & 0xff00) | divisor; 143 b = factor & 0xff; 144 145 r = ch341_control_out(dev, 0x9a, 0x1312, a); 146 if (!r) 147 r = ch341_control_out(dev, 0x9a, 0x0f2c, b); 148 149 return r; 150 } 151 152 static int ch341_set_handshake(struct usb_device *dev, u8 control) 153 { 154 return ch341_control_out(dev, 0xa4, ~control, 0); 155 } 156 157 static int ch341_get_status(struct usb_device *dev, struct ch341_private *priv) 158 { 159 char *buffer; 160 int r; 161 const unsigned size = 8; 162 unsigned long flags; 163 164 buffer = kmalloc(size, GFP_KERNEL); 165 if (!buffer) 166 return -ENOMEM; 167 168 r = ch341_control_in(dev, 0x95, 0x0706, 0, buffer, size); 169 if (r < 0) 170 goto out; 171 172 /* setup the private status if available */ 173 if (r == 2) { 174 r = 0; 175 spin_lock_irqsave(&priv->lock, flags); 176 priv->line_status = (~(*buffer)) & CH341_BITS_MODEM_STAT; 177 priv->multi_status_change = 0; 178 spin_unlock_irqrestore(&priv->lock, flags); 179 } else 180 r = -EPROTO; 181 182 out: kfree(buffer); 183 return r; 184 } 185 186 /* -------------------------------------------------------------------------- */ 187 188 static int ch341_configure(struct usb_device *dev, struct ch341_private *priv) 189 { 190 char *buffer; 191 int r; 192 const unsigned size = 8; 193 194 buffer = kmalloc(size, GFP_KERNEL); 195 if (!buffer) 196 return -ENOMEM; 197 198 /* expect two bytes 0x27 0x00 */ 199 r = ch341_control_in(dev, 0x5f, 0, 0, buffer, size); 200 if (r < 0) 201 goto out; 202 203 r = ch341_control_out(dev, 0xa1, 0, 0); 204 if (r < 0) 205 goto out; 206 207 r = ch341_set_baudrate(dev, priv); 208 if (r < 0) 209 goto out; 210 211 /* expect two bytes 0x56 0x00 */ 212 r = ch341_control_in(dev, 0x95, 0x2518, 0, buffer, size); 213 if (r < 0) 214 goto out; 215 216 r = ch341_control_out(dev, 0x9a, 0x2518, 0x0050); 217 if (r < 0) 218 goto out; 219 220 /* expect 0xff 0xee */ 221 r = ch341_get_status(dev, priv); 222 if (r < 0) 223 goto out; 224 225 r = ch341_control_out(dev, 0xa1, 0x501f, 0xd90a); 226 if (r < 0) 227 goto out; 228 229 r = ch341_set_baudrate(dev, priv); 230 if (r < 0) 231 goto out; 232 233 r = ch341_set_handshake(dev, priv->line_control); 234 if (r < 0) 235 goto out; 236 237 /* expect 0x9f 0xee */ 238 r = ch341_get_status(dev, priv); 239 240 out: kfree(buffer); 241 return r; 242 } 243 244 static int ch341_port_probe(struct usb_serial_port *port) 245 { 246 struct ch341_private *priv; 247 int r; 248 249 priv = kzalloc(sizeof(struct ch341_private), GFP_KERNEL); 250 if (!priv) 251 return -ENOMEM; 252 253 spin_lock_init(&priv->lock); 254 priv->baud_rate = DEFAULT_BAUD_RATE; 255 priv->line_control = CH341_BIT_RTS | CH341_BIT_DTR; 256 257 r = ch341_configure(port->serial->dev, priv); 258 if (r < 0) 259 goto error; 260 261 usb_set_serial_port_data(port, priv); 262 return 0; 263 264 error: kfree(priv); 265 return r; 266 } 267 268 static int ch341_port_remove(struct usb_serial_port *port) 269 { 270 struct ch341_private *priv; 271 272 priv = usb_get_serial_port_data(port); 273 kfree(priv); 274 275 return 0; 276 } 277 278 static int ch341_carrier_raised(struct usb_serial_port *port) 279 { 280 struct ch341_private *priv = usb_get_serial_port_data(port); 281 if (priv->line_status & CH341_BIT_DCD) 282 return 1; 283 return 0; 284 } 285 286 static void ch341_dtr_rts(struct usb_serial_port *port, int on) 287 { 288 struct ch341_private *priv = usb_get_serial_port_data(port); 289 unsigned long flags; 290 291 /* drop DTR and RTS */ 292 spin_lock_irqsave(&priv->lock, flags); 293 if (on) 294 priv->line_control |= CH341_BIT_RTS | CH341_BIT_DTR; 295 else 296 priv->line_control &= ~(CH341_BIT_RTS | CH341_BIT_DTR); 297 spin_unlock_irqrestore(&priv->lock, flags); 298 ch341_set_handshake(port->serial->dev, priv->line_control); 299 wake_up_interruptible(&port->delta_msr_wait); 300 } 301 302 static void ch341_close(struct usb_serial_port *port) 303 { 304 usb_serial_generic_close(port); 305 usb_kill_urb(port->interrupt_in_urb); 306 } 307 308 309 /* open this device, set default parameters */ 310 static int ch341_open(struct tty_struct *tty, struct usb_serial_port *port) 311 { 312 struct usb_serial *serial = port->serial; 313 struct ch341_private *priv = usb_get_serial_port_data(port); 314 int r; 315 316 priv->baud_rate = DEFAULT_BAUD_RATE; 317 318 r = ch341_configure(serial->dev, priv); 319 if (r) 320 goto out; 321 322 r = ch341_set_handshake(serial->dev, priv->line_control); 323 if (r) 324 goto out; 325 326 r = ch341_set_baudrate(serial->dev, priv); 327 if (r) 328 goto out; 329 330 dev_dbg(&port->dev, "%s - submitting interrupt urb", __func__); 331 r = usb_submit_urb(port->interrupt_in_urb, GFP_KERNEL); 332 if (r) { 333 dev_err(&port->dev, "%s - failed submitting interrupt urb," 334 " error %d\n", __func__, r); 335 ch341_close(port); 336 goto out; 337 } 338 339 r = usb_serial_generic_open(tty, port); 340 341 out: return r; 342 } 343 344 /* Old_termios contains the original termios settings and 345 * tty->termios contains the new setting to be used. 346 */ 347 static void ch341_set_termios(struct tty_struct *tty, 348 struct usb_serial_port *port, struct ktermios *old_termios) 349 { 350 struct ch341_private *priv = usb_get_serial_port_data(port); 351 unsigned baud_rate; 352 unsigned long flags; 353 354 baud_rate = tty_get_baud_rate(tty); 355 356 priv->baud_rate = baud_rate; 357 358 if (baud_rate) { 359 spin_lock_irqsave(&priv->lock, flags); 360 priv->line_control |= (CH341_BIT_DTR | CH341_BIT_RTS); 361 spin_unlock_irqrestore(&priv->lock, flags); 362 ch341_set_baudrate(port->serial->dev, priv); 363 } else { 364 spin_lock_irqsave(&priv->lock, flags); 365 priv->line_control &= ~(CH341_BIT_DTR | CH341_BIT_RTS); 366 spin_unlock_irqrestore(&priv->lock, flags); 367 } 368 369 ch341_set_handshake(port->serial->dev, priv->line_control); 370 371 /* Unimplemented: 372 * (cflag & CSIZE) : data bits [5, 8] 373 * (cflag & PARENB) : parity {NONE, EVEN, ODD} 374 * (cflag & CSTOPB) : stop bits [1, 2] 375 */ 376 } 377 378 static void ch341_break_ctl(struct tty_struct *tty, int break_state) 379 { 380 const uint16_t ch341_break_reg = 381 CH341_REG_BREAK1 | ((uint16_t) CH341_REG_BREAK2 << 8); 382 struct usb_serial_port *port = tty->driver_data; 383 int r; 384 uint16_t reg_contents; 385 uint8_t *break_reg; 386 387 break_reg = kmalloc(2, GFP_KERNEL); 388 if (!break_reg) { 389 dev_err(&port->dev, "%s - kmalloc failed\n", __func__); 390 return; 391 } 392 393 r = ch341_control_in(port->serial->dev, CH341_REQ_READ_REG, 394 ch341_break_reg, 0, break_reg, 2); 395 if (r < 0) { 396 dev_err(&port->dev, "%s - USB control read error (%d)\n", 397 __func__, r); 398 goto out; 399 } 400 dev_dbg(&port->dev, "%s - initial ch341 break register contents - reg1: %x, reg2: %x\n", 401 __func__, break_reg[0], break_reg[1]); 402 if (break_state != 0) { 403 dev_dbg(&port->dev, "%s - Enter break state requested\n", __func__); 404 break_reg[0] &= ~CH341_NBREAK_BITS_REG1; 405 break_reg[1] &= ~CH341_NBREAK_BITS_REG2; 406 } else { 407 dev_dbg(&port->dev, "%s - Leave break state requested\n", __func__); 408 break_reg[0] |= CH341_NBREAK_BITS_REG1; 409 break_reg[1] |= CH341_NBREAK_BITS_REG2; 410 } 411 dev_dbg(&port->dev, "%s - New ch341 break register contents - reg1: %x, reg2: %x\n", 412 __func__, break_reg[0], break_reg[1]); 413 reg_contents = get_unaligned_le16(break_reg); 414 r = ch341_control_out(port->serial->dev, CH341_REQ_WRITE_REG, 415 ch341_break_reg, reg_contents); 416 if (r < 0) 417 dev_err(&port->dev, "%s - USB control write error (%d)\n", 418 __func__, r); 419 out: 420 kfree(break_reg); 421 } 422 423 static int ch341_tiocmset(struct tty_struct *tty, 424 unsigned int set, unsigned int clear) 425 { 426 struct usb_serial_port *port = tty->driver_data; 427 struct ch341_private *priv = usb_get_serial_port_data(port); 428 unsigned long flags; 429 u8 control; 430 431 spin_lock_irqsave(&priv->lock, flags); 432 if (set & TIOCM_RTS) 433 priv->line_control |= CH341_BIT_RTS; 434 if (set & TIOCM_DTR) 435 priv->line_control |= CH341_BIT_DTR; 436 if (clear & TIOCM_RTS) 437 priv->line_control &= ~CH341_BIT_RTS; 438 if (clear & TIOCM_DTR) 439 priv->line_control &= ~CH341_BIT_DTR; 440 control = priv->line_control; 441 spin_unlock_irqrestore(&priv->lock, flags); 442 443 return ch341_set_handshake(port->serial->dev, control); 444 } 445 446 static void ch341_read_int_callback(struct urb *urb) 447 { 448 struct usb_serial_port *port = (struct usb_serial_port *) urb->context; 449 unsigned char *data = urb->transfer_buffer; 450 unsigned int actual_length = urb->actual_length; 451 int status; 452 453 switch (urb->status) { 454 case 0: 455 /* success */ 456 break; 457 case -ECONNRESET: 458 case -ENOENT: 459 case -ESHUTDOWN: 460 /* this urb is terminated, clean up */ 461 dev_dbg(&urb->dev->dev, "%s - urb shutting down with status: %d\n", 462 __func__, urb->status); 463 return; 464 default: 465 dev_dbg(&urb->dev->dev, "%s - nonzero urb status received: %d\n", 466 __func__, urb->status); 467 goto exit; 468 } 469 470 usb_serial_debug_data(&port->dev, __func__, 471 urb->actual_length, urb->transfer_buffer); 472 473 if (actual_length >= 4) { 474 struct ch341_private *priv = usb_get_serial_port_data(port); 475 unsigned long flags; 476 u8 prev_line_status = priv->line_status; 477 478 spin_lock_irqsave(&priv->lock, flags); 479 priv->line_status = (~(data[2])) & CH341_BITS_MODEM_STAT; 480 if ((data[1] & CH341_MULT_STAT)) 481 priv->multi_status_change = 1; 482 spin_unlock_irqrestore(&priv->lock, flags); 483 484 if ((priv->line_status ^ prev_line_status) & CH341_BIT_DCD) { 485 struct tty_struct *tty = tty_port_tty_get(&port->port); 486 if (tty) 487 usb_serial_handle_dcd_change(port, tty, 488 priv->line_status & CH341_BIT_DCD); 489 tty_kref_put(tty); 490 } 491 492 wake_up_interruptible(&port->delta_msr_wait); 493 } 494 495 exit: 496 status = usb_submit_urb(urb, GFP_ATOMIC); 497 if (status) 498 dev_err(&urb->dev->dev, 499 "%s - usb_submit_urb failed with result %d\n", 500 __func__, status); 501 } 502 503 static int wait_modem_info(struct usb_serial_port *port, unsigned int arg) 504 { 505 struct ch341_private *priv = usb_get_serial_port_data(port); 506 unsigned long flags; 507 u8 prevstatus; 508 u8 status; 509 u8 changed; 510 u8 multi_change = 0; 511 512 spin_lock_irqsave(&priv->lock, flags); 513 prevstatus = priv->line_status; 514 priv->multi_status_change = 0; 515 spin_unlock_irqrestore(&priv->lock, flags); 516 517 while (!multi_change) { 518 interruptible_sleep_on(&port->delta_msr_wait); 519 /* see if a signal did it */ 520 if (signal_pending(current)) 521 return -ERESTARTSYS; 522 523 if (port->serial->disconnected) 524 return -EIO; 525 526 spin_lock_irqsave(&priv->lock, flags); 527 status = priv->line_status; 528 multi_change = priv->multi_status_change; 529 spin_unlock_irqrestore(&priv->lock, flags); 530 531 changed = prevstatus ^ status; 532 533 if (((arg & TIOCM_RNG) && (changed & CH341_BIT_RI)) || 534 ((arg & TIOCM_DSR) && (changed & CH341_BIT_DSR)) || 535 ((arg & TIOCM_CD) && (changed & CH341_BIT_DCD)) || 536 ((arg & TIOCM_CTS) && (changed & CH341_BIT_CTS))) { 537 return 0; 538 } 539 prevstatus = status; 540 } 541 542 return 0; 543 } 544 545 static int ch341_ioctl(struct tty_struct *tty, 546 unsigned int cmd, unsigned long arg) 547 { 548 struct usb_serial_port *port = tty->driver_data; 549 550 dev_dbg(&port->dev, "%s (%d) cmd = 0x%04x\n", __func__, port->number, cmd); 551 552 switch (cmd) { 553 case TIOCMIWAIT: 554 dev_dbg(&port->dev, "%s (%d) TIOCMIWAIT\n", __func__, port->number); 555 return wait_modem_info(port, arg); 556 557 default: 558 dev_dbg(&port->dev, "%s not supported = 0x%04x\n", __func__, cmd); 559 break; 560 } 561 562 return -ENOIOCTLCMD; 563 } 564 565 static int ch341_tiocmget(struct tty_struct *tty) 566 { 567 struct usb_serial_port *port = tty->driver_data; 568 struct ch341_private *priv = usb_get_serial_port_data(port); 569 unsigned long flags; 570 u8 mcr; 571 u8 status; 572 unsigned int result; 573 574 spin_lock_irqsave(&priv->lock, flags); 575 mcr = priv->line_control; 576 status = priv->line_status; 577 spin_unlock_irqrestore(&priv->lock, flags); 578 579 result = ((mcr & CH341_BIT_DTR) ? TIOCM_DTR : 0) 580 | ((mcr & CH341_BIT_RTS) ? TIOCM_RTS : 0) 581 | ((status & CH341_BIT_CTS) ? TIOCM_CTS : 0) 582 | ((status & CH341_BIT_DSR) ? TIOCM_DSR : 0) 583 | ((status & CH341_BIT_RI) ? TIOCM_RI : 0) 584 | ((status & CH341_BIT_DCD) ? TIOCM_CD : 0); 585 586 dev_dbg(&port->dev, "%s - result = %x\n", __func__, result); 587 588 return result; 589 } 590 591 static int ch341_reset_resume(struct usb_serial *serial) 592 { 593 struct ch341_private *priv; 594 595 priv = usb_get_serial_port_data(serial->port[0]); 596 597 /* reconfigure ch341 serial port after bus-reset */ 598 ch341_configure(serial->dev, priv); 599 600 return 0; 601 } 602 603 static struct usb_serial_driver ch341_device = { 604 .driver = { 605 .owner = THIS_MODULE, 606 .name = "ch341-uart", 607 }, 608 .id_table = id_table, 609 .num_ports = 1, 610 .open = ch341_open, 611 .dtr_rts = ch341_dtr_rts, 612 .carrier_raised = ch341_carrier_raised, 613 .close = ch341_close, 614 .ioctl = ch341_ioctl, 615 .set_termios = ch341_set_termios, 616 .break_ctl = ch341_break_ctl, 617 .tiocmget = ch341_tiocmget, 618 .tiocmset = ch341_tiocmset, 619 .read_int_callback = ch341_read_int_callback, 620 .port_probe = ch341_port_probe, 621 .port_remove = ch341_port_remove, 622 .reset_resume = ch341_reset_resume, 623 }; 624 625 static struct usb_serial_driver * const serial_drivers[] = { 626 &ch341_device, NULL 627 }; 628 629 module_usb_serial_driver(serial_drivers, id_table); 630 631 MODULE_LICENSE("GPL"); 632