Line data    Source code

       1             : /* SPDX-License-Identifier: GPL-2.0 */
       2             : #ifndef IOPRIO_H
       3             : #define IOPRIO_H
       4             : 
       5             : #include <linux/sched.h>
       6             : #include <linux/sched/rt.h>
       7             : #include <linux/iocontext.h>
       8             : 
       9             : #include <uapi/linux/ioprio.h>
      10             : 
      11             : /*
      12             :  * Default IO priority.
      13             :  */
      14             : #define IOPRIO_DEFAULT  IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0)
      15             : 
      16             : /*
      17             :  * Check that a priority value has a valid class.
      18             :  */
      19             : static inline bool ioprio_valid(unsigned short ioprio)
      20             : {
      21             :         unsigned short class = IOPRIO_PRIO_CLASS(ioprio);
      22             : 
      23             :         return class > IOPRIO_CLASS_NONE && class <= IOPRIO_CLASS_IDLE;
      24             : }
      25             : 
      26             : /*
      27             :  * if process has set io priority explicitly, use that. if not, convert
      28             :  * the cpu scheduler nice value to an io priority
      29             :  */
      30             : static inline int task_nice_ioprio(struct task_struct *task)
      31             : {
      32             :         return (task_nice(task) + 20) / 5;
      33             : }
      34             : 
      35             : /*
      36             :  * This is for the case where the task hasn't asked for a specific IO class.
      37             :  * Check for idle and rt task process, and return appropriate IO class.
      38             :  */
      39             : static inline int task_nice_ioclass(struct task_struct *task)
      40             : {
      41             :         if (task->policy == SCHED_IDLE)
      42             :                 return IOPRIO_CLASS_IDLE;
      43             :         else if (task_is_realtime(task))
      44             :                 return IOPRIO_CLASS_RT;
      45             :         else
      46             :                 return IOPRIO_CLASS_BE;
      47             : }
      48             : 
      49             : #ifdef CONFIG_BLOCK
      50             : int __get_task_ioprio(struct task_struct *p);
      51             : #else
      52             : static inline int __get_task_ioprio(struct task_struct *p)
      53             : {
      54             :         return IOPRIO_DEFAULT;
      55             : }
      56             : #endif /* CONFIG_BLOCK */
      57             : 
      58             : static inline int get_current_ioprio(void)
      59             : {
      60  3090812321 :         return __get_task_ioprio(current);
      61             : }
      62             : 
      63             : extern int set_task_ioprio(struct task_struct *task, int ioprio);
      64             : 
      65             : #ifdef CONFIG_BLOCK
      66             : extern int ioprio_check_cap(int ioprio);
      67             : #else
      68             : static inline int ioprio_check_cap(int ioprio)
      69             : {
      70             :         return -ENOTBLK;
      71             : }
      72             : #endif /* CONFIG_BLOCK */
      73             : 
      74             : #endif