Line data    Source code

       1             : /* SPDX-License-Identifier: GPL-2.0 */
       2             : #ifndef _LINUX_PSI_H
       3             : #define _LINUX_PSI_H
       4             : 
       5             : #include <linux/jump_label.h>
       6             : #include <linux/psi_types.h>
       7             : #include <linux/sched.h>
       8             : #include <linux/poll.h>
       9             : #include <linux/cgroup-defs.h>
      10             : #include <linux/cgroup.h>
      11             : 
      12             : struct seq_file;
      13             : struct css_set;
      14             : 
      15             : #ifdef CONFIG_PSI
      16             : 
      17             : extern struct static_key_false psi_disabled;
      18             : extern struct psi_group psi_system;
      19             : 
      20             : void psi_init(void);
      21             : 
      22             : void psi_memstall_enter(unsigned long *flags);
      23             : void psi_memstall_leave(unsigned long *flags);
      24             : 
      25             : int psi_show(struct seq_file *s, struct psi_group *group, enum psi_res res);
      26             : struct psi_trigger *psi_trigger_create(struct psi_group *group, char *buf,
      27             :                                        enum psi_res res, struct file *file,
      28             :                                        struct kernfs_open_file *of);
      29             : void psi_trigger_destroy(struct psi_trigger *t);
      30             : 
      31             : __poll_t psi_trigger_poll(void **trigger_ptr, struct file *file,
      32             :                         poll_table *wait);
      33             : 
      34             : #ifdef CONFIG_CGROUPS
      35             : static inline struct psi_group *cgroup_psi(struct cgroup *cgrp)
      36             : {
      37             :         return cgroup_ino(cgrp) == 1 ? &psi_system : cgrp->psi;
      38             : }
      39             : 
      40             : int psi_cgroup_alloc(struct cgroup *cgrp);
      41             : void psi_cgroup_free(struct cgroup *cgrp);
      42             : void cgroup_move_task(struct task_struct *p, struct css_set *to);
      43             : void psi_cgroup_restart(struct psi_group *group);
      44             : #endif
      45             : 
      46             : #else /* CONFIG_PSI */
      47             : 
      48             : static inline void psi_init(void) {}
      49             : 
      50           0 : static inline void psi_memstall_enter(unsigned long *flags) {}
      51             : static inline void psi_memstall_leave(unsigned long *flags) {}
      52             : 
      53             : #ifdef CONFIG_CGROUPS
      54             : static inline int psi_cgroup_alloc(struct cgroup *cgrp)
      55             : {
      56             :         return 0;
      57             : }
      58             : static inline void psi_cgroup_free(struct cgroup *cgrp)
      59             : {
      60             : }
      61             : static inline void cgroup_move_task(struct task_struct *p, struct css_set *to)
      62             : {
      63             :         rcu_assign_pointer(p->cgroups, to);
      64             : }
      65             : static inline void psi_cgroup_restart(struct psi_group *group) {}
      66             : #endif
      67             : 
      68             : #endif /* CONFIG_PSI */
      69             : 
      70             : #endif /* _LINUX_PSI_H */