cgroup.h 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610
  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. #include <linux/sched.h>
  11. #include <linux/cpumask.h>
  12. #include <linux/nodemask.h>
  13. #include <linux/rcupdate.h>
  14. #include <linux/cgroupstats.h>
  15. #include <linux/prio_heap.h>
  16. #include <linux/rwsem.h>
  17. #include <linux/idr.h>
  18. #include <linux/workqueue.h>
  19. #ifdef CONFIG_CGROUPS
  20. struct cgroupfs_root;
  21. struct cgroup_subsys;
  22. struct inode;
  23. struct cgroup;
  24. struct css_id;
  25. extern int cgroup_init_early(void);
  26. extern int cgroup_init(void);
  27. extern void cgroup_lock(void);
  28. extern int cgroup_lock_is_held(void);
  29. extern bool cgroup_lock_live_group(struct cgroup *cgrp);
  30. extern void cgroup_unlock(void);
  31. extern void cgroup_fork(struct task_struct *p);
  32. extern void cgroup_fork_callbacks(struct task_struct *p);
  33. extern void cgroup_post_fork(struct task_struct *p);
  34. extern void cgroup_exit(struct task_struct *p, int run_callbacks);
  35. extern int cgroupstats_build(struct cgroupstats *stats,
  36. struct dentry *dentry);
  37. extern int cgroup_load_subsys(struct cgroup_subsys *ss);
  38. extern void cgroup_unload_subsys(struct cgroup_subsys *ss);
  39. extern const struct file_operations proc_cgroup_operations;
  40. /* Define the enumeration of all builtin cgroup subsystems */
  41. #define SUBSYS(_x) _x ## _subsys_id,
  42. enum cgroup_subsys_id {
  43. #include <linux/cgroup_subsys.h>
  44. CGROUP_BUILTIN_SUBSYS_COUNT
  45. };
  46. #undef SUBSYS
  47. /*
  48. * This define indicates the maximum number of subsystems that can be loaded
  49. * at once. We limit to this many since cgroupfs_root has subsys_bits to keep
  50. * track of all of them.
  51. */
  52. #define CGROUP_SUBSYS_COUNT (BITS_PER_BYTE*sizeof(unsigned long))
  53. /* Per-subsystem/per-cgroup state maintained by the system. */
  54. struct cgroup_subsys_state {
  55. /*
  56. * The cgroup that this subsystem is attached to. Useful
  57. * for subsystems that want to know about the cgroup
  58. * hierarchy structure
  59. */
  60. struct cgroup *cgroup;
  61. /*
  62. * State maintained by the cgroup system to allow subsystems
  63. * to be "busy". Should be accessed via css_get(),
  64. * css_tryget() and and css_put().
  65. */
  66. atomic_t refcnt;
  67. unsigned long flags;
  68. /* ID for this css, if possible */
  69. struct css_id __rcu *id;
  70. /* Used to put @cgroup->dentry on the last css_put() */
  71. struct work_struct dput_work;
  72. };
  73. /* bits in struct cgroup_subsys_state flags field */
  74. enum {
  75. CSS_ROOT, /* This CSS is the root of the subsystem */
  76. };
  77. /* Caller must verify that the css is not for root cgroup */
  78. static inline void __css_get(struct cgroup_subsys_state *css, int count)
  79. {
  80. atomic_add(count, &css->refcnt);
  81. }
  82. /*
  83. * Call css_get() to hold a reference on the css; it can be used
  84. * for a reference obtained via:
  85. * - an existing ref-counted reference to the css
  86. * - task->cgroups for a locked task
  87. */
  88. static inline void css_get(struct cgroup_subsys_state *css)
  89. {
  90. /* We don't need to reference count the root state */
  91. if (!test_bit(CSS_ROOT, &css->flags))
  92. __css_get(css, 1);
  93. }
  94. /*
  95. * Call css_tryget() to take a reference on a css if your existing
  96. * (known-valid) reference isn't already ref-counted. Returns false if
  97. * the css has been destroyed.
  98. */
  99. extern bool __css_tryget(struct cgroup_subsys_state *css);
  100. static inline bool css_tryget(struct cgroup_subsys_state *css)
  101. {
  102. if (test_bit(CSS_ROOT, &css->flags))
  103. return true;
  104. return __css_tryget(css);
  105. }
  106. /*
  107. * css_put() should be called to release a reference taken by
  108. * css_get() or css_tryget()
  109. */
  110. extern void __css_put(struct cgroup_subsys_state *css);
  111. static inline void css_put(struct cgroup_subsys_state *css)
  112. {
  113. if (!test_bit(CSS_ROOT, &css->flags))
  114. __css_put(css);
  115. }
  116. /* bits in struct cgroup flags field */
  117. enum {
  118. /* Control Group is dead */
  119. CGRP_REMOVED,
  120. /*
  121. * Control Group has previously had a child cgroup or a task,
  122. * but no longer (only if CGRP_NOTIFY_ON_RELEASE is set)
  123. */
  124. CGRP_RELEASABLE,
  125. /* Control Group requires release notifications to userspace */
  126. CGRP_NOTIFY_ON_RELEASE,
  127. /*
  128. * Clone cgroup values when creating a new child cgroup
  129. */
  130. CGRP_CLONE_CHILDREN,
  131. };
  132. struct cgroup {
  133. unsigned long flags; /* "unsigned long" so bitops work */
  134. /*
  135. * count users of this cgroup. >0 means busy, but doesn't
  136. * necessarily indicate the number of tasks in the cgroup
  137. */
  138. atomic_t count;
  139. /*
  140. * We link our 'sibling' struct into our parent's 'children'.
  141. * Our children link their 'sibling' into our 'children'.
  142. */
  143. struct list_head sibling; /* my parent's children */
  144. struct list_head children; /* my children */
  145. struct list_head files; /* my files */
  146. struct cgroup *parent; /* my parent */
  147. struct dentry __rcu *dentry; /* cgroup fs entry, RCU protected */
  148. /* Private pointers for each registered subsystem */
  149. struct cgroup_subsys_state *subsys[CGROUP_SUBSYS_COUNT];
  150. struct cgroupfs_root *root;
  151. struct cgroup *top_cgroup;
  152. /*
  153. * List of cg_cgroup_links pointing at css_sets with
  154. * tasks in this cgroup. Protected by css_set_lock
  155. */
  156. struct list_head css_sets;
  157. struct list_head allcg_node; /* cgroupfs_root->allcg_list */
  158. struct list_head cft_q_node; /* used during cftype add/rm */
  159. /*
  160. * Linked list running through all cgroups that can
  161. * potentially be reaped by the release agent. Protected by
  162. * release_list_lock
  163. */
  164. struct list_head release_list;
  165. /*
  166. * list of pidlists, up to two for each namespace (one for procs, one
  167. * for tasks); created on demand.
  168. */
  169. struct list_head pidlists;
  170. struct mutex pidlist_mutex;
  171. /* For RCU-protected deletion */
  172. struct rcu_head rcu_head;
  173. /* List of events which userspace want to receive */
  174. struct list_head event_list;
  175. spinlock_t event_list_lock;
  176. };
  177. /*
  178. * A css_set is a structure holding pointers to a set of
  179. * cgroup_subsys_state objects. This saves space in the task struct
  180. * object and speeds up fork()/exit(), since a single inc/dec and a
  181. * list_add()/del() can bump the reference count on the entire cgroup
  182. * set for a task.
  183. */
  184. struct css_set {
  185. /* Reference count */
  186. atomic_t refcount;
  187. /*
  188. * List running through all cgroup groups in the same hash
  189. * slot. Protected by css_set_lock
  190. */
  191. struct hlist_node hlist;
  192. /*
  193. * List running through all tasks using this cgroup
  194. * group. Protected by css_set_lock
  195. */
  196. struct list_head tasks;
  197. /*
  198. * List of cg_cgroup_link objects on link chains from
  199. * cgroups referenced from this css_set. Protected by
  200. * css_set_lock
  201. */
  202. struct list_head cg_links;
  203. /*
  204. * Set of subsystem states, one for each subsystem. This array
  205. * is immutable after creation apart from the init_css_set
  206. * during subsystem registration (at boot time) and modular subsystem
  207. * loading/unloading.
  208. */
  209. struct cgroup_subsys_state *subsys[CGROUP_SUBSYS_COUNT];
  210. /* For RCU-protected deletion */
  211. struct rcu_head rcu_head;
  212. };
  213. /*
  214. * cgroup_map_cb is an abstract callback API for reporting map-valued
  215. * control files
  216. */
  217. struct cgroup_map_cb {
  218. int (*fill)(struct cgroup_map_cb *cb, const char *key, u64 value);
  219. void *state;
  220. };
  221. /*
  222. * struct cftype: handler definitions for cgroup control files
  223. *
  224. * When reading/writing to a file:
  225. * - the cgroup to use is file->f_dentry->d_parent->d_fsdata
  226. * - the 'cftype' of the file is file->f_dentry->d_fsdata
  227. */
  228. /* cftype->flags */
  229. #define CFTYPE_ONLY_ON_ROOT (1U << 0) /* only create on root cg */
  230. #define CFTYPE_NOT_ON_ROOT (1U << 1) /* don't create onp root cg */
  231. #define MAX_CFTYPE_NAME 64
  232. struct cftype {
  233. /*
  234. * By convention, the name should begin with the name of the
  235. * subsystem, followed by a period. Zero length string indicates
  236. * end of cftype array.
  237. */
  238. char name[MAX_CFTYPE_NAME];
  239. int private;
  240. /*
  241. * If not 0, file mode is set to this value, otherwise it will
  242. * be figured out automatically
  243. */
  244. umode_t mode;
  245. /*
  246. * If non-zero, defines the maximum length of string that can
  247. * be passed to write_string; defaults to 64
  248. */
  249. size_t max_write_len;
  250. /* CFTYPE_* flags */
  251. unsigned int flags;
  252. int (*open)(struct inode *inode, struct file *file);
  253. ssize_t (*read)(struct cgroup *cgrp, struct cftype *cft,
  254. struct file *file,
  255. char __user *buf, size_t nbytes, loff_t *ppos);
  256. /*
  257. * read_u64() is a shortcut for the common case of returning a
  258. * single integer. Use it in place of read()
  259. */
  260. u64 (*read_u64)(struct cgroup *cgrp, struct cftype *cft);
  261. /*
  262. * read_s64() is a signed version of read_u64()
  263. */
  264. s64 (*read_s64)(struct cgroup *cgrp, struct cftype *cft);
  265. /*
  266. * read_map() is used for defining a map of key/value
  267. * pairs. It should call cb->fill(cb, key, value) for each
  268. * entry. The key/value pairs (and their ordering) should not
  269. * change between reboots.
  270. */
  271. int (*read_map)(struct cgroup *cont, struct cftype *cft,
  272. struct cgroup_map_cb *cb);
  273. /*
  274. * read_seq_string() is used for outputting a simple sequence
  275. * using seqfile.
  276. */
  277. int (*read_seq_string)(struct cgroup *cont, struct cftype *cft,
  278. struct seq_file *m);
  279. ssize_t (*write)(struct cgroup *cgrp, struct cftype *cft,
  280. struct file *file,
  281. const char __user *buf, size_t nbytes, loff_t *ppos);
  282. /*
  283. * write_u64() is a shortcut for the common case of accepting
  284. * a single integer (as parsed by simple_strtoull) from
  285. * userspace. Use in place of write(); return 0 or error.
  286. */
  287. int (*write_u64)(struct cgroup *cgrp, struct cftype *cft, u64 val);
  288. /*
  289. * write_s64() is a signed version of write_u64()
  290. */
  291. int (*write_s64)(struct cgroup *cgrp, struct cftype *cft, s64 val);
  292. /*
  293. * write_string() is passed a nul-terminated kernelspace
  294. * buffer of maximum length determined by max_write_len.
  295. * Returns 0 or -ve error code.
  296. */
  297. int (*write_string)(struct cgroup *cgrp, struct cftype *cft,
  298. const char *buffer);
  299. /*
  300. * trigger() callback can be used to get some kick from the
  301. * userspace, when the actual string written is not important
  302. * at all. The private field can be used to determine the
  303. * kick type for multiplexing.
  304. */
  305. int (*trigger)(struct cgroup *cgrp, unsigned int event);
  306. int (*release)(struct inode *inode, struct file *file);
  307. /*
  308. * register_event() callback will be used to add new userspace
  309. * waiter for changes related to the cftype. Implement it if
  310. * you want to provide this functionality. Use eventfd_signal()
  311. * on eventfd to send notification to userspace.
  312. */
  313. int (*register_event)(struct cgroup *cgrp, struct cftype *cft,
  314. struct eventfd_ctx *eventfd, const char *args);
  315. /*
  316. * unregister_event() callback will be called when userspace
  317. * closes the eventfd or on cgroup removing.
  318. * This callback must be implemented, if you want provide
  319. * notification functionality.
  320. */
  321. void (*unregister_event)(struct cgroup *cgrp, struct cftype *cft,
  322. struct eventfd_ctx *eventfd);
  323. };
  324. /*
  325. * cftype_sets describe cftypes belonging to a subsystem and are chained at
  326. * cgroup_subsys->cftsets. Each cftset points to an array of cftypes
  327. * terminated by zero length name.
  328. */
  329. struct cftype_set {
  330. struct list_head node; /* chained at subsys->cftsets */
  331. const struct cftype *cfts;
  332. };
  333. struct cgroup_scanner {
  334. struct cgroup *cg;
  335. int (*test_task)(struct task_struct *p, struct cgroup_scanner *scan);
  336. void (*process_task)(struct task_struct *p,
  337. struct cgroup_scanner *scan);
  338. struct ptr_heap *heap;
  339. void *data;
  340. };
  341. int cgroup_add_cftypes(struct cgroup_subsys *ss, const struct cftype *cfts);
  342. int cgroup_rm_cftypes(struct cgroup_subsys *ss, const struct cftype *cfts);
  343. int cgroup_is_removed(const struct cgroup *cgrp);
  344. int cgroup_path(const struct cgroup *cgrp, char *buf, int buflen);
  345. int cgroup_task_count(const struct cgroup *cgrp);
  346. /* Return true if cgrp is a descendant of the task's cgroup */
  347. int cgroup_is_descendant(const struct cgroup *cgrp, struct task_struct *task);
  348. /*
  349. * Control Group taskset, used to pass around set of tasks to cgroup_subsys
  350. * methods.
  351. */
  352. struct cgroup_taskset;
  353. struct task_struct *cgroup_taskset_first(struct cgroup_taskset *tset);
  354. struct task_struct *cgroup_taskset_next(struct cgroup_taskset *tset);
  355. struct cgroup *cgroup_taskset_cur_cgroup(struct cgroup_taskset *tset);
  356. int cgroup_taskset_size(struct cgroup_taskset *tset);
  357. /**
  358. * cgroup_taskset_for_each - iterate cgroup_taskset
  359. * @task: the loop cursor
  360. * @skip_cgrp: skip if task's cgroup matches this, %NULL to iterate through all
  361. * @tset: taskset to iterate
  362. */
  363. #define cgroup_taskset_for_each(task, skip_cgrp, tset) \
  364. for ((task) = cgroup_taskset_first((tset)); (task); \
  365. (task) = cgroup_taskset_next((tset))) \
  366. if (!(skip_cgrp) || \
  367. cgroup_taskset_cur_cgroup((tset)) != (skip_cgrp))
  368. /*
  369. * Control Group subsystem type.
  370. * See Documentation/cgroups/cgroups.txt for details
  371. */
  372. struct cgroup_subsys {
  373. struct cgroup_subsys_state *(*create)(struct cgroup *cgrp);
  374. int (*pre_destroy)(struct cgroup *cgrp);
  375. void (*destroy)(struct cgroup *cgrp);
  376. int (*can_attach)(struct cgroup *cgrp, struct cgroup_taskset *tset);
  377. void (*cancel_attach)(struct cgroup *cgrp, struct cgroup_taskset *tset);
  378. void (*attach)(struct cgroup *cgrp, struct cgroup_taskset *tset);
  379. void (*fork)(struct task_struct *task);
  380. void (*exit)(struct cgroup *cgrp, struct cgroup *old_cgrp,
  381. struct task_struct *task);
  382. void (*post_clone)(struct cgroup *cgrp);
  383. void (*bind)(struct cgroup *root);
  384. int subsys_id;
  385. int active;
  386. int disabled;
  387. int early_init;
  388. /*
  389. * True if this subsys uses ID. ID is not available before cgroup_init()
  390. * (not available in early_init time.)
  391. */
  392. bool use_id;
  393. #define MAX_CGROUP_TYPE_NAMELEN 32
  394. const char *name;
  395. /*
  396. * Link to parent, and list entry in parent's children.
  397. * Protected by cgroup_lock()
  398. */
  399. struct cgroupfs_root *root;
  400. struct list_head sibling;
  401. /* used when use_id == true */
  402. struct idr idr;
  403. spinlock_t id_lock;
  404. /* list of cftype_sets */
  405. struct list_head cftsets;
  406. /* base cftypes, automatically [de]registered with subsys itself */
  407. struct cftype *base_cftypes;
  408. struct cftype_set base_cftset;
  409. /* should be defined only by modular subsystems */
  410. struct module *module;
  411. };
  412. #define SUBSYS(_x) extern struct cgroup_subsys _x ## _subsys;
  413. #include <linux/cgroup_subsys.h>
  414. #undef SUBSYS
  415. static inline struct cgroup_subsys_state *cgroup_subsys_state(
  416. struct cgroup *cgrp, int subsys_id)
  417. {
  418. return cgrp->subsys[subsys_id];
  419. }
  420. /*
  421. * function to get the cgroup_subsys_state which allows for extra
  422. * rcu_dereference_check() conditions, such as locks used during the
  423. * cgroup_subsys::attach() methods.
  424. */
  425. #define task_subsys_state_check(task, subsys_id, __c) \
  426. rcu_dereference_check(task->cgroups->subsys[subsys_id], \
  427. lockdep_is_held(&task->alloc_lock) || \
  428. cgroup_lock_is_held() || (__c))
  429. static inline struct cgroup_subsys_state *
  430. task_subsys_state(struct task_struct *task, int subsys_id)
  431. {
  432. return task_subsys_state_check(task, subsys_id, false);
  433. }
  434. static inline struct cgroup* task_cgroup(struct task_struct *task,
  435. int subsys_id)
  436. {
  437. return task_subsys_state(task, subsys_id)->cgroup;
  438. }
  439. /* A cgroup_iter should be treated as an opaque object */
  440. struct cgroup_iter {
  441. struct list_head *cg_link;
  442. struct list_head *task;
  443. };
  444. /*
  445. * To iterate across the tasks in a cgroup:
  446. *
  447. * 1) call cgroup_iter_start to initialize an iterator
  448. *
  449. * 2) call cgroup_iter_next() to retrieve member tasks until it
  450. * returns NULL or until you want to end the iteration
  451. *
  452. * 3) call cgroup_iter_end() to destroy the iterator.
  453. *
  454. * Or, call cgroup_scan_tasks() to iterate through every task in a
  455. * cgroup - cgroup_scan_tasks() holds the css_set_lock when calling
  456. * the test_task() callback, but not while calling the process_task()
  457. * callback.
  458. */
  459. void cgroup_iter_start(struct cgroup *cgrp, struct cgroup_iter *it);
  460. struct task_struct *cgroup_iter_next(struct cgroup *cgrp,
  461. struct cgroup_iter *it);
  462. void cgroup_iter_end(struct cgroup *cgrp, struct cgroup_iter *it);
  463. int cgroup_scan_tasks(struct cgroup_scanner *scan);
  464. int cgroup_attach_task(struct cgroup *, struct task_struct *);
  465. int cgroup_attach_task_all(struct task_struct *from, struct task_struct *);
  466. /*
  467. * CSS ID is ID for cgroup_subsys_state structs under subsys. This only works
  468. * if cgroup_subsys.use_id == true. It can be used for looking up and scanning.
  469. * CSS ID is assigned at cgroup allocation (create) automatically
  470. * and removed when subsys calls free_css_id() function. This is because
  471. * the lifetime of cgroup_subsys_state is subsys's matter.
  472. *
  473. * Looking up and scanning function should be called under rcu_read_lock().
  474. * Taking cgroup_mutex is not necessary for following calls.
  475. * But the css returned by this routine can be "not populated yet" or "being
  476. * destroyed". The caller should check css and cgroup's status.
  477. */
  478. /*
  479. * Typically Called at ->destroy(), or somewhere the subsys frees
  480. * cgroup_subsys_state.
  481. */
  482. void free_css_id(struct cgroup_subsys *ss, struct cgroup_subsys_state *css);
  483. /* Find a cgroup_subsys_state which has given ID */
  484. struct cgroup_subsys_state *css_lookup(struct cgroup_subsys *ss, int id);
  485. /*
  486. * Get a cgroup whose id is greater than or equal to id under tree of root.
  487. * Returning a cgroup_subsys_state or NULL.
  488. */
  489. struct cgroup_subsys_state *css_get_next(struct cgroup_subsys *ss, int id,
  490. struct cgroup_subsys_state *root, int *foundid);
  491. /* Returns true if root is ancestor of cg */
  492. bool css_is_ancestor(struct cgroup_subsys_state *cg,
  493. const struct cgroup_subsys_state *root);
  494. /* Get id and depth of css */
  495. unsigned short css_id(struct cgroup_subsys_state *css);
  496. unsigned short css_depth(struct cgroup_subsys_state *css);
  497. struct cgroup_subsys_state *cgroup_css_from_dir(struct file *f, int id);
  498. #else /* !CONFIG_CGROUPS */
  499. static inline int cgroup_init_early(void) { return 0; }
  500. static inline int cgroup_init(void) { return 0; }
  501. static inline void cgroup_fork(struct task_struct *p) {}
  502. static inline void cgroup_fork_callbacks(struct task_struct *p) {}
  503. static inline void cgroup_post_fork(struct task_struct *p) {}
  504. static inline void cgroup_exit(struct task_struct *p, int callbacks) {}
  505. static inline void cgroup_lock(void) {}
  506. static inline void cgroup_unlock(void) {}
  507. static inline int cgroupstats_build(struct cgroupstats *stats,
  508. struct dentry *dentry)
  509. {
  510. return -EINVAL;
  511. }
  512. /* No cgroups - nothing to do */
  513. static inline int cgroup_attach_task_all(struct task_struct *from,
  514. struct task_struct *t)
  515. {
  516. return 0;
  517. }
  518. #endif /* !CONFIG_CGROUPS */
  519. #endif /* _LINUX_CGROUP_H */