1/* net/atm/common.c - ATM sockets (common part for PVC and SVC) */
2
3/* Written 1995-2000 by Werner Almesberger, EPFL LRC/ICA */
4
5#define pr_fmt(fmt) KBUILD_MODNAME ":%s: " fmt, __func__
6
7#include <linux/module.h>
8#include <linux/kmod.h>
9#include <linux/net.h>		/* struct socket, struct proto_ops */
10#include <linux/atm.h>		/* ATM stuff */
11#include <linux/atmdev.h>
12#include <linux/socket.h>	/* SOL_SOCKET */
13#include <linux/errno.h>	/* error codes */
14#include <linux/capability.h>
15#include <linux/mm.h>
16#include <linux/sched.h>
17#include <linux/time.h>		/* struct timeval */
18#include <linux/skbuff.h>
19#include <linux/bitops.h>
20#include <linux/init.h>
21#include <linux/slab.h>
22#include <net/sock.h>		/* struct sock */
23#include <linux/uaccess.h>
24#include <linux/poll.h>
25
26#include <linux/atomic.h>
27
28#include "resources.h"		/* atm_find_dev */
29#include "common.h"		/* prototypes */
30#include "protocols.h"		/* atm_init_<transport> */
31#include "addr.h"		/* address registry */
32#include "signaling.h"		/* for WAITING and sigd_attach */
33
34struct hlist_head vcc_hash[VCC_HTABLE_SIZE];
35EXPORT_SYMBOL(vcc_hash);
36
37DEFINE_RWLOCK(vcc_sklist_lock);
38EXPORT_SYMBOL(vcc_sklist_lock);
39
40static ATOMIC_NOTIFIER_HEAD(atm_dev_notify_chain);
41
42static void __vcc_insert_socket(struct sock *sk)
43{
44	struct atm_vcc *vcc = atm_sk(sk);
45	struct hlist_head *head = &vcc_hash[vcc->vci & (VCC_HTABLE_SIZE - 1)];
46	sk->sk_hash = vcc->vci & (VCC_HTABLE_SIZE - 1);
47	sk_add_node(sk, head);
48}
49
50void vcc_insert_socket(struct sock *sk)
51{
52	write_lock_irq(&vcc_sklist_lock);
53	__vcc_insert_socket(sk);
54	write_unlock_irq(&vcc_sklist_lock);
55}
56EXPORT_SYMBOL(vcc_insert_socket);
57
58static void vcc_remove_socket(struct sock *sk)
59{
60	write_lock_irq(&vcc_sklist_lock);
61	sk_del_node_init(sk);
62	write_unlock_irq(&vcc_sklist_lock);
63}
64
65static struct sk_buff *alloc_tx(struct atm_vcc *vcc, unsigned int size)
66{
67	struct sk_buff *skb;
68	struct sock *sk = sk_atm(vcc);
69
70	if (sk_wmem_alloc_get(sk) && !atm_may_send(vcc, size)) {
71		pr_debug("Sorry: wmem_alloc = %d, size = %d, sndbuf = %d\n",
72			 sk_wmem_alloc_get(sk), size, sk->sk_sndbuf);
73		return NULL;
74	}
75	while (!(skb = alloc_skb(size, GFP_KERNEL)))
76		schedule();
77	pr_debug("%d += %d\n", sk_wmem_alloc_get(sk), skb->truesize);
78	atomic_add(skb->truesize, &sk->sk_wmem_alloc);
79	return skb;
80}
81
82static void vcc_sock_destruct(struct sock *sk)
83{
84	if (atomic_read(&sk->sk_rmem_alloc))
85		printk(KERN_DEBUG "%s: rmem leakage (%d bytes) detected.\n",
86		       __func__, atomic_read(&sk->sk_rmem_alloc));
87
88	if (atomic_read(&sk->sk_wmem_alloc))
89		printk(KERN_DEBUG "%s: wmem leakage (%d bytes) detected.\n",
90		       __func__, atomic_read(&sk->sk_wmem_alloc));
91}
92
93static void vcc_def_wakeup(struct sock *sk)
94{
95	struct socket_wq *wq;
96
97	rcu_read_lock();
98	wq = rcu_dereference(sk->sk_wq);
99	if (wq_has_sleeper(wq))
100		wake_up(&wq->wait);
101	rcu_read_unlock();
102}
103
104static inline int vcc_writable(struct sock *sk)
105{
106	struct atm_vcc *vcc = atm_sk(sk);
107
108	return (vcc->qos.txtp.max_sdu +
109		atomic_read(&sk->sk_wmem_alloc)) <= sk->sk_sndbuf;
110}
111
112static void vcc_write_space(struct sock *sk)
113{
114	struct socket_wq *wq;
115
116	rcu_read_lock();
117
118	if (vcc_writable(sk)) {
119		wq = rcu_dereference(sk->sk_wq);
120		if (wq_has_sleeper(wq))
121			wake_up_interruptible(&wq->wait);
122
123		sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
124	}
125
126	rcu_read_unlock();
127}
128
129static void vcc_release_cb(struct sock *sk)
130{
131	struct atm_vcc *vcc = atm_sk(sk);
132
133	if (vcc->release_cb)
134		vcc->release_cb(vcc);
135}
136
137static struct proto vcc_proto = {
138	.name	  = "VCC",
139	.owner	  = THIS_MODULE,
140	.obj_size = sizeof(struct atm_vcc),
141	.release_cb = vcc_release_cb,
142};
143
144int vcc_create(struct net *net, struct socket *sock, int protocol, int family, int kern)
145{
146	struct sock *sk;
147	struct atm_vcc *vcc;
148
149	sock->sk = NULL;
150	if (sock->type == SOCK_STREAM)
151		return -EINVAL;
152	sk = sk_alloc(net, family, GFP_KERNEL, &vcc_proto, kern);
153	if (!sk)
154		return -ENOMEM;
155	sock_init_data(sock, sk);
156	sk->sk_state_change = vcc_def_wakeup;
157	sk->sk_write_space = vcc_write_space;
158
159	vcc = atm_sk(sk);
160	vcc->dev = NULL;
161	memset(&vcc->local, 0, sizeof(struct sockaddr_atmsvc));
162	memset(&vcc->remote, 0, sizeof(struct sockaddr_atmsvc));
163	vcc->qos.txtp.max_sdu = 1 << 16; /* for meta VCs */
164	atomic_set(&sk->sk_wmem_alloc, 1);
165	atomic_set(&sk->sk_rmem_alloc, 0);
166	vcc->push = NULL;
167	vcc->pop = NULL;
168	vcc->owner = NULL;
169	vcc->push_oam = NULL;
170	vcc->release_cb = NULL;
171	vcc->vpi = vcc->vci = 0; /* no VCI/VPI yet */
172	vcc->atm_options = vcc->aal_options = 0;
173	sk->sk_destruct = vcc_sock_destruct;
174	return 0;
175}
176
177static void vcc_destroy_socket(struct sock *sk)
178{
179	struct atm_vcc *vcc = atm_sk(sk);
180	struct sk_buff *skb;
181
182	set_bit(ATM_VF_CLOSE, &vcc->flags);
183	clear_bit(ATM_VF_READY, &vcc->flags);
184	if (vcc->dev) {
185		if (vcc->dev->ops->close)
186			vcc->dev->ops->close(vcc);
187		if (vcc->push)
188			vcc->push(vcc, NULL); /* atmarpd has no push */
189		module_put(vcc->owner);
190
191		while ((skb = skb_dequeue(&sk->sk_receive_queue)) != NULL) {
192			atm_return(vcc, skb->truesize);
193			kfree_skb(skb);
194		}
195
196		module_put(vcc->dev->ops->owner);
197		atm_dev_put(vcc->dev);
198	}
199
200	vcc_remove_socket(sk);
201}
202
203int vcc_release(struct socket *sock)
204{
205	struct sock *sk = sock->sk;
206
207	if (sk) {
208		lock_sock(sk);
209		vcc_destroy_socket(sock->sk);
210		release_sock(sk);
211		sock_put(sk);
212	}
213
214	return 0;
215}
216
217void vcc_release_async(struct atm_vcc *vcc, int reply)
218{
219	struct sock *sk = sk_atm(vcc);
220
221	set_bit(ATM_VF_CLOSE, &vcc->flags);
222	sk->sk_shutdown |= RCV_SHUTDOWN;
223	sk->sk_err = -reply;
224	clear_bit(ATM_VF_WAITING, &vcc->flags);
225	sk->sk_state_change(sk);
226}
227EXPORT_SYMBOL(vcc_release_async);
228
229void vcc_process_recv_queue(struct atm_vcc *vcc)
230{
231	struct sk_buff_head queue, *rq;
232	struct sk_buff *skb, *tmp;
233	unsigned long flags;
234
235	__skb_queue_head_init(&queue);
236	rq = &sk_atm(vcc)->sk_receive_queue;
237
238	spin_lock_irqsave(&rq->lock, flags);
239	skb_queue_splice_init(rq, &queue);
240	spin_unlock_irqrestore(&rq->lock, flags);
241
242	skb_queue_walk_safe(&queue, skb, tmp) {
243		__skb_unlink(skb, &queue);
244		vcc->push(vcc, skb);
245	}
246}
247EXPORT_SYMBOL(vcc_process_recv_queue);
248
249void atm_dev_signal_change(struct atm_dev *dev, char signal)
250{
251	pr_debug("%s signal=%d dev=%p number=%d dev->signal=%d\n",
252		__func__, signal, dev, dev->number, dev->signal);
253
254	/* atm driver sending invalid signal */
255	WARN_ON(signal < ATM_PHY_SIG_LOST || signal > ATM_PHY_SIG_FOUND);
256
257	if (dev->signal == signal)
258		return; /* no change */
259
260	dev->signal = signal;
261
262	atomic_notifier_call_chain(&atm_dev_notify_chain, signal, dev);
263}
264EXPORT_SYMBOL(atm_dev_signal_change);
265
266void atm_dev_release_vccs(struct atm_dev *dev)
267{
268	int i;
269
270	write_lock_irq(&vcc_sklist_lock);
271	for (i = 0; i < VCC_HTABLE_SIZE; i++) {
272		struct hlist_head *head = &vcc_hash[i];
273		struct hlist_node *tmp;
274		struct sock *s;
275		struct atm_vcc *vcc;
276
277		sk_for_each_safe(s, tmp, head) {
278			vcc = atm_sk(s);
279			if (vcc->dev == dev) {
280				vcc_release_async(vcc, -EPIPE);
281				sk_del_node_init(s);
282			}
283		}
284	}
285	write_unlock_irq(&vcc_sklist_lock);
286}
287EXPORT_SYMBOL(atm_dev_release_vccs);
288
289static int adjust_tp(struct atm_trafprm *tp, unsigned char aal)
290{
291	int max_sdu;
292
293	if (!tp->traffic_class)
294		return 0;
295	switch (aal) {
296	case ATM_AAL0:
297		max_sdu = ATM_CELL_SIZE-1;
298		break;
299	case ATM_AAL34:
300		max_sdu = ATM_MAX_AAL34_PDU;
301		break;
302	default:
303		pr_warn("AAL problems ... (%d)\n", aal);
304		/* fall through */
305	case ATM_AAL5:
306		max_sdu = ATM_MAX_AAL5_PDU;
307	}
308	if (!tp->max_sdu)
309		tp->max_sdu = max_sdu;
310	else if (tp->max_sdu > max_sdu)
311		return -EINVAL;
312	if (!tp->max_cdv)
313		tp->max_cdv = ATM_MAX_CDV;
314	return 0;
315}
316
317static int check_ci(const struct atm_vcc *vcc, short vpi, int vci)
318{
319	struct hlist_head *head = &vcc_hash[vci & (VCC_HTABLE_SIZE - 1)];
320	struct sock *s;
321	struct atm_vcc *walk;
322
323	sk_for_each(s, head) {
324		walk = atm_sk(s);
325		if (walk->dev != vcc->dev)
326			continue;
327		if (test_bit(ATM_VF_ADDR, &walk->flags) && walk->vpi == vpi &&
328		    walk->vci == vci && ((walk->qos.txtp.traffic_class !=
329		    ATM_NONE && vcc->qos.txtp.traffic_class != ATM_NONE) ||
330		    (walk->qos.rxtp.traffic_class != ATM_NONE &&
331		    vcc->qos.rxtp.traffic_class != ATM_NONE)))
332			return -EADDRINUSE;
333	}
334
335	/* allow VCCs with same VPI/VCI iff they don't collide on
336	   TX/RX (but we may refuse such sharing for other reasons,
337	   e.g. if protocol requires to have both channels) */
338
339	return 0;
340}
341
342static int find_ci(const struct atm_vcc *vcc, short *vpi, int *vci)
343{
344	static short p;        /* poor man's per-device cache */
345	static int c;
346	short old_p;
347	int old_c;
348	int err;
349
350	if (*vpi != ATM_VPI_ANY && *vci != ATM_VCI_ANY) {
351		err = check_ci(vcc, *vpi, *vci);
352		return err;
353	}
354	/* last scan may have left values out of bounds for current device */
355	if (*vpi != ATM_VPI_ANY)
356		p = *vpi;
357	else if (p >= 1 << vcc->dev->ci_range.vpi_bits)
358		p = 0;
359	if (*vci != ATM_VCI_ANY)
360		c = *vci;
361	else if (c < ATM_NOT_RSV_VCI || c >= 1 << vcc->dev->ci_range.vci_bits)
362			c = ATM_NOT_RSV_VCI;
363	old_p = p;
364	old_c = c;
365	do {
366		if (!check_ci(vcc, p, c)) {
367			*vpi = p;
368			*vci = c;
369			return 0;
370		}
371		if (*vci == ATM_VCI_ANY) {
372			c++;
373			if (c >= 1 << vcc->dev->ci_range.vci_bits)
374				c = ATM_NOT_RSV_VCI;
375		}
376		if ((c == ATM_NOT_RSV_VCI || *vci != ATM_VCI_ANY) &&
377		    *vpi == ATM_VPI_ANY) {
378			p++;
379			if (p >= 1 << vcc->dev->ci_range.vpi_bits)
380				p = 0;
381		}
382	} while (old_p != p || old_c != c);
383	return -EADDRINUSE;
384}
385
386static int __vcc_connect(struct atm_vcc *vcc, struct atm_dev *dev, short vpi,
387			 int vci)
388{
389	struct sock *sk = sk_atm(vcc);
390	int error;
391
392	if ((vpi != ATM_VPI_UNSPEC && vpi != ATM_VPI_ANY &&
393	    vpi >> dev->ci_range.vpi_bits) || (vci != ATM_VCI_UNSPEC &&
394	    vci != ATM_VCI_ANY && vci >> dev->ci_range.vci_bits))
395		return -EINVAL;
396	if (vci > 0 && vci < ATM_NOT_RSV_VCI && !capable(CAP_NET_BIND_SERVICE))
397		return -EPERM;
398	error = -ENODEV;
399	if (!try_module_get(dev->ops->owner))
400		return error;
401	vcc->dev = dev;
402	write_lock_irq(&vcc_sklist_lock);
403	if (test_bit(ATM_DF_REMOVED, &dev->flags) ||
404	    (error = find_ci(vcc, &vpi, &vci))) {
405		write_unlock_irq(&vcc_sklist_lock);
406		goto fail_module_put;
407	}
408	vcc->vpi = vpi;
409	vcc->vci = vci;
410	__vcc_insert_socket(sk);
411	write_unlock_irq(&vcc_sklist_lock);
412	switch (vcc->qos.aal) {
413	case ATM_AAL0:
414		error = atm_init_aal0(vcc);
415		vcc->stats = &dev->stats.aal0;
416		break;
417	case ATM_AAL34:
418		error = atm_init_aal34(vcc);
419		vcc->stats = &dev->stats.aal34;
420		break;
421	case ATM_NO_AAL:
422		/* ATM_AAL5 is also used in the "0 for default" case */
423		vcc->qos.aal = ATM_AAL5;
424		/* fall through */
425	case ATM_AAL5:
426		error = atm_init_aal5(vcc);
427		vcc->stats = &dev->stats.aal5;
428		break;
429	default:
430		error = -EPROTOTYPE;
431	}
432	if (!error)
433		error = adjust_tp(&vcc->qos.txtp, vcc->qos.aal);
434	if (!error)
435		error = adjust_tp(&vcc->qos.rxtp, vcc->qos.aal);
436	if (error)
437		goto fail;
438	pr_debug("VCC %d.%d, AAL %d\n", vpi, vci, vcc->qos.aal);
439	pr_debug("  TX: %d, PCR %d..%d, SDU %d\n",
440		 vcc->qos.txtp.traffic_class,
441		 vcc->qos.txtp.min_pcr,
442		 vcc->qos.txtp.max_pcr,
443		 vcc->qos.txtp.max_sdu);
444	pr_debug("  RX: %d, PCR %d..%d, SDU %d\n",
445		 vcc->qos.rxtp.traffic_class,
446		 vcc->qos.rxtp.min_pcr,
447		 vcc->qos.rxtp.max_pcr,
448		 vcc->qos.rxtp.max_sdu);
449
450	if (dev->ops->open) {
451		error = dev->ops->open(vcc);
452		if (error)
453			goto fail;
454	}
455	return 0;
456
457fail:
458	vcc_remove_socket(sk);
459fail_module_put:
460	module_put(dev->ops->owner);
461	/* ensure we get dev module ref count correct */
462	vcc->dev = NULL;
463	return error;
464}
465
466int vcc_connect(struct socket *sock, int itf, short vpi, int vci)
467{
468	struct atm_dev *dev;
469	struct atm_vcc *vcc = ATM_SD(sock);
470	int error;
471
472	pr_debug("(vpi %d, vci %d)\n", vpi, vci);
473	if (sock->state == SS_CONNECTED)
474		return -EISCONN;
475	if (sock->state != SS_UNCONNECTED)
476		return -EINVAL;
477	if (!(vpi || vci))
478		return -EINVAL;
479
480	if (vpi != ATM_VPI_UNSPEC && vci != ATM_VCI_UNSPEC)
481		clear_bit(ATM_VF_PARTIAL, &vcc->flags);
482	else
483		if (test_bit(ATM_VF_PARTIAL, &vcc->flags))
484			return -EINVAL;
485	pr_debug("(TX: cl %d,bw %d-%d,sdu %d; "
486		 "RX: cl %d,bw %d-%d,sdu %d,AAL %s%d)\n",
487		 vcc->qos.txtp.traffic_class, vcc->qos.txtp.min_pcr,
488		 vcc->qos.txtp.max_pcr, vcc->qos.txtp.max_sdu,
489		 vcc->qos.rxtp.traffic_class, vcc->qos.rxtp.min_pcr,
490		 vcc->qos.rxtp.max_pcr, vcc->qos.rxtp.max_sdu,
491		 vcc->qos.aal == ATM_AAL5 ? "" :
492		 vcc->qos.aal == ATM_AAL0 ? "" : " ??? code ",
493		 vcc->qos.aal == ATM_AAL0 ? 0 : vcc->qos.aal);
494	if (!test_bit(ATM_VF_HASQOS, &vcc->flags))
495		return -EBADFD;
496	if (vcc->qos.txtp.traffic_class == ATM_ANYCLASS ||
497	    vcc->qos.rxtp.traffic_class == ATM_ANYCLASS)
498		return -EINVAL;
499	if (likely(itf != ATM_ITF_ANY)) {
500		dev = try_then_request_module(atm_dev_lookup(itf),
501					      "atm-device-%d", itf);
502	} else {
503		dev = NULL;
504		mutex_lock(&atm_dev_mutex);
505		if (!list_empty(&atm_devs)) {
506			dev = list_entry(atm_devs.next,
507					 struct atm_dev, dev_list);
508			atm_dev_hold(dev);
509		}
510		mutex_unlock(&atm_dev_mutex);
511	}
512	if (!dev)
513		return -ENODEV;
514	error = __vcc_connect(vcc, dev, vpi, vci);
515	if (error) {
516		atm_dev_put(dev);
517		return error;
518	}
519	if (vpi == ATM_VPI_UNSPEC || vci == ATM_VCI_UNSPEC)
520		set_bit(ATM_VF_PARTIAL, &vcc->flags);
521	if (test_bit(ATM_VF_READY, &ATM_SD(sock)->flags))
522		sock->state = SS_CONNECTED;
523	return 0;
524}
525
526int vcc_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
527		int flags)
528{
529	struct sock *sk = sock->sk;
530	struct atm_vcc *vcc;
531	struct sk_buff *skb;
532	int copied, error = -EINVAL;
533
534	if (sock->state != SS_CONNECTED)
535		return -ENOTCONN;
536
537	/* only handle MSG_DONTWAIT and MSG_PEEK */
538	if (flags & ~(MSG_DONTWAIT | MSG_PEEK))
539		return -EOPNOTSUPP;
540
541	vcc = ATM_SD(sock);
542	if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
543	    test_bit(ATM_VF_CLOSE, &vcc->flags) ||
544	    !test_bit(ATM_VF_READY, &vcc->flags))
545		return 0;
546
547	skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &error);
548	if (!skb)
549		return error;
550
551	copied = skb->len;
552	if (copied > size) {
553		copied = size;
554		msg->msg_flags |= MSG_TRUNC;
555	}
556
557	error = skb_copy_datagram_msg(skb, 0, msg, copied);
558	if (error)
559		return error;
560	sock_recv_ts_and_drops(msg, sk, skb);
561
562	if (!(flags & MSG_PEEK)) {
563		pr_debug("%d -= %d\n", atomic_read(&sk->sk_rmem_alloc),
564			 skb->truesize);
565		atm_return(vcc, skb->truesize);
566	}
567
568	skb_free_datagram(sk, skb);
569	return copied;
570}
571
572int vcc_sendmsg(struct socket *sock, struct msghdr *m, size_t size)
573{
574	struct sock *sk = sock->sk;
575	DEFINE_WAIT(wait);
576	struct atm_vcc *vcc;
577	struct sk_buff *skb;
578	int eff, error;
579
580	lock_sock(sk);
581	if (sock->state != SS_CONNECTED) {
582		error = -ENOTCONN;
583		goto out;
584	}
585	if (m->msg_name) {
586		error = -EISCONN;
587		goto out;
588	}
589	vcc = ATM_SD(sock);
590	if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
591	    test_bit(ATM_VF_CLOSE, &vcc->flags) ||
592	    !test_bit(ATM_VF_READY, &vcc->flags)) {
593		error = -EPIPE;
594		send_sig(SIGPIPE, current, 0);
595		goto out;
596	}
597	if (!size) {
598		error = 0;
599		goto out;
600	}
601	if (size > vcc->qos.txtp.max_sdu) {
602		error = -EMSGSIZE;
603		goto out;
604	}
605
606	eff = (size+3) & ~3; /* align to word boundary */
607	prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
608	error = 0;
609	while (!(skb = alloc_tx(vcc, eff))) {
610		if (m->msg_flags & MSG_DONTWAIT) {
611			error = -EAGAIN;
612			break;
613		}
614		schedule();
615		if (signal_pending(current)) {
616			error = -ERESTARTSYS;
617			break;
618		}
619		if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
620		    test_bit(ATM_VF_CLOSE, &vcc->flags) ||
621		    !test_bit(ATM_VF_READY, &vcc->flags)) {
622			error = -EPIPE;
623			send_sig(SIGPIPE, current, 0);
624			break;
625		}
626		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
627	}
628	finish_wait(sk_sleep(sk), &wait);
629	if (error)
630		goto out;
631	skb->dev = NULL; /* for paths shared with net_device interfaces */
632	ATM_SKB(skb)->atm_options = vcc->atm_options;
633	if (copy_from_iter(skb_put(skb, size), size, &m->msg_iter) != size) {
634		kfree_skb(skb);
635		error = -EFAULT;
636		goto out;
637	}
638	if (eff != size)
639		memset(skb->data + size, 0, eff-size);
640	error = vcc->dev->ops->send(vcc, skb);
641	error = error ? error : size;
642out:
643	release_sock(sk);
644	return error;
645}
646
647unsigned int vcc_poll(struct file *file, struct socket *sock, poll_table *wait)
648{
649	struct sock *sk = sock->sk;
650	struct atm_vcc *vcc;
651	unsigned int mask;
652
653	sock_poll_wait(file, sk_sleep(sk), wait);
654	mask = 0;
655
656	vcc = ATM_SD(sock);
657
658	/* exceptional events */
659	if (sk->sk_err)
660		mask = POLLERR;
661
662	if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
663	    test_bit(ATM_VF_CLOSE, &vcc->flags))
664		mask |= POLLHUP;
665
666	/* readable? */
667	if (!skb_queue_empty(&sk->sk_receive_queue))
668		mask |= POLLIN | POLLRDNORM;
669
670	/* writable? */
671	if (sock->state == SS_CONNECTING &&
672	    test_bit(ATM_VF_WAITING, &vcc->flags))
673		return mask;
674
675	if (vcc->qos.txtp.traffic_class != ATM_NONE &&
676	    vcc_writable(sk))
677		mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
678
679	return mask;
680}
681
682static int atm_change_qos(struct atm_vcc *vcc, struct atm_qos *qos)
683{
684	int error;
685
686	/*
687	 * Don't let the QoS change the already connected AAL type nor the
688	 * traffic class.
689	 */
690	if (qos->aal != vcc->qos.aal ||
691	    qos->rxtp.traffic_class != vcc->qos.rxtp.traffic_class ||
692	    qos->txtp.traffic_class != vcc->qos.txtp.traffic_class)
693		return -EINVAL;
694	error = adjust_tp(&qos->txtp, qos->aal);
695	if (!error)
696		error = adjust_tp(&qos->rxtp, qos->aal);
697	if (error)
698		return error;
699	if (!vcc->dev->ops->change_qos)
700		return -EOPNOTSUPP;
701	if (sk_atm(vcc)->sk_family == AF_ATMPVC)
702		return vcc->dev->ops->change_qos(vcc, qos, ATM_MF_SET);
703	return svc_change_qos(vcc, qos);
704}
705
706static int check_tp(const struct atm_trafprm *tp)
707{
708	/* @@@ Should be merged with adjust_tp */
709	if (!tp->traffic_class || tp->traffic_class == ATM_ANYCLASS)
710		return 0;
711	if (tp->traffic_class != ATM_UBR && !tp->min_pcr && !tp->pcr &&
712	    !tp->max_pcr)
713		return -EINVAL;
714	if (tp->min_pcr == ATM_MAX_PCR)
715		return -EINVAL;
716	if (tp->min_pcr && tp->max_pcr && tp->max_pcr != ATM_MAX_PCR &&
717	    tp->min_pcr > tp->max_pcr)
718		return -EINVAL;
719	/*
720	 * We allow pcr to be outside [min_pcr,max_pcr], because later
721	 * adjustment may still push it in the valid range.
722	 */
723	return 0;
724}
725
726static int check_qos(const struct atm_qos *qos)
727{
728	int error;
729
730	if (!qos->txtp.traffic_class && !qos->rxtp.traffic_class)
731		return -EINVAL;
732	if (qos->txtp.traffic_class != qos->rxtp.traffic_class &&
733	    qos->txtp.traffic_class && qos->rxtp.traffic_class &&
734	    qos->txtp.traffic_class != ATM_ANYCLASS &&
735	    qos->rxtp.traffic_class != ATM_ANYCLASS)
736		return -EINVAL;
737	error = check_tp(&qos->txtp);
738	if (error)
739		return error;
740	return check_tp(&qos->rxtp);
741}
742
743int vcc_setsockopt(struct socket *sock, int level, int optname,
744		   char __user *optval, unsigned int optlen)
745{
746	struct atm_vcc *vcc;
747	unsigned long value;
748	int error;
749
750	if (__SO_LEVEL_MATCH(optname, level) && optlen != __SO_SIZE(optname))
751		return -EINVAL;
752
753	vcc = ATM_SD(sock);
754	switch (optname) {
755	case SO_ATMQOS:
756	{
757		struct atm_qos qos;
758
759		if (copy_from_user(&qos, optval, sizeof(qos)))
760			return -EFAULT;
761		error = check_qos(&qos);
762		if (error)
763			return error;
764		if (sock->state == SS_CONNECTED)
765			return atm_change_qos(vcc, &qos);
766		if (sock->state != SS_UNCONNECTED)
767			return -EBADFD;
768		vcc->qos = qos;
769		set_bit(ATM_VF_HASQOS, &vcc->flags);
770		return 0;
771	}
772	case SO_SETCLP:
773		if (get_user(value, (unsigned long __user *)optval))
774			return -EFAULT;
775		if (value)
776			vcc->atm_options |= ATM_ATMOPT_CLP;
777		else
778			vcc->atm_options &= ~ATM_ATMOPT_CLP;
779		return 0;
780	default:
781		if (level == SOL_SOCKET)
782			return -EINVAL;
783		break;
784	}
785	if (!vcc->dev || !vcc->dev->ops->setsockopt)
786		return -EINVAL;
787	return vcc->dev->ops->setsockopt(vcc, level, optname, optval, optlen);
788}
789
790int vcc_getsockopt(struct socket *sock, int level, int optname,
791		   char __user *optval, int __user *optlen)
792{
793	struct atm_vcc *vcc;
794	int len;
795
796	if (get_user(len, optlen))
797		return -EFAULT;
798	if (__SO_LEVEL_MATCH(optname, level) && len != __SO_SIZE(optname))
799		return -EINVAL;
800
801	vcc = ATM_SD(sock);
802	switch (optname) {
803	case SO_ATMQOS:
804		if (!test_bit(ATM_VF_HASQOS, &vcc->flags))
805			return -EINVAL;
806		return copy_to_user(optval, &vcc->qos, sizeof(vcc->qos))
807			? -EFAULT : 0;
808	case SO_SETCLP:
809		return put_user(vcc->atm_options & ATM_ATMOPT_CLP ? 1 : 0,
810				(unsigned long __user *)optval) ? -EFAULT : 0;
811	case SO_ATMPVC:
812	{
813		struct sockaddr_atmpvc pvc;
814
815		if (!vcc->dev || !test_bit(ATM_VF_ADDR, &vcc->flags))
816			return -ENOTCONN;
817		memset(&pvc, 0, sizeof(pvc));
818		pvc.sap_family = AF_ATMPVC;
819		pvc.sap_addr.itf = vcc->dev->number;
820		pvc.sap_addr.vpi = vcc->vpi;
821		pvc.sap_addr.vci = vcc->vci;
822		return copy_to_user(optval, &pvc, sizeof(pvc)) ? -EFAULT : 0;
823	}
824	default:
825		if (level == SOL_SOCKET)
826			return -EINVAL;
827		break;
828	}
829	if (!vcc->dev || !vcc->dev->ops->getsockopt)
830		return -EINVAL;
831	return vcc->dev->ops->getsockopt(vcc, level, optname, optval, len);
832}
833
834int register_atmdevice_notifier(struct notifier_block *nb)
835{
836	return atomic_notifier_chain_register(&atm_dev_notify_chain, nb);
837}
838EXPORT_SYMBOL_GPL(register_atmdevice_notifier);
839
840void unregister_atmdevice_notifier(struct notifier_block *nb)
841{
842	atomic_notifier_chain_unregister(&atm_dev_notify_chain, nb);
843}
844EXPORT_SYMBOL_GPL(unregister_atmdevice_notifier);
845
846static int __init atm_init(void)
847{
848	int error;
849
850	error = proto_register(&vcc_proto, 0);
851	if (error < 0)
852		goto out;
853	error = atmpvc_init();
854	if (error < 0) {
855		pr_err("atmpvc_init() failed with %d\n", error);
856		goto out_unregister_vcc_proto;
857	}
858	error = atmsvc_init();
859	if (error < 0) {
860		pr_err("atmsvc_init() failed with %d\n", error);
861		goto out_atmpvc_exit;
862	}
863	error = atm_proc_init();
864	if (error < 0) {
865		pr_err("atm_proc_init() failed with %d\n", error);
866		goto out_atmsvc_exit;
867	}
868	error = atm_sysfs_init();
869	if (error < 0) {
870		pr_err("atm_sysfs_init() failed with %d\n", error);
871		goto out_atmproc_exit;
872	}
873out:
874	return error;
875out_atmproc_exit:
876	atm_proc_exit();
877out_atmsvc_exit:
878	atmsvc_exit();
879out_atmpvc_exit:
880	atmsvc_exit();
881out_unregister_vcc_proto:
882	proto_unregister(&vcc_proto);
883	goto out;
884}
885
886static void __exit atm_exit(void)
887{
888	atm_proc_exit();
889	atm_sysfs_exit();
890	atmsvc_exit();
891	atmpvc_exit();
892	proto_unregister(&vcc_proto);
893}
894
895subsys_initcall(atm_init);
896
897module_exit(atm_exit);
898
899MODULE_LICENSE("GPL");
900MODULE_ALIAS_NETPROTO(PF_ATMPVC);
901MODULE_ALIAS_NETPROTO(PF_ATMSVC);
902