1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21#include <linux/types.h>
22#include <linux/string.h>
23#include <linux/slab.h>
24#include <linux/module.h>
25#include <linux/capability.h>
26#include <linux/pagemap.h>
27#include <linux/errno.h>
28#include <linux/socket.h>
29#include <linux/in.h>
30#include <linux/net.h>
31#include <linux/mm.h>
32#include <linux/un.h>
33#include <linux/udp.h>
34#include <linux/tcp.h>
35#include <linux/sunrpc/clnt.h>
36#include <linux/sunrpc/addr.h>
37#include <linux/sunrpc/sched.h>
38#include <linux/sunrpc/svcsock.h>
39#include <linux/sunrpc/xprtsock.h>
40#include <linux/file.h>
41#ifdef CONFIG_SUNRPC_BACKCHANNEL
42#include <linux/sunrpc/bc_xprt.h>
43#endif
44
45#include <net/sock.h>
46#include <net/checksum.h>
47#include <net/udp.h>
48#include <net/tcp.h>
49
50#include <trace/events/sunrpc.h>
51
52#include "sunrpc.h"
53
54static void xs_close(struct rpc_xprt *xprt);
55
56
57
58
59static unsigned int xprt_udp_slot_table_entries = RPC_DEF_SLOT_TABLE;
60static unsigned int xprt_tcp_slot_table_entries = RPC_MIN_SLOT_TABLE;
61static unsigned int xprt_max_tcp_slot_table_entries = RPC_MAX_SLOT_TABLE;
62
63static unsigned int xprt_min_resvport = RPC_DEF_MIN_RESVPORT;
64static unsigned int xprt_max_resvport = RPC_DEF_MAX_RESVPORT;
65
66#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
67
68#define XS_TCP_LINGER_TO (15U * HZ)
69static unsigned int xs_tcp_fin_timeout __read_mostly = XS_TCP_LINGER_TO;
70
71
72
73
74
75
76
77
78
79
80static unsigned int min_slot_table_size = RPC_MIN_SLOT_TABLE;
81static unsigned int max_slot_table_size = RPC_MAX_SLOT_TABLE;
82static unsigned int max_tcp_slot_table_limit = RPC_MAX_SLOT_TABLE_LIMIT;
83static unsigned int xprt_min_resvport_limit = RPC_MIN_RESVPORT;
84static unsigned int xprt_max_resvport_limit = RPC_MAX_RESVPORT;
85
86static struct ctl_table_header *sunrpc_table_header;
87
88
89
90
91
92static struct ctl_table xs_tunables_table[] = {
93 {
94 .procname = "udp_slot_table_entries",
95 .data = &xprt_udp_slot_table_entries,
96 .maxlen = sizeof(unsigned int),
97 .mode = 0644,
98 .proc_handler = proc_dointvec_minmax,
99 .extra1 = &min_slot_table_size,
100 .extra2 = &max_slot_table_size
101 },
102 {
103 .procname = "tcp_slot_table_entries",
104 .data = &xprt_tcp_slot_table_entries,
105 .maxlen = sizeof(unsigned int),
106 .mode = 0644,
107 .proc_handler = proc_dointvec_minmax,
108 .extra1 = &min_slot_table_size,
109 .extra2 = &max_slot_table_size
110 },
111 {
112 .procname = "tcp_max_slot_table_entries",
113 .data = &xprt_max_tcp_slot_table_entries,
114 .maxlen = sizeof(unsigned int),
115 .mode = 0644,
116 .proc_handler = proc_dointvec_minmax,
117 .extra1 = &min_slot_table_size,
118 .extra2 = &max_tcp_slot_table_limit
119 },
120 {
121 .procname = "min_resvport",
122 .data = &xprt_min_resvport,
123 .maxlen = sizeof(unsigned int),
124 .mode = 0644,
125 .proc_handler = proc_dointvec_minmax,
126 .extra1 = &xprt_min_resvport_limit,
127 .extra2 = &xprt_max_resvport
128 },
129 {
130 .procname = "max_resvport",
131 .data = &xprt_max_resvport,
132 .maxlen = sizeof(unsigned int),
133 .mode = 0644,
134 .proc_handler = proc_dointvec_minmax,
135 .extra1 = &xprt_min_resvport,
136 .extra2 = &xprt_max_resvport_limit
137 },
138 {
139 .procname = "tcp_fin_timeout",
140 .data = &xs_tcp_fin_timeout,
141 .maxlen = sizeof(xs_tcp_fin_timeout),
142 .mode = 0644,
143 .proc_handler = proc_dointvec_jiffies,
144 },
145 { },
146};
147
148static struct ctl_table sunrpc_table[] = {
149 {
150 .procname = "sunrpc",
151 .mode = 0555,
152 .child = xs_tunables_table
153 },
154 { },
155};
156
157#endif
158
159
160
161
162#define XS_BIND_TO (60U * HZ)
163
164
165
166
167
168#define XS_UDP_REEST_TO (2U * HZ)
169
170
171
172
173
174
175
176
177
178
179#define XS_TCP_INIT_REEST_TO (3U * HZ)
180
181
182
183
184
185
186#define XS_IDLE_DISC_TO (5U * 60 * HZ)
187
188#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
189# undef RPC_DEBUG_DATA
190# define RPCDBG_FACILITY RPCDBG_TRANS
191#endif
192
193#ifdef RPC_DEBUG_DATA
194static void xs_pktdump(char *msg, u32 *packet, unsigned int count)
195{
196 u8 *buf = (u8 *) packet;
197 int j;
198
199 dprintk("RPC: %s\n", msg);
200 for (j = 0; j < count && j < 128; j += 4) {
201 if (!(j & 31)) {
202 if (j)
203 dprintk("\n");
204 dprintk("0x%04x ", j);
205 }
206 dprintk("%02x%02x%02x%02x ",
207 buf[j], buf[j+1], buf[j+2], buf[j+3]);
208 }
209 dprintk("\n");
210}
211#else
212static inline void xs_pktdump(char *msg, u32 *packet, unsigned int count)
213{
214
215}
216#endif
217
218static inline struct rpc_xprt *xprt_from_sock(struct sock *sk)
219{
220 return (struct rpc_xprt *) sk->sk_user_data;
221}
222
223static inline struct sockaddr *xs_addr(struct rpc_xprt *xprt)
224{
225 return (struct sockaddr *) &xprt->addr;
226}
227
228static inline struct sockaddr_un *xs_addr_un(struct rpc_xprt *xprt)
229{
230 return (struct sockaddr_un *) &xprt->addr;
231}
232
233static inline struct sockaddr_in *xs_addr_in(struct rpc_xprt *xprt)
234{
235 return (struct sockaddr_in *) &xprt->addr;
236}
237
238static inline struct sockaddr_in6 *xs_addr_in6(struct rpc_xprt *xprt)
239{
240 return (struct sockaddr_in6 *) &xprt->addr;
241}
242
243static void xs_format_common_peer_addresses(struct rpc_xprt *xprt)
244{
245 struct sockaddr *sap = xs_addr(xprt);
246 struct sockaddr_in6 *sin6;
247 struct sockaddr_in *sin;
248 struct sockaddr_un *sun;
249 char buf[128];
250
251 switch (sap->sa_family) {
252 case AF_LOCAL:
253 sun = xs_addr_un(xprt);
254 strlcpy(buf, sun->sun_path, sizeof(buf));
255 xprt->address_strings[RPC_DISPLAY_ADDR] =
256 kstrdup(buf, GFP_KERNEL);
257 break;
258 case AF_INET:
259 (void)rpc_ntop(sap, buf, sizeof(buf));
260 xprt->address_strings[RPC_DISPLAY_ADDR] =
261 kstrdup(buf, GFP_KERNEL);
262 sin = xs_addr_in(xprt);
263 snprintf(buf, sizeof(buf), "%08x", ntohl(sin->sin_addr.s_addr));
264 break;
265 case AF_INET6:
266 (void)rpc_ntop(sap, buf, sizeof(buf));
267 xprt->address_strings[RPC_DISPLAY_ADDR] =
268 kstrdup(buf, GFP_KERNEL);
269 sin6 = xs_addr_in6(xprt);
270 snprintf(buf, sizeof(buf), "%pi6", &sin6->sin6_addr);
271 break;
272 default:
273 BUG();
274 }
275
276 xprt->address_strings[RPC_DISPLAY_HEX_ADDR] = kstrdup(buf, GFP_KERNEL);
277}
278
279static void xs_format_common_peer_ports(struct rpc_xprt *xprt)
280{
281 struct sockaddr *sap = xs_addr(xprt);
282 char buf[128];
283
284 snprintf(buf, sizeof(buf), "%u", rpc_get_port(sap));
285 xprt->address_strings[RPC_DISPLAY_PORT] = kstrdup(buf, GFP_KERNEL);
286
287 snprintf(buf, sizeof(buf), "%4hx", rpc_get_port(sap));
288 xprt->address_strings[RPC_DISPLAY_HEX_PORT] = kstrdup(buf, GFP_KERNEL);
289}
290
291static void xs_format_peer_addresses(struct rpc_xprt *xprt,
292 const char *protocol,
293 const char *netid)
294{
295 xprt->address_strings[RPC_DISPLAY_PROTO] = protocol;
296 xprt->address_strings[RPC_DISPLAY_NETID] = netid;
297 xs_format_common_peer_addresses(xprt);
298 xs_format_common_peer_ports(xprt);
299}
300
301static void xs_update_peer_port(struct rpc_xprt *xprt)
302{
303 kfree(xprt->address_strings[RPC_DISPLAY_HEX_PORT]);
304 kfree(xprt->address_strings[RPC_DISPLAY_PORT]);
305
306 xs_format_common_peer_ports(xprt);
307}
308
309static void xs_free_peer_addresses(struct rpc_xprt *xprt)
310{
311 unsigned int i;
312
313 for (i = 0; i < RPC_DISPLAY_MAX; i++)
314 switch (i) {
315 case RPC_DISPLAY_PROTO:
316 case RPC_DISPLAY_NETID:
317 continue;
318 default:
319 kfree(xprt->address_strings[i]);
320 }
321}
322
323#define XS_SENDMSG_FLAGS (MSG_DONTWAIT | MSG_NOSIGNAL)
324
325static int xs_send_kvec(struct socket *sock, struct sockaddr *addr, int addrlen, struct kvec *vec, unsigned int base, int more)
326{
327 struct msghdr msg = {
328 .msg_name = addr,
329 .msg_namelen = addrlen,
330 .msg_flags = XS_SENDMSG_FLAGS | (more ? MSG_MORE : 0),
331 };
332 struct kvec iov = {
333 .iov_base = vec->iov_base + base,
334 .iov_len = vec->iov_len - base,
335 };
336
337 if (iov.iov_len != 0)
338 return kernel_sendmsg(sock, &msg, &iov, 1, iov.iov_len);
339 return kernel_sendmsg(sock, &msg, NULL, 0, 0);
340}
341
342static int xs_send_pagedata(struct socket *sock, struct xdr_buf *xdr, unsigned int base, int more, bool zerocopy, int *sent_p)
343{
344 ssize_t (*do_sendpage)(struct socket *sock, struct page *page,
345 int offset, size_t size, int flags);
346 struct page **ppage;
347 unsigned int remainder;
348 int err;
349
350 remainder = xdr->page_len - base;
351 base += xdr->page_base;
352 ppage = xdr->pages + (base >> PAGE_SHIFT);
353 base &= ~PAGE_MASK;
354 do_sendpage = sock->ops->sendpage;
355 if (!zerocopy)
356 do_sendpage = sock_no_sendpage;
357 for(;;) {
358 unsigned int len = min_t(unsigned int, PAGE_SIZE - base, remainder);
359 int flags = XS_SENDMSG_FLAGS;
360
361 remainder -= len;
362 if (more)
363 flags |= MSG_MORE;
364 if (remainder != 0)
365 flags |= MSG_SENDPAGE_NOTLAST | MSG_MORE;
366 err = do_sendpage(sock, *ppage, base, len, flags);
367 if (remainder == 0 || err != len)
368 break;
369 *sent_p += err;
370 ppage++;
371 base = 0;
372 }
373 if (err > 0) {
374 *sent_p += err;
375 err = 0;
376 }
377 return err;
378}
379
380
381
382
383
384
385
386
387
388
389
390
391static int xs_sendpages(struct socket *sock, struct sockaddr *addr, int addrlen, struct xdr_buf *xdr, unsigned int base, bool zerocopy, int *sent_p)
392{
393 unsigned int remainder = xdr->len - base;
394 int err = 0;
395 int sent = 0;
396
397 if (unlikely(!sock))
398 return -ENOTSOCK;
399
400 if (base != 0) {
401 addr = NULL;
402 addrlen = 0;
403 }
404
405 if (base < xdr->head[0].iov_len || addr != NULL) {
406 unsigned int len = xdr->head[0].iov_len - base;
407 remainder -= len;
408 err = xs_send_kvec(sock, addr, addrlen, &xdr->head[0], base, remainder != 0);
409 if (remainder == 0 || err != len)
410 goto out;
411 *sent_p += err;
412 base = 0;
413 } else
414 base -= xdr->head[0].iov_len;
415
416 if (base < xdr->page_len) {
417 unsigned int len = xdr->page_len - base;
418 remainder -= len;
419 err = xs_send_pagedata(sock, xdr, base, remainder != 0, zerocopy, &sent);
420 *sent_p += sent;
421 if (remainder == 0 || sent != len)
422 goto out;
423 base = 0;
424 } else
425 base -= xdr->page_len;
426
427 if (base >= xdr->tail[0].iov_len)
428 return 0;
429 err = xs_send_kvec(sock, NULL, 0, &xdr->tail[0], base, 0);
430out:
431 if (err > 0) {
432 *sent_p += err;
433 err = 0;
434 }
435 return err;
436}
437
438static void xs_nospace_callback(struct rpc_task *task)
439{
440 struct sock_xprt *transport = container_of(task->tk_rqstp->rq_xprt, struct sock_xprt, xprt);
441
442 transport->inet->sk_write_pending--;
443}
444
445
446
447
448
449
450static int xs_nospace(struct rpc_task *task)
451{
452 struct rpc_rqst *req = task->tk_rqstp;
453 struct rpc_xprt *xprt = req->rq_xprt;
454 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
455 struct sock *sk = transport->inet;
456 int ret = -EAGAIN;
457
458 dprintk("RPC: %5u xmit incomplete (%u left of %u)\n",
459 task->tk_pid, req->rq_slen - req->rq_bytes_sent,
460 req->rq_slen);
461
462
463 spin_lock_bh(&xprt->transport_lock);
464
465
466 if (xprt_connected(xprt)) {
467
468 sk->sk_write_pending++;
469 xprt_wait_for_buffer_space(task, xs_nospace_callback);
470 } else
471 ret = -ENOTCONN;
472
473 spin_unlock_bh(&xprt->transport_lock);
474
475
476 sk->sk_write_space(sk);
477 return ret;
478}
479
480
481
482
483static inline void xs_encode_stream_record_marker(struct xdr_buf *buf)
484{
485 u32 reclen = buf->len - sizeof(rpc_fraghdr);
486 rpc_fraghdr *base = buf->head[0].iov_base;
487 *base = cpu_to_be32(RPC_LAST_STREAM_FRAGMENT | reclen);
488}
489
490
491
492
493
494
495
496
497
498
499
500
501static int xs_local_send_request(struct rpc_task *task)
502{
503 struct rpc_rqst *req = task->tk_rqstp;
504 struct rpc_xprt *xprt = req->rq_xprt;
505 struct sock_xprt *transport =
506 container_of(xprt, struct sock_xprt, xprt);
507 struct xdr_buf *xdr = &req->rq_snd_buf;
508 int status;
509 int sent = 0;
510
511 xs_encode_stream_record_marker(&req->rq_snd_buf);
512
513 xs_pktdump("packet data:",
514 req->rq_svec->iov_base, req->rq_svec->iov_len);
515
516 status = xs_sendpages(transport->sock, NULL, 0, xdr, req->rq_bytes_sent,
517 true, &sent);
518 dprintk("RPC: %s(%u) = %d\n",
519 __func__, xdr->len - req->rq_bytes_sent, status);
520
521 if (status == -EAGAIN && sock_writeable(transport->inet))
522 status = -ENOBUFS;
523
524 if (likely(sent > 0) || status == 0) {
525 req->rq_bytes_sent += sent;
526 req->rq_xmit_bytes_sent += sent;
527 if (likely(req->rq_bytes_sent >= req->rq_slen)) {
528 req->rq_bytes_sent = 0;
529 return 0;
530 }
531 status = -EAGAIN;
532 }
533
534 switch (status) {
535 case -ENOBUFS:
536 break;
537 case -EAGAIN:
538 status = xs_nospace(task);
539 break;
540 default:
541 dprintk("RPC: sendmsg returned unrecognized error %d\n",
542 -status);
543 case -EPIPE:
544 xs_close(xprt);
545 status = -ENOTCONN;
546 }
547
548 return status;
549}
550
551
552
553
554
555
556
557
558
559
560
561
562static int xs_udp_send_request(struct rpc_task *task)
563{
564 struct rpc_rqst *req = task->tk_rqstp;
565 struct rpc_xprt *xprt = req->rq_xprt;
566 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
567 struct xdr_buf *xdr = &req->rq_snd_buf;
568 int sent = 0;
569 int status;
570
571 xs_pktdump("packet data:",
572 req->rq_svec->iov_base,
573 req->rq_svec->iov_len);
574
575 if (!xprt_bound(xprt))
576 return -ENOTCONN;
577 status = xs_sendpages(transport->sock, xs_addr(xprt), xprt->addrlen,
578 xdr, req->rq_bytes_sent, true, &sent);
579
580 dprintk("RPC: xs_udp_send_request(%u) = %d\n",
581 xdr->len - req->rq_bytes_sent, status);
582
583
584 if (status == -EPERM)
585 goto process_status;
586
587 if (status == -EAGAIN && sock_writeable(transport->inet))
588 status = -ENOBUFS;
589
590 if (sent > 0 || status == 0) {
591 req->rq_xmit_bytes_sent += sent;
592 if (sent >= req->rq_slen)
593 return 0;
594
595 status = -EAGAIN;
596 }
597
598process_status:
599 switch (status) {
600 case -ENOTSOCK:
601 status = -ENOTCONN;
602
603 break;
604 case -EAGAIN:
605 status = xs_nospace(task);
606 break;
607 case -ENETUNREACH:
608 case -ENOBUFS:
609 case -EPIPE:
610 case -ECONNREFUSED:
611 case -EPERM:
612
613
614 break;
615 default:
616 dprintk("RPC: sendmsg returned unrecognized error %d\n",
617 -status);
618 }
619
620 return status;
621}
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637static int xs_tcp_send_request(struct rpc_task *task)
638{
639 struct rpc_rqst *req = task->tk_rqstp;
640 struct rpc_xprt *xprt = req->rq_xprt;
641 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
642 struct xdr_buf *xdr = &req->rq_snd_buf;
643 bool zerocopy = true;
644 bool vm_wait = false;
645 int status;
646 int sent;
647
648 xs_encode_stream_record_marker(&req->rq_snd_buf);
649
650 xs_pktdump("packet data:",
651 req->rq_svec->iov_base,
652 req->rq_svec->iov_len);
653
654
655
656
657 if (task->tk_flags & RPC_TASK_SENT)
658 zerocopy = false;
659
660
661
662
663 while (1) {
664 sent = 0;
665 status = xs_sendpages(transport->sock, NULL, 0, xdr,
666 req->rq_bytes_sent, zerocopy, &sent);
667
668 dprintk("RPC: xs_tcp_send_request(%u) = %d\n",
669 xdr->len - req->rq_bytes_sent, status);
670
671
672
673 req->rq_bytes_sent += sent;
674 req->rq_xmit_bytes_sent += sent;
675 if (likely(req->rq_bytes_sent >= req->rq_slen)) {
676 req->rq_bytes_sent = 0;
677 return 0;
678 }
679
680 WARN_ON_ONCE(sent == 0 && status == 0);
681
682 if (status == -EAGAIN ) {
683
684
685
686
687 if (test_bit(SOCK_NOSPACE, &transport->sock->flags))
688 break;
689
690
691
692 if (sent == 0) {
693 status = -ENOBUFS;
694 if (vm_wait)
695 break;
696
697
698
699 vm_wait = true;
700 }
701 continue;
702 }
703 if (status < 0)
704 break;
705 vm_wait = false;
706 }
707
708 switch (status) {
709 case -ENOTSOCK:
710 status = -ENOTCONN;
711
712 break;
713 case -EAGAIN:
714 status = xs_nospace(task);
715 break;
716 case -ECONNRESET:
717 case -ECONNREFUSED:
718 case -ENOTCONN:
719 case -EADDRINUSE:
720 case -ENOBUFS:
721 case -EPIPE:
722 break;
723 default:
724 dprintk("RPC: sendmsg returned unrecognized error %d\n",
725 -status);
726 }
727
728 return status;
729}
730
731
732
733
734
735
736
737
738
739
740static void xs_tcp_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task)
741{
742 struct rpc_rqst *req;
743
744 if (task != xprt->snd_task)
745 return;
746 if (task == NULL)
747 goto out_release;
748 req = task->tk_rqstp;
749 if (req == NULL)
750 goto out_release;
751 if (req->rq_bytes_sent == 0)
752 goto out_release;
753 if (req->rq_bytes_sent == req->rq_snd_buf.len)
754 goto out_release;
755 set_bit(XPRT_CLOSE_WAIT, &xprt->state);
756out_release:
757 xprt_release_xprt(xprt, task);
758}
759
760static void xs_save_old_callbacks(struct sock_xprt *transport, struct sock *sk)
761{
762 transport->old_data_ready = sk->sk_data_ready;
763 transport->old_state_change = sk->sk_state_change;
764 transport->old_write_space = sk->sk_write_space;
765 transport->old_error_report = sk->sk_error_report;
766}
767
768static void xs_restore_old_callbacks(struct sock_xprt *transport, struct sock *sk)
769{
770 sk->sk_data_ready = transport->old_data_ready;
771 sk->sk_state_change = transport->old_state_change;
772 sk->sk_write_space = transport->old_write_space;
773 sk->sk_error_report = transport->old_error_report;
774}
775
776static void xs_sock_reset_state_flags(struct rpc_xprt *xprt)
777{
778 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
779
780 clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
781}
782
783static void xs_sock_reset_connection_flags(struct rpc_xprt *xprt)
784{
785 smp_mb__before_atomic();
786 clear_bit(XPRT_CLOSE_WAIT, &xprt->state);
787 clear_bit(XPRT_CLOSING, &xprt->state);
788 xs_sock_reset_state_flags(xprt);
789 smp_mb__after_atomic();
790}
791
792static void xs_sock_mark_closed(struct rpc_xprt *xprt)
793{
794 xs_sock_reset_connection_flags(xprt);
795
796 xprt_disconnect_done(xprt);
797}
798
799
800
801
802
803
804
805
806static void xs_error_report(struct sock *sk)
807{
808 struct rpc_xprt *xprt;
809 int err;
810
811 read_lock_bh(&sk->sk_callback_lock);
812 if (!(xprt = xprt_from_sock(sk)))
813 goto out;
814
815 err = -sk->sk_err;
816 if (err == 0)
817 goto out;
818
819 if (sk->sk_state == TCP_CLOSE)
820 xs_sock_mark_closed(xprt);
821 dprintk("RPC: xs_error_report client %p, error=%d...\n",
822 xprt, -err);
823 trace_rpc_socket_error(xprt, sk->sk_socket, err);
824 xprt_wake_pending_tasks(xprt, err);
825 out:
826 read_unlock_bh(&sk->sk_callback_lock);
827}
828
829static void xs_reset_transport(struct sock_xprt *transport)
830{
831 struct socket *sock = transport->sock;
832 struct sock *sk = transport->inet;
833 struct rpc_xprt *xprt = &transport->xprt;
834
835 if (sk == NULL)
836 return;
837
838 if (atomic_read(&transport->xprt.swapper))
839 sk_clear_memalloc(sk);
840
841 kernel_sock_shutdown(sock, SHUT_RDWR);
842
843 mutex_lock(&transport->recv_mutex);
844 write_lock_bh(&sk->sk_callback_lock);
845 transport->inet = NULL;
846 transport->sock = NULL;
847
848 sk->sk_user_data = NULL;
849
850 xs_restore_old_callbacks(transport, sk);
851 xprt_clear_connected(xprt);
852 write_unlock_bh(&sk->sk_callback_lock);
853 xs_sock_reset_connection_flags(xprt);
854 mutex_unlock(&transport->recv_mutex);
855
856 trace_rpc_socket_close(xprt, sock);
857 sock_release(sock);
858}
859
860
861
862
863
864
865
866
867
868
869
870static void xs_close(struct rpc_xprt *xprt)
871{
872 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
873
874 dprintk("RPC: xs_close xprt %p\n", xprt);
875
876 xs_reset_transport(transport);
877 xprt->reestablish_timeout = 0;
878
879 xprt_disconnect_done(xprt);
880}
881
882static void xs_inject_disconnect(struct rpc_xprt *xprt)
883{
884 dprintk("RPC: injecting transport disconnect on xprt=%p\n",
885 xprt);
886 xprt_disconnect_done(xprt);
887}
888
889static void xs_xprt_free(struct rpc_xprt *xprt)
890{
891 xs_free_peer_addresses(xprt);
892 xprt_free(xprt);
893}
894
895
896
897
898
899
900static void xs_destroy(struct rpc_xprt *xprt)
901{
902 struct sock_xprt *transport = container_of(xprt,
903 struct sock_xprt, xprt);
904 dprintk("RPC: xs_destroy xprt %p\n", xprt);
905
906 cancel_delayed_work_sync(&transport->connect_worker);
907 xs_close(xprt);
908 cancel_work_sync(&transport->recv_worker);
909 xs_xprt_free(xprt);
910 module_put(THIS_MODULE);
911}
912
913static int xs_local_copy_to_xdr(struct xdr_buf *xdr, struct sk_buff *skb)
914{
915 struct xdr_skb_reader desc = {
916 .skb = skb,
917 .offset = sizeof(rpc_fraghdr),
918 .count = skb->len - sizeof(rpc_fraghdr),
919 };
920
921 if (xdr_partial_copy_from_skb(xdr, 0, &desc, xdr_skb_read_bits) < 0)
922 return -1;
923 if (desc.count)
924 return -1;
925 return 0;
926}
927
928
929
930
931
932
933
934
935
936static void xs_local_data_read_skb(struct rpc_xprt *xprt,
937 struct sock *sk,
938 struct sk_buff *skb)
939{
940 struct rpc_task *task;
941 struct rpc_rqst *rovr;
942 int repsize, copied;
943 u32 _xid;
944 __be32 *xp;
945
946 repsize = skb->len - sizeof(rpc_fraghdr);
947 if (repsize < 4) {
948 dprintk("RPC: impossible RPC reply size %d\n", repsize);
949 return;
950 }
951
952
953 xp = skb_header_pointer(skb, sizeof(rpc_fraghdr), sizeof(_xid), &_xid);
954 if (xp == NULL)
955 return;
956
957
958 spin_lock_bh(&xprt->transport_lock);
959 rovr = xprt_lookup_rqst(xprt, *xp);
960 if (!rovr)
961 goto out_unlock;
962 task = rovr->rq_task;
963
964 copied = rovr->rq_private_buf.buflen;
965 if (copied > repsize)
966 copied = repsize;
967
968 if (xs_local_copy_to_xdr(&rovr->rq_private_buf, skb)) {
969 dprintk("RPC: sk_buff copy failed\n");
970 goto out_unlock;
971 }
972
973 xprt_complete_rqst(task, copied);
974
975 out_unlock:
976 spin_unlock_bh(&xprt->transport_lock);
977}
978
979static void xs_local_data_receive(struct sock_xprt *transport)
980{
981 struct sk_buff *skb;
982 struct sock *sk;
983 int err;
984
985 mutex_lock(&transport->recv_mutex);
986 sk = transport->inet;
987 if (sk == NULL)
988 goto out;
989 for (;;) {
990 skb = skb_recv_datagram(sk, 0, 1, &err);
991 if (skb != NULL) {
992 xs_local_data_read_skb(&transport->xprt, sk, skb);
993 skb_free_datagram(sk, skb);
994 continue;
995 }
996 if (!test_and_clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state))
997 break;
998 }
999out:
1000 mutex_unlock(&transport->recv_mutex);
1001}
1002
1003static void xs_local_data_receive_workfn(struct work_struct *work)
1004{
1005 struct sock_xprt *transport =
1006 container_of(work, struct sock_xprt, recv_worker);
1007 xs_local_data_receive(transport);
1008}
1009
1010
1011
1012
1013
1014
1015
1016
1017static void xs_udp_data_read_skb(struct rpc_xprt *xprt,
1018 struct sock *sk,
1019 struct sk_buff *skb)
1020{
1021 struct rpc_task *task;
1022 struct rpc_rqst *rovr;
1023 int repsize, copied;
1024 u32 _xid;
1025 __be32 *xp;
1026
1027 repsize = skb->len;
1028 if (repsize < 4) {
1029 dprintk("RPC: impossible RPC reply size %d!\n", repsize);
1030 return;
1031 }
1032
1033
1034 xp = skb_header_pointer(skb, 0, sizeof(_xid), &_xid);
1035 if (xp == NULL)
1036 return;
1037
1038
1039 spin_lock_bh(&xprt->transport_lock);
1040 rovr = xprt_lookup_rqst(xprt, *xp);
1041 if (!rovr)
1042 goto out_unlock;
1043 task = rovr->rq_task;
1044
1045 if ((copied = rovr->rq_private_buf.buflen) > repsize)
1046 copied = repsize;
1047
1048
1049 if (csum_partial_copy_to_xdr(&rovr->rq_private_buf, skb)) {
1050 __UDPX_INC_STATS(sk, UDP_MIB_INERRORS);
1051 goto out_unlock;
1052 }
1053
1054 __UDPX_INC_STATS(sk, UDP_MIB_INDATAGRAMS);
1055
1056 xprt_adjust_cwnd(xprt, task, copied);
1057 xprt_complete_rqst(task, copied);
1058
1059 out_unlock:
1060 spin_unlock_bh(&xprt->transport_lock);
1061}
1062
1063static void xs_udp_data_receive(struct sock_xprt *transport)
1064{
1065 struct sk_buff *skb;
1066 struct sock *sk;
1067 int err;
1068
1069 mutex_lock(&transport->recv_mutex);
1070 sk = transport->inet;
1071 if (sk == NULL)
1072 goto out;
1073 for (;;) {
1074 skb = skb_recv_datagram(sk, 0, 1, &err);
1075 if (skb != NULL) {
1076 xs_udp_data_read_skb(&transport->xprt, sk, skb);
1077 skb_free_datagram_locked(sk, skb);
1078 continue;
1079 }
1080 if (!test_and_clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state))
1081 break;
1082 }
1083out:
1084 mutex_unlock(&transport->recv_mutex);
1085}
1086
1087static void xs_udp_data_receive_workfn(struct work_struct *work)
1088{
1089 struct sock_xprt *transport =
1090 container_of(work, struct sock_xprt, recv_worker);
1091 xs_udp_data_receive(transport);
1092}
1093
1094
1095
1096
1097
1098
1099static void xs_data_ready(struct sock *sk)
1100{
1101 struct rpc_xprt *xprt;
1102
1103 read_lock_bh(&sk->sk_callback_lock);
1104 dprintk("RPC: xs_data_ready...\n");
1105 xprt = xprt_from_sock(sk);
1106 if (xprt != NULL) {
1107 struct sock_xprt *transport = container_of(xprt,
1108 struct sock_xprt, xprt);
1109 transport->old_data_ready(sk);
1110
1111
1112
1113 if (xprt->reestablish_timeout)
1114 xprt->reestablish_timeout = 0;
1115 if (!test_and_set_bit(XPRT_SOCK_DATA_READY, &transport->sock_state))
1116 queue_work(xprtiod_workqueue, &transport->recv_worker);
1117 }
1118 read_unlock_bh(&sk->sk_callback_lock);
1119}
1120
1121
1122
1123
1124
1125static void xs_tcp_force_close(struct rpc_xprt *xprt)
1126{
1127 xprt_force_disconnect(xprt);
1128}
1129
1130static inline void xs_tcp_read_fraghdr(struct rpc_xprt *xprt, struct xdr_skb_reader *desc)
1131{
1132 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
1133 size_t len, used;
1134 char *p;
1135
1136 p = ((char *) &transport->tcp_fraghdr) + transport->tcp_offset;
1137 len = sizeof(transport->tcp_fraghdr) - transport->tcp_offset;
1138 used = xdr_skb_read_bits(desc, p, len);
1139 transport->tcp_offset += used;
1140 if (used != len)
1141 return;
1142
1143 transport->tcp_reclen = ntohl(transport->tcp_fraghdr);
1144 if (transport->tcp_reclen & RPC_LAST_STREAM_FRAGMENT)
1145 transport->tcp_flags |= TCP_RCV_LAST_FRAG;
1146 else
1147 transport->tcp_flags &= ~TCP_RCV_LAST_FRAG;
1148 transport->tcp_reclen &= RPC_FRAGMENT_SIZE_MASK;
1149
1150 transport->tcp_flags &= ~TCP_RCV_COPY_FRAGHDR;
1151 transport->tcp_offset = 0;
1152
1153
1154 if (unlikely(transport->tcp_reclen < 8)) {
1155 dprintk("RPC: invalid TCP record fragment length\n");
1156 xs_tcp_force_close(xprt);
1157 return;
1158 }
1159 dprintk("RPC: reading TCP record fragment of length %d\n",
1160 transport->tcp_reclen);
1161}
1162
1163static void xs_tcp_check_fraghdr(struct sock_xprt *transport)
1164{
1165 if (transport->tcp_offset == transport->tcp_reclen) {
1166 transport->tcp_flags |= TCP_RCV_COPY_FRAGHDR;
1167 transport->tcp_offset = 0;
1168 if (transport->tcp_flags & TCP_RCV_LAST_FRAG) {
1169 transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
1170 transport->tcp_flags |= TCP_RCV_COPY_XID;
1171 transport->tcp_copied = 0;
1172 }
1173 }
1174}
1175
1176static inline void xs_tcp_read_xid(struct sock_xprt *transport, struct xdr_skb_reader *desc)
1177{
1178 size_t len, used;
1179 char *p;
1180
1181 len = sizeof(transport->tcp_xid) - transport->tcp_offset;
1182 dprintk("RPC: reading XID (%Zu bytes)\n", len);
1183 p = ((char *) &transport->tcp_xid) + transport->tcp_offset;
1184 used = xdr_skb_read_bits(desc, p, len);
1185 transport->tcp_offset += used;
1186 if (used != len)
1187 return;
1188 transport->tcp_flags &= ~TCP_RCV_COPY_XID;
1189 transport->tcp_flags |= TCP_RCV_READ_CALLDIR;
1190 transport->tcp_copied = 4;
1191 dprintk("RPC: reading %s XID %08x\n",
1192 (transport->tcp_flags & TCP_RPC_REPLY) ? "reply for"
1193 : "request with",
1194 ntohl(transport->tcp_xid));
1195 xs_tcp_check_fraghdr(transport);
1196}
1197
1198static inline void xs_tcp_read_calldir(struct sock_xprt *transport,
1199 struct xdr_skb_reader *desc)
1200{
1201 size_t len, used;
1202 u32 offset;
1203 char *p;
1204
1205
1206
1207
1208
1209
1210
1211 offset = transport->tcp_offset - sizeof(transport->tcp_xid);
1212 len = sizeof(transport->tcp_calldir) - offset;
1213 dprintk("RPC: reading CALL/REPLY flag (%Zu bytes)\n", len);
1214 p = ((char *) &transport->tcp_calldir) + offset;
1215 used = xdr_skb_read_bits(desc, p, len);
1216 transport->tcp_offset += used;
1217 if (used != len)
1218 return;
1219 transport->tcp_flags &= ~TCP_RCV_READ_CALLDIR;
1220
1221
1222
1223
1224 switch (ntohl(transport->tcp_calldir)) {
1225 case RPC_REPLY:
1226 transport->tcp_flags |= TCP_RCV_COPY_CALLDIR;
1227 transport->tcp_flags |= TCP_RCV_COPY_DATA;
1228 transport->tcp_flags |= TCP_RPC_REPLY;
1229 break;
1230 case RPC_CALL:
1231 transport->tcp_flags |= TCP_RCV_COPY_CALLDIR;
1232 transport->tcp_flags |= TCP_RCV_COPY_DATA;
1233 transport->tcp_flags &= ~TCP_RPC_REPLY;
1234 break;
1235 default:
1236 dprintk("RPC: invalid request message type\n");
1237 xs_tcp_force_close(&transport->xprt);
1238 }
1239 xs_tcp_check_fraghdr(transport);
1240}
1241
1242static inline void xs_tcp_read_common(struct rpc_xprt *xprt,
1243 struct xdr_skb_reader *desc,
1244 struct rpc_rqst *req)
1245{
1246 struct sock_xprt *transport =
1247 container_of(xprt, struct sock_xprt, xprt);
1248 struct xdr_buf *rcvbuf;
1249 size_t len;
1250 ssize_t r;
1251
1252 rcvbuf = &req->rq_private_buf;
1253
1254 if (transport->tcp_flags & TCP_RCV_COPY_CALLDIR) {
1255
1256
1257
1258 memcpy(rcvbuf->head[0].iov_base + transport->tcp_copied,
1259 &transport->tcp_calldir,
1260 sizeof(transport->tcp_calldir));
1261 transport->tcp_copied += sizeof(transport->tcp_calldir);
1262 transport->tcp_flags &= ~TCP_RCV_COPY_CALLDIR;
1263 }
1264
1265 len = desc->count;
1266 if (len > transport->tcp_reclen - transport->tcp_offset) {
1267 struct xdr_skb_reader my_desc;
1268
1269 len = transport->tcp_reclen - transport->tcp_offset;
1270 memcpy(&my_desc, desc, sizeof(my_desc));
1271 my_desc.count = len;
1272 r = xdr_partial_copy_from_skb(rcvbuf, transport->tcp_copied,
1273 &my_desc, xdr_skb_read_bits);
1274 desc->count -= r;
1275 desc->offset += r;
1276 } else
1277 r = xdr_partial_copy_from_skb(rcvbuf, transport->tcp_copied,
1278 desc, xdr_skb_read_bits);
1279
1280 if (r > 0) {
1281 transport->tcp_copied += r;
1282 transport->tcp_offset += r;
1283 }
1284 if (r != len) {
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294 transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
1295 dprintk("RPC: XID %08x truncated request\n",
1296 ntohl(transport->tcp_xid));
1297 dprintk("RPC: xprt = %p, tcp_copied = %lu, "
1298 "tcp_offset = %u, tcp_reclen = %u\n",
1299 xprt, transport->tcp_copied,
1300 transport->tcp_offset, transport->tcp_reclen);
1301 return;
1302 }
1303
1304 dprintk("RPC: XID %08x read %Zd bytes\n",
1305 ntohl(transport->tcp_xid), r);
1306 dprintk("RPC: xprt = %p, tcp_copied = %lu, tcp_offset = %u, "
1307 "tcp_reclen = %u\n", xprt, transport->tcp_copied,
1308 transport->tcp_offset, transport->tcp_reclen);
1309
1310 if (transport->tcp_copied == req->rq_private_buf.buflen)
1311 transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
1312 else if (transport->tcp_offset == transport->tcp_reclen) {
1313 if (transport->tcp_flags & TCP_RCV_LAST_FRAG)
1314 transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
1315 }
1316}
1317
1318
1319
1320
1321
1322static inline int xs_tcp_read_reply(struct rpc_xprt *xprt,
1323 struct xdr_skb_reader *desc)
1324{
1325 struct sock_xprt *transport =
1326 container_of(xprt, struct sock_xprt, xprt);
1327 struct rpc_rqst *req;
1328
1329 dprintk("RPC: read reply XID %08x\n", ntohl(transport->tcp_xid));
1330
1331
1332 spin_lock_bh(&xprt->transport_lock);
1333 req = xprt_lookup_rqst(xprt, transport->tcp_xid);
1334 if (!req) {
1335 dprintk("RPC: XID %08x request not found!\n",
1336 ntohl(transport->tcp_xid));
1337 spin_unlock_bh(&xprt->transport_lock);
1338 return -1;
1339 }
1340
1341 xs_tcp_read_common(xprt, desc, req);
1342
1343 if (!(transport->tcp_flags & TCP_RCV_COPY_DATA))
1344 xprt_complete_rqst(req->rq_task, transport->tcp_copied);
1345
1346 spin_unlock_bh(&xprt->transport_lock);
1347 return 0;
1348}
1349
1350#if defined(CONFIG_SUNRPC_BACKCHANNEL)
1351
1352
1353
1354
1355
1356
1357
1358static int xs_tcp_read_callback(struct rpc_xprt *xprt,
1359 struct xdr_skb_reader *desc)
1360{
1361 struct sock_xprt *transport =
1362 container_of(xprt, struct sock_xprt, xprt);
1363 struct rpc_rqst *req;
1364
1365
1366 spin_lock_bh(&xprt->transport_lock);
1367 req = xprt_lookup_bc_request(xprt, transport->tcp_xid);
1368 if (req == NULL) {
1369 spin_unlock_bh(&xprt->transport_lock);
1370 printk(KERN_WARNING "Callback slot table overflowed\n");
1371 xprt_force_disconnect(xprt);
1372 return -1;
1373 }
1374
1375 dprintk("RPC: read callback XID %08x\n", ntohl(req->rq_xid));
1376 xs_tcp_read_common(xprt, desc, req);
1377
1378 if (!(transport->tcp_flags & TCP_RCV_COPY_DATA))
1379 xprt_complete_bc_request(req, transport->tcp_copied);
1380 spin_unlock_bh(&xprt->transport_lock);
1381
1382 return 0;
1383}
1384
1385static inline int _xs_tcp_read_data(struct rpc_xprt *xprt,
1386 struct xdr_skb_reader *desc)
1387{
1388 struct sock_xprt *transport =
1389 container_of(xprt, struct sock_xprt, xprt);
1390
1391 return (transport->tcp_flags & TCP_RPC_REPLY) ?
1392 xs_tcp_read_reply(xprt, desc) :
1393 xs_tcp_read_callback(xprt, desc);
1394}
1395
1396static int xs_tcp_bc_up(struct svc_serv *serv, struct net *net)
1397{
1398 int ret;
1399
1400 ret = svc_create_xprt(serv, "tcp-bc", net, PF_INET, 0,
1401 SVC_SOCK_ANONYMOUS);
1402 if (ret < 0)
1403 return ret;
1404 return 0;
1405}
1406
1407static size_t xs_tcp_bc_maxpayload(struct rpc_xprt *xprt)
1408{
1409 return PAGE_SIZE;
1410}
1411#else
1412static inline int _xs_tcp_read_data(struct rpc_xprt *xprt,
1413 struct xdr_skb_reader *desc)
1414{
1415 return xs_tcp_read_reply(xprt, desc);
1416}
1417#endif
1418
1419
1420
1421
1422
1423static void xs_tcp_read_data(struct rpc_xprt *xprt,
1424 struct xdr_skb_reader *desc)
1425{
1426 struct sock_xprt *transport =
1427 container_of(xprt, struct sock_xprt, xprt);
1428
1429 if (_xs_tcp_read_data(xprt, desc) == 0)
1430 xs_tcp_check_fraghdr(transport);
1431 else {
1432
1433
1434
1435
1436 transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
1437 }
1438}
1439
1440static inline void xs_tcp_read_discard(struct sock_xprt *transport, struct xdr_skb_reader *desc)
1441{
1442 size_t len;
1443
1444 len = transport->tcp_reclen - transport->tcp_offset;
1445 if (len > desc->count)
1446 len = desc->count;
1447 desc->count -= len;
1448 desc->offset += len;
1449 transport->tcp_offset += len;
1450 dprintk("RPC: discarded %Zu bytes\n", len);
1451 xs_tcp_check_fraghdr(transport);
1452}
1453
1454static int xs_tcp_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb, unsigned int offset, size_t len)
1455{
1456 struct rpc_xprt *xprt = rd_desc->arg.data;
1457 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
1458 struct xdr_skb_reader desc = {
1459 .skb = skb,
1460 .offset = offset,
1461 .count = len,
1462 };
1463
1464 dprintk("RPC: xs_tcp_data_recv started\n");
1465 do {
1466 trace_xs_tcp_data_recv(transport);
1467
1468
1469 if (transport->tcp_flags & TCP_RCV_COPY_FRAGHDR) {
1470 xs_tcp_read_fraghdr(xprt, &desc);
1471 continue;
1472 }
1473
1474 if (transport->tcp_flags & TCP_RCV_COPY_XID) {
1475 xs_tcp_read_xid(transport, &desc);
1476 continue;
1477 }
1478
1479 if (transport->tcp_flags & TCP_RCV_READ_CALLDIR) {
1480 xs_tcp_read_calldir(transport, &desc);
1481 continue;
1482 }
1483
1484 if (transport->tcp_flags & TCP_RCV_COPY_DATA) {
1485 xs_tcp_read_data(xprt, &desc);
1486 continue;
1487 }
1488
1489 xs_tcp_read_discard(transport, &desc);
1490 } while (desc.count);
1491 trace_xs_tcp_data_recv(transport);
1492 dprintk("RPC: xs_tcp_data_recv done\n");
1493 return len - desc.count;
1494}
1495
1496static void xs_tcp_data_receive(struct sock_xprt *transport)
1497{
1498 struct rpc_xprt *xprt = &transport->xprt;
1499 struct sock *sk;
1500 read_descriptor_t rd_desc = {
1501 .count = 2*1024*1024,
1502 .arg.data = xprt,
1503 };
1504 unsigned long total = 0;
1505 int read = 0;
1506
1507 mutex_lock(&transport->recv_mutex);
1508 sk = transport->inet;
1509 if (sk == NULL)
1510 goto out;
1511
1512
1513 for (;;) {
1514 lock_sock(sk);
1515 read = tcp_read_sock(sk, &rd_desc, xs_tcp_data_recv);
1516 if (read <= 0) {
1517 clear_bit(XPRT_SOCK_DATA_READY, &transport->sock_state);
1518 release_sock(sk);
1519 if (!test_bit(XPRT_SOCK_DATA_READY, &transport->sock_state))
1520 break;
1521 } else {
1522 release_sock(sk);
1523 total += read;
1524 }
1525 rd_desc.count = 65536;
1526 }
1527out:
1528 mutex_unlock(&transport->recv_mutex);
1529 trace_xs_tcp_data_ready(xprt, read, total);
1530}
1531
1532static void xs_tcp_data_receive_workfn(struct work_struct *work)
1533{
1534 struct sock_xprt *transport =
1535 container_of(work, struct sock_xprt, recv_worker);
1536 xs_tcp_data_receive(transport);
1537}
1538
1539
1540
1541
1542
1543
1544static void xs_tcp_state_change(struct sock *sk)
1545{
1546 struct rpc_xprt *xprt;
1547 struct sock_xprt *transport;
1548
1549 read_lock_bh(&sk->sk_callback_lock);
1550 if (!(xprt = xprt_from_sock(sk)))
1551 goto out;
1552 dprintk("RPC: xs_tcp_state_change client %p...\n", xprt);
1553 dprintk("RPC: state %x conn %d dead %d zapped %d sk_shutdown %d\n",
1554 sk->sk_state, xprt_connected(xprt),
1555 sock_flag(sk, SOCK_DEAD),
1556 sock_flag(sk, SOCK_ZAPPED),
1557 sk->sk_shutdown);
1558
1559 transport = container_of(xprt, struct sock_xprt, xprt);
1560 trace_rpc_socket_state_change(xprt, sk->sk_socket);
1561 switch (sk->sk_state) {
1562 case TCP_ESTABLISHED:
1563 spin_lock(&xprt->transport_lock);
1564 if (!xprt_test_and_set_connected(xprt)) {
1565
1566
1567 transport->tcp_offset = 0;
1568 transport->tcp_reclen = 0;
1569 transport->tcp_copied = 0;
1570 transport->tcp_flags =
1571 TCP_RCV_COPY_FRAGHDR | TCP_RCV_COPY_XID;
1572 xprt->connect_cookie++;
1573 clear_bit(XPRT_SOCK_CONNECTING, &transport->sock_state);
1574 xprt_clear_connecting(xprt);
1575
1576 xprt_wake_pending_tasks(xprt, -EAGAIN);
1577 }
1578 spin_unlock(&xprt->transport_lock);
1579 break;
1580 case TCP_FIN_WAIT1:
1581
1582 xprt->connect_cookie++;
1583 xprt->reestablish_timeout = 0;
1584 set_bit(XPRT_CLOSING, &xprt->state);
1585 smp_mb__before_atomic();
1586 clear_bit(XPRT_CONNECTED, &xprt->state);
1587 clear_bit(XPRT_CLOSE_WAIT, &xprt->state);
1588 smp_mb__after_atomic();
1589 break;
1590 case TCP_CLOSE_WAIT:
1591
1592 xprt->connect_cookie++;
1593 clear_bit(XPRT_CONNECTED, &xprt->state);
1594 xs_tcp_force_close(xprt);
1595 case TCP_CLOSING:
1596
1597
1598
1599
1600 if (xprt->reestablish_timeout < XS_TCP_INIT_REEST_TO)
1601 xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
1602 break;
1603 case TCP_LAST_ACK:
1604 set_bit(XPRT_CLOSING, &xprt->state);
1605 smp_mb__before_atomic();
1606 clear_bit(XPRT_CONNECTED, &xprt->state);
1607 smp_mb__after_atomic();
1608 break;
1609 case TCP_CLOSE:
1610 if (test_and_clear_bit(XPRT_SOCK_CONNECTING,
1611 &transport->sock_state))
1612 xprt_clear_connecting(xprt);
1613 xs_sock_mark_closed(xprt);
1614 }
1615 out:
1616 read_unlock_bh(&sk->sk_callback_lock);
1617}
1618
1619static void xs_write_space(struct sock *sk)
1620{
1621 struct socket_wq *wq;
1622 struct rpc_xprt *xprt;
1623
1624 if (!sk->sk_socket)
1625 return;
1626 clear_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
1627
1628 if (unlikely(!(xprt = xprt_from_sock(sk))))
1629 return;
1630 rcu_read_lock();
1631 wq = rcu_dereference(sk->sk_wq);
1632 if (!wq || test_and_clear_bit(SOCKWQ_ASYNC_NOSPACE, &wq->flags) == 0)
1633 goto out;
1634
1635 xprt_write_space(xprt);
1636out:
1637 rcu_read_unlock();
1638}
1639
1640
1641
1642
1643
1644
1645
1646
1647
1648
1649
1650static void xs_udp_write_space(struct sock *sk)
1651{
1652 read_lock_bh(&sk->sk_callback_lock);
1653
1654
1655 if (sock_writeable(sk))
1656 xs_write_space(sk);
1657
1658 read_unlock_bh(&sk->sk_callback_lock);
1659}
1660
1661
1662
1663
1664
1665
1666
1667
1668
1669
1670
1671static void xs_tcp_write_space(struct sock *sk)
1672{
1673 read_lock_bh(&sk->sk_callback_lock);
1674
1675
1676 if (sk_stream_is_writeable(sk))
1677 xs_write_space(sk);
1678
1679 read_unlock_bh(&sk->sk_callback_lock);
1680}
1681
1682static void xs_udp_do_set_buffer_size(struct rpc_xprt *xprt)
1683{
1684 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
1685 struct sock *sk = transport->inet;
1686
1687 if (transport->rcvsize) {
1688 sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
1689 sk->sk_rcvbuf = transport->rcvsize * xprt->max_reqs * 2;
1690 }
1691 if (transport->sndsize) {
1692 sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
1693 sk->sk_sndbuf = transport->sndsize * xprt->max_reqs * 2;
1694 sk->sk_write_space(sk);
1695 }
1696}
1697
1698
1699
1700
1701
1702
1703
1704
1705
1706static void xs_udp_set_buffer_size(struct rpc_xprt *xprt, size_t sndsize, size_t rcvsize)
1707{
1708 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
1709
1710 transport->sndsize = 0;
1711 if (sndsize)
1712 transport->sndsize = sndsize + 1024;
1713 transport->rcvsize = 0;
1714 if (rcvsize)
1715 transport->rcvsize = rcvsize + 1024;
1716
1717 xs_udp_do_set_buffer_size(xprt);
1718}
1719
1720
1721
1722
1723
1724
1725
1726static void xs_udp_timer(struct rpc_xprt *xprt, struct rpc_task *task)
1727{
1728 xprt_adjust_cwnd(xprt, task, -ETIMEDOUT);
1729}
1730
1731static unsigned short xs_get_random_port(void)
1732{
1733 unsigned short range = xprt_max_resvport - xprt_min_resvport + 1;
1734 unsigned short rand = (unsigned short) prandom_u32() % range;
1735 return rand + xprt_min_resvport;
1736}
1737
1738
1739
1740
1741
1742
1743
1744
1745static void xs_sock_set_reuseport(struct socket *sock)
1746{
1747 int opt = 1;
1748
1749 kernel_setsockopt(sock, SOL_SOCKET, SO_REUSEPORT,
1750 (char *)&opt, sizeof(opt));
1751}
1752
1753static unsigned short xs_sock_getport(struct socket *sock)
1754{
1755 struct sockaddr_storage buf;
1756 int buflen;
1757 unsigned short port = 0;
1758
1759 if (kernel_getsockname(sock, (struct sockaddr *)&buf, &buflen) < 0)
1760 goto out;
1761 switch (buf.ss_family) {
1762 case AF_INET6:
1763 port = ntohs(((struct sockaddr_in6 *)&buf)->sin6_port);
1764 break;
1765 case AF_INET:
1766 port = ntohs(((struct sockaddr_in *)&buf)->sin_port);
1767 }
1768out:
1769 return port;
1770}
1771
1772
1773
1774
1775
1776
1777
1778static void xs_set_port(struct rpc_xprt *xprt, unsigned short port)
1779{
1780 dprintk("RPC: setting port for xprt %p to %u\n", xprt, port);
1781
1782 rpc_set_port(xs_addr(xprt), port);
1783 xs_update_peer_port(xprt);
1784}
1785
1786static void xs_set_srcport(struct sock_xprt *transport, struct socket *sock)
1787{
1788 if (transport->srcport == 0)
1789 transport->srcport = xs_sock_getport(sock);
1790}
1791
1792static unsigned short xs_get_srcport(struct sock_xprt *transport)
1793{
1794 unsigned short port = transport->srcport;
1795
1796 if (port == 0 && transport->xprt.resvport)
1797 port = xs_get_random_port();
1798 return port;
1799}
1800
1801static unsigned short xs_next_srcport(struct sock_xprt *transport, unsigned short port)
1802{
1803 if (transport->srcport != 0)
1804 transport->srcport = 0;
1805 if (!transport->xprt.resvport)
1806 return 0;
1807 if (port <= xprt_min_resvport || port > xprt_max_resvport)
1808 return xprt_max_resvport;
1809 return --port;
1810}
1811static int xs_bind(struct sock_xprt *transport, struct socket *sock)
1812{
1813 struct sockaddr_storage myaddr;
1814 int err, nloop = 0;
1815 unsigned short port = xs_get_srcport(transport);
1816 unsigned short last;
1817
1818
1819
1820
1821
1822
1823
1824
1825
1826
1827
1828
1829
1830
1831
1832
1833 if (port == 0)
1834 return 0;
1835
1836 memcpy(&myaddr, &transport->srcaddr, transport->xprt.addrlen);
1837 do {
1838 rpc_set_port((struct sockaddr *)&myaddr, port);
1839 err = kernel_bind(sock, (struct sockaddr *)&myaddr,
1840 transport->xprt.addrlen);
1841 if (err == 0) {
1842 transport->srcport = port;
1843 break;
1844 }
1845 last = port;
1846 port = xs_next_srcport(transport, port);
1847 if (port > last)
1848 nloop++;
1849 } while (err == -EADDRINUSE && nloop != 2);
1850
1851 if (myaddr.ss_family == AF_INET)
1852 dprintk("RPC: %s %pI4:%u: %s (%d)\n", __func__,
1853 &((struct sockaddr_in *)&myaddr)->sin_addr,
1854 port, err ? "failed" : "ok", err);
1855 else
1856 dprintk("RPC: %s %pI6:%u: %s (%d)\n", __func__,
1857 &((struct sockaddr_in6 *)&myaddr)->sin6_addr,
1858 port, err ? "failed" : "ok", err);
1859 return err;
1860}
1861
1862
1863
1864
1865static void xs_local_rpcbind(struct rpc_task *task)
1866{
1867 xprt_set_bound(task->tk_xprt);
1868}
1869
1870static void xs_local_set_port(struct rpc_xprt *xprt, unsigned short port)
1871{
1872}
1873
1874#ifdef CONFIG_DEBUG_LOCK_ALLOC
1875static struct lock_class_key xs_key[2];
1876static struct lock_class_key xs_slock_key[2];
1877
1878static inline void xs_reclassify_socketu(struct socket *sock)
1879{
1880 struct sock *sk = sock->sk;
1881
1882 sock_lock_init_class_and_name(sk, "slock-AF_LOCAL-RPC",
1883 &xs_slock_key[1], "sk_lock-AF_LOCAL-RPC", &xs_key[1]);
1884}
1885
1886static inline void xs_reclassify_socket4(struct socket *sock)
1887{
1888 struct sock *sk = sock->sk;
1889
1890 sock_lock_init_class_and_name(sk, "slock-AF_INET-RPC",
1891 &xs_slock_key[0], "sk_lock-AF_INET-RPC", &xs_key[0]);
1892}
1893
1894static inline void xs_reclassify_socket6(struct socket *sock)
1895{
1896 struct sock *sk = sock->sk;
1897
1898 sock_lock_init_class_and_name(sk, "slock-AF_INET6-RPC",
1899 &xs_slock_key[1], "sk_lock-AF_INET6-RPC", &xs_key[1]);
1900}
1901
1902static inline void xs_reclassify_socket(int family, struct socket *sock)
1903{
1904 if (WARN_ON_ONCE(!sock_allow_reclassification(sock->sk)))
1905 return;
1906
1907 switch (family) {
1908 case AF_LOCAL:
1909 xs_reclassify_socketu(sock);
1910 break;
1911 case AF_INET:
1912 xs_reclassify_socket4(sock);
1913 break;
1914 case AF_INET6:
1915 xs_reclassify_socket6(sock);
1916 break;
1917 }
1918}
1919#else
1920static inline void xs_reclassify_socket(int family, struct socket *sock)
1921{
1922}
1923#endif
1924
1925static void xs_dummy_setup_socket(struct work_struct *work)
1926{
1927}
1928
1929static struct socket *xs_create_sock(struct rpc_xprt *xprt,
1930 struct sock_xprt *transport, int family, int type,
1931 int protocol, bool reuseport)
1932{
1933 struct socket *sock;
1934 int err;
1935
1936 err = __sock_create(xprt->xprt_net, family, type, protocol, &sock, 1);
1937 if (err < 0) {
1938 dprintk("RPC: can't create %d transport socket (%d).\n",
1939 protocol, -err);
1940 goto out;
1941 }
1942 xs_reclassify_socket(family, sock);
1943
1944 if (reuseport)
1945 xs_sock_set_reuseport(sock);
1946
1947 err = xs_bind(transport, sock);
1948 if (err) {
1949 sock_release(sock);
1950 goto out;
1951 }
1952
1953 return sock;
1954out:
1955 return ERR_PTR(err);
1956}
1957
1958static int xs_local_finish_connecting(struct rpc_xprt *xprt,
1959 struct socket *sock)
1960{
1961 struct sock_xprt *transport = container_of(xprt, struct sock_xprt,
1962 xprt);
1963
1964 if (!transport->inet) {
1965 struct sock *sk = sock->sk;
1966
1967 write_lock_bh(&sk->sk_callback_lock);
1968
1969 xs_save_old_callbacks(transport, sk);
1970
1971 sk->sk_user_data = xprt;
1972 sk->sk_data_ready = xs_data_ready;
1973 sk->sk_write_space = xs_udp_write_space;
1974 sock_set_flag(sk, SOCK_FASYNC);
1975 sk->sk_error_report = xs_error_report;
1976 sk->sk_allocation = GFP_NOIO;
1977
1978 xprt_clear_connected(xprt);
1979
1980
1981 transport->sock = sock;
1982 transport->inet = sk;
1983
1984 write_unlock_bh(&sk->sk_callback_lock);
1985 }
1986
1987
1988 xprt->stat.connect_count++;
1989 xprt->stat.connect_start = jiffies;
1990 return kernel_connect(sock, xs_addr(xprt), xprt->addrlen, 0);
1991}
1992
1993
1994
1995
1996
1997static int xs_local_setup_socket(struct sock_xprt *transport)
1998{
1999 struct rpc_xprt *xprt = &transport->xprt;
2000 struct socket *sock;
2001 int status = -EIO;
2002
2003 status = __sock_create(xprt->xprt_net, AF_LOCAL,
2004 SOCK_STREAM, 0, &sock, 1);
2005 if (status < 0) {
2006 dprintk("RPC: can't create AF_LOCAL "
2007 "transport socket (%d).\n", -status);
2008 goto out;
2009 }
2010 xs_reclassify_socket(AF_LOCAL, sock);
2011
2012 dprintk("RPC: worker connecting xprt %p via AF_LOCAL to %s\n",
2013 xprt, xprt->address_strings[RPC_DISPLAY_ADDR]);
2014
2015 status = xs_local_finish_connecting(xprt, sock);
2016 trace_rpc_socket_connect(xprt, sock, status);
2017 switch (status) {
2018 case 0:
2019 dprintk("RPC: xprt %p connected to %s\n",
2020 xprt, xprt->address_strings[RPC_DISPLAY_ADDR]);
2021 xprt_set_connected(xprt);
2022 case -ENOBUFS:
2023 break;
2024 case -ENOENT:
2025 dprintk("RPC: xprt %p: socket %s does not exist\n",
2026 xprt, xprt->address_strings[RPC_DISPLAY_ADDR]);
2027 break;
2028 case -ECONNREFUSED:
2029 dprintk("RPC: xprt %p: connection refused for %s\n",
2030 xprt, xprt->address_strings[RPC_DISPLAY_ADDR]);
2031 break;
2032 default:
2033 printk(KERN_ERR "%s: unhandled error (%d) connecting to %s\n",
2034 __func__, -status,
2035 xprt->address_strings[RPC_DISPLAY_ADDR]);
2036 }
2037
2038out:
2039 xprt_clear_connecting(xprt);
2040 xprt_wake_pending_tasks(xprt, status);
2041 return status;
2042}
2043
2044static void xs_local_connect(struct rpc_xprt *xprt, struct rpc_task *task)
2045{
2046 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2047 int ret;
2048
2049 if (RPC_IS_ASYNC(task)) {
2050
2051
2052
2053
2054
2055
2056
2057
2058
2059 rpc_exit(task, -ENOTCONN);
2060 return;
2061 }
2062 ret = xs_local_setup_socket(transport);
2063 if (ret && !RPC_IS_SOFTCONN(task))
2064 msleep_interruptible(15000);
2065}
2066
2067#if IS_ENABLED(CONFIG_SUNRPC_SWAP)
2068
2069
2070
2071
2072
2073static void xs_set_memalloc(struct rpc_xprt *xprt)
2074{
2075 struct sock_xprt *transport = container_of(xprt, struct sock_xprt,
2076 xprt);
2077
2078
2079
2080
2081
2082 if (!transport->inet)
2083 return;
2084 if (atomic_read(&xprt->swapper))
2085 sk_set_memalloc(transport->inet);
2086}
2087
2088
2089
2090
2091
2092
2093
2094
2095static int
2096xs_enable_swap(struct rpc_xprt *xprt)
2097{
2098 struct sock_xprt *xs = container_of(xprt, struct sock_xprt, xprt);
2099
2100 if (atomic_inc_return(&xprt->swapper) != 1)
2101 return 0;
2102 if (wait_on_bit_lock(&xprt->state, XPRT_LOCKED, TASK_KILLABLE))
2103 return -ERESTARTSYS;
2104 if (xs->inet)
2105 sk_set_memalloc(xs->inet);
2106 xprt_release_xprt(xprt, NULL);
2107 return 0;
2108}
2109
2110
2111
2112
2113
2114
2115
2116
2117static void
2118xs_disable_swap(struct rpc_xprt *xprt)
2119{
2120 struct sock_xprt *xs = container_of(xprt, struct sock_xprt, xprt);
2121
2122 if (!atomic_dec_and_test(&xprt->swapper))
2123 return;
2124 if (wait_on_bit_lock(&xprt->state, XPRT_LOCKED, TASK_KILLABLE))
2125 return;
2126 if (xs->inet)
2127 sk_clear_memalloc(xs->inet);
2128 xprt_release_xprt(xprt, NULL);
2129}
2130#else
2131static void xs_set_memalloc(struct rpc_xprt *xprt)
2132{
2133}
2134
2135static int
2136xs_enable_swap(struct rpc_xprt *xprt)
2137{
2138 return -EINVAL;
2139}
2140
2141static void
2142xs_disable_swap(struct rpc_xprt *xprt)
2143{
2144}
2145#endif
2146
2147static void xs_udp_finish_connecting(struct rpc_xprt *xprt, struct socket *sock)
2148{
2149 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2150
2151 if (!transport->inet) {
2152 struct sock *sk = sock->sk;
2153
2154 write_lock_bh(&sk->sk_callback_lock);
2155
2156 xs_save_old_callbacks(transport, sk);
2157
2158 sk->sk_user_data = xprt;
2159 sk->sk_data_ready = xs_data_ready;
2160 sk->sk_write_space = xs_udp_write_space;
2161 sock_set_flag(sk, SOCK_FASYNC);
2162 sk->sk_allocation = GFP_NOIO;
2163
2164 xprt_set_connected(xprt);
2165
2166
2167 transport->sock = sock;
2168 transport->inet = sk;
2169
2170 xs_set_memalloc(xprt);
2171
2172 write_unlock_bh(&sk->sk_callback_lock);
2173 }
2174 xs_udp_do_set_buffer_size(xprt);
2175
2176 xprt->stat.connect_start = jiffies;
2177}
2178
2179static void xs_udp_setup_socket(struct work_struct *work)
2180{
2181 struct sock_xprt *transport =
2182 container_of(work, struct sock_xprt, connect_worker.work);
2183 struct rpc_xprt *xprt = &transport->xprt;
2184 struct socket *sock = transport->sock;
2185 int status = -EIO;
2186
2187 sock = xs_create_sock(xprt, transport,
2188 xs_addr(xprt)->sa_family, SOCK_DGRAM,
2189 IPPROTO_UDP, false);
2190 if (IS_ERR(sock))
2191 goto out;
2192
2193 dprintk("RPC: worker connecting xprt %p via %s to "
2194 "%s (port %s)\n", xprt,
2195 xprt->address_strings[RPC_DISPLAY_PROTO],
2196 xprt->address_strings[RPC_DISPLAY_ADDR],
2197 xprt->address_strings[RPC_DISPLAY_PORT]);
2198
2199 xs_udp_finish_connecting(xprt, sock);
2200 trace_rpc_socket_connect(xprt, sock, 0);
2201 status = 0;
2202out:
2203 xprt_unlock_connect(xprt, transport);
2204 xprt_clear_connecting(xprt);
2205 xprt_wake_pending_tasks(xprt, status);
2206}
2207
2208
2209
2210
2211
2212
2213
2214
2215static void xs_tcp_shutdown(struct rpc_xprt *xprt)
2216{
2217 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2218 struct socket *sock = transport->sock;
2219
2220 if (sock == NULL)
2221 return;
2222 if (xprt_connected(xprt)) {
2223 kernel_sock_shutdown(sock, SHUT_RDWR);
2224 trace_rpc_socket_shutdown(xprt, sock);
2225 } else
2226 xs_reset_transport(transport);
2227}
2228
2229static int xs_tcp_finish_connecting(struct rpc_xprt *xprt, struct socket *sock)
2230{
2231 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2232 int ret = -ENOTCONN;
2233
2234 if (!transport->inet) {
2235 struct sock *sk = sock->sk;
2236 unsigned int keepidle = xprt->timeout->to_initval / HZ;
2237 unsigned int keepcnt = xprt->timeout->to_retries + 1;
2238 unsigned int opt_on = 1;
2239 unsigned int timeo;
2240 unsigned int addr_pref = IPV6_PREFER_SRC_PUBLIC;
2241
2242
2243 kernel_setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE,
2244 (char *)&opt_on, sizeof(opt_on));
2245 kernel_setsockopt(sock, SOL_TCP, TCP_KEEPIDLE,
2246 (char *)&keepidle, sizeof(keepidle));
2247 kernel_setsockopt(sock, SOL_TCP, TCP_KEEPINTVL,
2248 (char *)&keepidle, sizeof(keepidle));
2249 kernel_setsockopt(sock, SOL_TCP, TCP_KEEPCNT,
2250 (char *)&keepcnt, sizeof(keepcnt));
2251
2252
2253
2254
2255
2256
2257
2258
2259 kernel_setsockopt(sock, SOL_IPV6, IPV6_ADDR_PREFERENCES,
2260 (char *)&addr_pref, sizeof(addr_pref));
2261
2262
2263 timeo = jiffies_to_msecs(xprt->timeout->to_initval) *
2264 (xprt->timeout->to_retries + 1);
2265 kernel_setsockopt(sock, SOL_TCP, TCP_USER_TIMEOUT,
2266 (char *)&timeo, sizeof(timeo));
2267
2268 write_lock_bh(&sk->sk_callback_lock);
2269
2270 xs_save_old_callbacks(transport, sk);
2271
2272 sk->sk_user_data = xprt;
2273 sk->sk_data_ready = xs_data_ready;
2274 sk->sk_state_change = xs_tcp_state_change;
2275 sk->sk_write_space = xs_tcp_write_space;
2276 sock_set_flag(sk, SOCK_FASYNC);
2277 sk->sk_error_report = xs_error_report;
2278 sk->sk_allocation = GFP_NOIO;
2279
2280
2281 sock_reset_flag(sk, SOCK_LINGER);
2282 tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF;
2283
2284 xprt_clear_connected(xprt);
2285
2286
2287 transport->sock = sock;
2288 transport->inet = sk;
2289
2290 write_unlock_bh(&sk->sk_callback_lock);
2291 }
2292
2293 if (!xprt_bound(xprt))
2294 goto out;
2295
2296 xs_set_memalloc(xprt);
2297
2298
2299 xprt->stat.connect_count++;
2300 xprt->stat.connect_start = jiffies;
2301 set_bit(XPRT_SOCK_CONNECTING, &transport->sock_state);
2302 ret = kernel_connect(sock, xs_addr(xprt), xprt->addrlen, O_NONBLOCK);
2303 switch (ret) {
2304 case 0:
2305 xs_set_srcport(transport, sock);
2306 case -EINPROGRESS:
2307
2308 if (xprt->reestablish_timeout < XS_TCP_INIT_REEST_TO)
2309 xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
2310 break;
2311 case -EADDRNOTAVAIL:
2312
2313 transport->srcport = 0;
2314 }
2315out:
2316 return ret;
2317}
2318
2319
2320
2321
2322
2323
2324static void xs_tcp_setup_socket(struct work_struct *work)
2325{
2326 struct sock_xprt *transport =
2327 container_of(work, struct sock_xprt, connect_worker.work);
2328 struct socket *sock = transport->sock;
2329 struct rpc_xprt *xprt = &transport->xprt;
2330 int status = -EIO;
2331
2332 if (!sock) {
2333 sock = xs_create_sock(xprt, transport,
2334 xs_addr(xprt)->sa_family, SOCK_STREAM,
2335 IPPROTO_TCP, true);
2336 if (IS_ERR(sock)) {
2337 status = PTR_ERR(sock);
2338 goto out;
2339 }
2340 }
2341
2342 dprintk("RPC: worker connecting xprt %p via %s to "
2343 "%s (port %s)\n", xprt,
2344 xprt->address_strings[RPC_DISPLAY_PROTO],
2345 xprt->address_strings[RPC_DISPLAY_ADDR],
2346 xprt->address_strings[RPC_DISPLAY_PORT]);
2347
2348 status = xs_tcp_finish_connecting(xprt, sock);
2349 trace_rpc_socket_connect(xprt, sock, status);
2350 dprintk("RPC: %p connect status %d connected %d sock state %d\n",
2351 xprt, -status, xprt_connected(xprt),
2352 sock->sk->sk_state);
2353 switch (status) {
2354 default:
2355 printk("%s: connect returned unhandled error %d\n",
2356 __func__, status);
2357 case -EADDRNOTAVAIL:
2358
2359
2360
2361 xs_tcp_force_close(xprt);
2362 break;
2363 case 0:
2364 case -EINPROGRESS:
2365 case -EALREADY:
2366 xprt_unlock_connect(xprt, transport);
2367 return;
2368 case -EINVAL:
2369
2370
2371
2372 case -ECONNREFUSED:
2373 case -ECONNRESET:
2374 case -ENETUNREACH:
2375 case -EADDRINUSE:
2376 case -ENOBUFS:
2377
2378 xs_tcp_force_close(xprt);
2379 goto out;
2380 }
2381 status = -EAGAIN;
2382out:
2383 xprt_unlock_connect(xprt, transport);
2384 xprt_clear_connecting(xprt);
2385 xprt_wake_pending_tasks(xprt, status);
2386}
2387
2388static unsigned long xs_reconnect_delay(const struct rpc_xprt *xprt)
2389{
2390 unsigned long start, now = jiffies;
2391
2392 start = xprt->stat.connect_start + xprt->reestablish_timeout;
2393 if (time_after(start, now))
2394 return start - now;
2395 return 0;
2396}
2397
2398static void xs_reconnect_backoff(struct rpc_xprt *xprt)
2399{
2400 xprt->reestablish_timeout <<= 1;
2401 if (xprt->reestablish_timeout > xprt->max_reconnect_timeout)
2402 xprt->reestablish_timeout = xprt->max_reconnect_timeout;
2403 if (xprt->reestablish_timeout < XS_TCP_INIT_REEST_TO)
2404 xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
2405}
2406
2407
2408
2409
2410
2411
2412
2413
2414
2415
2416
2417
2418
2419
2420
2421static void xs_connect(struct rpc_xprt *xprt, struct rpc_task *task)
2422{
2423 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2424 unsigned long delay = 0;
2425
2426 WARN_ON_ONCE(!xprt_lock_connect(xprt, task, transport));
2427
2428 if (transport->sock != NULL) {
2429 dprintk("RPC: xs_connect delayed xprt %p for %lu "
2430 "seconds\n",
2431 xprt, xprt->reestablish_timeout / HZ);
2432
2433
2434 xs_reset_transport(transport);
2435
2436 delay = xs_reconnect_delay(xprt);
2437 xs_reconnect_backoff(xprt);
2438
2439 } else
2440 dprintk("RPC: xs_connect scheduled xprt %p\n", xprt);
2441
2442 queue_delayed_work(xprtiod_workqueue,
2443 &transport->connect_worker,
2444 delay);
2445}
2446
2447
2448
2449
2450
2451
2452
2453static void xs_local_print_stats(struct rpc_xprt *xprt, struct seq_file *seq)
2454{
2455 long idle_time = 0;
2456
2457 if (xprt_connected(xprt))
2458 idle_time = (long)(jiffies - xprt->last_used) / HZ;
2459
2460 seq_printf(seq, "\txprt:\tlocal %lu %lu %lu %ld %lu %lu %lu "
2461 "%llu %llu %lu %llu %llu\n",
2462 xprt->stat.bind_count,
2463 xprt->stat.connect_count,
2464 xprt->stat.connect_time,
2465 idle_time,
2466 xprt->stat.sends,
2467 xprt->stat.recvs,
2468 xprt->stat.bad_xids,
2469 xprt->stat.req_u,
2470 xprt->stat.bklog_u,
2471 xprt->stat.max_slots,
2472 xprt->stat.sending_u,
2473 xprt->stat.pending_u);
2474}
2475
2476
2477
2478
2479
2480
2481
2482static void xs_udp_print_stats(struct rpc_xprt *xprt, struct seq_file *seq)
2483{
2484 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2485
2486 seq_printf(seq, "\txprt:\tudp %u %lu %lu %lu %lu %llu %llu "
2487 "%lu %llu %llu\n",
2488 transport->srcport,
2489 xprt->stat.bind_count,
2490 xprt->stat.sends,
2491 xprt->stat.recvs,
2492 xprt->stat.bad_xids,
2493 xprt->stat.req_u,
2494 xprt->stat.bklog_u,
2495 xprt->stat.max_slots,
2496 xprt->stat.sending_u,
2497 xprt->stat.pending_u);
2498}
2499
2500
2501
2502
2503
2504
2505
2506static void xs_tcp_print_stats(struct rpc_xprt *xprt, struct seq_file *seq)
2507{
2508 struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
2509 long idle_time = 0;
2510
2511 if (xprt_connected(xprt))
2512 idle_time = (long)(jiffies - xprt->last_used) / HZ;
2513
2514 seq_printf(seq, "\txprt:\ttcp %u %lu %lu %lu %ld %lu %lu %lu "
2515 "%llu %llu %lu %llu %llu\n",
2516 transport->srcport,
2517 xprt->stat.bind_count,
2518 xprt->stat.connect_count,
2519 xprt->stat.connect_time,
2520 idle_time,
2521 xprt->stat.sends,
2522 xprt->stat.recvs,
2523 xprt->stat.bad_xids,
2524 xprt->stat.req_u,
2525 xprt->stat.bklog_u,
2526 xprt->stat.max_slots,
2527 xprt->stat.sending_u,
2528 xprt->stat.pending_u);
2529}
2530
2531
2532
2533
2534
2535
2536static void *bc_malloc(struct rpc_task *task, size_t size)
2537{
2538 struct page *page;
2539 struct rpc_buffer *buf;
2540
2541 WARN_ON_ONCE(size > PAGE_SIZE - sizeof(struct rpc_buffer));
2542 if (size > PAGE_SIZE - sizeof(struct rpc_buffer))
2543 return NULL;
2544
2545 page = alloc_page(GFP_KERNEL);
2546 if (!page)
2547 return NULL;
2548
2549 buf = page_address(page);
2550 buf->len = PAGE_SIZE;
2551
2552 return buf->data;
2553}
2554
2555
2556
2557
2558static void bc_free(void *buffer)
2559{
2560 struct rpc_buffer *buf;
2561
2562 if (!buffer)
2563 return;
2564
2565 buf = container_of(buffer, struct rpc_buffer, data);
2566 free_page((unsigned long)buf);
2567}
2568
2569
2570
2571
2572
2573static int bc_sendto(struct rpc_rqst *req)
2574{
2575 int len;
2576 struct xdr_buf *xbufp = &req->rq_snd_buf;
2577 struct rpc_xprt *xprt = req->rq_xprt;
2578 struct sock_xprt *transport =
2579 container_of(xprt, struct sock_xprt, xprt);
2580 struct socket *sock = transport->sock;
2581 unsigned long headoff;
2582 unsigned long tailoff;
2583
2584 xs_encode_stream_record_marker(xbufp);
2585
2586 tailoff = (unsigned long)xbufp->tail[0].iov_base & ~PAGE_MASK;
2587 headoff = (unsigned long)xbufp->head[0].iov_base & ~PAGE_MASK;
2588 len = svc_send_common(sock, xbufp,
2589 virt_to_page(xbufp->head[0].iov_base), headoff,
2590 xbufp->tail[0].iov_base, tailoff);
2591
2592 if (len != xbufp->len) {
2593 printk(KERN_NOTICE "Error sending entire callback!\n");
2594 len = -EAGAIN;
2595 }
2596
2597 return len;
2598}
2599
2600
2601
2602
2603static int bc_send_request(struct rpc_task *task)
2604{
2605 struct rpc_rqst *req = task->tk_rqstp;
2606 struct svc_xprt *xprt;
2607 int len;
2608
2609 dprintk("sending request with xid: %08x\n", ntohl(req->rq_xid));
2610
2611
2612
2613 xprt = req->rq_xprt->bc_xprt;
2614
2615
2616
2617
2618
2619 if (!mutex_trylock(&xprt->xpt_mutex)) {
2620 rpc_sleep_on(&xprt->xpt_bc_pending, task, NULL);
2621 if (!mutex_trylock(&xprt->xpt_mutex))
2622 return -EAGAIN;
2623 rpc_wake_up_queued_task(&xprt->xpt_bc_pending, task);
2624 }
2625 if (test_bit(XPT_DEAD, &xprt->xpt_flags))
2626 len = -ENOTCONN;
2627 else
2628 len = bc_sendto(req);
2629 mutex_unlock(&xprt->xpt_mutex);
2630
2631 if (len > 0)
2632 len = 0;
2633
2634 return len;
2635}
2636
2637
2638
2639
2640
2641static void bc_close(struct rpc_xprt *xprt)
2642{
2643}
2644
2645
2646
2647
2648
2649
2650static void bc_destroy(struct rpc_xprt *xprt)
2651{
2652 dprintk("RPC: bc_destroy xprt %p\n", xprt);
2653
2654 xs_xprt_free(xprt);
2655 module_put(THIS_MODULE);
2656}
2657
2658static struct rpc_xprt_ops xs_local_ops = {
2659 .reserve_xprt = xprt_reserve_xprt,
2660 .release_xprt = xs_tcp_release_xprt,
2661 .alloc_slot = xprt_alloc_slot,
2662 .rpcbind = xs_local_rpcbind,
2663 .set_port = xs_local_set_port,
2664 .connect = xs_local_connect,
2665 .buf_alloc = rpc_malloc,
2666 .buf_free = rpc_free,
2667 .send_request = xs_local_send_request,
2668 .set_retrans_timeout = xprt_set_retrans_timeout_def,
2669 .close = xs_close,
2670 .destroy = xs_destroy,
2671 .print_stats = xs_local_print_stats,
2672 .enable_swap = xs_enable_swap,
2673 .disable_swap = xs_disable_swap,
2674};
2675
2676static struct rpc_xprt_ops xs_udp_ops = {
2677 .set_buffer_size = xs_udp_set_buffer_size,
2678 .reserve_xprt = xprt_reserve_xprt_cong,
2679 .release_xprt = xprt_release_xprt_cong,
2680 .alloc_slot = xprt_alloc_slot,
2681 .rpcbind = rpcb_getport_async,
2682 .set_port = xs_set_port,
2683 .connect = xs_connect,
2684 .buf_alloc = rpc_malloc,
2685 .buf_free = rpc_free,
2686 .send_request = xs_udp_send_request,
2687 .set_retrans_timeout = xprt_set_retrans_timeout_rtt,
2688 .timer = xs_udp_timer,
2689 .release_request = xprt_release_rqst_cong,
2690 .close = xs_close,
2691 .destroy = xs_destroy,
2692 .print_stats = xs_udp_print_stats,
2693 .enable_swap = xs_enable_swap,
2694 .disable_swap = xs_disable_swap,
2695 .inject_disconnect = xs_inject_disconnect,
2696};
2697
2698static struct rpc_xprt_ops xs_tcp_ops = {
2699 .reserve_xprt = xprt_reserve_xprt,
2700 .release_xprt = xs_tcp_release_xprt,
2701 .alloc_slot = xprt_lock_and_alloc_slot,
2702 .rpcbind = rpcb_getport_async,
2703 .set_port = xs_set_port,
2704 .connect = xs_connect,
2705 .buf_alloc = rpc_malloc,
2706 .buf_free = rpc_free,
2707 .send_request = xs_tcp_send_request,
2708 .set_retrans_timeout = xprt_set_retrans_timeout_def,
2709 .close = xs_tcp_shutdown,
2710 .destroy = xs_destroy,
2711 .print_stats = xs_tcp_print_stats,
2712 .enable_swap = xs_enable_swap,
2713 .disable_swap = xs_disable_swap,
2714 .inject_disconnect = xs_inject_disconnect,
2715#ifdef CONFIG_SUNRPC_BACKCHANNEL
2716 .bc_setup = xprt_setup_bc,
2717 .bc_up = xs_tcp_bc_up,
2718 .bc_maxpayload = xs_tcp_bc_maxpayload,
2719 .bc_free_rqst = xprt_free_bc_rqst,
2720 .bc_destroy = xprt_destroy_bc,
2721#endif
2722};
2723
2724
2725
2726
2727
2728static struct rpc_xprt_ops bc_tcp_ops = {
2729 .reserve_xprt = xprt_reserve_xprt,
2730 .release_xprt = xprt_release_xprt,
2731 .alloc_slot = xprt_alloc_slot,
2732 .buf_alloc = bc_malloc,
2733 .buf_free = bc_free,
2734 .send_request = bc_send_request,
2735 .set_retrans_timeout = xprt_set_retrans_timeout_def,
2736 .close = bc_close,
2737 .destroy = bc_destroy,
2738 .print_stats = xs_tcp_print_stats,
2739 .enable_swap = xs_enable_swap,
2740 .disable_swap = xs_disable_swap,
2741 .inject_disconnect = xs_inject_disconnect,
2742};
2743
2744static int xs_init_anyaddr(const int family, struct sockaddr *sap)
2745{
2746 static const struct sockaddr_in sin = {
2747 .sin_family = AF_INET,
2748 .sin_addr.s_addr = htonl(INADDR_ANY),
2749 };
2750 static const struct sockaddr_in6 sin6 = {
2751 .sin6_family = AF_INET6,
2752 .sin6_addr = IN6ADDR_ANY_INIT,
2753 };
2754
2755 switch (family) {
2756 case AF_LOCAL:
2757 break;
2758 case AF_INET:
2759 memcpy(sap, &sin, sizeof(sin));
2760 break;
2761 case AF_INET6:
2762 memcpy(sap, &sin6, sizeof(sin6));
2763 break;
2764 default:
2765 dprintk("RPC: %s: Bad address family\n", __func__);
2766 return -EAFNOSUPPORT;
2767 }
2768 return 0;
2769}
2770
2771static struct rpc_xprt *xs_setup_xprt(struct xprt_create *args,
2772 unsigned int slot_table_size,
2773 unsigned int max_slot_table_size)
2774{
2775 struct rpc_xprt *xprt;
2776 struct sock_xprt *new;
2777
2778 if (args->addrlen > sizeof(xprt->addr)) {
2779 dprintk("RPC: xs_setup_xprt: address too large\n");
2780 return ERR_PTR(-EBADF);
2781 }
2782
2783 xprt = xprt_alloc(args->net, sizeof(*new), slot_table_size,
2784 max_slot_table_size);
2785 if (xprt == NULL) {
2786 dprintk("RPC: xs_setup_xprt: couldn't allocate "
2787 "rpc_xprt\n");
2788 return ERR_PTR(-ENOMEM);
2789 }
2790
2791 new = container_of(xprt, struct sock_xprt, xprt);
2792 mutex_init(&new->recv_mutex);
2793 memcpy(&xprt->addr, args->dstaddr, args->addrlen);
2794 xprt->addrlen = args->addrlen;
2795 if (args->srcaddr)
2796 memcpy(&new->srcaddr, args->srcaddr, args->addrlen);
2797 else {
2798 int err;
2799 err = xs_init_anyaddr(args->dstaddr->sa_family,
2800 (struct sockaddr *)&new->srcaddr);
2801 if (err != 0) {
2802 xprt_free(xprt);
2803 return ERR_PTR(err);
2804 }
2805 }
2806
2807 return xprt;
2808}
2809
2810static const struct rpc_timeout xs_local_default_timeout = {
2811 .to_initval = 10 * HZ,
2812 .to_maxval = 10 * HZ,
2813 .to_retries = 2,
2814};
2815
2816
2817
2818
2819
2820
2821
2822static struct rpc_xprt *xs_setup_local(struct xprt_create *args)
2823{
2824 struct sockaddr_un *sun = (struct sockaddr_un *)args->dstaddr;
2825 struct sock_xprt *transport;
2826 struct rpc_xprt *xprt;
2827 struct rpc_xprt *ret;
2828
2829 xprt = xs_setup_xprt(args, xprt_tcp_slot_table_entries,
2830 xprt_max_tcp_slot_table_entries);
2831 if (IS_ERR(xprt))
2832 return xprt;
2833 transport = container_of(xprt, struct sock_xprt, xprt);
2834
2835 xprt->prot = 0;
2836 xprt->tsh_size = sizeof(rpc_fraghdr) / sizeof(u32);
2837 xprt->max_payload = RPC_MAX_FRAGMENT_SIZE;
2838
2839 xprt->bind_timeout = XS_BIND_TO;
2840 xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
2841 xprt->idle_timeout = XS_IDLE_DISC_TO;
2842
2843 xprt->ops = &xs_local_ops;
2844 xprt->timeout = &xs_local_default_timeout;
2845
2846 INIT_WORK(&transport->recv_worker, xs_local_data_receive_workfn);
2847 INIT_DELAYED_WORK(&transport->connect_worker,
2848 xs_dummy_setup_socket);
2849
2850 switch (sun->sun_family) {
2851 case AF_LOCAL:
2852 if (sun->sun_path[0] != '/') {
2853 dprintk("RPC: bad AF_LOCAL address: %s\n",
2854 sun->sun_path);
2855 ret = ERR_PTR(-EINVAL);
2856 goto out_err;
2857 }
2858 xprt_set_bound(xprt);
2859 xs_format_peer_addresses(xprt, "local", RPCBIND_NETID_LOCAL);
2860 ret = ERR_PTR(xs_local_setup_socket(transport));
2861 if (ret)
2862 goto out_err;
2863 break;
2864 default:
2865 ret = ERR_PTR(-EAFNOSUPPORT);
2866 goto out_err;
2867 }
2868
2869 dprintk("RPC: set up xprt to %s via AF_LOCAL\n",
2870 xprt->address_strings[RPC_DISPLAY_ADDR]);
2871
2872 if (try_module_get(THIS_MODULE))
2873 return xprt;
2874 ret = ERR_PTR(-EINVAL);
2875out_err:
2876 xs_xprt_free(xprt);
2877 return ret;
2878}
2879
2880static const struct rpc_timeout xs_udp_default_timeout = {
2881 .to_initval = 5 * HZ,
2882 .to_maxval = 30 * HZ,
2883 .to_increment = 5 * HZ,
2884 .to_retries = 5,
2885};
2886
2887
2888
2889
2890
2891
2892static struct rpc_xprt *xs_setup_udp(struct xprt_create *args)
2893{
2894 struct sockaddr *addr = args->dstaddr;
2895 struct rpc_xprt *xprt;
2896 struct sock_xprt *transport;
2897 struct rpc_xprt *ret;
2898
2899 xprt = xs_setup_xprt(args, xprt_udp_slot_table_entries,
2900 xprt_udp_slot_table_entries);
2901 if (IS_ERR(xprt))
2902 return xprt;
2903 transport = container_of(xprt, struct sock_xprt, xprt);
2904
2905 xprt->prot = IPPROTO_UDP;
2906 xprt->tsh_size = 0;
2907
2908 xprt->max_payload = (1U << 16) - (MAX_HEADER << 3);
2909
2910 xprt->bind_timeout = XS_BIND_TO;
2911 xprt->reestablish_timeout = XS_UDP_REEST_TO;
2912 xprt->idle_timeout = XS_IDLE_DISC_TO;
2913
2914 xprt->ops = &xs_udp_ops;
2915
2916 xprt->timeout = &xs_udp_default_timeout;
2917
2918 INIT_WORK(&transport->recv_worker, xs_udp_data_receive_workfn);
2919 INIT_DELAYED_WORK(&transport->connect_worker, xs_udp_setup_socket);
2920
2921 switch (addr->sa_family) {
2922 case AF_INET:
2923 if (((struct sockaddr_in *)addr)->sin_port != htons(0))
2924 xprt_set_bound(xprt);
2925
2926 xs_format_peer_addresses(xprt, "udp", RPCBIND_NETID_UDP);
2927 break;
2928 case AF_INET6:
2929 if (((struct sockaddr_in6 *)addr)->sin6_port != htons(0))
2930 xprt_set_bound(xprt);
2931
2932 xs_format_peer_addresses(xprt, "udp", RPCBIND_NETID_UDP6);
2933 break;
2934 default:
2935 ret = ERR_PTR(-EAFNOSUPPORT);
2936 goto out_err;
2937 }
2938
2939 if (xprt_bound(xprt))
2940 dprintk("RPC: set up xprt to %s (port %s) via %s\n",
2941 xprt->address_strings[RPC_DISPLAY_ADDR],
2942 xprt->address_strings[RPC_DISPLAY_PORT],
2943 xprt->address_strings[RPC_DISPLAY_PROTO]);
2944 else
2945 dprintk("RPC: set up xprt to %s (autobind) via %s\n",
2946 xprt->address_strings[RPC_DISPLAY_ADDR],
2947 xprt->address_strings[RPC_DISPLAY_PROTO]);
2948
2949 if (try_module_get(THIS_MODULE))
2950 return xprt;
2951 ret = ERR_PTR(-EINVAL);
2952out_err:
2953 xs_xprt_free(xprt);
2954 return ret;
2955}
2956
2957static const struct rpc_timeout xs_tcp_default_timeout = {
2958 .to_initval = 60 * HZ,
2959 .to_maxval = 60 * HZ,
2960 .to_retries = 2,
2961};
2962
2963
2964
2965
2966
2967
2968static struct rpc_xprt *xs_setup_tcp(struct xprt_create *args)
2969{
2970 struct sockaddr *addr = args->dstaddr;
2971 struct rpc_xprt *xprt;
2972 struct sock_xprt *transport;
2973 struct rpc_xprt *ret;
2974 unsigned int max_slot_table_size = xprt_max_tcp_slot_table_entries;
2975
2976 if (args->flags & XPRT_CREATE_INFINITE_SLOTS)
2977 max_slot_table_size = RPC_MAX_SLOT_TABLE_LIMIT;
2978
2979 xprt = xs_setup_xprt(args, xprt_tcp_slot_table_entries,
2980 max_slot_table_size);
2981 if (IS_ERR(xprt))
2982 return xprt;
2983 transport = container_of(xprt, struct sock_xprt, xprt);
2984
2985 xprt->prot = IPPROTO_TCP;
2986 xprt->tsh_size = sizeof(rpc_fraghdr) / sizeof(u32);
2987 xprt->max_payload = RPC_MAX_FRAGMENT_SIZE;
2988
2989 xprt->bind_timeout = XS_BIND_TO;
2990 xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
2991 xprt->idle_timeout = XS_IDLE_DISC_TO;
2992
2993 xprt->ops = &xs_tcp_ops;
2994 xprt->timeout = &xs_tcp_default_timeout;
2995
2996 xprt->max_reconnect_timeout = xprt->timeout->to_maxval;
2997
2998 INIT_WORK(&transport->recv_worker, xs_tcp_data_receive_workfn);
2999 INIT_DELAYED_WORK(&transport->connect_worker, xs_tcp_setup_socket);
3000
3001 switch (addr->sa_family) {
3002 case AF_INET:
3003 if (((struct sockaddr_in *)addr)->sin_port != htons(0))
3004 xprt_set_bound(xprt);
3005
3006 xs_format_peer_addresses(xprt, "tcp", RPCBIND_NETID_TCP);
3007 break;
3008 case AF_INET6:
3009 if (((struct sockaddr_in6 *)addr)->sin6_port != htons(0))
3010 xprt_set_bound(xprt);
3011
3012 xs_format_peer_addresses(xprt, "tcp", RPCBIND_NETID_TCP6);
3013 break;
3014 default:
3015 ret = ERR_PTR(-EAFNOSUPPORT);
3016 goto out_err;
3017 }
3018
3019 if (xprt_bound(xprt))
3020 dprintk("RPC: set up xprt to %s (port %s) via %s\n",
3021 xprt->address_strings[RPC_DISPLAY_ADDR],
3022 xprt->address_strings[RPC_DISPLAY_PORT],
3023 xprt->address_strings[RPC_DISPLAY_PROTO]);
3024 else
3025 dprintk("RPC: set up xprt to %s (autobind) via %s\n",
3026 xprt->address_strings[RPC_DISPLAY_ADDR],
3027 xprt->address_strings[RPC_DISPLAY_PROTO]);
3028
3029 if (try_module_get(THIS_MODULE))
3030 return xprt;
3031 ret = ERR_PTR(-EINVAL);
3032out_err:
3033 xs_xprt_free(xprt);
3034 return ret;
3035}
3036
3037
3038
3039
3040
3041
3042static struct rpc_xprt *xs_setup_bc_tcp(struct xprt_create *args)
3043{
3044 struct sockaddr *addr = args->dstaddr;
3045 struct rpc_xprt *xprt;
3046 struct sock_xprt *transport;
3047 struct svc_sock *bc_sock;
3048 struct rpc_xprt *ret;
3049
3050 xprt = xs_setup_xprt(args, xprt_tcp_slot_table_entries,
3051 xprt_tcp_slot_table_entries);
3052 if (IS_ERR(xprt))
3053 return xprt;
3054 transport = container_of(xprt, struct sock_xprt, xprt);
3055
3056 xprt->prot = IPPROTO_TCP;
3057 xprt->tsh_size = sizeof(rpc_fraghdr) / sizeof(u32);
3058 xprt->max_payload = RPC_MAX_FRAGMENT_SIZE;
3059 xprt->timeout = &xs_tcp_default_timeout;
3060
3061
3062 xprt_set_bound(xprt);
3063 xprt->bind_timeout = 0;
3064 xprt->reestablish_timeout = 0;
3065 xprt->idle_timeout = 0;
3066
3067 xprt->ops = &bc_tcp_ops;
3068
3069 switch (addr->sa_family) {
3070 case AF_INET:
3071 xs_format_peer_addresses(xprt, "tcp",
3072 RPCBIND_NETID_TCP);
3073 break;
3074 case AF_INET6:
3075 xs_format_peer_addresses(xprt, "tcp",
3076 RPCBIND_NETID_TCP6);
3077 break;
3078 default:
3079 ret = ERR_PTR(-EAFNOSUPPORT);
3080 goto out_err;
3081 }
3082
3083 dprintk("RPC: set up xprt to %s (port %s) via %s\n",
3084 xprt->address_strings[RPC_DISPLAY_ADDR],
3085 xprt->address_strings[RPC_DISPLAY_PORT],
3086 xprt->address_strings[RPC_DISPLAY_PROTO]);
3087
3088
3089
3090
3091
3092
3093
3094 xprt_get(xprt);
3095 args->bc_xprt->xpt_bc_xprt = xprt;
3096 xprt->bc_xprt = args->bc_xprt;
3097 bc_sock = container_of(args->bc_xprt, struct svc_sock, sk_xprt);
3098 transport->sock = bc_sock->sk_sock;
3099 transport->inet = bc_sock->sk_sk;
3100
3101
3102
3103
3104
3105 xprt_set_connected(xprt);
3106
3107 if (try_module_get(THIS_MODULE))
3108 return xprt;
3109
3110 args->bc_xprt->xpt_bc_xprt = NULL;
3111 args->bc_xprt->xpt_bc_xps = NULL;
3112 xprt_put(xprt);
3113 ret = ERR_PTR(-EINVAL);
3114out_err:
3115 xs_xprt_free(xprt);
3116 return ret;
3117}
3118
3119static struct xprt_class xs_local_transport = {
3120 .list = LIST_HEAD_INIT(xs_local_transport.list),
3121 .name = "named UNIX socket",
3122 .owner = THIS_MODULE,
3123 .ident = XPRT_TRANSPORT_LOCAL,
3124 .setup = xs_setup_local,
3125};
3126
3127static struct xprt_class xs_udp_transport = {
3128 .list = LIST_HEAD_INIT(xs_udp_transport.list),
3129 .name = "udp",
3130 .owner = THIS_MODULE,
3131 .ident = XPRT_TRANSPORT_UDP,
3132 .setup = xs_setup_udp,
3133};
3134
3135static struct xprt_class xs_tcp_transport = {
3136 .list = LIST_HEAD_INIT(xs_tcp_transport.list),
3137 .name = "tcp",
3138 .owner = THIS_MODULE,
3139 .ident = XPRT_TRANSPORT_TCP,
3140 .setup = xs_setup_tcp,
3141};
3142
3143static struct xprt_class xs_bc_tcp_transport = {
3144 .list = LIST_HEAD_INIT(xs_bc_tcp_transport.list),
3145 .name = "tcp NFSv4.1 backchannel",
3146 .owner = THIS_MODULE,
3147 .ident = XPRT_TRANSPORT_BC_TCP,
3148 .setup = xs_setup_bc_tcp,
3149};
3150
3151
3152
3153
3154
3155int init_socket_xprt(void)
3156{
3157#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
3158 if (!sunrpc_table_header)
3159 sunrpc_table_header = register_sysctl_table(sunrpc_table);
3160#endif
3161
3162 xprt_register_transport(&xs_local_transport);
3163 xprt_register_transport(&xs_udp_transport);
3164 xprt_register_transport(&xs_tcp_transport);
3165 xprt_register_transport(&xs_bc_tcp_transport);
3166
3167 return 0;
3168}
3169
3170
3171
3172
3173
3174void cleanup_socket_xprt(void)
3175{
3176#if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
3177 if (sunrpc_table_header) {
3178 unregister_sysctl_table(sunrpc_table_header);
3179 sunrpc_table_header = NULL;
3180 }
3181#endif
3182
3183 xprt_unregister_transport(&xs_local_transport);
3184 xprt_unregister_transport(&xs_udp_transport);
3185 xprt_unregister_transport(&xs_tcp_transport);
3186 xprt_unregister_transport(&xs_bc_tcp_transport);
3187}
3188
3189static int param_set_uint_minmax(const char *val,
3190 const struct kernel_param *kp,
3191 unsigned int min, unsigned int max)
3192{
3193 unsigned int num;
3194 int ret;
3195
3196 if (!val)
3197 return -EINVAL;
3198 ret = kstrtouint(val, 0, &num);
3199 if (ret == -EINVAL || num < min || num > max)
3200 return -EINVAL;
3201 *((unsigned int *)kp->arg) = num;
3202 return 0;
3203}
3204
3205static int param_set_portnr(const char *val, const struct kernel_param *kp)
3206{
3207 if (kp->arg == &xprt_min_resvport)
3208 return param_set_uint_minmax(val, kp,
3209 RPC_MIN_RESVPORT,
3210 xprt_max_resvport);
3211 return param_set_uint_minmax(val, kp,
3212 xprt_min_resvport,
3213 RPC_MAX_RESVPORT);
3214}
3215
3216static const struct kernel_param_ops param_ops_portnr = {
3217 .set = param_set_portnr,
3218 .get = param_get_uint,
3219};
3220
3221#define param_check_portnr(name, p) \
3222 __param_check(name, p, unsigned int);
3223
3224module_param_named(min_resvport, xprt_min_resvport, portnr, 0644);
3225module_param_named(max_resvport, xprt_max_resvport, portnr, 0644);
3226
3227static int param_set_slot_table_size(const char *val,
3228 const struct kernel_param *kp)
3229{
3230 return param_set_uint_minmax(val, kp,
3231 RPC_MIN_SLOT_TABLE,
3232 RPC_MAX_SLOT_TABLE);
3233}
3234
3235static const struct kernel_param_ops param_ops_slot_table_size = {
3236 .set = param_set_slot_table_size,
3237 .get = param_get_uint,
3238};
3239
3240#define param_check_slot_table_size(name, p) \
3241 __param_check(name, p, unsigned int);
3242
3243static int param_set_max_slot_table_size(const char *val,
3244 const struct kernel_param *kp)
3245{
3246 return param_set_uint_minmax(val, kp,
3247 RPC_MIN_SLOT_TABLE,
3248 RPC_MAX_SLOT_TABLE_LIMIT);
3249}
3250
3251static const struct kernel_param_ops param_ops_max_slot_table_size = {
3252 .set = param_set_max_slot_table_size,
3253 .get = param_get_uint,
3254};
3255
3256#define param_check_max_slot_table_size(name, p) \
3257 __param_check(name, p, unsigned int);
3258
3259module_param_named(tcp_slot_table_entries, xprt_tcp_slot_table_entries,
3260 slot_table_size, 0644);
3261module_param_named(tcp_max_slot_table_entries, xprt_max_tcp_slot_table_entries,
3262 max_slot_table_size, 0644);
3263module_param_named(udp_slot_table_entries, xprt_udp_slot_table_entries,
3264 slot_table_size, 0644);
3265
3266