2015-12-17 20:56:48 +03:00
/*
* Copyright ( c ) 2002 - 2004 Roland McGrath < roland @ redhat . com >
2016-01-07 03:31:33 +03:00
* Copyright ( c ) 2009 - 2016 Dmitry V . Levin < ldv @ altlinux . org >
2015-12-17 20:56:48 +03:00
* All rights reserved .
*
* Redistribution and use in source and binary forms , with or without
* modification , are permitted provided that the following conditions
* are met :
* 1. Redistributions of source code must retain the above copyright
* notice , this list of conditions and the following disclaimer .
* 2. Redistributions in binary form must reproduce the above copyright
* notice , this list of conditions and the following disclaimer in the
* documentation and / or other materials provided with the distribution .
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission .
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ` ` AS IS ' ' AND ANY EXPRESS OR
* IMPLIED WARRANTIES , INCLUDING , BUT NOT LIMITED TO , THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED .
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT , INDIRECT ,
* INCIDENTAL , SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT
* NOT LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
* DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
* ( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
*/
2014-12-11 22:25:02 +03:00
# include "defs.h"
2016-01-07 03:31:33 +03:00
# include <sched.h>
static unsigned int
get_cpuset_size ( void )
{
static unsigned int cpuset_size ;
if ( ! cpuset_size ) {
2016-12-05 05:31:52 +03:00
/*
* If the cpuset size passed to sched_getaffinity is less
* than necessary to store the bitmask , the kernel does not
* look at the mask pointer and fails with EINVAL .
*
* If the cpuset size is large enough , the kernel fails with
* EFAULT on inaccessible mask pointers .
*
* This undocumented kernel feature can be used to probe
* the kernel and find out the minimal valid cpuset size
* without allocating any memory for the CPU affinity mask .
*/
2016-01-07 03:31:33 +03:00
cpuset_size = 128 ;
while ( cpuset_size & &
2016-12-05 05:34:56 +03:00
sched_getaffinity ( 0 , cpuset_size , NULL ) = = - 1 & &
2016-01-07 03:31:33 +03:00
EINVAL = = errno ) {
cpuset_size < < = 1 ;
}
if ( ! cpuset_size )
cpuset_size = 128 ;
}
return cpuset_size ;
}
2014-12-11 22:25:02 +03:00
static void
2015-07-20 22:22:10 +03:00
print_affinitylist ( struct tcb * tcp , const unsigned long addr , const unsigned int len )
2014-12-11 22:25:02 +03:00
{
2016-01-07 03:31:33 +03:00
const unsigned int max_size = get_cpuset_size ( ) ;
const unsigned int umove_size = len < max_size ? len : max_size ;
const unsigned int size =
( umove_size + current_wordsize - 1 ) & - current_wordsize ;
const unsigned int ncpu = size * 8 ;
void * cpu ;
2014-12-11 22:25:02 +03:00
2015-07-20 22:22:10 +03:00
if ( ! verbose ( tcp ) | | ( exiting ( tcp ) & & syserror ( tcp ) ) | |
2016-01-07 03:31:33 +03:00
! addr | | ! len | | ! ( cpu = calloc ( size , 1 ) ) ) {
2015-07-20 22:22:10 +03:00
printaddr ( addr ) ;
return ;
2014-12-11 22:25:02 +03:00
}
2015-07-20 22:22:10 +03:00
2016-01-07 03:31:33 +03:00
if ( ! umoven_or_printaddr ( tcp , addr , umove_size , cpu ) ) {
int i = 0 ;
const char * sep = " " ;
2015-07-20 22:22:10 +03:00
2016-01-07 03:31:33 +03:00
tprints ( " [ " ) ;
for ( ; ; i + + ) {
i = next_set_bit ( cpu , i , ncpu ) ;
if ( i < 0 )
break ;
tprintf ( " %s%d " , sep , i ) ;
2016-11-30 13:42:13 +03:00
sep = " , " ;
2015-07-20 22:22:10 +03:00
}
2016-01-07 03:31:33 +03:00
if ( size < len )
tprintf ( " %s... " , sep ) ;
tprints ( " ] " ) ;
2014-12-11 22:25:02 +03:00
}
2016-01-07 03:31:33 +03:00
free ( cpu ) ;
2014-12-11 22:25:02 +03:00
}
2015-04-07 04:36:50 +03:00
SYS_FUNC ( sched_setaffinity )
2014-12-11 22:25:02 +03:00
{
2016-01-07 03:31:33 +03:00
const int pid = tcp - > u_arg [ 0 ] ;
const unsigned int len = tcp - > u_arg [ 1 ] ;
tprintf ( " %d, %u, " , pid , len ) ;
print_affinitylist ( tcp , tcp - > u_arg [ 2 ] , len ) ;
2015-07-20 22:24:43 +03:00
return RVAL_DECODED ;
2014-12-11 22:25:02 +03:00
}
2015-04-07 04:36:50 +03:00
SYS_FUNC ( sched_getaffinity )
2014-12-11 22:25:02 +03:00
{
2016-01-07 03:31:33 +03:00
const int pid = tcp - > u_arg [ 0 ] ;
const unsigned int len = tcp - > u_arg [ 1 ] ;
2014-12-11 22:25:02 +03:00
if ( entering ( tcp ) ) {
2016-01-07 03:31:33 +03:00
tprintf ( " %d, %u, " , pid , len ) ;
2014-12-11 22:25:02 +03:00
} else {
2015-07-20 22:22:10 +03:00
print_affinitylist ( tcp , tcp - > u_arg [ 2 ] , tcp - > u_rval ) ;
2014-12-11 22:25:02 +03:00
}
return 0 ;
}