Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 1 | /* |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 2 | * Copyright (C) 2001 - 2007 Jeff Dike (jdike@{linux.intel,addtoit}.com) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 3 | * Licensed under the GPL |
| 4 | */ |
| 5 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 6 | #include "linux/completion.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 7 | #include "linux/interrupt.h" |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 8 | #include "linux/list.h" |
Daniel Walker | 2aa9c5d | 2008-02-04 22:31:27 -0800 | [diff] [blame] | 9 | #include "linux/mutex.h" |
Tejun Heo | 5a0e3ad | 2010-03-24 17:04:11 +0900 | [diff] [blame] | 10 | #include "linux/slab.h" |
Alexey Dobriyan | d43c36d | 2009-10-07 17:09:06 +0400 | [diff] [blame] | 11 | #include "linux/workqueue.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 12 | #include "asm/atomic.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 13 | #include "init.h" |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 14 | #include "irq_kern.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 15 | #include "os.h" |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 16 | #include "port.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 17 | |
| 18 | struct port_list { |
| 19 | struct list_head list; |
| 20 | atomic_t wait_count; |
| 21 | int has_connection; |
| 22 | struct completion done; |
| 23 | int port; |
| 24 | int fd; |
| 25 | spinlock_t lock; |
| 26 | struct list_head pending; |
| 27 | struct list_head connections; |
| 28 | }; |
| 29 | |
| 30 | struct port_dev { |
| 31 | struct port_list *port; |
| 32 | int helper_pid; |
| 33 | int telnetd_pid; |
| 34 | }; |
| 35 | |
| 36 | struct connection { |
| 37 | struct list_head list; |
| 38 | int fd; |
| 39 | int helper_pid; |
| 40 | int socket[2]; |
| 41 | int telnetd_pid; |
| 42 | struct port_list *port; |
| 43 | }; |
| 44 | |
Al Viro | 7bea96f | 2006-10-08 22:49:34 +0100 | [diff] [blame] | 45 | static irqreturn_t pipe_interrupt(int irq, void *data) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 46 | { |
| 47 | struct connection *conn = data; |
| 48 | int fd; |
| 49 | |
| 50 | fd = os_rcv_fd(conn->socket[0], &conn->helper_pid); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 51 | if (fd < 0) { |
| 52 | if (fd == -EAGAIN) |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 53 | return IRQ_NONE; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 54 | |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 55 | printk(KERN_ERR "pipe_interrupt : os_rcv_fd returned %d\n", |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 56 | -fd); |
| 57 | os_close_file(conn->fd); |
| 58 | } |
| 59 | |
| 60 | list_del(&conn->list); |
| 61 | |
| 62 | conn->fd = fd; |
| 63 | list_add(&conn->list, &conn->port->connections); |
| 64 | |
| 65 | complete(&conn->port->done); |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 66 | return IRQ_HANDLED; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 67 | } |
| 68 | |
| 69 | #define NO_WAITER_MSG \ |
| 70 | "****\n" \ |
| 71 | "There are currently no UML consoles waiting for port connections.\n" \ |
| 72 | "Either disconnect from one to make it available or activate some more\n" \ |
| 73 | "by enabling more consoles in the UML /etc/inittab.\n" \ |
| 74 | "****\n" |
| 75 | |
| 76 | static int port_accept(struct port_list *port) |
| 77 | { |
| 78 | struct connection *conn; |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 79 | int fd, socket[2], pid; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 80 | |
| 81 | fd = port_connection(port->fd, socket, &pid); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 82 | if (fd < 0) { |
| 83 | if (fd != -EAGAIN) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 84 | printk(KERN_ERR "port_accept : port_connection " |
| 85 | "returned %d\n", -fd); |
| 86 | goto out; |
| 87 | } |
| 88 | |
| 89 | conn = kmalloc(sizeof(*conn), GFP_ATOMIC); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 90 | if (conn == NULL) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 91 | printk(KERN_ERR "port_accept : failed to allocate " |
| 92 | "connection\n"); |
| 93 | goto out_close; |
| 94 | } |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 95 | *conn = ((struct connection) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 96 | { .list = LIST_HEAD_INIT(conn->list), |
| 97 | .fd = fd, |
| 98 | .socket = { socket[0], socket[1] }, |
| 99 | .telnetd_pid = pid, |
| 100 | .port = port }); |
| 101 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 102 | if (um_request_irq(TELNETD_IRQ, socket[0], IRQ_READ, pipe_interrupt, |
Yong Zhang | c0b79a9 | 2011-09-22 16:58:46 +0800 | [diff] [blame] | 103 | IRQF_SHARED | IRQF_SAMPLE_RANDOM, |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 104 | "telnetd", conn)) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 105 | printk(KERN_ERR "port_accept : failed to get IRQ for " |
| 106 | "telnetd\n"); |
| 107 | goto out_free; |
| 108 | } |
| 109 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 110 | if (atomic_read(&port->wait_count) == 0) { |
Jeff Dike | a6ea4cc | 2007-05-06 14:51:43 -0700 | [diff] [blame] | 111 | os_write_file(fd, NO_WAITER_MSG, sizeof(NO_WAITER_MSG)); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 112 | printk(KERN_ERR "No one waiting for port\n"); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 113 | } |
| 114 | list_add(&conn->list, &port->pending); |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 115 | return 1; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 116 | |
| 117 | out_free: |
| 118 | kfree(conn); |
| 119 | out_close: |
| 120 | os_close_file(fd); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 121 | os_kill_process(pid, 1); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 122 | out: |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 123 | return 0; |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 124 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 125 | |
Daniel Walker | 2aa9c5d | 2008-02-04 22:31:27 -0800 | [diff] [blame] | 126 | static DEFINE_MUTEX(ports_mutex); |
Jeff Dike | c59bce6 | 2007-02-10 01:44:04 -0800 | [diff] [blame] | 127 | static LIST_HEAD(ports); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 128 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 129 | static void port_work_proc(struct work_struct *unused) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 130 | { |
| 131 | struct port_list *port; |
| 132 | struct list_head *ele; |
| 133 | unsigned long flags; |
| 134 | |
| 135 | local_irq_save(flags); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 136 | list_for_each(ele, &ports) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 137 | port = list_entry(ele, struct port_list, list); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 138 | if (!port->has_connection) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 139 | continue; |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 140 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 141 | reactivate_fd(port->fd, ACCEPT_IRQ); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 142 | while (port_accept(port)) |
| 143 | ; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 144 | port->has_connection = 0; |
| 145 | } |
| 146 | local_irq_restore(flags); |
| 147 | } |
| 148 | |
David Howells | 6d5aefb | 2006-12-05 19:36:26 +0000 | [diff] [blame] | 149 | DECLARE_WORK(port_work, port_work_proc); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 150 | |
Al Viro | 7bea96f | 2006-10-08 22:49:34 +0100 | [diff] [blame] | 151 | static irqreturn_t port_interrupt(int irq, void *data) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 152 | { |
| 153 | struct port_list *port = data; |
| 154 | |
| 155 | port->has_connection = 1; |
| 156 | schedule_work(&port_work); |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 157 | return IRQ_HANDLED; |
| 158 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 159 | |
| 160 | void *port_data(int port_num) |
| 161 | { |
| 162 | struct list_head *ele; |
| 163 | struct port_list *port; |
| 164 | struct port_dev *dev = NULL; |
| 165 | int fd; |
| 166 | |
Daniel Walker | 2aa9c5d | 2008-02-04 22:31:27 -0800 | [diff] [blame] | 167 | mutex_lock(&ports_mutex); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 168 | list_for_each(ele, &ports) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 169 | port = list_entry(ele, struct port_list, list); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 170 | if (port->port == port_num) |
| 171 | goto found; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 172 | } |
| 173 | port = kmalloc(sizeof(struct port_list), GFP_KERNEL); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 174 | if (port == NULL) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 175 | printk(KERN_ERR "Allocation of port list failed\n"); |
| 176 | goto out; |
| 177 | } |
| 178 | |
| 179 | fd = port_listen_fd(port_num); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 180 | if (fd < 0) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 181 | printk(KERN_ERR "binding to port %d failed, errno = %d\n", |
| 182 | port_num, -fd); |
| 183 | goto out_free; |
| 184 | } |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 185 | |
| 186 | if (um_request_irq(ACCEPT_IRQ, fd, IRQ_READ, port_interrupt, |
Yong Zhang | c0b79a9 | 2011-09-22 16:58:46 +0800 | [diff] [blame] | 187 | IRQF_SHARED | IRQF_SAMPLE_RANDOM, |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 188 | "port", port)) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 189 | printk(KERN_ERR "Failed to get IRQ for port %d\n", port_num); |
| 190 | goto out_close; |
| 191 | } |
| 192 | |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 193 | *port = ((struct port_list) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 194 | { .list = LIST_HEAD_INIT(port->list), |
| 195 | .wait_count = ATOMIC_INIT(0), |
| 196 | .has_connection = 0, |
| 197 | .port = port_num, |
| 198 | .fd = fd, |
| 199 | .pending = LIST_HEAD_INIT(port->pending), |
| 200 | .connections = LIST_HEAD_INIT(port->connections) }); |
| 201 | spin_lock_init(&port->lock); |
| 202 | init_completion(&port->done); |
| 203 | list_add(&port->list, &ports); |
| 204 | |
| 205 | found: |
| 206 | dev = kmalloc(sizeof(struct port_dev), GFP_KERNEL); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 207 | if (dev == NULL) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 208 | printk(KERN_ERR "Allocation of port device entry failed\n"); |
| 209 | goto out; |
| 210 | } |
| 211 | |
| 212 | *dev = ((struct port_dev) { .port = port, |
| 213 | .helper_pid = -1, |
| 214 | .telnetd_pid = -1 }); |
| 215 | goto out; |
| 216 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 217 | out_close: |
| 218 | os_close_file(fd); |
Jeff Dike | 79f6623 | 2007-10-16 01:26:40 -0700 | [diff] [blame] | 219 | out_free: |
| 220 | kfree(port); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 221 | out: |
Daniel Walker | 2aa9c5d | 2008-02-04 22:31:27 -0800 | [diff] [blame] | 222 | mutex_unlock(&ports_mutex); |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 223 | return dev; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 224 | } |
| 225 | |
| 226 | int port_wait(void *data) |
| 227 | { |
| 228 | struct port_dev *dev = data; |
| 229 | struct connection *conn; |
| 230 | struct port_list *port = dev->port; |
| 231 | int fd; |
| 232 | |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 233 | atomic_inc(&port->wait_count); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 234 | while (1) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 235 | fd = -ERESTARTSYS; |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 236 | if (wait_for_completion_interruptible(&port->done)) |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 237 | goto out; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 238 | |
| 239 | spin_lock(&port->lock); |
| 240 | |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 241 | conn = list_entry(port->connections.next, struct connection, |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 242 | list); |
| 243 | list_del(&conn->list); |
| 244 | spin_unlock(&port->lock); |
| 245 | |
| 246 | os_shutdown_socket(conn->socket[0], 1, 1); |
| 247 | os_close_file(conn->socket[0]); |
| 248 | os_shutdown_socket(conn->socket[1], 1, 1); |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 249 | os_close_file(conn->socket[1]); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 250 | |
| 251 | /* This is done here because freeing an IRQ can't be done |
| 252 | * within the IRQ handler. So, pipe_interrupt always ups |
| 253 | * the semaphore regardless of whether it got a successful |
Jeff Dike | 67608e0 | 2007-02-10 01:44:02 -0800 | [diff] [blame] | 254 | * connection. Then we loop here throwing out failed |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 255 | * connections until a good one is found. |
| 256 | */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 257 | free_irq(TELNETD_IRQ, conn); |
| 258 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 259 | if (conn->fd >= 0) |
| 260 | break; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 261 | os_close_file(conn->fd); |
| 262 | kfree(conn); |
| 263 | } |
| 264 | |
| 265 | fd = conn->fd; |
| 266 | dev->helper_pid = conn->helper_pid; |
| 267 | dev->telnetd_pid = conn->telnetd_pid; |
| 268 | kfree(conn); |
| 269 | out: |
| 270 | atomic_dec(&port->wait_count); |
| 271 | return fd; |
| 272 | } |
| 273 | |
| 274 | void port_remove_dev(void *d) |
| 275 | { |
| 276 | struct port_dev *dev = d; |
| 277 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 278 | if (dev->helper_pid != -1) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 279 | os_kill_process(dev->helper_pid, 0); |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 280 | if (dev->telnetd_pid != -1) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 281 | os_kill_process(dev->telnetd_pid, 1); |
| 282 | dev->helper_pid = -1; |
| 283 | dev->telnetd_pid = -1; |
| 284 | } |
| 285 | |
| 286 | void port_kern_free(void *d) |
| 287 | { |
| 288 | struct port_dev *dev = d; |
| 289 | |
| 290 | port_remove_dev(dev); |
| 291 | kfree(dev); |
| 292 | } |
| 293 | |
| 294 | static void free_port(void) |
| 295 | { |
| 296 | struct list_head *ele; |
| 297 | struct port_list *port; |
| 298 | |
Jeff Dike | e99525f | 2007-10-16 01:26:41 -0700 | [diff] [blame] | 299 | list_for_each(ele, &ports) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 300 | port = list_entry(ele, struct port_list, list); |
| 301 | free_irq_by_fd(port->fd); |
| 302 | os_close_file(port->fd); |
| 303 | } |
| 304 | } |
| 305 | |
| 306 | __uml_exitcall(free_port); |