1 /* 2 * IUCV protocol stack for Linux on zSeries 3 * 4 * Copyright IBM Corp. 2006, 2009 5 * 6 * Author(s): Jennifer Hunt <jenhunt@us.ibm.com> 7 * Hendrik Brueckner <brueckner@linux.vnet.ibm.com> 8 * PM functions: 9 * Ursula Braun <ursula.braun@de.ibm.com> 10 */ 11 12 #define KMSG_COMPONENT "af_iucv" 13 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt 14 15 #include <linux/module.h> 16 #include <linux/types.h> 17 #include <linux/list.h> 18 #include <linux/errno.h> 19 #include <linux/kernel.h> 20 #include <linux/sched.h> 21 #include <linux/slab.h> 22 #include <linux/skbuff.h> 23 #include <linux/init.h> 24 #include <linux/poll.h> 25 #include <net/sock.h> 26 #include <asm/ebcdic.h> 27 #include <asm/cpcmd.h> 28 #include <linux/kmod.h> 29 30 #include <net/iucv/iucv.h> 31 #include <net/iucv/af_iucv.h> 32 33 #define VERSION "1.1" 34 35 static char iucv_userid[80]; 36 37 static const struct proto_ops iucv_sock_ops; 38 39 static struct proto iucv_proto = { 40 .name = "AF_IUCV", 41 .owner = THIS_MODULE, 42 .obj_size = sizeof(struct iucv_sock), 43 }; 44 45 /* special AF_IUCV IPRM messages */ 46 static const u8 iprm_shutdown[8] = 47 {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01}; 48 49 #define TRGCLS_SIZE (sizeof(((struct iucv_message *)0)->class)) 50 51 /* macros to set/get socket control buffer at correct offset */ 52 #define CB_TAG(skb) ((skb)->cb) /* iucv message tag */ 53 #define CB_TAG_LEN (sizeof(((struct iucv_message *) 0)->tag)) 54 #define CB_TRGCLS(skb) ((skb)->cb + CB_TAG_LEN) /* iucv msg target class */ 55 #define CB_TRGCLS_LEN (TRGCLS_SIZE) 56 57 #define __iucv_sock_wait(sk, condition, timeo, ret) \ 58 do { \ 59 DEFINE_WAIT(__wait); \ 60 long __timeo = timeo; \ 61 ret = 0; \ 62 prepare_to_wait(sk_sleep(sk), &__wait, TASK_INTERRUPTIBLE); \ 63 while (!(condition)) { \ 64 if (!__timeo) { \ 65 ret = -EAGAIN; \ 66 break; \ 67 } \ 68 if (signal_pending(current)) { \ 69 ret = sock_intr_errno(__timeo); \ 70 break; \ 71 } \ 72 release_sock(sk); \ 73 __timeo = schedule_timeout(__timeo); \ 74 lock_sock(sk); \ 75 ret = sock_error(sk); \ 76 if (ret) \ 77 break; \ 78 } \ 79 finish_wait(sk_sleep(sk), &__wait); \ 80 } while (0) 81 82 #define iucv_sock_wait(sk, condition, timeo) \ 83 ({ \ 84 int __ret = 0; \ 85 if (!(condition)) \ 86 __iucv_sock_wait(sk, condition, timeo, __ret); \ 87 __ret; \ 88 }) 89 90 static void iucv_sock_kill(struct sock *sk); 91 static void iucv_sock_close(struct sock *sk); 92 93 /* Call Back functions */ 94 static void iucv_callback_rx(struct iucv_path *, struct iucv_message *); 95 static void iucv_callback_txdone(struct iucv_path *, struct iucv_message *); 96 static void iucv_callback_connack(struct iucv_path *, u8 ipuser[16]); 97 static int iucv_callback_connreq(struct iucv_path *, u8 ipvmid[8], 98 u8 ipuser[16]); 99 static void iucv_callback_connrej(struct iucv_path *, u8 ipuser[16]); 100 static void iucv_callback_shutdown(struct iucv_path *, u8 ipuser[16]); 101 102 static struct iucv_sock_list iucv_sk_list = { 103 .lock = __RW_LOCK_UNLOCKED(iucv_sk_list.lock), 104 .autobind_name = ATOMIC_INIT(0) 105 }; 106 107 static struct iucv_handler af_iucv_handler = { 108 .path_pending = iucv_callback_connreq, 109 .path_complete = iucv_callback_connack, 110 .path_severed = iucv_callback_connrej, 111 .message_pending = iucv_callback_rx, 112 .message_complete = iucv_callback_txdone, 113 .path_quiesced = iucv_callback_shutdown, 114 }; 115 116 static inline void high_nmcpy(unsigned char *dst, char *src) 117 { 118 memcpy(dst, src, 8); 119 } 120 121 static inline void low_nmcpy(unsigned char *dst, char *src) 122 { 123 memcpy(&dst[8], src, 8); 124 } 125 126 static int afiucv_pm_prepare(struct device *dev) 127 { 128 #ifdef CONFIG_PM_DEBUG 129 printk(KERN_WARNING "afiucv_pm_prepare\n"); 130 #endif 131 return 0; 132 } 133 134 static void afiucv_pm_complete(struct device *dev) 135 { 136 #ifdef CONFIG_PM_DEBUG 137 printk(KERN_WARNING "afiucv_pm_complete\n"); 138 #endif 139 } 140 141 /** 142 * afiucv_pm_freeze() - Freeze PM callback 143 * @dev: AFIUCV dummy device 144 * 145 * Sever all established IUCV communication pathes 146 */ 147 static int afiucv_pm_freeze(struct device *dev) 148 { 149 struct iucv_sock *iucv; 150 struct sock *sk; 151 struct hlist_node *node; 152 int err = 0; 153 154 #ifdef CONFIG_PM_DEBUG 155 printk(KERN_WARNING "afiucv_pm_freeze\n"); 156 #endif 157 read_lock(&iucv_sk_list.lock); 158 sk_for_each(sk, node, &iucv_sk_list.head) { 159 iucv = iucv_sk(sk); 160 skb_queue_purge(&iucv->send_skb_q); 161 skb_queue_purge(&iucv->backlog_skb_q); 162 switch (sk->sk_state) { 163 case IUCV_SEVERED: 164 case IUCV_DISCONN: 165 case IUCV_CLOSING: 166 case IUCV_CONNECTED: 167 if (iucv->path) { 168 err = iucv_path_sever(iucv->path, NULL); 169 iucv_path_free(iucv->path); 170 iucv->path = NULL; 171 } 172 break; 173 case IUCV_OPEN: 174 case IUCV_BOUND: 175 case IUCV_LISTEN: 176 case IUCV_CLOSED: 177 default: 178 break; 179 } 180 } 181 read_unlock(&iucv_sk_list.lock); 182 return err; 183 } 184 185 /** 186 * afiucv_pm_restore_thaw() - Thaw and restore PM callback 187 * @dev: AFIUCV dummy device 188 * 189 * socket clean up after freeze 190 */ 191 static int afiucv_pm_restore_thaw(struct device *dev) 192 { 193 struct sock *sk; 194 struct hlist_node *node; 195 196 #ifdef CONFIG_PM_DEBUG 197 printk(KERN_WARNING "afiucv_pm_restore_thaw\n"); 198 #endif 199 read_lock(&iucv_sk_list.lock); 200 sk_for_each(sk, node, &iucv_sk_list.head) { 201 switch (sk->sk_state) { 202 case IUCV_CONNECTED: 203 sk->sk_err = EPIPE; 204 sk->sk_state = IUCV_DISCONN; 205 sk->sk_state_change(sk); 206 break; 207 case IUCV_DISCONN: 208 case IUCV_SEVERED: 209 case IUCV_CLOSING: 210 case IUCV_LISTEN: 211 case IUCV_BOUND: 212 case IUCV_OPEN: 213 default: 214 break; 215 } 216 } 217 read_unlock(&iucv_sk_list.lock); 218 return 0; 219 } 220 221 static const struct dev_pm_ops afiucv_pm_ops = { 222 .prepare = afiucv_pm_prepare, 223 .complete = afiucv_pm_complete, 224 .freeze = afiucv_pm_freeze, 225 .thaw = afiucv_pm_restore_thaw, 226 .restore = afiucv_pm_restore_thaw, 227 }; 228 229 static struct device_driver af_iucv_driver = { 230 .owner = THIS_MODULE, 231 .name = "afiucv", 232 .bus = &iucv_bus, 233 .pm = &afiucv_pm_ops, 234 }; 235 236 /* dummy device used as trigger for PM functions */ 237 static struct device *af_iucv_dev; 238 239 /** 240 * iucv_msg_length() - Returns the length of an iucv message. 241 * @msg: Pointer to struct iucv_message, MUST NOT be NULL 242 * 243 * The function returns the length of the specified iucv message @msg of data 244 * stored in a buffer and of data stored in the parameter list (PRMDATA). 245 * 246 * For IUCV_IPRMDATA, AF_IUCV uses the following convention to transport socket 247 * data: 248 * PRMDATA[0..6] socket data (max 7 bytes); 249 * PRMDATA[7] socket data length value (len is 0xff - PRMDATA[7]) 250 * 251 * The socket data length is computed by subtracting the socket data length 252 * value from 0xFF. 253 * If the socket data len is greater 7, then PRMDATA can be used for special 254 * notifications (see iucv_sock_shutdown); and further, 255 * if the socket data len is > 7, the function returns 8. 256 * 257 * Use this function to allocate socket buffers to store iucv message data. 258 */ 259 static inline size_t iucv_msg_length(struct iucv_message *msg) 260 { 261 size_t datalen; 262 263 if (msg->flags & IUCV_IPRMDATA) { 264 datalen = 0xff - msg->rmmsg[7]; 265 return (datalen < 8) ? datalen : 8; 266 } 267 return msg->length; 268 } 269 270 /** 271 * iucv_sock_in_state() - check for specific states 272 * @sk: sock structure 273 * @state: first iucv sk state 274 * @state: second iucv sk state 275 * 276 * Returns true if the socket in either in the first or second state. 277 */ 278 static int iucv_sock_in_state(struct sock *sk, int state, int state2) 279 { 280 return (sk->sk_state == state || sk->sk_state == state2); 281 } 282 283 /** 284 * iucv_below_msglim() - function to check if messages can be sent 285 * @sk: sock structure 286 * 287 * Returns true if the send queue length is lower than the message limit. 288 * Always returns true if the socket is not connected (no iucv path for 289 * checking the message limit). 290 */ 291 static inline int iucv_below_msglim(struct sock *sk) 292 { 293 struct iucv_sock *iucv = iucv_sk(sk); 294 295 if (sk->sk_state != IUCV_CONNECTED) 296 return 1; 297 return (skb_queue_len(&iucv->send_skb_q) < iucv->path->msglim); 298 } 299 300 /** 301 * iucv_sock_wake_msglim() - Wake up thread waiting on msg limit 302 */ 303 static void iucv_sock_wake_msglim(struct sock *sk) 304 { 305 struct socket_wq *wq; 306 307 rcu_read_lock(); 308 wq = rcu_dereference(sk->sk_wq); 309 if (wq_has_sleeper(wq)) 310 wake_up_interruptible_all(&wq->wait); 311 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT); 312 rcu_read_unlock(); 313 } 314 315 /* Timers */ 316 static void iucv_sock_timeout(unsigned long arg) 317 { 318 struct sock *sk = (struct sock *)arg; 319 320 bh_lock_sock(sk); 321 sk->sk_err = ETIMEDOUT; 322 sk->sk_state_change(sk); 323 bh_unlock_sock(sk); 324 325 iucv_sock_kill(sk); 326 sock_put(sk); 327 } 328 329 static void iucv_sock_clear_timer(struct sock *sk) 330 { 331 sk_stop_timer(sk, &sk->sk_timer); 332 } 333 334 static struct sock *__iucv_get_sock_by_name(char *nm) 335 { 336 struct sock *sk; 337 struct hlist_node *node; 338 339 sk_for_each(sk, node, &iucv_sk_list.head) 340 if (!memcmp(&iucv_sk(sk)->src_name, nm, 8)) 341 return sk; 342 343 return NULL; 344 } 345 346 static void iucv_sock_destruct(struct sock *sk) 347 { 348 skb_queue_purge(&sk->sk_receive_queue); 349 skb_queue_purge(&sk->sk_write_queue); 350 } 351 352 /* Cleanup Listen */ 353 static void iucv_sock_cleanup_listen(struct sock *parent) 354 { 355 struct sock *sk; 356 357 /* Close non-accepted connections */ 358 while ((sk = iucv_accept_dequeue(parent, NULL))) { 359 iucv_sock_close(sk); 360 iucv_sock_kill(sk); 361 } 362 363 parent->sk_state = IUCV_CLOSED; 364 } 365 366 /* Kill socket (only if zapped and orphaned) */ 367 static void iucv_sock_kill(struct sock *sk) 368 { 369 if (!sock_flag(sk, SOCK_ZAPPED) || sk->sk_socket) 370 return; 371 372 iucv_sock_unlink(&iucv_sk_list, sk); 373 sock_set_flag(sk, SOCK_DEAD); 374 sock_put(sk); 375 } 376 377 /* Close an IUCV socket */ 378 static void iucv_sock_close(struct sock *sk) 379 { 380 unsigned char user_data[16]; 381 struct iucv_sock *iucv = iucv_sk(sk); 382 unsigned long timeo; 383 384 iucv_sock_clear_timer(sk); 385 lock_sock(sk); 386 387 switch (sk->sk_state) { 388 case IUCV_LISTEN: 389 iucv_sock_cleanup_listen(sk); 390 break; 391 392 case IUCV_CONNECTED: 393 case IUCV_DISCONN: 394 sk->sk_state = IUCV_CLOSING; 395 sk->sk_state_change(sk); 396 397 if (!skb_queue_empty(&iucv->send_skb_q)) { 398 if (sock_flag(sk, SOCK_LINGER) && sk->sk_lingertime) 399 timeo = sk->sk_lingertime; 400 else 401 timeo = IUCV_DISCONN_TIMEOUT; 402 iucv_sock_wait(sk, 403 iucv_sock_in_state(sk, IUCV_CLOSED, 0), 404 timeo); 405 } 406 407 case IUCV_CLOSING: /* fall through */ 408 sk->sk_state = IUCV_CLOSED; 409 sk->sk_state_change(sk); 410 411 if (iucv->path) { 412 low_nmcpy(user_data, iucv->src_name); 413 high_nmcpy(user_data, iucv->dst_name); 414 ASCEBC(user_data, sizeof(user_data)); 415 iucv_path_sever(iucv->path, user_data); 416 iucv_path_free(iucv->path); 417 iucv->path = NULL; 418 } 419 420 sk->sk_err = ECONNRESET; 421 sk->sk_state_change(sk); 422 423 skb_queue_purge(&iucv->send_skb_q); 424 skb_queue_purge(&iucv->backlog_skb_q); 425 break; 426 427 default: 428 /* nothing to do here */ 429 break; 430 } 431 432 /* mark socket for deletion by iucv_sock_kill() */ 433 sock_set_flag(sk, SOCK_ZAPPED); 434 435 release_sock(sk); 436 } 437 438 static void iucv_sock_init(struct sock *sk, struct sock *parent) 439 { 440 if (parent) 441 sk->sk_type = parent->sk_type; 442 } 443 444 static struct sock *iucv_sock_alloc(struct socket *sock, int proto, gfp_t prio) 445 { 446 struct sock *sk; 447 448 sk = sk_alloc(&init_net, PF_IUCV, prio, &iucv_proto); 449 if (!sk) 450 return NULL; 451 452 sock_init_data(sock, sk); 453 INIT_LIST_HEAD(&iucv_sk(sk)->accept_q); 454 spin_lock_init(&iucv_sk(sk)->accept_q_lock); 455 skb_queue_head_init(&iucv_sk(sk)->send_skb_q); 456 INIT_LIST_HEAD(&iucv_sk(sk)->message_q.list); 457 spin_lock_init(&iucv_sk(sk)->message_q.lock); 458 skb_queue_head_init(&iucv_sk(sk)->backlog_skb_q); 459 iucv_sk(sk)->send_tag = 0; 460 iucv_sk(sk)->flags = 0; 461 iucv_sk(sk)->msglimit = IUCV_QUEUELEN_DEFAULT; 462 iucv_sk(sk)->path = NULL; 463 memset(&iucv_sk(sk)->src_user_id , 0, 32); 464 465 sk->sk_destruct = iucv_sock_destruct; 466 sk->sk_sndtimeo = IUCV_CONN_TIMEOUT; 467 sk->sk_allocation = GFP_DMA; 468 469 sock_reset_flag(sk, SOCK_ZAPPED); 470 471 sk->sk_protocol = proto; 472 sk->sk_state = IUCV_OPEN; 473 474 setup_timer(&sk->sk_timer, iucv_sock_timeout, (unsigned long)sk); 475 476 iucv_sock_link(&iucv_sk_list, sk); 477 return sk; 478 } 479 480 /* Create an IUCV socket */ 481 static int iucv_sock_create(struct net *net, struct socket *sock, int protocol, 482 int kern) 483 { 484 struct sock *sk; 485 486 if (protocol && protocol != PF_IUCV) 487 return -EPROTONOSUPPORT; 488 489 sock->state = SS_UNCONNECTED; 490 491 switch (sock->type) { 492 case SOCK_STREAM: 493 sock->ops = &iucv_sock_ops; 494 break; 495 case SOCK_SEQPACKET: 496 /* currently, proto ops can handle both sk types */ 497 sock->ops = &iucv_sock_ops; 498 break; 499 default: 500 return -ESOCKTNOSUPPORT; 501 } 502 503 sk = iucv_sock_alloc(sock, protocol, GFP_KERNEL); 504 if (!sk) 505 return -ENOMEM; 506 507 iucv_sock_init(sk, NULL); 508 509 return 0; 510 } 511 512 void iucv_sock_link(struct iucv_sock_list *l, struct sock *sk) 513 { 514 write_lock_bh(&l->lock); 515 sk_add_node(sk, &l->head); 516 write_unlock_bh(&l->lock); 517 } 518 519 void iucv_sock_unlink(struct iucv_sock_list *l, struct sock *sk) 520 { 521 write_lock_bh(&l->lock); 522 sk_del_node_init(sk); 523 write_unlock_bh(&l->lock); 524 } 525 526 void iucv_accept_enqueue(struct sock *parent, struct sock *sk) 527 { 528 unsigned long flags; 529 struct iucv_sock *par = iucv_sk(parent); 530 531 sock_hold(sk); 532 spin_lock_irqsave(&par->accept_q_lock, flags); 533 list_add_tail(&iucv_sk(sk)->accept_q, &par->accept_q); 534 spin_unlock_irqrestore(&par->accept_q_lock, flags); 535 iucv_sk(sk)->parent = parent; 536 sk_acceptq_added(parent); 537 } 538 539 void iucv_accept_unlink(struct sock *sk) 540 { 541 unsigned long flags; 542 struct iucv_sock *par = iucv_sk(iucv_sk(sk)->parent); 543 544 spin_lock_irqsave(&par->accept_q_lock, flags); 545 list_del_init(&iucv_sk(sk)->accept_q); 546 spin_unlock_irqrestore(&par->accept_q_lock, flags); 547 sk_acceptq_removed(iucv_sk(sk)->parent); 548 iucv_sk(sk)->parent = NULL; 549 sock_put(sk); 550 } 551 552 struct sock *iucv_accept_dequeue(struct sock *parent, struct socket *newsock) 553 { 554 struct iucv_sock *isk, *n; 555 struct sock *sk; 556 557 list_for_each_entry_safe(isk, n, &iucv_sk(parent)->accept_q, accept_q) { 558 sk = (struct sock *) isk; 559 lock_sock(sk); 560 561 if (sk->sk_state == IUCV_CLOSED) { 562 iucv_accept_unlink(sk); 563 release_sock(sk); 564 continue; 565 } 566 567 if (sk->sk_state == IUCV_CONNECTED || 568 sk->sk_state == IUCV_SEVERED || 569 sk->sk_state == IUCV_DISCONN || /* due to PM restore */ 570 !newsock) { 571 iucv_accept_unlink(sk); 572 if (newsock) 573 sock_graft(sk, newsock); 574 575 if (sk->sk_state == IUCV_SEVERED) 576 sk->sk_state = IUCV_DISCONN; 577 578 release_sock(sk); 579 return sk; 580 } 581 582 release_sock(sk); 583 } 584 return NULL; 585 } 586 587 /* Bind an unbound socket */ 588 static int iucv_sock_bind(struct socket *sock, struct sockaddr *addr, 589 int addr_len) 590 { 591 struct sockaddr_iucv *sa = (struct sockaddr_iucv *) addr; 592 struct sock *sk = sock->sk; 593 struct iucv_sock *iucv; 594 int err; 595 596 /* Verify the input sockaddr */ 597 if (!addr || addr->sa_family != AF_IUCV) 598 return -EINVAL; 599 600 lock_sock(sk); 601 if (sk->sk_state != IUCV_OPEN) { 602 err = -EBADFD; 603 goto done; 604 } 605 606 write_lock_bh(&iucv_sk_list.lock); 607 608 iucv = iucv_sk(sk); 609 if (__iucv_get_sock_by_name(sa->siucv_name)) { 610 err = -EADDRINUSE; 611 goto done_unlock; 612 } 613 if (iucv->path) { 614 err = 0; 615 goto done_unlock; 616 } 617 618 /* Bind the socket */ 619 memcpy(iucv->src_name, sa->siucv_name, 8); 620 621 /* Copy the user id */ 622 memcpy(iucv->src_user_id, iucv_userid, 8); 623 sk->sk_state = IUCV_BOUND; 624 err = 0; 625 626 done_unlock: 627 /* Release the socket list lock */ 628 write_unlock_bh(&iucv_sk_list.lock); 629 done: 630 release_sock(sk); 631 return err; 632 } 633 634 /* Automatically bind an unbound socket */ 635 static int iucv_sock_autobind(struct sock *sk) 636 { 637 struct iucv_sock *iucv = iucv_sk(sk); 638 char query_buffer[80]; 639 char name[12]; 640 int err = 0; 641 642 /* Set the userid and name */ 643 cpcmd("QUERY USERID", query_buffer, sizeof(query_buffer), &err); 644 if (unlikely(err)) 645 return -EPROTO; 646 647 memcpy(iucv->src_user_id, query_buffer, 8); 648 649 write_lock_bh(&iucv_sk_list.lock); 650 651 sprintf(name, "%08x", atomic_inc_return(&iucv_sk_list.autobind_name)); 652 while (__iucv_get_sock_by_name(name)) { 653 sprintf(name, "%08x", 654 atomic_inc_return(&iucv_sk_list.autobind_name)); 655 } 656 657 write_unlock_bh(&iucv_sk_list.lock); 658 659 memcpy(&iucv->src_name, name, 8); 660 661 return err; 662 } 663 664 /* Connect an unconnected socket */ 665 static int iucv_sock_connect(struct socket *sock, struct sockaddr *addr, 666 int alen, int flags) 667 { 668 struct sockaddr_iucv *sa = (struct sockaddr_iucv *) addr; 669 struct sock *sk = sock->sk; 670 struct iucv_sock *iucv; 671 unsigned char user_data[16]; 672 int err; 673 674 if (addr->sa_family != AF_IUCV || alen < sizeof(struct sockaddr_iucv)) 675 return -EINVAL; 676 677 if (sk->sk_state != IUCV_OPEN && sk->sk_state != IUCV_BOUND) 678 return -EBADFD; 679 680 if (sk->sk_type != SOCK_STREAM && sk->sk_type != SOCK_SEQPACKET) 681 return -EINVAL; 682 683 if (sk->sk_state == IUCV_OPEN) { 684 err = iucv_sock_autobind(sk); 685 if (unlikely(err)) 686 return err; 687 } 688 689 lock_sock(sk); 690 691 /* Set the destination information */ 692 memcpy(iucv_sk(sk)->dst_user_id, sa->siucv_user_id, 8); 693 memcpy(iucv_sk(sk)->dst_name, sa->siucv_name, 8); 694 695 high_nmcpy(user_data, sa->siucv_name); 696 low_nmcpy(user_data, iucv_sk(sk)->src_name); 697 ASCEBC(user_data, sizeof(user_data)); 698 699 iucv = iucv_sk(sk); 700 /* Create path. */ 701 iucv->path = iucv_path_alloc(iucv->msglimit, 702 IUCV_IPRMDATA, GFP_KERNEL); 703 if (!iucv->path) { 704 err = -ENOMEM; 705 goto done; 706 } 707 err = iucv_path_connect(iucv->path, &af_iucv_handler, 708 sa->siucv_user_id, NULL, user_data, sk); 709 if (err) { 710 iucv_path_free(iucv->path); 711 iucv->path = NULL; 712 switch (err) { 713 case 0x0b: /* Target communicator is not logged on */ 714 err = -ENETUNREACH; 715 break; 716 case 0x0d: /* Max connections for this guest exceeded */ 717 case 0x0e: /* Max connections for target guest exceeded */ 718 err = -EAGAIN; 719 break; 720 case 0x0f: /* Missing IUCV authorization */ 721 err = -EACCES; 722 break; 723 default: 724 err = -ECONNREFUSED; 725 break; 726 } 727 goto done; 728 } 729 730 if (sk->sk_state != IUCV_CONNECTED) { 731 err = iucv_sock_wait(sk, iucv_sock_in_state(sk, IUCV_CONNECTED, 732 IUCV_DISCONN), 733 sock_sndtimeo(sk, flags & O_NONBLOCK)); 734 } 735 736 if (sk->sk_state == IUCV_DISCONN) { 737 err = -ECONNREFUSED; 738 } 739 740 if (err) { 741 iucv_path_sever(iucv->path, NULL); 742 iucv_path_free(iucv->path); 743 iucv->path = NULL; 744 } 745 746 done: 747 release_sock(sk); 748 return err; 749 } 750 751 /* Move a socket into listening state. */ 752 static int iucv_sock_listen(struct socket *sock, int backlog) 753 { 754 struct sock *sk = sock->sk; 755 int err; 756 757 lock_sock(sk); 758 759 err = -EINVAL; 760 if (sk->sk_state != IUCV_BOUND) 761 goto done; 762 763 if (sock->type != SOCK_STREAM && sock->type != SOCK_SEQPACKET) 764 goto done; 765 766 sk->sk_max_ack_backlog = backlog; 767 sk->sk_ack_backlog = 0; 768 sk->sk_state = IUCV_LISTEN; 769 err = 0; 770 771 done: 772 release_sock(sk); 773 return err; 774 } 775 776 /* Accept a pending connection */ 777 static int iucv_sock_accept(struct socket *sock, struct socket *newsock, 778 int flags) 779 { 780 DECLARE_WAITQUEUE(wait, current); 781 struct sock *sk = sock->sk, *nsk; 782 long timeo; 783 int err = 0; 784 785 lock_sock_nested(sk, SINGLE_DEPTH_NESTING); 786 787 if (sk->sk_state != IUCV_LISTEN) { 788 err = -EBADFD; 789 goto done; 790 } 791 792 timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK); 793 794 /* Wait for an incoming connection */ 795 add_wait_queue_exclusive(sk_sleep(sk), &wait); 796 while (!(nsk = iucv_accept_dequeue(sk, newsock))) { 797 set_current_state(TASK_INTERRUPTIBLE); 798 if (!timeo) { 799 err = -EAGAIN; 800 break; 801 } 802 803 release_sock(sk); 804 timeo = schedule_timeout(timeo); 805 lock_sock_nested(sk, SINGLE_DEPTH_NESTING); 806 807 if (sk->sk_state != IUCV_LISTEN) { 808 err = -EBADFD; 809 break; 810 } 811 812 if (signal_pending(current)) { 813 err = sock_intr_errno(timeo); 814 break; 815 } 816 } 817 818 set_current_state(TASK_RUNNING); 819 remove_wait_queue(sk_sleep(sk), &wait); 820 821 if (err) 822 goto done; 823 824 newsock->state = SS_CONNECTED; 825 826 done: 827 release_sock(sk); 828 return err; 829 } 830 831 static int iucv_sock_getname(struct socket *sock, struct sockaddr *addr, 832 int *len, int peer) 833 { 834 struct sockaddr_iucv *siucv = (struct sockaddr_iucv *) addr; 835 struct sock *sk = sock->sk; 836 837 addr->sa_family = AF_IUCV; 838 *len = sizeof(struct sockaddr_iucv); 839 840 if (peer) { 841 memcpy(siucv->siucv_user_id, iucv_sk(sk)->dst_user_id, 8); 842 memcpy(siucv->siucv_name, &iucv_sk(sk)->dst_name, 8); 843 } else { 844 memcpy(siucv->siucv_user_id, iucv_sk(sk)->src_user_id, 8); 845 memcpy(siucv->siucv_name, iucv_sk(sk)->src_name, 8); 846 } 847 memset(&siucv->siucv_port, 0, sizeof(siucv->siucv_port)); 848 memset(&siucv->siucv_addr, 0, sizeof(siucv->siucv_addr)); 849 memset(siucv->siucv_nodeid, 0, sizeof(siucv->siucv_nodeid)); 850 851 return 0; 852 } 853 854 /** 855 * iucv_send_iprm() - Send socket data in parameter list of an iucv message. 856 * @path: IUCV path 857 * @msg: Pointer to a struct iucv_message 858 * @skb: The socket data to send, skb->len MUST BE <= 7 859 * 860 * Send the socket data in the parameter list in the iucv message 861 * (IUCV_IPRMDATA). The socket data is stored at index 0 to 6 in the parameter 862 * list and the socket data len at index 7 (last byte). 863 * See also iucv_msg_length(). 864 * 865 * Returns the error code from the iucv_message_send() call. 866 */ 867 static int iucv_send_iprm(struct iucv_path *path, struct iucv_message *msg, 868 struct sk_buff *skb) 869 { 870 u8 prmdata[8]; 871 872 memcpy(prmdata, (void *) skb->data, skb->len); 873 prmdata[7] = 0xff - (u8) skb->len; 874 return iucv_message_send(path, msg, IUCV_IPRMDATA, 0, 875 (void *) prmdata, 8); 876 } 877 878 static int iucv_sock_sendmsg(struct kiocb *iocb, struct socket *sock, 879 struct msghdr *msg, size_t len) 880 { 881 struct sock *sk = sock->sk; 882 struct iucv_sock *iucv = iucv_sk(sk); 883 struct sk_buff *skb; 884 struct iucv_message txmsg; 885 struct cmsghdr *cmsg; 886 int cmsg_done; 887 long timeo; 888 char user_id[9]; 889 char appl_id[9]; 890 int err; 891 int noblock = msg->msg_flags & MSG_DONTWAIT; 892 893 err = sock_error(sk); 894 if (err) 895 return err; 896 897 if (msg->msg_flags & MSG_OOB) 898 return -EOPNOTSUPP; 899 900 /* SOCK_SEQPACKET: we do not support segmented records */ 901 if (sk->sk_type == SOCK_SEQPACKET && !(msg->msg_flags & MSG_EOR)) 902 return -EOPNOTSUPP; 903 904 lock_sock(sk); 905 906 if (sk->sk_shutdown & SEND_SHUTDOWN) { 907 err = -EPIPE; 908 goto out; 909 } 910 911 /* Return if the socket is not in connected state */ 912 if (sk->sk_state != IUCV_CONNECTED) { 913 err = -ENOTCONN; 914 goto out; 915 } 916 917 /* initialize defaults */ 918 cmsg_done = 0; /* check for duplicate headers */ 919 txmsg.class = 0; 920 921 /* iterate over control messages */ 922 for (cmsg = CMSG_FIRSTHDR(msg); cmsg; 923 cmsg = CMSG_NXTHDR(msg, cmsg)) { 924 925 if (!CMSG_OK(msg, cmsg)) { 926 err = -EINVAL; 927 goto out; 928 } 929 930 if (cmsg->cmsg_level != SOL_IUCV) 931 continue; 932 933 if (cmsg->cmsg_type & cmsg_done) { 934 err = -EINVAL; 935 goto out; 936 } 937 cmsg_done |= cmsg->cmsg_type; 938 939 switch (cmsg->cmsg_type) { 940 case SCM_IUCV_TRGCLS: 941 if (cmsg->cmsg_len != CMSG_LEN(TRGCLS_SIZE)) { 942 err = -EINVAL; 943 goto out; 944 } 945 946 /* set iucv message target class */ 947 memcpy(&txmsg.class, 948 (void *) CMSG_DATA(cmsg), TRGCLS_SIZE); 949 950 break; 951 952 default: 953 err = -EINVAL; 954 goto out; 955 break; 956 } 957 } 958 959 /* allocate one skb for each iucv message: 960 * this is fine for SOCK_SEQPACKET (unless we want to support 961 * segmented records using the MSG_EOR flag), but 962 * for SOCK_STREAM we might want to improve it in future */ 963 skb = sock_alloc_send_skb(sk, len, noblock, &err); 964 if (!skb) 965 goto out; 966 if (memcpy_fromiovec(skb_put(skb, len), msg->msg_iov, len)) { 967 err = -EFAULT; 968 goto fail; 969 } 970 971 /* wait if outstanding messages for iucv path has reached */ 972 timeo = sock_sndtimeo(sk, noblock); 973 err = iucv_sock_wait(sk, iucv_below_msglim(sk), timeo); 974 if (err) 975 goto fail; 976 977 /* return -ECONNRESET if the socket is no longer connected */ 978 if (sk->sk_state != IUCV_CONNECTED) { 979 err = -ECONNRESET; 980 goto fail; 981 } 982 983 /* increment and save iucv message tag for msg_completion cbk */ 984 txmsg.tag = iucv->send_tag++; 985 memcpy(CB_TAG(skb), &txmsg.tag, CB_TAG_LEN); 986 skb_queue_tail(&iucv->send_skb_q, skb); 987 988 if (((iucv->path->flags & IUCV_IPRMDATA) & iucv->flags) 989 && skb->len <= 7) { 990 err = iucv_send_iprm(iucv->path, &txmsg, skb); 991 992 /* on success: there is no message_complete callback 993 * for an IPRMDATA msg; remove skb from send queue */ 994 if (err == 0) { 995 skb_unlink(skb, &iucv->send_skb_q); 996 kfree_skb(skb); 997 } 998 999 /* this error should never happen since the 1000 * IUCV_IPRMDATA path flag is set... sever path */ 1001 if (err == 0x15) { 1002 iucv_path_sever(iucv->path, NULL); 1003 skb_unlink(skb, &iucv->send_skb_q); 1004 err = -EPIPE; 1005 goto fail; 1006 } 1007 } else 1008 err = iucv_message_send(iucv->path, &txmsg, 0, 0, 1009 (void *) skb->data, skb->len); 1010 if (err) { 1011 if (err == 3) { 1012 user_id[8] = 0; 1013 memcpy(user_id, iucv->dst_user_id, 8); 1014 appl_id[8] = 0; 1015 memcpy(appl_id, iucv->dst_name, 8); 1016 pr_err("Application %s on z/VM guest %s" 1017 " exceeds message limit\n", 1018 appl_id, user_id); 1019 err = -EAGAIN; 1020 } else 1021 err = -EPIPE; 1022 skb_unlink(skb, &iucv->send_skb_q); 1023 goto fail; 1024 } 1025 1026 release_sock(sk); 1027 return len; 1028 1029 fail: 1030 kfree_skb(skb); 1031 out: 1032 release_sock(sk); 1033 return err; 1034 } 1035 1036 /* iucv_fragment_skb() - Fragment a single IUCV message into multiple skb's 1037 * 1038 * Locking: must be called with message_q.lock held 1039 */ 1040 static int iucv_fragment_skb(struct sock *sk, struct sk_buff *skb, int len) 1041 { 1042 int dataleft, size, copied = 0; 1043 struct sk_buff *nskb; 1044 1045 dataleft = len; 1046 while (dataleft) { 1047 if (dataleft >= sk->sk_rcvbuf / 4) 1048 size = sk->sk_rcvbuf / 4; 1049 else 1050 size = dataleft; 1051 1052 nskb = alloc_skb(size, GFP_ATOMIC | GFP_DMA); 1053 if (!nskb) 1054 return -ENOMEM; 1055 1056 /* copy target class to control buffer of new skb */ 1057 memcpy(CB_TRGCLS(nskb), CB_TRGCLS(skb), CB_TRGCLS_LEN); 1058 1059 /* copy data fragment */ 1060 memcpy(nskb->data, skb->data + copied, size); 1061 copied += size; 1062 dataleft -= size; 1063 1064 skb_reset_transport_header(nskb); 1065 skb_reset_network_header(nskb); 1066 nskb->len = size; 1067 1068 skb_queue_tail(&iucv_sk(sk)->backlog_skb_q, nskb); 1069 } 1070 1071 return 0; 1072 } 1073 1074 /* iucv_process_message() - Receive a single outstanding IUCV message 1075 * 1076 * Locking: must be called with message_q.lock held 1077 */ 1078 static void iucv_process_message(struct sock *sk, struct sk_buff *skb, 1079 struct iucv_path *path, 1080 struct iucv_message *msg) 1081 { 1082 int rc; 1083 unsigned int len; 1084 1085 len = iucv_msg_length(msg); 1086 1087 /* store msg target class in the second 4 bytes of skb ctrl buffer */ 1088 /* Note: the first 4 bytes are reserved for msg tag */ 1089 memcpy(CB_TRGCLS(skb), &msg->class, CB_TRGCLS_LEN); 1090 1091 /* check for special IPRM messages (e.g. iucv_sock_shutdown) */ 1092 if ((msg->flags & IUCV_IPRMDATA) && len > 7) { 1093 if (memcmp(msg->rmmsg, iprm_shutdown, 8) == 0) { 1094 skb->data = NULL; 1095 skb->len = 0; 1096 } 1097 } else { 1098 rc = iucv_message_receive(path, msg, msg->flags & IUCV_IPRMDATA, 1099 skb->data, len, NULL); 1100 if (rc) { 1101 kfree_skb(skb); 1102 return; 1103 } 1104 /* we need to fragment iucv messages for SOCK_STREAM only; 1105 * for SOCK_SEQPACKET, it is only relevant if we support 1106 * record segmentation using MSG_EOR (see also recvmsg()) */ 1107 if (sk->sk_type == SOCK_STREAM && 1108 skb->truesize >= sk->sk_rcvbuf / 4) { 1109 rc = iucv_fragment_skb(sk, skb, len); 1110 kfree_skb(skb); 1111 skb = NULL; 1112 if (rc) { 1113 iucv_path_sever(path, NULL); 1114 return; 1115 } 1116 skb = skb_dequeue(&iucv_sk(sk)->backlog_skb_q); 1117 } else { 1118 skb_reset_transport_header(skb); 1119 skb_reset_network_header(skb); 1120 skb->len = len; 1121 } 1122 } 1123 1124 if (sock_queue_rcv_skb(sk, skb)) 1125 skb_queue_head(&iucv_sk(sk)->backlog_skb_q, skb); 1126 } 1127 1128 /* iucv_process_message_q() - Process outstanding IUCV messages 1129 * 1130 * Locking: must be called with message_q.lock held 1131 */ 1132 static void iucv_process_message_q(struct sock *sk) 1133 { 1134 struct iucv_sock *iucv = iucv_sk(sk); 1135 struct sk_buff *skb; 1136 struct sock_msg_q *p, *n; 1137 1138 list_for_each_entry_safe(p, n, &iucv->message_q.list, list) { 1139 skb = alloc_skb(iucv_msg_length(&p->msg), GFP_ATOMIC | GFP_DMA); 1140 if (!skb) 1141 break; 1142 iucv_process_message(sk, skb, p->path, &p->msg); 1143 list_del(&p->list); 1144 kfree(p); 1145 if (!skb_queue_empty(&iucv->backlog_skb_q)) 1146 break; 1147 } 1148 } 1149 1150 static int iucv_sock_recvmsg(struct kiocb *iocb, struct socket *sock, 1151 struct msghdr *msg, size_t len, int flags) 1152 { 1153 int noblock = flags & MSG_DONTWAIT; 1154 struct sock *sk = sock->sk; 1155 struct iucv_sock *iucv = iucv_sk(sk); 1156 unsigned int copied, rlen; 1157 struct sk_buff *skb, *rskb, *cskb; 1158 int err = 0; 1159 1160 if ((sk->sk_state == IUCV_DISCONN || sk->sk_state == IUCV_SEVERED) && 1161 skb_queue_empty(&iucv->backlog_skb_q) && 1162 skb_queue_empty(&sk->sk_receive_queue) && 1163 list_empty(&iucv->message_q.list)) 1164 return 0; 1165 1166 if (flags & (MSG_OOB)) 1167 return -EOPNOTSUPP; 1168 1169 /* receive/dequeue next skb: 1170 * the function understands MSG_PEEK and, thus, does not dequeue skb */ 1171 skb = skb_recv_datagram(sk, flags, noblock, &err); 1172 if (!skb) { 1173 if (sk->sk_shutdown & RCV_SHUTDOWN) 1174 return 0; 1175 return err; 1176 } 1177 1178 rlen = skb->len; /* real length of skb */ 1179 copied = min_t(unsigned int, rlen, len); 1180 1181 cskb = skb; 1182 if (memcpy_toiovec(msg->msg_iov, cskb->data, copied)) { 1183 if (!(flags & MSG_PEEK)) 1184 skb_queue_head(&sk->sk_receive_queue, skb); 1185 return -EFAULT; 1186 } 1187 1188 /* SOCK_SEQPACKET: set MSG_TRUNC if recv buf size is too small */ 1189 if (sk->sk_type == SOCK_SEQPACKET) { 1190 if (copied < rlen) 1191 msg->msg_flags |= MSG_TRUNC; 1192 /* each iucv message contains a complete record */ 1193 msg->msg_flags |= MSG_EOR; 1194 } 1195 1196 /* create control message to store iucv msg target class: 1197 * get the trgcls from the control buffer of the skb due to 1198 * fragmentation of original iucv message. */ 1199 err = put_cmsg(msg, SOL_IUCV, SCM_IUCV_TRGCLS, 1200 CB_TRGCLS_LEN, CB_TRGCLS(skb)); 1201 if (err) { 1202 if (!(flags & MSG_PEEK)) 1203 skb_queue_head(&sk->sk_receive_queue, skb); 1204 return err; 1205 } 1206 1207 /* Mark read part of skb as used */ 1208 if (!(flags & MSG_PEEK)) { 1209 1210 /* SOCK_STREAM: re-queue skb if it contains unreceived data */ 1211 if (sk->sk_type == SOCK_STREAM) { 1212 skb_pull(skb, copied); 1213 if (skb->len) { 1214 skb_queue_head(&sk->sk_receive_queue, skb); 1215 goto done; 1216 } 1217 } 1218 1219 kfree_skb(skb); 1220 1221 /* Queue backlog skbs */ 1222 spin_lock_bh(&iucv->message_q.lock); 1223 rskb = skb_dequeue(&iucv->backlog_skb_q); 1224 while (rskb) { 1225 if (sock_queue_rcv_skb(sk, rskb)) { 1226 skb_queue_head(&iucv->backlog_skb_q, 1227 rskb); 1228 break; 1229 } else { 1230 rskb = skb_dequeue(&iucv->backlog_skb_q); 1231 } 1232 } 1233 if (skb_queue_empty(&iucv->backlog_skb_q)) { 1234 if (!list_empty(&iucv->message_q.list)) 1235 iucv_process_message_q(sk); 1236 } 1237 spin_unlock_bh(&iucv->message_q.lock); 1238 } 1239 1240 done: 1241 /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */ 1242 if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC)) 1243 copied = rlen; 1244 1245 return copied; 1246 } 1247 1248 static inline unsigned int iucv_accept_poll(struct sock *parent) 1249 { 1250 struct iucv_sock *isk, *n; 1251 struct sock *sk; 1252 1253 list_for_each_entry_safe(isk, n, &iucv_sk(parent)->accept_q, accept_q) { 1254 sk = (struct sock *) isk; 1255 1256 if (sk->sk_state == IUCV_CONNECTED) 1257 return POLLIN | POLLRDNORM; 1258 } 1259 1260 return 0; 1261 } 1262 1263 unsigned int iucv_sock_poll(struct file *file, struct socket *sock, 1264 poll_table *wait) 1265 { 1266 struct sock *sk = sock->sk; 1267 unsigned int mask = 0; 1268 1269 sock_poll_wait(file, sk_sleep(sk), wait); 1270 1271 if (sk->sk_state == IUCV_LISTEN) 1272 return iucv_accept_poll(sk); 1273 1274 if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue)) 1275 mask |= POLLERR; 1276 1277 if (sk->sk_shutdown & RCV_SHUTDOWN) 1278 mask |= POLLRDHUP; 1279 1280 if (sk->sk_shutdown == SHUTDOWN_MASK) 1281 mask |= POLLHUP; 1282 1283 if (!skb_queue_empty(&sk->sk_receive_queue) || 1284 (sk->sk_shutdown & RCV_SHUTDOWN)) 1285 mask |= POLLIN | POLLRDNORM; 1286 1287 if (sk->sk_state == IUCV_CLOSED) 1288 mask |= POLLHUP; 1289 1290 if (sk->sk_state == IUCV_DISCONN || sk->sk_state == IUCV_SEVERED) 1291 mask |= POLLIN; 1292 1293 if (sock_writeable(sk)) 1294 mask |= POLLOUT | POLLWRNORM | POLLWRBAND; 1295 else 1296 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags); 1297 1298 return mask; 1299 } 1300 1301 static int iucv_sock_shutdown(struct socket *sock, int how) 1302 { 1303 struct sock *sk = sock->sk; 1304 struct iucv_sock *iucv = iucv_sk(sk); 1305 struct iucv_message txmsg; 1306 int err = 0; 1307 1308 how++; 1309 1310 if ((how & ~SHUTDOWN_MASK) || !how) 1311 return -EINVAL; 1312 1313 lock_sock(sk); 1314 switch (sk->sk_state) { 1315 case IUCV_DISCONN: 1316 case IUCV_CLOSING: 1317 case IUCV_SEVERED: 1318 case IUCV_CLOSED: 1319 err = -ENOTCONN; 1320 goto fail; 1321 1322 default: 1323 sk->sk_shutdown |= how; 1324 break; 1325 } 1326 1327 if (how == SEND_SHUTDOWN || how == SHUTDOWN_MASK) { 1328 txmsg.class = 0; 1329 txmsg.tag = 0; 1330 err = iucv_message_send(iucv->path, &txmsg, IUCV_IPRMDATA, 0, 1331 (void *) iprm_shutdown, 8); 1332 if (err) { 1333 switch (err) { 1334 case 1: 1335 err = -ENOTCONN; 1336 break; 1337 case 2: 1338 err = -ECONNRESET; 1339 break; 1340 default: 1341 err = -ENOTCONN; 1342 break; 1343 } 1344 } 1345 } 1346 1347 if (how == RCV_SHUTDOWN || how == SHUTDOWN_MASK) { 1348 err = iucv_path_quiesce(iucv_sk(sk)->path, NULL); 1349 if (err) 1350 err = -ENOTCONN; 1351 1352 skb_queue_purge(&sk->sk_receive_queue); 1353 } 1354 1355 /* Wake up anyone sleeping in poll */ 1356 sk->sk_state_change(sk); 1357 1358 fail: 1359 release_sock(sk); 1360 return err; 1361 } 1362 1363 static int iucv_sock_release(struct socket *sock) 1364 { 1365 struct sock *sk = sock->sk; 1366 int err = 0; 1367 1368 if (!sk) 1369 return 0; 1370 1371 iucv_sock_close(sk); 1372 1373 /* Unregister with IUCV base support */ 1374 if (iucv_sk(sk)->path) { 1375 iucv_path_sever(iucv_sk(sk)->path, NULL); 1376 iucv_path_free(iucv_sk(sk)->path); 1377 iucv_sk(sk)->path = NULL; 1378 } 1379 1380 sock_orphan(sk); 1381 iucv_sock_kill(sk); 1382 return err; 1383 } 1384 1385 /* getsockopt and setsockopt */ 1386 static int iucv_sock_setsockopt(struct socket *sock, int level, int optname, 1387 char __user *optval, unsigned int optlen) 1388 { 1389 struct sock *sk = sock->sk; 1390 struct iucv_sock *iucv = iucv_sk(sk); 1391 int val; 1392 int rc; 1393 1394 if (level != SOL_IUCV) 1395 return -ENOPROTOOPT; 1396 1397 if (optlen < sizeof(int)) 1398 return -EINVAL; 1399 1400 if (get_user(val, (int __user *) optval)) 1401 return -EFAULT; 1402 1403 rc = 0; 1404 1405 lock_sock(sk); 1406 switch (optname) { 1407 case SO_IPRMDATA_MSG: 1408 if (val) 1409 iucv->flags |= IUCV_IPRMDATA; 1410 else 1411 iucv->flags &= ~IUCV_IPRMDATA; 1412 break; 1413 case SO_MSGLIMIT: 1414 switch (sk->sk_state) { 1415 case IUCV_OPEN: 1416 case IUCV_BOUND: 1417 if (val < 1 || val > (u16)(~0)) 1418 rc = -EINVAL; 1419 else 1420 iucv->msglimit = val; 1421 break; 1422 default: 1423 rc = -EINVAL; 1424 break; 1425 } 1426 break; 1427 default: 1428 rc = -ENOPROTOOPT; 1429 break; 1430 } 1431 release_sock(sk); 1432 1433 return rc; 1434 } 1435 1436 static int iucv_sock_getsockopt(struct socket *sock, int level, int optname, 1437 char __user *optval, int __user *optlen) 1438 { 1439 struct sock *sk = sock->sk; 1440 struct iucv_sock *iucv = iucv_sk(sk); 1441 int val, len; 1442 1443 if (level != SOL_IUCV) 1444 return -ENOPROTOOPT; 1445 1446 if (get_user(len, optlen)) 1447 return -EFAULT; 1448 1449 if (len < 0) 1450 return -EINVAL; 1451 1452 len = min_t(unsigned int, len, sizeof(int)); 1453 1454 switch (optname) { 1455 case SO_IPRMDATA_MSG: 1456 val = (iucv->flags & IUCV_IPRMDATA) ? 1 : 0; 1457 break; 1458 case SO_MSGLIMIT: 1459 lock_sock(sk); 1460 val = (iucv->path != NULL) ? iucv->path->msglim /* connected */ 1461 : iucv->msglimit; /* default */ 1462 release_sock(sk); 1463 break; 1464 default: 1465 return -ENOPROTOOPT; 1466 } 1467 1468 if (put_user(len, optlen)) 1469 return -EFAULT; 1470 if (copy_to_user(optval, &val, len)) 1471 return -EFAULT; 1472 1473 return 0; 1474 } 1475 1476 1477 /* Callback wrappers - called from iucv base support */ 1478 static int iucv_callback_connreq(struct iucv_path *path, 1479 u8 ipvmid[8], u8 ipuser[16]) 1480 { 1481 unsigned char user_data[16]; 1482 unsigned char nuser_data[16]; 1483 unsigned char src_name[8]; 1484 struct hlist_node *node; 1485 struct sock *sk, *nsk; 1486 struct iucv_sock *iucv, *niucv; 1487 int err; 1488 1489 memcpy(src_name, ipuser, 8); 1490 EBCASC(src_name, 8); 1491 /* Find out if this path belongs to af_iucv. */ 1492 read_lock(&iucv_sk_list.lock); 1493 iucv = NULL; 1494 sk = NULL; 1495 sk_for_each(sk, node, &iucv_sk_list.head) 1496 if (sk->sk_state == IUCV_LISTEN && 1497 !memcmp(&iucv_sk(sk)->src_name, src_name, 8)) { 1498 /* 1499 * Found a listening socket with 1500 * src_name == ipuser[0-7]. 1501 */ 1502 iucv = iucv_sk(sk); 1503 break; 1504 } 1505 read_unlock(&iucv_sk_list.lock); 1506 if (!iucv) 1507 /* No socket found, not one of our paths. */ 1508 return -EINVAL; 1509 1510 bh_lock_sock(sk); 1511 1512 /* Check if parent socket is listening */ 1513 low_nmcpy(user_data, iucv->src_name); 1514 high_nmcpy(user_data, iucv->dst_name); 1515 ASCEBC(user_data, sizeof(user_data)); 1516 if (sk->sk_state != IUCV_LISTEN) { 1517 err = iucv_path_sever(path, user_data); 1518 iucv_path_free(path); 1519 goto fail; 1520 } 1521 1522 /* Check for backlog size */ 1523 if (sk_acceptq_is_full(sk)) { 1524 err = iucv_path_sever(path, user_data); 1525 iucv_path_free(path); 1526 goto fail; 1527 } 1528 1529 /* Create the new socket */ 1530 nsk = iucv_sock_alloc(NULL, sk->sk_type, GFP_ATOMIC); 1531 if (!nsk) { 1532 err = iucv_path_sever(path, user_data); 1533 iucv_path_free(path); 1534 goto fail; 1535 } 1536 1537 niucv = iucv_sk(nsk); 1538 iucv_sock_init(nsk, sk); 1539 1540 /* Set the new iucv_sock */ 1541 memcpy(niucv->dst_name, ipuser + 8, 8); 1542 EBCASC(niucv->dst_name, 8); 1543 memcpy(niucv->dst_user_id, ipvmid, 8); 1544 memcpy(niucv->src_name, iucv->src_name, 8); 1545 memcpy(niucv->src_user_id, iucv->src_user_id, 8); 1546 niucv->path = path; 1547 1548 /* Call iucv_accept */ 1549 high_nmcpy(nuser_data, ipuser + 8); 1550 memcpy(nuser_data + 8, niucv->src_name, 8); 1551 ASCEBC(nuser_data + 8, 8); 1552 1553 /* set message limit for path based on msglimit of accepting socket */ 1554 niucv->msglimit = iucv->msglimit; 1555 path->msglim = iucv->msglimit; 1556 err = iucv_path_accept(path, &af_iucv_handler, nuser_data, nsk); 1557 if (err) { 1558 err = iucv_path_sever(path, user_data); 1559 iucv_path_free(path); 1560 iucv_sock_kill(nsk); 1561 goto fail; 1562 } 1563 1564 iucv_accept_enqueue(sk, nsk); 1565 1566 /* Wake up accept */ 1567 nsk->sk_state = IUCV_CONNECTED; 1568 sk->sk_data_ready(sk, 1); 1569 err = 0; 1570 fail: 1571 bh_unlock_sock(sk); 1572 return 0; 1573 } 1574 1575 static void iucv_callback_connack(struct iucv_path *path, u8 ipuser[16]) 1576 { 1577 struct sock *sk = path->private; 1578 1579 sk->sk_state = IUCV_CONNECTED; 1580 sk->sk_state_change(sk); 1581 } 1582 1583 static void iucv_callback_rx(struct iucv_path *path, struct iucv_message *msg) 1584 { 1585 struct sock *sk = path->private; 1586 struct iucv_sock *iucv = iucv_sk(sk); 1587 struct sk_buff *skb; 1588 struct sock_msg_q *save_msg; 1589 int len; 1590 1591 if (sk->sk_shutdown & RCV_SHUTDOWN) { 1592 iucv_message_reject(path, msg); 1593 return; 1594 } 1595 1596 spin_lock(&iucv->message_q.lock); 1597 1598 if (!list_empty(&iucv->message_q.list) || 1599 !skb_queue_empty(&iucv->backlog_skb_q)) 1600 goto save_message; 1601 1602 len = atomic_read(&sk->sk_rmem_alloc); 1603 len += iucv_msg_length(msg) + sizeof(struct sk_buff); 1604 if (len > sk->sk_rcvbuf) 1605 goto save_message; 1606 1607 skb = alloc_skb(iucv_msg_length(msg), GFP_ATOMIC | GFP_DMA); 1608 if (!skb) 1609 goto save_message; 1610 1611 iucv_process_message(sk, skb, path, msg); 1612 goto out_unlock; 1613 1614 save_message: 1615 save_msg = kzalloc(sizeof(struct sock_msg_q), GFP_ATOMIC | GFP_DMA); 1616 if (!save_msg) 1617 goto out_unlock; 1618 save_msg->path = path; 1619 save_msg->msg = *msg; 1620 1621 list_add_tail(&save_msg->list, &iucv->message_q.list); 1622 1623 out_unlock: 1624 spin_unlock(&iucv->message_q.lock); 1625 } 1626 1627 static void iucv_callback_txdone(struct iucv_path *path, 1628 struct iucv_message *msg) 1629 { 1630 struct sock *sk = path->private; 1631 struct sk_buff *this = NULL; 1632 struct sk_buff_head *list = &iucv_sk(sk)->send_skb_q; 1633 struct sk_buff *list_skb = list->next; 1634 unsigned long flags; 1635 1636 if (!skb_queue_empty(list)) { 1637 spin_lock_irqsave(&list->lock, flags); 1638 1639 while (list_skb != (struct sk_buff *)list) { 1640 if (!memcmp(&msg->tag, CB_TAG(list_skb), CB_TAG_LEN)) { 1641 this = list_skb; 1642 break; 1643 } 1644 list_skb = list_skb->next; 1645 } 1646 if (this) 1647 __skb_unlink(this, list); 1648 1649 spin_unlock_irqrestore(&list->lock, flags); 1650 1651 if (this) { 1652 kfree_skb(this); 1653 /* wake up any process waiting for sending */ 1654 iucv_sock_wake_msglim(sk); 1655 } 1656 } 1657 BUG_ON(!this); 1658 1659 if (sk->sk_state == IUCV_CLOSING) { 1660 if (skb_queue_empty(&iucv_sk(sk)->send_skb_q)) { 1661 sk->sk_state = IUCV_CLOSED; 1662 sk->sk_state_change(sk); 1663 } 1664 } 1665 1666 } 1667 1668 static void iucv_callback_connrej(struct iucv_path *path, u8 ipuser[16]) 1669 { 1670 struct sock *sk = path->private; 1671 1672 if (!list_empty(&iucv_sk(sk)->accept_q)) 1673 sk->sk_state = IUCV_SEVERED; 1674 else 1675 sk->sk_state = IUCV_DISCONN; 1676 1677 sk->sk_state_change(sk); 1678 } 1679 1680 /* called if the other communication side shuts down its RECV direction; 1681 * in turn, the callback sets SEND_SHUTDOWN to disable sending of data. 1682 */ 1683 static void iucv_callback_shutdown(struct iucv_path *path, u8 ipuser[16]) 1684 { 1685 struct sock *sk = path->private; 1686 1687 bh_lock_sock(sk); 1688 if (sk->sk_state != IUCV_CLOSED) { 1689 sk->sk_shutdown |= SEND_SHUTDOWN; 1690 sk->sk_state_change(sk); 1691 } 1692 bh_unlock_sock(sk); 1693 } 1694 1695 static const struct proto_ops iucv_sock_ops = { 1696 .family = PF_IUCV, 1697 .owner = THIS_MODULE, 1698 .release = iucv_sock_release, 1699 .bind = iucv_sock_bind, 1700 .connect = iucv_sock_connect, 1701 .listen = iucv_sock_listen, 1702 .accept = iucv_sock_accept, 1703 .getname = iucv_sock_getname, 1704 .sendmsg = iucv_sock_sendmsg, 1705 .recvmsg = iucv_sock_recvmsg, 1706 .poll = iucv_sock_poll, 1707 .ioctl = sock_no_ioctl, 1708 .mmap = sock_no_mmap, 1709 .socketpair = sock_no_socketpair, 1710 .shutdown = iucv_sock_shutdown, 1711 .setsockopt = iucv_sock_setsockopt, 1712 .getsockopt = iucv_sock_getsockopt, 1713 }; 1714 1715 static const struct net_proto_family iucv_sock_family_ops = { 1716 .family = AF_IUCV, 1717 .owner = THIS_MODULE, 1718 .create = iucv_sock_create, 1719 }; 1720 1721 static int __init afiucv_init(void) 1722 { 1723 int err; 1724 1725 if (!MACHINE_IS_VM) { 1726 pr_err("The af_iucv module cannot be loaded" 1727 " without z/VM\n"); 1728 err = -EPROTONOSUPPORT; 1729 goto out; 1730 } 1731 cpcmd("QUERY USERID", iucv_userid, sizeof(iucv_userid), &err); 1732 if (unlikely(err)) { 1733 WARN_ON(err); 1734 err = -EPROTONOSUPPORT; 1735 goto out; 1736 } 1737 1738 err = iucv_register(&af_iucv_handler, 0); 1739 if (err) 1740 goto out; 1741 err = proto_register(&iucv_proto, 0); 1742 if (err) 1743 goto out_iucv; 1744 err = sock_register(&iucv_sock_family_ops); 1745 if (err) 1746 goto out_proto; 1747 /* establish dummy device */ 1748 err = driver_register(&af_iucv_driver); 1749 if (err) 1750 goto out_sock; 1751 af_iucv_dev = kzalloc(sizeof(struct device), GFP_KERNEL); 1752 if (!af_iucv_dev) { 1753 err = -ENOMEM; 1754 goto out_driver; 1755 } 1756 dev_set_name(af_iucv_dev, "af_iucv"); 1757 af_iucv_dev->bus = &iucv_bus; 1758 af_iucv_dev->parent = iucv_root; 1759 af_iucv_dev->release = (void (*)(struct device *))kfree; 1760 af_iucv_dev->driver = &af_iucv_driver; 1761 err = device_register(af_iucv_dev); 1762 if (err) 1763 goto out_driver; 1764 1765 return 0; 1766 1767 out_driver: 1768 driver_unregister(&af_iucv_driver); 1769 out_sock: 1770 sock_unregister(PF_IUCV); 1771 out_proto: 1772 proto_unregister(&iucv_proto); 1773 out_iucv: 1774 iucv_unregister(&af_iucv_handler, 0); 1775 out: 1776 return err; 1777 } 1778 1779 static void __exit afiucv_exit(void) 1780 { 1781 device_unregister(af_iucv_dev); 1782 driver_unregister(&af_iucv_driver); 1783 sock_unregister(PF_IUCV); 1784 proto_unregister(&iucv_proto); 1785 iucv_unregister(&af_iucv_handler, 0); 1786 } 1787 1788 module_init(afiucv_init); 1789 module_exit(afiucv_exit); 1790 1791 MODULE_AUTHOR("Jennifer Hunt <jenhunt@us.ibm.com>"); 1792 MODULE_DESCRIPTION("IUCV Sockets ver " VERSION); 1793 MODULE_VERSION(VERSION); 1794 MODULE_LICENSE("GPL"); 1795 MODULE_ALIAS_NETPROTO(PF_IUCV); 1796