66 lines
1.5 KiB
C
66 lines
1.5 KiB
C
|
/* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
|
||
|
|
/*
|
||
|
|
* Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved.
|
||
|
|
* Copyright (c) 2015 System Fabric Works, Inc. All rights reserved.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef RXE_TASK_H
|
||
|
|
#define RXE_TASK_H
|
||
|
|
|
||
|
|
enum {
|
||
|
|
TASK_STATE_START = 0,
|
||
|
|
TASK_STATE_BUSY = 1,
|
||
|
|
TASK_STATE_ARMED = 2,
|
||
|
|
};
|
||
|
|
|
||
|
|
/*
|
||
|
|
* data structure to describe a 'task' which is a short
|
||
|
|
* function that returns 0 as long as it needs to be
|
||
|
|
* called again.
|
||
|
|
*/
|
||
|
|
struct rxe_task {
|
||
|
|
struct tasklet_struct tasklet;
|
||
|
|
int state;
|
||
|
|
spinlock_t state_lock; /* spinlock for task state */
|
||
|
|
void *arg;
|
||
|
|
int (*func)(void *arg);
|
||
|
|
int ret;
|
||
|
|
bool destroyed;
|
||
|
|
};
|
||
|
|
|
||
|
|
/*
|
||
|
|
* init rxe_task structure
|
||
|
|
* arg => parameter to pass to fcn
|
||
|
|
* func => function to call until it returns != 0
|
||
|
|
*/
|
||
|
|
int rxe_init_task(struct rxe_task *task, void *arg, int (*func)(void *));
|
||
|
|
|
||
|
|
/* cleanup task */
|
||
|
|
void rxe_cleanup_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
/*
|
||
|
|
* raw call to func in loop without any checking
|
||
|
|
* can call when tasklets are disabled
|
||
|
|
*/
|
||
|
|
int __rxe_do_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
/*
|
||
|
|
* common function called by any of the main tasklets
|
||
|
|
* If there is any chance that there is additional
|
||
|
|
* work to do someone must reschedule the task before
|
||
|
|
* leaving
|
||
|
|
*/
|
||
|
|
void rxe_do_task(struct tasklet_struct *t);
|
||
|
|
|
||
|
|
void rxe_run_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
void rxe_sched_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
/* keep a task from scheduling */
|
||
|
|
void rxe_disable_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
/* allow task to run */
|
||
|
|
void rxe_enable_task(struct rxe_task *task);
|
||
|
|
|
||
|
|
#endif /* RXE_TASK_H */
|