xref: /openbmc/linux/drivers/tty/hvc/hvc_console.c (revision b58c6630)
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 	/* check if we need to re-register the kernel console */
306 	hvc_check_console(index);
307 
308 	return 0;
309 }
310 EXPORT_SYMBOL_GPL(hvc_instantiate);
311 
312 /* Wake the sleeping khvcd */
313 void hvc_kick(void)
314 {
315 	hvc_kicked = 1;
316 	wake_up_process(hvc_task);
317 }
318 EXPORT_SYMBOL_GPL(hvc_kick);
319 
320 static void hvc_unthrottle(struct tty_struct *tty)
321 {
322 	hvc_kick();
323 }
324 
325 static int hvc_install(struct tty_driver *driver, struct tty_struct *tty)
326 {
327 	struct hvc_struct *hp;
328 	int rc;
329 
330 	/* Auto increments kref reference if found. */
331 	hp = hvc_get_by_index(tty->index);
332 	if (!hp)
333 		return -ENODEV;
334 
335 	tty->driver_data = hp;
336 
337 	rc = tty_port_install(&hp->port, driver, tty);
338 	if (rc)
339 		tty_port_put(&hp->port);
340 	return rc;
341 }
342 
343 /*
344  * The TTY interface won't be used until after the vio layer has exposed the vty
345  * adapter to the kernel.
346  */
347 static int hvc_open(struct tty_struct *tty, struct file * filp)
348 {
349 	struct hvc_struct *hp = tty->driver_data;
350 	unsigned long flags;
351 	int rc = 0;
352 
353 	spin_lock_irqsave(&hp->port.lock, flags);
354 	/* Check and then increment for fast path open. */
355 	if (hp->port.count++ > 0) {
356 		spin_unlock_irqrestore(&hp->port.lock, flags);
357 		hvc_kick();
358 		return 0;
359 	} /* else count == 0 */
360 	spin_unlock_irqrestore(&hp->port.lock, flags);
361 
362 	tty_port_tty_set(&hp->port, tty);
363 
364 	if (hp->ops->notifier_add)
365 		rc = hp->ops->notifier_add(hp, hp->data);
366 
367 	/*
368 	 * If the notifier fails we return an error.  The tty layer
369 	 * will call hvc_close() after a failed open but we don't want to clean
370 	 * up there so we'll clean up here and clear out the previously set
371 	 * tty fields and return the kref reference.
372 	 */
373 	if (rc) {
374 		tty_port_tty_set(&hp->port, NULL);
375 		tty->driver_data = NULL;
376 		tty_port_put(&hp->port);
377 		printk(KERN_ERR "hvc_open: request_irq failed with rc %d.\n", rc);
378 	} else
379 		/* We are ready... raise DTR/RTS */
380 		if (C_BAUD(tty))
381 			if (hp->ops->dtr_rts)
382 				hp->ops->dtr_rts(hp, 1);
383 
384 	/* Force wakeup of the polling thread */
385 	hvc_kick();
386 
387 	return rc;
388 }
389 
390 static void hvc_close(struct tty_struct *tty, struct file * filp)
391 {
392 	struct hvc_struct *hp;
393 	unsigned long flags;
394 
395 	if (tty_hung_up_p(filp))
396 		return;
397 
398 	/*
399 	 * No driver_data means that this close was issued after a failed
400 	 * hvc_open by the tty layer's release_dev() function and we can just
401 	 * exit cleanly because the kref reference wasn't made.
402 	 */
403 	if (!tty->driver_data)
404 		return;
405 
406 	hp = tty->driver_data;
407 
408 	spin_lock_irqsave(&hp->port.lock, flags);
409 
410 	if (--hp->port.count == 0) {
411 		spin_unlock_irqrestore(&hp->port.lock, flags);
412 		/* We are done with the tty pointer now. */
413 		tty_port_tty_set(&hp->port, NULL);
414 
415 		if (C_HUPCL(tty))
416 			if (hp->ops->dtr_rts)
417 				hp->ops->dtr_rts(hp, 0);
418 
419 		if (hp->ops->notifier_del)
420 			hp->ops->notifier_del(hp, hp->data);
421 
422 		/* cancel pending tty resize work */
423 		cancel_work_sync(&hp->tty_resize);
424 
425 		/*
426 		 * Chain calls chars_in_buffer() and returns immediately if
427 		 * there is no buffered data otherwise sleeps on a wait queue
428 		 * waking periodically to check chars_in_buffer().
429 		 */
430 		tty_wait_until_sent(tty, HVC_CLOSE_WAIT);
431 	} else {
432 		if (hp->port.count < 0)
433 			printk(KERN_ERR "hvc_close %X: oops, count is %d\n",
434 				hp->vtermno, hp->port.count);
435 		spin_unlock_irqrestore(&hp->port.lock, flags);
436 	}
437 }
438 
439 static void hvc_cleanup(struct tty_struct *tty)
440 {
441 	struct hvc_struct *hp = tty->driver_data;
442 
443 	tty_port_put(&hp->port);
444 }
445 
446 static void hvc_hangup(struct tty_struct *tty)
447 {
448 	struct hvc_struct *hp = tty->driver_data;
449 	unsigned long flags;
450 
451 	if (!hp)
452 		return;
453 
454 	/* cancel pending tty resize work */
455 	cancel_work_sync(&hp->tty_resize);
456 
457 	spin_lock_irqsave(&hp->port.lock, flags);
458 
459 	/*
460 	 * The N_TTY line discipline has problems such that in a close vs
461 	 * open->hangup case this can be called after the final close so prevent
462 	 * that from happening for now.
463 	 */
464 	if (hp->port.count <= 0) {
465 		spin_unlock_irqrestore(&hp->port.lock, flags);
466 		return;
467 	}
468 
469 	hp->port.count = 0;
470 	spin_unlock_irqrestore(&hp->port.lock, flags);
471 	tty_port_tty_set(&hp->port, NULL);
472 
473 	hp->n_outbuf = 0;
474 
475 	if (hp->ops->notifier_hangup)
476 		hp->ops->notifier_hangup(hp, hp->data);
477 }
478 
479 /*
480  * Push buffered characters whether they were just recently buffered or waiting
481  * on a blocked hypervisor.  Call this function with hp->lock held.
482  */
483 static int hvc_push(struct hvc_struct *hp)
484 {
485 	int n;
486 
487 	n = hp->ops->put_chars(hp->vtermno, hp->outbuf, hp->n_outbuf);
488 	if (n <= 0) {
489 		if (n == 0 || n == -EAGAIN) {
490 			hp->do_wakeup = 1;
491 			return 0;
492 		}
493 		/* throw away output on error; this happens when
494 		   there is no session connected to the vterm. */
495 		hp->n_outbuf = 0;
496 	} else
497 		hp->n_outbuf -= n;
498 	if (hp->n_outbuf > 0)
499 		memmove(hp->outbuf, hp->outbuf + n, hp->n_outbuf);
500 	else
501 		hp->do_wakeup = 1;
502 
503 	return n;
504 }
505 
506 static int hvc_write(struct tty_struct *tty, const unsigned char *buf, int count)
507 {
508 	struct hvc_struct *hp = tty->driver_data;
509 	unsigned long flags;
510 	int rsize, written = 0;
511 
512 	/* This write was probably executed during a tty close. */
513 	if (!hp)
514 		return -EPIPE;
515 
516 	/* FIXME what's this (unprotected) check for? */
517 	if (hp->port.count <= 0)
518 		return -EIO;
519 
520 	while (count > 0) {
521 		int ret = 0;
522 
523 		spin_lock_irqsave(&hp->lock, flags);
524 
525 		rsize = hp->outbuf_size - hp->n_outbuf;
526 
527 		if (rsize) {
528 			if (rsize > count)
529 				rsize = count;
530 			memcpy(hp->outbuf + hp->n_outbuf, buf, rsize);
531 			count -= rsize;
532 			buf += rsize;
533 			hp->n_outbuf += rsize;
534 			written += rsize;
535 		}
536 
537 		if (hp->n_outbuf > 0)
538 			ret = hvc_push(hp);
539 
540 		spin_unlock_irqrestore(&hp->lock, flags);
541 
542 		if (!ret)
543 			break;
544 
545 		if (count) {
546 			if (hp->n_outbuf > 0)
547 				hvc_flush(hp);
548 			cond_resched();
549 		}
550 	}
551 
552 	/*
553 	 * Racy, but harmless, kick thread if there is still pending data.
554 	 */
555 	if (hp->n_outbuf)
556 		hvc_kick();
557 
558 	return written;
559 }
560 
561 /**
562  * hvc_set_winsz() - Resize the hvc tty terminal window.
563  * @work:	work structure.
564  *
565  * The routine shall not be called within an atomic context because it
566  * might sleep.
567  *
568  * Locking:	hp->lock
569  */
570 static void hvc_set_winsz(struct work_struct *work)
571 {
572 	struct hvc_struct *hp;
573 	unsigned long hvc_flags;
574 	struct tty_struct *tty;
575 	struct winsize ws;
576 
577 	hp = container_of(work, struct hvc_struct, tty_resize);
578 
579 	tty = tty_port_tty_get(&hp->port);
580 	if (!tty)
581 		return;
582 
583 	spin_lock_irqsave(&hp->lock, hvc_flags);
584 	ws = hp->ws;
585 	spin_unlock_irqrestore(&hp->lock, hvc_flags);
586 
587 	tty_do_resize(tty, &ws);
588 	tty_kref_put(tty);
589 }
590 
591 /*
592  * This is actually a contract between the driver and the tty layer outlining
593  * how much write room the driver can guarantee will be sent OR BUFFERED.  This
594  * driver MUST honor the return value.
595  */
596 static int hvc_write_room(struct tty_struct *tty)
597 {
598 	struct hvc_struct *hp = tty->driver_data;
599 
600 	if (!hp)
601 		return 0;
602 
603 	return hp->outbuf_size - hp->n_outbuf;
604 }
605 
606 static int hvc_chars_in_buffer(struct tty_struct *tty)
607 {
608 	struct hvc_struct *hp = tty->driver_data;
609 
610 	if (!hp)
611 		return 0;
612 	return hp->n_outbuf;
613 }
614 
615 /*
616  * timeout will vary between the MIN and MAX values defined here.  By default
617  * and during console activity we will use a default MIN_TIMEOUT of 10.  When
618  * the console is idle, we increase the timeout value on each pass through
619  * msleep until we reach the max.  This may be noticeable as a brief (average
620  * one second) delay on the console before the console responds to input when
621  * there has been no input for some time.
622  */
623 #define MIN_TIMEOUT		(10)
624 #define MAX_TIMEOUT		(2000)
625 static u32 timeout = MIN_TIMEOUT;
626 
627 /*
628  * Maximum number of bytes to get from the console driver if hvc_poll is
629  * called from driver (and can't sleep). Any more than this and we break
630  * and start polling with khvcd. This value was derived from from an OpenBMC
631  * console with the OPAL driver that results in about 0.25ms interrupts off
632  * latency.
633  */
634 #define HVC_ATOMIC_READ_MAX	128
635 
636 #define HVC_POLL_READ	0x00000001
637 #define HVC_POLL_WRITE	0x00000002
638 
639 static int __hvc_poll(struct hvc_struct *hp, bool may_sleep)
640 {
641 	struct tty_struct *tty;
642 	int i, n, count, poll_mask = 0;
643 	char buf[N_INBUF] __ALIGNED__;
644 	unsigned long flags;
645 	int read_total = 0;
646 	int written_total = 0;
647 
648 	spin_lock_irqsave(&hp->lock, flags);
649 
650 	/* Push pending writes */
651 	if (hp->n_outbuf > 0)
652 		written_total = hvc_push(hp);
653 
654 	/* Reschedule us if still some write pending */
655 	if (hp->n_outbuf > 0) {
656 		poll_mask |= HVC_POLL_WRITE;
657 		/* If hvc_push() was not able to write, sleep a few msecs */
658 		timeout = (written_total) ? 0 : MIN_TIMEOUT;
659 	}
660 
661 	if (may_sleep) {
662 		spin_unlock_irqrestore(&hp->lock, flags);
663 		cond_resched();
664 		spin_lock_irqsave(&hp->lock, flags);
665 	}
666 
667 	/* No tty attached, just skip */
668 	tty = tty_port_tty_get(&hp->port);
669 	if (tty == NULL)
670 		goto bail;
671 
672 	/* Now check if we can get data (are we throttled ?) */
673 	if (tty_throttled(tty))
674 		goto out;
675 
676 	/* If we aren't notifier driven and aren't throttled, we always
677 	 * request a reschedule
678 	 */
679 	if (!hp->irq_requested)
680 		poll_mask |= HVC_POLL_READ;
681 
682  read_again:
683 	/* Read data if any */
684 	count = tty_buffer_request_room(&hp->port, N_INBUF);
685 
686 	/* If flip is full, just reschedule a later read */
687 	if (count == 0) {
688 		poll_mask |= HVC_POLL_READ;
689 		goto out;
690 	}
691 
692 	n = hp->ops->get_chars(hp->vtermno, buf, count);
693 	if (n <= 0) {
694 		/* Hangup the tty when disconnected from host */
695 		if (n == -EPIPE) {
696 			spin_unlock_irqrestore(&hp->lock, flags);
697 			tty_hangup(tty);
698 			spin_lock_irqsave(&hp->lock, flags);
699 		} else if ( n == -EAGAIN ) {
700 			/*
701 			 * Some back-ends can only ensure a certain min
702 			 * num of bytes read, which may be > 'count'.
703 			 * Let the tty clear the flip buff to make room.
704 			 */
705 			poll_mask |= HVC_POLL_READ;
706 		}
707 		goto out;
708 	}
709 
710 	for (i = 0; i < n; ++i) {
711 #ifdef CONFIG_MAGIC_SYSRQ
712 		if (hp->index == hvc_console.index) {
713 			/* Handle the SysRq Hack */
714 			/* XXX should support a sequence */
715 			if (buf[i] == '\x0f') {	/* ^O */
716 				/* if ^O is pressed again, reset
717 				 * sysrq_pressed and flip ^O char */
718 				sysrq_pressed = !sysrq_pressed;
719 				if (sysrq_pressed)
720 					continue;
721 			} else if (sysrq_pressed) {
722 				handle_sysrq(buf[i]);
723 				sysrq_pressed = 0;
724 				continue;
725 			}
726 		}
727 #endif /* CONFIG_MAGIC_SYSRQ */
728 		tty_insert_flip_char(&hp->port, buf[i], 0);
729 	}
730 	read_total += n;
731 
732 	if (may_sleep) {
733 		/* Keep going until the flip is full */
734 		spin_unlock_irqrestore(&hp->lock, flags);
735 		cond_resched();
736 		spin_lock_irqsave(&hp->lock, flags);
737 		goto read_again;
738 	} else if (read_total < HVC_ATOMIC_READ_MAX) {
739 		/* Break and defer if it's a large read in atomic */
740 		goto read_again;
741 	}
742 
743 	/*
744 	 * Latency break, schedule another poll immediately.
745 	 */
746 	poll_mask |= HVC_POLL_READ;
747 
748  out:
749 	/* Wakeup write queue if necessary */
750 	if (hp->do_wakeup) {
751 		hp->do_wakeup = 0;
752 		tty_wakeup(tty);
753 	}
754  bail:
755 	spin_unlock_irqrestore(&hp->lock, flags);
756 
757 	if (read_total) {
758 		/* Activity is occurring, so reset the polling backoff value to
759 		   a minimum for performance. */
760 		timeout = MIN_TIMEOUT;
761 
762 		tty_flip_buffer_push(&hp->port);
763 	}
764 	tty_kref_put(tty);
765 
766 	return poll_mask;
767 }
768 
769 int hvc_poll(struct hvc_struct *hp)
770 {
771 	return __hvc_poll(hp, false);
772 }
773 EXPORT_SYMBOL_GPL(hvc_poll);
774 
775 /**
776  * __hvc_resize() - Update terminal window size information.
777  * @hp:		HVC console pointer
778  * @ws:		Terminal window size structure
779  *
780  * Stores the specified window size information in the hvc structure of @hp.
781  * The function schedule the tty resize update.
782  *
783  * Locking:	Locking free; the function MUST be called holding hp->lock
784  */
785 void __hvc_resize(struct hvc_struct *hp, struct winsize ws)
786 {
787 	hp->ws = ws;
788 	schedule_work(&hp->tty_resize);
789 }
790 EXPORT_SYMBOL_GPL(__hvc_resize);
791 
792 /*
793  * This kthread is either polling or interrupt driven.  This is determined by
794  * calling hvc_poll() who determines whether a console adapter support
795  * interrupts.
796  */
797 static int khvcd(void *unused)
798 {
799 	int poll_mask;
800 	struct hvc_struct *hp;
801 
802 	set_freezable();
803 	do {
804 		poll_mask = 0;
805 		hvc_kicked = 0;
806 		try_to_freeze();
807 		wmb();
808 		if (!cpus_are_in_xmon()) {
809 			mutex_lock(&hvc_structs_mutex);
810 			list_for_each_entry(hp, &hvc_structs, next) {
811 				poll_mask |= __hvc_poll(hp, true);
812 				cond_resched();
813 			}
814 			mutex_unlock(&hvc_structs_mutex);
815 		} else
816 			poll_mask |= HVC_POLL_READ;
817 		if (hvc_kicked)
818 			continue;
819 		set_current_state(TASK_INTERRUPTIBLE);
820 		if (!hvc_kicked) {
821 			if (poll_mask == 0)
822 				schedule();
823 			else {
824 				unsigned long j_timeout;
825 
826 				if (timeout < MAX_TIMEOUT)
827 					timeout += (timeout >> 6) + 1;
828 
829 				/*
830 				 * We don't use msleep_interruptible otherwise
831 				 * "kick" will fail to wake us up
832 				 */
833 				j_timeout = msecs_to_jiffies(timeout) + 1;
834 				schedule_timeout_interruptible(j_timeout);
835 			}
836 		}
837 		__set_current_state(TASK_RUNNING);
838 	} while (!kthread_should_stop());
839 
840 	return 0;
841 }
842 
843 static int hvc_tiocmget(struct tty_struct *tty)
844 {
845 	struct hvc_struct *hp = tty->driver_data;
846 
847 	if (!hp || !hp->ops->tiocmget)
848 		return -EINVAL;
849 	return hp->ops->tiocmget(hp);
850 }
851 
852 static int hvc_tiocmset(struct tty_struct *tty,
853 			unsigned int set, unsigned int clear)
854 {
855 	struct hvc_struct *hp = tty->driver_data;
856 
857 	if (!hp || !hp->ops->tiocmset)
858 		return -EINVAL;
859 	return hp->ops->tiocmset(hp, set, clear);
860 }
861 
862 #ifdef CONFIG_CONSOLE_POLL
863 static int hvc_poll_init(struct tty_driver *driver, int line, char *options)
864 {
865 	return 0;
866 }
867 
868 static int hvc_poll_get_char(struct tty_driver *driver, int line)
869 {
870 	struct tty_struct *tty = driver->ttys[0];
871 	struct hvc_struct *hp = tty->driver_data;
872 	int n;
873 	char ch;
874 
875 	n = hp->ops->get_chars(hp->vtermno, &ch, 1);
876 
877 	if (n <= 0)
878 		return NO_POLL_CHAR;
879 
880 	return ch;
881 }
882 
883 static void hvc_poll_put_char(struct tty_driver *driver, int line, char ch)
884 {
885 	struct tty_struct *tty = driver->ttys[0];
886 	struct hvc_struct *hp = tty->driver_data;
887 	int n;
888 
889 	do {
890 		n = hp->ops->put_chars(hp->vtermno, &ch, 1);
891 	} while (n <= 0);
892 }
893 #endif
894 
895 static const struct tty_operations hvc_ops = {
896 	.install = hvc_install,
897 	.open = hvc_open,
898 	.close = hvc_close,
899 	.cleanup = hvc_cleanup,
900 	.write = hvc_write,
901 	.hangup = hvc_hangup,
902 	.unthrottle = hvc_unthrottle,
903 	.write_room = hvc_write_room,
904 	.chars_in_buffer = hvc_chars_in_buffer,
905 	.tiocmget = hvc_tiocmget,
906 	.tiocmset = hvc_tiocmset,
907 #ifdef CONFIG_CONSOLE_POLL
908 	.poll_init = hvc_poll_init,
909 	.poll_get_char = hvc_poll_get_char,
910 	.poll_put_char = hvc_poll_put_char,
911 #endif
912 };
913 
914 static const struct tty_port_operations hvc_port_ops = {
915 	.destruct = hvc_port_destruct,
916 };
917 
918 struct hvc_struct *hvc_alloc(uint32_t vtermno, int data,
919 			     const struct hv_ops *ops,
920 			     int outbuf_size)
921 {
922 	struct hvc_struct *hp;
923 	int i;
924 
925 	/* We wait until a driver actually comes along */
926 	if (atomic_inc_not_zero(&hvc_needs_init)) {
927 		int err = hvc_init();
928 		if (err)
929 			return ERR_PTR(err);
930 	}
931 
932 	hp = kzalloc(ALIGN(sizeof(*hp), sizeof(long)) + outbuf_size,
933 			GFP_KERNEL);
934 	if (!hp)
935 		return ERR_PTR(-ENOMEM);
936 
937 	hp->vtermno = vtermno;
938 	hp->data = data;
939 	hp->ops = ops;
940 	hp->outbuf_size = outbuf_size;
941 	hp->outbuf = &((char *)hp)[ALIGN(sizeof(*hp), sizeof(long))];
942 
943 	tty_port_init(&hp->port);
944 	hp->port.ops = &hvc_port_ops;
945 
946 	INIT_WORK(&hp->tty_resize, hvc_set_winsz);
947 	spin_lock_init(&hp->lock);
948 	mutex_lock(&hvc_structs_mutex);
949 
950 	/*
951 	 * find index to use:
952 	 * see if this vterm id matches one registered for console.
953 	 */
954 	for (i=0; i < MAX_NR_HVC_CONSOLES; i++)
955 		if (vtermnos[i] == hp->vtermno &&
956 		    cons_ops[i] == hp->ops)
957 			break;
958 
959 	if (i >= MAX_NR_HVC_CONSOLES) {
960 
961 		/* find 'empty' slot for console */
962 		for (i = 0; i < MAX_NR_HVC_CONSOLES && vtermnos[i] != -1; i++) {
963 		}
964 
965 		/* no matching slot, just use a counter */
966 		if (i == MAX_NR_HVC_CONSOLES)
967 			i = ++last_hvc + MAX_NR_HVC_CONSOLES;
968 	}
969 
970 	hp->index = i;
971 	if (i < MAX_NR_HVC_CONSOLES) {
972 		cons_ops[i] = ops;
973 		vtermnos[i] = vtermno;
974 	}
975 
976 	list_add_tail(&(hp->next), &hvc_structs);
977 	mutex_unlock(&hvc_structs_mutex);
978 
979 	/* check if we need to re-register the kernel console */
980 	hvc_check_console(i);
981 
982 	return hp;
983 }
984 EXPORT_SYMBOL_GPL(hvc_alloc);
985 
986 int hvc_remove(struct hvc_struct *hp)
987 {
988 	unsigned long flags;
989 	struct tty_struct *tty;
990 
991 	tty = tty_port_tty_get(&hp->port);
992 
993 	console_lock();
994 	spin_lock_irqsave(&hp->lock, flags);
995 	if (hp->index < MAX_NR_HVC_CONSOLES) {
996 		vtermnos[hp->index] = -1;
997 		cons_ops[hp->index] = NULL;
998 	}
999 
1000 	/* Don't whack hp->irq because tty_hangup() will need to free the irq. */
1001 
1002 	spin_unlock_irqrestore(&hp->lock, flags);
1003 	console_unlock();
1004 
1005 	/*
1006 	 * We 'put' the instance that was grabbed when the kref instance
1007 	 * was initialized using kref_init().  Let the last holder of this
1008 	 * kref cause it to be removed, which will probably be the tty_vhangup
1009 	 * below.
1010 	 */
1011 	tty_port_put(&hp->port);
1012 
1013 	/*
1014 	 * This function call will auto chain call hvc_hangup.
1015 	 */
1016 	if (tty) {
1017 		tty_vhangup(tty);
1018 		tty_kref_put(tty);
1019 	}
1020 	return 0;
1021 }
1022 EXPORT_SYMBOL_GPL(hvc_remove);
1023 
1024 /* Driver initialization: called as soon as someone uses hvc_alloc(). */
1025 static int hvc_init(void)
1026 {
1027 	struct tty_driver *drv;
1028 	int err;
1029 
1030 	/* We need more than hvc_count adapters due to hotplug additions. */
1031 	drv = alloc_tty_driver(HVC_ALLOC_TTY_ADAPTERS);
1032 	if (!drv) {
1033 		err = -ENOMEM;
1034 		goto out;
1035 	}
1036 
1037 	drv->driver_name = "hvc";
1038 	drv->name = "hvc";
1039 	drv->major = HVC_MAJOR;
1040 	drv->minor_start = HVC_MINOR;
1041 	drv->type = TTY_DRIVER_TYPE_SYSTEM;
1042 	drv->init_termios = tty_std_termios;
1043 	drv->flags = TTY_DRIVER_REAL_RAW | TTY_DRIVER_RESET_TERMIOS;
1044 	tty_set_operations(drv, &hvc_ops);
1045 
1046 	/* Always start the kthread because there can be hotplug vty adapters
1047 	 * added later. */
1048 	hvc_task = kthread_run(khvcd, NULL, "khvcd");
1049 	if (IS_ERR(hvc_task)) {
1050 		printk(KERN_ERR "Couldn't create kthread for console.\n");
1051 		err = PTR_ERR(hvc_task);
1052 		goto put_tty;
1053 	}
1054 
1055 	err = tty_register_driver(drv);
1056 	if (err) {
1057 		printk(KERN_ERR "Couldn't register hvc console driver\n");
1058 		goto stop_thread;
1059 	}
1060 
1061 	/*
1062 	 * Make sure tty is fully registered before allowing it to be
1063 	 * found by hvc_console_device.
1064 	 */
1065 	smp_mb();
1066 	hvc_driver = drv;
1067 	return 0;
1068 
1069 stop_thread:
1070 	kthread_stop(hvc_task);
1071 	hvc_task = NULL;
1072 put_tty:
1073 	put_tty_driver(drv);
1074 out:
1075 	return err;
1076 }
1077