1 /* 2 * Copyright (C) 2001 - 2007 Jeff Dike (jdike@{linux.intel,addtoit}.com) 3 * Licensed under the GPL 4 */ 5 6 #include "linux/completion.h" 7 #include "linux/interrupt.h" 8 #include "linux/list.h" 9 #include "linux/mutex.h" 10 #include "linux/slab.h" 11 #include "linux/workqueue.h" 12 #include "asm/atomic.h" 13 #include "init.h" 14 #include "irq_kern.h" 15 #include "os.h" 16 #include "port.h" 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 45 static irqreturn_t pipe_interrupt(int irq, void *data) 46 { 47 struct connection *conn = data; 48 int fd; 49 50 fd = os_rcv_fd(conn->socket[0], &conn->helper_pid); 51 if (fd < 0) { 52 if (fd == -EAGAIN) 53 return IRQ_NONE; 54 55 printk(KERN_ERR "pipe_interrupt : os_rcv_fd returned %d\n", 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); 66 return IRQ_HANDLED; 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; 79 int fd, socket[2], pid; 80 81 fd = port_connection(port->fd, socket, &pid); 82 if (fd < 0) { 83 if (fd != -EAGAIN) 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); 90 if (conn == NULL) { 91 printk(KERN_ERR "port_accept : failed to allocate " 92 "connection\n"); 93 goto out_close; 94 } 95 *conn = ((struct connection) 96 { .list = LIST_HEAD_INIT(conn->list), 97 .fd = fd, 98 .socket = { socket[0], socket[1] }, 99 .telnetd_pid = pid, 100 .port = port }); 101 102 if (um_request_irq(TELNETD_IRQ, socket[0], IRQ_READ, pipe_interrupt, 103 IRQF_DISABLED | IRQF_SHARED | IRQF_SAMPLE_RANDOM, 104 "telnetd", conn)) { 105 printk(KERN_ERR "port_accept : failed to get IRQ for " 106 "telnetd\n"); 107 goto out_free; 108 } 109 110 if (atomic_read(&port->wait_count) == 0) { 111 os_write_file(fd, NO_WAITER_MSG, sizeof(NO_WAITER_MSG)); 112 printk(KERN_ERR "No one waiting for port\n"); 113 } 114 list_add(&conn->list, &port->pending); 115 return 1; 116 117 out_free: 118 kfree(conn); 119 out_close: 120 os_close_file(fd); 121 os_kill_process(pid, 1); 122 out: 123 return 0; 124 } 125 126 static DEFINE_MUTEX(ports_mutex); 127 static LIST_HEAD(ports); 128 129 static void port_work_proc(struct work_struct *unused) 130 { 131 struct port_list *port; 132 struct list_head *ele; 133 unsigned long flags; 134 135 local_irq_save(flags); 136 list_for_each(ele, &ports) { 137 port = list_entry(ele, struct port_list, list); 138 if (!port->has_connection) 139 continue; 140 141 reactivate_fd(port->fd, ACCEPT_IRQ); 142 while (port_accept(port)) 143 ; 144 port->has_connection = 0; 145 } 146 local_irq_restore(flags); 147 } 148 149 DECLARE_WORK(port_work, port_work_proc); 150 151 static irqreturn_t port_interrupt(int irq, void *data) 152 { 153 struct port_list *port = data; 154 155 port->has_connection = 1; 156 schedule_work(&port_work); 157 return IRQ_HANDLED; 158 } 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 167 mutex_lock(&ports_mutex); 168 list_for_each(ele, &ports) { 169 port = list_entry(ele, struct port_list, list); 170 if (port->port == port_num) 171 goto found; 172 } 173 port = kmalloc(sizeof(struct port_list), GFP_KERNEL); 174 if (port == NULL) { 175 printk(KERN_ERR "Allocation of port list failed\n"); 176 goto out; 177 } 178 179 fd = port_listen_fd(port_num); 180 if (fd < 0) { 181 printk(KERN_ERR "binding to port %d failed, errno = %d\n", 182 port_num, -fd); 183 goto out_free; 184 } 185 186 if (um_request_irq(ACCEPT_IRQ, fd, IRQ_READ, port_interrupt, 187 IRQF_DISABLED | IRQF_SHARED | IRQF_SAMPLE_RANDOM, 188 "port", port)) { 189 printk(KERN_ERR "Failed to get IRQ for port %d\n", port_num); 190 goto out_close; 191 } 192 193 *port = ((struct port_list) 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); 207 if (dev == NULL) { 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 217 out_close: 218 os_close_file(fd); 219 out_free: 220 kfree(port); 221 out: 222 mutex_unlock(&ports_mutex); 223 return dev; 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 233 atomic_inc(&port->wait_count); 234 while (1) { 235 fd = -ERESTARTSYS; 236 if (wait_for_completion_interruptible(&port->done)) 237 goto out; 238 239 spin_lock(&port->lock); 240 241 conn = list_entry(port->connections.next, struct connection, 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); 249 os_close_file(conn->socket[1]); 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 254 * connection. Then we loop here throwing out failed 255 * connections until a good one is found. 256 */ 257 free_irq(TELNETD_IRQ, conn); 258 259 if (conn->fd >= 0) 260 break; 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 278 if (dev->helper_pid != -1) 279 os_kill_process(dev->helper_pid, 0); 280 if (dev->telnetd_pid != -1) 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 299 list_for_each(ele, &ports) { 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); 307