?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
netfilter: nft_set_rbtree: fix panic when destroying set by GC
[linux/fpc-iii.git] / net / dccp / proto.c
blob0d56e36a6db7b77dcdeb9697dd81bf62895e6e4c
1 /*
2 * net/dccp/proto.c
3 *
4 * An implementation of the DCCP protocol
5 * Arnaldo Carvalho de Melo <acme@conectiva.com.br>
6 *
7 * This program is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License version 2 as
9 * published by the Free Software Foundation.
10 */
11
12 #include <linux/dccp.h>
13 #include <linux/module.h>
14 #include <linux/types.h>
15 #include <linux/sched.h>
16 #include <linux/kernel.h>
17 #include <linux/skbuff.h>
18 #include <linux/netdevice.h>
19 #include <linux/in.h>
20 #include <linux/if_arp.h>
21 #include <linux/init.h>
22 #include <linux/random.h>
23 #include <linux/slab.h>
24 #include <net/checksum.h>
25
26 #include <net/inet_sock.h>
27 #include <net/inet_common.h>
28 #include <net/sock.h>
29 #include <net/xfrm.h>
30
31 #include <asm/ioctls.h>
32 #include <linux/spinlock.h>
33 #include <linux/timer.h>
34 #include <linux/delay.h>
35 #include <linux/poll.h>
36
37 #include "ccid.h"
38 #include "dccp.h"
39 #include "feat.h"
40
41 #define CREATE_TRACE_POINTS
42 #include "trace.h"
43
44 DEFINE_SNMP_STAT(struct dccp_mib, dccp_statistics) __read_mostly;
45
46 EXPORT_SYMBOL_GPL(dccp_statistics);
47
48 struct percpu_counter dccp_orphan_count;
49 EXPORT_SYMBOL_GPL(dccp_orphan_count);
50
51 struct inet_hashinfo dccp_hashinfo;
52 EXPORT_SYMBOL_GPL(dccp_hashinfo);
53
54 /* the maximum queue length for tx in packets. 0 is no limit */
55 int sysctl_dccp_tx_qlen __read_mostly = 5;
56
57 #ifdef CONFIG_IP_DCCP_DEBUG
58 static const char *dccp_state_name(const int state)
59 {
60 static const char *const dccp_state_names[] = {
61 [DCCP_OPEN] = "OPEN",
62 [DCCP_REQUESTING] = "REQUESTING",
63 [DCCP_PARTOPEN] = "PARTOPEN",
64 [DCCP_LISTEN] = "LISTEN",
65 [DCCP_RESPOND] = "RESPOND",
66 [DCCP_CLOSING] = "CLOSING",
67 [DCCP_ACTIVE_CLOSEREQ] = "CLOSEREQ",
68 [DCCP_PASSIVE_CLOSE] = "PASSIVE_CLOSE",
69 [DCCP_PASSIVE_CLOSEREQ] = "PASSIVE_CLOSEREQ",
70 [DCCP_TIME_WAIT] = "TIME_WAIT",
71 [DCCP_CLOSED] = "CLOSED",
72 };
73
74 if (state >= DCCP_MAX_STATES)
75 return "INVALID STATE!";
76 else
77 return dccp_state_names[state];
78 }
79 #endif
80
81 void dccp_set_state(struct sock *sk, const int state)
82 {
83 const int oldstate = sk->sk_state;
84
85 dccp_pr_debug("%s(%p) %s --> %s\n", dccp_role(sk), sk,
86 dccp_state_name(oldstate), dccp_state_name(state));
87 WARN_ON(state == oldstate);
88
89 switch (state) {
90 case DCCP_OPEN:
91 if (oldstate != DCCP_OPEN)
92 DCCP_INC_STATS(DCCP_MIB_CURRESTAB);
93 /* Client retransmits all Confirm options until entering OPEN */
94 if (oldstate == DCCP_PARTOPEN)
95 dccp_feat_list_purge(&dccp_sk(sk)->dccps_featneg);
96 break;
97
98 case DCCP_CLOSED:
99 if (oldstate == DCCP_OPEN || oldstate == DCCP_ACTIVE_CLOSEREQ ||
100 oldstate == DCCP_CLOSING)
101 DCCP_INC_STATS(DCCP_MIB_ESTABRESETS);
102
103 sk->sk_prot->unhash(sk);
104 if (inet_csk(sk)->icsk_bind_hash != NULL &&
105 !(sk->sk_userlocks & SOCK_BINDPORT_LOCK))
106 inet_put_port(sk);
107 /* fall through */
108 default:
109 if (oldstate == DCCP_OPEN)
110 DCCP_DEC_STATS(DCCP_MIB_CURRESTAB);
111 }
112
113 /* Change state AFTER socket is unhashed to avoid closed
114 * socket sitting in hash tables.
115 */
116 inet_sk_set_state(sk, state);
117 }
118
119 EXPORT_SYMBOL_GPL(dccp_set_state);
120
121 static void dccp_finish_passive_close(struct sock *sk)
122 {
123 switch (sk->sk_state) {
124 case DCCP_PASSIVE_CLOSE:
125 /* Node (client or server) has received Close packet. */
126 dccp_send_reset(sk, DCCP_RESET_CODE_CLOSED);
127 dccp_set_state(sk, DCCP_CLOSED);
128 break;
129 case DCCP_PASSIVE_CLOSEREQ:
130 /*
131 * Client received CloseReq. We set the `active' flag so that
132 * dccp_send_close() retransmits the Close as per RFC 4340, 8.3.
133 */
134 dccp_send_close(sk, 1);
135 dccp_set_state(sk, DCCP_CLOSING);
136 }
137 }
138
139 void dccp_done(struct sock *sk)
140 {
141 dccp_set_state(sk, DCCP_CLOSED);
142 dccp_clear_xmit_timers(sk);
143
144 sk->sk_shutdown = SHUTDOWN_MASK;
145
146 if (!sock_flag(sk, SOCK_DEAD))
147 sk->sk_state_change(sk);
148 else
149 inet_csk_destroy_sock(sk);
150 }
151
152 EXPORT_SYMBOL_GPL(dccp_done);
153
154 const char *dccp_packet_name(const int type)
155 {
156 static const char *const dccp_packet_names[] = {
157 [DCCP_PKT_REQUEST] = "REQUEST",
158 [DCCP_PKT_RESPONSE] = "RESPONSE",
159 [DCCP_PKT_DATA] = "DATA",
160 [DCCP_PKT_ACK] = "ACK",
161 [DCCP_PKT_DATAACK] = "DATAACK",
162 [DCCP_PKT_CLOSEREQ] = "CLOSEREQ",
163 [DCCP_PKT_CLOSE] = "CLOSE",
164 [DCCP_PKT_RESET] = "RESET",
165 [DCCP_PKT_SYNC] = "SYNC",
166 [DCCP_PKT_SYNCACK] = "SYNCACK",
167 };
168
169 if (type >= DCCP_NR_PKT_TYPES)
170 return "INVALID";
171 else
172 return dccp_packet_names[type];
173 }
174
175 EXPORT_SYMBOL_GPL(dccp_packet_name);
176
177 static void dccp_sk_destruct(struct sock *sk)
178 {
179 struct dccp_sock *dp = dccp_sk(sk);
180
181 ccid_hc_tx_delete(dp->dccps_hc_tx_ccid, sk);
182 dp->dccps_hc_tx_ccid = NULL;
183 inet_sock_destruct(sk);
184 }
185
186 int dccp_init_sock(struct sock *sk, const __u8 ctl_sock_initialized)
187 {
188 struct dccp_sock *dp = dccp_sk(sk);
189 struct inet_connection_sock *icsk = inet_csk(sk);
190
191 icsk->icsk_rto = DCCP_TIMEOUT_INIT;
192 icsk->icsk_syn_retries = sysctl_dccp_request_retries;
193 sk->sk_state = DCCP_CLOSED;
194 sk->sk_write_space = dccp_write_space;
195 sk->sk_destruct = dccp_sk_destruct;
196 icsk->icsk_sync_mss = dccp_sync_mss;
197 dp->dccps_mss_cache = 536;
198 dp->dccps_rate_last = jiffies;
199 dp->dccps_role = DCCP_ROLE_UNDEFINED;
200 dp->dccps_service = DCCP_SERVICE_CODE_IS_ABSENT;
201 dp->dccps_tx_qlen = sysctl_dccp_tx_qlen;
202
203 dccp_init_xmit_timers(sk);
204
205 INIT_LIST_HEAD(&dp->dccps_featneg);
206 /* control socket doesn't need feat nego */
207 if (likely(ctl_sock_initialized))
208 return dccp_feat_init(sk);
209 return 0;
210 }
211
212 EXPORT_SYMBOL_GPL(dccp_init_sock);
213
214 void dccp_destroy_sock(struct sock *sk)
215 {
216 struct dccp_sock *dp = dccp_sk(sk);
217
218 __skb_queue_purge(&sk->sk_write_queue);
219 if (sk->sk_send_head != NULL) {
220 kfree_skb(sk->sk_send_head);
221 sk->sk_send_head = NULL;
222 }
223
224 /* Clean up a referenced DCCP bind bucket. */
225 if (inet_csk(sk)->icsk_bind_hash != NULL)
226 inet_put_port(sk);
227
228 kfree(dp->dccps_service_list);
229 dp->dccps_service_list = NULL;
230
231 if (dp->dccps_hc_rx_ackvec != NULL) {
232 dccp_ackvec_free(dp->dccps_hc_rx_ackvec);
233 dp->dccps_hc_rx_ackvec = NULL;
234 }
235 ccid_hc_rx_delete(dp->dccps_hc_rx_ccid, sk);
236 dp->dccps_hc_rx_ccid = NULL;
237
238 /* clean up feature negotiation state */
239 dccp_feat_list_purge(&dp->dccps_featneg);
240 }
241
242 EXPORT_SYMBOL_GPL(dccp_destroy_sock);
243
244 static inline int dccp_listen_start(struct sock *sk, int backlog)
245 {
246 struct dccp_sock *dp = dccp_sk(sk);
247
248 dp->dccps_role = DCCP_ROLE_LISTEN;
249 /* do not start to listen if feature negotiation setup fails */
250 if (dccp_feat_finalise_settings(dp))
251 return -EPROTO;
252 return inet_csk_listen_start(sk, backlog);
253 }
254
255 static inline int dccp_need_reset(int state)
256 {
257 return state != DCCP_CLOSED && state != DCCP_LISTEN &&
258 state != DCCP_REQUESTING;
259 }
260
261 int dccp_disconnect(struct sock *sk, int flags)
262 {
263 struct inet_connection_sock *icsk = inet_csk(sk);
264 struct inet_sock *inet = inet_sk(sk);
265 struct dccp_sock *dp = dccp_sk(sk);
266 int err = 0;
267 const int old_state = sk->sk_state;
268
269 if (old_state != DCCP_CLOSED)
270 dccp_set_state(sk, DCCP_CLOSED);
271
272 /*
273 * This corresponds to the ABORT function of RFC793, sec. 3.8
274 * TCP uses a RST segment, DCCP a Reset packet with Code 2, "Aborted".
275 */
276 if (old_state == DCCP_LISTEN) {
277 inet_csk_listen_stop(sk);
278 } else if (dccp_need_reset(old_state)) {
279 dccp_send_reset(sk, DCCP_RESET_CODE_ABORTED);
280 sk->sk_err = ECONNRESET;
281 } else if (old_state == DCCP_REQUESTING)
282 sk->sk_err = ECONNRESET;
283
284 dccp_clear_xmit_timers(sk);
285 ccid_hc_rx_delete(dp->dccps_hc_rx_ccid, sk);
286 dp->dccps_hc_rx_ccid = NULL;
287
288 __skb_queue_purge(&sk->sk_receive_queue);
289 __skb_queue_purge(&sk->sk_write_queue);
290 if (sk->sk_send_head != NULL) {
291 __kfree_skb(sk->sk_send_head);
292 sk->sk_send_head = NULL;
293 }
294
295 inet->inet_dport = 0;
296
297 if (!(sk->sk_userlocks & SOCK_BINDADDR_LOCK))
298 inet_reset_saddr(sk);
299
300 sk->sk_shutdown = 0;
301 sock_reset_flag(sk, SOCK_DONE);
302
303 icsk->icsk_backoff = 0;
304 inet_csk_delack_init(sk);
305 __sk_dst_reset(sk);
306
307 WARN_ON(inet->inet_num && !icsk->icsk_bind_hash);
308
309 sk->sk_error_report(sk);
310 return err;
311 }
312
313 EXPORT_SYMBOL_GPL(dccp_disconnect);
314
315 /*
316 * Wait for a DCCP event.
317 *
318 * Note that we don't need to lock the socket, as the upper poll layers
319 * take care of normal races (between the test and the event) and we don't
320 * go look at any of the socket buffers directly.
321 */
322 __poll_t dccp_poll(struct file *file, struct socket *sock,
323 poll_table *wait)
324 {
325 __poll_t mask;
326 struct sock *sk = sock->sk;
327
328 sock_poll_wait(file, sk_sleep(sk), wait);
329 if (sk->sk_state == DCCP_LISTEN)
330 return inet_csk_listen_poll(sk);
331
332 /* Socket is not locked. We are protected from async events
333 by poll logic and correct handling of state changes
334 made by another threads is impossible in any case.
335 */
336
337 mask = 0;
338 if (sk->sk_err)
339 mask = EPOLLERR;
340
341 if (sk->sk_shutdown == SHUTDOWN_MASK || sk->sk_state == DCCP_CLOSED)
342 mask |= EPOLLHUP;
343 if (sk->sk_shutdown & RCV_SHUTDOWN)
344 mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
345
346 /* Connected? */
347 if ((1 << sk->sk_state) & ~(DCCPF_REQUESTING | DCCPF_RESPOND)) {
348 if (atomic_read(&sk->sk_rmem_alloc) > 0)
349 mask |= EPOLLIN | EPOLLRDNORM;
350
351 if (!(sk->sk_shutdown & SEND_SHUTDOWN)) {
352 if (sk_stream_is_writeable(sk)) {
353 mask |= EPOLLOUT | EPOLLWRNORM;
354 } else { /* send SIGIO later */
355 sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
356 set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
357
358 /* Race breaker. If space is freed after
359 * wspace test but before the flags are set,
360 * IO signal will be lost.
361 */
362 if (sk_stream_is_writeable(sk))
363 mask |= EPOLLOUT | EPOLLWRNORM;
364 }
365 }
366 }
367 return mask;
368 }
369
370 EXPORT_SYMBOL_GPL(dccp_poll);
371
372 int dccp_ioctl(struct sock *sk, int cmd, unsigned long arg)
373 {
374 int rc = -ENOTCONN;
375
376 lock_sock(sk);
377
378 if (sk->sk_state == DCCP_LISTEN)
379 goto out;
380
381 switch (cmd) {
382 case SIOCINQ: {
383 struct sk_buff *skb;
384 unsigned long amount = 0;
385
386 skb = skb_peek(&sk->sk_receive_queue);
387 if (skb != NULL) {
388 /*
389 * We will only return the amount of this packet since
390 * that is all that will be read.
391 */
392 amount = skb->len;
393 }
394 rc = put_user(amount, (int __user *)arg);
395 }
396 break;
397 default:
398 rc = -ENOIOCTLCMD;
399 break;
400 }
401 out:
402 release_sock(sk);
403 return rc;
404 }
405
406 EXPORT_SYMBOL_GPL(dccp_ioctl);
407
408 static int dccp_setsockopt_service(struct sock *sk, const __be32 service,
409 char __user *optval, unsigned int optlen)
410 {
411 struct dccp_sock *dp = dccp_sk(sk);
412 struct dccp_service_list *sl = NULL;
413
414 if (service == DCCP_SERVICE_INVALID_VALUE ||
415 optlen > DCCP_SERVICE_LIST_MAX_LEN * sizeof(u32))
416 return -EINVAL;
417
418 if (optlen > sizeof(service)) {
419 sl = kmalloc(optlen, GFP_KERNEL);
420 if (sl == NULL)
421 return -ENOMEM;
422
423 sl->dccpsl_nr = optlen / sizeof(u32) - 1;
424 if (copy_from_user(sl->dccpsl_list,
425 optval + sizeof(service),
426 optlen - sizeof(service)) ||
427 dccp_list_has_service(sl, DCCP_SERVICE_INVALID_VALUE)) {
428 kfree(sl);
429 return -EFAULT;
430 }
431 }
432
433 lock_sock(sk);
434 dp->dccps_service = service;
435
436 kfree(dp->dccps_service_list);
437
438 dp->dccps_service_list = sl;
439 release_sock(sk);
440 return 0;
441 }
442
443 static int dccp_setsockopt_cscov(struct sock *sk, int cscov, bool rx)
444 {
445 u8 *list, len;
446 int i, rc;
447
448 if (cscov < 0 || cscov > 15)
449 return -EINVAL;
450 /*
451 * Populate a list of permissible values, in the range cscov...15. This
452 * is necessary since feature negotiation of single values only works if
453 * both sides incidentally choose the same value. Since the list starts
454 * lowest-value first, negotiation will pick the smallest shared value.
455 */
456 if (cscov == 0)
457 return 0;
458 len = 16 - cscov;
459
460 list = kmalloc(len, GFP_KERNEL);
461 if (list == NULL)
462 return -ENOBUFS;
463
464 for (i = 0; i < len; i++)
465 list[i] = cscov++;
466
467 rc = dccp_feat_register_sp(sk, DCCPF_MIN_CSUM_COVER, rx, list, len);
468
469 if (rc == 0) {
470 if (rx)
471 dccp_sk(sk)->dccps_pcrlen = cscov;
472 else
473 dccp_sk(sk)->dccps_pcslen = cscov;
474 }
475 kfree(list);
476 return rc;
477 }
478
479 static int dccp_setsockopt_ccid(struct sock *sk, int type,
480 char __user *optval, unsigned int optlen)
481 {
482 u8 *val;
483 int rc = 0;
484
485 if (optlen < 1 || optlen > DCCP_FEAT_MAX_SP_VALS)
486 return -EINVAL;
487
488 val = memdup_user(optval, optlen);
489 if (IS_ERR(val))
490 return PTR_ERR(val);
491
492 lock_sock(sk);
493 if (type == DCCP_SOCKOPT_TX_CCID || type == DCCP_SOCKOPT_CCID)
494 rc = dccp_feat_register_sp(sk, DCCPF_CCID, 1, val, optlen);
495
496 if (!rc && (type == DCCP_SOCKOPT_RX_CCID || type == DCCP_SOCKOPT_CCID))
497 rc = dccp_feat_register_sp(sk, DCCPF_CCID, 0, val, optlen);
498 release_sock(sk);
499
500 kfree(val);
501 return rc;
502 }
503
504 static int do_dccp_setsockopt(struct sock *sk, int level, int optname,
505 char __user *optval, unsigned int optlen)
506 {
507 struct dccp_sock *dp = dccp_sk(sk);
508 int val, err = 0;
509
510 switch (optname) {
511 case DCCP_SOCKOPT_PACKET_SIZE:
512 DCCP_WARN("sockopt(PACKET_SIZE) is deprecated: fix your app\n");
513 return 0;
514 case DCCP_SOCKOPT_CHANGE_L:
515 case DCCP_SOCKOPT_CHANGE_R:
516 DCCP_WARN("sockopt(CHANGE_L/R) is deprecated: fix your app\n");
517 return 0;
518 case DCCP_SOCKOPT_CCID:
519 case DCCP_SOCKOPT_RX_CCID:
520 case DCCP_SOCKOPT_TX_CCID:
521 return dccp_setsockopt_ccid(sk, optname, optval, optlen);
522 }
523
524 if (optlen < (int)sizeof(int))
525 return -EINVAL;
526
527 if (get_user(val, (int __user *)optval))
528 return -EFAULT;
529
530 if (optname == DCCP_SOCKOPT_SERVICE)
531 return dccp_setsockopt_service(sk, val, optval, optlen);
532
533 lock_sock(sk);
534 switch (optname) {
535 case DCCP_SOCKOPT_SERVER_TIMEWAIT:
536 if (dp->dccps_role != DCCP_ROLE_SERVER)
537 err = -EOPNOTSUPP;
538 else
539 dp->dccps_server_timewait = (val != 0);
540 break;
541 case DCCP_SOCKOPT_SEND_CSCOV:
542 err = dccp_setsockopt_cscov(sk, val, false);
543 break;
544 case DCCP_SOCKOPT_RECV_CSCOV:
545 err = dccp_setsockopt_cscov(sk, val, true);
546 break;
547 case DCCP_SOCKOPT_QPOLICY_ID:
548 if (sk->sk_state != DCCP_CLOSED)
549 err = -EISCONN;
550 else if (val < 0 || val >= DCCPQ_POLICY_MAX)
551 err = -EINVAL;
552 else
553 dp->dccps_qpolicy = val;
554 break;
555 case DCCP_SOCKOPT_QPOLICY_TXQLEN:
556 if (val < 0)
557 err = -EINVAL;
558 else
559 dp->dccps_tx_qlen = val;
560 break;
561 default:
562 err = -ENOPROTOOPT;
563 break;
564 }
565 release_sock(sk);
566
567 return err;
568 }
569
570 int dccp_setsockopt(struct sock *sk, int level, int optname,
571 char __user *optval, unsigned int optlen)
572 {
573 if (level != SOL_DCCP)
574 return inet_csk(sk)->icsk_af_ops->setsockopt(sk, level,
575 optname, optval,
576 optlen);
577 return do_dccp_setsockopt(sk, level, optname, optval, optlen);
578 }
579
580 EXPORT_SYMBOL_GPL(dccp_setsockopt);
581
582 #ifdef CONFIG_COMPAT
583 int compat_dccp_setsockopt(struct sock *sk, int level, int optname,
584 char __user *optval, unsigned int optlen)
585 {
586 if (level != SOL_DCCP)
587 return inet_csk_compat_setsockopt(sk, level, optname,
588 optval, optlen);
589 return do_dccp_setsockopt(sk, level, optname, optval, optlen);
590 }
591
592 EXPORT_SYMBOL_GPL(compat_dccp_setsockopt);
593 #endif
594
595 static int dccp_getsockopt_service(struct sock *sk, int len,
596 __be32 __user *optval,
597 int __user *optlen)
598 {
599 const struct dccp_sock *dp = dccp_sk(sk);
600 const struct dccp_service_list *sl;
601 int err = -ENOENT, slen = 0, total_len = sizeof(u32);
602
603 lock_sock(sk);
604 if ((sl = dp->dccps_service_list) != NULL) {
605 slen = sl->dccpsl_nr * sizeof(u32);
606 total_len += slen;
607 }
608
609 err = -EINVAL;
610 if (total_len > len)
611 goto out;
612
613 err = 0;
614 if (put_user(total_len, optlen) ||
615 put_user(dp->dccps_service, optval) ||
616 (sl != NULL && copy_to_user(optval + 1, sl->dccpsl_list, slen)))
617 err = -EFAULT;
618 out:
619 release_sock(sk);
620 return err;
621 }
622
623 static int do_dccp_getsockopt(struct sock *sk, int level, int optname,
624 char __user *optval, int __user *optlen)
625 {
626 struct dccp_sock *dp;
627 int val, len;
628
629 if (get_user(len, optlen))
630 return -EFAULT;
631
632 if (len < (int)sizeof(int))
633 return -EINVAL;
634
635 dp = dccp_sk(sk);
636
637 switch (optname) {
638 case DCCP_SOCKOPT_PACKET_SIZE:
639 DCCP_WARN("sockopt(PACKET_SIZE) is deprecated: fix your app\n");
640 return 0;
641 case DCCP_SOCKOPT_SERVICE:
642 return dccp_getsockopt_service(sk, len,
643 (__be32 __user *)optval, optlen);
644 case DCCP_SOCKOPT_GET_CUR_MPS:
645 val = dp->dccps_mss_cache;
646 break;
647 case DCCP_SOCKOPT_AVAILABLE_CCIDS:
648 return ccid_getsockopt_builtin_ccids(sk, len, optval, optlen);
649 case DCCP_SOCKOPT_TX_CCID:
650 val = ccid_get_current_tx_ccid(dp);
651 if (val < 0)
652 return -ENOPROTOOPT;
653 break;
654 case DCCP_SOCKOPT_RX_CCID:
655 val = ccid_get_current_rx_ccid(dp);
656 if (val < 0)
657 return -ENOPROTOOPT;
658 break;
659 case DCCP_SOCKOPT_SERVER_TIMEWAIT:
660 val = dp->dccps_server_timewait;
661 break;
662 case DCCP_SOCKOPT_SEND_CSCOV:
663 val = dp->dccps_pcslen;
664 break;
665 case DCCP_SOCKOPT_RECV_CSCOV:
666 val = dp->dccps_pcrlen;
667 break;
668 case DCCP_SOCKOPT_QPOLICY_ID:
669 val = dp->dccps_qpolicy;
670 break;
671 case DCCP_SOCKOPT_QPOLICY_TXQLEN:
672 val = dp->dccps_tx_qlen;
673 break;
674 case 128 ... 191:
675 return ccid_hc_rx_getsockopt(dp->dccps_hc_rx_ccid, sk, optname,
676 len, (u32 __user *)optval, optlen);
677 case 192 ... 255:
678 return ccid_hc_tx_getsockopt(dp->dccps_hc_tx_ccid, sk, optname,
679 len, (u32 __user *)optval, optlen);
680 default:
681 return -ENOPROTOOPT;
682 }
683
684 len = sizeof(val);
685 if (put_user(len, optlen) || copy_to_user(optval, &val, len))
686 return -EFAULT;
687
688 return 0;
689 }
690
691 int dccp_getsockopt(struct sock *sk, int level, int optname,
692 char __user *optval, int __user *optlen)
693 {
694 if (level != SOL_DCCP)
695 return inet_csk(sk)->icsk_af_ops->getsockopt(sk, level,
696 optname, optval,
697 optlen);
698 return do_dccp_getsockopt(sk, level, optname, optval, optlen);
699 }
700
701 EXPORT_SYMBOL_GPL(dccp_getsockopt);
702
703 #ifdef CONFIG_COMPAT
704 int compat_dccp_getsockopt(struct sock *sk, int level, int optname,
705 char __user *optval, int __user *optlen)
706 {
707 if (level != SOL_DCCP)
708 return inet_csk_compat_getsockopt(sk, level, optname,
709 optval, optlen);
710 return do_dccp_getsockopt(sk, level, optname, optval, optlen);
711 }
712
713 EXPORT_SYMBOL_GPL(compat_dccp_getsockopt);
714 #endif
715
716 static int dccp_msghdr_parse(struct msghdr *msg, struct sk_buff *skb)
717 {
718 struct cmsghdr *cmsg;
719
720 /*
721 * Assign an (opaque) qpolicy priority value to skb->priority.
722 *
723 * We are overloading this skb field for use with the qpolicy subystem.
724 * The skb->priority is normally used for the SO_PRIORITY option, which
725 * is initialised from sk_priority. Since the assignment of sk_priority
726 * to skb->priority happens later (on layer 3), we overload this field
727 * for use with queueing priorities as long as the skb is on layer 4.
728 * The default priority value (if nothing is set) is 0.
729 */
730 skb->priority = 0;
731
732 for_each_cmsghdr(cmsg, msg) {
733 if (!CMSG_OK(msg, cmsg))
734 return -EINVAL;
735
736 if (cmsg->cmsg_level != SOL_DCCP)
737 continue;
738
739 if (cmsg->cmsg_type <= DCCP_SCM_QPOLICY_MAX &&
740 !dccp_qpolicy_param_ok(skb->sk, cmsg->cmsg_type))
741 return -EINVAL;
742
743 switch (cmsg->cmsg_type) {
744 case DCCP_SCM_PRIORITY:
745 if (cmsg->cmsg_len != CMSG_LEN(sizeof(__u32)))
746 return -EINVAL;
747 skb->priority = *(__u32 *)CMSG_DATA(cmsg);
748 break;
749 default:
750 return -EINVAL;
751 }
752 }
753 return 0;
754 }
755
756 int dccp_sendmsg(struct sock *sk, struct msghdr *msg, size_t len)
757 {
758 const struct dccp_sock *dp = dccp_sk(sk);
759 const int flags = msg->msg_flags;
760 const int noblock = flags & MSG_DONTWAIT;
761 struct sk_buff *skb;
762 int rc, size;
763 long timeo;
764
765 trace_dccp_probe(sk, len);
766
767 if (len > dp->dccps_mss_cache)
768 return -EMSGSIZE;
769
770 lock_sock(sk);
771
772 if (dccp_qpolicy_full(sk)) {
773 rc = -EAGAIN;
774 goto out_release;
775 }
776
777 timeo = sock_sndtimeo(sk, noblock);
778
779 /*
780 * We have to use sk_stream_wait_connect here to set sk_write_pending,
781 * so that the trick in dccp_rcv_request_sent_state_process.
782 */
783 /* Wait for a connection to finish. */
784 if ((1 << sk->sk_state) & ~(DCCPF_OPEN | DCCPF_PARTOPEN))
785 if ((rc = sk_stream_wait_connect(sk, &timeo)) != 0)
786 goto out_release;
787
788 size = sk->sk_prot->max_header + len;
789 release_sock(sk);
790 skb = sock_alloc_send_skb(sk, size, noblock, &rc);
791 lock_sock(sk);
792 if (skb == NULL)
793 goto out_release;
794
795 if (sk->sk_state == DCCP_CLOSED) {
796 rc = -ENOTCONN;
797 goto out_discard;
798 }
799
800 skb_reserve(skb, sk->sk_prot->max_header);
801 rc = memcpy_from_msg(skb_put(skb, len), msg, len);
802 if (rc != 0)
803 goto out_discard;
804
805 rc = dccp_msghdr_parse(msg, skb);
806 if (rc != 0)
807 goto out_discard;
808
809 dccp_qpolicy_push(sk, skb);
810 /*
811 * The xmit_timer is set if the TX CCID is rate-based and will expire
812 * when congestion control permits to release further packets into the
813 * network. Window-based CCIDs do not use this timer.
814 */
815 if (!timer_pending(&dp->dccps_xmit_timer))
816 dccp_write_xmit(sk);
817 out_release:
818 release_sock(sk);
819 return rc ? : len;
820 out_discard:
821 kfree_skb(skb);
822 goto out_release;
823 }
824
825 EXPORT_SYMBOL_GPL(dccp_sendmsg);
826
827 int dccp_recvmsg(struct sock *sk, struct msghdr *msg, size_t len, int nonblock,
828 int flags, int *addr_len)
829 {
830 const struct dccp_hdr *dh;
831 long timeo;
832
833 lock_sock(sk);
834
835 if (sk->sk_state == DCCP_LISTEN) {
836 len = -ENOTCONN;
837 goto out;
838 }
839
840 timeo = sock_rcvtimeo(sk, nonblock);
841
842 do {
843 struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
844
845 if (skb == NULL)
846 goto verify_sock_status;
847
848 dh = dccp_hdr(skb);
849
850 switch (dh->dccph_type) {
851 case DCCP_PKT_DATA:
852 case DCCP_PKT_DATAACK:
853 goto found_ok_skb;
854
855 case DCCP_PKT_CLOSE:
856 case DCCP_PKT_CLOSEREQ:
857 if (!(flags & MSG_PEEK))
858 dccp_finish_passive_close(sk);
859 /* fall through */
860 case DCCP_PKT_RESET:
861 dccp_pr_debug("found fin (%s) ok!\n",
862 dccp_packet_name(dh->dccph_type));
863 len = 0;
864 goto found_fin_ok;
865 default:
866 dccp_pr_debug("packet_type=%s\n",
867 dccp_packet_name(dh->dccph_type));
868 sk_eat_skb(sk, skb);
869 }
870 verify_sock_status:
871 if (sock_flag(sk, SOCK_DONE)) {
872 len = 0;
873 break;
874 }
875
876 if (sk->sk_err) {
877 len = sock_error(sk);
878 break;
879 }
880
881 if (sk->sk_shutdown & RCV_SHUTDOWN) {
882 len = 0;
883 break;
884 }
885
886 if (sk->sk_state == DCCP_CLOSED) {
887 if (!sock_flag(sk, SOCK_DONE)) {
888 /* This occurs when user tries to read
889 * from never connected socket.
890 */
891 len = -ENOTCONN;
892 break;
893 }
894 len = 0;
895 break;
896 }
897
898 if (!timeo) {
899 len = -EAGAIN;
900 break;
901 }
902
903 if (signal_pending(current)) {
904 len = sock_intr_errno(timeo);
905 break;
906 }
907
908 sk_wait_data(sk, &timeo, NULL);
909 continue;
910 found_ok_skb:
911 if (len > skb->len)
912 len = skb->len;
913 else if (len < skb->len)
914 msg->msg_flags |= MSG_TRUNC;
915
916 if (skb_copy_datagram_msg(skb, 0, msg, len)) {
917 /* Exception. Bailout! */
918 len = -EFAULT;
919 break;
920 }
921 if (flags & MSG_TRUNC)
922 len = skb->len;
923 found_fin_ok:
924 if (!(flags & MSG_PEEK))
925 sk_eat_skb(sk, skb);
926 break;
927 } while (1);
928 out:
929 release_sock(sk);
930 return len;
931 }
932
933 EXPORT_SYMBOL_GPL(dccp_recvmsg);
934
935 int inet_dccp_listen(struct socket *sock, int backlog)
936 {
937 struct sock *sk = sock->sk;
938 unsigned char old_state;
939 int err;
940
941 lock_sock(sk);
942
943 err = -EINVAL;
944 if (sock->state != SS_UNCONNECTED || sock->type != SOCK_DCCP)
945 goto out;
946
947 old_state = sk->sk_state;
948 if (!((1 << old_state) & (DCCPF_CLOSED | DCCPF_LISTEN)))
949 goto out;
950
951 /* Really, if the socket is already in listen state
952 * we can only allow the backlog to be adjusted.
953 */
954 if (old_state != DCCP_LISTEN) {
955 /*
956 * FIXME: here it probably should be sk->sk_prot->listen_start
957 * see tcp_listen_start
958 */
959 err = dccp_listen_start(sk, backlog);
960 if (err)
961 goto out;
962 }
963 sk->sk_max_ack_backlog = backlog;
964 err = 0;
965
966 out:
967 release_sock(sk);
968 return err;
969 }
970
971 EXPORT_SYMBOL_GPL(inet_dccp_listen);
972
973 static void dccp_terminate_connection(struct sock *sk)
974 {
975 u8 next_state = DCCP_CLOSED;
976
977 switch (sk->sk_state) {
978 case DCCP_PASSIVE_CLOSE:
979 case DCCP_PASSIVE_CLOSEREQ:
980 dccp_finish_passive_close(sk);
981 break;
982 case DCCP_PARTOPEN:
983 dccp_pr_debug("Stop PARTOPEN timer (%p)\n", sk);
984 inet_csk_clear_xmit_timer(sk, ICSK_TIME_DACK);
985 /* fall through */
986 case DCCP_OPEN:
987 dccp_send_close(sk, 1);
988
989 if (dccp_sk(sk)->dccps_role == DCCP_ROLE_SERVER &&
990 !dccp_sk(sk)->dccps_server_timewait)
991 next_state = DCCP_ACTIVE_CLOSEREQ;
992 else
993 next_state = DCCP_CLOSING;
994 /* fall through */
995 default:
996 dccp_set_state(sk, next_state);
997 }
998 }
999
1000 void dccp_close(struct sock *sk, long timeout)
1001 {
1002 struct dccp_sock *dp = dccp_sk(sk);
1003 struct sk_buff *skb;
1004 u32 data_was_unread = 0;
1005 int state;
1006
1007 lock_sock(sk);
1008
1009 sk->sk_shutdown = SHUTDOWN_MASK;
1010
1011 if (sk->sk_state == DCCP_LISTEN) {
1012 dccp_set_state(sk, DCCP_CLOSED);
1013
1014 /* Special case. */
1015 inet_csk_listen_stop(sk);
1016
1017 goto adjudge_to_death;
1018 }
1019
1020 sk_stop_timer(sk, &dp->dccps_xmit_timer);
1021
1022 /*
1023 * We need to flush the recv. buffs. We do this only on the
1024 * descriptor close, not protocol-sourced closes, because the
1025 *reader process may not have drained the data yet!
1026 */
1027 while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
1028 data_was_unread += skb->len;
1029 __kfree_skb(skb);
1030 }
1031
1032 /* If socket has been already reset kill it. */
1033 if (sk->sk_state == DCCP_CLOSED)
1034 goto adjudge_to_death;
1035
1036 if (data_was_unread) {
1037 /* Unread data was tossed, send an appropriate Reset Code */
1038 DCCP_WARN("ABORT with %u bytes unread\n", data_was_unread);
1039 dccp_send_reset(sk, DCCP_RESET_CODE_ABORTED);
1040 dccp_set_state(sk, DCCP_CLOSED);
1041 } else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
1042 /* Check zero linger _after_ checking for unread data. */
1043 sk->sk_prot->disconnect(sk, 0);
1044 } else if (sk->sk_state != DCCP_CLOSED) {
1045 /*
1046 * Normal connection termination. May need to wait if there are
1047 * still packets in the TX queue that are delayed by the CCID.
1048 */
1049 dccp_flush_write_queue(sk, &timeout);
1050 dccp_terminate_connection(sk);
1051 }
1052
1053 /*
1054 * Flush write queue. This may be necessary in several cases:
1055 * - we have been closed by the peer but still have application data;
1056 * - abortive termination (unread data or zero linger time),
1057 * - normal termination but queue could not be flushed within time limit
1058 */
1059 __skb_queue_purge(&sk->sk_write_queue);
1060
1061 sk_stream_wait_close(sk, timeout);
1062
1063 adjudge_to_death:
1064 state = sk->sk_state;
1065 sock_hold(sk);
1066 sock_orphan(sk);
1067
1068 /*
1069 * It is the last release_sock in its life. It will remove backlog.
1070 */
1071 release_sock(sk);
1072 /*
1073 * Now socket is owned by kernel and we acquire BH lock
1074 * to finish close. No need to check for user refs.
1075 */
1076 local_bh_disable();
1077 bh_lock_sock(sk);
1078 WARN_ON(sock_owned_by_user(sk));
1079
1080 percpu_counter_inc(sk->sk_prot->orphan_count);
1081
1082 /* Have we already been destroyed by a softirq or backlog? */
1083 if (state != DCCP_CLOSED && sk->sk_state == DCCP_CLOSED)
1084 goto out;
1085
1086 if (sk->sk_state == DCCP_CLOSED)
1087 inet_csk_destroy_sock(sk);
1088
1089 /* Otherwise, socket is reprieved until protocol close. */
1090
1091 out:
1092 bh_unlock_sock(sk);
1093 local_bh_enable();
1094 sock_put(sk);
1095 }
1096
1097 EXPORT_SYMBOL_GPL(dccp_close);
1098
1099 void dccp_shutdown(struct sock *sk, int how)
1100 {
1101 dccp_pr_debug("called shutdown(%x)\n", how);
1102 }
1103
1104 EXPORT_SYMBOL_GPL(dccp_shutdown);
1105
1106 static inline int __init dccp_mib_init(void)
1107 {
1108 dccp_statistics = alloc_percpu(struct dccp_mib);
1109 if (!dccp_statistics)
1110 return -ENOMEM;
1111 return 0;
1112 }
1113
1114 static inline void dccp_mib_exit(void)
1115 {
1116 free_percpu(dccp_statistics);
1117 }
1118
1119 static int thash_entries;
1120 module_param(thash_entries, int, 0444);
1121 MODULE_PARM_DESC(thash_entries, "Number of ehash buckets");
1122
1123 #ifdef CONFIG_IP_DCCP_DEBUG
1124 bool dccp_debug;
1125 module_param(dccp_debug, bool, 0644);
1126 MODULE_PARM_DESC(dccp_debug, "Enable debug messages");
1127
1128 EXPORT_SYMBOL_GPL(dccp_debug);
1129 #endif
1130
1131 static int __init dccp_init(void)
1132 {
1133 unsigned long goal;
1134 int ehash_order, bhash_order, i;
1135 int rc;
1136
1137 BUILD_BUG_ON(sizeof(struct dccp_skb_cb) >
1138 FIELD_SIZEOF(struct sk_buff, cb));
1139 rc = percpu_counter_init(&dccp_orphan_count, 0, GFP_KERNEL);
1140 if (rc)
1141 goto out_fail;
1142 rc = -ENOBUFS;
1143 inet_hashinfo_init(&dccp_hashinfo);
1144 dccp_hashinfo.bind_bucket_cachep =
1145 kmem_cache_create("dccp_bind_bucket",
1146 sizeof(struct inet_bind_bucket), 0,
1147 SLAB_HWCACHE_ALIGN, NULL);
1148 if (!dccp_hashinfo.bind_bucket_cachep)
1149 goto out_free_percpu;
1150
1151 /*
1152 * Size and allocate the main established and bind bucket
1153 * hash tables.
1154 *
1155 * The methodology is similar to that of the buffer cache.
1156 */
1157 if (totalram_pages >= (128 * 1024))
1158 goal = totalram_pages >> (21 - PAGE_SHIFT);
1159 else
1160 goal = totalram_pages >> (23 - PAGE_SHIFT);
1161
1162 if (thash_entries)
1163 goal = (thash_entries *
1164 sizeof(struct inet_ehash_bucket)) >> PAGE_SHIFT;
1165 for (ehash_order = 0; (1UL << ehash_order) < goal; ehash_order++)
1166 ;
1167 do {
1168 unsigned long hash_size = (1UL << ehash_order) * PAGE_SIZE /
1169 sizeof(struct inet_ehash_bucket);
1170
1171 while (hash_size & (hash_size - 1))
1172 hash_size--;
1173 dccp_hashinfo.ehash_mask = hash_size - 1;
1174 dccp_hashinfo.ehash = (struct inet_ehash_bucket *)
1175 __get_free_pages(GFP_ATOMIC|__GFP_NOWARN, ehash_order);
1176 } while (!dccp_hashinfo.ehash && --ehash_order > 0);
1177
1178 if (!dccp_hashinfo.ehash) {
1179 DCCP_CRIT("Failed to allocate DCCP established hash table");
1180 goto out_free_bind_bucket_cachep;
1181 }
1182
1183 for (i = 0; i <= dccp_hashinfo.ehash_mask; i++)
1184 INIT_HLIST_NULLS_HEAD(&dccp_hashinfo.ehash[i].chain, i);
1185
1186 if (inet_ehash_locks_alloc(&dccp_hashinfo))
1187 goto out_free_dccp_ehash;
1188
1189 bhash_order = ehash_order;
1190
1191 do {
1192 dccp_hashinfo.bhash_size = (1UL << bhash_order) * PAGE_SIZE /
1193 sizeof(struct inet_bind_hashbucket);
1194 if ((dccp_hashinfo.bhash_size > (64 * 1024)) &&
1195 bhash_order > 0)
1196 continue;
1197 dccp_hashinfo.bhash = (struct inet_bind_hashbucket *)
1198 __get_free_pages(GFP_ATOMIC|__GFP_NOWARN, bhash_order);
1199 } while (!dccp_hashinfo.bhash && --bhash_order >= 0);
1200
1201 if (!dccp_hashinfo.bhash) {
1202 DCCP_CRIT("Failed to allocate DCCP bind hash table");
1203 goto out_free_dccp_locks;
1204 }
1205
1206 for (i = 0; i < dccp_hashinfo.bhash_size; i++) {
1207 spin_lock_init(&dccp_hashinfo.bhash[i].lock);
1208 INIT_HLIST_HEAD(&dccp_hashinfo.bhash[i].chain);
1209 }
1210
1211 rc = dccp_mib_init();
1212 if (rc)
1213 goto out_free_dccp_bhash;
1214
1215 rc = dccp_ackvec_init();
1216 if (rc)
1217 goto out_free_dccp_mib;
1218
1219 rc = dccp_sysctl_init();
1220 if (rc)
1221 goto out_ackvec_exit;
1222
1223 rc = ccid_initialize_builtins();
1224 if (rc)
1225 goto out_sysctl_exit;
1226
1227 dccp_timestamping_init();
1228
1229 return 0;
1230
1231 out_sysctl_exit:
1232 dccp_sysctl_exit();
1233 out_ackvec_exit:
1234 dccp_ackvec_exit();
1235 out_free_dccp_mib:
1236 dccp_mib_exit();
1237 out_free_dccp_bhash:
1238 free_pages((unsigned long)dccp_hashinfo.bhash, bhash_order);
1239 out_free_dccp_locks:
1240 inet_ehash_locks_free(&dccp_hashinfo);
1241 out_free_dccp_ehash:
1242 free_pages((unsigned long)dccp_hashinfo.ehash, ehash_order);
1243 out_free_bind_bucket_cachep:
1244 kmem_cache_destroy(dccp_hashinfo.bind_bucket_cachep);
1245 out_free_percpu:
1246 percpu_counter_destroy(&dccp_orphan_count);
1247 out_fail:
1248 dccp_hashinfo.bhash = NULL;
1249 dccp_hashinfo.ehash = NULL;
1250 dccp_hashinfo.bind_bucket_cachep = NULL;
1251 return rc;
1252 }
1253
1254 static void __exit dccp_fini(void)
1255 {
1256 ccid_cleanup_builtins();
1257 dccp_mib_exit();
1258 free_pages((unsigned long)dccp_hashinfo.bhash,
1259 get_order(dccp_hashinfo.bhash_size *
1260 sizeof(struct inet_bind_hashbucket)));
1261 free_pages((unsigned long)dccp_hashinfo.ehash,
1262 get_order((dccp_hashinfo.ehash_mask + 1) *
1263 sizeof(struct inet_ehash_bucket)));
1264 inet_ehash_locks_free(&dccp_hashinfo);
1265 kmem_cache_destroy(dccp_hashinfo.bind_bucket_cachep);
1266 dccp_ackvec_exit();
1267 dccp_sysctl_exit();
1268 percpu_counter_destroy(&dccp_orphan_count);
1269 }
1270
1271 module_init(dccp_init);
1272 module_exit(dccp_fini);
1273
1274 MODULE_LICENSE("GPL");
1275 MODULE_AUTHOR("Arnaldo Carvalho de Melo <acme@conectiva.com.br>");
1276 MODULE_DESCRIPTION("DCCP - Datagram Congestion Controlled Protocol");
Linux with added FPC-III support