xprtsock.c 40 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522
  1. /*
  2. * linux/net/sunrpc/xprtsock.c
  3. *
  4. * Client-side transport implementation for sockets.
  5. *
  6. * TCP callback races fixes (C) 1998 Red Hat Software <alan@redhat.com>
  7. * TCP send fixes (C) 1998 Red Hat Software <alan@redhat.com>
  8. * TCP NFS related read + write fixes
  9. * (C) 1999 Dave Airlie, University of Limerick, Ireland <airlied@linux.ie>
  10. *
  11. * Rewrite of larges part of the code in order to stabilize TCP stuff.
  12. * Fix behaviour when socket buffer is full.
  13. * (C) 1999 Trond Myklebust <trond.myklebust@fys.uio.no>
  14. *
  15. * IP socket transport implementation, (C) 2005 Chuck Lever <cel@netapp.com>
  16. */
  17. #include <linux/types.h>
  18. #include <linux/slab.h>
  19. #include <linux/capability.h>
  20. #include <linux/sched.h>
  21. #include <linux/pagemap.h>
  22. #include <linux/errno.h>
  23. #include <linux/socket.h>
  24. #include <linux/in.h>
  25. #include <linux/net.h>
  26. #include <linux/mm.h>
  27. #include <linux/udp.h>
  28. #include <linux/tcp.h>
  29. #include <linux/sunrpc/clnt.h>
  30. #include <linux/sunrpc/sched.h>
  31. #include <linux/file.h>
  32. #include <net/sock.h>
  33. #include <net/checksum.h>
  34. #include <net/udp.h>
  35. #include <net/tcp.h>
  36. /*
  37. * xprtsock tunables
  38. */
  39. unsigned int xprt_udp_slot_table_entries = RPC_DEF_SLOT_TABLE;
  40. unsigned int xprt_tcp_slot_table_entries = RPC_DEF_SLOT_TABLE;
  41. unsigned int xprt_min_resvport = RPC_DEF_MIN_RESVPORT;
  42. unsigned int xprt_max_resvport = RPC_DEF_MAX_RESVPORT;
  43. /*
  44. * How many times to try sending a request on a socket before waiting
  45. * for the socket buffer to clear.
  46. */
  47. #define XS_SENDMSG_RETRY (10U)
  48. /*
  49. * Time out for an RPC UDP socket connect. UDP socket connects are
  50. * synchronous, but we set a timeout anyway in case of resource
  51. * exhaustion on the local host.
  52. */
  53. #define XS_UDP_CONN_TO (5U * HZ)
  54. /*
  55. * Wait duration for an RPC TCP connection to be established. Solaris
  56. * NFS over TCP uses 60 seconds, for example, which is in line with how
  57. * long a server takes to reboot.
  58. */
  59. #define XS_TCP_CONN_TO (60U * HZ)
  60. /*
  61. * Wait duration for a reply from the RPC portmapper.
  62. */
  63. #define XS_BIND_TO (60U * HZ)
  64. /*
  65. * Delay if a UDP socket connect error occurs. This is most likely some
  66. * kind of resource problem on the local host.
  67. */
  68. #define XS_UDP_REEST_TO (2U * HZ)
  69. /*
  70. * The reestablish timeout allows clients to delay for a bit before attempting
  71. * to reconnect to a server that just dropped our connection.
  72. *
  73. * We implement an exponential backoff when trying to reestablish a TCP
  74. * transport connection with the server. Some servers like to drop a TCP
  75. * connection when they are overworked, so we start with a short timeout and
  76. * increase over time if the server is down or not responding.
  77. */
  78. #define XS_TCP_INIT_REEST_TO (3U * HZ)
  79. #define XS_TCP_MAX_REEST_TO (5U * 60 * HZ)
  80. /*
  81. * TCP idle timeout; client drops the transport socket if it is idle
  82. * for this long. Note that we also timeout UDP sockets to prevent
  83. * holding port numbers when there is no RPC traffic.
  84. */
  85. #define XS_IDLE_DISC_TO (5U * 60 * HZ)
  86. #ifdef RPC_DEBUG
  87. # undef RPC_DEBUG_DATA
  88. # define RPCDBG_FACILITY RPCDBG_TRANS
  89. #endif
  90. #ifdef RPC_DEBUG_DATA
  91. static void xs_pktdump(char *msg, u32 *packet, unsigned int count)
  92. {
  93. u8 *buf = (u8 *) packet;
  94. int j;
  95. dprintk("RPC: %s\n", msg);
  96. for (j = 0; j < count && j < 128; j += 4) {
  97. if (!(j & 31)) {
  98. if (j)
  99. dprintk("\n");
  100. dprintk("0x%04x ", j);
  101. }
  102. dprintk("%02x%02x%02x%02x ",
  103. buf[j], buf[j+1], buf[j+2], buf[j+3]);
  104. }
  105. dprintk("\n");
  106. }
  107. #else
  108. static inline void xs_pktdump(char *msg, u32 *packet, unsigned int count)
  109. {
  110. /* NOP */
  111. }
  112. #endif
  113. struct sock_xprt {
  114. struct rpc_xprt xprt;
  115. /*
  116. * Network layer
  117. */
  118. struct socket * sock;
  119. struct sock * inet;
  120. /*
  121. * State of TCP reply receive
  122. */
  123. __be32 tcp_fraghdr,
  124. tcp_xid;
  125. u32 tcp_offset,
  126. tcp_reclen;
  127. unsigned long tcp_copied,
  128. tcp_flags;
  129. /*
  130. * Connection of transports
  131. */
  132. struct work_struct connect_worker;
  133. unsigned short port;
  134. /*
  135. * UDP socket buffer size parameters
  136. */
  137. size_t rcvsize,
  138. sndsize;
  139. /*
  140. * Saved socket callback addresses
  141. */
  142. void (*old_data_ready)(struct sock *, int);
  143. void (*old_state_change)(struct sock *);
  144. void (*old_write_space)(struct sock *);
  145. };
  146. /*
  147. * TCP receive state flags
  148. */
  149. #define TCP_RCV_LAST_FRAG (1UL << 0)
  150. #define TCP_RCV_COPY_FRAGHDR (1UL << 1)
  151. #define TCP_RCV_COPY_XID (1UL << 2)
  152. #define TCP_RCV_COPY_DATA (1UL << 3)
  153. static void xs_format_peer_addresses(struct rpc_xprt *xprt)
  154. {
  155. struct sockaddr_in *addr = (struct sockaddr_in *) &xprt->addr;
  156. char *buf;
  157. buf = kzalloc(20, GFP_KERNEL);
  158. if (buf) {
  159. snprintf(buf, 20, "%u.%u.%u.%u",
  160. NIPQUAD(addr->sin_addr.s_addr));
  161. }
  162. xprt->address_strings[RPC_DISPLAY_ADDR] = buf;
  163. buf = kzalloc(8, GFP_KERNEL);
  164. if (buf) {
  165. snprintf(buf, 8, "%u",
  166. ntohs(addr->sin_port));
  167. }
  168. xprt->address_strings[RPC_DISPLAY_PORT] = buf;
  169. if (xprt->prot == IPPROTO_UDP)
  170. xprt->address_strings[RPC_DISPLAY_PROTO] = "udp";
  171. else
  172. xprt->address_strings[RPC_DISPLAY_PROTO] = "tcp";
  173. buf = kzalloc(48, GFP_KERNEL);
  174. if (buf) {
  175. snprintf(buf, 48, "addr=%u.%u.%u.%u port=%u proto=%s",
  176. NIPQUAD(addr->sin_addr.s_addr),
  177. ntohs(addr->sin_port),
  178. xprt->prot == IPPROTO_UDP ? "udp" : "tcp");
  179. }
  180. xprt->address_strings[RPC_DISPLAY_ALL] = buf;
  181. }
  182. static void xs_free_peer_addresses(struct rpc_xprt *xprt)
  183. {
  184. kfree(xprt->address_strings[RPC_DISPLAY_ADDR]);
  185. kfree(xprt->address_strings[RPC_DISPLAY_PORT]);
  186. kfree(xprt->address_strings[RPC_DISPLAY_ALL]);
  187. }
  188. #define XS_SENDMSG_FLAGS (MSG_DONTWAIT | MSG_NOSIGNAL)
  189. static int xs_send_kvec(struct socket *sock, struct sockaddr *addr, int addrlen, struct kvec *vec, unsigned int base, int more)
  190. {
  191. struct msghdr msg = {
  192. .msg_name = addr,
  193. .msg_namelen = addrlen,
  194. .msg_flags = XS_SENDMSG_FLAGS | (more ? MSG_MORE : 0),
  195. };
  196. struct kvec iov = {
  197. .iov_base = vec->iov_base + base,
  198. .iov_len = vec->iov_len - base,
  199. };
  200. if (iov.iov_len != 0)
  201. return kernel_sendmsg(sock, &msg, &iov, 1, iov.iov_len);
  202. return kernel_sendmsg(sock, &msg, NULL, 0, 0);
  203. }
  204. static int xs_send_pagedata(struct socket *sock, struct xdr_buf *xdr, unsigned int base, int more)
  205. {
  206. struct page **ppage;
  207. unsigned int remainder;
  208. int err, sent = 0;
  209. remainder = xdr->page_len - base;
  210. base += xdr->page_base;
  211. ppage = xdr->pages + (base >> PAGE_SHIFT);
  212. base &= ~PAGE_MASK;
  213. for(;;) {
  214. unsigned int len = min_t(unsigned int, PAGE_SIZE - base, remainder);
  215. int flags = XS_SENDMSG_FLAGS;
  216. remainder -= len;
  217. if (remainder != 0 || more)
  218. flags |= MSG_MORE;
  219. err = sock->ops->sendpage(sock, *ppage, base, len, flags);
  220. if (remainder == 0 || err != len)
  221. break;
  222. sent += err;
  223. ppage++;
  224. base = 0;
  225. }
  226. if (sent == 0)
  227. return err;
  228. if (err > 0)
  229. sent += err;
  230. return sent;
  231. }
  232. /**
  233. * xs_sendpages - write pages directly to a socket
  234. * @sock: socket to send on
  235. * @addr: UDP only -- address of destination
  236. * @addrlen: UDP only -- length of destination address
  237. * @xdr: buffer containing this request
  238. * @base: starting position in the buffer
  239. *
  240. */
  241. static int xs_sendpages(struct socket *sock, struct sockaddr *addr, int addrlen, struct xdr_buf *xdr, unsigned int base)
  242. {
  243. unsigned int remainder = xdr->len - base;
  244. int err, sent = 0;
  245. if (unlikely(!sock))
  246. return -ENOTCONN;
  247. clear_bit(SOCK_ASYNC_NOSPACE, &sock->flags);
  248. if (base != 0) {
  249. addr = NULL;
  250. addrlen = 0;
  251. }
  252. if (base < xdr->head[0].iov_len || addr != NULL) {
  253. unsigned int len = xdr->head[0].iov_len - base;
  254. remainder -= len;
  255. err = xs_send_kvec(sock, addr, addrlen, &xdr->head[0], base, remainder != 0);
  256. if (remainder == 0 || err != len)
  257. goto out;
  258. sent += err;
  259. base = 0;
  260. } else
  261. base -= xdr->head[0].iov_len;
  262. if (base < xdr->page_len) {
  263. unsigned int len = xdr->page_len - base;
  264. remainder -= len;
  265. err = xs_send_pagedata(sock, xdr, base, remainder != 0);
  266. if (remainder == 0 || err != len)
  267. goto out;
  268. sent += err;
  269. base = 0;
  270. } else
  271. base -= xdr->page_len;
  272. if (base >= xdr->tail[0].iov_len)
  273. return sent;
  274. err = xs_send_kvec(sock, NULL, 0, &xdr->tail[0], base, 0);
  275. out:
  276. if (sent == 0)
  277. return err;
  278. if (err > 0)
  279. sent += err;
  280. return sent;
  281. }
  282. /**
  283. * xs_nospace - place task on wait queue if transmit was incomplete
  284. * @task: task to put to sleep
  285. *
  286. */
  287. static void xs_nospace(struct rpc_task *task)
  288. {
  289. struct rpc_rqst *req = task->tk_rqstp;
  290. struct rpc_xprt *xprt = req->rq_xprt;
  291. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  292. dprintk("RPC: %4d xmit incomplete (%u left of %u)\n",
  293. task->tk_pid, req->rq_slen - req->rq_bytes_sent,
  294. req->rq_slen);
  295. if (test_bit(SOCK_ASYNC_NOSPACE, &transport->sock->flags)) {
  296. /* Protect against races with write_space */
  297. spin_lock_bh(&xprt->transport_lock);
  298. /* Don't race with disconnect */
  299. if (!xprt_connected(xprt))
  300. task->tk_status = -ENOTCONN;
  301. else if (test_bit(SOCK_NOSPACE, &transport->sock->flags))
  302. xprt_wait_for_buffer_space(task);
  303. spin_unlock_bh(&xprt->transport_lock);
  304. } else
  305. /* Keep holding the socket if it is blocked */
  306. rpc_delay(task, HZ>>4);
  307. }
  308. /**
  309. * xs_udp_send_request - write an RPC request to a UDP socket
  310. * @task: address of RPC task that manages the state of an RPC request
  311. *
  312. * Return values:
  313. * 0: The request has been sent
  314. * EAGAIN: The socket was blocked, please call again later to
  315. * complete the request
  316. * ENOTCONN: Caller needs to invoke connect logic then call again
  317. * other: Some other error occured, the request was not sent
  318. */
  319. static int xs_udp_send_request(struct rpc_task *task)
  320. {
  321. struct rpc_rqst *req = task->tk_rqstp;
  322. struct rpc_xprt *xprt = req->rq_xprt;
  323. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  324. struct xdr_buf *xdr = &req->rq_snd_buf;
  325. int status;
  326. xs_pktdump("packet data:",
  327. req->rq_svec->iov_base,
  328. req->rq_svec->iov_len);
  329. req->rq_xtime = jiffies;
  330. status = xs_sendpages(transport->sock,
  331. (struct sockaddr *) &xprt->addr,
  332. xprt->addrlen, xdr,
  333. req->rq_bytes_sent);
  334. dprintk("RPC: xs_udp_send_request(%u) = %d\n",
  335. xdr->len - req->rq_bytes_sent, status);
  336. if (likely(status >= (int) req->rq_slen))
  337. return 0;
  338. /* Still some bytes left; set up for a retry later. */
  339. if (status > 0)
  340. status = -EAGAIN;
  341. switch (status) {
  342. case -ENETUNREACH:
  343. case -EPIPE:
  344. case -ECONNREFUSED:
  345. /* When the server has died, an ICMP port unreachable message
  346. * prompts ECONNREFUSED. */
  347. break;
  348. case -EAGAIN:
  349. xs_nospace(task);
  350. break;
  351. default:
  352. dprintk("RPC: sendmsg returned unrecognized error %d\n",
  353. -status);
  354. break;
  355. }
  356. return status;
  357. }
  358. static inline void xs_encode_tcp_record_marker(struct xdr_buf *buf)
  359. {
  360. u32 reclen = buf->len - sizeof(rpc_fraghdr);
  361. rpc_fraghdr *base = buf->head[0].iov_base;
  362. *base = htonl(RPC_LAST_STREAM_FRAGMENT | reclen);
  363. }
  364. /**
  365. * xs_tcp_send_request - write an RPC request to a TCP socket
  366. * @task: address of RPC task that manages the state of an RPC request
  367. *
  368. * Return values:
  369. * 0: The request has been sent
  370. * EAGAIN: The socket was blocked, please call again later to
  371. * complete the request
  372. * ENOTCONN: Caller needs to invoke connect logic then call again
  373. * other: Some other error occured, the request was not sent
  374. *
  375. * XXX: In the case of soft timeouts, should we eventually give up
  376. * if sendmsg is not able to make progress?
  377. */
  378. static int xs_tcp_send_request(struct rpc_task *task)
  379. {
  380. struct rpc_rqst *req = task->tk_rqstp;
  381. struct rpc_xprt *xprt = req->rq_xprt;
  382. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  383. struct xdr_buf *xdr = &req->rq_snd_buf;
  384. int status, retry = 0;
  385. xs_encode_tcp_record_marker(&req->rq_snd_buf);
  386. xs_pktdump("packet data:",
  387. req->rq_svec->iov_base,
  388. req->rq_svec->iov_len);
  389. /* Continue transmitting the packet/record. We must be careful
  390. * to cope with writespace callbacks arriving _after_ we have
  391. * called sendmsg(). */
  392. while (1) {
  393. req->rq_xtime = jiffies;
  394. status = xs_sendpages(transport->sock,
  395. NULL, 0, xdr, req->rq_bytes_sent);
  396. dprintk("RPC: xs_tcp_send_request(%u) = %d\n",
  397. xdr->len - req->rq_bytes_sent, status);
  398. if (unlikely(status < 0))
  399. break;
  400. /* If we've sent the entire packet, immediately
  401. * reset the count of bytes sent. */
  402. req->rq_bytes_sent += status;
  403. task->tk_bytes_sent += status;
  404. if (likely(req->rq_bytes_sent >= req->rq_slen)) {
  405. req->rq_bytes_sent = 0;
  406. return 0;
  407. }
  408. status = -EAGAIN;
  409. if (retry++ > XS_SENDMSG_RETRY)
  410. break;
  411. }
  412. switch (status) {
  413. case -EAGAIN:
  414. xs_nospace(task);
  415. break;
  416. case -ECONNREFUSED:
  417. case -ECONNRESET:
  418. case -ENOTCONN:
  419. case -EPIPE:
  420. status = -ENOTCONN;
  421. break;
  422. default:
  423. dprintk("RPC: sendmsg returned unrecognized error %d\n",
  424. -status);
  425. xprt_disconnect(xprt);
  426. break;
  427. }
  428. return status;
  429. }
  430. /**
  431. * xs_tcp_release_xprt - clean up after a tcp transmission
  432. * @xprt: transport
  433. * @task: rpc task
  434. *
  435. * This cleans up if an error causes us to abort the transmission of a request.
  436. * In this case, the socket may need to be reset in order to avoid confusing
  437. * the server.
  438. */
  439. static void xs_tcp_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task)
  440. {
  441. struct rpc_rqst *req;
  442. if (task != xprt->snd_task)
  443. return;
  444. if (task == NULL)
  445. goto out_release;
  446. req = task->tk_rqstp;
  447. if (req->rq_bytes_sent == 0)
  448. goto out_release;
  449. if (req->rq_bytes_sent == req->rq_snd_buf.len)
  450. goto out_release;
  451. set_bit(XPRT_CLOSE_WAIT, &task->tk_xprt->state);
  452. out_release:
  453. xprt_release_xprt(xprt, task);
  454. }
  455. /**
  456. * xs_close - close a socket
  457. * @xprt: transport
  458. *
  459. * This is used when all requests are complete; ie, no DRC state remains
  460. * on the server we want to save.
  461. */
  462. static void xs_close(struct rpc_xprt *xprt)
  463. {
  464. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  465. struct socket *sock = transport->sock;
  466. struct sock *sk = transport->inet;
  467. if (!sk)
  468. goto clear_close_wait;
  469. dprintk("RPC: xs_close xprt %p\n", xprt);
  470. write_lock_bh(&sk->sk_callback_lock);
  471. transport->inet = NULL;
  472. transport->sock = NULL;
  473. sk->sk_user_data = NULL;
  474. sk->sk_data_ready = transport->old_data_ready;
  475. sk->sk_state_change = transport->old_state_change;
  476. sk->sk_write_space = transport->old_write_space;
  477. write_unlock_bh(&sk->sk_callback_lock);
  478. sk->sk_no_check = 0;
  479. sock_release(sock);
  480. clear_close_wait:
  481. smp_mb__before_clear_bit();
  482. clear_bit(XPRT_CLOSE_WAIT, &xprt->state);
  483. smp_mb__after_clear_bit();
  484. }
  485. /**
  486. * xs_destroy - prepare to shutdown a transport
  487. * @xprt: doomed transport
  488. *
  489. */
  490. static void xs_destroy(struct rpc_xprt *xprt)
  491. {
  492. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  493. dprintk("RPC: xs_destroy xprt %p\n", xprt);
  494. cancel_delayed_work(&transport->connect_worker);
  495. flush_scheduled_work();
  496. xprt_disconnect(xprt);
  497. xs_close(xprt);
  498. xs_free_peer_addresses(xprt);
  499. kfree(xprt->slot);
  500. kfree(xprt);
  501. }
  502. static inline struct rpc_xprt *xprt_from_sock(struct sock *sk)
  503. {
  504. return (struct rpc_xprt *) sk->sk_user_data;
  505. }
  506. /**
  507. * xs_udp_data_ready - "data ready" callback for UDP sockets
  508. * @sk: socket with data to read
  509. * @len: how much data to read
  510. *
  511. */
  512. static void xs_udp_data_ready(struct sock *sk, int len)
  513. {
  514. struct rpc_task *task;
  515. struct rpc_xprt *xprt;
  516. struct rpc_rqst *rovr;
  517. struct sk_buff *skb;
  518. int err, repsize, copied;
  519. u32 _xid;
  520. __be32 *xp;
  521. read_lock(&sk->sk_callback_lock);
  522. dprintk("RPC: xs_udp_data_ready...\n");
  523. if (!(xprt = xprt_from_sock(sk)))
  524. goto out;
  525. if ((skb = skb_recv_datagram(sk, 0, 1, &err)) == NULL)
  526. goto out;
  527. if (xprt->shutdown)
  528. goto dropit;
  529. repsize = skb->len - sizeof(struct udphdr);
  530. if (repsize < 4) {
  531. dprintk("RPC: impossible RPC reply size %d!\n", repsize);
  532. goto dropit;
  533. }
  534. /* Copy the XID from the skb... */
  535. xp = skb_header_pointer(skb, sizeof(struct udphdr),
  536. sizeof(_xid), &_xid);
  537. if (xp == NULL)
  538. goto dropit;
  539. /* Look up and lock the request corresponding to the given XID */
  540. spin_lock(&xprt->transport_lock);
  541. rovr = xprt_lookup_rqst(xprt, *xp);
  542. if (!rovr)
  543. goto out_unlock;
  544. task = rovr->rq_task;
  545. if ((copied = rovr->rq_private_buf.buflen) > repsize)
  546. copied = repsize;
  547. /* Suck it into the iovec, verify checksum if not done by hw. */
  548. if (csum_partial_copy_to_xdr(&rovr->rq_private_buf, skb))
  549. goto out_unlock;
  550. /* Something worked... */
  551. dst_confirm(skb->dst);
  552. xprt_adjust_cwnd(task, copied);
  553. xprt_update_rtt(task);
  554. xprt_complete_rqst(task, copied);
  555. out_unlock:
  556. spin_unlock(&xprt->transport_lock);
  557. dropit:
  558. skb_free_datagram(sk, skb);
  559. out:
  560. read_unlock(&sk->sk_callback_lock);
  561. }
  562. static inline void xs_tcp_read_fraghdr(struct rpc_xprt *xprt, struct xdr_skb_reader *desc)
  563. {
  564. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  565. size_t len, used;
  566. char *p;
  567. p = ((char *) &transport->tcp_fraghdr) + transport->tcp_offset;
  568. len = sizeof(transport->tcp_fraghdr) - transport->tcp_offset;
  569. used = xdr_skb_read_bits(desc, p, len);
  570. transport->tcp_offset += used;
  571. if (used != len)
  572. return;
  573. transport->tcp_reclen = ntohl(transport->tcp_fraghdr);
  574. if (transport->tcp_reclen & RPC_LAST_STREAM_FRAGMENT)
  575. transport->tcp_flags |= TCP_RCV_LAST_FRAG;
  576. else
  577. transport->tcp_flags &= ~TCP_RCV_LAST_FRAG;
  578. transport->tcp_reclen &= RPC_FRAGMENT_SIZE_MASK;
  579. transport->tcp_flags &= ~TCP_RCV_COPY_FRAGHDR;
  580. transport->tcp_offset = 0;
  581. /* Sanity check of the record length */
  582. if (unlikely(transport->tcp_reclen < 4)) {
  583. dprintk("RPC: invalid TCP record fragment length\n");
  584. xprt_disconnect(xprt);
  585. return;
  586. }
  587. dprintk("RPC: reading TCP record fragment of length %d\n",
  588. transport->tcp_reclen);
  589. }
  590. static void xs_tcp_check_fraghdr(struct sock_xprt *transport)
  591. {
  592. if (transport->tcp_offset == transport->tcp_reclen) {
  593. transport->tcp_flags |= TCP_RCV_COPY_FRAGHDR;
  594. transport->tcp_offset = 0;
  595. if (transport->tcp_flags & TCP_RCV_LAST_FRAG) {
  596. transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
  597. transport->tcp_flags |= TCP_RCV_COPY_XID;
  598. transport->tcp_copied = 0;
  599. }
  600. }
  601. }
  602. static inline void xs_tcp_read_xid(struct sock_xprt *transport, struct xdr_skb_reader *desc)
  603. {
  604. size_t len, used;
  605. char *p;
  606. len = sizeof(transport->tcp_xid) - transport->tcp_offset;
  607. dprintk("RPC: reading XID (%Zu bytes)\n", len);
  608. p = ((char *) &transport->tcp_xid) + transport->tcp_offset;
  609. used = xdr_skb_read_bits(desc, p, len);
  610. transport->tcp_offset += used;
  611. if (used != len)
  612. return;
  613. transport->tcp_flags &= ~TCP_RCV_COPY_XID;
  614. transport->tcp_flags |= TCP_RCV_COPY_DATA;
  615. transport->tcp_copied = 4;
  616. dprintk("RPC: reading reply for XID %08x\n",
  617. ntohl(transport->tcp_xid));
  618. xs_tcp_check_fraghdr(transport);
  619. }
  620. static inline void xs_tcp_read_request(struct rpc_xprt *xprt, struct xdr_skb_reader *desc)
  621. {
  622. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  623. struct rpc_rqst *req;
  624. struct xdr_buf *rcvbuf;
  625. size_t len;
  626. ssize_t r;
  627. /* Find and lock the request corresponding to this xid */
  628. spin_lock(&xprt->transport_lock);
  629. req = xprt_lookup_rqst(xprt, transport->tcp_xid);
  630. if (!req) {
  631. transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
  632. dprintk("RPC: XID %08x request not found!\n",
  633. ntohl(transport->tcp_xid));
  634. spin_unlock(&xprt->transport_lock);
  635. return;
  636. }
  637. rcvbuf = &req->rq_private_buf;
  638. len = desc->count;
  639. if (len > transport->tcp_reclen - transport->tcp_offset) {
  640. struct xdr_skb_reader my_desc;
  641. len = transport->tcp_reclen - transport->tcp_offset;
  642. memcpy(&my_desc, desc, sizeof(my_desc));
  643. my_desc.count = len;
  644. r = xdr_partial_copy_from_skb(rcvbuf, transport->tcp_copied,
  645. &my_desc, xdr_skb_read_bits);
  646. desc->count -= r;
  647. desc->offset += r;
  648. } else
  649. r = xdr_partial_copy_from_skb(rcvbuf, transport->tcp_copied,
  650. desc, xdr_skb_read_bits);
  651. if (r > 0) {
  652. transport->tcp_copied += r;
  653. transport->tcp_offset += r;
  654. }
  655. if (r != len) {
  656. /* Error when copying to the receive buffer,
  657. * usually because we weren't able to allocate
  658. * additional buffer pages. All we can do now
  659. * is turn off TCP_RCV_COPY_DATA, so the request
  660. * will not receive any additional updates,
  661. * and time out.
  662. * Any remaining data from this record will
  663. * be discarded.
  664. */
  665. transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
  666. dprintk("RPC: XID %08x truncated request\n",
  667. ntohl(transport->tcp_xid));
  668. dprintk("RPC: xprt = %p, tcp_copied = %lu, tcp_offset = %u, tcp_reclen = %u\n",
  669. xprt, transport->tcp_copied, transport->tcp_offset,
  670. transport->tcp_reclen);
  671. goto out;
  672. }
  673. dprintk("RPC: XID %08x read %Zd bytes\n",
  674. ntohl(transport->tcp_xid), r);
  675. dprintk("RPC: xprt = %p, tcp_copied = %lu, tcp_offset = %u, tcp_reclen = %u\n",
  676. xprt, transport->tcp_copied, transport->tcp_offset,
  677. transport->tcp_reclen);
  678. if (transport->tcp_copied == req->rq_private_buf.buflen)
  679. transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
  680. else if (transport->tcp_offset == transport->tcp_reclen) {
  681. if (transport->tcp_flags & TCP_RCV_LAST_FRAG)
  682. transport->tcp_flags &= ~TCP_RCV_COPY_DATA;
  683. }
  684. out:
  685. if (!(transport->tcp_flags & TCP_RCV_COPY_DATA))
  686. xprt_complete_rqst(req->rq_task, transport->tcp_copied);
  687. spin_unlock(&xprt->transport_lock);
  688. xs_tcp_check_fraghdr(transport);
  689. }
  690. static inline void xs_tcp_read_discard(struct sock_xprt *transport, struct xdr_skb_reader *desc)
  691. {
  692. size_t len;
  693. len = transport->tcp_reclen - transport->tcp_offset;
  694. if (len > desc->count)
  695. len = desc->count;
  696. desc->count -= len;
  697. desc->offset += len;
  698. transport->tcp_offset += len;
  699. dprintk("RPC: discarded %Zu bytes\n", len);
  700. xs_tcp_check_fraghdr(transport);
  701. }
  702. static int xs_tcp_data_recv(read_descriptor_t *rd_desc, struct sk_buff *skb, unsigned int offset, size_t len)
  703. {
  704. struct rpc_xprt *xprt = rd_desc->arg.data;
  705. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  706. struct xdr_skb_reader desc = {
  707. .skb = skb,
  708. .offset = offset,
  709. .count = len,
  710. };
  711. dprintk("RPC: xs_tcp_data_recv started\n");
  712. do {
  713. /* Read in a new fragment marker if necessary */
  714. /* Can we ever really expect to get completely empty fragments? */
  715. if (transport->tcp_flags & TCP_RCV_COPY_FRAGHDR) {
  716. xs_tcp_read_fraghdr(xprt, &desc);
  717. continue;
  718. }
  719. /* Read in the xid if necessary */
  720. if (transport->tcp_flags & TCP_RCV_COPY_XID) {
  721. xs_tcp_read_xid(transport, &desc);
  722. continue;
  723. }
  724. /* Read in the request data */
  725. if (transport->tcp_flags & TCP_RCV_COPY_DATA) {
  726. xs_tcp_read_request(xprt, &desc);
  727. continue;
  728. }
  729. /* Skip over any trailing bytes on short reads */
  730. xs_tcp_read_discard(transport, &desc);
  731. } while (desc.count);
  732. dprintk("RPC: xs_tcp_data_recv done\n");
  733. return len - desc.count;
  734. }
  735. /**
  736. * xs_tcp_data_ready - "data ready" callback for TCP sockets
  737. * @sk: socket with data to read
  738. * @bytes: how much data to read
  739. *
  740. */
  741. static void xs_tcp_data_ready(struct sock *sk, int bytes)
  742. {
  743. struct rpc_xprt *xprt;
  744. read_descriptor_t rd_desc;
  745. read_lock(&sk->sk_callback_lock);
  746. dprintk("RPC: xs_tcp_data_ready...\n");
  747. if (!(xprt = xprt_from_sock(sk)))
  748. goto out;
  749. if (xprt->shutdown)
  750. goto out;
  751. /* We use rd_desc to pass struct xprt to xs_tcp_data_recv */
  752. rd_desc.arg.data = xprt;
  753. rd_desc.count = 65536;
  754. tcp_read_sock(sk, &rd_desc, xs_tcp_data_recv);
  755. out:
  756. read_unlock(&sk->sk_callback_lock);
  757. }
  758. /**
  759. * xs_tcp_state_change - callback to handle TCP socket state changes
  760. * @sk: socket whose state has changed
  761. *
  762. */
  763. static void xs_tcp_state_change(struct sock *sk)
  764. {
  765. struct rpc_xprt *xprt;
  766. read_lock(&sk->sk_callback_lock);
  767. if (!(xprt = xprt_from_sock(sk)))
  768. goto out;
  769. dprintk("RPC: xs_tcp_state_change client %p...\n", xprt);
  770. dprintk("RPC: state %x conn %d dead %d zapped %d\n",
  771. sk->sk_state, xprt_connected(xprt),
  772. sock_flag(sk, SOCK_DEAD),
  773. sock_flag(sk, SOCK_ZAPPED));
  774. switch (sk->sk_state) {
  775. case TCP_ESTABLISHED:
  776. spin_lock_bh(&xprt->transport_lock);
  777. if (!xprt_test_and_set_connected(xprt)) {
  778. struct sock_xprt *transport = container_of(xprt,
  779. struct sock_xprt, xprt);
  780. /* Reset TCP record info */
  781. transport->tcp_offset = 0;
  782. transport->tcp_reclen = 0;
  783. transport->tcp_copied = 0;
  784. transport->tcp_flags =
  785. TCP_RCV_COPY_FRAGHDR | TCP_RCV_COPY_XID;
  786. xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
  787. xprt_wake_pending_tasks(xprt, 0);
  788. }
  789. spin_unlock_bh(&xprt->transport_lock);
  790. break;
  791. case TCP_SYN_SENT:
  792. case TCP_SYN_RECV:
  793. break;
  794. case TCP_CLOSE_WAIT:
  795. /* Try to schedule an autoclose RPC calls */
  796. set_bit(XPRT_CLOSE_WAIT, &xprt->state);
  797. if (test_and_set_bit(XPRT_LOCKED, &xprt->state) == 0)
  798. schedule_work(&xprt->task_cleanup);
  799. default:
  800. xprt_disconnect(xprt);
  801. }
  802. out:
  803. read_unlock(&sk->sk_callback_lock);
  804. }
  805. /**
  806. * xs_udp_write_space - callback invoked when socket buffer space
  807. * becomes available
  808. * @sk: socket whose state has changed
  809. *
  810. * Called when more output buffer space is available for this socket.
  811. * We try not to wake our writers until they can make "significant"
  812. * progress, otherwise we'll waste resources thrashing kernel_sendmsg
  813. * with a bunch of small requests.
  814. */
  815. static void xs_udp_write_space(struct sock *sk)
  816. {
  817. read_lock(&sk->sk_callback_lock);
  818. /* from net/core/sock.c:sock_def_write_space */
  819. if (sock_writeable(sk)) {
  820. struct socket *sock;
  821. struct rpc_xprt *xprt;
  822. if (unlikely(!(sock = sk->sk_socket)))
  823. goto out;
  824. if (unlikely(!(xprt = xprt_from_sock(sk))))
  825. goto out;
  826. if (unlikely(!test_and_clear_bit(SOCK_NOSPACE, &sock->flags)))
  827. goto out;
  828. xprt_write_space(xprt);
  829. }
  830. out:
  831. read_unlock(&sk->sk_callback_lock);
  832. }
  833. /**
  834. * xs_tcp_write_space - callback invoked when socket buffer space
  835. * becomes available
  836. * @sk: socket whose state has changed
  837. *
  838. * Called when more output buffer space is available for this socket.
  839. * We try not to wake our writers until they can make "significant"
  840. * progress, otherwise we'll waste resources thrashing kernel_sendmsg
  841. * with a bunch of small requests.
  842. */
  843. static void xs_tcp_write_space(struct sock *sk)
  844. {
  845. read_lock(&sk->sk_callback_lock);
  846. /* from net/core/stream.c:sk_stream_write_space */
  847. if (sk_stream_wspace(sk) >= sk_stream_min_wspace(sk)) {
  848. struct socket *sock;
  849. struct rpc_xprt *xprt;
  850. if (unlikely(!(sock = sk->sk_socket)))
  851. goto out;
  852. if (unlikely(!(xprt = xprt_from_sock(sk))))
  853. goto out;
  854. if (unlikely(!test_and_clear_bit(SOCK_NOSPACE, &sock->flags)))
  855. goto out;
  856. xprt_write_space(xprt);
  857. }
  858. out:
  859. read_unlock(&sk->sk_callback_lock);
  860. }
  861. static void xs_udp_do_set_buffer_size(struct rpc_xprt *xprt)
  862. {
  863. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  864. struct sock *sk = transport->inet;
  865. if (transport->rcvsize) {
  866. sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
  867. sk->sk_rcvbuf = transport->rcvsize * xprt->max_reqs * 2;
  868. }
  869. if (transport->sndsize) {
  870. sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
  871. sk->sk_sndbuf = transport->sndsize * xprt->max_reqs * 2;
  872. sk->sk_write_space(sk);
  873. }
  874. }
  875. /**
  876. * xs_udp_set_buffer_size - set send and receive limits
  877. * @xprt: generic transport
  878. * @sndsize: requested size of send buffer, in bytes
  879. * @rcvsize: requested size of receive buffer, in bytes
  880. *
  881. * Set socket send and receive buffer size limits.
  882. */
  883. static void xs_udp_set_buffer_size(struct rpc_xprt *xprt, size_t sndsize, size_t rcvsize)
  884. {
  885. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  886. transport->sndsize = 0;
  887. if (sndsize)
  888. transport->sndsize = sndsize + 1024;
  889. transport->rcvsize = 0;
  890. if (rcvsize)
  891. transport->rcvsize = rcvsize + 1024;
  892. xs_udp_do_set_buffer_size(xprt);
  893. }
  894. /**
  895. * xs_udp_timer - called when a retransmit timeout occurs on a UDP transport
  896. * @task: task that timed out
  897. *
  898. * Adjust the congestion window after a retransmit timeout has occurred.
  899. */
  900. static void xs_udp_timer(struct rpc_task *task)
  901. {
  902. xprt_adjust_cwnd(task, -ETIMEDOUT);
  903. }
  904. static unsigned short xs_get_random_port(void)
  905. {
  906. unsigned short range = xprt_max_resvport - xprt_min_resvport;
  907. unsigned short rand = (unsigned short) net_random() % range;
  908. return rand + xprt_min_resvport;
  909. }
  910. /**
  911. * xs_set_port - reset the port number in the remote endpoint address
  912. * @xprt: generic transport
  913. * @port: new port number
  914. *
  915. */
  916. static void xs_set_port(struct rpc_xprt *xprt, unsigned short port)
  917. {
  918. struct sockaddr_in *sap = (struct sockaddr_in *) &xprt->addr;
  919. dprintk("RPC: setting port for xprt %p to %u\n", xprt, port);
  920. sap->sin_port = htons(port);
  921. }
  922. static int xs_bindresvport(struct sock_xprt *transport, struct socket *sock)
  923. {
  924. struct sockaddr_in myaddr = {
  925. .sin_family = AF_INET,
  926. };
  927. int err;
  928. unsigned short port = transport->port;
  929. do {
  930. myaddr.sin_port = htons(port);
  931. err = kernel_bind(sock, (struct sockaddr *) &myaddr,
  932. sizeof(myaddr));
  933. if (err == 0) {
  934. transport->port = port;
  935. dprintk("RPC: xs_bindresvport bound to port %u\n",
  936. port);
  937. return 0;
  938. }
  939. if (port <= xprt_min_resvport)
  940. port = xprt_max_resvport;
  941. else
  942. port--;
  943. } while (err == -EADDRINUSE && port != transport->port);
  944. dprintk("RPC: can't bind to reserved port (%d).\n", -err);
  945. return err;
  946. }
  947. /**
  948. * xs_udp_connect_worker - set up a UDP socket
  949. * @args: RPC transport to connect
  950. *
  951. * Invoked by a work queue tasklet.
  952. */
  953. static void xs_udp_connect_worker(void *args)
  954. {
  955. struct sock_xprt *transport = (struct sock_xprt *)args;
  956. struct rpc_xprt *xprt = &transport->xprt;
  957. struct socket *sock = transport->sock;
  958. int err, status = -EIO;
  959. if (xprt->shutdown || !xprt_bound(xprt))
  960. goto out;
  961. /* Start by resetting any existing state */
  962. xs_close(xprt);
  963. if ((err = sock_create_kern(PF_INET, SOCK_DGRAM, IPPROTO_UDP, &sock)) < 0) {
  964. dprintk("RPC: can't create UDP transport socket (%d).\n", -err);
  965. goto out;
  966. }
  967. if (xprt->resvport && xs_bindresvport(transport, sock) < 0) {
  968. sock_release(sock);
  969. goto out;
  970. }
  971. dprintk("RPC: worker connecting xprt %p to address: %s\n",
  972. xprt, xprt->address_strings[RPC_DISPLAY_ALL]);
  973. if (!transport->inet) {
  974. struct sock *sk = sock->sk;
  975. write_lock_bh(&sk->sk_callback_lock);
  976. sk->sk_user_data = xprt;
  977. transport->old_data_ready = sk->sk_data_ready;
  978. transport->old_state_change = sk->sk_state_change;
  979. transport->old_write_space = sk->sk_write_space;
  980. sk->sk_data_ready = xs_udp_data_ready;
  981. sk->sk_write_space = xs_udp_write_space;
  982. sk->sk_no_check = UDP_CSUM_NORCV;
  983. sk->sk_allocation = GFP_ATOMIC;
  984. xprt_set_connected(xprt);
  985. /* Reset to new socket */
  986. transport->sock = sock;
  987. transport->inet = sk;
  988. write_unlock_bh(&sk->sk_callback_lock);
  989. }
  990. xs_udp_do_set_buffer_size(xprt);
  991. status = 0;
  992. out:
  993. xprt_wake_pending_tasks(xprt, status);
  994. xprt_clear_connecting(xprt);
  995. }
  996. /*
  997. * We need to preserve the port number so the reply cache on the server can
  998. * find our cached RPC replies when we get around to reconnecting.
  999. */
  1000. static void xs_tcp_reuse_connection(struct rpc_xprt *xprt)
  1001. {
  1002. int result;
  1003. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  1004. struct sockaddr any;
  1005. dprintk("RPC: disconnecting xprt %p to reuse port\n", xprt);
  1006. /*
  1007. * Disconnect the transport socket by doing a connect operation
  1008. * with AF_UNSPEC. This should return immediately...
  1009. */
  1010. memset(&any, 0, sizeof(any));
  1011. any.sa_family = AF_UNSPEC;
  1012. result = kernel_connect(transport->sock, &any, sizeof(any), 0);
  1013. if (result)
  1014. dprintk("RPC: AF_UNSPEC connect return code %d\n",
  1015. result);
  1016. }
  1017. /**
  1018. * xs_tcp_connect_worker - connect a TCP socket to a remote endpoint
  1019. * @args: RPC transport to connect
  1020. *
  1021. * Invoked by a work queue tasklet.
  1022. */
  1023. static void xs_tcp_connect_worker(void *args)
  1024. {
  1025. struct sock_xprt *transport = (struct sock_xprt *)args;
  1026. struct rpc_xprt *xprt = &transport->xprt;
  1027. struct socket *sock = transport->sock;
  1028. int err, status = -EIO;
  1029. if (xprt->shutdown || !xprt_bound(xprt))
  1030. goto out;
  1031. if (!sock) {
  1032. /* start from scratch */
  1033. if ((err = sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &sock)) < 0) {
  1034. dprintk("RPC: can't create TCP transport socket (%d).\n", -err);
  1035. goto out;
  1036. }
  1037. if (xprt->resvport && xs_bindresvport(transport, sock) < 0) {
  1038. sock_release(sock);
  1039. goto out;
  1040. }
  1041. } else
  1042. /* "close" the socket, preserving the local port */
  1043. xs_tcp_reuse_connection(xprt);
  1044. dprintk("RPC: worker connecting xprt %p to address: %s\n",
  1045. xprt, xprt->address_strings[RPC_DISPLAY_ALL]);
  1046. if (!transport->inet) {
  1047. struct sock *sk = sock->sk;
  1048. write_lock_bh(&sk->sk_callback_lock);
  1049. sk->sk_user_data = xprt;
  1050. transport->old_data_ready = sk->sk_data_ready;
  1051. transport->old_state_change = sk->sk_state_change;
  1052. transport->old_write_space = sk->sk_write_space;
  1053. sk->sk_data_ready = xs_tcp_data_ready;
  1054. sk->sk_state_change = xs_tcp_state_change;
  1055. sk->sk_write_space = xs_tcp_write_space;
  1056. sk->sk_allocation = GFP_ATOMIC;
  1057. /* socket options */
  1058. sk->sk_userlocks |= SOCK_BINDPORT_LOCK;
  1059. sock_reset_flag(sk, SOCK_LINGER);
  1060. tcp_sk(sk)->linger2 = 0;
  1061. tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF;
  1062. xprt_clear_connected(xprt);
  1063. /* Reset to new socket */
  1064. transport->sock = sock;
  1065. transport->inet = sk;
  1066. write_unlock_bh(&sk->sk_callback_lock);
  1067. }
  1068. /* Tell the socket layer to start connecting... */
  1069. xprt->stat.connect_count++;
  1070. xprt->stat.connect_start = jiffies;
  1071. status = kernel_connect(sock, (struct sockaddr *) &xprt->addr,
  1072. xprt->addrlen, O_NONBLOCK);
  1073. dprintk("RPC: %p connect status %d connected %d sock state %d\n",
  1074. xprt, -status, xprt_connected(xprt), sock->sk->sk_state);
  1075. if (status < 0) {
  1076. switch (status) {
  1077. case -EINPROGRESS:
  1078. case -EALREADY:
  1079. goto out_clear;
  1080. case -ECONNREFUSED:
  1081. case -ECONNRESET:
  1082. /* retry with existing socket, after a delay */
  1083. break;
  1084. default:
  1085. /* get rid of existing socket, and retry */
  1086. xs_close(xprt);
  1087. break;
  1088. }
  1089. }
  1090. out:
  1091. xprt_wake_pending_tasks(xprt, status);
  1092. out_clear:
  1093. xprt_clear_connecting(xprt);
  1094. }
  1095. /**
  1096. * xs_connect - connect a socket to a remote endpoint
  1097. * @task: address of RPC task that manages state of connect request
  1098. *
  1099. * TCP: If the remote end dropped the connection, delay reconnecting.
  1100. *
  1101. * UDP socket connects are synchronous, but we use a work queue anyway
  1102. * to guarantee that even unprivileged user processes can set up a
  1103. * socket on a privileged port.
  1104. *
  1105. * If a UDP socket connect fails, the delay behavior here prevents
  1106. * retry floods (hard mounts).
  1107. */
  1108. static void xs_connect(struct rpc_task *task)
  1109. {
  1110. struct rpc_xprt *xprt = task->tk_xprt;
  1111. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  1112. if (xprt_test_and_set_connecting(xprt))
  1113. return;
  1114. if (transport->sock != NULL) {
  1115. dprintk("RPC: xs_connect delayed xprt %p for %lu seconds\n",
  1116. xprt, xprt->reestablish_timeout / HZ);
  1117. schedule_delayed_work(&transport->connect_worker,
  1118. xprt->reestablish_timeout);
  1119. xprt->reestablish_timeout <<= 1;
  1120. if (xprt->reestablish_timeout > XS_TCP_MAX_REEST_TO)
  1121. xprt->reestablish_timeout = XS_TCP_MAX_REEST_TO;
  1122. } else {
  1123. dprintk("RPC: xs_connect scheduled xprt %p\n", xprt);
  1124. schedule_work(&transport->connect_worker);
  1125. /* flush_scheduled_work can sleep... */
  1126. if (!RPC_IS_ASYNC(task))
  1127. flush_scheduled_work();
  1128. }
  1129. }
  1130. /**
  1131. * xs_udp_print_stats - display UDP socket-specifc stats
  1132. * @xprt: rpc_xprt struct containing statistics
  1133. * @seq: output file
  1134. *
  1135. */
  1136. static void xs_udp_print_stats(struct rpc_xprt *xprt, struct seq_file *seq)
  1137. {
  1138. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  1139. seq_printf(seq, "\txprt:\tudp %u %lu %lu %lu %lu %Lu %Lu\n",
  1140. transport->port,
  1141. xprt->stat.bind_count,
  1142. xprt->stat.sends,
  1143. xprt->stat.recvs,
  1144. xprt->stat.bad_xids,
  1145. xprt->stat.req_u,
  1146. xprt->stat.bklog_u);
  1147. }
  1148. /**
  1149. * xs_tcp_print_stats - display TCP socket-specifc stats
  1150. * @xprt: rpc_xprt struct containing statistics
  1151. * @seq: output file
  1152. *
  1153. */
  1154. static void xs_tcp_print_stats(struct rpc_xprt *xprt, struct seq_file *seq)
  1155. {
  1156. struct sock_xprt *transport = container_of(xprt, struct sock_xprt, xprt);
  1157. long idle_time = 0;
  1158. if (xprt_connected(xprt))
  1159. idle_time = (long)(jiffies - xprt->last_used) / HZ;
  1160. seq_printf(seq, "\txprt:\ttcp %u %lu %lu %lu %ld %lu %lu %lu %Lu %Lu\n",
  1161. transport->port,
  1162. xprt->stat.bind_count,
  1163. xprt->stat.connect_count,
  1164. xprt->stat.connect_time,
  1165. idle_time,
  1166. xprt->stat.sends,
  1167. xprt->stat.recvs,
  1168. xprt->stat.bad_xids,
  1169. xprt->stat.req_u,
  1170. xprt->stat.bklog_u);
  1171. }
  1172. static struct rpc_xprt_ops xs_udp_ops = {
  1173. .set_buffer_size = xs_udp_set_buffer_size,
  1174. .reserve_xprt = xprt_reserve_xprt_cong,
  1175. .release_xprt = xprt_release_xprt_cong,
  1176. .rpcbind = rpc_getport,
  1177. .set_port = xs_set_port,
  1178. .connect = xs_connect,
  1179. .buf_alloc = rpc_malloc,
  1180. .buf_free = rpc_free,
  1181. .send_request = xs_udp_send_request,
  1182. .set_retrans_timeout = xprt_set_retrans_timeout_rtt,
  1183. .timer = xs_udp_timer,
  1184. .release_request = xprt_release_rqst_cong,
  1185. .close = xs_close,
  1186. .destroy = xs_destroy,
  1187. .print_stats = xs_udp_print_stats,
  1188. };
  1189. static struct rpc_xprt_ops xs_tcp_ops = {
  1190. .reserve_xprt = xprt_reserve_xprt,
  1191. .release_xprt = xs_tcp_release_xprt,
  1192. .rpcbind = rpc_getport,
  1193. .set_port = xs_set_port,
  1194. .connect = xs_connect,
  1195. .buf_alloc = rpc_malloc,
  1196. .buf_free = rpc_free,
  1197. .send_request = xs_tcp_send_request,
  1198. .set_retrans_timeout = xprt_set_retrans_timeout_def,
  1199. .close = xs_close,
  1200. .destroy = xs_destroy,
  1201. .print_stats = xs_tcp_print_stats,
  1202. };
  1203. static struct rpc_xprt *xs_setup_xprt(struct sockaddr *addr, size_t addrlen, unsigned int slot_table_size)
  1204. {
  1205. struct rpc_xprt *xprt;
  1206. struct sock_xprt *new;
  1207. if (addrlen > sizeof(xprt->addr)) {
  1208. dprintk("RPC: xs_setup_xprt: address too large\n");
  1209. return ERR_PTR(-EBADF);
  1210. }
  1211. new = kzalloc(sizeof(*new), GFP_KERNEL);
  1212. if (new == NULL) {
  1213. dprintk("RPC: xs_setup_xprt: couldn't allocate rpc_xprt\n");
  1214. return ERR_PTR(-ENOMEM);
  1215. }
  1216. xprt = &new->xprt;
  1217. xprt->max_reqs = slot_table_size;
  1218. xprt->slot = kcalloc(xprt->max_reqs, sizeof(struct rpc_rqst), GFP_KERNEL);
  1219. if (xprt->slot == NULL) {
  1220. kfree(xprt);
  1221. dprintk("RPC: xs_setup_xprt: couldn't allocate slot table\n");
  1222. return ERR_PTR(-ENOMEM);
  1223. }
  1224. memcpy(&xprt->addr, addr, addrlen);
  1225. xprt->addrlen = addrlen;
  1226. new->port = xs_get_random_port();
  1227. return xprt;
  1228. }
  1229. /**
  1230. * xs_setup_udp - Set up transport to use a UDP socket
  1231. * @addr: address of remote server
  1232. * @addrlen: length of address in bytes
  1233. * @to: timeout parameters
  1234. *
  1235. */
  1236. struct rpc_xprt *xs_setup_udp(struct sockaddr *addr, size_t addrlen, struct rpc_timeout *to)
  1237. {
  1238. struct rpc_xprt *xprt;
  1239. struct sock_xprt *transport;
  1240. xprt = xs_setup_xprt(addr, addrlen, xprt_udp_slot_table_entries);
  1241. if (IS_ERR(xprt))
  1242. return xprt;
  1243. transport = container_of(xprt, struct sock_xprt, xprt);
  1244. if (ntohs(((struct sockaddr_in *)addr)->sin_port) != 0)
  1245. xprt_set_bound(xprt);
  1246. xprt->prot = IPPROTO_UDP;
  1247. xprt->tsh_size = 0;
  1248. /* XXX: header size can vary due to auth type, IPv6, etc. */
  1249. xprt->max_payload = (1U << 16) - (MAX_HEADER << 3);
  1250. INIT_WORK(&transport->connect_worker, xs_udp_connect_worker, transport);
  1251. xprt->bind_timeout = XS_BIND_TO;
  1252. xprt->connect_timeout = XS_UDP_CONN_TO;
  1253. xprt->reestablish_timeout = XS_UDP_REEST_TO;
  1254. xprt->idle_timeout = XS_IDLE_DISC_TO;
  1255. xprt->ops = &xs_udp_ops;
  1256. if (to)
  1257. xprt->timeout = *to;
  1258. else
  1259. xprt_set_timeout(&xprt->timeout, 5, 5 * HZ);
  1260. xs_format_peer_addresses(xprt);
  1261. dprintk("RPC: set up transport to address %s\n",
  1262. xprt->address_strings[RPC_DISPLAY_ALL]);
  1263. return xprt;
  1264. }
  1265. /**
  1266. * xs_setup_tcp - Set up transport to use a TCP socket
  1267. * @addr: address of remote server
  1268. * @addrlen: length of address in bytes
  1269. * @to: timeout parameters
  1270. *
  1271. */
  1272. struct rpc_xprt *xs_setup_tcp(struct sockaddr *addr, size_t addrlen, struct rpc_timeout *to)
  1273. {
  1274. struct rpc_xprt *xprt;
  1275. struct sock_xprt *transport;
  1276. xprt = xs_setup_xprt(addr, addrlen, xprt_tcp_slot_table_entries);
  1277. if (IS_ERR(xprt))
  1278. return xprt;
  1279. transport = container_of(xprt, struct sock_xprt, xprt);
  1280. if (ntohs(((struct sockaddr_in *)addr)->sin_port) != 0)
  1281. xprt_set_bound(xprt);
  1282. xprt->prot = IPPROTO_TCP;
  1283. xprt->tsh_size = sizeof(rpc_fraghdr) / sizeof(u32);
  1284. xprt->max_payload = RPC_MAX_FRAGMENT_SIZE;
  1285. INIT_WORK(&transport->connect_worker, xs_tcp_connect_worker, transport);
  1286. xprt->bind_timeout = XS_BIND_TO;
  1287. xprt->connect_timeout = XS_TCP_CONN_TO;
  1288. xprt->reestablish_timeout = XS_TCP_INIT_REEST_TO;
  1289. xprt->idle_timeout = XS_IDLE_DISC_TO;
  1290. xprt->ops = &xs_tcp_ops;
  1291. if (to)
  1292. xprt->timeout = *to;
  1293. else
  1294. xprt_set_timeout(&xprt->timeout, 2, 60 * HZ);
  1295. xs_format_peer_addresses(xprt);
  1296. dprintk("RPC: set up transport to address %s\n",
  1297. xprt->address_strings[RPC_DISPLAY_ALL]);
  1298. return xprt;
  1299. }
  1300. /**
  1301. * init_socket_xprt - stub
  1302. *
  1303. */
  1304. int init_socket_xprt(void)
  1305. {
  1306. return 0;
  1307. }
  1308. /**
  1309. * cleanup_socket_xprt - stub
  1310. *
  1311. */
  1312. void cleanup_socket_xprt(void)
  1313. {
  1314. return;
  1315. }