2009-12-03 20:59:42 +03:00
/*
* Common Block IO controller cgroup interface
*
* Based on ideas and code from CFQ , CFS and BFQ :
* Copyright ( C ) 2003 Jens Axboe < axboe @ kernel . dk >
*
* Copyright ( C ) 2008 Fabio Checconi < fabio @ gandalf . sssup . it >
* Paolo Valente < paolo . valente @ unimore . it >
*
* Copyright ( C ) 2009 Vivek Goyal < vgoyal @ redhat . com >
* Nauman Rafique < nauman @ google . com >
*/
# include <linux/ioprio.h>
2009-12-03 20:59:49 +03:00
# include <linux/kdev_t.h>
2009-12-04 18:36:41 +03:00
# include <linux/module.h>
2009-12-07 11:29:39 +03:00
# include <linux/err.h>
2010-04-02 02:01:41 +04:00
# include <linux/blkdev.h>
include cleanup: Update gfp.h and slab.h includes to prepare for breaking implicit slab.h inclusion from percpu.h
percpu.h is included by sched.h and module.h and thus ends up being
included when building most .c files. percpu.h includes slab.h which
in turn includes gfp.h making everything defined by the two files
universally available and complicating inclusion dependencies.
percpu.h -> slab.h dependency is about to be removed. Prepare for
this change by updating users of gfp and slab facilities include those
headers directly instead of assuming availability. As this conversion
needs to touch large number of source files, the following script is
used as the basis of conversion.
http://userweb.kernel.org/~tj/misc/slabh-sweep.py
The script does the followings.
* Scan files for gfp and slab usages and update includes such that
only the necessary includes are there. ie. if only gfp is used,
gfp.h, if slab is used, slab.h.
* When the script inserts a new include, it looks at the include
blocks and try to put the new include such that its order conforms
to its surrounding. It's put in the include block which contains
core kernel includes, in the same order that the rest are ordered -
alphabetical, Christmas tree, rev-Xmas-tree or at the end if there
doesn't seem to be any matching order.
* If the script can't find a place to put a new include (mostly
because the file doesn't have fitting include block), it prints out
an error message indicating which .h file needs to be added to the
file.
The conversion was done in the following steps.
1. The initial automatic conversion of all .c files updated slightly
over 4000 files, deleting around 700 includes and adding ~480 gfp.h
and ~3000 slab.h inclusions. The script emitted errors for ~400
files.
2. Each error was manually checked. Some didn't need the inclusion,
some needed manual addition while adding it to implementation .h or
embedding .c file was more appropriate for others. This step added
inclusions to around 150 files.
3. The script was run again and the output was compared to the edits
from #2 to make sure no file was left behind.
4. Several build tests were done and a couple of problems were fixed.
e.g. lib/decompress_*.c used malloc/free() wrappers around slab
APIs requiring slab.h to be added manually.
5. The script was run on all .h files but without automatically
editing them as sprinkling gfp.h and slab.h inclusions around .h
files could easily lead to inclusion dependency hell. Most gfp.h
inclusion directives were ignored as stuff from gfp.h was usually
wildly available and often used in preprocessor macros. Each
slab.h inclusion directive was examined and added manually as
necessary.
6. percpu.h was updated not to include slab.h.
7. Build test were done on the following configurations and failures
were fixed. CONFIG_GCOV_KERNEL was turned off for all tests (as my
distributed build env didn't work with gcov compiles) and a few
more options had to be turned off depending on archs to make things
build (like ipr on powerpc/64 which failed due to missing writeq).
* x86 and x86_64 UP and SMP allmodconfig and a custom test config.
* powerpc and powerpc64 SMP allmodconfig
* sparc and sparc64 SMP allmodconfig
* ia64 SMP allmodconfig
* s390 SMP allmodconfig
* alpha SMP allmodconfig
* um on x86_64 SMP allmodconfig
8. percpu.h modifications were reverted so that it could be applied as
a separate patch and serve as bisection point.
Given the fact that I had only a couple of failures from tests on step
6, I'm fairly confident about the coverage of this conversion patch.
If there is a breakage, it's likely to be something in one of the arch
headers which should be easily discoverable easily on most builds of
the specific arch.
Signed-off-by: Tejun Heo <tj@kernel.org>
Guess-its-ok-by: Christoph Lameter <cl@linux-foundation.org>
Cc: Ingo Molnar <mingo@redhat.com>
Cc: Lee Schermerhorn <Lee.Schermerhorn@hp.com>
2010-03-24 11:04:11 +03:00
# include <linux/slab.h>
2010-04-13 12:05:49 +04:00
# include <linux/genhd.h>
2012-03-06 01:15:00 +04:00
# include <linux/delay.h>
2012-03-20 02:10:56 +04:00
# include <linux/atomic.h>
2012-03-06 01:15:00 +04:00
# include "blk-cgroup.h"
2012-03-06 01:15:12 +04:00
# include "blk.h"
2009-12-04 18:36:42 +03:00
2010-04-09 10:31:19 +04:00
# define MAX_KEY_LEN 100
2012-04-14 00:11:26 +04:00
static DEFINE_MUTEX ( blkcg_pol_mutex ) ;
2012-03-06 01:15:13 +04:00
static DEFINE_MUTEX ( all_q_mutex ) ;
static LIST_HEAD ( all_q_list ) ;
2012-04-02 01:38:44 +04:00
struct blkio_cgroup blkio_root_cgroup = { . cfq_weight = 2 * CFQ_WEIGHT_DEFAULT } ;
2009-12-04 18:36:41 +03:00
EXPORT_SYMBOL_GPL ( blkio_root_cgroup ) ;
2012-03-06 01:15:04 +04:00
static struct blkio_policy_type * blkio_policy [ BLKIO_NR_POLICIES ] ;
2009-12-03 20:59:42 +03:00
struct blkio_cgroup * cgroup_to_blkio_cgroup ( struct cgroup * cgroup )
{
return container_of ( cgroup_subsys_state ( cgroup , blkio_subsys_id ) ,
struct blkio_cgroup , css ) ;
}
2009-12-04 18:36:41 +03:00
EXPORT_SYMBOL_GPL ( cgroup_to_blkio_cgroup ) ;
2009-12-03 20:59:42 +03:00
2012-03-06 01:15:28 +04:00
static struct blkio_cgroup * task_blkio_cgroup ( struct task_struct * tsk )
2011-05-16 17:24:08 +04:00
{
return container_of ( task_subsys_state ( tsk , blkio_subsys_id ) ,
struct blkio_cgroup , css ) ;
}
2012-03-06 01:15:28 +04:00
struct blkio_cgroup * bio_blkio_cgroup ( struct bio * bio )
{
if ( bio & & bio - > bi_css )
return container_of ( bio - > bi_css , struct blkio_cgroup , css ) ;
return task_blkio_cgroup ( current ) ;
}
EXPORT_SYMBOL_GPL ( bio_blkio_cgroup ) ;
2011-05-16 17:24:08 +04:00
2012-03-06 01:15:14 +04:00
/**
* blkg_free - free a blkg
* @ blkg : blkg to free
*
* Free @ blkg which may be partially allocated .
*/
static void blkg_free ( struct blkio_group * blkg )
{
2012-03-06 01:15:20 +04:00
int i ;
2012-03-06 01:15:16 +04:00
if ( ! blkg )
return ;
2012-03-06 01:15:20 +04:00
for ( i = 0 ; i < BLKIO_NR_POLICIES ; i + + ) {
2012-04-02 01:38:44 +04:00
struct blkio_policy_type * pol = blkio_policy [ i ] ;
2012-03-06 01:15:20 +04:00
struct blkg_policy_data * pd = blkg - > pd [ i ] ;
2012-04-02 01:38:44 +04:00
if ( ! pd )
continue ;
if ( pol & & pol - > ops . blkio_exit_group_fn )
pol - > ops . blkio_exit_group_fn ( blkg ) ;
kfree ( pd ) ;
2012-03-06 01:15:14 +04:00
}
2012-03-06 01:15:20 +04:00
2012-03-06 01:15:16 +04:00
kfree ( blkg ) ;
2012-03-06 01:15:14 +04:00
}
/**
* blkg_alloc - allocate a blkg
* @ blkcg : block cgroup the new blkg is associated with
* @ q : request_queue the new blkg is associated with
*
2012-03-06 01:15:20 +04:00
* Allocate a new blkg assocating @ blkcg and @ q .
2012-03-06 01:15:14 +04:00
*/
static struct blkio_group * blkg_alloc ( struct blkio_cgroup * blkcg ,
2012-03-06 01:15:20 +04:00
struct request_queue * q )
2012-03-06 01:15:14 +04:00
{
struct blkio_group * blkg ;
2012-03-06 01:15:20 +04:00
int i ;
2012-03-06 01:15:14 +04:00
/* alloc and init base part */
blkg = kzalloc_node ( sizeof ( * blkg ) , GFP_ATOMIC , q - > node ) ;
if ( ! blkg )
return NULL ;
2012-03-06 01:15:22 +04:00
blkg - > q = q ;
2012-03-06 01:15:20 +04:00
INIT_LIST_HEAD ( & blkg - > q_node ) ;
2012-03-06 01:15:14 +04:00
blkg - > blkcg = blkcg ;
2012-03-06 01:15:15 +04:00
blkg - > refcnt = 1 ;
2012-03-06 01:15:14 +04:00
cgroup_path ( blkcg - > css . cgroup , blkg - > path , sizeof ( blkg - > path ) ) ;
2012-03-06 01:15:20 +04:00
for ( i = 0 ; i < BLKIO_NR_POLICIES ; i + + ) {
struct blkio_policy_type * pol = blkio_policy [ i ] ;
struct blkg_policy_data * pd ;
2012-03-06 01:15:14 +04:00
2012-03-06 01:15:20 +04:00
if ( ! pol )
continue ;
/* alloc per-policy data and attach it to blkg */
pd = kzalloc_node ( sizeof ( * pd ) + pol - > pdata_size , GFP_ATOMIC ,
q - > node ) ;
if ( ! pd ) {
blkg_free ( blkg ) ;
return NULL ;
}
2012-03-06 01:15:16 +04:00
2012-03-06 01:15:20 +04:00
blkg - > pd [ i ] = pd ;
pd - > blkg = blkg ;
2012-03-06 01:15:14 +04:00
}
2012-03-06 01:15:16 +04:00
/* invoke per-policy init */
2012-03-06 01:15:20 +04:00
for ( i = 0 ; i < BLKIO_NR_POLICIES ; i + + ) {
struct blkio_policy_type * pol = blkio_policy [ i ] ;
if ( pol )
pol - > ops . blkio_init_group_fn ( blkg ) ;
}
2012-03-06 01:15:14 +04:00
return blkg ;
}
2012-03-06 01:15:06 +04:00
struct blkio_group * blkg_lookup_create ( struct blkio_cgroup * blkcg ,
struct request_queue * q ,
bool for_root )
__releases ( q - > queue_lock ) __acquires ( q - > queue_lock )
2011-05-19 23:38:28 +04:00
{
2012-03-08 22:53:56 +04:00
struct blkio_group * blkg ;
2011-05-19 23:38:28 +04:00
2012-03-06 01:15:06 +04:00
WARN_ON_ONCE ( ! rcu_read_lock_held ( ) ) ;
lockdep_assert_held ( q - > queue_lock ) ;
/*
* This could be the first entry point of blkcg implementation and
* we shouldn ' t allow anything to go through for a bypassing queue .
* The following can be removed if blkg lookup is guaranteed to
* fail on a bypassing queue .
*/
if ( unlikely ( blk_queue_bypass ( q ) ) & & ! for_root )
return ERR_PTR ( blk_queue_dead ( q ) ? - EINVAL : - EBUSY ) ;
2012-03-06 01:15:20 +04:00
blkg = blkg_lookup ( blkcg , q ) ;
2012-03-06 01:15:06 +04:00
if ( blkg )
return blkg ;
2012-03-06 01:15:11 +04:00
/* blkg holds a reference to blkcg */
2012-03-06 01:15:06 +04:00
if ( ! css_tryget ( & blkcg - > css ) )
return ERR_PTR ( - EINVAL ) ;
/*
* Allocate and initialize .
*/
2012-03-08 22:53:56 +04:00
blkg = blkg_alloc ( blkcg , q ) ;
2012-03-06 01:15:06 +04:00
/* did alloc fail? */
2012-03-08 22:53:56 +04:00
if ( unlikely ( ! blkg ) ) {
2012-03-06 01:15:06 +04:00
blkg = ERR_PTR ( - ENOMEM ) ;
goto out ;
}
/* insert */
spin_lock ( & blkcg - > lock ) ;
2009-12-03 20:59:42 +03:00
hlist_add_head_rcu ( & blkg - > blkcg_node , & blkcg - > blkg_list ) ;
2012-03-06 01:15:20 +04:00
list_add ( & blkg - > q_node , & q - > blkg_list ) ;
2012-03-06 01:15:06 +04:00
spin_unlock ( & blkcg - > lock ) ;
out :
return blkg ;
2009-12-03 20:59:42 +03:00
}
2012-03-06 01:15:06 +04:00
EXPORT_SYMBOL_GPL ( blkg_lookup_create ) ;
2009-12-03 20:59:42 +03:00
/* called under rcu_read_lock(). */
2012-03-06 01:15:06 +04:00
struct blkio_group * blkg_lookup ( struct blkio_cgroup * blkcg ,
2012-03-06 01:15:20 +04:00
struct request_queue * q )
2009-12-03 20:59:42 +03:00
{
struct blkio_group * blkg ;
struct hlist_node * n ;
2012-03-06 01:15:03 +04:00
hlist_for_each_entry_rcu ( blkg , n , & blkcg - > blkg_list , blkcg_node )
2012-03-06 01:15:20 +04:00
if ( blkg - > q = = q )
2009-12-03 20:59:42 +03:00
return blkg ;
return NULL ;
}
2012-03-06 01:15:06 +04:00
EXPORT_SYMBOL_GPL ( blkg_lookup ) ;
2009-12-03 20:59:42 +03:00
2012-03-06 01:15:20 +04:00
static void blkg_destroy ( struct blkio_group * blkg )
2012-03-06 01:15:19 +04:00
{
struct request_queue * q = blkg - > q ;
2012-03-06 01:15:21 +04:00
struct blkio_cgroup * blkcg = blkg - > blkcg ;
2012-03-06 01:15:19 +04:00
lockdep_assert_held ( q - > queue_lock ) ;
2012-03-06 01:15:21 +04:00
lockdep_assert_held ( & blkcg - > lock ) ;
2012-03-06 01:15:19 +04:00
/* Something wrong if we are trying to remove same group twice */
2012-03-06 01:15:20 +04:00
WARN_ON_ONCE ( list_empty ( & blkg - > q_node ) ) ;
2012-03-06 01:15:21 +04:00
WARN_ON_ONCE ( hlist_unhashed ( & blkg - > blkcg_node ) ) ;
2012-03-06 01:15:20 +04:00
list_del_init ( & blkg - > q_node ) ;
2012-03-06 01:15:21 +04:00
hlist_del_init_rcu ( & blkg - > blkcg_node ) ;
2012-03-06 01:15:19 +04:00
/*
* Put the reference taken at the time of creation so that when all
* queues are gone , group can be destroyed .
*/
blkg_put ( blkg ) ;
}
2012-03-06 01:15:20 +04:00
/*
* XXX : This updates blkg policy data in - place for root blkg , which is
* necessary across elevator switch and policy registration as root blkgs
* aren ' t shot down . This broken and racy implementation is temporary .
* Eventually , blkg shoot down will be replaced by proper in - place update .
*/
2012-04-14 00:11:27 +04:00
void update_root_blkg_pd ( struct request_queue * q ,
const struct blkio_policy_type * pol )
2012-03-06 01:15:20 +04:00
{
struct blkio_group * blkg = blkg_lookup ( & blkio_root_cgroup , q ) ;
struct blkg_policy_data * pd ;
if ( ! blkg )
return ;
2012-04-14 00:11:27 +04:00
kfree ( blkg - > pd [ pol - > plid ] ) ;
blkg - > pd [ pol - > plid ] = NULL ;
2012-03-06 01:15:20 +04:00
if ( ! pol )
return ;
pd = kzalloc ( sizeof ( * pd ) + pol - > pdata_size , GFP_KERNEL ) ;
WARN_ON_ONCE ( ! pd ) ;
2012-04-14 00:11:27 +04:00
blkg - > pd [ pol - > plid ] = pd ;
2012-03-06 01:15:20 +04:00
pd - > blkg = blkg ;
pol - > ops . blkio_init_group_fn ( blkg ) ;
}
EXPORT_SYMBOL_GPL ( update_root_blkg_pd ) ;
2012-03-06 01:15:21 +04:00
/**
* blkg_destroy_all - destroy all blkgs associated with a request_queue
* @ q : request_queue of interest
* @ destroy_root : whether to destroy root blkg or not
*
* Destroy blkgs associated with @ q . If @ destroy_root is % true , all are
* destroyed ; otherwise , root blkg is left alone .
*/
2012-03-06 01:15:20 +04:00
void blkg_destroy_all ( struct request_queue * q , bool destroy_root )
2012-03-06 01:15:00 +04:00
{
2012-03-06 01:15:19 +04:00
struct blkio_group * blkg , * n ;
2012-03-06 01:15:00 +04:00
2012-03-06 01:15:21 +04:00
spin_lock_irq ( q - > queue_lock ) ;
2012-03-06 01:15:00 +04:00
2012-03-06 01:15:21 +04:00
list_for_each_entry_safe ( blkg , n , & q - > blkg_list , q_node ) {
struct blkio_cgroup * blkcg = blkg - > blkcg ;
2012-03-06 01:15:00 +04:00
2012-03-06 01:15:21 +04:00
/* skip root? */
if ( ! destroy_root & & blkg - > blkcg = = & blkio_root_cgroup )
continue ;
2012-03-06 01:15:00 +04:00
2012-03-06 01:15:21 +04:00
spin_lock ( & blkcg - > lock ) ;
blkg_destroy ( blkg ) ;
spin_unlock ( & blkcg - > lock ) ;
2012-03-06 01:15:00 +04:00
}
2012-03-06 01:15:21 +04:00
spin_unlock_irq ( q - > queue_lock ) ;
2012-03-06 01:15:00 +04:00
}
2012-03-06 01:15:19 +04:00
EXPORT_SYMBOL_GPL ( blkg_destroy_all ) ;
2012-03-06 01:15:00 +04:00
2012-03-06 01:15:15 +04:00
static void blkg_rcu_free ( struct rcu_head * rcu_head )
{
blkg_free ( container_of ( rcu_head , struct blkio_group , rcu_head ) ) ;
}
void __blkg_release ( struct blkio_group * blkg )
{
/* release the extra blkcg reference this blkg has been holding */
css_put ( & blkg - > blkcg - > css ) ;
/*
* A group is freed in rcu manner . But having an rcu lock does not
* mean that one can access all the fields of blkg and assume these
* are valid . For example , don ' t try to follow throtl_data and
* request queue links .
*
* Having a reference to blkg under an rcu allows acess to only
* values local to groups like group stats and group rate limits
*/
call_rcu ( & blkg - > rcu_head , blkg_rcu_free ) ;
}
EXPORT_SYMBOL_GPL ( __blkg_release ) ;
2010-04-02 02:01:24 +04:00
static int
2010-04-09 10:31:19 +04:00
blkiocg_reset_stats ( struct cgroup * cgroup , struct cftype * cftype , u64 val )
2010-04-02 02:01:24 +04:00
{
2012-03-08 22:53:58 +04:00
struct blkio_cgroup * blkcg = cgroup_to_blkio_cgroup ( cgroup ) ;
2010-04-02 02:01:24 +04:00
struct blkio_group * blkg ;
struct hlist_node * n ;
2012-04-14 00:11:26 +04:00
int i ;
2010-04-02 02:01:24 +04:00
2012-04-14 00:11:26 +04:00
mutex_lock ( & blkcg_pol_mutex ) ;
2010-04-02 02:01:24 +04:00
spin_lock_irq ( & blkcg - > lock ) ;
2012-03-08 22:53:58 +04:00
/*
* Note that stat reset is racy - it doesn ' t synchronize against
* stat updates . This is a debug feature which shouldn ' t exist
* anyway . If you get hit by a race , retry .
*/
2010-04-02 02:01:24 +04:00
hlist_for_each_entry ( blkg , n , & blkcg - > blkg_list , blkcg_node ) {
2012-04-14 00:11:26 +04:00
for ( i = 0 ; i < BLKIO_NR_POLICIES ; i + + ) {
struct blkio_policy_type * pol = blkio_policy [ i ] ;
2012-03-06 01:15:16 +04:00
2012-04-14 00:11:26 +04:00
if ( pol & & pol - > ops . blkio_reset_group_stats_fn )
2012-04-02 01:38:44 +04:00
pol - > ops . blkio_reset_group_stats_fn ( blkg ) ;
2012-04-14 00:11:26 +04:00
}
2010-04-02 02:01:24 +04:00
}
2011-05-19 23:38:30 +04:00
2010-04-02 02:01:24 +04:00
spin_unlock_irq ( & blkcg - > lock ) ;
2012-04-14 00:11:26 +04:00
mutex_unlock ( & blkcg_pol_mutex ) ;
2010-04-02 02:01:24 +04:00
return 0 ;
}
2012-04-02 01:38:42 +04:00
static const char * blkg_dev_name ( struct blkio_group * blkg )
2010-04-02 02:01:24 +04:00
{
2012-04-02 01:38:42 +04:00
/* some drivers (floppy) instantiate a queue w/o disk registered */
if ( blkg - > q - > backing_dev_info . dev )
return dev_name ( blkg - > q - > backing_dev_info . dev ) ;
return NULL ;
2010-04-02 02:01:24 +04:00
}
2012-04-02 01:38:42 +04:00
/**
* blkcg_print_blkgs - helper for printing per - blkg data
* @ sf : seq_file to print to
* @ blkcg : blkcg of interest
* @ prfill : fill function to print out a blkg
* @ pol : policy in question
* @ data : data to be passed to @ prfill
* @ show_total : to print out sum of prfill return values or not
*
* This function invokes @ prfill on each blkg of @ blkcg if pd for the
* policy specified by @ pol exists . @ prfill is invoked with @ sf , the
* policy data and @ data . If @ show_total is % true , the sum of the return
* values from @ prfill is printed with " Total " label at the end .
*
* This is to be used to construct print functions for
* cftype - > read_seq_string method .
*/
2012-04-02 01:38:43 +04:00
void blkcg_print_blkgs ( struct seq_file * sf , struct blkio_cgroup * blkcg ,
2012-04-02 01:38:44 +04:00
u64 ( * prfill ) ( struct seq_file * , void * , int ) ,
2012-04-14 00:11:27 +04:00
const struct blkio_policy_type * pol , int data ,
bool show_total )
2011-05-19 23:38:28 +04:00
{
2012-04-02 01:38:42 +04:00
struct blkio_group * blkg ;
struct hlist_node * n ;
u64 total = 0 ;
2011-05-19 23:38:28 +04:00
2012-04-02 01:38:42 +04:00
spin_lock_irq ( & blkcg - > lock ) ;
hlist_for_each_entry ( blkg , n , & blkcg - > blkg_list , blkcg_node )
2012-04-14 00:11:27 +04:00
if ( blkg - > pd [ pol - > plid ] )
total + = prfill ( sf , blkg - > pd [ pol - > plid ] - > pdata , data ) ;
2012-04-02 01:38:42 +04:00
spin_unlock_irq ( & blkcg - > lock ) ;
if ( show_total )
seq_printf ( sf , " Total %llu \n " , ( unsigned long long ) total ) ;
}
2012-04-02 01:38:43 +04:00
EXPORT_SYMBOL_GPL ( blkcg_print_blkgs ) ;
2012-04-02 01:38:42 +04:00
/**
* __blkg_prfill_u64 - prfill helper for a single u64 value
* @ sf : seq_file to print to
2012-04-02 01:38:44 +04:00
* @ pdata : policy private data of interest
2012-04-02 01:38:42 +04:00
* @ v : value to print
*
2012-04-02 01:38:44 +04:00
* Print @ v to @ sf for the device assocaited with @ pdata .
2012-04-02 01:38:42 +04:00
*/
2012-04-02 01:38:44 +04:00
u64 __blkg_prfill_u64 ( struct seq_file * sf , void * pdata , u64 v )
2012-04-02 01:38:42 +04:00
{
2012-04-02 01:38:44 +04:00
const char * dname = blkg_dev_name ( pdata_to_blkg ( pdata ) ) ;
2012-04-02 01:38:42 +04:00
if ( ! dname )
return 0 ;
seq_printf ( sf , " %s %llu \n " , dname , ( unsigned long long ) v ) ;
return v ;
}
2012-04-02 01:38:43 +04:00
EXPORT_SYMBOL_GPL ( __blkg_prfill_u64 ) ;
2012-04-02 01:38:42 +04:00
/**
* __blkg_prfill_rwstat - prfill helper for a blkg_rwstat
* @ sf : seq_file to print to
2012-04-02 01:38:44 +04:00
* @ pdata : policy private data of interest
2012-04-02 01:38:42 +04:00
* @ rwstat : rwstat to print
*
2012-04-02 01:38:44 +04:00
* Print @ rwstat to @ sf for the device assocaited with @ pdata .
2012-04-02 01:38:42 +04:00
*/
2012-04-02 01:38:44 +04:00
u64 __blkg_prfill_rwstat ( struct seq_file * sf , void * pdata ,
2012-04-02 01:38:43 +04:00
const struct blkg_rwstat * rwstat )
2012-04-02 01:38:42 +04:00
{
static const char * rwstr [ ] = {
[ BLKG_RWSTAT_READ ] = " Read " ,
[ BLKG_RWSTAT_WRITE ] = " Write " ,
[ BLKG_RWSTAT_SYNC ] = " Sync " ,
[ BLKG_RWSTAT_ASYNC ] = " Async " ,
} ;
2012-04-02 01:38:44 +04:00
const char * dname = blkg_dev_name ( pdata_to_blkg ( pdata ) ) ;
2012-04-02 01:38:42 +04:00
u64 v ;
int i ;
if ( ! dname )
return 0 ;
for ( i = 0 ; i < BLKG_RWSTAT_NR ; i + + )
seq_printf ( sf , " %s %s %llu \n " , dname , rwstr [ i ] ,
( unsigned long long ) rwstat - > cnt [ i ] ) ;
v = rwstat - > cnt [ BLKG_RWSTAT_READ ] + rwstat - > cnt [ BLKG_RWSTAT_WRITE ] ;
seq_printf ( sf , " %s Total %llu \n " , dname , ( unsigned long long ) v ) ;
return v ;
}
2012-04-02 01:38:45 +04:00
/**
* blkg_prfill_stat - prfill callback for blkg_stat
* @ sf : seq_file to print to
* @ pdata : policy private data of interest
* @ off : offset to the blkg_stat in @ pdata
*
* prfill callback for printing a blkg_stat .
*/
u64 blkg_prfill_stat ( struct seq_file * sf , void * pdata , int off )
2012-04-02 01:38:42 +04:00
{
2012-04-02 01:38:44 +04:00
return __blkg_prfill_u64 ( sf , pdata , blkg_stat_read ( pdata + off ) ) ;
2012-04-02 01:38:42 +04:00
}
2012-04-02 01:38:45 +04:00
EXPORT_SYMBOL_GPL ( blkg_prfill_stat ) ;
2012-04-02 01:38:42 +04:00
2012-04-02 01:38:45 +04:00
/**
* blkg_prfill_rwstat - prfill callback for blkg_rwstat
* @ sf : seq_file to print to
* @ pdata : policy private data of interest
* @ off : offset to the blkg_rwstat in @ pdata
*
* prfill callback for printing a blkg_rwstat .
*/
u64 blkg_prfill_rwstat ( struct seq_file * sf , void * pdata , int off )
2012-04-02 01:38:42 +04:00
{
2012-04-02 01:38:44 +04:00
struct blkg_rwstat rwstat = blkg_rwstat_read ( pdata + off ) ;
2012-04-02 01:38:42 +04:00
2012-04-02 01:38:44 +04:00
return __blkg_prfill_rwstat ( sf , pdata , & rwstat ) ;
2012-04-02 01:38:42 +04:00
}
2012-04-02 01:38:45 +04:00
EXPORT_SYMBOL_GPL ( blkg_prfill_rwstat ) ;
2012-04-02 01:38:42 +04:00
2012-04-02 01:38:43 +04:00
/**
* blkg_conf_prep - parse and prepare for per - blkg config update
* @ blkcg : target block cgroup
* @ input : input string
* @ ctx : blkg_conf_ctx to be filled
*
* Parse per - blkg config update from @ input and initialize @ ctx with the
* result . @ ctx - > blkg points to the blkg to be updated and @ ctx - > v the new
* value . This function returns with RCU read locked and must be paired
* with blkg_conf_finish ( ) .
*/
2012-04-02 01:38:43 +04:00
int blkg_conf_prep ( struct blkio_cgroup * blkcg , const char * input ,
struct blkg_conf_ctx * ctx )
2012-04-02 01:38:43 +04:00
__acquires ( rcu )
2010-04-13 12:05:49 +04:00
{
2012-04-02 01:38:43 +04:00
struct gendisk * disk ;
struct blkio_group * blkg ;
2012-04-02 01:38:43 +04:00
unsigned int major , minor ;
unsigned long long v ;
int part , ret ;
2010-04-13 12:05:49 +04:00
2012-04-02 01:38:43 +04:00
if ( sscanf ( input , " %u:%u %llu " , & major , & minor , & v ) ! = 3 )
return - EINVAL ;
2012-04-02 01:38:43 +04:00
2012-04-02 01:38:43 +04:00
disk = get_gendisk ( MKDEV ( major , minor ) , & part ) ;
2012-03-06 01:15:08 +04:00
if ( ! disk | | part )
2012-04-02 01:38:43 +04:00
return - EINVAL ;
2012-03-06 01:15:07 +04:00
rcu_read_lock ( ) ;
2012-03-06 01:15:08 +04:00
spin_lock_irq ( disk - > queue - > queue_lock ) ;
2012-04-02 01:38:42 +04:00
blkg = blkg_lookup_create ( blkcg , disk - > queue , false ) ;
2012-03-06 01:15:08 +04:00
spin_unlock_irq ( disk - > queue - > queue_lock ) ;
2012-03-06 01:15:07 +04:00
2012-03-06 01:15:08 +04:00
if ( IS_ERR ( blkg ) ) {
ret = PTR_ERR ( blkg ) ;
2012-04-02 01:38:43 +04:00
rcu_read_unlock ( ) ;
put_disk ( disk ) ;
/*
* If queue was bypassing , we should retry . Do so after a
* short msleep ( ) . It isn ' t strictly necessary but queue
* can be bypassing for some time and it ' s always nice to
* avoid busy looping .
*/
if ( ret = = - EBUSY ) {
msleep ( 10 ) ;
ret = restart_syscall ( ) ;
2010-09-16 01:06:36 +04:00
}
2012-04-02 01:38:43 +04:00
return ret ;
2010-09-16 01:06:33 +04:00
}
2012-04-02 01:38:43 +04:00
ctx - > disk = disk ;
ctx - > blkg = blkg ;
2012-04-02 01:38:43 +04:00
ctx - > v = v ;
return 0 ;
2010-04-13 12:05:49 +04:00
}
2012-04-02 01:38:43 +04:00
EXPORT_SYMBOL_GPL ( blkg_conf_prep ) ;
2010-04-13 12:05:49 +04:00
2012-04-02 01:38:43 +04:00
/**
* blkg_conf_finish - finish up per - blkg config update
* @ ctx : blkg_conf_ctx intiailized by blkg_conf_prep ( )
*
* Finish up after per - blkg config update . This function must be paired
* with blkg_conf_prep ( ) .
*/
2012-04-02 01:38:43 +04:00
void blkg_conf_finish ( struct blkg_conf_ctx * ctx )
2012-04-02 01:38:43 +04:00
__releases ( rcu )
2010-04-13 12:05:49 +04:00
{
2012-04-02 01:38:43 +04:00
rcu_read_unlock ( ) ;
put_disk ( ctx - > disk ) ;
2010-04-13 12:05:49 +04:00
}
2012-04-02 01:38:43 +04:00
EXPORT_SYMBOL_GPL ( blkg_conf_finish ) ;
2010-04-13 12:05:49 +04:00
2009-12-03 20:59:42 +03:00
struct cftype blkio_files [ ] = {
2010-04-09 10:31:19 +04:00
{
. name = " reset_stats " ,
. write_u64 = blkiocg_reset_stats ,
2009-12-03 20:59:49 +03:00
} ,
2012-04-01 23:09:55 +04:00
{ } /* terminate */
2009-12-03 20:59:42 +03:00
} ;
2012-03-06 01:15:21 +04:00
/**
* blkiocg_pre_destroy - cgroup pre_destroy callback
* @ cgroup : cgroup of interest
*
* This function is called when @ cgroup is about to go away and responsible
* for shooting down all blkgs associated with @ cgroup . blkgs should be
* removed while holding both q and blkcg locks . As blkcg lock is nested
* inside q lock , this function performs reverse double lock dancing .
*
* This is the blkcg counterpart of ioc_release_fn ( ) .
*/
2012-04-01 23:30:01 +04:00
static int blkiocg_pre_destroy ( struct cgroup * cgroup )
2009-12-03 20:59:42 +03:00
{
struct blkio_cgroup * blkcg = cgroup_to_blkio_cgroup ( cgroup ) ;
2009-12-03 20:59:47 +03:00
2012-03-06 01:15:21 +04:00
spin_lock_irq ( & blkcg - > lock ) ;
2012-03-06 01:15:11 +04:00
2012-03-06 01:15:21 +04:00
while ( ! hlist_empty ( & blkcg - > blkg_list ) ) {
struct blkio_group * blkg = hlist_entry ( blkcg - > blkg_list . first ,
struct blkio_group , blkcg_node ) ;
2012-03-06 01:15:22 +04:00
struct request_queue * q = blkg - > q ;
2009-12-03 20:59:47 +03:00
2012-03-06 01:15:21 +04:00
if ( spin_trylock ( q - > queue_lock ) ) {
blkg_destroy ( blkg ) ;
spin_unlock ( q - > queue_lock ) ;
} else {
spin_unlock_irq ( & blkcg - > lock ) ;
cpu_relax ( ) ;
2012-03-29 22:57:08 +04:00
spin_lock_irq ( & blkcg - > lock ) ;
2010-05-03 16:28:55 +04:00
}
2012-03-06 01:15:21 +04:00
}
2009-12-03 20:59:47 +03:00
2012-03-06 01:15:21 +04:00
spin_unlock_irq ( & blkcg - > lock ) ;
2012-03-06 01:15:11 +04:00
return 0 ;
}
2012-04-01 23:30:01 +04:00
static void blkiocg_destroy ( struct cgroup * cgroup )
2012-03-06 01:15:11 +04:00
{
struct blkio_cgroup * blkcg = cgroup_to_blkio_cgroup ( cgroup ) ;
2010-03-11 02:22:11 +03:00
if ( blkcg ! = & blkio_root_cgroup )
kfree ( blkcg ) ;
2009-12-03 20:59:42 +03:00
}
2012-01-31 09:47:36 +04:00
static struct cgroup_subsys_state * blkiocg_create ( struct cgroup * cgroup )
2009-12-03 20:59:42 +03:00
{
2012-03-20 02:10:56 +04:00
static atomic64_t id_seq = ATOMIC64_INIT ( 0 ) ;
2010-05-07 10:57:00 +04:00
struct blkio_cgroup * blkcg ;
struct cgroup * parent = cgroup - > parent ;
2009-12-03 20:59:42 +03:00
2010-05-07 10:57:00 +04:00
if ( ! parent ) {
2009-12-03 20:59:42 +03:00
blkcg = & blkio_root_cgroup ;
goto done ;
}
blkcg = kzalloc ( sizeof ( * blkcg ) , GFP_KERNEL ) ;
if ( ! blkcg )
return ERR_PTR ( - ENOMEM ) ;
2012-04-02 01:38:44 +04:00
blkcg - > cfq_weight = CFQ_WEIGHT_DEFAULT ;
2012-03-20 02:10:56 +04:00
blkcg - > id = atomic64_inc_return ( & id_seq ) ; /* root is 0, start from 1 */
2009-12-03 20:59:42 +03:00
done :
spin_lock_init ( & blkcg - > lock ) ;
INIT_HLIST_HEAD ( & blkcg - > blkg_list ) ;
return & blkcg - > css ;
}
2012-03-06 01:15:12 +04:00
/**
* blkcg_init_queue - initialize blkcg part of request queue
* @ q : request_queue to initialize
*
* Called from blk_alloc_queue_node ( ) . Responsible for initializing blkcg
* part of new request_queue @ q .
*
* RETURNS :
* 0 on success , - errno on failure .
*/
int blkcg_init_queue ( struct request_queue * q )
{
2012-03-06 01:15:13 +04:00
int ret ;
2012-03-06 01:15:12 +04:00
might_sleep ( ) ;
2012-03-06 01:15:13 +04:00
ret = blk_throtl_init ( q ) ;
if ( ret )
return ret ;
mutex_lock ( & all_q_mutex ) ;
INIT_LIST_HEAD ( & q - > all_q_node ) ;
list_add_tail ( & q - > all_q_node , & all_q_list ) ;
mutex_unlock ( & all_q_mutex ) ;
return 0 ;
2012-03-06 01:15:12 +04:00
}
/**
* blkcg_drain_queue - drain blkcg part of request_queue
* @ q : request_queue to drain
*
* Called from blk_drain_queue ( ) . Responsible for draining blkcg part .
*/
void blkcg_drain_queue ( struct request_queue * q )
{
lockdep_assert_held ( q - > queue_lock ) ;
blk_throtl_drain ( q ) ;
}
/**
* blkcg_exit_queue - exit and release blkcg part of request_queue
* @ q : request_queue being released
*
* Called from blk_release_queue ( ) . Responsible for exiting blkcg part .
*/
void blkcg_exit_queue ( struct request_queue * q )
{
2012-03-06 01:15:13 +04:00
mutex_lock ( & all_q_mutex ) ;
list_del_init ( & q - > all_q_node ) ;
mutex_unlock ( & all_q_mutex ) ;
2012-03-06 01:15:20 +04:00
blkg_destroy_all ( q , true ) ;
2012-03-06 01:15:12 +04:00
blk_throtl_exit ( q ) ;
}
2009-12-03 20:59:42 +03:00
/*
* We cannot support shared io contexts , as we have no mean to support
* two tasks with the same ioc in two different groups without major rework
* of the main cic data structures . For now we allow a task to change
* its cgroup only if it ' s the only owner of its ioc .
*/
2012-01-31 09:47:36 +04:00
static int blkiocg_can_attach ( struct cgroup * cgrp , struct cgroup_taskset * tset )
2009-12-03 20:59:42 +03:00
{
2011-12-13 06:12:21 +04:00
struct task_struct * task ;
2009-12-03 20:59:42 +03:00
struct io_context * ioc ;
int ret = 0 ;
/* task_lock() is needed to avoid races with exit_io_context() */
2011-12-13 06:12:21 +04:00
cgroup_taskset_for_each ( task , cgrp , tset ) {
task_lock ( task ) ;
ioc = task - > io_context ;
if ( ioc & & atomic_read ( & ioc - > nr_tasks ) > 1 )
ret = - EINVAL ;
task_unlock ( task ) ;
if ( ret )
break ;
}
2009-12-03 20:59:42 +03:00
return ret ;
}
2012-03-06 01:15:13 +04:00
static void blkcg_bypass_start ( void )
__acquires ( & all_q_mutex )
{
struct request_queue * q ;
mutex_lock ( & all_q_mutex ) ;
list_for_each_entry ( q , & all_q_list , all_q_node ) {
blk_queue_bypass_start ( q ) ;
2012-03-06 01:15:20 +04:00
blkg_destroy_all ( q , false ) ;
2012-03-06 01:15:13 +04:00
}
}
static void blkcg_bypass_end ( void )
__releases ( & all_q_mutex )
{
struct request_queue * q ;
list_for_each_entry ( q , & all_q_list , all_q_node )
blk_queue_bypass_end ( q ) ;
mutex_unlock ( & all_q_mutex ) ;
}
2012-04-01 23:09:55 +04:00
struct cgroup_subsys blkio_subsys = {
. name = " blkio " ,
. create = blkiocg_create ,
. can_attach = blkiocg_can_attach ,
2012-04-01 23:30:01 +04:00
. pre_destroy = blkiocg_pre_destroy ,
2012-04-01 23:09:55 +04:00
. destroy = blkiocg_destroy ,
. subsys_id = blkio_subsys_id ,
2012-04-01 23:09:55 +04:00
. base_cftypes = blkio_files ,
2012-04-01 23:09:55 +04:00
. module = THIS_MODULE ,
} ;
EXPORT_SYMBOL_GPL ( blkio_subsys ) ;
2009-12-04 18:36:42 +03:00
void blkio_policy_register ( struct blkio_policy_type * blkiop )
{
2012-03-06 01:15:20 +04:00
struct request_queue * q ;
2012-04-14 00:11:26 +04:00
mutex_lock ( & blkcg_pol_mutex ) ;
2012-03-06 01:15:13 +04:00
blkcg_bypass_start ( ) ;
2012-03-06 01:15:04 +04:00
BUG_ON ( blkio_policy [ blkiop - > plid ] ) ;
blkio_policy [ blkiop - > plid ] = blkiop ;
2012-03-06 01:15:20 +04:00
list_for_each_entry ( q , & all_q_list , all_q_node )
2012-04-14 00:11:27 +04:00
update_root_blkg_pd ( q , blkiop ) ;
2012-04-14 00:11:26 +04:00
2012-03-06 01:15:13 +04:00
blkcg_bypass_end ( ) ;
2012-04-02 01:38:43 +04:00
if ( blkiop - > cftypes )
WARN_ON ( cgroup_add_cftypes ( & blkio_subsys , blkiop - > cftypes ) ) ;
2012-04-14 00:11:26 +04:00
mutex_unlock ( & blkcg_pol_mutex ) ;
2009-12-04 18:36:42 +03:00
}
EXPORT_SYMBOL_GPL ( blkio_policy_register ) ;
void blkio_policy_unregister ( struct blkio_policy_type * blkiop )
{
2012-03-06 01:15:20 +04:00
struct request_queue * q ;
2012-04-14 00:11:26 +04:00
mutex_lock ( & blkcg_pol_mutex ) ;
2012-04-02 01:38:43 +04:00
if ( blkiop - > cftypes )
cgroup_rm_cftypes ( & blkio_subsys , blkiop - > cftypes ) ;
2012-03-06 01:15:13 +04:00
blkcg_bypass_start ( ) ;
2012-03-06 01:15:04 +04:00
BUG_ON ( blkio_policy [ blkiop - > plid ] ! = blkiop ) ;
blkio_policy [ blkiop - > plid ] = NULL ;
2012-03-06 01:15:20 +04:00
list_for_each_entry ( q , & all_q_list , all_q_node )
2012-04-14 00:11:27 +04:00
update_root_blkg_pd ( q , blkiop ) ;
2012-03-06 01:15:13 +04:00
blkcg_bypass_end ( ) ;
2012-04-14 00:11:26 +04:00
mutex_unlock ( & blkcg_pol_mutex ) ;
2009-12-04 18:36:42 +03:00
}
EXPORT_SYMBOL_GPL ( blkio_policy_unregister ) ;