| 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
| 2 | /* |
| 3 | * linux/include/linux/sunrpc/sched.h |
| 4 | * |
| 5 | * Scheduling primitives for kernel Sun RPC. |
| 6 | * |
| 7 | * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de> |
| 8 | */ |
| 9 | |
| 10 | #ifndef _LINUX_SUNRPC_SCHED_H_ |
| 11 | #define _LINUX_SUNRPC_SCHED_H_ |
| 12 | |
| 13 | #include <linux/timer.h> |
| 14 | #include <linux/ktime.h> |
| 15 | #include <linux/sunrpc/types.h> |
| 16 | #include <linux/spinlock.h> |
| 17 | #include <linux/wait_bit.h> |
| 18 | #include <linux/workqueue.h> |
| 19 | #include <linux/sunrpc/xdr.h> |
| 20 | |
| 21 | /* |
| 22 | * This is the actual RPC procedure call info. |
| 23 | */ |
| 24 | struct rpc_procinfo; |
| 25 | struct rpc_message { |
| 26 | const struct rpc_procinfo *rpc_proc; /* Procedure information */ |
| 27 | void * rpc_argp; /* Arguments */ |
| 28 | void * rpc_resp; /* Result */ |
| 29 | const struct cred * rpc_cred; /* Credentials */ |
| 30 | }; |
| 31 | |
| 32 | struct rpc_call_ops; |
| 33 | struct rpc_wait_queue; |
| 34 | struct rpc_wait { |
| 35 | struct list_head list; /* wait queue links */ |
| 36 | struct list_head links; /* Links to related tasks */ |
| 37 | struct list_head timer_list; /* Timer list */ |
| 38 | }; |
| 39 | |
| 40 | /* |
| 41 | * This describes a timeout strategy |
| 42 | */ |
| 43 | struct rpc_timeout { |
| 44 | unsigned long to_initval, /* initial timeout */ |
| 45 | to_maxval, /* max timeout */ |
| 46 | to_increment; /* if !exponential */ |
| 47 | unsigned int to_retries; /* max # of retries */ |
| 48 | unsigned char to_exponential; |
| 49 | }; |
| 50 | |
| 51 | /* |
| 52 | * This is the RPC task struct |
| 53 | */ |
| 54 | struct rpc_task { |
| 55 | atomic_t tk_count; /* Reference count */ |
| 56 | int tk_status; /* result of last operation */ |
| 57 | struct list_head tk_task; /* global list of tasks */ |
| 58 | |
| 59 | /* |
| 60 | * callback to be executed after waking up |
| 61 | * action next procedure for async tasks |
| 62 | */ |
| 63 | void (*tk_callback)(struct rpc_task *); |
| 64 | void (*tk_action)(struct rpc_task *); |
| 65 | |
| 66 | unsigned long tk_timeout; /* timeout for rpc_sleep() */ |
| 67 | unsigned long tk_runstate; /* Task run status */ |
| 68 | |
| 69 | struct rpc_wait_queue *tk_waitqueue; /* RPC wait queue we're on */ |
| 70 | union { |
| 71 | struct work_struct tk_work; /* Async task work queue */ |
| 72 | struct rpc_wait tk_wait; /* RPC wait */ |
| 73 | } u; |
| 74 | |
| 75 | /* |
| 76 | * RPC call state |
| 77 | */ |
| 78 | struct rpc_message tk_msg; /* RPC call info */ |
| 79 | void * tk_calldata; /* Caller private data */ |
| 80 | const struct rpc_call_ops *tk_ops; /* Caller callbacks */ |
| 81 | |
| 82 | struct rpc_clnt * tk_client; /* RPC client */ |
| 83 | struct rpc_xprt * tk_xprt; /* Transport */ |
| 84 | struct rpc_cred * tk_op_cred; /* cred being operated on */ |
| 85 | |
| 86 | struct rpc_rqst * tk_rqstp; /* RPC request */ |
| 87 | |
| 88 | struct workqueue_struct *tk_workqueue; /* Normally rpciod, but could |
| 89 | * be any workqueue |
| 90 | */ |
| 91 | ktime_t tk_start; /* RPC task init timestamp */ |
| 92 | |
| 93 | pid_t tk_owner; /* Process id for batching tasks */ |
| 94 | |
| 95 | int tk_rpc_status; /* Result of last RPC operation */ |
| 96 | unsigned short tk_flags; /* misc flags */ |
| 97 | unsigned short tk_timeouts; /* maj timeouts */ |
| 98 | |
| 99 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) || IS_ENABLED(CONFIG_TRACEPOINTS) |
| 100 | unsigned short tk_pid; /* debugging aid */ |
| 101 | #endif |
| 102 | unsigned char tk_priority : 2,/* Task priority */ |
| 103 | tk_garb_retry : 2, |
| 104 | tk_cred_retry : 2; |
| 105 | }; |
| 106 | |
| 107 | typedef void (*rpc_action)(struct rpc_task *); |
| 108 | |
| 109 | struct rpc_call_ops { |
| 110 | void (*rpc_call_prepare)(struct rpc_task *, void *); |
| 111 | void (*rpc_call_done)(struct rpc_task *, void *); |
| 112 | void (*rpc_count_stats)(struct rpc_task *, void *); |
| 113 | void (*rpc_release)(void *); |
| 114 | }; |
| 115 | |
| 116 | struct rpc_task_setup { |
| 117 | struct rpc_task *task; |
| 118 | struct rpc_clnt *rpc_client; |
| 119 | struct rpc_xprt *rpc_xprt; |
| 120 | struct rpc_cred *rpc_op_cred; /* credential being operated on */ |
| 121 | const struct rpc_message *rpc_message; |
| 122 | const struct rpc_call_ops *callback_ops; |
| 123 | void *callback_data; |
| 124 | struct workqueue_struct *workqueue; |
| 125 | unsigned short flags; |
| 126 | signed char priority; |
| 127 | }; |
| 128 | |
| 129 | /* |
| 130 | * RPC task flags |
| 131 | */ |
| 132 | #define RPC_TASK_ASYNC 0x0001 /* is an async task */ |
| 133 | #define RPC_TASK_SWAPPER 0x0002 /* is swapping in/out */ |
| 134 | #define RPC_TASK_MOVEABLE 0x0004 /* nfs4.1+ rpc tasks */ |
| 135 | #define RPC_TASK_NULLCREDS 0x0010 /* Use AUTH_NULL credential */ |
| 136 | #define RPC_CALL_MAJORSEEN 0x0020 /* major timeout seen */ |
| 137 | #define RPC_TASK_NETUNREACH_FATAL 0x0040 /* ENETUNREACH is fatal */ |
| 138 | #define RPC_TASK_DYNAMIC 0x0080 /* task was kmalloc'ed */ |
| 139 | #define RPC_TASK_NO_ROUND_ROBIN 0x0100 /* send requests on "main" xprt */ |
| 140 | #define RPC_TASK_SOFT 0x0200 /* Use soft timeouts */ |
| 141 | #define RPC_TASK_SOFTCONN 0x0400 /* Fail if can't connect */ |
| 142 | #define RPC_TASK_SENT 0x0800 /* message was sent */ |
| 143 | #define RPC_TASK_TIMEOUT 0x1000 /* fail with ETIMEDOUT on timeout */ |
| 144 | #define RPC_TASK_NOCONNECT 0x2000 /* return ENOTCONN if not connected */ |
| 145 | #define RPC_TASK_NO_RETRANS_TIMEOUT 0x4000 /* wait forever for a reply */ |
| 146 | #define RPC_TASK_CRED_NOREF 0x8000 /* No refcount on the credential */ |
| 147 | |
| 148 | #define RPC_IS_ASYNC(t) ((t)->tk_flags & RPC_TASK_ASYNC) |
| 149 | #define RPC_IS_SWAPPER(t) ((t)->tk_flags & RPC_TASK_SWAPPER) |
| 150 | #define RPC_IS_SOFT(t) ((t)->tk_flags & (RPC_TASK_SOFT|RPC_TASK_TIMEOUT)) |
| 151 | #define RPC_IS_SOFTCONN(t) ((t)->tk_flags & RPC_TASK_SOFTCONN) |
| 152 | #define RPC_WAS_SENT(t) ((t)->tk_flags & RPC_TASK_SENT) |
| 153 | #define RPC_IS_MOVEABLE(t) ((t)->tk_flags & RPC_TASK_MOVEABLE) |
| 154 | |
| 155 | enum { |
| 156 | RPC_TASK_RUNNING, |
| 157 | RPC_TASK_QUEUED, |
| 158 | RPC_TASK_ACTIVE, |
| 159 | RPC_TASK_NEED_XMIT, |
| 160 | RPC_TASK_NEED_RECV, |
| 161 | RPC_TASK_MSG_PIN_WAIT, |
| 162 | }; |
| 163 | |
| 164 | #define rpc_test_and_set_running(t) \ |
| 165 | test_and_set_bit(RPC_TASK_RUNNING, &(t)->tk_runstate) |
| 166 | #define rpc_clear_running(t) clear_bit(RPC_TASK_RUNNING, &(t)->tk_runstate) |
| 167 | |
| 168 | #define RPC_IS_QUEUED(t) test_bit(RPC_TASK_QUEUED, &(t)->tk_runstate) |
| 169 | #define rpc_set_queued(t) set_bit(RPC_TASK_QUEUED, &(t)->tk_runstate) |
| 170 | #define rpc_clear_queued(t) clear_bit(RPC_TASK_QUEUED, &(t)->tk_runstate) |
| 171 | |
| 172 | #define RPC_IS_ACTIVATED(t) test_bit(RPC_TASK_ACTIVE, &(t)->tk_runstate) |
| 173 | |
| 174 | #define RPC_SIGNALLED(t) (READ_ONCE(task->tk_rpc_status) == -ERESTARTSYS) |
| 175 | |
| 176 | /* |
| 177 | * Task priorities. |
| 178 | * Note: if you change these, you must also change |
| 179 | * the task initialization definitions below. |
| 180 | */ |
| 181 | #define RPC_PRIORITY_LOW (-1) |
| 182 | #define RPC_PRIORITY_NORMAL (0) |
| 183 | #define RPC_PRIORITY_HIGH (1) |
| 184 | #define RPC_PRIORITY_PRIVILEGED (2) |
| 185 | #define RPC_NR_PRIORITY (1 + RPC_PRIORITY_PRIVILEGED - RPC_PRIORITY_LOW) |
| 186 | |
| 187 | struct rpc_timer { |
| 188 | struct list_head list; |
| 189 | unsigned long expires; |
| 190 | struct delayed_work dwork; |
| 191 | }; |
| 192 | |
| 193 | /* |
| 194 | * RPC synchronization objects |
| 195 | */ |
| 196 | struct rpc_wait_queue { |
| 197 | spinlock_t lock; |
| 198 | struct list_head tasks[RPC_NR_PRIORITY]; /* task queue for each priority level */ |
| 199 | unsigned char maxpriority; /* maximum priority (0 if queue is not a priority queue) */ |
| 200 | unsigned char priority; /* current priority */ |
| 201 | unsigned char nr; /* # tasks remaining for cookie */ |
| 202 | unsigned int qlen; /* total # tasks waiting in queue */ |
| 203 | struct rpc_timer timer_list; |
| 204 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) || IS_ENABLED(CONFIG_TRACEPOINTS) |
| 205 | const char * name; |
| 206 | #endif |
| 207 | }; |
| 208 | |
| 209 | /* |
| 210 | * This is the # requests to send consecutively |
| 211 | * from a single cookie. The aim is to improve |
| 212 | * performance of NFS operations such as read/write. |
| 213 | */ |
| 214 | #define RPC_IS_PRIORITY(q) ((q)->maxpriority > 0) |
| 215 | |
| 216 | /* |
| 217 | * Function prototypes |
| 218 | */ |
| 219 | struct rpc_task *rpc_new_task(const struct rpc_task_setup *); |
| 220 | struct rpc_task *rpc_run_task(const struct rpc_task_setup *); |
| 221 | struct rpc_task *rpc_run_bc_task(struct rpc_rqst *req, |
| 222 | struct rpc_timeout *timeout); |
| 223 | void rpc_put_task(struct rpc_task *); |
| 224 | void rpc_put_task_async(struct rpc_task *); |
| 225 | bool rpc_task_set_rpc_status(struct rpc_task *task, int rpc_status); |
| 226 | void rpc_task_try_cancel(struct rpc_task *task, int error); |
| 227 | void rpc_signal_task(struct rpc_task *); |
| 228 | void rpc_exit_task(struct rpc_task *); |
| 229 | void rpc_exit(struct rpc_task *, int); |
| 230 | void rpc_release_calldata(const struct rpc_call_ops *, void *); |
| 231 | void rpc_killall_tasks(struct rpc_clnt *); |
| 232 | unsigned long rpc_cancel_tasks(struct rpc_clnt *clnt, int error, |
| 233 | bool (*fnmatch)(const struct rpc_task *, |
| 234 | const void *), |
| 235 | const void *data); |
| 236 | void rpc_execute(struct rpc_task *); |
| 237 | void rpc_init_priority_wait_queue(struct rpc_wait_queue *, const char *); |
| 238 | void rpc_init_wait_queue(struct rpc_wait_queue *, const char *); |
| 239 | void rpc_destroy_wait_queue(struct rpc_wait_queue *); |
| 240 | unsigned long rpc_task_timeout(const struct rpc_task *task); |
| 241 | void rpc_sleep_on_timeout(struct rpc_wait_queue *queue, |
| 242 | struct rpc_task *task, |
| 243 | rpc_action action, |
| 244 | unsigned long timeout); |
| 245 | void rpc_sleep_on(struct rpc_wait_queue *, struct rpc_task *, |
| 246 | rpc_action action); |
| 247 | void rpc_sleep_on_priority_timeout(struct rpc_wait_queue *queue, |
| 248 | struct rpc_task *task, |
| 249 | unsigned long timeout, |
| 250 | int priority); |
| 251 | void rpc_sleep_on_priority(struct rpc_wait_queue *, |
| 252 | struct rpc_task *, |
| 253 | int priority); |
| 254 | void rpc_wake_up_queued_task(struct rpc_wait_queue *, |
| 255 | struct rpc_task *); |
| 256 | void rpc_wake_up_queued_task_set_status(struct rpc_wait_queue *, |
| 257 | struct rpc_task *, |
| 258 | int); |
| 259 | void rpc_wake_up(struct rpc_wait_queue *); |
| 260 | struct rpc_task *rpc_wake_up_next(struct rpc_wait_queue *); |
| 261 | struct rpc_task *rpc_wake_up_first_on_wq(struct workqueue_struct *wq, |
| 262 | struct rpc_wait_queue *, |
| 263 | bool (*)(struct rpc_task *, void *), |
| 264 | void *); |
| 265 | struct rpc_task *rpc_wake_up_first(struct rpc_wait_queue *, |
| 266 | bool (*)(struct rpc_task *, void *), |
| 267 | void *); |
| 268 | void rpc_wake_up_status(struct rpc_wait_queue *, int); |
| 269 | void rpc_delay(struct rpc_task *, unsigned long); |
| 270 | int rpc_malloc(struct rpc_task *); |
| 271 | void rpc_free(struct rpc_task *); |
| 272 | int rpciod_up(void); |
| 273 | void rpciod_down(void); |
| 274 | int rpc_wait_for_completion_task(struct rpc_task *task); |
| 275 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) |
| 276 | struct net; |
| 277 | void rpc_show_tasks(struct net *); |
| 278 | #endif |
| 279 | int rpc_init_mempool(void); |
| 280 | void rpc_destroy_mempool(void); |
| 281 | extern struct workqueue_struct *rpciod_workqueue; |
| 282 | extern struct workqueue_struct *xprtiod_workqueue; |
| 283 | void rpc_prepare_task(struct rpc_task *task); |
| 284 | gfp_t rpc_task_gfp_mask(void); |
| 285 | |
| 286 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) || IS_ENABLED(CONFIG_TRACEPOINTS) |
| 287 | static inline const char * rpc_qname(const struct rpc_wait_queue *q) |
| 288 | { |
| 289 | return ((q && q->name) ? q->name : "unknown" ); |
| 290 | } |
| 291 | |
| 292 | static inline void rpc_assign_waitqueue_name(struct rpc_wait_queue *q, |
| 293 | const char *name) |
| 294 | { |
| 295 | q->name = name; |
| 296 | } |
| 297 | #else |
| 298 | static inline void rpc_assign_waitqueue_name(struct rpc_wait_queue *q, |
| 299 | const char *name) |
| 300 | { |
| 301 | } |
| 302 | #endif |
| 303 | |
| 304 | #if IS_ENABLED(CONFIG_SUNRPC_SWAP) |
| 305 | int rpc_clnt_swap_activate(struct rpc_clnt *clnt); |
| 306 | void rpc_clnt_swap_deactivate(struct rpc_clnt *clnt); |
| 307 | #else |
| 308 | static inline int |
| 309 | rpc_clnt_swap_activate(struct rpc_clnt *clnt) |
| 310 | { |
| 311 | return -EINVAL; |
| 312 | } |
| 313 | |
| 314 | static inline void |
| 315 | rpc_clnt_swap_deactivate(struct rpc_clnt *clnt) |
| 316 | { |
| 317 | } |
| 318 | #endif /* CONFIG_SUNRPC_SWAP */ |
| 319 | |
| 320 | #endif /* _LINUX_SUNRPC_SCHED_H_ */ |
| 321 | |