CPUInfo.m

   1 //
   2 //  CPUInfo.m
   3 //  CPU Usage
   4 //
   5 //  Created by Peter Hosey on 2006-06-21.
   6 //  Copyright 2006 Peter Hosey. All rights reserved.
   7 //
   8 //  Modified by Christopher Bowns, starting 2007-1-1.
   9
  10 #ifndef NSLOG_DEBUG
  11 #define NSLOG_DEBUG
  12 #endif
  13
  14 #import "CPUInfo.h"
  15
  16 #include <sys/types.h>
  17 //sqrtf, ceilf
  18 #include <math.h>
  19 //sysctl and its parameters
  20 #include <sys/sysctl.h>
  21 //errno, strerror
  22 #include <sys/errno.h>
  23 #include <string.h>
  24
  25 @implementation CPUInfo
  26
  27 /*
  28         Accomplishes:
  29                 inits cpudata, inptr, outptr
  30                 retrieves lastProcessorInfo for refresh method to go to town.
  31
  32         if cannot get processor info, returns nil.
  33 */
  34 - (CPUInfo *) initWithCapacity:(unsigned)numItems
  35 {
  36
  37         /*
  38                 from CPU usage
  39         */
  40         //We could get the number of processors the same way that we get the CPU usage info, but that allocates memory.
  41 /*      enum { miblen = 2U };
  42         int mib[miblen] = { CTL_HW, HW_NCPU };
  43         size_t sizeOfNumCPUs = sizeof(numCPUs);
  44         int status = sysctl(mib, miblen,
  45                    &numCPUs, &sizeOfNumCPUs,
  46 */                 /*newp*/ // NULL, /*newlen*/ 0U);
  47 //      if(status != 0) {
  48                 numCPUs = 1; // TODO we're going to assume one CPU for the moment.
  49 //              NSLog(@"%s error status, assuming one CPU", _cmd);
  50 //      }
  51
  52
  53
  54
  55
  56
  57         self = [super init];
  58
  59         size = numItems;
  60         cpudata = calloc(numItems, sizeof(CPUData));
  61         if (cpudata == NULL) {
  62                 NSLog (@"%s Failed to allocate buffer for CPUInfo", _cmd);
  63                 return (nil);
  64         }
  65         // Set up our data structure ptrs
  66         inptr = 0;
  67         outptr = -1;
  68
  69         // set the lastProcessorInfo array so we can get to work when we call refresh:
  70         natural_t numProcessors_nobodyCares = 0U;
  71         kern_return_t err = host_processor_info(mach_host_self(), PROCESSOR_CPU_LOAD_INFO, (natural_t *)&numProcessors_nobodyCares, (processor_info_array_t *)&lastProcessorInfo, (mach_msg_type_number_t *)&numLastProcessorInfo);
  72         if(err != KERN_SUCCESS) {
  73                 NSLog(@"%s failed to get cpu statistics", _cmd);
  74
  75                 // if we can't get this info, then we're toast. Exit gracefully.
  76                 return (nil);
  77         }
  78
  79         return (self);
  80 }
  81
  82
  83 /*
  84         Assumptions:
  85                 init has run.
  86                 lastProcessorInfo and numLastProcessorInfo contain valid data from prior refresh of CPU info
  87         Accomplishes:
  88                 retrieves current data
  89                 calculates avg cpu load over last period
  90                 updates cpudata array with info
  91                 deallocates old numLastProcInfo and lastProcessorInfo and set ptrs to new data
  92 */
  93 - (void)refresh
  94 {
  95         processor_info_array_t processorInfo;
  96         natural_t numProcessors_nobodyCares = 0U;
  97         mach_msg_type_number_t numProcessorInfo;
  98
  99         kern_return_t err = host_processor_info(mach_host_self(), PROCESSOR_CPU_LOAD_INFO, (natural_t *)&numProcessors_nobodyCares, (processor_info_array_t *)&processorInfo, (mach_msg_type_number_t *)&numProcessorInfo);
 100         if(err != KERN_SUCCESS) {
 101                 NSLog(@"%s failed to get cpu statistics", _cmd);
 102         }
 103
 104         /*
 105                 TODO make this multicore. First, we're gonna need a multicore machine to test it on.
 106         */
 107         // for(unsigned i = 0U; i < numCPUs; ++i)
 108
 109         unsigned int inUse, total, user, sys, nice, idle;
 110
 111         user = processorInfo[CPU_STATE_USER] - lastProcessorInfo[CPU_STATE_USER];
 112         sys = processorInfo[CPU_STATE_SYSTEM] - lastProcessorInfo[CPU_STATE_SYSTEM];
 113         nice = processorInfo[CPU_STATE_NICE] - lastProcessorInfo[CPU_STATE_NICE];
 114         idle = processorInfo[CPU_STATE_IDLE] - lastProcessorInfo[CPU_STATE_IDLE];
 115
 116         inUse = user + sys + nice;
 117         total = inUse + idle;
 118
 119         cpudata[inptr].user = (double)user / (double)total;
 120         cpudata[inptr].sys = (double)sys / (double)total;
 121         cpudata[inptr].nice = (double)nice / (double)total;
 122         cpudata[inptr].idle = (double)idle / (double)total;
 123
 124
 125         // deallocate the last one, since we don't want memory leaks.
 126         /*
 127                 TODO refresh: test in Shark / malloc debug: this DOES leak if we take this dealloc out, right?
 128         */
 129         if(lastProcessorInfo) {
 130                 size_t lastProcessorInfoSize = sizeof(integer_t) * numLastProcessorInfo;
 131                 vm_deallocate(mach_task_self(), (vm_address_t)lastProcessorInfo, lastProcessorInfoSize);
 132         }
 133
 134         lastProcessorInfo = processorInfo;
 135         numLastProcessorInfo = numProcessorInfo;
 136
 137         if (++inptr >= size) // advance our data ptr
 138                 inptr = 0;
 139 }
 140
 141
 142 - (void)startIterate
 143 {
 144         outptr = inptr;
 145 }
 146
 147
 148 - (BOOL)getNext:(CPUDataPtr)ptr
 149 {
 150         if (outptr == -1)
 151                 return (FALSE);
 152         *ptr = cpudata[outptr++];
 153         if (outptr >= size)
 154                 outptr = 0;
 155         if (outptr == inptr)
 156                 outptr = -1;
 157         return (TRUE);
 158 }
 159
 160
 161 - (void)getCurrent:(CPUDataPtr)ptr
 162 {
 163         *ptr = cpudata[inptr ? inptr - 1 : size - 1];
 164 }
 165
 166
 167 - (void)getLast:(CPUDataPtr)ptr
 168 {
 169         *ptr = cpudata[inptr > 1 ? inptr - 2 : size + inptr - 2];
 170 }
 171
 172
 173 - (int)getSize
 174 {
 175         return (size);
 176 }
 177
 178 @end