1 | /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */ |
2 | /* |
3 | * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved. |
4 | * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved. |
5 | */ |
6 | |
7 | #ifndef RXE_TASK_H |
8 | #define RXE_TASK_H |
9 | |
10 | enum { |
11 | TASK_STATE_IDLE = 0, |
12 | TASK_STATE_BUSY = 1, |
13 | TASK_STATE_ARMED = 2, |
14 | TASK_STATE_DRAINING = 3, |
15 | TASK_STATE_DRAINED = 4, |
16 | TASK_STATE_INVALID = 5, |
17 | }; |
18 | |
19 | /* |
20 | * data structure to describe a 'task' which is a short |
21 | * function that returns 0 as long as it needs to be |
22 | * called again. |
23 | */ |
24 | struct rxe_task { |
25 | struct work_struct work; |
26 | int state; |
27 | spinlock_t lock; |
28 | struct rxe_qp *qp; |
29 | int (*func)(struct rxe_qp *qp); |
30 | int ret; |
31 | long num_sched; |
32 | long num_done; |
33 | }; |
34 | |
35 | int rxe_alloc_wq(void); |
36 | |
37 | void rxe_destroy_wq(void); |
38 | |
39 | /* |
40 | * init rxe_task structure |
41 | * qp => parameter to pass to func |
42 | * func => function to call until it returns != 0 |
43 | */ |
44 | int rxe_init_task(struct rxe_task *task, struct rxe_qp *qp, |
45 | int (*func)(struct rxe_qp *)); |
46 | |
47 | /* cleanup task */ |
48 | void rxe_cleanup_task(struct rxe_task *task); |
49 | |
50 | void rxe_run_task(struct rxe_task *task); |
51 | |
52 | void rxe_sched_task(struct rxe_task *task); |
53 | |
54 | /* keep a task from scheduling */ |
55 | void rxe_disable_task(struct rxe_task *task); |
56 | |
57 | /* allow task to run */ |
58 | void rxe_enable_task(struct rxe_task *task); |
59 | |
60 | #endif /* RXE_TASK_H */ |
61 | |