osfmk/kperf/filter.c

   1 /*
   2  * Copyright (c) 2011 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28
  29 /*  Toy filtering. Allow system-wide or filtering on 4 PIDs */
  30
  31 #include <mach/mach_types.h>
  32 #include <sys/types.h> /* NULL */
  33 // #include <libkern/libkern.h>
  34
  35 #include <kperf/context.h>
  36 #include <kperf/filter.h>
  37
  38 // Filter params... dodge for now
  39 #define NPIDS (4)
  40 int pid_list[NPIDS];
  41
  42 // function to determine whether we should take a sample
  43 int
  44 kperf_filter_should_sample(struct kperf_context *context)
  45 {
  46         int i, restricted = 0;
  47
  48         /* see if the pids are restricted */
  49         for( i = 0; i < NPIDS; i++ )
  50         {
  51                 if( context->cur_pid == pid_list[i] )
  52                         return 1;
  53
  54                 if( pid_list[i] != -1 )
  55                         restricted = 1;
  56         }
  57
  58         /* wasn't in the pid list, but something was */
  59         if( restricted )
  60                 return 0;
  61
  62         /* not fitered, sample it */
  63         return 1;
  64 }
  65
  66 /* check whether pid filtering is enabled */
  67 int
  68 kperf_filter_on_pid(void)
  69 {
  70         int i;
  71
  72         for( i = 0; i < NPIDS; i++ )
  73                 if( pid_list[i] != -1 )
  74                         return 1;
  75
  76         return 0;
  77 }
  78
  79 /* create a list of pids to filter */
  80 void
  81 kperf_filter_pid_list( int *outcount, int **outv )
  82 {
  83         int i, found = 0;
  84
  85         for( i = 0; i < NPIDS; i++ )
  86                 if( pid_list[i] != -1 )
  87                         found = 1;
  88
  89         if( !found )
  90         {
  91                 *outcount = 0;
  92                 *outv = NULL;
  93                 return;
  94         }
  95
  96         /* just return our list */
  97         *outcount = NPIDS;
  98         *outv = pid_list;
  99 }
 100
 101 /* free a list we created*/
 102 void
 103 kperf_filter_free_pid_list( int *incount, int **inv )
 104 {
 105         // no op
 106         (void) incount;
 107         (void) inv;
 108 }
 109
 110 /* init the filters to nothing */
 111 void
 112 kperf_filter_init(void)
 113 {
 114         int i;
 115         for( i = 0; i < NPIDS; i++ )
 116                 pid_list[i] = -1;
 117 }