[linux-block.git] / include / linux / misc_cgroup.h

/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Miscellaneous cgroup controller.
 *
 * Copyright 2020 Google LLC
 * Author: Vipin Sharma <vipinsh@google.com>
 */
#ifndef _MISC_CGROUP_H_
#define _MISC_CGROUP_H_

/**
 * Types of misc cgroup entries supported by the host.
 */
enum misc_res_type {
#ifdef CONFIG_KVM_AMD_SEV
	/* AMD SEV ASIDs resource */
	MISC_CG_RES_SEV,
	/* AMD SEV-ES ASIDs resource */
	MISC_CG_RES_SEV_ES,
#endif
	MISC_CG_RES_TYPES
};

struct misc_cg;

#ifdef CONFIG_CGROUP_MISC

#include <linux/cgroup.h>

/**
 * struct misc_res: Per cgroup per misc type resource
 * @max: Maximum limit on the resource.
 * @usage: Current usage of the resource.
 * @failed: True if charged failed for the resource in a cgroup.
 */
struct misc_res {
	unsigned long max;
	atomic_long_t usage;
	atomic_long_t events;
};

/**
 * struct misc_cg - Miscellaneous controller's cgroup structure.
 * @css: cgroup subsys state object.
 * @res: Array of misc resources usage in the cgroup.
 */
struct misc_cg {
	struct cgroup_subsys_state css;

	/* misc.events */
	struct cgroup_file events_file;

	struct misc_res res[MISC_CG_RES_TYPES];
};

unsigned long misc_cg_res_total_usage(enum misc_res_type type);
int misc_cg_set_capacity(enum misc_res_type type, unsigned long capacity);
int misc_cg_try_charge(enum misc_res_type type, struct misc_cg *cg,
		       unsigned long amount);
void misc_cg_uncharge(enum misc_res_type type, struct misc_cg *cg,
		      unsigned long amount);

/**
 * css_misc() - Get misc cgroup from the css.
 * @css: cgroup subsys state object.
 *
 * Context: Any context.
 * Return:
 * * %NULL - If @css is null.
 * * struct misc_cg* - misc cgroup pointer of the passed css.
 */
static inline struct misc_cg *css_misc(struct cgroup_subsys_state *css)
{
	return css ? container_of(css, struct misc_cg, css) : NULL;
}

/*
 * get_current_misc_cg() - Find and get the misc cgroup of the current task.
 *
 * Returned cgroup has its ref count increased by 1. Caller must call
 * put_misc_cg() to return the reference.
 *
 * Return: Misc cgroup to which the current task belongs to.
 */
static inline struct misc_cg *get_current_misc_cg(void)
{
	return css_misc(task_get_css(current, misc_cgrp_id));
}

/*
 * put_misc_cg() - Put the misc cgroup and reduce its ref count.
 * @cg - cgroup to put.
 */
static inline void put_misc_cg(struct misc_cg *cg)
{
	if (cg)
		css_put(&cg->css);
}

#else /* !CONFIG_CGROUP_MISC */

static inline unsigned long misc_cg_res_total_usage(enum misc_res_type type)
{
	return 0;
}

static inline int misc_cg_set_capacity(enum misc_res_type type,
				       unsigned long capacity)
{
	return 0;
}

static inline int misc_cg_try_charge(enum misc_res_type type,
				     struct misc_cg *cg,
				     unsigned long amount)
{
	return 0;
}

static inline void misc_cg_uncharge(enum misc_res_type type,
				    struct misc_cg *cg,
				    unsigned long amount)
{
}

static inline struct misc_cg *get_current_misc_cg(void)
{
	return NULL;
}

static inline void put_misc_cg(struct misc_cg *cg)
{
}

