• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  *  IUCV protocol stack for Linux on zSeries
4  *
5  *  Copyright IBM Corp. 2006, 2009
6  *
7  *  Author(s):	Jennifer Hunt <jenhunt@us.ibm.com>
8  *		Hendrik Brueckner <brueckner@linux.vnet.ibm.com>
9  *  PM functions:
10  *		Ursula Braun <ursula.braun@de.ibm.com>
11  */
12 
13 #define KMSG_COMPONENT "af_iucv"
14 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
15 
16 #include <linux/module.h>
17 #include <linux/netdevice.h>
18 #include <linux/types.h>
19 #include <linux/list.h>
20 #include <linux/errno.h>
21 #include <linux/kernel.h>
22 #include <linux/sched/signal.h>
23 #include <linux/slab.h>
24 #include <linux/skbuff.h>
25 #include <linux/init.h>
26 #include <linux/poll.h>
27 #include <linux/security.h>
28 #include <net/sock.h>
29 #include <asm/ebcdic.h>
30 #include <asm/cpcmd.h>
31 #include <linux/kmod.h>
32 
33 #include <net/iucv/af_iucv.h>
34 
35 #define VERSION "1.2"
36 
37 static char iucv_userid[80];
38 
39 static const struct proto_ops iucv_sock_ops;
40 
41 static struct proto iucv_proto = {
42 	.name		= "AF_IUCV",
43 	.owner		= THIS_MODULE,
44 	.obj_size	= sizeof(struct iucv_sock),
45 };
46 
47 static struct iucv_interface *pr_iucv;
48 
49 /* special AF_IUCV IPRM messages */
50 static const u8 iprm_shutdown[8] =
51 	{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01};
52 
53 #define TRGCLS_SIZE	FIELD_SIZEOF(struct iucv_message, class)
54 
55 #define __iucv_sock_wait(sk, condition, timeo, ret)			\
56 do {									\
57 	DEFINE_WAIT(__wait);						\
58 	long __timeo = timeo;						\
59 	ret = 0;							\
60 	prepare_to_wait(sk_sleep(sk), &__wait, TASK_INTERRUPTIBLE);	\
61 	while (!(condition)) {						\
62 		if (!__timeo) {						\
63 			ret = -EAGAIN;					\
64 			break;						\
65 		}							\
66 		if (signal_pending(current)) {				\
67 			ret = sock_intr_errno(__timeo);			\
68 			break;						\
69 		}							\
70 		release_sock(sk);					\
71 		__timeo = schedule_timeout(__timeo);			\
72 		lock_sock(sk);						\
73 		ret = sock_error(sk);					\
74 		if (ret)						\
75 			break;						\
76 	}								\
77 	finish_wait(sk_sleep(sk), &__wait);				\
78 } while (0)
79 
80 #define iucv_sock_wait(sk, condition, timeo)				\
81 ({									\
82 	int __ret = 0;							\
83 	if (!(condition))						\
84 		__iucv_sock_wait(sk, condition, timeo, __ret);		\
85 	__ret;								\
86 })
87 
88 static void iucv_sock_kill(struct sock *sk);
89 static void iucv_sock_close(struct sock *sk);
90 static void iucv_sever_path(struct sock *, int);
91 
92 static int afiucv_hs_rcv(struct sk_buff *skb, struct net_device *dev,
93 	struct packet_type *pt, struct net_device *orig_dev);
94 static int afiucv_hs_send(struct iucv_message *imsg, struct sock *sock,
95 		   struct sk_buff *skb, u8 flags);
96 static void afiucv_hs_callback_txnotify(struct sk_buff *, enum iucv_tx_notify);
97 
98 /* Call Back functions */
99 static void iucv_callback_rx(struct iucv_path *, struct iucv_message *);
100 static void iucv_callback_txdone(struct iucv_path *, struct iucv_message *);
101 static void iucv_callback_connack(struct iucv_path *, u8 *);
102 static int iucv_callback_connreq(struct iucv_path *, u8 *, u8 *);
103 static void iucv_callback_connrej(struct iucv_path *, u8 *);
104 static void iucv_callback_shutdown(struct iucv_path *, u8 *);
105 
106 static struct iucv_sock_list iucv_sk_list = {
107 	.lock = __RW_LOCK_UNLOCKED(iucv_sk_list.lock),
108 	.autobind_name = ATOMIC_INIT(0)
109 };
110 
111 static struct iucv_handler af_iucv_handler = {
112 	.path_pending	  = iucv_callback_connreq,
113 	.path_complete	  = iucv_callback_connack,
114 	.path_severed	  = iucv_callback_connrej,
115 	.message_pending  = iucv_callback_rx,
116 	.message_complete = iucv_callback_txdone,
117 	.path_quiesced	  = iucv_callback_shutdown,
118 };
119 
high_nmcpy(unsigned char * dst,char * src)120 static inline void high_nmcpy(unsigned char *dst, char *src)
121 {
122        memcpy(dst, src, 8);
123 }
124 
low_nmcpy(unsigned char * dst,char * src)125 static inline void low_nmcpy(unsigned char *dst, char *src)
126 {
127        memcpy(&dst[8], src, 8);
128 }
129 
afiucv_pm_prepare(struct device * dev)130 static int afiucv_pm_prepare(struct device *dev)
131 {
132 #ifdef CONFIG_PM_DEBUG
133 	printk(KERN_WARNING "afiucv_pm_prepare\n");
134 #endif
135 	return 0;
136 }
137 
afiucv_pm_complete(struct device * dev)138 static void afiucv_pm_complete(struct device *dev)
139 {
140 #ifdef CONFIG_PM_DEBUG
141 	printk(KERN_WARNING "afiucv_pm_complete\n");
142 #endif
143 }
144 
145 /**
146  * afiucv_pm_freeze() - Freeze PM callback
147  * @dev:	AFIUCV dummy device
148  *
149  * Sever all established IUCV communication pathes
150  */
afiucv_pm_freeze(struct device * dev)151 static int afiucv_pm_freeze(struct device *dev)
152 {
153 	struct iucv_sock *iucv;
154 	struct sock *sk;
155 
156 #ifdef CONFIG_PM_DEBUG
157 	printk(KERN_WARNING "afiucv_pm_freeze\n");
158 #endif
159 	read_lock(&iucv_sk_list.lock);
160 	sk_for_each(sk, &iucv_sk_list.head) {
161 		iucv = iucv_sk(sk);
162 		switch (sk->sk_state) {
163 		case IUCV_DISCONN:
164 		case IUCV_CLOSING:
165 		case IUCV_CONNECTED:
166 			iucv_sever_path(sk, 0);
167 			break;
168 		case IUCV_OPEN:
169 		case IUCV_BOUND:
170 		case IUCV_LISTEN:
171 		case IUCV_CLOSED:
172 		default:
173 			break;
174 		}
175 		skb_queue_purge(&iucv->send_skb_q);
176 		skb_queue_purge(&iucv->backlog_skb_q);
177 	}
178 	read_unlock(&iucv_sk_list.lock);
179 	return 0;
180 }
181 
182 /**
183  * afiucv_pm_restore_thaw() - Thaw and restore PM callback
184  * @dev:	AFIUCV dummy device
185  *
186  * socket clean up after freeze
187  */
afiucv_pm_restore_thaw(struct device * dev)188 static int afiucv_pm_restore_thaw(struct device *dev)
189 {
190 	struct sock *sk;
191 
192 #ifdef CONFIG_PM_DEBUG
193 	printk(KERN_WARNING "afiucv_pm_restore_thaw\n");
194 #endif
195 	read_lock(&iucv_sk_list.lock);
196 	sk_for_each(sk, &iucv_sk_list.head) {
197 		switch (sk->sk_state) {
198 		case IUCV_CONNECTED:
199 			sk->sk_err = EPIPE;
200 			sk->sk_state = IUCV_DISCONN;
201 			sk->sk_state_change(sk);
202 			break;
203 		case IUCV_DISCONN:
204 		case IUCV_CLOSING:
205 		case IUCV_LISTEN:
206 		case IUCV_BOUND:
207 		case IUCV_OPEN:
208 		default:
209 			break;
210 		}
211 	}
212 	read_unlock(&iucv_sk_list.lock);
213 	return 0;
214 }
215 
216 static const struct dev_pm_ops afiucv_pm_ops = {
217 	.prepare = afiucv_pm_prepare,
218 	.complete = afiucv_pm_complete,
219 	.freeze = afiucv_pm_freeze,
220 	.thaw = afiucv_pm_restore_thaw,
221 	.restore = afiucv_pm_restore_thaw,
222 };
223 
224 static struct device_driver af_iucv_driver = {
225 	.owner = THIS_MODULE,
226 	.name = "afiucv",
227 	.bus  = NULL,
228 	.pm   = &afiucv_pm_ops,
229 };
230 
231 /* dummy device used as trigger for PM functions */
232 static struct device *af_iucv_dev;
233 
234 /**
235  * iucv_msg_length() - Returns the length of an iucv message.
236  * @msg:	Pointer to struct iucv_message, MUST NOT be NULL
237  *
238  * The function returns the length of the specified iucv message @msg of data
239  * stored in a buffer and of data stored in the parameter list (PRMDATA).
240  *
241  * For IUCV_IPRMDATA, AF_IUCV uses the following convention to transport socket
242  * data:
243  *	PRMDATA[0..6]	socket data (max 7 bytes);
244  *	PRMDATA[7]	socket data length value (len is 0xff - PRMDATA[7])
245  *
246  * The socket data length is computed by subtracting the socket data length
247  * value from 0xFF.
248  * If the socket data len is greater 7, then PRMDATA can be used for special
249  * notifications (see iucv_sock_shutdown); and further,
250  * if the socket data len is > 7, the function returns 8.
251  *
252  * Use this function to allocate socket buffers to store iucv message data.
253  */
iucv_msg_length(struct iucv_message * msg)254 static inline size_t iucv_msg_length(struct iucv_message *msg)
255 {
256 	size_t datalen;
257 
258 	if (msg->flags & IUCV_IPRMDATA) {
259 		datalen = 0xff - msg->rmmsg[7];
260 		return (datalen < 8) ? datalen : 8;
261 	}
262 	return msg->length;
263 }
264 
265 /**
266  * iucv_sock_in_state() - check for specific states
267  * @sk:		sock structure
268  * @state:	first iucv sk state
269  * @state:	second iucv sk state
270  *
271  * Returns true if the socket in either in the first or second state.
272  */
iucv_sock_in_state(struct sock * sk,int state,int state2)273 static int iucv_sock_in_state(struct sock *sk, int state, int state2)
274 {
275 	return (sk->sk_state == state || sk->sk_state == state2);
276 }
277 
278 /**
279  * iucv_below_msglim() - function to check if messages can be sent
280  * @sk:		sock structure
281  *
282  * Returns true if the send queue length is lower than the message limit.
283  * Always returns true if the socket is not connected (no iucv path for
284  * checking the message limit).
285  */
iucv_below_msglim(struct sock * sk)286 static inline int iucv_below_msglim(struct sock *sk)
287 {
288 	struct iucv_sock *iucv = iucv_sk(sk);
289 
290 	if (sk->sk_state != IUCV_CONNECTED)
291 		return 1;
292 	if (iucv->transport == AF_IUCV_TRANS_IUCV)
293 		return (skb_queue_len(&iucv->send_skb_q) < iucv->path->msglim);
294 	else
295 		return ((atomic_read(&iucv->msg_sent) < iucv->msglimit_peer) &&
296 			(atomic_read(&iucv->pendings) <= 0));
297 }
298 
299 /**
300  * iucv_sock_wake_msglim() - Wake up thread waiting on msg limit
301  */
iucv_sock_wake_msglim(struct sock * sk)302 static void iucv_sock_wake_msglim(struct sock *sk)
303 {
304 	struct socket_wq *wq;
305 
306 	rcu_read_lock();
307 	wq = rcu_dereference(sk->sk_wq);
308 	if (skwq_has_sleeper(wq))
309 		wake_up_interruptible_all(&wq->wait);
310 	sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
311 	rcu_read_unlock();
312 }
313 
314 /**
315  * afiucv_hs_send() - send a message through HiperSockets transport
316  */
afiucv_hs_send(struct iucv_message * imsg,struct sock * sock,struct sk_buff * skb,u8 flags)317 static int afiucv_hs_send(struct iucv_message *imsg, struct sock *sock,
318 		   struct sk_buff *skb, u8 flags)
319 {
320 	struct iucv_sock *iucv = iucv_sk(sock);
321 	struct af_iucv_trans_hdr *phs_hdr;
322 	struct sk_buff *nskb;
323 	int err, confirm_recv = 0;
324 
325 	phs_hdr = skb_push(skb, sizeof(*phs_hdr));
326 	memset(phs_hdr, 0, sizeof(*phs_hdr));
327 	skb_reset_network_header(skb);
328 
329 	phs_hdr->magic = ETH_P_AF_IUCV;
330 	phs_hdr->version = 1;
331 	phs_hdr->flags = flags;
332 	if (flags == AF_IUCV_FLAG_SYN)
333 		phs_hdr->window = iucv->msglimit;
334 	else if ((flags == AF_IUCV_FLAG_WIN) || !flags) {
335 		confirm_recv = atomic_read(&iucv->msg_recv);
336 		phs_hdr->window = confirm_recv;
337 		if (confirm_recv)
338 			phs_hdr->flags = phs_hdr->flags | AF_IUCV_FLAG_WIN;
339 	}
340 	memcpy(phs_hdr->destUserID, iucv->dst_user_id, 8);
341 	memcpy(phs_hdr->destAppName, iucv->dst_name, 8);
342 	memcpy(phs_hdr->srcUserID, iucv->src_user_id, 8);
343 	memcpy(phs_hdr->srcAppName, iucv->src_name, 8);
344 	ASCEBC(phs_hdr->destUserID, sizeof(phs_hdr->destUserID));
345 	ASCEBC(phs_hdr->destAppName, sizeof(phs_hdr->destAppName));
346 	ASCEBC(phs_hdr->srcUserID, sizeof(phs_hdr->srcUserID));
347 	ASCEBC(phs_hdr->srcAppName, sizeof(phs_hdr->srcAppName));
348 	if (imsg)
349 		memcpy(&phs_hdr->iucv_hdr, imsg, sizeof(struct iucv_message));
350 
351 	skb->dev = iucv->hs_dev;
352 	if (!skb->dev) {
353 		err = -ENODEV;
354 		goto err_free;
355 	}
356 
357 	dev_hard_header(skb, skb->dev, ETH_P_AF_IUCV, NULL, NULL, skb->len);
358 
359 	if (!(skb->dev->flags & IFF_UP) || !netif_carrier_ok(skb->dev)) {
360 		err = -ENETDOWN;
361 		goto err_free;
362 	}
363 	if (skb->len > skb->dev->mtu) {
364 		if (sock->sk_type == SOCK_SEQPACKET) {
365 			err = -EMSGSIZE;
366 			goto err_free;
367 		}
368 		skb_trim(skb, skb->dev->mtu);
369 	}
370 	skb->protocol = cpu_to_be16(ETH_P_AF_IUCV);
371 
372 	__skb_header_release(skb);
373 	nskb = skb_clone(skb, GFP_ATOMIC);
374 	if (!nskb) {
375 		err = -ENOMEM;
376 		goto err_free;
377 	}
378 
379 	skb_queue_tail(&iucv->send_skb_q, nskb);
380 	err = dev_queue_xmit(skb);
381 	if (net_xmit_eval(err)) {
382 		skb_unlink(nskb, &iucv->send_skb_q);
383 		kfree_skb(nskb);
384 	} else {
385 		atomic_sub(confirm_recv, &iucv->msg_recv);
386 		WARN_ON(atomic_read(&iucv->msg_recv) < 0);
387 	}
388 	return net_xmit_eval(err);
389 
390 err_free:
391 	kfree_skb(skb);
392 	return err;
393 }
394 
__iucv_get_sock_by_name(char * nm)395 static struct sock *__iucv_get_sock_by_name(char *nm)
396 {
397 	struct sock *sk;
398 
399 	sk_for_each(sk, &iucv_sk_list.head)
400 		if (!memcmp(&iucv_sk(sk)->src_name, nm, 8))
401 			return sk;
402 
403 	return NULL;
404 }
405 
iucv_sock_destruct(struct sock * sk)406 static void iucv_sock_destruct(struct sock *sk)
407 {
408 	skb_queue_purge(&sk->sk_receive_queue);
409 	skb_queue_purge(&sk->sk_error_queue);
410 
411 	sk_mem_reclaim(sk);
412 
413 	if (!sock_flag(sk, SOCK_DEAD)) {
414 		pr_err("Attempt to release alive iucv socket %p\n", sk);
415 		return;
416 	}
417 
418 	WARN_ON(atomic_read(&sk->sk_rmem_alloc));
419 	WARN_ON(refcount_read(&sk->sk_wmem_alloc));
420 	WARN_ON(sk->sk_wmem_queued);
421 	WARN_ON(sk->sk_forward_alloc);
422 }
423 
424 /* Cleanup Listen */
iucv_sock_cleanup_listen(struct sock * parent)425 static void iucv_sock_cleanup_listen(struct sock *parent)
426 {
427 	struct sock *sk;
428 
429 	/* Close non-accepted connections */
430 	while ((sk = iucv_accept_dequeue(parent, NULL))) {
431 		iucv_sock_close(sk);
432 		iucv_sock_kill(sk);
433 	}
434 
435 	parent->sk_state = IUCV_CLOSED;
436 }
437 
438 /* Kill socket (only if zapped and orphaned) */
iucv_sock_kill(struct sock * sk)439 static void iucv_sock_kill(struct sock *sk)
440 {
441 	if (!sock_flag(sk, SOCK_ZAPPED) || sk->sk_socket)
442 		return;
443 
444 	iucv_sock_unlink(&iucv_sk_list, sk);
445 	sock_set_flag(sk, SOCK_DEAD);
446 	sock_put(sk);
447 }
448 
449 /* Terminate an IUCV path */
iucv_sever_path(struct sock * sk,int with_user_data)450 static void iucv_sever_path(struct sock *sk, int with_user_data)
451 {
452 	unsigned char user_data[16];
453 	struct iucv_sock *iucv = iucv_sk(sk);
454 	struct iucv_path *path = iucv->path;
455 
456 	if (iucv->path) {
457 		iucv->path = NULL;
458 		if (with_user_data) {
459 			low_nmcpy(user_data, iucv->src_name);
460 			high_nmcpy(user_data, iucv->dst_name);
461 			ASCEBC(user_data, sizeof(user_data));
462 			pr_iucv->path_sever(path, user_data);
463 		} else
464 			pr_iucv->path_sever(path, NULL);
465 		iucv_path_free(path);
466 	}
467 }
468 
469 /* Send controlling flags through an IUCV socket for HIPER transport */
iucv_send_ctrl(struct sock * sk,u8 flags)470 static int iucv_send_ctrl(struct sock *sk, u8 flags)
471 {
472 	struct iucv_sock *iucv = iucv_sk(sk);
473 	int err = 0;
474 	int blen;
475 	struct sk_buff *skb;
476 	u8 shutdown = 0;
477 
478 	blen = sizeof(struct af_iucv_trans_hdr) +
479 	       LL_RESERVED_SPACE(iucv->hs_dev);
480 	if (sk->sk_shutdown & SEND_SHUTDOWN) {
481 		/* controlling flags should be sent anyway */
482 		shutdown = sk->sk_shutdown;
483 		sk->sk_shutdown &= RCV_SHUTDOWN;
484 	}
485 	skb = sock_alloc_send_skb(sk, blen, 1, &err);
486 	if (skb) {
487 		skb_reserve(skb, blen);
488 		err = afiucv_hs_send(NULL, sk, skb, flags);
489 	}
490 	if (shutdown)
491 		sk->sk_shutdown = shutdown;
492 	return err;
493 }
494 
495 /* Close an IUCV socket */
iucv_sock_close(struct sock * sk)496 static void iucv_sock_close(struct sock *sk)
497 {
498 	struct iucv_sock *iucv = iucv_sk(sk);
499 	unsigned long timeo;
500 	int err = 0;
501 
502 	lock_sock(sk);
503 
504 	switch (sk->sk_state) {
505 	case IUCV_LISTEN:
506 		iucv_sock_cleanup_listen(sk);
507 		break;
508 
509 	case IUCV_CONNECTED:
510 		if (iucv->transport == AF_IUCV_TRANS_HIPER) {
511 			err = iucv_send_ctrl(sk, AF_IUCV_FLAG_FIN);
512 			sk->sk_state = IUCV_DISCONN;
513 			sk->sk_state_change(sk);
514 		}
515 		/* fall through */
516 
517 	case IUCV_DISCONN:
518 		sk->sk_state = IUCV_CLOSING;
519 		sk->sk_state_change(sk);
520 
521 		if (!err && !skb_queue_empty(&iucv->send_skb_q)) {
522 			if (sock_flag(sk, SOCK_LINGER) && sk->sk_lingertime)
523 				timeo = sk->sk_lingertime;
524 			else
525 				timeo = IUCV_DISCONN_TIMEOUT;
526 			iucv_sock_wait(sk,
527 					iucv_sock_in_state(sk, IUCV_CLOSED, 0),
528 					timeo);
529 		}
530 		/* fall through */
531 
532 	case IUCV_CLOSING:
533 		sk->sk_state = IUCV_CLOSED;
534 		sk->sk_state_change(sk);
535 
536 		sk->sk_err = ECONNRESET;
537 		sk->sk_state_change(sk);
538 
539 		skb_queue_purge(&iucv->send_skb_q);
540 		skb_queue_purge(&iucv->backlog_skb_q);
541 		/* fall through */
542 
543 	default:
544 		iucv_sever_path(sk, 1);
545 	}
546 
547 	if (iucv->hs_dev) {
548 		dev_put(iucv->hs_dev);
549 		iucv->hs_dev = NULL;
550 		sk->sk_bound_dev_if = 0;
551 	}
552 
553 	/* mark socket for deletion by iucv_sock_kill() */
554 	sock_set_flag(sk, SOCK_ZAPPED);
555 
556 	release_sock(sk);
557 }
558 
iucv_sock_init(struct sock * sk,struct sock * parent)559 static void iucv_sock_init(struct sock *sk, struct sock *parent)
560 {
561 	if (parent) {
562 		sk->sk_type = parent->sk_type;
563 		security_sk_clone(parent, sk);
564 	}
565 }
566 
iucv_sock_alloc(struct socket * sock,int proto,gfp_t prio,int kern)567 static struct sock *iucv_sock_alloc(struct socket *sock, int proto, gfp_t prio, int kern)
568 {
569 	struct sock *sk;
570 	struct iucv_sock *iucv;
571 
572 	sk = sk_alloc(&init_net, PF_IUCV, prio, &iucv_proto, kern);
573 	if (!sk)
574 		return NULL;
575 	iucv = iucv_sk(sk);
576 
577 	sock_init_data(sock, sk);
578 	INIT_LIST_HEAD(&iucv->accept_q);
579 	spin_lock_init(&iucv->accept_q_lock);
580 	skb_queue_head_init(&iucv->send_skb_q);
581 	INIT_LIST_HEAD(&iucv->message_q.list);
582 	spin_lock_init(&iucv->message_q.lock);
583 	skb_queue_head_init(&iucv->backlog_skb_q);
584 	iucv->send_tag = 0;
585 	atomic_set(&iucv->pendings, 0);
586 	iucv->flags = 0;
587 	iucv->msglimit = 0;
588 	atomic_set(&iucv->msg_sent, 0);
589 	atomic_set(&iucv->msg_recv, 0);
590 	iucv->path = NULL;
591 	iucv->sk_txnotify = afiucv_hs_callback_txnotify;
592 	memset(&iucv->src_user_id , 0, 32);
593 	if (pr_iucv)
594 		iucv->transport = AF_IUCV_TRANS_IUCV;
595 	else
596 		iucv->transport = AF_IUCV_TRANS_HIPER;
597 
598 	sk->sk_destruct = iucv_sock_destruct;
599 	sk->sk_sndtimeo = IUCV_CONN_TIMEOUT;
600 
601 	sock_reset_flag(sk, SOCK_ZAPPED);
602 
603 	sk->sk_protocol = proto;
604 	sk->sk_state	= IUCV_OPEN;
605 
606 	iucv_sock_link(&iucv_sk_list, sk);
607 	return sk;
608 }
609 
610 /* Create an IUCV socket */
iucv_sock_create(struct net * net,struct socket * sock,int protocol,int kern)611 static int iucv_sock_create(struct net *net, struct socket *sock, int protocol,
612 			    int kern)
613 {
614 	struct sock *sk;
615 
616 	if (protocol && protocol != PF_IUCV)
617 		return -EPROTONOSUPPORT;
618 
619 	sock->state = SS_UNCONNECTED;
620 
621 	switch (sock->type) {
622 	case SOCK_STREAM:
623 		sock->ops = &iucv_sock_ops;
624 		break;
625 	case SOCK_SEQPACKET:
626 		/* currently, proto ops can handle both sk types */
627 		sock->ops = &iucv_sock_ops;
628 		break;
629 	default:
630 		return -ESOCKTNOSUPPORT;
631 	}
632 
633 	sk = iucv_sock_alloc(sock, protocol, GFP_KERNEL, kern);
634 	if (!sk)
635 		return -ENOMEM;
636 
637 	iucv_sock_init(sk, NULL);
638 
639 	return 0;
640 }
641 
iucv_sock_link(struct iucv_sock_list * l,struct sock * sk)642 void iucv_sock_link(struct iucv_sock_list *l, struct sock *sk)
643 {
644 	write_lock_bh(&l->lock);
645 	sk_add_node(sk, &l->head);
646 	write_unlock_bh(&l->lock);
647 }
648 
iucv_sock_unlink(struct iucv_sock_list * l,struct sock * sk)649 void iucv_sock_unlink(struct iucv_sock_list *l, struct sock *sk)
650 {
651 	write_lock_bh(&l->lock);
652 	sk_del_node_init(sk);
653 	write_unlock_bh(&l->lock);
654 }
655 
iucv_accept_enqueue(struct sock * parent,struct sock * sk)656 void iucv_accept_enqueue(struct sock *parent, struct sock *sk)
657 {
658 	unsigned long flags;
659 	struct iucv_sock *par = iucv_sk(parent);
660 
661 	sock_hold(sk);
662 	spin_lock_irqsave(&par->accept_q_lock, flags);
663 	list_add_tail(&iucv_sk(sk)->accept_q, &par->accept_q);
664 	spin_unlock_irqrestore(&par->accept_q_lock, flags);
665 	iucv_sk(sk)->parent = parent;
666 	sk_acceptq_added(parent);
667 }
668 
iucv_accept_unlink(struct sock * sk)669 void iucv_accept_unlink(struct sock *sk)
670 {
671 	unsigned long flags;
672 	struct iucv_sock *par = iucv_sk(iucv_sk(sk)->parent);
673 
674 	spin_lock_irqsave(&par->accept_q_lock, flags);
675 	list_del_init(&iucv_sk(sk)->accept_q);
676 	spin_unlock_irqrestore(&par->accept_q_lock, flags);
677 	sk_acceptq_removed(iucv_sk(sk)->parent);
678 	iucv_sk(sk)->parent = NULL;
679 	sock_put(sk);
680 }
681 
iucv_accept_dequeue(struct sock * parent,struct socket * newsock)682 struct sock *iucv_accept_dequeue(struct sock *parent, struct socket *newsock)
683 {
684 	struct iucv_sock *isk, *n;
685 	struct sock *sk;
686 
687 	list_for_each_entry_safe(isk, n, &iucv_sk(parent)->accept_q, accept_q) {
688 		sk = (struct sock *) isk;
689 		lock_sock(sk);
690 
691 		if (sk->sk_state == IUCV_CLOSED) {
692 			iucv_accept_unlink(sk);
693 			release_sock(sk);
694 			continue;
695 		}
696 
697 		if (sk->sk_state == IUCV_CONNECTED ||
698 		    sk->sk_state == IUCV_DISCONN ||
699 		    !newsock) {
700 			iucv_accept_unlink(sk);
701 			if (newsock)
702 				sock_graft(sk, newsock);
703 
704 			release_sock(sk);
705 			return sk;
706 		}
707 
708 		release_sock(sk);
709 	}
710 	return NULL;
711 }
712 
__iucv_auto_name(struct iucv_sock * iucv)713 static void __iucv_auto_name(struct iucv_sock *iucv)
714 {
715 	char name[12];
716 
717 	sprintf(name, "%08x", atomic_inc_return(&iucv_sk_list.autobind_name));
718 	while (__iucv_get_sock_by_name(name)) {
719 		sprintf(name, "%08x",
720 			atomic_inc_return(&iucv_sk_list.autobind_name));
721 	}
722 	memcpy(iucv->src_name, name, 8);
723 }
724 
725 /* Bind an unbound socket */
iucv_sock_bind(struct socket * sock,struct sockaddr * addr,int addr_len)726 static int iucv_sock_bind(struct socket *sock, struct sockaddr *addr,
727 			  int addr_len)
728 {
729 	struct sockaddr_iucv *sa = (struct sockaddr_iucv *) addr;
730 	struct sock *sk = sock->sk;
731 	struct iucv_sock *iucv;
732 	int err = 0;
733 	struct net_device *dev;
734 	char uid[9];
735 
736 	/* Verify the input sockaddr */
737 	if (addr_len < sizeof(struct sockaddr_iucv) ||
738 	    addr->sa_family != AF_IUCV)
739 		return -EINVAL;
740 
741 	lock_sock(sk);
742 	if (sk->sk_state != IUCV_OPEN) {
743 		err = -EBADFD;
744 		goto done;
745 	}
746 
747 	write_lock_bh(&iucv_sk_list.lock);
748 
749 	iucv = iucv_sk(sk);
750 	if (__iucv_get_sock_by_name(sa->siucv_name)) {
751 		err = -EADDRINUSE;
752 		goto done_unlock;
753 	}
754 	if (iucv->path)
755 		goto done_unlock;
756 
757 	/* Bind the socket */
758 	if (pr_iucv)
759 		if (!memcmp(sa->siucv_user_id, iucv_userid, 8))
760 			goto vm_bind; /* VM IUCV transport */
761 
762 	/* try hiper transport */
763 	memcpy(uid, sa->siucv_user_id, sizeof(uid));
764 	ASCEBC(uid, 8);
765 	rcu_read_lock();
766 	for_each_netdev_rcu(&init_net, dev) {
767 		if (!memcmp(dev->perm_addr, uid, 8)) {
768 			memcpy(iucv->src_user_id, sa->siucv_user_id, 8);
769 			/* Check for unitialized siucv_name */
770 			if (strncmp(sa->siucv_name, "        ", 8) == 0)
771 				__iucv_auto_name(iucv);
772 			else
773 				memcpy(iucv->src_name, sa->siucv_name, 8);
774 			sk->sk_bound_dev_if = dev->ifindex;
775 			iucv->hs_dev = dev;
776 			dev_hold(dev);
777 			sk->sk_state = IUCV_BOUND;
778 			iucv->transport = AF_IUCV_TRANS_HIPER;
779 			if (!iucv->msglimit)
780 				iucv->msglimit = IUCV_HIPER_MSGLIM_DEFAULT;
781 			rcu_read_unlock();
782 			goto done_unlock;
783 		}
784 	}
785 	rcu_read_unlock();
786 vm_bind:
787 	if (pr_iucv) {
788 		/* use local userid for backward compat */
789 		memcpy(iucv->src_name, sa->siucv_name, 8);
790 		memcpy(iucv->src_user_id, iucv_userid, 8);
791 		sk->sk_state = IUCV_BOUND;
792 		iucv->transport = AF_IUCV_TRANS_IUCV;
793 		sk->sk_allocation |= GFP_DMA;
794 		if (!iucv->msglimit)
795 			iucv->msglimit = IUCV_QUEUELEN_DEFAULT;
796 		goto done_unlock;
797 	}
798 	/* found no dev to bind */
799 	err = -ENODEV;
800 done_unlock:
801 	/* Release the socket list lock */
802 	write_unlock_bh(&iucv_sk_list.lock);
803 done:
804 	release_sock(sk);
805 	return err;
806 }
807 
808 /* Automatically bind an unbound socket */
iucv_sock_autobind(struct sock * sk)809 static int iucv_sock_autobind(struct sock *sk)
810 {
811 	struct iucv_sock *iucv = iucv_sk(sk);
812 	int err = 0;
813 
814 	if (unlikely(!pr_iucv))
815 		return -EPROTO;
816 
817 	memcpy(iucv->src_user_id, iucv_userid, 8);
818 	iucv->transport = AF_IUCV_TRANS_IUCV;
819 	sk->sk_allocation |= GFP_DMA;
820 
821 	write_lock_bh(&iucv_sk_list.lock);
822 	__iucv_auto_name(iucv);
823 	write_unlock_bh(&iucv_sk_list.lock);
824 
825 	if (!iucv->msglimit)
826 		iucv->msglimit = IUCV_QUEUELEN_DEFAULT;
827 
828 	return err;
829 }
830 
afiucv_path_connect(struct socket * sock,struct sockaddr * addr)831 static int afiucv_path_connect(struct socket *sock, struct sockaddr *addr)
832 {
833 	struct sockaddr_iucv *sa = (struct sockaddr_iucv *) addr;
834 	struct sock *sk = sock->sk;
835 	struct iucv_sock *iucv = iucv_sk(sk);
836 	unsigned char user_data[16];
837 	int err;
838 
839 	high_nmcpy(user_data, sa->siucv_name);
840 	low_nmcpy(user_data, iucv->src_name);
841 	ASCEBC(user_data, sizeof(user_data));
842 
843 	/* Create path. */
844 	iucv->path = iucv_path_alloc(iucv->msglimit,
845 				     IUCV_IPRMDATA, GFP_KERNEL);
846 	if (!iucv->path) {
847 		err = -ENOMEM;
848 		goto done;
849 	}
850 	err = pr_iucv->path_connect(iucv->path, &af_iucv_handler,
851 				    sa->siucv_user_id, NULL, user_data,
852 				    sk);
853 	if (err) {
854 		iucv_path_free(iucv->path);
855 		iucv->path = NULL;
856 		switch (err) {
857 		case 0x0b:	/* Target communicator is not logged on */
858 			err = -ENETUNREACH;
859 			break;
860 		case 0x0d:	/* Max connections for this guest exceeded */
861 		case 0x0e:	/* Max connections for target guest exceeded */
862 			err = -EAGAIN;
863 			break;
864 		case 0x0f:	/* Missing IUCV authorization */
865 			err = -EACCES;
866 			break;
867 		default:
868 			err = -ECONNREFUSED;
869 			break;
870 		}
871 	}
872 done:
873 	return err;
874 }
875 
876 /* Connect an unconnected socket */
iucv_sock_connect(struct socket * sock,struct sockaddr * addr,int alen,int flags)877 static int iucv_sock_connect(struct socket *sock, struct sockaddr *addr,
878 			     int alen, int flags)
879 {
880 	struct sockaddr_iucv *sa = (struct sockaddr_iucv *) addr;
881 	struct sock *sk = sock->sk;
882 	struct iucv_sock *iucv = iucv_sk(sk);
883 	int err;
884 
885 	if (alen < sizeof(struct sockaddr_iucv) || addr->sa_family != AF_IUCV)
886 		return -EINVAL;
887 
888 	if (sk->sk_state != IUCV_OPEN && sk->sk_state != IUCV_BOUND)
889 		return -EBADFD;
890 
891 	if (sk->sk_state == IUCV_OPEN &&
892 	    iucv->transport == AF_IUCV_TRANS_HIPER)
893 		return -EBADFD; /* explicit bind required */
894 
895 	if (sk->sk_type != SOCK_STREAM && sk->sk_type != SOCK_SEQPACKET)
896 		return -EINVAL;
897 
898 	if (sk->sk_state == IUCV_OPEN) {
899 		err = iucv_sock_autobind(sk);
900 		if (unlikely(err))
901 			return err;
902 	}
903 
904 	lock_sock(sk);
905 
906 	/* Set the destination information */
907 	memcpy(iucv->dst_user_id, sa->siucv_user_id, 8);
908 	memcpy(iucv->dst_name, sa->siucv_name, 8);
909 
910 	if (iucv->transport == AF_IUCV_TRANS_HIPER)
911 		err = iucv_send_ctrl(sock->sk, AF_IUCV_FLAG_SYN);
912 	else
913 		err = afiucv_path_connect(sock, addr);
914 	if (err)
915 		goto done;
916 
917 	if (sk->sk_state != IUCV_CONNECTED)
918 		err = iucv_sock_wait(sk, iucv_sock_in_state(sk, IUCV_CONNECTED,
919 							    IUCV_DISCONN),
920 				     sock_sndtimeo(sk, flags & O_NONBLOCK));
921 
922 	if (sk->sk_state == IUCV_DISCONN || sk->sk_state == IUCV_CLOSED)
923 		err = -ECONNREFUSED;
924 
925 	if (err && iucv->transport == AF_IUCV_TRANS_IUCV)
926 		iucv_sever_path(sk, 0);
927 
928 done:
929 	release_sock(sk);
930 	return err;
931 }
932 
933 /* Move a socket into listening state. */
iucv_sock_listen(struct socket * sock,int backlog)934 static int iucv_sock_listen(struct socket *sock, int backlog)
935 {
936 	struct sock *sk = sock->sk;
937 	int err;
938 
939 	lock_sock(sk);
940 
941 	err = -EINVAL;
942 	if (sk->sk_state != IUCV_BOUND)
943 		goto done;
944 
945 	if (sock->type != SOCK_STREAM && sock->type != SOCK_SEQPACKET)
946 		goto done;
947 
948 	sk->sk_max_ack_backlog = backlog;
949 	sk->sk_ack_backlog = 0;
950 	sk->sk_state = IUCV_LISTEN;
951 	err = 0;
952 
953 done:
954 	release_sock(sk);
955 	return err;
956 }
957 
958 /* Accept a pending connection */
iucv_sock_accept(struct socket * sock,struct socket * newsock,int flags,bool kern)959 static int iucv_sock_accept(struct socket *sock, struct socket *newsock,
960 			    int flags, bool kern)
961 {
962 	DECLARE_WAITQUEUE(wait, current);
963 	struct sock *sk = sock->sk, *nsk;
964 	long timeo;
965 	int err = 0;
966 
967 	lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
968 
969 	if (sk->sk_state != IUCV_LISTEN) {
970 		err = -EBADFD;
971 		goto done;
972 	}
973 
974 	timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
975 
976 	/* Wait for an incoming connection */
977 	add_wait_queue_exclusive(sk_sleep(sk), &wait);
978 	while (!(nsk = iucv_accept_dequeue(sk, newsock))) {
979 		set_current_state(TASK_INTERRUPTIBLE);
980 		if (!timeo) {
981 			err = -EAGAIN;
982 			break;
983 		}
984 
985 		release_sock(sk);
986 		timeo = schedule_timeout(timeo);
987 		lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
988 
989 		if (sk->sk_state != IUCV_LISTEN) {
990 			err = -EBADFD;
991 			break;
992 		}
993 
994 		if (signal_pending(current)) {
995 			err = sock_intr_errno(timeo);
996 			break;
997 		}
998 	}
999 
1000 	set_current_state(TASK_RUNNING);
1001 	remove_wait_queue(sk_sleep(sk), &wait);
1002 
1003 	if (err)
1004 		goto done;
1005 
1006 	newsock->state = SS_CONNECTED;
1007 
1008 done:
1009 	release_sock(sk);
1010 	return err;
1011 }
1012 
iucv_sock_getname(struct socket * sock,struct sockaddr * addr,int peer)1013 static int iucv_sock_getname(struct socket *sock, struct sockaddr *addr,
1014 			     int peer)
1015 {
1016 	struct sockaddr_iucv *siucv = (struct sockaddr_iucv *) addr;
1017 	struct sock *sk = sock->sk;
1018 	struct iucv_sock *iucv = iucv_sk(sk);
1019 
1020 	addr->sa_family = AF_IUCV;
1021 
1022 	if (peer) {
1023 		memcpy(siucv->siucv_user_id, iucv->dst_user_id, 8);
1024 		memcpy(siucv->siucv_name, iucv->dst_name, 8);
1025 	} else {
1026 		memcpy(siucv->siucv_user_id, iucv->src_user_id, 8);
1027 		memcpy(siucv->siucv_name, iucv->src_name, 8);
1028 	}
1029 	memset(&siucv->siucv_port, 0, sizeof(siucv->siucv_port));
1030 	memset(&siucv->siucv_addr, 0, sizeof(siucv->siucv_addr));
1031 	memset(&siucv->siucv_nodeid, 0, sizeof(siucv->siucv_nodeid));
1032 
1033 	return sizeof(struct sockaddr_iucv);
1034 }
1035 
1036 /**
1037  * iucv_send_iprm() - Send socket data in parameter list of an iucv message.
1038  * @path:	IUCV path
1039  * @msg:	Pointer to a struct iucv_message
1040  * @skb:	The socket data to send, skb->len MUST BE <= 7
1041  *
1042  * Send the socket data in the parameter list in the iucv message
1043  * (IUCV_IPRMDATA). The socket data is stored at index 0 to 6 in the parameter
1044  * list and the socket data len at index 7 (last byte).
1045  * See also iucv_msg_length().
1046  *
1047  * Returns the error code from the iucv_message_send() call.
1048  */
iucv_send_iprm(struct iucv_path * path,struct iucv_message * msg,struct sk_buff * skb)1049 static int iucv_send_iprm(struct iucv_path *path, struct iucv_message *msg,
1050 			  struct sk_buff *skb)
1051 {
1052 	u8 prmdata[8];
1053 
1054 	memcpy(prmdata, (void *) skb->data, skb->len);
1055 	prmdata[7] = 0xff - (u8) skb->len;
1056 	return pr_iucv->message_send(path, msg, IUCV_IPRMDATA, 0,
1057 				 (void *) prmdata, 8);
1058 }
1059 
iucv_sock_sendmsg(struct socket * sock,struct msghdr * msg,size_t len)1060 static int iucv_sock_sendmsg(struct socket *sock, struct msghdr *msg,
1061 			     size_t len)
1062 {
1063 	struct sock *sk = sock->sk;
1064 	struct iucv_sock *iucv = iucv_sk(sk);
1065 	size_t headroom = 0;
1066 	size_t linear;
1067 	struct sk_buff *skb;
1068 	struct iucv_message txmsg = {0};
1069 	struct cmsghdr *cmsg;
1070 	int cmsg_done;
1071 	long timeo;
1072 	char user_id[9];
1073 	char appl_id[9];
1074 	int err;
1075 	int noblock = msg->msg_flags & MSG_DONTWAIT;
1076 
1077 	err = sock_error(sk);
1078 	if (err)
1079 		return err;
1080 
1081 	if (msg->msg_flags & MSG_OOB)
1082 		return -EOPNOTSUPP;
1083 
1084 	/* SOCK_SEQPACKET: we do not support segmented records */
1085 	if (sk->sk_type == SOCK_SEQPACKET && !(msg->msg_flags & MSG_EOR))
1086 		return -EOPNOTSUPP;
1087 
1088 	lock_sock(sk);
1089 
1090 	if (sk->sk_shutdown & SEND_SHUTDOWN) {
1091 		err = -EPIPE;
1092 		goto out;
1093 	}
1094 
1095 	/* Return if the socket is not in connected state */
1096 	if (sk->sk_state != IUCV_CONNECTED) {
1097 		err = -ENOTCONN;
1098 		goto out;
1099 	}
1100 
1101 	/* initialize defaults */
1102 	cmsg_done   = 0;	/* check for duplicate headers */
1103 	txmsg.class = 0;
1104 
1105 	/* iterate over control messages */
1106 	for_each_cmsghdr(cmsg, msg) {
1107 		if (!CMSG_OK(msg, cmsg)) {
1108 			err = -EINVAL;
1109 			goto out;
1110 		}
1111 
1112 		if (cmsg->cmsg_level != SOL_IUCV)
1113 			continue;
1114 
1115 		if (cmsg->cmsg_type & cmsg_done) {
1116 			err = -EINVAL;
1117 			goto out;
1118 		}
1119 		cmsg_done |= cmsg->cmsg_type;
1120 
1121 		switch (cmsg->cmsg_type) {
1122 		case SCM_IUCV_TRGCLS:
1123 			if (cmsg->cmsg_len != CMSG_LEN(TRGCLS_SIZE)) {
1124 				err = -EINVAL;
1125 				goto out;
1126 			}
1127 
1128 			/* set iucv message target class */
1129 			memcpy(&txmsg.class,
1130 				(void *) CMSG_DATA(cmsg), TRGCLS_SIZE);
1131 
1132 			break;
1133 
1134 		default:
1135 			err = -EINVAL;
1136 			goto out;
1137 		}
1138 	}
1139 
1140 	/* allocate one skb for each iucv message:
1141 	 * this is fine for SOCK_SEQPACKET (unless we want to support
1142 	 * segmented records using the MSG_EOR flag), but
1143 	 * for SOCK_STREAM we might want to improve it in future */
1144 	if (iucv->transport == AF_IUCV_TRANS_HIPER) {
1145 		headroom = sizeof(struct af_iucv_trans_hdr) +
1146 			   LL_RESERVED_SPACE(iucv->hs_dev);
1147 		linear = len;
1148 	} else {
1149 		if (len < PAGE_SIZE) {
1150 			linear = len;
1151 		} else {
1152 			/* In nonlinear "classic" iucv skb,
1153 			 * reserve space for iucv_array
1154 			 */
1155 			headroom = sizeof(struct iucv_array) *
1156 				   (MAX_SKB_FRAGS + 1);
1157 			linear = PAGE_SIZE - headroom;
1158 		}
1159 	}
1160 	skb = sock_alloc_send_pskb(sk, headroom + linear, len - linear,
1161 				   noblock, &err, 0);
1162 	if (!skb)
1163 		goto out;
1164 	if (headroom)
1165 		skb_reserve(skb, headroom);
1166 	skb_put(skb, linear);
1167 	skb->len = len;
1168 	skb->data_len = len - linear;
1169 	err = skb_copy_datagram_from_iter(skb, 0, &msg->msg_iter, len);
1170 	if (err)
1171 		goto fail;
1172 
1173 	/* wait if outstanding messages for iucv path has reached */
1174 	timeo = sock_sndtimeo(sk, noblock);
1175 	err = iucv_sock_wait(sk, iucv_below_msglim(sk), timeo);
1176 	if (err)
1177 		goto fail;
1178 
1179 	/* return -ECONNRESET if the socket is no longer connected */
1180 	if (sk->sk_state != IUCV_CONNECTED) {
1181 		err = -ECONNRESET;
1182 		goto fail;
1183 	}
1184 
1185 	/* increment and save iucv message tag for msg_completion cbk */
1186 	txmsg.tag = iucv->send_tag++;
1187 	IUCV_SKB_CB(skb)->tag = txmsg.tag;
1188 
1189 	if (iucv->transport == AF_IUCV_TRANS_HIPER) {
1190 		atomic_inc(&iucv->msg_sent);
1191 		err = afiucv_hs_send(&txmsg, sk, skb, 0);
1192 		if (err) {
1193 			atomic_dec(&iucv->msg_sent);
1194 			goto out;
1195 		}
1196 	} else { /* Classic VM IUCV transport */
1197 		skb_queue_tail(&iucv->send_skb_q, skb);
1198 
1199 		if (((iucv->path->flags & IUCV_IPRMDATA) & iucv->flags) &&
1200 		    skb->len <= 7) {
1201 			err = iucv_send_iprm(iucv->path, &txmsg, skb);
1202 
1203 			/* on success: there is no message_complete callback */
1204 			/* for an IPRMDATA msg; remove skb from send queue   */
1205 			if (err == 0) {
1206 				skb_unlink(skb, &iucv->send_skb_q);
1207 				kfree_skb(skb);
1208 			}
1209 
1210 			/* this error should never happen since the	*/
1211 			/* IUCV_IPRMDATA path flag is set... sever path */
1212 			if (err == 0x15) {
1213 				pr_iucv->path_sever(iucv->path, NULL);
1214 				skb_unlink(skb, &iucv->send_skb_q);
1215 				err = -EPIPE;
1216 				goto fail;
1217 			}
1218 		} else if (skb_is_nonlinear(skb)) {
1219 			struct iucv_array *iba = (struct iucv_array *)skb->head;
1220 			int i;
1221 
1222 			/* skip iucv_array lying in the headroom */
1223 			iba[0].address = (u32)(addr_t)skb->data;
1224 			iba[0].length = (u32)skb_headlen(skb);
1225 			for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
1226 				skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
1227 
1228 				iba[i + 1].address =
1229 					(u32)(addr_t)skb_frag_address(frag);
1230 				iba[i + 1].length = (u32)skb_frag_size(frag);
1231 			}
1232 			err = pr_iucv->message_send(iucv->path, &txmsg,
1233 						    IUCV_IPBUFLST, 0,
1234 						    (void *)iba, skb->len);
1235 		} else { /* non-IPRM Linear skb */
1236 			err = pr_iucv->message_send(iucv->path, &txmsg,
1237 					0, 0, (void *)skb->data, skb->len);
1238 		}
1239 		if (err) {
1240 			if (err == 3) {
1241 				user_id[8] = 0;
1242 				memcpy(user_id, iucv->dst_user_id, 8);
1243 				appl_id[8] = 0;
1244 				memcpy(appl_id, iucv->dst_name, 8);
1245 				pr_err(
1246 		"Application %s on z/VM guest %s exceeds message limit\n",
1247 					appl_id, user_id);
1248 				err = -EAGAIN;
1249 			} else {
1250 				err = -EPIPE;
1251 			}
1252 			skb_unlink(skb, &iucv->send_skb_q);
1253 			goto fail;
1254 		}
1255 	}
1256 
1257 	release_sock(sk);
1258 	return len;
1259 
1260 fail:
1261 	kfree_skb(skb);
1262 out:
1263 	release_sock(sk);
1264 	return err;
1265 }
1266 
alloc_iucv_recv_skb(unsigned long len)1267 static struct sk_buff *alloc_iucv_recv_skb(unsigned long len)
1268 {
1269 	size_t headroom, linear;
1270 	struct sk_buff *skb;
1271 	int err;
1272 
1273 	if (len < PAGE_SIZE) {
1274 		headroom = 0;
1275 		linear = len;
1276 	} else {
1277 		headroom = sizeof(struct iucv_array) * (MAX_SKB_FRAGS + 1);
1278 		linear = PAGE_SIZE - headroom;
1279 	}
1280 	skb = alloc_skb_with_frags(headroom + linear, len - linear,
1281 				   0, &err, GFP_ATOMIC | GFP_DMA);
1282 	WARN_ONCE(!skb,
1283 		  "alloc of recv iucv skb len=%lu failed with errcode=%d\n",
1284 		  len, err);
1285 	if (skb) {
1286 		if (headroom)
1287 			skb_reserve(skb, headroom);
1288 		skb_put(skb, linear);
1289 		skb->len = len;
1290 		skb->data_len = len - linear;
1291 	}
1292 	return skb;
1293 }
1294 
1295 /* iucv_process_message() - Receive a single outstanding IUCV message
1296  *
1297  * Locking: must be called with message_q.lock held
1298  */
iucv_process_message(struct sock * sk,struct sk_buff * skb,struct iucv_path * path,struct iucv_message * msg)1299 static void iucv_process_message(struct sock *sk, struct sk_buff *skb,
1300 				 struct iucv_path *path,
1301 				 struct iucv_message *msg)
1302 {
1303 	int rc;
1304 	unsigned int len;
1305 
1306 	len = iucv_msg_length(msg);
1307 
1308 	/* store msg target class in the second 4 bytes of skb ctrl buffer */
1309 	/* Note: the first 4 bytes are reserved for msg tag */
1310 	IUCV_SKB_CB(skb)->class = msg->class;
1311 
1312 	/* check for special IPRM messages (e.g. iucv_sock_shutdown) */
1313 	if ((msg->flags & IUCV_IPRMDATA) && len > 7) {
1314 		if (memcmp(msg->rmmsg, iprm_shutdown, 8) == 0) {
1315 			skb->data = NULL;
1316 			skb->len = 0;
1317 		}
1318 	} else {
1319 		if (skb_is_nonlinear(skb)) {
1320 			struct iucv_array *iba = (struct iucv_array *)skb->head;
1321 			int i;
1322 
1323 			iba[0].address = (u32)(addr_t)skb->data;
1324 			iba[0].length = (u32)skb_headlen(skb);
1325 			for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
1326 				skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
1327 
1328 				iba[i + 1].address =
1329 					(u32)(addr_t)skb_frag_address(frag);
1330 				iba[i + 1].length = (u32)skb_frag_size(frag);
1331 			}
1332 			rc = pr_iucv->message_receive(path, msg,
1333 					      IUCV_IPBUFLST,
1334 					      (void *)iba, len, NULL);
1335 		} else {
1336 			rc = pr_iucv->message_receive(path, msg,
1337 					      msg->flags & IUCV_IPRMDATA,
1338 					      skb->data, len, NULL);
1339 		}
1340 		if (rc) {
1341 			kfree_skb(skb);
1342 			return;
1343 		}
1344 		WARN_ON_ONCE(skb->len != len);
1345 	}
1346 
1347 	IUCV_SKB_CB(skb)->offset = 0;
1348 	if (sk_filter(sk, skb)) {
1349 		atomic_inc(&sk->sk_drops);	/* skb rejected by filter */
1350 		kfree_skb(skb);
1351 		return;
1352 	}
1353 	if (__sock_queue_rcv_skb(sk, skb))	/* handle rcv queue full */
1354 		skb_queue_tail(&iucv_sk(sk)->backlog_skb_q, skb);
1355 }
1356 
1357 /* iucv_process_message_q() - Process outstanding IUCV messages
1358  *
1359  * Locking: must be called with message_q.lock held
1360  */
iucv_process_message_q(struct sock * sk)1361 static void iucv_process_message_q(struct sock *sk)
1362 {
1363 	struct iucv_sock *iucv = iucv_sk(sk);
1364 	struct sk_buff *skb;
1365 	struct sock_msg_q *p, *n;
1366 
1367 	list_for_each_entry_safe(p, n, &iucv->message_q.list, list) {
1368 		skb = alloc_iucv_recv_skb(iucv_msg_length(&p->msg));
1369 		if (!skb)
1370 			break;
1371 		iucv_process_message(sk, skb, p->path, &p->msg);
1372 		list_del(&p->list);
1373 		kfree(p);
1374 		if (!skb_queue_empty(&iucv->backlog_skb_q))
1375 			break;
1376 	}
1377 }
1378 
iucv_sock_recvmsg(struct socket * sock,struct msghdr * msg,size_t len,int flags)1379 static int iucv_sock_recvmsg(struct socket *sock, struct msghdr *msg,
1380 			     size_t len, int flags)
1381 {
1382 	int noblock = flags & MSG_DONTWAIT;
1383 	struct sock *sk = sock->sk;
1384 	struct iucv_sock *iucv = iucv_sk(sk);
1385 	unsigned int copied, rlen;
1386 	struct sk_buff *skb, *rskb, *cskb;
1387 	int err = 0;
1388 	u32 offset;
1389 
1390 	if ((sk->sk_state == IUCV_DISCONN) &&
1391 	    skb_queue_empty(&iucv->backlog_skb_q) &&
1392 	    skb_queue_empty(&sk->sk_receive_queue) &&
1393 	    list_empty(&iucv->message_q.list))
1394 		return 0;
1395 
1396 	if (flags & (MSG_OOB))
1397 		return -EOPNOTSUPP;
1398 
1399 	/* receive/dequeue next skb:
1400 	 * the function understands MSG_PEEK and, thus, does not dequeue skb */
1401 	skb = skb_recv_datagram(sk, flags, noblock, &err);
1402 	if (!skb) {
1403 		if (sk->sk_shutdown & RCV_SHUTDOWN)
1404 			return 0;
1405 		return err;
1406 	}
1407 
1408 	offset = IUCV_SKB_CB(skb)->offset;
1409 	rlen   = skb->len - offset;		/* real length of skb */
1410 	copied = min_t(unsigned int, rlen, len);
1411 	if (!rlen)
1412 		sk->sk_shutdown = sk->sk_shutdown | RCV_SHUTDOWN;
1413 
1414 	cskb = skb;
1415 	if (skb_copy_datagram_msg(cskb, offset, msg, copied)) {
1416 		if (!(flags & MSG_PEEK))
1417 			skb_queue_head(&sk->sk_receive_queue, skb);
1418 		return -EFAULT;
1419 	}
1420 
1421 	/* SOCK_SEQPACKET: set MSG_TRUNC if recv buf size is too small */
1422 	if (sk->sk_type == SOCK_SEQPACKET) {
1423 		if (copied < rlen)
1424 			msg->msg_flags |= MSG_TRUNC;
1425 		/* each iucv message contains a complete record */
1426 		msg->msg_flags |= MSG_EOR;
1427 	}
1428 
1429 	/* create control message to store iucv msg target class:
1430 	 * get the trgcls from the control buffer of the skb due to
1431 	 * fragmentation of original iucv message. */
1432 	err = put_cmsg(msg, SOL_IUCV, SCM_IUCV_TRGCLS,
1433 		       sizeof(IUCV_SKB_CB(skb)->class),
1434 		       (void *)&IUCV_SKB_CB(skb)->class);
1435 	if (err) {
1436 		if (!(flags & MSG_PEEK))
1437 			skb_queue_head(&sk->sk_receive_queue, skb);
1438 		return err;
1439 	}
1440 
1441 	/* Mark read part of skb as used */
1442 	if (!(flags & MSG_PEEK)) {
1443 
1444 		/* SOCK_STREAM: re-queue skb if it contains unreceived data */
1445 		if (sk->sk_type == SOCK_STREAM) {
1446 			if (copied < rlen) {
1447 				IUCV_SKB_CB(skb)->offset = offset + copied;
1448 				skb_queue_head(&sk->sk_receive_queue, skb);
1449 				goto done;
1450 			}
1451 		}
1452 
1453 		kfree_skb(skb);
1454 		if (iucv->transport == AF_IUCV_TRANS_HIPER) {
1455 			atomic_inc(&iucv->msg_recv);
1456 			if (atomic_read(&iucv->msg_recv) > iucv->msglimit) {
1457 				WARN_ON(1);
1458 				iucv_sock_close(sk);
1459 				return -EFAULT;
1460 			}
1461 		}
1462 
1463 		/* Queue backlog skbs */
1464 		spin_lock_bh(&iucv->message_q.lock);
1465 		rskb = skb_dequeue(&iucv->backlog_skb_q);
1466 		while (rskb) {
1467 			IUCV_SKB_CB(rskb)->offset = 0;
1468 			if (__sock_queue_rcv_skb(sk, rskb)) {
1469 				/* handle rcv queue full */
1470 				skb_queue_head(&iucv->backlog_skb_q,
1471 						rskb);
1472 				break;
1473 			}
1474 			rskb = skb_dequeue(&iucv->backlog_skb_q);
1475 		}
1476 		if (skb_queue_empty(&iucv->backlog_skb_q)) {
1477 			if (!list_empty(&iucv->message_q.list))
1478 				iucv_process_message_q(sk);
1479 			if (atomic_read(&iucv->msg_recv) >=
1480 							iucv->msglimit / 2) {
1481 				err = iucv_send_ctrl(sk, AF_IUCV_FLAG_WIN);
1482 				if (err) {
1483 					sk->sk_state = IUCV_DISCONN;
1484 					sk->sk_state_change(sk);
1485 				}
1486 			}
1487 		}
1488 		spin_unlock_bh(&iucv->message_q.lock);
1489 	}
1490 
1491 done:
1492 	/* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
1493 	if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
1494 		copied = rlen;
1495 
1496 	return copied;
1497 }
1498 
iucv_accept_poll(struct sock * parent)1499 static inline __poll_t iucv_accept_poll(struct sock *parent)
1500 {
1501 	struct iucv_sock *isk, *n;
1502 	struct sock *sk;
1503 
1504 	list_for_each_entry_safe(isk, n, &iucv_sk(parent)->accept_q, accept_q) {
1505 		sk = (struct sock *) isk;
1506 
1507 		if (sk->sk_state == IUCV_CONNECTED)
1508 			return EPOLLIN | EPOLLRDNORM;
1509 	}
1510 
1511 	return 0;
1512 }
1513 
iucv_sock_poll(struct file * file,struct socket * sock,poll_table * wait)1514 __poll_t iucv_sock_poll(struct file *file, struct socket *sock,
1515 			    poll_table *wait)
1516 {
1517 	struct sock *sk = sock->sk;
1518 	__poll_t mask = 0;
1519 
1520 	sock_poll_wait(file, sock, wait);
1521 
1522 	if (sk->sk_state == IUCV_LISTEN)
1523 		return iucv_accept_poll(sk);
1524 
1525 	if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
1526 		mask |= EPOLLERR |
1527 			(sock_flag(sk, SOCK_SELECT_ERR_QUEUE) ? EPOLLPRI : 0);
1528 
1529 	if (sk->sk_shutdown & RCV_SHUTDOWN)
1530 		mask |= EPOLLRDHUP;
1531 
1532 	if (sk->sk_shutdown == SHUTDOWN_MASK)
1533 		mask |= EPOLLHUP;
1534 
1535 	if (!skb_queue_empty(&sk->sk_receive_queue) ||
1536 	    (sk->sk_shutdown & RCV_SHUTDOWN))
1537 		mask |= EPOLLIN | EPOLLRDNORM;
1538 
1539 	if (sk->sk_state == IUCV_CLOSED)
1540 		mask |= EPOLLHUP;
1541 
1542 	if (sk->sk_state == IUCV_DISCONN)
1543 		mask |= EPOLLIN;
1544 
1545 	if (sock_writeable(sk) && iucv_below_msglim(sk))
1546 		mask |= EPOLLOUT | EPOLLWRNORM | EPOLLWRBAND;
1547 	else
1548 		sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
1549 
1550 	return mask;
1551 }
1552 
iucv_sock_shutdown(struct socket * sock,int how)1553 static int iucv_sock_shutdown(struct socket *sock, int how)
1554 {
1555 	struct sock *sk = sock->sk;
1556 	struct iucv_sock *iucv = iucv_sk(sk);
1557 	struct iucv_message txmsg;
1558 	int err = 0;
1559 
1560 	how++;
1561 
1562 	if ((how & ~SHUTDOWN_MASK) || !how)
1563 		return -EINVAL;
1564 
1565 	lock_sock(sk);
1566 	switch (sk->sk_state) {
1567 	case IUCV_LISTEN:
1568 	case IUCV_DISCONN:
1569 	case IUCV_CLOSING:
1570 	case IUCV_CLOSED:
1571 		err = -ENOTCONN;
1572 		goto fail;
1573 	default:
1574 		break;
1575 	}
1576 
1577 	if (how == SEND_SHUTDOWN || how == SHUTDOWN_MASK) {
1578 		if (iucv->transport == AF_IUCV_TRANS_IUCV) {
1579 			txmsg.class = 0;
1580 			txmsg.tag = 0;
1581 			err = pr_iucv->message_send(iucv->path, &txmsg,
1582 				IUCV_IPRMDATA, 0, (void *) iprm_shutdown, 8);
1583 			if (err) {
1584 				switch (err) {
1585 				case 1:
1586 					err = -ENOTCONN;
1587 					break;
1588 				case 2:
1589 					err = -ECONNRESET;
1590 					break;
1591 				default:
1592 					err = -ENOTCONN;
1593 					break;
1594 				}
1595 			}
1596 		} else
1597 			iucv_send_ctrl(sk, AF_IUCV_FLAG_SHT);
1598 	}
1599 
1600 	sk->sk_shutdown |= how;
1601 	if (how == RCV_SHUTDOWN || how == SHUTDOWN_MASK) {
1602 		if ((iucv->transport == AF_IUCV_TRANS_IUCV) &&
1603 		    iucv->path) {
1604 			err = pr_iucv->path_quiesce(iucv->path, NULL);
1605 			if (err)
1606 				err = -ENOTCONN;
1607 /*			skb_queue_purge(&sk->sk_receive_queue); */
1608 		}
1609 		skb_queue_purge(&sk->sk_receive_queue);
1610 	}
1611 
1612 	/* Wake up anyone sleeping in poll */
1613 	sk->sk_state_change(sk);
1614 
1615 fail:
1616 	release_sock(sk);
1617 	return err;
1618 }
1619 
iucv_sock_release(struct socket * sock)1620 static int iucv_sock_release(struct socket *sock)
1621 {
1622 	struct sock *sk = sock->sk;
1623 	int err = 0;
1624 
1625 	if (!sk)
1626 		return 0;
1627 
1628 	iucv_sock_close(sk);
1629 
1630 	sock_orphan(sk);
1631 	iucv_sock_kill(sk);
1632 	return err;
1633 }
1634 
1635 /* getsockopt and setsockopt */
iucv_sock_setsockopt(struct socket * sock,int level,int optname,char __user * optval,unsigned int optlen)1636 static int iucv_sock_setsockopt(struct socket *sock, int level, int optname,
1637 				char __user *optval, unsigned int optlen)
1638 {
1639 	struct sock *sk = sock->sk;
1640 	struct iucv_sock *iucv = iucv_sk(sk);
1641 	int val;
1642 	int rc;
1643 
1644 	if (level != SOL_IUCV)
1645 		return -ENOPROTOOPT;
1646 
1647 	if (optlen < sizeof(int))
1648 		return -EINVAL;
1649 
1650 	if (get_user(val, (int __user *) optval))
1651 		return -EFAULT;
1652 
1653 	rc = 0;
1654 
1655 	lock_sock(sk);
1656 	switch (optname) {
1657 	case SO_IPRMDATA_MSG:
1658 		if (val)
1659 			iucv->flags |= IUCV_IPRMDATA;
1660 		else
1661 			iucv->flags &= ~IUCV_IPRMDATA;
1662 		break;
1663 	case SO_MSGLIMIT:
1664 		switch (sk->sk_state) {
1665 		case IUCV_OPEN:
1666 		case IUCV_BOUND:
1667 			if (val < 1 || val > (u16)(~0))
1668 				rc = -EINVAL;
1669 			else
1670 				iucv->msglimit = val;
1671 			break;
1672 		default:
1673 			rc = -EINVAL;
1674 			break;
1675 		}
1676 		break;
1677 	default:
1678 		rc = -ENOPROTOOPT;
1679 		break;
1680 	}
1681 	release_sock(sk);
1682 
1683 	return rc;
1684 }
1685 
iucv_sock_getsockopt(struct socket * sock,int level,int optname,char __user * optval,int __user * optlen)1686 static int iucv_sock_getsockopt(struct socket *sock, int level, int optname,
1687 				char __user *optval, int __user *optlen)
1688 {
1689 	struct sock *sk = sock->sk;
1690 	struct iucv_sock *iucv = iucv_sk(sk);
1691 	unsigned int val;
1692 	int len;
1693 
1694 	if (level != SOL_IUCV)
1695 		return -ENOPROTOOPT;
1696 
1697 	if (get_user(len, optlen))
1698 		return -EFAULT;
1699 
1700 	if (len < 0)
1701 		return -EINVAL;
1702 
1703 	len = min_t(unsigned int, len, sizeof(int));
1704 
1705 	switch (optname) {
1706 	case SO_IPRMDATA_MSG:
1707 		val = (iucv->flags & IUCV_IPRMDATA) ? 1 : 0;
1708 		break;
1709 	case SO_MSGLIMIT:
1710 		lock_sock(sk);
1711 		val = (iucv->path != NULL) ? iucv->path->msglim	/* connected */
1712 					   : iucv->msglimit;	/* default */
1713 		release_sock(sk);
1714 		break;
1715 	case SO_MSGSIZE:
1716 		if (sk->sk_state == IUCV_OPEN)
1717 			return -EBADFD;
1718 		val = (iucv->hs_dev) ? iucv->hs_dev->mtu -
1719 				sizeof(struct af_iucv_trans_hdr) - ETH_HLEN :
1720 				0x7fffffff;
1721 		break;
1722 	default:
1723 		return -ENOPROTOOPT;
1724 	}
1725 
1726 	if (put_user(len, optlen))
1727 		return -EFAULT;
1728 	if (copy_to_user(optval, &val, len))
1729 		return -EFAULT;
1730 
1731 	return 0;
1732 }
1733 
1734 
1735 /* Callback wrappers - called from iucv base support */
iucv_callback_connreq(struct iucv_path * path,u8 ipvmid[8],u8 ipuser[16])1736 static int iucv_callback_connreq(struct iucv_path *path,
1737 				 u8 ipvmid[8], u8 ipuser[16])
1738 {
1739 	unsigned char user_data[16];
1740 	unsigned char nuser_data[16];
1741 	unsigned char src_name[8];
1742 	struct sock *sk, *nsk;
1743 	struct iucv_sock *iucv, *niucv;
1744 	int err;
1745 
1746 	memcpy(src_name, ipuser, 8);
1747 	EBCASC(src_name, 8);
1748 	/* Find out if this path belongs to af_iucv. */
1749 	read_lock(&iucv_sk_list.lock);
1750 	iucv = NULL;
1751 	sk = NULL;
1752 	sk_for_each(sk, &iucv_sk_list.head)
1753 		if (sk->sk_state == IUCV_LISTEN &&
1754 		    !memcmp(&iucv_sk(sk)->src_name, src_name, 8)) {
1755 			/*
1756 			 * Found a listening socket with
1757 			 * src_name == ipuser[0-7].
1758 			 */
1759 			iucv = iucv_sk(sk);
1760 			break;
1761 		}
1762 	read_unlock(&iucv_sk_list.lock);
1763 	if (!iucv)
1764 		/* No socket found, not one of our paths. */
1765 		return -EINVAL;
1766 
1767 	bh_lock_sock(sk);
1768 
1769 	/* Check if parent socket is listening */
1770 	low_nmcpy(user_data, iucv->src_name);
1771 	high_nmcpy(user_data, iucv->dst_name);
1772 	ASCEBC(user_data, sizeof(user_data));
1773 	if (sk->sk_state != IUCV_LISTEN) {
1774 		err = pr_iucv->path_sever(path, user_data);
1775 		iucv_path_free(path);
1776 		goto fail;
1777 	}
1778 
1779 	/* Check for backlog size */
1780 	if (sk_acceptq_is_full(sk)) {
1781 		err = pr_iucv->path_sever(path, user_data);
1782 		iucv_path_free(path);
1783 		goto fail;
1784 	}
1785 
1786 	/* Create the new socket */
1787 	nsk = iucv_sock_alloc(NULL, sk->sk_type, GFP_ATOMIC, 0);
1788 	if (!nsk) {
1789 		err = pr_iucv->path_sever(path, user_data);
1790 		iucv_path_free(path);
1791 		goto fail;
1792 	}
1793 
1794 	niucv = iucv_sk(nsk);
1795 	iucv_sock_init(nsk, sk);
1796 	niucv->transport = AF_IUCV_TRANS_IUCV;
1797 	nsk->sk_allocation |= GFP_DMA;
1798 
1799 	/* Set the new iucv_sock */
1800 	memcpy(niucv->dst_name, ipuser + 8, 8);
1801 	EBCASC(niucv->dst_name, 8);
1802 	memcpy(niucv->dst_user_id, ipvmid, 8);
1803 	memcpy(niucv->src_name, iucv->src_name, 8);
1804 	memcpy(niucv->src_user_id, iucv->src_user_id, 8);
1805 	niucv->path = path;
1806 
1807 	/* Call iucv_accept */
1808 	high_nmcpy(nuser_data, ipuser + 8);
1809 	memcpy(nuser_data + 8, niucv->src_name, 8);
1810 	ASCEBC(nuser_data + 8, 8);
1811 
1812 	/* set message limit for path based on msglimit of accepting socket */
1813 	niucv->msglimit = iucv->msglimit;
1814 	path->msglim = iucv->msglimit;
1815 	err = pr_iucv->path_accept(path, &af_iucv_handler, nuser_data, nsk);
1816 	if (err) {
1817 		iucv_sever_path(nsk, 1);
1818 		iucv_sock_kill(nsk);
1819 		goto fail;
1820 	}
1821 
1822 	iucv_accept_enqueue(sk, nsk);
1823 
1824 	/* Wake up accept */
1825 	nsk->sk_state = IUCV_CONNECTED;
1826 	sk->sk_data_ready(sk);
1827 	err = 0;
1828 fail:
1829 	bh_unlock_sock(sk);
1830 	return 0;
1831 }
1832 
iucv_callback_connack(struct iucv_path * path,u8 ipuser[16])1833 static void iucv_callback_connack(struct iucv_path *path, u8 ipuser[16])
1834 {
1835 	struct sock *sk = path->private;
1836 
1837 	sk->sk_state = IUCV_CONNECTED;
1838 	sk->sk_state_change(sk);
1839 }
1840 
iucv_callback_rx(struct iucv_path * path,struct iucv_message * msg)1841 static void iucv_callback_rx(struct iucv_path *path, struct iucv_message *msg)
1842 {
1843 	struct sock *sk = path->private;
1844 	struct iucv_sock *iucv = iucv_sk(sk);
1845 	struct sk_buff *skb;
1846 	struct sock_msg_q *save_msg;
1847 	int len;
1848 
1849 	if (sk->sk_shutdown & RCV_SHUTDOWN) {
1850 		pr_iucv->message_reject(path, msg);
1851 		return;
1852 	}
1853 
1854 	spin_lock(&iucv->message_q.lock);
1855 
1856 	if (!list_empty(&iucv->message_q.list) ||
1857 	    !skb_queue_empty(&iucv->backlog_skb_q))
1858 		goto save_message;
1859 
1860 	len = atomic_read(&sk->sk_rmem_alloc);
1861 	len += SKB_TRUESIZE(iucv_msg_length(msg));
1862 	if (len > sk->sk_rcvbuf)
1863 		goto save_message;
1864 
1865 	skb = alloc_iucv_recv_skb(iucv_msg_length(msg));
1866 	if (!skb)
1867 		goto save_message;
1868 
1869 	iucv_process_message(sk, skb, path, msg);
1870 	goto out_unlock;
1871 
1872 save_message:
1873 	save_msg = kzalloc(sizeof(struct sock_msg_q), GFP_ATOMIC | GFP_DMA);
1874 	if (!save_msg)
1875 		goto out_unlock;
1876 	save_msg->path = path;
1877 	save_msg->msg = *msg;
1878 
1879 	list_add_tail(&save_msg->list, &iucv->message_q.list);
1880 
1881 out_unlock:
1882 	spin_unlock(&iucv->message_q.lock);
1883 }
1884 
iucv_callback_txdone(struct iucv_path * path,struct iucv_message * msg)1885 static void iucv_callback_txdone(struct iucv_path *path,
1886 				 struct iucv_message *msg)
1887 {
1888 	struct sock *sk = path->private;
1889 	struct sk_buff *this = NULL;
1890 	struct sk_buff_head *list = &iucv_sk(sk)->send_skb_q;
1891 	struct sk_buff *list_skb;
1892 	unsigned long flags;
1893 
1894 	bh_lock_sock(sk);
1895 
1896 	spin_lock_irqsave(&list->lock, flags);
1897 	skb_queue_walk(list, list_skb) {
1898 		if (msg->tag == IUCV_SKB_CB(list_skb)->tag) {
1899 			this = list_skb;
1900 			break;
1901 		}
1902 	}
1903 	if (this)
1904 		__skb_unlink(this, list);
1905 	spin_unlock_irqrestore(&list->lock, flags);
1906 
1907 	if (this) {
1908 		kfree_skb(this);
1909 		/* wake up any process waiting for sending */
1910 		iucv_sock_wake_msglim(sk);
1911 	}
1912 
1913 	if (sk->sk_state == IUCV_CLOSING) {
1914 		if (skb_queue_empty(&iucv_sk(sk)->send_skb_q)) {
1915 			sk->sk_state = IUCV_CLOSED;
1916 			sk->sk_state_change(sk);
1917 		}
1918 	}
1919 	bh_unlock_sock(sk);
1920 
1921 }
1922 
iucv_callback_connrej(struct iucv_path * path,u8 ipuser[16])1923 static void iucv_callback_connrej(struct iucv_path *path, u8 ipuser[16])
1924 {
1925 	struct sock *sk = path->private;
1926 
1927 	if (sk->sk_state == IUCV_CLOSED)
1928 		return;
1929 
1930 	bh_lock_sock(sk);
1931 	iucv_sever_path(sk, 1);
1932 	sk->sk_state = IUCV_DISCONN;
1933 
1934 	sk->sk_state_change(sk);
1935 	bh_unlock_sock(sk);
1936 }
1937 
1938 /* called if the other communication side shuts down its RECV direction;
1939  * in turn, the callback sets SEND_SHUTDOWN to disable sending of data.
1940  */
iucv_callback_shutdown(struct iucv_path * path,u8 ipuser[16])1941 static void iucv_callback_shutdown(struct iucv_path *path, u8 ipuser[16])
1942 {
1943 	struct sock *sk = path->private;
1944 
1945 	bh_lock_sock(sk);
1946 	if (sk->sk_state != IUCV_CLOSED) {
1947 		sk->sk_shutdown |= SEND_SHUTDOWN;
1948 		sk->sk_state_change(sk);
1949 	}
1950 	bh_unlock_sock(sk);
1951 }
1952 
1953 /***************** HiperSockets transport callbacks ********************/
afiucv_swap_src_dest(struct sk_buff * skb)1954 static void afiucv_swap_src_dest(struct sk_buff *skb)
1955 {
1956 	struct af_iucv_trans_hdr *trans_hdr = iucv_trans_hdr(skb);
1957 	char tmpID[8];
1958 	char tmpName[8];
1959 
1960 	ASCEBC(trans_hdr->destUserID, sizeof(trans_hdr->destUserID));
1961 	ASCEBC(trans_hdr->destAppName, sizeof(trans_hdr->destAppName));
1962 	ASCEBC(trans_hdr->srcUserID, sizeof(trans_hdr->srcUserID));
1963 	ASCEBC(trans_hdr->srcAppName, sizeof(trans_hdr->srcAppName));
1964 	memcpy(tmpID, trans_hdr->srcUserID, 8);
1965 	memcpy(tmpName, trans_hdr->srcAppName, 8);
1966 	memcpy(trans_hdr->srcUserID, trans_hdr->destUserID, 8);
1967 	memcpy(trans_hdr->srcAppName, trans_hdr->destAppName, 8);
1968 	memcpy(trans_hdr->destUserID, tmpID, 8);
1969 	memcpy(trans_hdr->destAppName, tmpName, 8);
1970 	skb_push(skb, ETH_HLEN);
1971 	memset(skb->data, 0, ETH_HLEN);
1972 }
1973 
1974 /**
1975  * afiucv_hs_callback_syn - react on received SYN
1976  **/
afiucv_hs_callback_syn(struct sock * sk,struct sk_buff * skb)1977 static int afiucv_hs_callback_syn(struct sock *sk, struct sk_buff *skb)
1978 {
1979 	struct af_iucv_trans_hdr *trans_hdr = iucv_trans_hdr(skb);
1980 	struct sock *nsk;
1981 	struct iucv_sock *iucv, *niucv;
1982 	int err;
1983 
1984 	iucv = iucv_sk(sk);
1985 	if (!iucv) {
1986 		/* no sock - connection refused */
1987 		afiucv_swap_src_dest(skb);
1988 		trans_hdr->flags = AF_IUCV_FLAG_SYN | AF_IUCV_FLAG_FIN;
1989 		err = dev_queue_xmit(skb);
1990 		goto out;
1991 	}
1992 
1993 	nsk = iucv_sock_alloc(NULL, sk->sk_type, GFP_ATOMIC, 0);
1994 	bh_lock_sock(sk);
1995 	if ((sk->sk_state != IUCV_LISTEN) ||
1996 	    sk_acceptq_is_full(sk) ||
1997 	    !nsk) {
1998 		/* error on server socket - connection refused */
1999 		afiucv_swap_src_dest(skb);
2000 		trans_hdr->flags = AF_IUCV_FLAG_SYN | AF_IUCV_FLAG_FIN;
2001 		err = dev_queue_xmit(skb);
2002 		iucv_sock_kill(nsk);
2003 		bh_unlock_sock(sk);
2004 		goto out;
2005 	}
2006 
2007 	niucv = iucv_sk(nsk);
2008 	iucv_sock_init(nsk, sk);
2009 	niucv->transport = AF_IUCV_TRANS_HIPER;
2010 	niucv->msglimit = iucv->msglimit;
2011 	if (!trans_hdr->window)
2012 		niucv->msglimit_peer = IUCV_HIPER_MSGLIM_DEFAULT;
2013 	else
2014 		niucv->msglimit_peer = trans_hdr->window;
2015 	memcpy(niucv->dst_name, trans_hdr->srcAppName, 8);
2016 	memcpy(niucv->dst_user_id, trans_hdr->srcUserID, 8);
2017 	memcpy(niucv->src_name, iucv->src_name, 8);
2018 	memcpy(niucv->src_user_id, iucv->src_user_id, 8);
2019 	nsk->sk_bound_dev_if = sk->sk_bound_dev_if;
2020 	niucv->hs_dev = iucv->hs_dev;
2021 	dev_hold(niucv->hs_dev);
2022 	afiucv_swap_src_dest(skb);
2023 	trans_hdr->flags = AF_IUCV_FLAG_SYN | AF_IUCV_FLAG_ACK;
2024 	trans_hdr->window = niucv->msglimit;
2025 	/* if receiver acks the xmit connection is established */
2026 	err = dev_queue_xmit(skb);
2027 	if (!err) {
2028 		iucv_accept_enqueue(sk, nsk);
2029 		nsk->sk_state = IUCV_CONNECTED;
2030 		sk->sk_data_ready(sk);
2031 	} else
2032 		iucv_sock_kill(nsk);
2033 	bh_unlock_sock(sk);
2034 
2035 out:
2036 	return NET_RX_SUCCESS;
2037 }
2038 
2039 /**
2040  * afiucv_hs_callback_synack() - react on received SYN-ACK
2041  **/
afiucv_hs_callback_synack(struct sock * sk,struct sk_buff * skb)2042 static int afiucv_hs_callback_synack(struct sock *sk, struct sk_buff *skb)
2043 {
2044 	struct iucv_sock *iucv = iucv_sk(sk);
2045 
2046 	if (!iucv)
2047 		goto out;
2048 	if (sk->sk_state != IUCV_BOUND)
2049 		goto out;
2050 	bh_lock_sock(sk);
2051 	iucv->msglimit_peer = iucv_trans_hdr(skb)->window;
2052 	sk->sk_state = IUCV_CONNECTED;
2053 	sk->sk_state_change(sk);
2054 	bh_unlock_sock(sk);
2055 out:
2056 	kfree_skb(skb);
2057 	return NET_RX_SUCCESS;
2058 }
2059 
2060 /**
2061  * afiucv_hs_callback_synfin() - react on received SYN_FIN
2062  **/
afiucv_hs_callback_synfin(struct sock * sk,struct sk_buff * skb)2063 static int afiucv_hs_callback_synfin(struct sock *sk, struct sk_buff *skb)
2064 {
2065 	struct iucv_sock *iucv = iucv_sk(sk);
2066 
2067 	if (!iucv)
2068 		goto out;
2069 	if (sk->sk_state != IUCV_BOUND)
2070 		goto out;
2071 	bh_lock_sock(sk);
2072 	sk->sk_state = IUCV_DISCONN;
2073 	sk->sk_state_change(sk);
2074 	bh_unlock_sock(sk);
2075 out:
2076 	kfree_skb(skb);
2077 	return NET_RX_SUCCESS;
2078 }
2079 
2080 /**
2081  * afiucv_hs_callback_fin() - react on received FIN
2082  **/
afiucv_hs_callback_fin(struct sock * sk,struct sk_buff * skb)2083 static int afiucv_hs_callback_fin(struct sock *sk, struct sk_buff *skb)
2084 {
2085 	struct iucv_sock *iucv = iucv_sk(sk);
2086 
2087 	/* other end of connection closed */
2088 	if (!iucv)
2089 		goto out;
2090 	bh_lock_sock(sk);
2091 	if (sk->sk_state == IUCV_CONNECTED) {
2092 		sk->sk_state = IUCV_DISCONN;
2093 		sk->sk_state_change(sk);
2094 	}
2095 	bh_unlock_sock(sk);
2096 out:
2097 	kfree_skb(skb);
2098 	return NET_RX_SUCCESS;
2099 }
2100 
2101 /**
2102  * afiucv_hs_callback_win() - react on received WIN
2103  **/
afiucv_hs_callback_win(struct sock * sk,struct sk_buff * skb)2104 static int afiucv_hs_callback_win(struct sock *sk, struct sk_buff *skb)
2105 {
2106 	struct iucv_sock *iucv = iucv_sk(sk);
2107 
2108 	if (!iucv)
2109 		return NET_RX_SUCCESS;
2110 
2111 	if (sk->sk_state != IUCV_CONNECTED)
2112 		return NET_RX_SUCCESS;
2113 
2114 	atomic_sub(iucv_trans_hdr(skb)->window, &iucv->msg_sent);
2115 	iucv_sock_wake_msglim(sk);
2116 	return NET_RX_SUCCESS;
2117 }
2118 
2119 /**
2120  * afiucv_hs_callback_rx() - react on received data
2121  **/
afiucv_hs_callback_rx(struct sock * sk,struct sk_buff * skb)2122 static int afiucv_hs_callback_rx(struct sock *sk, struct sk_buff *skb)
2123 {
2124 	struct iucv_sock *iucv = iucv_sk(sk);
2125 
2126 	if (!iucv) {
2127 		kfree_skb(skb);
2128 		return NET_RX_SUCCESS;
2129 	}
2130 
2131 	if (sk->sk_state != IUCV_CONNECTED) {
2132 		kfree_skb(skb);
2133 		return NET_RX_SUCCESS;
2134 	}
2135 
2136 	if (sk->sk_shutdown & RCV_SHUTDOWN) {
2137 		kfree_skb(skb);
2138 		return NET_RX_SUCCESS;
2139 	}
2140 
2141 	/* write stuff from iucv_msg to skb cb */
2142 	skb_pull(skb, sizeof(struct af_iucv_trans_hdr));
2143 	skb_reset_transport_header(skb);
2144 	skb_reset_network_header(skb);
2145 	IUCV_SKB_CB(skb)->offset = 0;
2146 	if (sk_filter(sk, skb)) {
2147 		atomic_inc(&sk->sk_drops);	/* skb rejected by filter */
2148 		kfree_skb(skb);
2149 		return NET_RX_SUCCESS;
2150 	}
2151 
2152 	spin_lock(&iucv->message_q.lock);
2153 	if (skb_queue_empty(&iucv->backlog_skb_q)) {
2154 		if (__sock_queue_rcv_skb(sk, skb))
2155 			/* handle rcv queue full */
2156 			skb_queue_tail(&iucv->backlog_skb_q, skb);
2157 	} else
2158 		skb_queue_tail(&iucv_sk(sk)->backlog_skb_q, skb);
2159 	spin_unlock(&iucv->message_q.lock);
2160 	return NET_RX_SUCCESS;
2161 }
2162 
2163 /**
2164  * afiucv_hs_rcv() - base function for arriving data through HiperSockets
2165  *                   transport
2166  *                   called from netif RX softirq
2167  **/
afiucv_hs_rcv(struct sk_buff * skb,struct net_device * dev,struct packet_type * pt,struct net_device * orig_dev)2168 static int afiucv_hs_rcv(struct sk_buff *skb, struct net_device *dev,
2169 	struct packet_type *pt, struct net_device *orig_dev)
2170 {
2171 	struct sock *sk;
2172 	struct iucv_sock *iucv;
2173 	struct af_iucv_trans_hdr *trans_hdr;
2174 	int err = NET_RX_SUCCESS;
2175 	char nullstring[8];
2176 
2177 	if (!pskb_may_pull(skb, sizeof(*trans_hdr))) {
2178 		WARN_ONCE(1, "AF_IUCV failed to receive skb, len=%u", skb->len);
2179 		kfree_skb(skb);
2180 		return NET_RX_SUCCESS;
2181 	}
2182 
2183 	trans_hdr = iucv_trans_hdr(skb);
2184 	EBCASC(trans_hdr->destAppName, sizeof(trans_hdr->destAppName));
2185 	EBCASC(trans_hdr->destUserID, sizeof(trans_hdr->destUserID));
2186 	EBCASC(trans_hdr->srcAppName, sizeof(trans_hdr->srcAppName));
2187 	EBCASC(trans_hdr->srcUserID, sizeof(trans_hdr->srcUserID));
2188 	memset(nullstring, 0, sizeof(nullstring));
2189 	iucv = NULL;
2190 	sk = NULL;
2191 	read_lock(&iucv_sk_list.lock);
2192 	sk_for_each(sk, &iucv_sk_list.head) {
2193 		if (trans_hdr->flags == AF_IUCV_FLAG_SYN) {
2194 			if ((!memcmp(&iucv_sk(sk)->src_name,
2195 				     trans_hdr->destAppName, 8)) &&
2196 			    (!memcmp(&iucv_sk(sk)->src_user_id,
2197 				     trans_hdr->destUserID, 8)) &&
2198 			    (!memcmp(&iucv_sk(sk)->dst_name, nullstring, 8)) &&
2199 			    (!memcmp(&iucv_sk(sk)->dst_user_id,
2200 				     nullstring, 8))) {
2201 				iucv = iucv_sk(sk);
2202 				break;
2203 			}
2204 		} else {
2205 			if ((!memcmp(&iucv_sk(sk)->src_name,
2206 				     trans_hdr->destAppName, 8)) &&
2207 			    (!memcmp(&iucv_sk(sk)->src_user_id,
2208 				     trans_hdr->destUserID, 8)) &&
2209 			    (!memcmp(&iucv_sk(sk)->dst_name,
2210 				     trans_hdr->srcAppName, 8)) &&
2211 			    (!memcmp(&iucv_sk(sk)->dst_user_id,
2212 				     trans_hdr->srcUserID, 8))) {
2213 				iucv = iucv_sk(sk);
2214 				break;
2215 			}
2216 		}
2217 	}
2218 	read_unlock(&iucv_sk_list.lock);
2219 	if (!iucv)
2220 		sk = NULL;
2221 
2222 	/* no sock
2223 	how should we send with no sock
2224 	1) send without sock no send rc checking?
2225 	2) introduce default sock to handle this cases
2226 
2227 	 SYN -> send SYN|ACK in good case, send SYN|FIN in bad case
2228 	 data -> send FIN
2229 	 SYN|ACK, SYN|FIN, FIN -> no action? */
2230 
2231 	switch (trans_hdr->flags) {
2232 	case AF_IUCV_FLAG_SYN:
2233 		/* connect request */
2234 		err = afiucv_hs_callback_syn(sk, skb);
2235 		break;
2236 	case (AF_IUCV_FLAG_SYN | AF_IUCV_FLAG_ACK):
2237 		/* connect request confirmed */
2238 		err = afiucv_hs_callback_synack(sk, skb);
2239 		break;
2240 	case (AF_IUCV_FLAG_SYN | AF_IUCV_FLAG_FIN):
2241 		/* connect request refused */
2242 		err = afiucv_hs_callback_synfin(sk, skb);
2243 		break;
2244 	case (AF_IUCV_FLAG_FIN):
2245 		/* close request */
2246 		err = afiucv_hs_callback_fin(sk, skb);
2247 		break;
2248 	case (AF_IUCV_FLAG_WIN):
2249 		err = afiucv_hs_callback_win(sk, skb);
2250 		if (skb->len == sizeof(struct af_iucv_trans_hdr)) {
2251 			kfree_skb(skb);
2252 			break;
2253 		}
2254 		/* fall through - and receive non-zero length data */
2255 	case (AF_IUCV_FLAG_SHT):
2256 		/* shutdown request */
2257 		/* fall through - and receive zero length data */
2258 	case 0:
2259 		/* plain data frame */
2260 		IUCV_SKB_CB(skb)->class = trans_hdr->iucv_hdr.class;
2261 		err = afiucv_hs_callback_rx(sk, skb);
2262 		break;
2263 	default:
2264 		kfree_skb(skb);
2265 	}
2266 
2267 	return err;
2268 }
2269 
2270 /**
2271  * afiucv_hs_callback_txnotify() - handle send notifcations from HiperSockets
2272  *                                 transport
2273  **/
afiucv_hs_callback_txnotify(struct sk_buff * skb,enum iucv_tx_notify n)2274 static void afiucv_hs_callback_txnotify(struct sk_buff *skb,
2275 					enum iucv_tx_notify n)
2276 {
2277 	struct sock *isk = skb->sk;
2278 	struct sock *sk = NULL;
2279 	struct iucv_sock *iucv = NULL;
2280 	struct sk_buff_head *list;
2281 	struct sk_buff *list_skb;
2282 	struct sk_buff *nskb;
2283 	unsigned long flags;
2284 
2285 	read_lock_irqsave(&iucv_sk_list.lock, flags);
2286 	sk_for_each(sk, &iucv_sk_list.head)
2287 		if (sk == isk) {
2288 			iucv = iucv_sk(sk);
2289 			break;
2290 		}
2291 	read_unlock_irqrestore(&iucv_sk_list.lock, flags);
2292 
2293 	if (!iucv || sock_flag(sk, SOCK_ZAPPED))
2294 		return;
2295 
2296 	list = &iucv->send_skb_q;
2297 	spin_lock_irqsave(&list->lock, flags);
2298 	skb_queue_walk_safe(list, list_skb, nskb) {
2299 		if (skb_shinfo(list_skb) == skb_shinfo(skb)) {
2300 			switch (n) {
2301 			case TX_NOTIFY_OK:
2302 				__skb_unlink(list_skb, list);
2303 				kfree_skb(list_skb);
2304 				iucv_sock_wake_msglim(sk);
2305 				break;
2306 			case TX_NOTIFY_PENDING:
2307 				atomic_inc(&iucv->pendings);
2308 				break;
2309 			case TX_NOTIFY_DELAYED_OK:
2310 				__skb_unlink(list_skb, list);
2311 				atomic_dec(&iucv->pendings);
2312 				if (atomic_read(&iucv->pendings) <= 0)
2313 					iucv_sock_wake_msglim(sk);
2314 				kfree_skb(list_skb);
2315 				break;
2316 			case TX_NOTIFY_UNREACHABLE:
2317 			case TX_NOTIFY_DELAYED_UNREACHABLE:
2318 			case TX_NOTIFY_TPQFULL: /* not yet used */
2319 			case TX_NOTIFY_GENERALERROR:
2320 			case TX_NOTIFY_DELAYED_GENERALERROR:
2321 				__skb_unlink(list_skb, list);
2322 				kfree_skb(list_skb);
2323 				if (sk->sk_state == IUCV_CONNECTED) {
2324 					sk->sk_state = IUCV_DISCONN;
2325 					sk->sk_state_change(sk);
2326 				}
2327 				break;
2328 			}
2329 			break;
2330 		}
2331 	}
2332 	spin_unlock_irqrestore(&list->lock, flags);
2333 
2334 	if (sk->sk_state == IUCV_CLOSING) {
2335 		if (skb_queue_empty(&iucv_sk(sk)->send_skb_q)) {
2336 			sk->sk_state = IUCV_CLOSED;
2337 			sk->sk_state_change(sk);
2338 		}
2339 	}
2340 
2341 }
2342 
2343 /*
2344  * afiucv_netdev_event: handle netdev notifier chain events
2345  */
afiucv_netdev_event(struct notifier_block * this,unsigned long event,void * ptr)2346 static int afiucv_netdev_event(struct notifier_block *this,
2347 			       unsigned long event, void *ptr)
2348 {
2349 	struct net_device *event_dev = netdev_notifier_info_to_dev(ptr);
2350 	struct sock *sk;
2351 	struct iucv_sock *iucv;
2352 
2353 	switch (event) {
2354 	case NETDEV_REBOOT:
2355 	case NETDEV_GOING_DOWN:
2356 		sk_for_each(sk, &iucv_sk_list.head) {
2357 			iucv = iucv_sk(sk);
2358 			if ((iucv->hs_dev == event_dev) &&
2359 			    (sk->sk_state == IUCV_CONNECTED)) {
2360 				if (event == NETDEV_GOING_DOWN)
2361 					iucv_send_ctrl(sk, AF_IUCV_FLAG_FIN);
2362 				sk->sk_state = IUCV_DISCONN;
2363 				sk->sk_state_change(sk);
2364 			}
2365 		}
2366 		break;
2367 	case NETDEV_DOWN:
2368 	case NETDEV_UNREGISTER:
2369 	default:
2370 		break;
2371 	}
2372 	return NOTIFY_DONE;
2373 }
2374 
2375 static struct notifier_block afiucv_netdev_notifier = {
2376 	.notifier_call = afiucv_netdev_event,
2377 };
2378 
2379 static const struct proto_ops iucv_sock_ops = {
2380 	.family		= PF_IUCV,
2381 	.owner		= THIS_MODULE,
2382 	.release	= iucv_sock_release,
2383 	.bind		= iucv_sock_bind,
2384 	.connect	= iucv_sock_connect,
2385 	.listen		= iucv_sock_listen,
2386 	.accept		= iucv_sock_accept,
2387 	.getname	= iucv_sock_getname,
2388 	.sendmsg	= iucv_sock_sendmsg,
2389 	.recvmsg	= iucv_sock_recvmsg,
2390 	.poll		= iucv_sock_poll,
2391 	.ioctl		= sock_no_ioctl,
2392 	.mmap		= sock_no_mmap,
2393 	.socketpair	= sock_no_socketpair,
2394 	.shutdown	= iucv_sock_shutdown,
2395 	.setsockopt	= iucv_sock_setsockopt,
2396 	.getsockopt	= iucv_sock_getsockopt,
2397 };
2398 
2399 static const struct net_proto_family iucv_sock_family_ops = {
2400 	.family	= AF_IUCV,
2401 	.owner	= THIS_MODULE,
2402 	.create	= iucv_sock_create,
2403 };
2404 
2405 static struct packet_type iucv_packet_type = {
2406 	.type = cpu_to_be16(ETH_P_AF_IUCV),
2407 	.func = afiucv_hs_rcv,
2408 };
2409 
afiucv_iucv_init(void)2410 static int afiucv_iucv_init(void)
2411 {
2412 	int err;
2413 
2414 	err = pr_iucv->iucv_register(&af_iucv_handler, 0);
2415 	if (err)
2416 		goto out;
2417 	/* establish dummy device */
2418 	af_iucv_driver.bus = pr_iucv->bus;
2419 	err = driver_register(&af_iucv_driver);
2420 	if (err)
2421 		goto out_iucv;
2422 	af_iucv_dev = kzalloc(sizeof(struct device), GFP_KERNEL);
2423 	if (!af_iucv_dev) {
2424 		err = -ENOMEM;
2425 		goto out_driver;
2426 	}
2427 	dev_set_name(af_iucv_dev, "af_iucv");
2428 	af_iucv_dev->bus = pr_iucv->bus;
2429 	af_iucv_dev->parent = pr_iucv->root;
2430 	af_iucv_dev->release = (void (*)(struct device *))kfree;
2431 	af_iucv_dev->driver = &af_iucv_driver;
2432 	err = device_register(af_iucv_dev);
2433 	if (err)
2434 		goto out_iucv_dev;
2435 	return 0;
2436 
2437 out_iucv_dev:
2438 	put_device(af_iucv_dev);
2439 out_driver:
2440 	driver_unregister(&af_iucv_driver);
2441 out_iucv:
2442 	pr_iucv->iucv_unregister(&af_iucv_handler, 0);
2443 out:
2444 	return err;
2445 }
2446 
afiucv_iucv_exit(void)2447 static void afiucv_iucv_exit(void)
2448 {
2449 	device_unregister(af_iucv_dev);
2450 	driver_unregister(&af_iucv_driver);
2451 	pr_iucv->iucv_unregister(&af_iucv_handler, 0);
2452 }
2453 
afiucv_init(void)2454 static int __init afiucv_init(void)
2455 {
2456 	int err;
2457 
2458 	if (MACHINE_IS_VM) {
2459 		cpcmd("QUERY USERID", iucv_userid, sizeof(iucv_userid), &err);
2460 		if (unlikely(err)) {
2461 			WARN_ON(err);
2462 			err = -EPROTONOSUPPORT;
2463 			goto out;
2464 		}
2465 
2466 		pr_iucv = try_then_request_module(symbol_get(iucv_if), "iucv");
2467 		if (!pr_iucv) {
2468 			printk(KERN_WARNING "iucv_if lookup failed\n");
2469 			memset(&iucv_userid, 0, sizeof(iucv_userid));
2470 		}
2471 	} else {
2472 		memset(&iucv_userid, 0, sizeof(iucv_userid));
2473 		pr_iucv = NULL;
2474 	}
2475 
2476 	err = proto_register(&iucv_proto, 0);
2477 	if (err)
2478 		goto out;
2479 	err = sock_register(&iucv_sock_family_ops);
2480 	if (err)
2481 		goto out_proto;
2482 
2483 	if (pr_iucv) {
2484 		err = afiucv_iucv_init();
2485 		if (err)
2486 			goto out_sock;
2487 	}
2488 
2489 	err = register_netdevice_notifier(&afiucv_netdev_notifier);
2490 	if (err)
2491 		goto out_notifier;
2492 
2493 	dev_add_pack(&iucv_packet_type);
2494 	return 0;
2495 
2496 out_notifier:
2497 	if (pr_iucv)
2498 		afiucv_iucv_exit();
2499 out_sock:
2500 	sock_unregister(PF_IUCV);
2501 out_proto:
2502 	proto_unregister(&iucv_proto);
2503 out:
2504 	if (pr_iucv)
2505 		symbol_put(iucv_if);
2506 	return err;
2507 }
2508 
afiucv_exit(void)2509 static void __exit afiucv_exit(void)
2510 {
2511 	if (pr_iucv) {
2512 		afiucv_iucv_exit();
2513 		symbol_put(iucv_if);
2514 	}
2515 
2516 	unregister_netdevice_notifier(&afiucv_netdev_notifier);
2517 	dev_remove_pack(&iucv_packet_type);
2518 	sock_unregister(PF_IUCV);
2519 	proto_unregister(&iucv_proto);
2520 }
2521 
2522 module_init(afiucv_init);
2523 module_exit(afiucv_exit);
2524 
2525 MODULE_AUTHOR("Jennifer Hunt <jenhunt@us.ibm.com>");
2526 MODULE_DESCRIPTION("IUCV Sockets ver " VERSION);
2527 MODULE_VERSION(VERSION);
2528 MODULE_LICENSE("GPL");
2529 MODULE_ALIAS_NETPROTO(PF_IUCV);
2530