xref: /openbmc/linux/drivers/tty/hvc/hvc_console.c (revision f220d3eb)
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