#endif /* CONFIG_CGROUP_MISC */
#endif /* _MISC_CGROUP_H_ */
Commit	Line	Data
a72232ea VS	1	/* SPDX-License-Identifier: GPL-2.0 */
	2	/*
	3	* Miscellaneous cgroup controller.
	4	*
	5	* Copyright 2020 Google LLC
	6	* Author: Vipin Sharma <vipinsh@google.com>
	7	*/
	8	#ifndef _MISC_CGROUP_H_
	9	#define _MISC_CGROUP_H_
	10
	11	/**
	12	* Types of misc cgroup entries supported by the host.
	13	*/
	14	enum misc_res_type {
7aef27f0 VS	15	#ifdef CONFIG_KVM_AMD_SEV
	16	/* AMD SEV ASIDs resource */
	17	MISC_CG_RES_SEV,
	18	/* AMD SEV-ES ASIDs resource */
	19	MISC_CG_RES_SEV_ES,
	20	#endif
a72232ea VS	21	MISC_CG_RES_TYPES
	22	};
	23
	24	struct misc_cg;
	25
	26	#ifdef CONFIG_CGROUP_MISC
	27
	28	#include <linux/cgroup.h>
	29
	30	/**
	31	* struct misc_res: Per cgroup per misc type resource
	32	* @max: Maximum limit on the resource.
	33	* @usage: Current usage of the resource.
	34	* @failed: True if charged failed for the resource in a cgroup.
	35	*/
	36	struct misc_res {
	37	unsigned long max;
	38	atomic_long_t usage;
f279294b	39	atomic_long_t events;
a72232ea VS	40	};
	41
	42	/**
	43	* struct misc_cg - Miscellaneous controller's cgroup structure.
	44	* @css: cgroup subsys state object.
	45	* @res: Array of misc resources usage in the cgroup.
	46	*/
	47	struct misc_cg {
	48	struct cgroup_subsys_state css;
f279294b CX	49
	50	/* misc.events */
	51	struct cgroup_file events_file;
	52
a72232ea VS	53	struct misc_res res[MISC_CG_RES_TYPES];
	54	};
	55
	56	unsigned long misc_cg_res_total_usage(enum misc_res_type type);
	57	int misc_cg_set_capacity(enum misc_res_type type, unsigned long capacity);
	58	int misc_cg_try_charge(enum misc_res_type type, struct misc_cg *cg,
	59	unsigned long amount);
	60	void misc_cg_uncharge(enum misc_res_type type, struct misc_cg *cg,
	61	unsigned long amount);
	62
	63	/**
	64	* css_misc() - Get misc cgroup from the css.
	65	* @css: cgroup subsys state object.
	66	*
	67	* Context: Any context.
	68	* Return:
	69	* * %NULL - If @css is null.
	70	* * struct misc_cg* - misc cgroup pointer of the passed css.
	71	*/
	72	static inline struct misc_cg css_misc(struct cgroup_subsys_state css)
	73	{
	74	return css ? container_of(css, struct misc_cg, css) : NULL;
	75	}
	76
	77	/*
	78	* get_current_misc_cg() - Find and get the misc cgroup of the current task.
	79	*
	80	* Returned cgroup has its ref count increased by 1. Caller must call
	81	* put_misc_cg() to return the reference.
	82	*
	83	* Return: Misc cgroup to which the current task belongs to.
	84	*/
	85	static inline struct misc_cg *get_current_misc_cg(void)
	86	{
	87	return css_misc(task_get_css(current, misc_cgrp_id));
	88	}
	89
	90	/*
	91	* put_misc_cg() - Put the misc cgroup and reduce its ref count.
	92	* @cg - cgroup to put.
	93	*/
	94	static inline void put_misc_cg(struct misc_cg *cg)
	95	{
	96	if (cg)
	97	css_put(&cg->css);
	98	}
	99
	100	#else /* !CONFIG_CGROUP_MISC */
	101
dd3f4e49	102	static inline unsigned long misc_cg_res_total_usage(enum misc_res_type type)
a72232ea VS	103	{
	104	return 0;
	105	}
	106
	107	static inline int misc_cg_set_capacity(enum misc_res_type type,
	108	unsigned long capacity)
	109	{
	110	return 0;
	111	}
	112
	113	static inline int misc_cg_try_charge(enum misc_res_type type,
	114	struct misc_cg *cg,
	115	unsigned long amount)
	116	{
	117	return 0;
	118	}
	119
	120	static inline void misc_cg_uncharge(enum misc_res_type type,
	121	struct misc_cg *cg,
	122	unsigned long amount)
	123	{
	124	}
	125
	126	static inline struct misc_cg *get_current_misc_cg(void)
	127	{
	128	return NULL;
	129	}
	130
	131	static inline void put_misc_cg(struct misc_cg *cg)
	132	{
	133	}
	134
	135	#endif /* CONFIG_CGROUP_MISC */
	136	#endif /* _MISC_CGROUP_H_ */