1 // SPDX-License-Identifier: GPL-2.0+ 2 /* 3 * Copyright (C) 2001 Anton Blanchard <anton@au.ibm.com>, IBM 4 * Copyright (C) 2001 Paul Mackerras <paulus@au.ibm.com>, IBM 5 * Copyright (C) 2004 Benjamin Herrenschmidt <benh@kernel.crashing.org>, IBM Corp. 6 * Copyright (C) 2004 IBM Corporation 7 * 8 * Additional Author(s): 9 * Ryan S. Arnold <rsa@us.ibm.com> 10 */ 11 12 #include <linux/console.h> 13 #include <linux/cpumask.h> 14 #include <linux/init.h> 15 #include <linux/kbd_kern.h> 16 #include <linux/kernel.h> 17 #include <linux/kthread.h> 18 #include <linux/list.h> 19 #include <linux/major.h> 20 #include <linux/atomic.h> 21 #include <linux/sysrq.h> 22 #include <linux/tty.h> 23 #include <linux/tty_flip.h> 24 #include <linux/sched.h> 25 #include <linux/spinlock.h> 26 #include <linux/delay.h> 27 #include <linux/freezer.h> 28 #include <linux/slab.h> 29 #include <linux/serial_core.h> 30 31 #include <linux/uaccess.h> 32 33 #include "hvc_console.h" 34 35 #define HVC_MAJOR 229 36 #define HVC_MINOR 0 37 38 /* 39 * Wait this long per iteration while trying to push buffered data to the 40 * hypervisor before allowing the tty to complete a close operation. 41 */ 42 #define HVC_CLOSE_WAIT (HZ/100) /* 1/10 of a second */ 43 44 /* 45 * These sizes are most efficient for vio, because they are the 46 * native transfer size. We could make them selectable in the 47 * future to better deal with backends that want other buffer sizes. 48 */ 49 #define N_OUTBUF 16 50 #define N_INBUF 16 51 52 #define __ALIGNED__ __attribute__((__aligned__(sizeof(long)))) 53 54 static struct tty_driver *hvc_driver; 55 static struct task_struct *hvc_task; 56 57 /* Picks up late kicks after list walk but before schedule() */ 58 static int hvc_kicked; 59 60 /* hvc_init is triggered from hvc_alloc, i.e. only when actually used */ 61 static atomic_t hvc_needs_init __read_mostly = ATOMIC_INIT(-1); 62 63 static int hvc_init(void); 64 65 #ifdef CONFIG_MAGIC_SYSRQ 66 static int sysrq_pressed; 67 #endif 68 69 /* dynamic list of hvc_struct instances */ 70 static LIST_HEAD(hvc_structs); 71 72 /* 73 * Protect the list of hvc_struct instances from inserts and removals during 74 * list traversal. 75 */ 76 static DEFINE_MUTEX(hvc_structs_mutex); 77 78 /* 79 * This value is used to assign a tty->index value to a hvc_struct based 80 * upon order of exposure via hvc_probe(), when we can not match it to 81 * a console candidate registered with hvc_instantiate(). 82 */ 83 static int last_hvc = -1; 84 85 /* 86 * Do not call this function with either the hvc_structs_mutex or the hvc_struct 87 * lock held. If successful, this function increments the kref reference 88 * count against the target hvc_struct so it should be released when finished. 89 */ 90 static struct hvc_struct *hvc_get_by_index(int index) 91 { 92 struct hvc_struct *hp; 93 unsigned long flags; 94 95 mutex_lock(&hvc_structs_mutex); 96 97 list_for_each_entry(hp, &hvc_structs, next) { 98 spin_lock_irqsave(&hp->lock, flags); 99 if (hp->index == index) { 100 tty_port_get(&hp->port); 101 spin_unlock_irqrestore(&hp->lock, flags); 102 mutex_unlock(&hvc_structs_mutex); 103 return hp; 104 } 105 spin_unlock_irqrestore(&hp->lock, flags); 106 } 107 hp = NULL; 108 mutex_unlock(&hvc_structs_mutex); 109 110 return hp; 111 } 112 113 static int __hvc_flush(const struct hv_ops *ops, uint32_t vtermno, bool wait) 114 { 115 if (wait) 116 might_sleep(); 117 118 if (ops->flush) 119 return ops->flush(vtermno, wait); 120 return 0; 121 } 122 123 static int hvc_console_flush(const struct hv_ops *ops, uint32_t vtermno) 124 { 125 return __hvc_flush(ops, vtermno, false); 126 } 127 128 /* 129 * Wait for the console to flush before writing more to it. This sleeps. 130 */ 131 static int hvc_flush(struct hvc_struct *hp) 132 { 133 return __hvc_flush(hp->ops, hp->vtermno, true); 134 } 135 136 /* 137 * Initial console vtermnos for console API usage prior to full console 138 * initialization. Any vty adapter outside this range will not have usable 139 * console interfaces but can still be used as a tty device. This has to be 140 * static because kmalloc will not work during early console init. 141 */ 142 static const struct hv_ops *cons_ops[MAX_NR_HVC_CONSOLES]; 143 static uint32_t vtermnos[MAX_NR_HVC_CONSOLES] = 144 {[0 ... MAX_NR_HVC_CONSOLES - 1] = -1}; 145 146 /* 147 * Console APIs, NOT TTY. These APIs are available immediately when 148 * hvc_console_setup() finds adapters. 149 */ 150 151 static void hvc_console_print(struct console *co, const char *b, 152 unsigned count) 153 { 154 char c[N_OUTBUF] __ALIGNED__; 155 unsigned i = 0, n = 0; 156 int r, donecr = 0, index = co->index; 157 158 /* Console access attempt outside of acceptable console range. */ 159 if (index >= MAX_NR_HVC_CONSOLES) 160 return; 161 162 /* This console adapter was removed so it is not usable. */ 163 if (vtermnos[index] == -1) 164 return; 165 166 while (count > 0 || i > 0) { 167 if (count > 0 && i < sizeof(c)) { 168 if (b[n] == '\n' && !donecr) { 169 c[i++] = '\r'; 170 donecr = 1; 171 } else { 172 c[i++] = b[n++]; 173 donecr = 0; 174 --count; 175 } 176 } else { 177 r = cons_ops[index]->put_chars(vtermnos[index], c, i); 178 if (r <= 0) { 179 /* throw away characters on error 180 * but spin in case of -EAGAIN */ 181 if (r != -EAGAIN) { 182 i = 0; 183 } else { 184 hvc_console_flush(cons_ops[index], 185 vtermnos[index]); 186 } 187 } else if (r > 0) { 188 i -= r; 189 if (i > 0) 190 memmove(c, c+r, i); 191 } 192 } 193 } 194 hvc_console_flush(cons_ops[index], vtermnos[index]); 195 } 196 197 static struct tty_driver *hvc_console_device(struct console *c, int *index) 198 { 199 if (vtermnos[c->index] == -1) 200 return NULL; 201 202 *index = c->index; 203 return hvc_driver; 204 } 205 206 static int hvc_console_setup(struct console *co, char *options) 207 { 208 if (co->index < 0 || co->index >= MAX_NR_HVC_CONSOLES) 209 return -ENODEV; 210 211 if (vtermnos[co->index] == -1) 212 return -ENODEV; 213 214 return 0; 215 } 216 217 static struct console hvc_console = { 218 .name = "hvc", 219 .write = hvc_console_print, 220 .device = hvc_console_device, 221 .setup = hvc_console_setup, 222 .flags = CON_PRINTBUFFER, 223 .index = -1, 224 }; 225 226 /* 227 * Early console initialization. Precedes driver initialization. 228 * 229 * (1) we are first, and the user specified another driver 230 * -- index will remain -1 231 * (2) we are first and the user specified no driver 232 * -- index will be set to 0, then we will fail setup. 233 * (3) we are first and the user specified our driver 234 * -- index will be set to user specified driver, and we will fail 235 * (4) we are after driver, and this initcall will register us 236 * -- if the user didn't specify a driver then the console will match 237 * 238 * Note that for cases 2 and 3, we will match later when the io driver 239 * calls hvc_instantiate() and call register again. 240 */ 241 static int __init hvc_console_init(void) 242 { 243 register_console(&hvc_console); 244 return 0; 245 } 246 console_initcall(hvc_console_init); 247 248 /* callback when the kboject ref count reaches zero. */ 249 static void hvc_port_destruct(struct tty_port *port) 250 { 251 struct hvc_struct *hp = container_of(port, struct hvc_struct, port); 252 unsigned long flags; 253 254 mutex_lock(&hvc_structs_mutex); 255 256 spin_lock_irqsave(&hp->lock, flags); 257 list_del(&(hp->next)); 258 spin_unlock_irqrestore(&hp->lock, flags); 259 260 mutex_unlock(&hvc_structs_mutex); 261 262 kfree(hp); 263 } 264 265 static void hvc_check_console(int index) 266 { 267 /* Already enabled, bail out */ 268 if (hvc_console.flags & CON_ENABLED) 269 return; 270 271 /* If this index is what the user requested, then register 272 * now (setup won't fail at this point). It's ok to just 273 * call register again if previously .setup failed. 274 */ 275 if (index == hvc_console.index) 276 register_console(&hvc_console); 277 } 278 279 /* 280 * hvc_instantiate() is an early console discovery method which locates 281 * consoles * prior to the vio subsystem discovering them. Hotplugged 282 * vty adapters do NOT get an hvc_instantiate() callback since they 283 * appear after early console init. 284 */ 285 int hvc_instantiate(uint32_t vtermno, int index, const struct hv_ops *ops) 286 { 287 struct hvc_struct *hp; 288 289 if (index < 0 || index >= MAX_NR_HVC_CONSOLES) 290 return -1; 291 292 if (vtermnos[index] != -1) 293 return -1; 294 295 /* make sure no no tty has been registered in this index */ 296 hp = hvc_get_by_index(index); 297 if (hp) { 298 tty_port_put(&hp->port); 299 return -1; 300 } 301 302 vtermnos[index] = vtermno; 303 cons_ops[index] = ops; 304 305 /* reserve all indices up to and including this index */ 306 if (last_hvc < index) 307 last_hvc = index; 308 309 /* check if we need to re-register the kernel console */ 310 hvc_check_console(index); 311 312 return 0; 313 } 314 EXPORT_SYMBOL_GPL(hvc_instantiate); 315 316 /* Wake the sleeping khvcd */ 317 void hvc_kick(void) 318 { 319 hvc_kicked = 1; 320 wake_up_process(hvc_task); 321 } 322 EXPORT_SYMBOL_GPL(hvc_kick); 323 324 static void hvc_unthrottle(struct tty_struct *tty) 325 { 326 hvc_kick(); 327 } 328 329 static int hvc_install(struct tty_driver *driver, struct tty_struct *tty) 330 { 331 struct hvc_struct *hp; 332 int rc; 333 334 /* Auto increments kref reference if found. */ 335 hp = hvc_get_by_index(tty->index); 336 if (!hp) 337 return -ENODEV; 338 339 tty->driver_data = hp; 340 341 rc = tty_port_install(&hp->port, driver, tty); 342 if (rc) 343 tty_port_put(&hp->port); 344 return rc; 345 } 346 347 /* 348 * The TTY interface won't be used until after the vio layer has exposed the vty 349 * adapter to the kernel. 350 */ 351 static int hvc_open(struct tty_struct *tty, struct file * filp) 352 { 353 struct hvc_struct *hp = tty->driver_data; 354 unsigned long flags; 355 int rc = 0; 356 357 spin_lock_irqsave(&hp->port.lock, flags); 358 /* Check and then increment for fast path open. */ 359 if (hp->port.count++ > 0) { 360 spin_unlock_irqrestore(&hp->port.lock, flags); 361 hvc_kick(); 362 return 0; 363 } /* else count == 0 */ 364 spin_unlock_irqrestore(&hp->port.lock, flags); 365 366 tty_port_tty_set(&hp->port, tty); 367 368 if (hp->ops->notifier_add) 369 rc = hp->ops->notifier_add(hp, hp->data); 370 371 /* 372 * If the notifier fails we return an error. The tty layer 373 * will call hvc_close() after a failed open but we don't want to clean 374 * up there so we'll clean up here and clear out the previously set 375 * tty fields and return the kref reference. 376 */ 377 if (rc) { 378 tty_port_tty_set(&hp->port, NULL); 379 tty->driver_data = NULL; 380 tty_port_put(&hp->port); 381 printk(KERN_ERR "hvc_open: request_irq failed with rc %d.\n", rc); 382 } else 383 /* We are ready... raise DTR/RTS */ 384 if (C_BAUD(tty)) 385 if (hp->ops->dtr_rts) 386 hp->ops->dtr_rts(hp, 1); 387 388 /* Force wakeup of the polling thread */ 389 hvc_kick(); 390 391 return rc; 392 } 393 394 static void hvc_close(struct tty_struct *tty, struct file * filp) 395 { 396 struct hvc_struct *hp; 397 unsigned long flags; 398 399 if (tty_hung_up_p(filp)) 400 return; 401 402 /* 403 * No driver_data means that this close was issued after a failed 404 * hvc_open by the tty layer's release_dev() function and we can just 405 * exit cleanly because the kref reference wasn't made. 406 */ 407 if (!tty->driver_data) 408 return; 409 410 hp = tty->driver_data; 411 412 spin_lock_irqsave(&hp->port.lock, flags); 413 414 if (--hp->port.count == 0) { 415 spin_unlock_irqrestore(&hp->port.lock, flags); 416 /* We are done with the tty pointer now. */ 417 tty_port_tty_set(&hp->port, NULL); 418 419 if (C_HUPCL(tty)) 420 if (hp->ops->dtr_rts) 421 hp->ops->dtr_rts(hp, 0); 422 423 if (hp->ops->notifier_del) 424 hp->ops->notifier_del(hp, hp->data); 425 426 /* cancel pending tty resize work */ 427 cancel_work_sync(&hp->tty_resize); 428 429 /* 430 * Chain calls chars_in_buffer() and returns immediately if 431 * there is no buffered data otherwise sleeps on a wait queue 432 * waking periodically to check chars_in_buffer(). 433 */ 434 tty_wait_until_sent(tty, HVC_CLOSE_WAIT); 435 } else { 436 if (hp->port.count < 0) 437 printk(KERN_ERR "hvc_close %X: oops, count is %d\n", 438 hp->vtermno, hp->port.count); 439 spin_unlock_irqrestore(&hp->port.lock, flags); 440 } 441 } 442 443 static void hvc_cleanup(struct tty_struct *tty) 444 { 445 struct hvc_struct *hp = tty->driver_data; 446 447 tty_port_put(&hp->port); 448 } 449 450 static void hvc_hangup(struct tty_struct *tty) 451 { 452 struct hvc_struct *hp = tty->driver_data; 453 unsigned long flags; 454 455 if (!hp) 456 return; 457 458 /* cancel pending tty resize work */ 459 cancel_work_sync(&hp->tty_resize); 460 461 spin_lock_irqsave(&hp->port.lock, flags); 462 463 /* 464 * The N_TTY line discipline has problems such that in a close vs 465 * open->hangup case this can be called after the final close so prevent 466 * that from happening for now. 467 */ 468 if (hp->port.count <= 0) { 469 spin_unlock_irqrestore(&hp->port.lock, flags); 470 return; 471 } 472 473 hp->port.count = 0; 474 spin_unlock_irqrestore(&hp->port.lock, flags); 475 tty_port_tty_set(&hp->port, NULL); 476 477 hp->n_outbuf = 0; 478 479 if (hp->ops->notifier_hangup) 480 hp->ops->notifier_hangup(hp, hp->data); 481 } 482 483 /* 484 * Push buffered characters whether they were just recently buffered or waiting 485 * on a blocked hypervisor. Call this function with hp->lock held. 486 */ 487 static int hvc_push(struct hvc_struct *hp) 488 { 489 int n; 490 491 n = hp->ops->put_chars(hp->vtermno, hp->outbuf, hp->n_outbuf); 492 if (n <= 0) { 493 if (n == 0 || n == -EAGAIN) { 494 hp->do_wakeup = 1; 495 return 0; 496 } 497 /* throw away output on error; this happens when 498 there is no session connected to the vterm. */ 499 hp->n_outbuf = 0; 500 } else 501 hp->n_outbuf -= n; 502 if (hp->n_outbuf > 0) 503 memmove(hp->outbuf, hp->outbuf + n, hp->n_outbuf); 504 else 505 hp->do_wakeup = 1; 506 507 return n; 508 } 509 510 static int hvc_write(struct tty_struct *tty, const unsigned char *buf, int count) 511 { 512 struct hvc_struct *hp = tty->driver_data; 513 unsigned long flags; 514 int rsize, written = 0; 515 516 /* This write was probably executed during a tty close. */ 517 if (!hp) 518 return -EPIPE; 519 520 /* FIXME what's this (unprotected) check for? */ 521 if (hp->port.count <= 0) 522 return -EIO; 523 524 while (count > 0) { 525 spin_lock_irqsave(&hp->lock, flags); 526 527 rsize = hp->outbuf_size - hp->n_outbuf; 528 529 if (rsize) { 530 if (rsize > count) 531 rsize = count; 532 memcpy(hp->outbuf + hp->n_outbuf, buf, rsize); 533 count -= rsize; 534 buf += rsize; 535 hp->n_outbuf += rsize; 536 written += rsize; 537 } 538 539 if (hp->n_outbuf > 0) 540 hvc_push(hp); 541 542 spin_unlock_irqrestore(&hp->lock, flags); 543 544 if (count) { 545 if (hp->n_outbuf > 0) 546 hvc_flush(hp); 547 cond_resched(); 548 } 549 } 550 551 /* 552 * Racy, but harmless, kick thread if there is still pending data. 553 */ 554 if (hp->n_outbuf) 555 hvc_kick(); 556 557 return written; 558 } 559 560 /** 561 * hvc_set_winsz() - Resize the hvc tty terminal window. 562 * @work: work structure. 563 * 564 * The routine shall not be called within an atomic context because it 565 * might sleep. 566 * 567 * Locking: hp->lock 568 */ 569 static void hvc_set_winsz(struct work_struct *work) 570 { 571 struct hvc_struct *hp; 572 unsigned long hvc_flags; 573 struct tty_struct *tty; 574 struct winsize ws; 575 576 hp = container_of(work, struct hvc_struct, tty_resize); 577 578 tty = tty_port_tty_get(&hp->port); 579 if (!tty) 580 return; 581 582 spin_lock_irqsave(&hp->lock, hvc_flags); 583 ws = hp->ws; 584 spin_unlock_irqrestore(&hp->lock, hvc_flags); 585 586 tty_do_resize(tty, &ws); 587 tty_kref_put(tty); 588 } 589 590 /* 591 * This is actually a contract between the driver and the tty layer outlining 592 * how much write room the driver can guarantee will be sent OR BUFFERED. This 593 * driver MUST honor the return value. 594 */ 595 static int hvc_write_room(struct tty_struct *tty) 596 { 597 struct hvc_struct *hp = tty->driver_data; 598 599 if (!hp) 600 return 0; 601 602 return hp->outbuf_size - hp->n_outbuf; 603 } 604 605 static int hvc_chars_in_buffer(struct tty_struct *tty) 606 { 607 struct hvc_struct *hp = tty->driver_data; 608 609 if (!hp) 610 return 0; 611 return hp->n_outbuf; 612 } 613 614 /* 615 * timeout will vary between the MIN and MAX values defined here. By default 616 * and during console activity we will use a default MIN_TIMEOUT of 10. When 617 * the console is idle, we increase the timeout value on each pass through 618 * msleep until we reach the max. This may be noticeable as a brief (average 619 * one second) delay on the console before the console responds to input when 620 * there has been no input for some time. 621 */ 622 #define MIN_TIMEOUT (10) 623 #define MAX_TIMEOUT (2000) 624 static u32 timeout = MIN_TIMEOUT; 625 626 #define HVC_POLL_READ 0x00000001 627 #define HVC_POLL_WRITE 0x00000002 628 629 static int __hvc_poll(struct hvc_struct *hp, bool may_sleep) 630 { 631 struct tty_struct *tty; 632 int i, n, count, poll_mask = 0; 633 char buf[N_INBUF] __ALIGNED__; 634 unsigned long flags; 635 int read_total = 0; 636 int written_total = 0; 637 638 spin_lock_irqsave(&hp->lock, flags); 639 640 /* Push pending writes */ 641 if (hp->n_outbuf > 0) 642 written_total = hvc_push(hp); 643 644 /* Reschedule us if still some write pending */ 645 if (hp->n_outbuf > 0) { 646 poll_mask |= HVC_POLL_WRITE; 647 /* If hvc_push() was not able to write, sleep a few msecs */ 648 timeout = (written_total) ? 0 : MIN_TIMEOUT; 649 } 650 651 if (may_sleep) { 652 spin_unlock_irqrestore(&hp->lock, flags); 653 cond_resched(); 654 spin_lock_irqsave(&hp->lock, flags); 655 } 656 657 /* No tty attached, just skip */ 658 tty = tty_port_tty_get(&hp->port); 659 if (tty == NULL) 660 goto bail; 661 662 /* Now check if we can get data (are we throttled ?) */ 663 if (tty_throttled(tty)) 664 goto out; 665 666 /* If we aren't notifier driven and aren't throttled, we always 667 * request a reschedule 668 */ 669 if (!hp->irq_requested) 670 poll_mask |= HVC_POLL_READ; 671 672 /* Read data if any */ 673 674 count = tty_buffer_request_room(&hp->port, N_INBUF); 675 676 /* If flip is full, just reschedule a later read */ 677 if (count == 0) { 678 poll_mask |= HVC_POLL_READ; 679 goto out; 680 } 681 682 n = hp->ops->get_chars(hp->vtermno, buf, count); 683 if (n <= 0) { 684 /* Hangup the tty when disconnected from host */ 685 if (n == -EPIPE) { 686 spin_unlock_irqrestore(&hp->lock, flags); 687 tty_hangup(tty); 688 spin_lock_irqsave(&hp->lock, flags); 689 } else if ( n == -EAGAIN ) { 690 /* 691 * Some back-ends can only ensure a certain min 692 * num of bytes read, which may be > 'count'. 693 * Let the tty clear the flip buff to make room. 694 */ 695 poll_mask |= HVC_POLL_READ; 696 } 697 goto out; 698 } 699 700 for (i = 0; i < n; ++i) { 701 #ifdef CONFIG_MAGIC_SYSRQ 702 if (hp->index == hvc_console.index) { 703 /* Handle the SysRq Hack */ 704 /* XXX should support a sequence */ 705 if (buf[i] == '\x0f') { /* ^O */ 706 /* if ^O is pressed again, reset 707 * sysrq_pressed and flip ^O char */ 708 sysrq_pressed = !sysrq_pressed; 709 if (sysrq_pressed) 710 continue; 711 } else if (sysrq_pressed) { 712 handle_sysrq(buf[i]); 713 sysrq_pressed = 0; 714 continue; 715 } 716 } 717 #endif /* CONFIG_MAGIC_SYSRQ */ 718 tty_insert_flip_char(&hp->port, buf[i], 0); 719 } 720 if (n == count) 721 poll_mask |= HVC_POLL_READ; 722 read_total = n; 723 724 out: 725 /* Wakeup write queue if necessary */ 726 if (hp->do_wakeup) { 727 hp->do_wakeup = 0; 728 tty_wakeup(tty); 729 } 730 bail: 731 spin_unlock_irqrestore(&hp->lock, flags); 732 733 if (read_total) { 734 /* Activity is occurring, so reset the polling backoff value to 735 a minimum for performance. */ 736 timeout = MIN_TIMEOUT; 737 738 tty_flip_buffer_push(&hp->port); 739 } 740 tty_kref_put(tty); 741 742 return poll_mask; 743 } 744 745 int hvc_poll(struct hvc_struct *hp) 746 { 747 return __hvc_poll(hp, false); 748 } 749 EXPORT_SYMBOL_GPL(hvc_poll); 750 751 /** 752 * __hvc_resize() - Update terminal window size information. 753 * @hp: HVC console pointer 754 * @ws: Terminal window size structure 755 * 756 * Stores the specified window size information in the hvc structure of @hp. 757 * The function schedule the tty resize update. 758 * 759 * Locking: Locking free; the function MUST be called holding hp->lock 760 */ 761 void __hvc_resize(struct hvc_struct *hp, struct winsize ws) 762 { 763 hp->ws = ws; 764 schedule_work(&hp->tty_resize); 765 } 766 EXPORT_SYMBOL_GPL(__hvc_resize); 767 768 /* 769 * This kthread is either polling or interrupt driven. This is determined by 770 * calling hvc_poll() who determines whether a console adapter support 771 * interrupts. 772 */ 773 static int khvcd(void *unused) 774 { 775 int poll_mask; 776 struct hvc_struct *hp; 777 778 set_freezable(); 779 do { 780 poll_mask = 0; 781 hvc_kicked = 0; 782 try_to_freeze(); 783 wmb(); 784 if (!cpus_are_in_xmon()) { 785 mutex_lock(&hvc_structs_mutex); 786 list_for_each_entry(hp, &hvc_structs, next) { 787 poll_mask |= __hvc_poll(hp, true); 788 cond_resched(); 789 } 790 mutex_unlock(&hvc_structs_mutex); 791 } else 792 poll_mask |= HVC_POLL_READ; 793 if (hvc_kicked) 794 continue; 795 set_current_state(TASK_INTERRUPTIBLE); 796 if (!hvc_kicked) { 797 if (poll_mask == 0) 798 schedule(); 799 else { 800 unsigned long j_timeout; 801 802 if (timeout < MAX_TIMEOUT) 803 timeout += (timeout >> 6) + 1; 804 805 /* 806 * We don't use msleep_interruptible otherwise 807 * "kick" will fail to wake us up 808 */ 809 j_timeout = msecs_to_jiffies(timeout) + 1; 810 schedule_timeout_interruptible(j_timeout); 811 } 812 } 813 __set_current_state(TASK_RUNNING); 814 } while (!kthread_should_stop()); 815 816 return 0; 817 } 818 819 static int hvc_tiocmget(struct tty_struct *tty) 820 { 821 struct hvc_struct *hp = tty->driver_data; 822 823 if (!hp || !hp->ops->tiocmget) 824 return -EINVAL; 825 return hp->ops->tiocmget(hp); 826 } 827 828 static int hvc_tiocmset(struct tty_struct *tty, 829 unsigned int set, unsigned int clear) 830 { 831 struct hvc_struct *hp = tty->driver_data; 832 833 if (!hp || !hp->ops->tiocmset) 834 return -EINVAL; 835 return hp->ops->tiocmset(hp, set, clear); 836 } 837 838 #ifdef CONFIG_CONSOLE_POLL 839 static int hvc_poll_init(struct tty_driver *driver, int line, char *options) 840 { 841 return 0; 842 } 843 844 static int hvc_poll_get_char(struct tty_driver *driver, int line) 845 { 846 struct tty_struct *tty = driver->ttys[0]; 847 struct hvc_struct *hp = tty->driver_data; 848 int n; 849 char ch; 850 851 n = hp->ops->get_chars(hp->vtermno, &ch, 1); 852 853 if (n <= 0) 854 return NO_POLL_CHAR; 855 856 return ch; 857 } 858 859 static void hvc_poll_put_char(struct tty_driver *driver, int line, char ch) 860 { 861 struct tty_struct *tty = driver->ttys[0]; 862 struct hvc_struct *hp = tty->driver_data; 863 int n; 864 865 do { 866 n = hp->ops->put_chars(hp->vtermno, &ch, 1); 867 } while (n <= 0); 868 } 869 #endif 870 871 static const struct tty_operations hvc_ops = { 872 .install = hvc_install, 873 .open = hvc_open, 874 .close = hvc_close, 875 .cleanup = hvc_cleanup, 876 .write = hvc_write, 877 .hangup = hvc_hangup, 878 .unthrottle = hvc_unthrottle, 879 .write_room = hvc_write_room, 880 .chars_in_buffer = hvc_chars_in_buffer, 881 .tiocmget = hvc_tiocmget, 882 .tiocmset = hvc_tiocmset, 883 #ifdef CONFIG_CONSOLE_POLL 884 .poll_init = hvc_poll_init, 885 .poll_get_char = hvc_poll_get_char, 886 .poll_put_char = hvc_poll_put_char, 887 #endif 888 }; 889 890 static const struct tty_port_operations hvc_port_ops = { 891 .destruct = hvc_port_destruct, 892 }; 893 894 struct hvc_struct *hvc_alloc(uint32_t vtermno, int data, 895 const struct hv_ops *ops, 896 int outbuf_size) 897 { 898 struct hvc_struct *hp; 899 int i; 900 901 /* We wait until a driver actually comes along */ 902 if (atomic_inc_not_zero(&hvc_needs_init)) { 903 int err = hvc_init(); 904 if (err) 905 return ERR_PTR(err); 906 } 907 908 hp = kzalloc(ALIGN(sizeof(*hp), sizeof(long)) + outbuf_size, 909 GFP_KERNEL); 910 if (!hp) 911 return ERR_PTR(-ENOMEM); 912 913 hp->vtermno = vtermno; 914 hp->data = data; 915 hp->ops = ops; 916 hp->outbuf_size = outbuf_size; 917 hp->outbuf = &((char *)hp)[ALIGN(sizeof(*hp), sizeof(long))]; 918 919 tty_port_init(&hp->port); 920 hp->port.ops = &hvc_port_ops; 921 922 INIT_WORK(&hp->tty_resize, hvc_set_winsz); 923 spin_lock_init(&hp->lock); 924 mutex_lock(&hvc_structs_mutex); 925 926 /* 927 * find index to use: 928 * see if this vterm id matches one registered for console. 929 */ 930 for (i=0; i < MAX_NR_HVC_CONSOLES; i++) 931 if (vtermnos[i] == hp->vtermno && 932 cons_ops[i] == hp->ops) 933 break; 934 935 /* no matching slot, just use a counter */ 936 if (i >= MAX_NR_HVC_CONSOLES) 937 i = ++last_hvc; 938 939 hp->index = i; 940 cons_ops[i] = ops; 941 vtermnos[i] = vtermno; 942 943 list_add_tail(&(hp->next), &hvc_structs); 944 mutex_unlock(&hvc_structs_mutex); 945 946 /* check if we need to re-register the kernel console */ 947 hvc_check_console(i); 948 949 return hp; 950 } 951 EXPORT_SYMBOL_GPL(hvc_alloc); 952 953 int hvc_remove(struct hvc_struct *hp) 954 { 955 unsigned long flags; 956 struct tty_struct *tty; 957 958 tty = tty_port_tty_get(&hp->port); 959 960 console_lock(); 961 spin_lock_irqsave(&hp->lock, flags); 962 if (hp->index < MAX_NR_HVC_CONSOLES) { 963 vtermnos[hp->index] = -1; 964 cons_ops[hp->index] = NULL; 965 } 966 967 /* Don't whack hp->irq because tty_hangup() will need to free the irq. */ 968 969 spin_unlock_irqrestore(&hp->lock, flags); 970 console_unlock(); 971 972 /* 973 * We 'put' the instance that was grabbed when the kref instance 974 * was initialized using kref_init(). Let the last holder of this 975 * kref cause it to be removed, which will probably be the tty_vhangup 976 * below. 977 */ 978 tty_port_put(&hp->port); 979 980 /* 981 * This function call will auto chain call hvc_hangup. 982 */ 983 if (tty) { 984 tty_vhangup(tty); 985 tty_kref_put(tty); 986 } 987 return 0; 988 } 989 EXPORT_SYMBOL_GPL(hvc_remove); 990 991 /* Driver initialization: called as soon as someone uses hvc_alloc(). */ 992 static int hvc_init(void) 993 { 994 struct tty_driver *drv; 995 int err; 996 997 /* We need more than hvc_count adapters due to hotplug additions. */ 998 drv = alloc_tty_driver(HVC_ALLOC_TTY_ADAPTERS); 999 if (!drv) { 1000 err = -ENOMEM; 1001 goto out; 1002 } 1003 1004 drv->driver_name = "hvc"; 1005 drv->name = "hvc"; 1006 drv->major = HVC_MAJOR; 1007 drv->minor_start = HVC_MINOR; 1008 drv->type = TTY_DRIVER_TYPE_SYSTEM; 1009 drv->init_termios = tty_std_termios; 1010 drv->flags = TTY_DRIVER_REAL_RAW | TTY_DRIVER_RESET_TERMIOS; 1011 tty_set_operations(drv, &hvc_ops); 1012 1013 /* Always start the kthread because there can be hotplug vty adapters 1014 * added later. */ 1015 hvc_task = kthread_run(khvcd, NULL, "khvcd"); 1016 if (IS_ERR(hvc_task)) { 1017 printk(KERN_ERR "Couldn't create kthread for console.\n"); 1018 err = PTR_ERR(hvc_task); 1019 goto put_tty; 1020 } 1021 1022 err = tty_register_driver(drv); 1023 if (err) { 1024 printk(KERN_ERR "Couldn't register hvc console driver\n"); 1025 goto stop_thread; 1026 } 1027 1028 /* 1029 * Make sure tty is fully registered before allowing it to be 1030 * found by hvc_console_device. 1031 */ 1032 smp_mb(); 1033 hvc_driver = drv; 1034 return 0; 1035 1036 stop_thread: 1037 kthread_stop(hvc_task); 1038 hvc_task = NULL; 1039 put_tty: 1040 put_tty_driver(drv); 1041 out: 1042 return err; 1043 } 1044