include/linux/cgroup.h

   1 #ifndef _LINUX_CGROUP_H
   2 #define _LINUX_CGROUP_H
   3 /*
   4  *  cgroup interface
   5  *
   6  *  Copyright (C) 2003 BULL SA
   7  *  Copyright (C) 2004-2006 Silicon Graphics, Inc.
   8  *
   9  */
  10
  11 #include <linux/sched.h>
  12 #include <linux/cpumask.h>
  13 #include <linux/nodemask.h>
  14 #include <linux/rculist.h>
  15 #include <linux/cgroupstats.h>
  16 #include <linux/rwsem.h>
  17 #include <linux/fs.h>
  18 #include <linux/seq_file.h>
  19 #include <linux/kernfs.h>
  20 #include <linux/jump_label.h>
  21
  22 #include <linux/cgroup-defs.h>
  23
  24 #ifdef CONFIG_CGROUPS
  25
  26 /*
  27  * All weight knobs on the default hierarhcy should use the following min,
  28  * default and max values.  The default value is the logarithmic center of
  29  * MIN and MAX and allows 100x to be expressed in both directions.
  30  */
  31 #define CGROUP_WEIGHT_MIN               1
  32 #define CGROUP_WEIGHT_DFL               100
  33 #define CGROUP_WEIGHT_MAX               10000
  34
  35 /* a css_task_iter should be treated as an opaque object */
  36 struct css_task_iter {
  37         struct cgroup_subsys            *ss;
  38
  39         struct list_head                *cset_pos;
  40         struct list_head                *cset_head;
  41
  42         struct list_head                *task_pos;
  43         struct list_head                *tasks_head;
  44         struct list_head                *mg_tasks_head;
  45 };
  46
  47 extern struct cgroup_root cgrp_dfl_root;
  48 extern struct css_set init_css_set;
  49
  50 #define SUBSYS(_x) extern struct cgroup_subsys _x ## _cgrp_subsys;
  51 #include <linux/cgroup_subsys.h>
  52 #undef SUBSYS
  53
  54 #define SUBSYS(_x)                                                              \
  55         extern struct static_key_true _x ## _cgrp_subsys_enabled_key;           \
  56         extern struct static_key_true _x ## _cgrp_subsys_on_dfl_key;
  57 #include <linux/cgroup_subsys.h>
  58 #undef SUBSYS
  59
  60 /**
  61  * cgroup_subsys_enabled - fast test on whether a subsys is enabled
  62  * @ss: subsystem in question
  63  */
  64 #define cgroup_subsys_enabled(ss)                                               \
  65         static_branch_likely(&ss ## _enabled_key)
  66
  67 /**
  68  * cgroup_subsys_on_dfl - fast test on whether a subsys is on default hierarchy
  69  * @ss: subsystem in question
  70  */
  71 #define cgroup_subsys_on_dfl(ss)                                                \
  72         static_branch_likely(&ss ## _on_dfl_key)
  73
  74 bool css_has_online_children(struct cgroup_subsys_state *css);
  75 struct cgroup_subsys_state *css_from_id(int id, struct cgroup_subsys *ss);
  76 struct cgroup_subsys_state *cgroup_get_e_css(struct cgroup *cgroup,
  77                                              struct cgroup_subsys *ss);
  78 struct cgroup_subsys_state *css_tryget_online_from_dir(struct dentry *dentry,
  79                                                        struct cgroup_subsys *ss);
  80
  81 bool cgroup_is_descendant(struct cgroup *cgrp, struct cgroup *ancestor);
  82 int cgroup_attach_task_all(struct task_struct *from, struct task_struct *);
  83 int cgroup_transfer_tasks(struct cgroup *to, struct cgroup *from);
  84
  85 int cgroup_add_dfl_cftypes(struct cgroup_subsys *ss, struct cftype *cfts);
  86 int cgroup_add_legacy_cftypes(struct cgroup_subsys *ss, struct cftype *cfts);
  87 int cgroup_rm_cftypes(struct cftype *cfts);
  88
  89 char *task_cgroup_path(struct task_struct *task, char *buf, size_t buflen);
  90 int cgroupstats_build(struct cgroupstats *stats, struct dentry *dentry);
  91 int proc_cgroup_show(struct seq_file *m, struct pid_namespace *ns,
  92                      struct pid *pid, struct task_struct *tsk);
  93
  94 void cgroup_fork(struct task_struct *p);
  95 extern int cgroup_can_fork(struct task_struct *p,
  96                            void *ss_priv[CGROUP_CANFORK_COUNT]);
  97 extern void cgroup_cancel_fork(struct task_struct *p,
  98                                void *ss_priv[CGROUP_CANFORK_COUNT]);
  99 extern void cgroup_post_fork(struct task_struct *p,
 100                              void *old_ss_priv[CGROUP_CANFORK_COUNT]);
 101 void cgroup_exit(struct task_struct *p);
 102
 103 int cgroup_init_early(void);
 104 int cgroup_init(void);
 105
 106 /*
 107  * Iteration helpers and macros.
 108  */
 109
 110 struct cgroup_subsys_state *css_next_child(struct cgroup_subsys_state *pos,
 111                                            struct cgroup_subsys_state *parent);
 112 struct cgroup_subsys_state *css_next_descendant_pre(struct cgroup_subsys_state *pos,
 113                                                     struct cgroup_subsys_state *css);
 114 struct cgroup_subsys_state *css_rightmost_descendant(struct cgroup_subsys_state *pos);
 115 struct cgroup_subsys_state *css_next_descendant_post(struct cgroup_subsys_state *pos,
 116                                                      struct cgroup_subsys_state *css);
 117
 118 struct task_struct *cgroup_taskset_first(struct cgroup_taskset *tset);
 119 struct task_struct *cgroup_taskset_next(struct cgroup_taskset *tset);
 120
 121 void css_task_iter_start(struct cgroup_subsys_state *css,
 122                          struct css_task_iter *it);
 123 struct task_struct *css_task_iter_next(struct css_task_iter *it);
 124 void css_task_iter_end(struct css_task_iter *it);
 125
 126 /**
 127  * css_for_each_child - iterate through children of a css
 128  * @pos: the css * to use as the loop cursor
 129  * @parent: css whose children to walk
 130  *
 131  * Walk @parent's children.  Must be called under rcu_read_lock().
 132  *
 133  * If a subsystem synchronizes ->css_online() and the start of iteration, a
 134  * css which finished ->css_online() is guaranteed to be visible in the
 135  * future iterations and will stay visible until the last reference is put.
 136  * A css which hasn't finished ->css_online() or already finished
 137  * ->css_offline() may show up during traversal.  It's each subsystem's
 138  * responsibility to synchronize against on/offlining.
 139  *
 140  * It is allowed to temporarily drop RCU read lock during iteration.  The
 141  * caller is responsible for ensuring that @pos remains accessible until
 142  * the start of the next iteration by, for example, bumping the css refcnt.
 143  */
 144 #define css_for_each_child(pos, parent)                                 \
 145         for ((pos) = css_next_child(NULL, (parent)); (pos);             \
 146              (pos) = css_next_child((pos), (parent)))
 147
 148 /**
 149  * css_for_each_descendant_pre - pre-order walk of a css's descendants
 150  * @pos: the css * to use as the loop cursor
 151  * @root: css whose descendants to walk
 152  *
 153  * Walk @root's descendants.  @root is included in the iteration and the
 154  * first node to be visited.  Must be called under rcu_read_lock().
 155  *
 156  * If a subsystem synchronizes ->css_online() and the start of iteration, a
 157  * css which finished ->css_online() is guaranteed to be visible in the
 158  * future iterations and will stay visible until the last reference is put.
 159  * A css which hasn't finished ->css_online() or already finished
 160  * ->css_offline() may show up during traversal.  It's each subsystem's
 161  * responsibility to synchronize against on/offlining.
 162  *
 163  * For example, the following guarantees that a descendant can't escape
 164  * state updates of its ancestors.
 165  *
 166  * my_online(@css)
 167  * {
 168  *      Lock @css's parent and @css;
 169  *      Inherit state from the parent;
 170  *      Unlock both.
 171  * }
 172  *
 173  * my_update_state(@css)
 174  * {
 175  *      css_for_each_descendant_pre(@pos, @css) {
 176  *              Lock @pos;
 177  *              if (@pos == @css)
 178  *                      Update @css's state;
 179  *              else
 180  *                      Verify @pos is alive and inherit state from its parent;
 181  *              Unlock @pos;
 182  *      }
 183  * }
 184  *
 185  * As long as the inheriting step, including checking the parent state, is
 186  * enclosed inside @pos locking, double-locking the parent isn't necessary
 187  * while inheriting.  The state update to the parent is guaranteed to be
 188  * visible by walking order and, as long as inheriting operations to the
 189  * same @pos are atomic to each other, multiple updates racing each other
 190  * still result in the correct state.  It's guaranateed that at least one
 191  * inheritance happens for any css after the latest update to its parent.
 192  *
 193  * If checking parent's state requires locking the parent, each inheriting
 194  * iteration should lock and unlock both @pos->parent and @pos.
 195  *
 196  * Alternatively, a subsystem may choose to use a single global lock to
 197  * synchronize ->css_online() and ->css_offline() against tree-walking
 198  * operations.
 199  *
 200  * It is allowed to temporarily drop RCU read lock during iteration.  The
 201  * caller is responsible for ensuring that @pos remains accessible until
 202  * the start of the next iteration by, for example, bumping the css refcnt.
 203  */
 204 #define css_for_each_descendant_pre(pos, css)                           \
 205         for ((pos) = css_next_descendant_pre(NULL, (css)); (pos);       \
 206              (pos) = css_next_descendant_pre((pos), (css)))
 207
 208 /**
 209  * css_for_each_descendant_post - post-order walk of a css's descendants
 210  * @pos: the css * to use as the loop cursor
 211  * @css: css whose descendants to walk
 212  *
 213  * Similar to css_for_each_descendant_pre() but performs post-order
 214  * traversal instead.  @root is included in the iteration and the last
 215  * node to be visited.
 216  *
 217  * If a subsystem synchronizes ->css_online() and the start of iteration, a
 218  * css which finished ->css_online() is guaranteed to be visible in the
 219  * future iterations and will stay visible until the last reference is put.
 220  * A css which hasn't finished ->css_online() or already finished
 221  * ->css_offline() may show up during traversal.  It's each subsystem's
 222  * responsibility to synchronize against on/offlining.
 223  *
 224  * Note that the walk visibility guarantee example described in pre-order
 225  * walk doesn't apply the same to post-order walks.
 226  */
 227 #define css_for_each_descendant_post(pos, css)                          \
 228         for ((pos) = css_next_descendant_post(NULL, (css)); (pos);      \
 229              (pos) = css_next_descendant_post((pos), (css)))
 230
 231 /**
 232  * cgroup_taskset_for_each - iterate cgroup_taskset
 233  * @task: the loop cursor
 234  * @tset: taskset to iterate
 235  */
 236 #define cgroup_taskset_for_each(task, tset)                             \
 237         for ((task) = cgroup_taskset_first((tset)); (task);             \
 238              (task) = cgroup_taskset_next((tset)))
 239
 240 /*
 241  * Inline functions.
 242  */
 243
 244 /**
 245  * css_get - obtain a reference on the specified css
 246  * @css: target css
 247  *
 248  * The caller must already have a reference.
 249  */
 250 static inline void css_get(struct cgroup_subsys_state *css)
 251 {
 252         if (!(css->flags & CSS_NO_REF))
 253                 percpu_ref_get(&css->refcnt);
 254 }
 255
 256 /**
 257  * css_get_many - obtain references on the specified css
 258  * @css: target css
 259  * @n: number of references to get
 260  *
 261  * The caller must already have a reference.
 262  */
 263 static inline void css_get_many(struct cgroup_subsys_state *css, unsigned int n)
 264 {
 265         if (!(css->flags & CSS_NO_REF))
 266                 percpu_ref_get_many(&css->refcnt, n);
 267 }
 268
 269 /**
 270  * css_tryget - try to obtain a reference on the specified css
 271  * @css: target css
 272  *
 273  * Obtain a reference on @css unless it already has reached zero and is
 274  * being released.  This function doesn't care whether @css is on or
 275  * offline.  The caller naturally needs to ensure that @css is accessible
 276  * but doesn't have to be holding a reference on it - IOW, RCU protected
 277  * access is good enough for this function.  Returns %true if a reference
 278  * count was successfully obtained; %false otherwise.
 279  */
 280 static inline bool css_tryget(struct cgroup_subsys_state *css)
 281 {
 282         if (!(css->flags & CSS_NO_REF))
 283                 return percpu_ref_tryget(&css->refcnt);
 284         return true;
 285 }
 286
 287 /**
 288  * css_tryget_online - try to obtain a reference on the specified css if online
 289  * @css: target css
 290  *
 291  * Obtain a reference on @css if it's online.  The caller naturally needs
 292  * to ensure that @css is accessible but doesn't have to be holding a
 293  * reference on it - IOW, RCU protected access is good enough for this
 294  * function.  Returns %true if a reference count was successfully obtained;
 295  * %false otherwise.
 296  */
 297 static inline bool css_tryget_online(struct cgroup_subsys_state *css)
 298 {
 299         if (!(css->flags & CSS_NO_REF))
 300                 return percpu_ref_tryget_live(&css->refcnt);
 301         return true;
 302 }
 303
 304 /**
 305  * css_put - put a css reference
 306  * @css: target css
 307  *
 308  * Put a reference obtained via css_get() and css_tryget_online().
 309  */
 310 static inline void css_put(struct cgroup_subsys_state *css)
 311 {
 312         if (!(css->flags & CSS_NO_REF))
 313                 percpu_ref_put(&css->refcnt);
 314 }
 315
 316 /**
 317  * css_put_many - put css references
 318  * @css: target css
 319  * @n: number of references to put
 320  *
 321  * Put references obtained via css_get() and css_tryget_online().
 322  */
 323 static inline void css_put_many(struct cgroup_subsys_state *css, unsigned int n)
 324 {
 325         if (!(css->flags & CSS_NO_REF))
 326                 percpu_ref_put_many(&css->refcnt, n);
 327 }
 328
 329 /**
 330  * task_css_set_check - obtain a task's css_set with extra access conditions
 331  * @task: the task to obtain css_set for
 332  * @__c: extra condition expression to be passed to rcu_dereference_check()
 333  *
 334  * A task's css_set is RCU protected, initialized and exited while holding
 335  * task_lock(), and can only be modified while holding both cgroup_mutex
 336  * and task_lock() while the task is alive.  This macro verifies that the
 337  * caller is inside proper critical section and returns @task's css_set.
 338  *
 339  * The caller can also specify additional allowed conditions via @__c, such
 340  * as locks used during the cgroup_subsys::attach() methods.
 341  */
 342 #ifdef CONFIG_PROVE_RCU
 343 extern struct mutex cgroup_mutex;
 344 extern struct rw_semaphore css_set_rwsem;
 345 #define task_css_set_check(task, __c)                                   \
 346         rcu_dereference_check((task)->cgroups,                          \
 347                 lockdep_is_held(&cgroup_mutex) ||                       \
 348                 lockdep_is_held(&css_set_rwsem) ||                      \
 349                 ((task)->flags & PF_EXITING) || (__c))
 350 #else
 351 #define task_css_set_check(task, __c)                                   \
 352         rcu_dereference((task)->cgroups)
 353 #endif
 354
 355 /**
 356  * task_css_check - obtain css for (task, subsys) w/ extra access conds
 357  * @task: the target task
 358  * @subsys_id: the target subsystem ID
 359  * @__c: extra condition expression to be passed to rcu_dereference_check()
 360  *
 361  * Return the cgroup_subsys_state for the (@task, @subsys_id) pair.  The
 362  * synchronization rules are the same as task_css_set_check().
 363  */
 364 #define task_css_check(task, subsys_id, __c)                            \
 365         task_css_set_check((task), (__c))->subsys[(subsys_id)]
 366
 367 /**
 368  * task_css_set - obtain a task's css_set
 369  * @task: the task to obtain css_set for
 370  *
 371  * See task_css_set_check().
 372  */
 373 static inline struct css_set *task_css_set(struct task_struct *task)
 374 {
 375         return task_css_set_check(task, false);
 376 }
 377
 378 /**
 379  * task_css - obtain css for (task, subsys)
 380  * @task: the target task
 381  * @subsys_id: the target subsystem ID
 382  *
 383  * See task_css_check().
 384  */
 385 static inline struct cgroup_subsys_state *task_css(struct task_struct *task,
 386                                                    int subsys_id)
 387 {
 388         return task_css_check(task, subsys_id, false);
 389 }
 390
 391 /**
 392  * task_get_css - find and get the css for (task, subsys)
 393  * @task: the target task
 394  * @subsys_id: the target subsystem ID
 395  *
 396  * Find the css for the (@task, @subsys_id) combination, increment a
 397  * reference on and return it.  This function is guaranteed to return a
 398  * valid css.
 399  */
 400 static inline struct cgroup_subsys_state *
 401 task_get_css(struct task_struct *task, int subsys_id)
 402 {
 403         struct cgroup_subsys_state *css;
 404
 405         rcu_read_lock();
 406         while (true) {
 407                 css = task_css(task, subsys_id);
 408                 if (likely(css_tryget_online(css)))
 409                         break;
 410                 cpu_relax();
 411         }
 412         rcu_read_unlock();
 413         return css;
 414 }
 415
 416 /**
 417  * task_css_is_root - test whether a task belongs to the root css
 418  * @task: the target task
 419  * @subsys_id: the target subsystem ID
 420  *
 421  * Test whether @task belongs to the root css on the specified subsystem.
 422  * May be invoked in any context.
 423  */
 424 static inline bool task_css_is_root(struct task_struct *task, int subsys_id)
 425 {
 426         return task_css_check(task, subsys_id, true) ==
 427                 init_css_set.subsys[subsys_id];
 428 }
 429
 430 static inline struct cgroup *task_cgroup(struct task_struct *task,
 431                                          int subsys_id)
 432 {
 433         return task_css(task, subsys_id)->cgroup;
 434 }
 435
 436 /* no synchronization, the result can only be used as a hint */
 437 static inline bool cgroup_has_tasks(struct cgroup *cgrp)
 438 {
 439         return !list_empty(&cgrp->cset_links);
 440 }
 441
 442 /* returns ino associated with a cgroup */
 443 static inline ino_t cgroup_ino(struct cgroup *cgrp)
 444 {
 445         return cgrp->kn->ino;
 446 }
 447
 448 /* cft/css accessors for cftype->write() operation */
 449 static inline struct cftype *of_cft(struct kernfs_open_file *of)
 450 {
 451         return of->kn->priv;
 452 }
 453
 454 struct cgroup_subsys_state *of_css(struct kernfs_open_file *of);
 455
 456 /* cft/css accessors for cftype->seq_*() operations */
 457 static inline struct cftype *seq_cft(struct seq_file *seq)
 458 {
 459         return of_cft(seq->private);
 460 }
 461
 462 static inline struct cgroup_subsys_state *seq_css(struct seq_file *seq)
 463 {
 464         return of_css(seq->private);
 465 }
 466
 467 /*
 468  * Name / path handling functions.  All are thin wrappers around the kernfs
 469  * counterparts and can be called under any context.
 470  */
 471
 472 static inline int cgroup_name(struct cgroup *cgrp, char *buf, size_t buflen)
 473 {
 474         return kernfs_name(cgrp->kn, buf, buflen);
 475 }
 476
 477 static inline char * __must_check cgroup_path(struct cgroup *cgrp, char *buf,
 478                                               size_t buflen)
 479 {
 480         return kernfs_path(cgrp->kn, buf, buflen);
 481 }
 482
 483 static inline void pr_cont_cgroup_name(struct cgroup *cgrp)
 484 {
 485         pr_cont_kernfs_name(cgrp->kn);
 486 }
 487
 488 static inline void pr_cont_cgroup_path(struct cgroup *cgrp)
 489 {
 490         pr_cont_kernfs_path(cgrp->kn);
 491 }
 492
 493 /**
 494  * cgroup_file_notify - generate a file modified event for a cgroup_file
 495  * @cfile: target cgroup_file
 496  *
 497  * @cfile must have been obtained by setting cftype->file_offset.
 498  */
 499 static inline void cgroup_file_notify(struct cgroup_file *cfile)
 500 {
 501         /* might not have been created due to one of the CFTYPE selector flags */
 502         if (cfile->kn)
 503                 kernfs_notify(cfile->kn);
 504 }
 505
 506 #else /* !CONFIG_CGROUPS */
 507
 508 struct cgroup_subsys_state;
 509
 510 static inline void css_put(struct cgroup_subsys_state *css) {}
 511 static inline int cgroup_attach_task_all(struct task_struct *from,
 512                                          struct task_struct *t) { return 0; }
 513 static inline int cgroupstats_build(struct cgroupstats *stats,
 514                                     struct dentry *dentry) { return -EINVAL; }
 515
 516 static inline void cgroup_fork(struct task_struct *p) {}
 517 static inline int cgroup_can_fork(struct task_struct *p,
 518                                   void *ss_priv[CGROUP_CANFORK_COUNT])
 519 { return 0; }
 520 static inline void cgroup_cancel_fork(struct task_struct *p,
 521                                       void *ss_priv[CGROUP_CANFORK_COUNT]) {}
 522 static inline void cgroup_post_fork(struct task_struct *p,
 523                                     void *ss_priv[CGROUP_CANFORK_COUNT]) {}
 524 static inline void cgroup_exit(struct task_struct *p) {}
 525
 526 static inline int cgroup_init_early(void) { return 0; }
 527 static inline int cgroup_init(void) { return 0; }
 528
 529 #endif /* !CONFIG_CGROUPS */
 530
 531 #endif /* _LINUX_CGROUP_H */