xprt.h 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474
  1. /*
  2. * linux/include/linux/sunrpc/xprt.h
  3. *
  4. * Declarations for the RPC transport interface.
  5. *
  6. * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
  7. */
  8. #ifndef _LINUX_SUNRPC_XPRT_H
  9. #define _LINUX_SUNRPC_XPRT_H
  10. #include <linux/uio.h>
  11. #include <linux/socket.h>
  12. #include <linux/in.h>
  13. #include <linux/ktime.h>
  14. #include <linux/kref.h>
  15. #include <linux/sunrpc/sched.h>
  16. #include <linux/sunrpc/xdr.h>
  17. #include <linux/sunrpc/msg_prot.h>
  18. #ifdef __KERNEL__
  19. #define RPC_MIN_SLOT_TABLE (2U)
  20. #define RPC_DEF_SLOT_TABLE (16U)
  21. #define RPC_MAX_SLOT_TABLE_LIMIT (65536U)
  22. #define RPC_MAX_SLOT_TABLE RPC_MAX_SLOT_TABLE_LIMIT
  23. #define RPC_CWNDSHIFT (8U)
  24. #define RPC_CWNDSCALE (1U << RPC_CWNDSHIFT)
  25. #define RPC_INITCWND RPC_CWNDSCALE
  26. #define RPC_MAXCWND(xprt) ((xprt)->max_reqs << RPC_CWNDSHIFT)
  27. #define RPCXPRT_CONGESTED(xprt) ((xprt)->cong >= (xprt)->cwnd)
  28. /*
  29. * This describes a timeout strategy
  30. */
  31. struct rpc_timeout {
  32. unsigned long to_initval, /* initial timeout */
  33. to_maxval, /* max timeout */
  34. to_increment; /* if !exponential */
  35. unsigned int to_retries; /* max # of retries */
  36. unsigned char to_exponential;
  37. };
  38. enum rpc_display_format_t {
  39. RPC_DISPLAY_ADDR = 0,
  40. RPC_DISPLAY_PORT,
  41. RPC_DISPLAY_PROTO,
  42. RPC_DISPLAY_HEX_ADDR,
  43. RPC_DISPLAY_HEX_PORT,
  44. RPC_DISPLAY_NETID,
  45. RPC_DISPLAY_MAX,
  46. };
  47. struct rpc_task;
  48. struct rpc_xprt;
  49. struct seq_file;
  50. struct svc_serv;
  51. struct net;
  52. /*
  53. * This describes a complete RPC request
  54. */
  55. struct rpc_rqst {
  56. /*
  57. * This is the user-visible part
  58. */
  59. struct rpc_xprt * rq_xprt; /* RPC client */
  60. struct xdr_buf rq_snd_buf; /* send buffer */
  61. struct xdr_buf rq_rcv_buf; /* recv buffer */
  62. /*
  63. * This is the private part
  64. */
  65. struct rpc_task * rq_task; /* RPC task data */
  66. struct rpc_cred * rq_cred; /* Bound cred */
  67. __be32 rq_xid; /* request XID */
  68. int rq_cong; /* has incremented xprt->cong */
  69. u32 rq_seqno; /* gss seq no. used on req. */
  70. int rq_enc_pages_num;
  71. struct page **rq_enc_pages; /* scratch pages for use by
  72. gss privacy code */
  73. void (*rq_release_snd_buf)(struct rpc_rqst *); /* release rq_enc_pages */
  74. struct list_head rq_list;
  75. __u32 * rq_buffer; /* XDR encode buffer */
  76. size_t rq_callsize,
  77. rq_rcvsize;
  78. size_t rq_xmit_bytes_sent; /* total bytes sent */
  79. size_t rq_reply_bytes_recvd; /* total reply bytes */
  80. /* received */
  81. struct xdr_buf rq_private_buf; /* The receive buffer
  82. * used in the softirq.
  83. */
  84. unsigned long rq_majortimeo; /* major timeout alarm */
  85. unsigned long rq_timeout; /* Current timeout value */
  86. ktime_t rq_rtt; /* round-trip time */
  87. unsigned int rq_retries; /* # of retries */
  88. unsigned int rq_connect_cookie;
  89. /* A cookie used to track the
  90. state of the transport
  91. connection */
  92. /*
  93. * Partial send handling
  94. */
  95. u32 rq_bytes_sent; /* Bytes we have sent */
  96. ktime_t rq_xtime; /* transmit time stamp */
  97. int rq_ntrans;
  98. #if defined(CONFIG_SUNRPC_BACKCHANNEL)
  99. struct list_head rq_bc_list; /* Callback service list */
  100. unsigned long rq_bc_pa_state; /* Backchannel prealloc state */
  101. struct list_head rq_bc_pa_list; /* Backchannel prealloc list */
  102. #endif /* CONFIG_SUNRPC_BACKCHANEL */
  103. };
  104. #define rq_svec rq_snd_buf.head
  105. #define rq_slen rq_snd_buf.len
  106. struct rpc_xprt_ops {
  107. void (*set_buffer_size)(struct rpc_xprt *xprt, size_t sndsize, size_t rcvsize);
  108. int (*reserve_xprt)(struct rpc_xprt *xprt, struct rpc_task *task);
  109. void (*release_xprt)(struct rpc_xprt *xprt, struct rpc_task *task);
  110. void (*alloc_slot)(struct rpc_xprt *xprt, struct rpc_task *task);
  111. void (*rpcbind)(struct rpc_task *task);
  112. void (*set_port)(struct rpc_xprt *xprt, unsigned short port);
  113. void (*connect)(struct rpc_xprt *xprt, struct rpc_task *task);
  114. void * (*buf_alloc)(struct rpc_task *task, size_t size);
  115. void (*buf_free)(void *buffer);
  116. int (*send_request)(struct rpc_task *task);
  117. void (*set_retrans_timeout)(struct rpc_task *task);
  118. void (*timer)(struct rpc_xprt *xprt, struct rpc_task *task);
  119. void (*release_request)(struct rpc_task *task);
  120. void (*close)(struct rpc_xprt *xprt);
  121. void (*destroy)(struct rpc_xprt *xprt);
  122. void (*print_stats)(struct rpc_xprt *xprt, struct seq_file *seq);
  123. int (*enable_swap)(struct rpc_xprt *xprt);
  124. void (*disable_swap)(struct rpc_xprt *xprt);
  125. void (*inject_disconnect)(struct rpc_xprt *xprt);
  126. int (*bc_setup)(struct rpc_xprt *xprt,
  127. unsigned int min_reqs);
  128. int (*bc_up)(struct svc_serv *serv, struct net *net);
  129. void (*bc_free_rqst)(struct rpc_rqst *rqst);
  130. void (*bc_destroy)(struct rpc_xprt *xprt,
  131. unsigned int max_reqs);
  132. };
  133. /*
  134. * RPC transport identifiers
  135. *
  136. * To preserve compatibility with the historical use of raw IP protocol
  137. * id's for transport selection, UDP and TCP identifiers are specified
  138. * with the previous values. No such restriction exists for new transports,
  139. * except that they may not collide with these values (17 and 6,
  140. * respectively).
  141. */
  142. #define XPRT_TRANSPORT_BC (1 << 31)
  143. enum xprt_transports {
  144. XPRT_TRANSPORT_UDP = IPPROTO_UDP,
  145. XPRT_TRANSPORT_TCP = IPPROTO_TCP,
  146. XPRT_TRANSPORT_BC_TCP = IPPROTO_TCP | XPRT_TRANSPORT_BC,
  147. XPRT_TRANSPORT_RDMA = 256,
  148. XPRT_TRANSPORT_BC_RDMA = XPRT_TRANSPORT_RDMA | XPRT_TRANSPORT_BC,
  149. XPRT_TRANSPORT_LOCAL = 257,
  150. };
  151. struct rpc_xprt {
  152. struct kref kref; /* Reference count */
  153. struct rpc_xprt_ops * ops; /* transport methods */
  154. const struct rpc_timeout *timeout; /* timeout parms */
  155. struct sockaddr_storage addr; /* server address */
  156. size_t addrlen; /* size of server address */
  157. int prot; /* IP protocol */
  158. unsigned long cong; /* current congestion */
  159. unsigned long cwnd; /* congestion window */
  160. size_t max_payload; /* largest RPC payload size,
  161. in bytes */
  162. unsigned int tsh_size; /* size of transport specific
  163. header */
  164. struct rpc_wait_queue binding; /* requests waiting on rpcbind */
  165. struct rpc_wait_queue sending; /* requests waiting to send */
  166. struct rpc_wait_queue pending; /* requests in flight */
  167. struct rpc_wait_queue backlog; /* waiting for slot */
  168. struct list_head free; /* free slots */
  169. unsigned int max_reqs; /* max number of slots */
  170. unsigned int min_reqs; /* min number of slots */
  171. atomic_t num_reqs; /* total slots */
  172. unsigned long state; /* transport state */
  173. unsigned char resvport : 1; /* use a reserved port */
  174. atomic_t swapper; /* we're swapping over this
  175. transport */
  176. unsigned int bind_index; /* bind function index */
  177. /*
  178. * Multipath
  179. */
  180. struct list_head xprt_switch;
  181. /*
  182. * Connection of transports
  183. */
  184. unsigned long bind_timeout,
  185. reestablish_timeout;
  186. unsigned int connect_cookie; /* A cookie that gets bumped
  187. every time the transport
  188. is reconnected */
  189. /*
  190. * Disconnection of idle transports
  191. */
  192. struct work_struct task_cleanup;
  193. struct timer_list timer;
  194. unsigned long last_used,
  195. idle_timeout;
  196. /*
  197. * Send stuff
  198. */
  199. spinlock_t transport_lock; /* lock transport info */
  200. spinlock_t reserve_lock; /* lock slot table */
  201. u32 xid; /* Next XID value to use */
  202. struct rpc_task * snd_task; /* Task blocked in send */
  203. struct svc_xprt *bc_xprt; /* NFSv4.1 backchannel */
  204. #if defined(CONFIG_SUNRPC_BACKCHANNEL)
  205. struct svc_serv *bc_serv; /* The RPC service which will */
  206. /* process the callback */
  207. int bc_alloc_count; /* Total number of preallocs */
  208. atomic_t bc_free_slots;
  209. spinlock_t bc_pa_lock; /* Protects the preallocated
  210. * items */
  211. struct list_head bc_pa_list; /* List of preallocated
  212. * backchannel rpc_rqst's */
  213. #endif /* CONFIG_SUNRPC_BACKCHANNEL */
  214. struct list_head recv;
  215. struct {
  216. unsigned long bind_count, /* total number of binds */
  217. connect_count, /* total number of connects */
  218. connect_start, /* connect start timestamp */
  219. connect_time, /* jiffies waiting for connect */
  220. sends, /* how many complete requests */
  221. recvs, /* how many complete requests */
  222. bad_xids, /* lookup_rqst didn't find XID */
  223. max_slots; /* max rpc_slots used */
  224. unsigned long long req_u, /* average requests on the wire */
  225. bklog_u, /* backlog queue utilization */
  226. sending_u, /* send q utilization */
  227. pending_u; /* pend q utilization */
  228. } stat;
  229. struct net *xprt_net;
  230. const char *servername;
  231. const char *address_strings[RPC_DISPLAY_MAX];
  232. #if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
  233. struct dentry *debugfs; /* debugfs directory */
  234. atomic_t inject_disconnect;
  235. #endif
  236. struct rcu_head rcu;
  237. };
  238. #if defined(CONFIG_SUNRPC_BACKCHANNEL)
  239. /*
  240. * Backchannel flags
  241. */
  242. #define RPC_BC_PA_IN_USE 0x0001 /* Preallocated backchannel */
  243. /* buffer in use */
  244. #endif /* CONFIG_SUNRPC_BACKCHANNEL */
  245. #if defined(CONFIG_SUNRPC_BACKCHANNEL)
  246. static inline int bc_prealloc(struct rpc_rqst *req)
  247. {
  248. return test_bit(RPC_BC_PA_IN_USE, &req->rq_bc_pa_state);
  249. }
  250. #else
  251. static inline int bc_prealloc(struct rpc_rqst *req)
  252. {
  253. return 0;
  254. }
  255. #endif /* CONFIG_SUNRPC_BACKCHANNEL */
  256. #define XPRT_CREATE_INFINITE_SLOTS (1U)
  257. #define XPRT_CREATE_NO_IDLE_TIMEOUT (1U << 1)
  258. struct xprt_create {
  259. int ident; /* XPRT_TRANSPORT identifier */
  260. struct net * net;
  261. struct sockaddr * srcaddr; /* optional local address */
  262. struct sockaddr * dstaddr; /* remote peer address */
  263. size_t addrlen;
  264. const char *servername;
  265. struct svc_xprt *bc_xprt; /* NFSv4.1 backchannel */
  266. unsigned int flags;
  267. };
  268. struct xprt_class {
  269. struct list_head list;
  270. int ident; /* XPRT_TRANSPORT identifier */
  271. struct rpc_xprt * (*setup)(struct xprt_create *);
  272. struct module *owner;
  273. char name[32];
  274. };
  275. /*
  276. * Generic internal transport functions
  277. */
  278. struct rpc_xprt *xprt_create_transport(struct xprt_create *args);
  279. void xprt_connect(struct rpc_task *task);
  280. void xprt_reserve(struct rpc_task *task);
  281. void xprt_retry_reserve(struct rpc_task *task);
  282. int xprt_reserve_xprt(struct rpc_xprt *xprt, struct rpc_task *task);
  283. int xprt_reserve_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task);
  284. void xprt_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task);
  285. void xprt_lock_and_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task);
  286. bool xprt_prepare_transmit(struct rpc_task *task);
  287. void xprt_transmit(struct rpc_task *task);
  288. void xprt_end_transmit(struct rpc_task *task);
  289. int xprt_adjust_timeout(struct rpc_rqst *req);
  290. void xprt_release_xprt(struct rpc_xprt *xprt, struct rpc_task *task);
  291. void xprt_release_xprt_cong(struct rpc_xprt *xprt, struct rpc_task *task);
  292. void xprt_release(struct rpc_task *task);
  293. struct rpc_xprt * xprt_get(struct rpc_xprt *xprt);
  294. void xprt_put(struct rpc_xprt *xprt);
  295. struct rpc_xprt * xprt_alloc(struct net *net, size_t size,
  296. unsigned int num_prealloc,
  297. unsigned int max_req);
  298. void xprt_free(struct rpc_xprt *);
  299. static inline __be32 *xprt_skip_transport_header(struct rpc_xprt *xprt, __be32 *p)
  300. {
  301. return p + xprt->tsh_size;
  302. }
  303. static inline int
  304. xprt_enable_swap(struct rpc_xprt *xprt)
  305. {
  306. return xprt->ops->enable_swap(xprt);
  307. }
  308. static inline void
  309. xprt_disable_swap(struct rpc_xprt *xprt)
  310. {
  311. xprt->ops->disable_swap(xprt);
  312. }
  313. /*
  314. * Transport switch helper functions
  315. */
  316. int xprt_register_transport(struct xprt_class *type);
  317. int xprt_unregister_transport(struct xprt_class *type);
  318. int xprt_load_transport(const char *);
  319. void xprt_set_retrans_timeout_def(struct rpc_task *task);
  320. void xprt_set_retrans_timeout_rtt(struct rpc_task *task);
  321. void xprt_wake_pending_tasks(struct rpc_xprt *xprt, int status);
  322. void xprt_wait_for_buffer_space(struct rpc_task *task, rpc_action action);
  323. void xprt_write_space(struct rpc_xprt *xprt);
  324. void xprt_adjust_cwnd(struct rpc_xprt *xprt, struct rpc_task *task, int result);
  325. struct rpc_rqst * xprt_lookup_rqst(struct rpc_xprt *xprt, __be32 xid);
  326. void xprt_complete_rqst(struct rpc_task *task, int copied);
  327. void xprt_release_rqst_cong(struct rpc_task *task);
  328. void xprt_disconnect_done(struct rpc_xprt *xprt);
  329. void xprt_force_disconnect(struct rpc_xprt *xprt);
  330. void xprt_conditional_disconnect(struct rpc_xprt *xprt, unsigned int cookie);
  331. bool xprt_lock_connect(struct rpc_xprt *, struct rpc_task *, void *);
  332. void xprt_unlock_connect(struct rpc_xprt *, void *);
  333. /*
  334. * Reserved bit positions in xprt->state
  335. */
  336. #define XPRT_LOCKED (0)
  337. #define XPRT_CONNECTED (1)
  338. #define XPRT_CONNECTING (2)
  339. #define XPRT_CLOSE_WAIT (3)
  340. #define XPRT_BOUND (4)
  341. #define XPRT_BINDING (5)
  342. #define XPRT_CLOSING (6)
  343. #define XPRT_CONGESTED (9)
  344. static inline void xprt_set_connected(struct rpc_xprt *xprt)
  345. {
  346. set_bit(XPRT_CONNECTED, &xprt->state);
  347. }
  348. static inline void xprt_clear_connected(struct rpc_xprt *xprt)
  349. {
  350. clear_bit(XPRT_CONNECTED, &xprt->state);
  351. }
  352. static inline int xprt_connected(struct rpc_xprt *xprt)
  353. {
  354. return test_bit(XPRT_CONNECTED, &xprt->state);
  355. }
  356. static inline int xprt_test_and_set_connected(struct rpc_xprt *xprt)
  357. {
  358. return test_and_set_bit(XPRT_CONNECTED, &xprt->state);
  359. }
  360. static inline int xprt_test_and_clear_connected(struct rpc_xprt *xprt)
  361. {
  362. return test_and_clear_bit(XPRT_CONNECTED, &xprt->state);
  363. }
  364. static inline void xprt_clear_connecting(struct rpc_xprt *xprt)
  365. {
  366. smp_mb__before_atomic();
  367. clear_bit(XPRT_CONNECTING, &xprt->state);
  368. smp_mb__after_atomic();
  369. }
  370. static inline int xprt_connecting(struct rpc_xprt *xprt)
  371. {
  372. return test_bit(XPRT_CONNECTING, &xprt->state);
  373. }
  374. static inline int xprt_test_and_set_connecting(struct rpc_xprt *xprt)
  375. {
  376. return test_and_set_bit(XPRT_CONNECTING, &xprt->state);
  377. }
  378. static inline void xprt_set_bound(struct rpc_xprt *xprt)
  379. {
  380. test_and_set_bit(XPRT_BOUND, &xprt->state);
  381. }
  382. static inline int xprt_bound(struct rpc_xprt *xprt)
  383. {
  384. return test_bit(XPRT_BOUND, &xprt->state);
  385. }
  386. static inline void xprt_clear_bound(struct rpc_xprt *xprt)
  387. {
  388. clear_bit(XPRT_BOUND, &xprt->state);
  389. }
  390. static inline void xprt_clear_binding(struct rpc_xprt *xprt)
  391. {
  392. smp_mb__before_atomic();
  393. clear_bit(XPRT_BINDING, &xprt->state);
  394. smp_mb__after_atomic();
  395. }
  396. static inline int xprt_test_and_set_binding(struct rpc_xprt *xprt)
  397. {
  398. return test_and_set_bit(XPRT_BINDING, &xprt->state);
  399. }
  400. #if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
  401. extern unsigned int rpc_inject_disconnect;
  402. static inline void xprt_inject_disconnect(struct rpc_xprt *xprt)
  403. {
  404. if (!rpc_inject_disconnect)
  405. return;
  406. if (atomic_dec_return(&xprt->inject_disconnect))
  407. return;
  408. atomic_set(&xprt->inject_disconnect, rpc_inject_disconnect);
  409. xprt->ops->inject_disconnect(xprt);
  410. }
  411. #else
  412. static inline void xprt_inject_disconnect(struct rpc_xprt *xprt)
  413. {
  414. }
  415. #endif
  416. #endif /* __KERNEL__*/
  417. #endif /* _LINUX_SUNRPC_XPRT_H */