1*c54f35caSApple OSS Distributions /*
2*c54f35caSApple OSS Distributions * Copyright (c) 2012-2013 Apple Computer, Inc. All Rights Reserved.
3*c54f35caSApple OSS Distributions *
4*c54f35caSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5*c54f35caSApple OSS Distributions *
6*c54f35caSApple OSS Distributions * This file contains Original Code and/or Modifications of Original Code
7*c54f35caSApple OSS Distributions * as defined in and that are subject to the Apple Public Source License
8*c54f35caSApple OSS Distributions * Version 2.0 (the 'License'). You may not use this file except in
9*c54f35caSApple OSS Distributions * compliance with the License. The rights granted to you under the License
10*c54f35caSApple OSS Distributions * may not be used to create, or enable the creation or redistribution of,
11*c54f35caSApple OSS Distributions * unlawful or unlicensed copies of an Apple operating system, or to
12*c54f35caSApple OSS Distributions * circumvent, violate, or enable the circumvention or violation of, any
13*c54f35caSApple OSS Distributions * terms of an Apple operating system software license agreement.
14*c54f35caSApple OSS Distributions *
15*c54f35caSApple OSS Distributions * Please obtain a copy of the License at
16*c54f35caSApple OSS Distributions * http://www.opensource.apple.com/apsl/ and read it before using this file.
17*c54f35caSApple OSS Distributions *
18*c54f35caSApple OSS Distributions * The Original Code and all software distributed under the License are
19*c54f35caSApple OSS Distributions * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20*c54f35caSApple OSS Distributions * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21*c54f35caSApple OSS Distributions * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22*c54f35caSApple OSS Distributions * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23*c54f35caSApple OSS Distributions * Please see the License for the specific language governing rights and
24*c54f35caSApple OSS Distributions * limitations under the License.
25*c54f35caSApple OSS Distributions *
26*c54f35caSApple OSS Distributions * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27*c54f35caSApple OSS Distributions */
28*c54f35caSApple OSS Distributions
29*c54f35caSApple OSS Distributions #define IOKIT_ENABLE_SHARED_PTR
30*c54f35caSApple OSS Distributions
31*c54f35caSApple OSS Distributions #include <libkern/c++/OSSharedPtr.h>
32*c54f35caSApple OSS Distributions #include <IOKit/IOKernelReportStructs.h>
33*c54f35caSApple OSS Distributions #include <IOKit/IOKernelReporters.h>
34*c54f35caSApple OSS Distributions #include "IOReporterDefs.h"
35*c54f35caSApple OSS Distributions
36*c54f35caSApple OSS Distributions
37*c54f35caSApple OSS Distributions #define super IOReporter
38*c54f35caSApple OSS Distributions OSDefineMetaClassAndStructors(IOStateReporter, IOReporter);
39*c54f35caSApple OSS Distributions
40*c54f35caSApple OSS Distributions
41*c54f35caSApple OSS Distributions /* static */
42*c54f35caSApple OSS Distributions OSSharedPtr<IOStateReporter>
with(IOService * reportingService,IOReportCategories categories,int nstates,IOReportUnit unit)43*c54f35caSApple OSS Distributions IOStateReporter::with(IOService *reportingService,
44*c54f35caSApple OSS Distributions IOReportCategories categories,
45*c54f35caSApple OSS Distributions int nstates,
46*c54f35caSApple OSS Distributions IOReportUnit unit /* = kIOReportUnitHWTicks*/)
47*c54f35caSApple OSS Distributions {
48*c54f35caSApple OSS Distributions OSSharedPtr<IOStateReporter> reporter;
49*c54f35caSApple OSS Distributions
50*c54f35caSApple OSS Distributions if (nstates > INT16_MAX) {
51*c54f35caSApple OSS Distributions return nullptr;
52*c54f35caSApple OSS Distributions }
53*c54f35caSApple OSS Distributions
54*c54f35caSApple OSS Distributions reporter = OSMakeShared<IOStateReporter>();
55*c54f35caSApple OSS Distributions if (!reporter) {
56*c54f35caSApple OSS Distributions return nullptr;
57*c54f35caSApple OSS Distributions }
58*c54f35caSApple OSS Distributions
59*c54f35caSApple OSS Distributions if (!reporter->initWith(reportingService, categories, (int16_t) nstates, unit)) {
60*c54f35caSApple OSS Distributions return nullptr;
61*c54f35caSApple OSS Distributions }
62*c54f35caSApple OSS Distributions
63*c54f35caSApple OSS Distributions return reporter;
64*c54f35caSApple OSS Distributions }
65*c54f35caSApple OSS Distributions
66*c54f35caSApple OSS Distributions bool
initWith(IOService * reportingService,IOReportCategories categories,int16_t nstates,IOReportUnit unit)67*c54f35caSApple OSS Distributions IOStateReporter::initWith(IOService *reportingService,
68*c54f35caSApple OSS Distributions IOReportCategories categories,
69*c54f35caSApple OSS Distributions int16_t nstates,
70*c54f35caSApple OSS Distributions IOReportUnit unit)
71*c54f35caSApple OSS Distributions {
72*c54f35caSApple OSS Distributions bool success = false;
73*c54f35caSApple OSS Distributions
74*c54f35caSApple OSS Distributions IOReportChannelType channelType = {
75*c54f35caSApple OSS Distributions .categories = categories,
76*c54f35caSApple OSS Distributions .report_format = kIOReportFormatState,
77*c54f35caSApple OSS Distributions .nelements = static_cast<uint16_t>(nstates),
78*c54f35caSApple OSS Distributions .element_idx = 0
79*c54f35caSApple OSS Distributions };
80*c54f35caSApple OSS Distributions
81*c54f35caSApple OSS Distributions if (super::init(reportingService, channelType, unit) != true) {
82*c54f35caSApple OSS Distributions IORLOG("ERROR super::initWith failed");
83*c54f35caSApple OSS Distributions success = false;
84*c54f35caSApple OSS Distributions goto finish;
85*c54f35caSApple OSS Distributions }
86*c54f35caSApple OSS Distributions
87*c54f35caSApple OSS Distributions _currentStates = NULL;
88*c54f35caSApple OSS Distributions _lastUpdateTimes = NULL;
89*c54f35caSApple OSS Distributions
90*c54f35caSApple OSS Distributions success = true;
91*c54f35caSApple OSS Distributions
92*c54f35caSApple OSS Distributions finish:
93*c54f35caSApple OSS Distributions return success;
94*c54f35caSApple OSS Distributions }
95*c54f35caSApple OSS Distributions
96*c54f35caSApple OSS Distributions
97*c54f35caSApple OSS Distributions void
free(void)98*c54f35caSApple OSS Distributions IOStateReporter::free(void)
99*c54f35caSApple OSS Distributions {
100*c54f35caSApple OSS Distributions if (_currentStates) {
101*c54f35caSApple OSS Distributions PREFL_MEMOP_PANIC(_nChannels, int);
102*c54f35caSApple OSS Distributions IOFreeData(_currentStates, (size_t)_nChannels * sizeof(int));
103*c54f35caSApple OSS Distributions }
104*c54f35caSApple OSS Distributions if (_lastUpdateTimes) {
105*c54f35caSApple OSS Distributions PREFL_MEMOP_PANIC(_nChannels, uint64_t);
106*c54f35caSApple OSS Distributions IOFreeData(_lastUpdateTimes, (size_t)_nChannels * sizeof(uint64_t));
107*c54f35caSApple OSS Distributions }
108*c54f35caSApple OSS Distributions
109*c54f35caSApple OSS Distributions super::free();
110*c54f35caSApple OSS Distributions }
111*c54f35caSApple OSS Distributions
112*c54f35caSApple OSS Distributions
113*c54f35caSApple OSS Distributions IOReturn
handleSwapPrepare(int newNChannels)114*c54f35caSApple OSS Distributions IOStateReporter::handleSwapPrepare(int newNChannels)
115*c54f35caSApple OSS Distributions {
116*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
117*c54f35caSApple OSS Distributions size_t newCurStatesSize, newTSSize;
118*c54f35caSApple OSS Distributions
119*c54f35caSApple OSS Distributions //IORLOG("handleSwapPrepare (state) _nChannels before = %u", _nChannels);
120*c54f35caSApple OSS Distributions
121*c54f35caSApple OSS Distributions IOREPORTER_CHECK_CONFIG_LOCK();
122*c54f35caSApple OSS Distributions
123*c54f35caSApple OSS Distributions if (_swapCurrentStates || _swapLastUpdateTimes) {
124*c54f35caSApple OSS Distributions panic("IOStateReporter::_swap* already in use");
125*c54f35caSApple OSS Distributions }
126*c54f35caSApple OSS Distributions
127*c54f35caSApple OSS Distributions // new currentStates buffer
128*c54f35caSApple OSS Distributions PREFL_MEMOP_FAIL(newNChannels, int);
129*c54f35caSApple OSS Distributions newCurStatesSize = (size_t)newNChannels * sizeof(int);
130*c54f35caSApple OSS Distributions _swapCurrentStates = (int*)IOMallocData(newCurStatesSize);
131*c54f35caSApple OSS Distributions if (_swapCurrentStates == NULL) {
132*c54f35caSApple OSS Distributions res = kIOReturnNoMemory; goto finish;
133*c54f35caSApple OSS Distributions }
134*c54f35caSApple OSS Distributions memset(_swapCurrentStates, -1, newCurStatesSize); // init w/"no state"
135*c54f35caSApple OSS Distributions
136*c54f35caSApple OSS Distributions // new timestamps buffer
137*c54f35caSApple OSS Distributions PREFL_MEMOP_FAIL(newNChannels, uint64_t);
138*c54f35caSApple OSS Distributions newTSSize = (size_t)newNChannels * sizeof(uint64_t);
139*c54f35caSApple OSS Distributions _swapLastUpdateTimes = (uint64_t *)IOMallocZeroData(newTSSize);
140*c54f35caSApple OSS Distributions if (_swapLastUpdateTimes == NULL) {
141*c54f35caSApple OSS Distributions res = kIOReturnNoMemory; goto finish;
142*c54f35caSApple OSS Distributions }
143*c54f35caSApple OSS Distributions
144*c54f35caSApple OSS Distributions res = super::handleSwapPrepare(newNChannels);
145*c54f35caSApple OSS Distributions
146*c54f35caSApple OSS Distributions finish:
147*c54f35caSApple OSS Distributions if (res) {
148*c54f35caSApple OSS Distributions if (_swapCurrentStates) {
149*c54f35caSApple OSS Distributions IOFreeData(_swapCurrentStates, newCurStatesSize);
150*c54f35caSApple OSS Distributions _swapCurrentStates = NULL;
151*c54f35caSApple OSS Distributions }
152*c54f35caSApple OSS Distributions if (_swapLastUpdateTimes) {
153*c54f35caSApple OSS Distributions IOFreeData(_swapLastUpdateTimes, newTSSize);
154*c54f35caSApple OSS Distributions _swapLastUpdateTimes = NULL;
155*c54f35caSApple OSS Distributions }
156*c54f35caSApple OSS Distributions }
157*c54f35caSApple OSS Distributions
158*c54f35caSApple OSS Distributions return res;
159*c54f35caSApple OSS Distributions }
160*c54f35caSApple OSS Distributions
161*c54f35caSApple OSS Distributions IOReturn
handleAddChannelSwap(uint64_t channelID,const OSSymbol * symChannelName)162*c54f35caSApple OSS Distributions IOStateReporter::handleAddChannelSwap(uint64_t channelID,
163*c54f35caSApple OSS Distributions const OSSymbol *symChannelName)
164*c54f35caSApple OSS Distributions {
165*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
166*c54f35caSApple OSS Distributions int cnt;
167*c54f35caSApple OSS Distributions int *tmpCurStates;
168*c54f35caSApple OSS Distributions uint64_t *tmpTimestamps;
169*c54f35caSApple OSS Distributions bool swapComplete = false;
170*c54f35caSApple OSS Distributions
171*c54f35caSApple OSS Distributions //IORLOG("IOStateReporter::handleSwap");
172*c54f35caSApple OSS Distributions
173*c54f35caSApple OSS Distributions if (!_swapCurrentStates || !_swapLastUpdateTimes) {
174*c54f35caSApple OSS Distributions IORLOG("IOReporter::handleSwap ERROR swap variables uninitialized!");
175*c54f35caSApple OSS Distributions goto finish;
176*c54f35caSApple OSS Distributions }
177*c54f35caSApple OSS Distributions
178*c54f35caSApple OSS Distributions IOREPORTER_CHECK_CONFIG_LOCK();
179*c54f35caSApple OSS Distributions IOREPORTER_CHECK_LOCK();
180*c54f35caSApple OSS Distributions
181*c54f35caSApple OSS Distributions // Copy any existing buffers
182*c54f35caSApple OSS Distributions if (_currentStates) {
183*c54f35caSApple OSS Distributions PREFL_MEMOP_FAIL(_nChannels, int);
184*c54f35caSApple OSS Distributions memcpy(_swapCurrentStates, _currentStates,
185*c54f35caSApple OSS Distributions (size_t)_nChannels * sizeof(int));
186*c54f35caSApple OSS Distributions
187*c54f35caSApple OSS Distributions if (!_lastUpdateTimes) {
188*c54f35caSApple OSS Distributions panic("IOStateReporter::handleAddChannelSwap _lastUpdateTimes unset despite non-NULL _currentStates");
189*c54f35caSApple OSS Distributions }
190*c54f35caSApple OSS Distributions PREFL_MEMOP_FAIL(_nChannels, uint64_t);
191*c54f35caSApple OSS Distributions memcpy(_swapLastUpdateTimes, _lastUpdateTimes,
192*c54f35caSApple OSS Distributions (size_t)_nChannels * sizeof(uint64_t));
193*c54f35caSApple OSS Distributions }
194*c54f35caSApple OSS Distributions
195*c54f35caSApple OSS Distributions // Update principal instance variables, keep old values in _swap* for cleanup
196*c54f35caSApple OSS Distributions tmpCurStates = _currentStates;
197*c54f35caSApple OSS Distributions _currentStates = _swapCurrentStates;
198*c54f35caSApple OSS Distributions _swapCurrentStates = tmpCurStates;
199*c54f35caSApple OSS Distributions
200*c54f35caSApple OSS Distributions tmpTimestamps = _lastUpdateTimes;
201*c54f35caSApple OSS Distributions _lastUpdateTimes = _swapLastUpdateTimes;
202*c54f35caSApple OSS Distributions _swapLastUpdateTimes = tmpTimestamps;
203*c54f35caSApple OSS Distributions
204*c54f35caSApple OSS Distributions swapComplete = true;
205*c54f35caSApple OSS Distributions
206*c54f35caSApple OSS Distributions // subclass success
207*c54f35caSApple OSS Distributions
208*c54f35caSApple OSS Distributions // invoke superclass(es): base class updates _nChannels & _nElements
209*c54f35caSApple OSS Distributions res = super::handleAddChannelSwap(channelID, symChannelName);
210*c54f35caSApple OSS Distributions if (res) {
211*c54f35caSApple OSS Distributions IORLOG("handleSwap(state) ERROR super::handleSwap failed!");
212*c54f35caSApple OSS Distributions goto finish;
213*c54f35caSApple OSS Distributions }
214*c54f35caSApple OSS Distributions
215*c54f35caSApple OSS Distributions // Channel added successfully, initialize the new channel's state_ids to 0..nStates-1
216*c54f35caSApple OSS Distributions for (cnt = 0; cnt < _channelDimension; cnt++) {
217*c54f35caSApple OSS Distributions handleSetStateID(channelID, cnt, (uint64_t)cnt);
218*c54f35caSApple OSS Distributions }
219*c54f35caSApple OSS Distributions
220*c54f35caSApple OSS Distributions finish:
221*c54f35caSApple OSS Distributions if (res && swapComplete) {
222*c54f35caSApple OSS Distributions // unswap so the unused buffers get cleaned up
223*c54f35caSApple OSS Distributions tmpCurStates = _currentStates;
224*c54f35caSApple OSS Distributions _currentStates = _swapCurrentStates;
225*c54f35caSApple OSS Distributions _swapCurrentStates = tmpCurStates;
226*c54f35caSApple OSS Distributions
227*c54f35caSApple OSS Distributions tmpTimestamps = _lastUpdateTimes;
228*c54f35caSApple OSS Distributions _lastUpdateTimes = _swapLastUpdateTimes;
229*c54f35caSApple OSS Distributions _swapLastUpdateTimes = tmpTimestamps;
230*c54f35caSApple OSS Distributions }
231*c54f35caSApple OSS Distributions
232*c54f35caSApple OSS Distributions return res;
233*c54f35caSApple OSS Distributions }
234*c54f35caSApple OSS Distributions
235*c54f35caSApple OSS Distributions
236*c54f35caSApple OSS Distributions void
handleSwapCleanup(int swapNChannels)237*c54f35caSApple OSS Distributions IOStateReporter::handleSwapCleanup(int swapNChannels)
238*c54f35caSApple OSS Distributions {
239*c54f35caSApple OSS Distributions IOREPORTER_CHECK_CONFIG_LOCK();
240*c54f35caSApple OSS Distributions
241*c54f35caSApple OSS Distributions super::handleSwapCleanup(swapNChannels);
242*c54f35caSApple OSS Distributions
243*c54f35caSApple OSS Distributions if (_swapCurrentStates) {
244*c54f35caSApple OSS Distributions PREFL_MEMOP_PANIC(swapNChannels, int);
245*c54f35caSApple OSS Distributions IOFreeData(_swapCurrentStates, (size_t)swapNChannels * sizeof(int));
246*c54f35caSApple OSS Distributions _swapCurrentStates = NULL;
247*c54f35caSApple OSS Distributions }
248*c54f35caSApple OSS Distributions if (_swapLastUpdateTimes) {
249*c54f35caSApple OSS Distributions PREFL_MEMOP_PANIC(swapNChannels, uint64_t);
250*c54f35caSApple OSS Distributions IOFreeData(_swapLastUpdateTimes, (size_t)swapNChannels * sizeof(uint64_t));
251*c54f35caSApple OSS Distributions _swapLastUpdateTimes = NULL;
252*c54f35caSApple OSS Distributions }
253*c54f35caSApple OSS Distributions }
254*c54f35caSApple OSS Distributions
255*c54f35caSApple OSS Distributions
256*c54f35caSApple OSS Distributions IOReturn
_getStateIndices(uint64_t channel_id,uint64_t state_id,int * channel_index,int * state_index)257*c54f35caSApple OSS Distributions IOStateReporter::_getStateIndices(uint64_t channel_id,
258*c54f35caSApple OSS Distributions uint64_t state_id,
259*c54f35caSApple OSS Distributions int *channel_index,
260*c54f35caSApple OSS Distributions int *state_index)
261*c54f35caSApple OSS Distributions {
262*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
263*c54f35caSApple OSS Distributions int cnt;
264*c54f35caSApple OSS Distributions IOStateReportValues *values;
265*c54f35caSApple OSS Distributions int element_index = 0;
266*c54f35caSApple OSS Distributions
267*c54f35caSApple OSS Distributions IOREPORTER_CHECK_LOCK();
268*c54f35caSApple OSS Distributions
269*c54f35caSApple OSS Distributions if (getChannelIndices(channel_id,
270*c54f35caSApple OSS Distributions channel_index,
271*c54f35caSApple OSS Distributions &element_index) != kIOReturnSuccess) {
272*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
273*c54f35caSApple OSS Distributions
274*c54f35caSApple OSS Distributions goto finish;
275*c54f35caSApple OSS Distributions }
276*c54f35caSApple OSS Distributions
277*c54f35caSApple OSS Distributions for (cnt = 0; cnt < _channelDimension; cnt++) {
278*c54f35caSApple OSS Distributions values = (IOStateReportValues *)getElementValues(element_index + cnt);
279*c54f35caSApple OSS Distributions
280*c54f35caSApple OSS Distributions if (values == NULL) {
281*c54f35caSApple OSS Distributions res = kIOReturnError;
282*c54f35caSApple OSS Distributions goto finish;
283*c54f35caSApple OSS Distributions }
284*c54f35caSApple OSS Distributions
285*c54f35caSApple OSS Distributions if (values->state_id == state_id) {
286*c54f35caSApple OSS Distributions *state_index = cnt;
287*c54f35caSApple OSS Distributions res = kIOReturnSuccess;
288*c54f35caSApple OSS Distributions goto finish;
289*c54f35caSApple OSS Distributions }
290*c54f35caSApple OSS Distributions }
291*c54f35caSApple OSS Distributions
292*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
293*c54f35caSApple OSS Distributions
294*c54f35caSApple OSS Distributions finish:
295*c54f35caSApple OSS Distributions return res;
296*c54f35caSApple OSS Distributions }
297*c54f35caSApple OSS Distributions
298*c54f35caSApple OSS Distributions
299*c54f35caSApple OSS Distributions IOReturn
setChannelState(uint64_t channel_id,uint64_t new_state_id)300*c54f35caSApple OSS Distributions IOStateReporter::setChannelState(uint64_t channel_id,
301*c54f35caSApple OSS Distributions uint64_t new_state_id)
302*c54f35caSApple OSS Distributions {
303*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
304*c54f35caSApple OSS Distributions int channel_index, new_state_index;
305*c54f35caSApple OSS Distributions uint64_t last_intransition = 0;
306*c54f35caSApple OSS Distributions uint64_t prev_state_residency = 0;
307*c54f35caSApple OSS Distributions
308*c54f35caSApple OSS Distributions lockReporter();
309*c54f35caSApple OSS Distributions
310*c54f35caSApple OSS Distributions if (_getStateIndices(channel_id, new_state_id, &channel_index, &new_state_index) == kIOReturnSuccess) {
311*c54f35caSApple OSS Distributions res = handleSetStateByIndices(channel_index, new_state_index,
312*c54f35caSApple OSS Distributions last_intransition,
313*c54f35caSApple OSS Distributions prev_state_residency);
314*c54f35caSApple OSS Distributions goto finish;
315*c54f35caSApple OSS Distributions }
316*c54f35caSApple OSS Distributions
317*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
318*c54f35caSApple OSS Distributions
319*c54f35caSApple OSS Distributions finish:
320*c54f35caSApple OSS Distributions unlockReporter();
321*c54f35caSApple OSS Distributions return res;
322*c54f35caSApple OSS Distributions }
323*c54f35caSApple OSS Distributions
324*c54f35caSApple OSS Distributions IOReturn
setChannelState(uint64_t channel_id,uint64_t new_state_id,uint64_t last_intransition,uint64_t prev_state_residency)325*c54f35caSApple OSS Distributions IOStateReporter::setChannelState(uint64_t channel_id,
326*c54f35caSApple OSS Distributions uint64_t new_state_id,
327*c54f35caSApple OSS Distributions uint64_t last_intransition,
328*c54f35caSApple OSS Distributions uint64_t prev_state_residency)
329*c54f35caSApple OSS Distributions {
330*c54f35caSApple OSS Distributions return setChannelState(channel_id, new_state_id);
331*c54f35caSApple OSS Distributions }
332*c54f35caSApple OSS Distributions
333*c54f35caSApple OSS Distributions IOReturn
overrideChannelState(uint64_t channel_id,uint64_t state_id,uint64_t time_in_state,uint64_t intransitions,uint64_t last_intransition)334*c54f35caSApple OSS Distributions IOStateReporter::overrideChannelState(uint64_t channel_id,
335*c54f35caSApple OSS Distributions uint64_t state_id,
336*c54f35caSApple OSS Distributions uint64_t time_in_state,
337*c54f35caSApple OSS Distributions uint64_t intransitions,
338*c54f35caSApple OSS Distributions uint64_t last_intransition /*=0*/)
339*c54f35caSApple OSS Distributions {
340*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
341*c54f35caSApple OSS Distributions int channel_index, state_index;
342*c54f35caSApple OSS Distributions
343*c54f35caSApple OSS Distributions lockReporter();
344*c54f35caSApple OSS Distributions
345*c54f35caSApple OSS Distributions if (_getStateIndices(channel_id, state_id, &channel_index, &state_index) == kIOReturnSuccess) {
346*c54f35caSApple OSS Distributions if (_lastUpdateTimes[channel_index]) {
347*c54f35caSApple OSS Distributions panic("overrideChannelState() cannot be used after setChannelState()!");
348*c54f35caSApple OSS Distributions }
349*c54f35caSApple OSS Distributions
350*c54f35caSApple OSS Distributions res = handleOverrideChannelStateByIndices(channel_index, state_index,
351*c54f35caSApple OSS Distributions time_in_state, intransitions,
352*c54f35caSApple OSS Distributions last_intransition);
353*c54f35caSApple OSS Distributions goto finish;
354*c54f35caSApple OSS Distributions }
355*c54f35caSApple OSS Distributions
356*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
357*c54f35caSApple OSS Distributions
358*c54f35caSApple OSS Distributions finish:
359*c54f35caSApple OSS Distributions unlockReporter();
360*c54f35caSApple OSS Distributions return res;
361*c54f35caSApple OSS Distributions }
362*c54f35caSApple OSS Distributions
363*c54f35caSApple OSS Distributions
364*c54f35caSApple OSS Distributions IOReturn
handleOverrideChannelStateByIndices(int channel_index,int state_index,uint64_t time_in_state,uint64_t intransitions,uint64_t last_intransition)365*c54f35caSApple OSS Distributions IOStateReporter::handleOverrideChannelStateByIndices(int channel_index,
366*c54f35caSApple OSS Distributions int state_index,
367*c54f35caSApple OSS Distributions uint64_t time_in_state,
368*c54f35caSApple OSS Distributions uint64_t intransitions,
369*c54f35caSApple OSS Distributions uint64_t last_intransition /*=0*/)
370*c54f35caSApple OSS Distributions {
371*c54f35caSApple OSS Distributions IOReturn kerr, result = kIOReturnError;
372*c54f35caSApple OSS Distributions IOStateReportValues state_values;
373*c54f35caSApple OSS Distributions int element_index;
374*c54f35caSApple OSS Distributions
375*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index >= _nChannels) {
376*c54f35caSApple OSS Distributions result = kIOReturnBadArgument; goto finish;
377*c54f35caSApple OSS Distributions }
378*c54f35caSApple OSS Distributions
379*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index > (_nElements - state_index)
380*c54f35caSApple OSS Distributions / _channelDimension) {
381*c54f35caSApple OSS Distributions result = kIOReturnOverrun; goto finish;
382*c54f35caSApple OSS Distributions }
383*c54f35caSApple OSS Distributions element_index = channel_index * _channelDimension + state_index;
384*c54f35caSApple OSS Distributions
385*c54f35caSApple OSS Distributions kerr = copyElementValues(element_index, (IOReportElementValues*)&state_values);
386*c54f35caSApple OSS Distributions if (kerr) {
387*c54f35caSApple OSS Distributions result = kerr; goto finish;
388*c54f35caSApple OSS Distributions }
389*c54f35caSApple OSS Distributions
390*c54f35caSApple OSS Distributions // last_intransition = 0 -> no current state ("residency summary only")
391*c54f35caSApple OSS Distributions state_values.last_intransition = last_intransition;
392*c54f35caSApple OSS Distributions state_values.intransitions = intransitions;
393*c54f35caSApple OSS Distributions state_values.upticks = time_in_state;
394*c54f35caSApple OSS Distributions
395*c54f35caSApple OSS Distributions // determines current time for metadata
396*c54f35caSApple OSS Distributions kerr = setElementValues(element_index, (IOReportElementValues *)&state_values);
397*c54f35caSApple OSS Distributions if (kerr) {
398*c54f35caSApple OSS Distributions result = kerr; goto finish;
399*c54f35caSApple OSS Distributions }
400*c54f35caSApple OSS Distributions
401*c54f35caSApple OSS Distributions // success
402*c54f35caSApple OSS Distributions result = kIOReturnSuccess;
403*c54f35caSApple OSS Distributions
404*c54f35caSApple OSS Distributions finish:
405*c54f35caSApple OSS Distributions return result;
406*c54f35caSApple OSS Distributions }
407*c54f35caSApple OSS Distributions
408*c54f35caSApple OSS Distributions
409*c54f35caSApple OSS Distributions IOReturn
incrementChannelState(uint64_t channel_id,uint64_t state_id,uint64_t time_in_state,uint64_t intransitions,uint64_t last_intransition)410*c54f35caSApple OSS Distributions IOStateReporter::incrementChannelState(uint64_t channel_id,
411*c54f35caSApple OSS Distributions uint64_t state_id,
412*c54f35caSApple OSS Distributions uint64_t time_in_state,
413*c54f35caSApple OSS Distributions uint64_t intransitions,
414*c54f35caSApple OSS Distributions uint64_t last_intransition /*=0*/)
415*c54f35caSApple OSS Distributions {
416*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
417*c54f35caSApple OSS Distributions int channel_index, state_index;
418*c54f35caSApple OSS Distributions
419*c54f35caSApple OSS Distributions lockReporter();
420*c54f35caSApple OSS Distributions
421*c54f35caSApple OSS Distributions if (_getStateIndices(channel_id, state_id, &channel_index, &state_index) == kIOReturnSuccess) {
422*c54f35caSApple OSS Distributions if (_lastUpdateTimes[channel_index]) {
423*c54f35caSApple OSS Distributions panic("incrementChannelState() cannot be used after setChannelState()!");
424*c54f35caSApple OSS Distributions }
425*c54f35caSApple OSS Distributions
426*c54f35caSApple OSS Distributions res = handleIncrementChannelStateByIndices(channel_index, state_index,
427*c54f35caSApple OSS Distributions time_in_state, intransitions,
428*c54f35caSApple OSS Distributions last_intransition);
429*c54f35caSApple OSS Distributions goto finish;
430*c54f35caSApple OSS Distributions }
431*c54f35caSApple OSS Distributions
432*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
433*c54f35caSApple OSS Distributions
434*c54f35caSApple OSS Distributions finish:
435*c54f35caSApple OSS Distributions unlockReporter();
436*c54f35caSApple OSS Distributions return res;
437*c54f35caSApple OSS Distributions }
438*c54f35caSApple OSS Distributions
439*c54f35caSApple OSS Distributions
440*c54f35caSApple OSS Distributions IOReturn
handleIncrementChannelStateByIndices(int channel_index,int state_index,uint64_t time_in_state,uint64_t intransitions,uint64_t last_intransition)441*c54f35caSApple OSS Distributions IOStateReporter::handleIncrementChannelStateByIndices(int channel_index,
442*c54f35caSApple OSS Distributions int state_index,
443*c54f35caSApple OSS Distributions uint64_t time_in_state,
444*c54f35caSApple OSS Distributions uint64_t intransitions,
445*c54f35caSApple OSS Distributions uint64_t last_intransition /*=0*/)
446*c54f35caSApple OSS Distributions {
447*c54f35caSApple OSS Distributions IOReturn kerr, result = kIOReturnError;
448*c54f35caSApple OSS Distributions IOStateReportValues state_values;
449*c54f35caSApple OSS Distributions int element_index;
450*c54f35caSApple OSS Distributions
451*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index >= _nChannels) {
452*c54f35caSApple OSS Distributions result = kIOReturnBadArgument; goto finish;
453*c54f35caSApple OSS Distributions }
454*c54f35caSApple OSS Distributions
455*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index > (_nElements - state_index)
456*c54f35caSApple OSS Distributions / _channelDimension) {
457*c54f35caSApple OSS Distributions result = kIOReturnOverrun; goto finish;
458*c54f35caSApple OSS Distributions }
459*c54f35caSApple OSS Distributions element_index = channel_index * _channelDimension + state_index;
460*c54f35caSApple OSS Distributions
461*c54f35caSApple OSS Distributions kerr = copyElementValues(element_index, (IOReportElementValues*)&state_values);
462*c54f35caSApple OSS Distributions if (kerr) {
463*c54f35caSApple OSS Distributions result = kerr;
464*c54f35caSApple OSS Distributions goto finish;
465*c54f35caSApple OSS Distributions }
466*c54f35caSApple OSS Distributions
467*c54f35caSApple OSS Distributions state_values.last_intransition = last_intransition;
468*c54f35caSApple OSS Distributions state_values.intransitions += intransitions;
469*c54f35caSApple OSS Distributions state_values.upticks += time_in_state;
470*c54f35caSApple OSS Distributions
471*c54f35caSApple OSS Distributions // determines current time for metadata
472*c54f35caSApple OSS Distributions kerr = setElementValues(element_index, (IOReportElementValues *)&state_values);
473*c54f35caSApple OSS Distributions if (kerr) {
474*c54f35caSApple OSS Distributions result = kerr;
475*c54f35caSApple OSS Distributions goto finish;
476*c54f35caSApple OSS Distributions }
477*c54f35caSApple OSS Distributions
478*c54f35caSApple OSS Distributions // success
479*c54f35caSApple OSS Distributions result = kIOReturnSuccess;
480*c54f35caSApple OSS Distributions
481*c54f35caSApple OSS Distributions finish:
482*c54f35caSApple OSS Distributions return result;
483*c54f35caSApple OSS Distributions }
484*c54f35caSApple OSS Distributions
485*c54f35caSApple OSS Distributions
486*c54f35caSApple OSS Distributions IOReturn
setState(uint64_t new_state_id)487*c54f35caSApple OSS Distributions IOStateReporter::setState(uint64_t new_state_id)
488*c54f35caSApple OSS Distributions {
489*c54f35caSApple OSS Distributions uint64_t last_intransition = 0;
490*c54f35caSApple OSS Distributions uint64_t prev_state_residency = 0;
491*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
492*c54f35caSApple OSS Distributions IOStateReportValues *values;
493*c54f35caSApple OSS Distributions int channel_index = 0, element_index = 0, new_state_index = 0;
494*c54f35caSApple OSS Distributions int cnt;
495*c54f35caSApple OSS Distributions
496*c54f35caSApple OSS Distributions lockReporter();
497*c54f35caSApple OSS Distributions
498*c54f35caSApple OSS Distributions if (_nChannels == 1) {
499*c54f35caSApple OSS Distributions for (cnt = 0; cnt < _channelDimension; cnt++) {
500*c54f35caSApple OSS Distributions new_state_index = element_index + cnt;
501*c54f35caSApple OSS Distributions
502*c54f35caSApple OSS Distributions values = (IOStateReportValues *)getElementValues(new_state_index);
503*c54f35caSApple OSS Distributions
504*c54f35caSApple OSS Distributions if (values == NULL) {
505*c54f35caSApple OSS Distributions res = kIOReturnError;
506*c54f35caSApple OSS Distributions goto finish;
507*c54f35caSApple OSS Distributions }
508*c54f35caSApple OSS Distributions
509*c54f35caSApple OSS Distributions if (values->state_id == new_state_id) {
510*c54f35caSApple OSS Distributions res = handleSetStateByIndices(channel_index, new_state_index,
511*c54f35caSApple OSS Distributions last_intransition,
512*c54f35caSApple OSS Distributions prev_state_residency);
513*c54f35caSApple OSS Distributions goto finish;
514*c54f35caSApple OSS Distributions }
515*c54f35caSApple OSS Distributions }
516*c54f35caSApple OSS Distributions }
517*c54f35caSApple OSS Distributions
518*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
519*c54f35caSApple OSS Distributions
520*c54f35caSApple OSS Distributions finish:
521*c54f35caSApple OSS Distributions unlockReporter();
522*c54f35caSApple OSS Distributions return res;
523*c54f35caSApple OSS Distributions }
524*c54f35caSApple OSS Distributions
525*c54f35caSApple OSS Distributions IOReturn
setState(uint64_t new_state_id,uint64_t last_intransition,uint64_t prev_state_residency)526*c54f35caSApple OSS Distributions IOStateReporter::setState(uint64_t new_state_id,
527*c54f35caSApple OSS Distributions uint64_t last_intransition,
528*c54f35caSApple OSS Distributions uint64_t prev_state_residency)
529*c54f35caSApple OSS Distributions {
530*c54f35caSApple OSS Distributions return setState(new_state_id);
531*c54f35caSApple OSS Distributions }
532*c54f35caSApple OSS Distributions
533*c54f35caSApple OSS Distributions IOReturn
setStateID(uint64_t channel_id,int state_index,uint64_t state_id)534*c54f35caSApple OSS Distributions IOStateReporter::setStateID(uint64_t channel_id,
535*c54f35caSApple OSS Distributions int state_index,
536*c54f35caSApple OSS Distributions uint64_t state_id)
537*c54f35caSApple OSS Distributions {
538*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
539*c54f35caSApple OSS Distributions
540*c54f35caSApple OSS Distributions lockReporter();
541*c54f35caSApple OSS Distributions
542*c54f35caSApple OSS Distributions res = handleSetStateID(channel_id, state_index, state_id);
543*c54f35caSApple OSS Distributions
544*c54f35caSApple OSS Distributions unlockReporter();
545*c54f35caSApple OSS Distributions
546*c54f35caSApple OSS Distributions return res;
547*c54f35caSApple OSS Distributions }
548*c54f35caSApple OSS Distributions
549*c54f35caSApple OSS Distributions
550*c54f35caSApple OSS Distributions IOReturn
handleSetStateID(uint64_t channel_id,int state_index,uint64_t state_id)551*c54f35caSApple OSS Distributions IOStateReporter::handleSetStateID(uint64_t channel_id,
552*c54f35caSApple OSS Distributions int state_index,
553*c54f35caSApple OSS Distributions uint64_t state_id)
554*c54f35caSApple OSS Distributions {
555*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
556*c54f35caSApple OSS Distributions IOStateReportValues state_values;
557*c54f35caSApple OSS Distributions int element_index = 0;
558*c54f35caSApple OSS Distributions
559*c54f35caSApple OSS Distributions IOREPORTER_CHECK_LOCK();
560*c54f35caSApple OSS Distributions
561*c54f35caSApple OSS Distributions if (getFirstElementIndex(channel_id, &element_index) == kIOReturnSuccess) {
562*c54f35caSApple OSS Distributions if (state_index >= _channelDimension) {
563*c54f35caSApple OSS Distributions res = kIOReturnBadArgument; goto finish;
564*c54f35caSApple OSS Distributions }
565*c54f35caSApple OSS Distributions if (_nElements - state_index <= element_index) {
566*c54f35caSApple OSS Distributions res = kIOReturnOverrun; goto finish;
567*c54f35caSApple OSS Distributions }
568*c54f35caSApple OSS Distributions element_index += state_index;
569*c54f35caSApple OSS Distributions
570*c54f35caSApple OSS Distributions if (copyElementValues(element_index, (IOReportElementValues *)&state_values) != kIOReturnSuccess) {
571*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
572*c54f35caSApple OSS Distributions goto finish;
573*c54f35caSApple OSS Distributions }
574*c54f35caSApple OSS Distributions
575*c54f35caSApple OSS Distributions state_values.state_id = state_id;
576*c54f35caSApple OSS Distributions
577*c54f35caSApple OSS Distributions res = setElementValues(element_index, (IOReportElementValues *)&state_values);
578*c54f35caSApple OSS Distributions }
579*c54f35caSApple OSS Distributions
580*c54f35caSApple OSS Distributions // FIXME: set a bit somewhere (reporter-wide?) that state_ids can no longer be
581*c54f35caSApple OSS Distributions // assumed to be contiguous
582*c54f35caSApple OSS Distributions finish:
583*c54f35caSApple OSS Distributions return res;
584*c54f35caSApple OSS Distributions }
585*c54f35caSApple OSS Distributions
586*c54f35caSApple OSS Distributions IOReturn
setStateByIndices(int channel_index,int new_state_index)587*c54f35caSApple OSS Distributions IOStateReporter::setStateByIndices(int channel_index,
588*c54f35caSApple OSS Distributions int new_state_index)
589*c54f35caSApple OSS Distributions {
590*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
591*c54f35caSApple OSS Distributions uint64_t last_intransition = 0;
592*c54f35caSApple OSS Distributions uint64_t prev_state_residency = 0;
593*c54f35caSApple OSS Distributions
594*c54f35caSApple OSS Distributions lockReporter();
595*c54f35caSApple OSS Distributions
596*c54f35caSApple OSS Distributions res = handleSetStateByIndices(channel_index, new_state_index,
597*c54f35caSApple OSS Distributions last_intransition, prev_state_residency);
598*c54f35caSApple OSS Distributions
599*c54f35caSApple OSS Distributions unlockReporter();
600*c54f35caSApple OSS Distributions
601*c54f35caSApple OSS Distributions return res;
602*c54f35caSApple OSS Distributions }
603*c54f35caSApple OSS Distributions
604*c54f35caSApple OSS Distributions IOReturn
setStateByIndices(int channel_index,int new_state_index,uint64_t last_intransition,uint64_t prev_state_residency)605*c54f35caSApple OSS Distributions IOStateReporter::setStateByIndices(int channel_index,
606*c54f35caSApple OSS Distributions int new_state_index,
607*c54f35caSApple OSS Distributions uint64_t last_intransition,
608*c54f35caSApple OSS Distributions uint64_t prev_state_residency)
609*c54f35caSApple OSS Distributions {
610*c54f35caSApple OSS Distributions return setStateByIndices(channel_index, new_state_index);
611*c54f35caSApple OSS Distributions }
612*c54f35caSApple OSS Distributions
613*c54f35caSApple OSS Distributions IOReturn
handleSetStateByIndices(int channel_index,int new_state_index,uint64_t last_intransition,uint64_t prev_state_residency)614*c54f35caSApple OSS Distributions IOStateReporter::handleSetStateByIndices(int channel_index,
615*c54f35caSApple OSS Distributions int new_state_index,
616*c54f35caSApple OSS Distributions uint64_t last_intransition,
617*c54f35caSApple OSS Distributions uint64_t prev_state_residency)
618*c54f35caSApple OSS Distributions {
619*c54f35caSApple OSS Distributions IOReturn res = kIOReturnError;
620*c54f35caSApple OSS Distributions
621*c54f35caSApple OSS Distributions IOStateReportValues curr_state_values, new_state_values;
622*c54f35caSApple OSS Distributions int curr_state_index = 0;
623*c54f35caSApple OSS Distributions int curr_element_index, new_element_index;
624*c54f35caSApple OSS Distributions uint64_t last_ch_update_time = 0;
625*c54f35caSApple OSS Distributions uint64_t recordTime = mach_absolute_time();
626*c54f35caSApple OSS Distributions
627*c54f35caSApple OSS Distributions IOREPORTER_CHECK_LOCK();
628*c54f35caSApple OSS Distributions
629*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index >= _nChannels) {
630*c54f35caSApple OSS Distributions res = kIOReturnBadArgument; goto finish;
631*c54f35caSApple OSS Distributions }
632*c54f35caSApple OSS Distributions
633*c54f35caSApple OSS Distributions // if no timestamp provided, last_intransition = time of recording (now)
634*c54f35caSApple OSS Distributions if (last_intransition == 0) {
635*c54f35caSApple OSS Distributions last_intransition = recordTime;
636*c54f35caSApple OSS Distributions }
637*c54f35caSApple OSS Distributions
638*c54f35caSApple OSS Distributions // First update target state if different than the current state
639*c54f35caSApple OSS Distributions // _currentStates[] initialized to -1 to detect first state transition
640*c54f35caSApple OSS Distributions curr_state_index = _currentStates[channel_index];
641*c54f35caSApple OSS Distributions if (new_state_index != curr_state_index) {
642*c54f35caSApple OSS Distributions // fetch element data
643*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index > (_nElements - new_state_index)
644*c54f35caSApple OSS Distributions / _channelDimension) {
645*c54f35caSApple OSS Distributions res = kIOReturnOverrun; goto finish;
646*c54f35caSApple OSS Distributions }
647*c54f35caSApple OSS Distributions new_element_index = channel_index * _channelDimension + new_state_index;
648*c54f35caSApple OSS Distributions if (copyElementValues(new_element_index,
649*c54f35caSApple OSS Distributions (IOReportElementValues *)&new_state_values)) {
650*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
651*c54f35caSApple OSS Distributions goto finish;
652*c54f35caSApple OSS Distributions }
653*c54f35caSApple OSS Distributions
654*c54f35caSApple OSS Distributions // Update new state's transition info
655*c54f35caSApple OSS Distributions new_state_values.intransitions += 1;
656*c54f35caSApple OSS Distributions new_state_values.last_intransition = last_intransition;
657*c54f35caSApple OSS Distributions
658*c54f35caSApple OSS Distributions // and store the values
659*c54f35caSApple OSS Distributions res = setElementValues(new_element_index,
660*c54f35caSApple OSS Distributions (IOReportElementValues *)&new_state_values,
661*c54f35caSApple OSS Distributions recordTime);
662*c54f35caSApple OSS Distributions
663*c54f35caSApple OSS Distributions if (res != kIOReturnSuccess) {
664*c54f35caSApple OSS Distributions goto finish;
665*c54f35caSApple OSS Distributions }
666*c54f35caSApple OSS Distributions
667*c54f35caSApple OSS Distributions _currentStates[channel_index] = new_state_index;
668*c54f35caSApple OSS Distributions }
669*c54f35caSApple OSS Distributions
670*c54f35caSApple OSS Distributions /* Now update time spent in any previous state
671*c54f35caSApple OSS Distributions * If new_state_index = curr_state_index, this updates time in the
672*c54f35caSApple OSS Distributions * current state. If this is the channel's first state transition,
673*c54f35caSApple OSS Distributions * the last update time will be zero.
674*c54f35caSApple OSS Distributions *
675*c54f35caSApple OSS Distributions * Note: While setState() should never be called on a channel being
676*c54f35caSApple OSS Distributions * updated with increment/overrideChannelState(), that's another way
677*c54f35caSApple OSS Distributions * that the last update time might not exist. Regardless, if there
678*c54f35caSApple OSS Distributions * is no basis for determining time spent in previous state, there's
679*c54f35caSApple OSS Distributions * nothing to update!
680*c54f35caSApple OSS Distributions */
681*c54f35caSApple OSS Distributions last_ch_update_time = _lastUpdateTimes[channel_index];
682*c54f35caSApple OSS Distributions if (last_ch_update_time != 0) {
683*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index > (_nElements - curr_state_index)
684*c54f35caSApple OSS Distributions / _channelDimension) {
685*c54f35caSApple OSS Distributions res = kIOReturnOverrun; goto finish;
686*c54f35caSApple OSS Distributions }
687*c54f35caSApple OSS Distributions curr_element_index = channel_index * _channelDimension + curr_state_index;
688*c54f35caSApple OSS Distributions if (copyElementValues(curr_element_index,
689*c54f35caSApple OSS Distributions (IOReportElementValues *)&curr_state_values)) {
690*c54f35caSApple OSS Distributions res = kIOReturnBadArgument;
691*c54f35caSApple OSS Distributions goto finish;
692*c54f35caSApple OSS Distributions }
693*c54f35caSApple OSS Distributions // compute the time spent in previous state, unless provided
694*c54f35caSApple OSS Distributions if (prev_state_residency == 0) {
695*c54f35caSApple OSS Distributions prev_state_residency = last_intransition - last_ch_update_time;
696*c54f35caSApple OSS Distributions }
697*c54f35caSApple OSS Distributions
698*c54f35caSApple OSS Distributions curr_state_values.upticks += prev_state_residency;
699*c54f35caSApple OSS Distributions
700*c54f35caSApple OSS Distributions res = setElementValues(curr_element_index,
701*c54f35caSApple OSS Distributions (IOReportElementValues*)&curr_state_values,
702*c54f35caSApple OSS Distributions recordTime);
703*c54f35caSApple OSS Distributions
704*c54f35caSApple OSS Distributions if (res != kIOReturnSuccess) {
705*c54f35caSApple OSS Distributions goto finish;
706*c54f35caSApple OSS Distributions }
707*c54f35caSApple OSS Distributions }
708*c54f35caSApple OSS Distributions
709*c54f35caSApple OSS Distributions // record basis for next "time in prior state" calculation
710*c54f35caSApple OSS Distributions // (also arms a panic in override/incrementChannelState())
711*c54f35caSApple OSS Distributions _lastUpdateTimes[channel_index] = last_intransition;
712*c54f35caSApple OSS Distributions
713*c54f35caSApple OSS Distributions finish:
714*c54f35caSApple OSS Distributions return res;
715*c54f35caSApple OSS Distributions }
716*c54f35caSApple OSS Distributions
717*c54f35caSApple OSS Distributions
718*c54f35caSApple OSS Distributions // blocks might make this slightly easier?
719*c54f35caSApple OSS Distributions uint64_t
getStateInTransitions(uint64_t channel_id,uint64_t state_id)720*c54f35caSApple OSS Distributions IOStateReporter::getStateInTransitions(uint64_t channel_id,
721*c54f35caSApple OSS Distributions uint64_t state_id)
722*c54f35caSApple OSS Distributions {
723*c54f35caSApple OSS Distributions return _getStateValue(channel_id, state_id, kInTransitions);
724*c54f35caSApple OSS Distributions }
725*c54f35caSApple OSS Distributions
726*c54f35caSApple OSS Distributions uint64_t
getStateResidencyTime(uint64_t channel_id,uint64_t state_id)727*c54f35caSApple OSS Distributions IOStateReporter::getStateResidencyTime(uint64_t channel_id,
728*c54f35caSApple OSS Distributions uint64_t state_id)
729*c54f35caSApple OSS Distributions {
730*c54f35caSApple OSS Distributions return _getStateValue(channel_id, state_id, kResidencyTime);
731*c54f35caSApple OSS Distributions }
732*c54f35caSApple OSS Distributions
733*c54f35caSApple OSS Distributions uint64_t
getStateLastTransitionTime(uint64_t channel_id,uint64_t state_id)734*c54f35caSApple OSS Distributions IOStateReporter::getStateLastTransitionTime(uint64_t channel_id,
735*c54f35caSApple OSS Distributions uint64_t state_id)
736*c54f35caSApple OSS Distributions {
737*c54f35caSApple OSS Distributions return _getStateValue(channel_id, state_id, kLastTransitionTime);
738*c54f35caSApple OSS Distributions }
739*c54f35caSApple OSS Distributions
740*c54f35caSApple OSS Distributions uint64_t
_getStateValue(uint64_t channel_id,uint64_t state_id,enum valueSelector value)741*c54f35caSApple OSS Distributions IOStateReporter::_getStateValue(uint64_t channel_id,
742*c54f35caSApple OSS Distributions uint64_t state_id,
743*c54f35caSApple OSS Distributions enum valueSelector value)
744*c54f35caSApple OSS Distributions {
745*c54f35caSApple OSS Distributions int channel_index = 0, element_index = 0, cnt;
746*c54f35caSApple OSS Distributions IOStateReportValues *values = NULL;
747*c54f35caSApple OSS Distributions uint64_t result = kIOReportInvalidValue;
748*c54f35caSApple OSS Distributions
749*c54f35caSApple OSS Distributions lockReporter();
750*c54f35caSApple OSS Distributions
751*c54f35caSApple OSS Distributions if (getChannelIndices(channel_id, &channel_index, &element_index) == kIOReturnSuccess) {
752*c54f35caSApple OSS Distributions if (updateChannelValues(channel_index) == kIOReturnSuccess) {
753*c54f35caSApple OSS Distributions for (cnt = 0; cnt < _channelDimension; cnt++) {
754*c54f35caSApple OSS Distributions values = (IOStateReportValues *)getElementValues(element_index);
755*c54f35caSApple OSS Distributions
756*c54f35caSApple OSS Distributions if (state_id == values->state_id) {
757*c54f35caSApple OSS Distributions switch (value) {
758*c54f35caSApple OSS Distributions case kInTransitions:
759*c54f35caSApple OSS Distributions result = values->intransitions;
760*c54f35caSApple OSS Distributions break;
761*c54f35caSApple OSS Distributions case kResidencyTime:
762*c54f35caSApple OSS Distributions result = values->upticks;
763*c54f35caSApple OSS Distributions break;
764*c54f35caSApple OSS Distributions case kLastTransitionTime:
765*c54f35caSApple OSS Distributions result = values->last_intransition;
766*c54f35caSApple OSS Distributions break;
767*c54f35caSApple OSS Distributions default:
768*c54f35caSApple OSS Distributions break;
769*c54f35caSApple OSS Distributions }
770*c54f35caSApple OSS Distributions
771*c54f35caSApple OSS Distributions break;
772*c54f35caSApple OSS Distributions }
773*c54f35caSApple OSS Distributions
774*c54f35caSApple OSS Distributions element_index++;
775*c54f35caSApple OSS Distributions }
776*c54f35caSApple OSS Distributions }
777*c54f35caSApple OSS Distributions }
778*c54f35caSApple OSS Distributions
779*c54f35caSApple OSS Distributions unlockReporter();
780*c54f35caSApple OSS Distributions return result;
781*c54f35caSApple OSS Distributions }
782*c54f35caSApple OSS Distributions
783*c54f35caSApple OSS Distributions
784*c54f35caSApple OSS Distributions uint64_t
getStateLastChannelUpdateTime(uint64_t channel_id)785*c54f35caSApple OSS Distributions IOStateReporter::getStateLastChannelUpdateTime(uint64_t channel_id)
786*c54f35caSApple OSS Distributions {
787*c54f35caSApple OSS Distributions int channel_index;
788*c54f35caSApple OSS Distributions uint64_t result = kIOReportInvalidValue;
789*c54f35caSApple OSS Distributions
790*c54f35caSApple OSS Distributions lockReporter();
791*c54f35caSApple OSS Distributions
792*c54f35caSApple OSS Distributions if (getChannelIndex(channel_id, &channel_index) == kIOReturnSuccess) {
793*c54f35caSApple OSS Distributions result = _lastUpdateTimes[channel_index];
794*c54f35caSApple OSS Distributions }
795*c54f35caSApple OSS Distributions
796*c54f35caSApple OSS Distributions unlockReporter();
797*c54f35caSApple OSS Distributions
798*c54f35caSApple OSS Distributions return result;
799*c54f35caSApple OSS Distributions }
800*c54f35caSApple OSS Distributions
801*c54f35caSApple OSS Distributions
802*c54f35caSApple OSS Distributions /* updateChannelValues() is called to refresh state before being
803*c54f35caSApple OSS Distributions * reported outside the reporter. In the case of IOStateReporter,
804*c54f35caSApple OSS Distributions * this is primarily an update to the "time in state" data.
805*c54f35caSApple OSS Distributions */
806*c54f35caSApple OSS Distributions IOReturn
updateChannelValues(int channel_index)807*c54f35caSApple OSS Distributions IOStateReporter::updateChannelValues(int channel_index)
808*c54f35caSApple OSS Distributions {
809*c54f35caSApple OSS Distributions IOReturn kerr, result = kIOReturnError;
810*c54f35caSApple OSS Distributions
811*c54f35caSApple OSS Distributions int state_index, element_idx;
812*c54f35caSApple OSS Distributions uint64_t currentTime;
813*c54f35caSApple OSS Distributions uint64_t last_ch_update_time;
814*c54f35caSApple OSS Distributions uint64_t time_in_state;
815*c54f35caSApple OSS Distributions IOStateReportValues state_values;
816*c54f35caSApple OSS Distributions
817*c54f35caSApple OSS Distributions IOREPORTER_CHECK_LOCK();
818*c54f35caSApple OSS Distributions
819*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index >= _nChannels) {
820*c54f35caSApple OSS Distributions result = kIOReturnBadArgument; goto finish;
821*c54f35caSApple OSS Distributions }
822*c54f35caSApple OSS Distributions
823*c54f35caSApple OSS Distributions /* First check to see whether this channel has begun self-
824*c54f35caSApple OSS Distributions * calculation of time in state. It's possible this channel
825*c54f35caSApple OSS Distributions * has yet to be initialized or that the driver is updating
826*c54f35caSApple OSS Distributions * the channel with override/incrementChannelState() which
827*c54f35caSApple OSS Distributions * never enable automatic time-in-state updates. In that case,
828*c54f35caSApple OSS Distributions * there is nothing to update and we return success.
829*c54f35caSApple OSS Distributions */
830*c54f35caSApple OSS Distributions last_ch_update_time = _lastUpdateTimes[channel_index];
831*c54f35caSApple OSS Distributions if (last_ch_update_time == 0) {
832*c54f35caSApple OSS Distributions result = kIOReturnSuccess; goto finish;
833*c54f35caSApple OSS Distributions }
834*c54f35caSApple OSS Distributions
835*c54f35caSApple OSS Distributions // figure out the current state (if any)
836*c54f35caSApple OSS Distributions state_index = _currentStates[channel_index];
837*c54f35caSApple OSS Distributions
838*c54f35caSApple OSS Distributions // e.g. given 4 4-state channels, the boundary is ch[3].st[3] <- _elems[15]
839*c54f35caSApple OSS Distributions if (channel_index < 0 || channel_index > (_nElements - state_index)
840*c54f35caSApple OSS Distributions / _channelDimension) {
841*c54f35caSApple OSS Distributions result = kIOReturnOverrun; goto finish;
842*c54f35caSApple OSS Distributions }
843*c54f35caSApple OSS Distributions element_idx = channel_index * _channelDimension + state_index;
844*c54f35caSApple OSS Distributions
845*c54f35caSApple OSS Distributions // get the current values
846*c54f35caSApple OSS Distributions kerr = copyElementValues(element_idx, (IOReportElementValues*)&state_values);
847*c54f35caSApple OSS Distributions if (kerr) {
848*c54f35caSApple OSS Distributions result = kerr; goto finish;
849*c54f35caSApple OSS Distributions }
850*c54f35caSApple OSS Distributions
851*c54f35caSApple OSS Distributions // calculate time in state
852*c54f35caSApple OSS Distributions currentTime = mach_absolute_time();
853*c54f35caSApple OSS Distributions time_in_state = currentTime - last_ch_update_time;
854*c54f35caSApple OSS Distributions state_values.upticks += time_in_state;
855*c54f35caSApple OSS Distributions
856*c54f35caSApple OSS Distributions // and store the values
857*c54f35caSApple OSS Distributions kerr = setElementValues(element_idx,
858*c54f35caSApple OSS Distributions (IOReportElementValues *)&state_values,
859*c54f35caSApple OSS Distributions currentTime);
860*c54f35caSApple OSS Distributions if (kerr) {
861*c54f35caSApple OSS Distributions result = kerr; goto finish;
862*c54f35caSApple OSS Distributions }
863*c54f35caSApple OSS Distributions
864*c54f35caSApple OSS Distributions // Record basis for next "prior time" calculation
865*c54f35caSApple OSS Distributions _lastUpdateTimes[channel_index] = currentTime;
866*c54f35caSApple OSS Distributions
867*c54f35caSApple OSS Distributions
868*c54f35caSApple OSS Distributions // success
869*c54f35caSApple OSS Distributions result = kIOReturnSuccess;
870*c54f35caSApple OSS Distributions
871*c54f35caSApple OSS Distributions finish:
872*c54f35caSApple OSS Distributions return result;
873*c54f35caSApple OSS Distributions }
874*c54f35caSApple OSS Distributions
875*c54f35caSApple OSS Distributions /* static */ OSPtr<IOReportLegendEntry>
createLegend(const uint64_t * channelIDs,const char ** channelNames,int channelCount,int nstates,IOReportCategories categories,IOReportUnit unit)876*c54f35caSApple OSS Distributions IOStateReporter::createLegend(const uint64_t *channelIDs,
877*c54f35caSApple OSS Distributions const char **channelNames,
878*c54f35caSApple OSS Distributions int channelCount,
879*c54f35caSApple OSS Distributions int nstates,
880*c54f35caSApple OSS Distributions IOReportCategories categories,
881*c54f35caSApple OSS Distributions IOReportUnit unit)
882*c54f35caSApple OSS Distributions {
883*c54f35caSApple OSS Distributions IOReportChannelType channelType = {
884*c54f35caSApple OSS Distributions .categories = categories,
885*c54f35caSApple OSS Distributions .report_format = kIOReportFormatState,
886*c54f35caSApple OSS Distributions .nelements = static_cast<uint16_t>(nstates),
887*c54f35caSApple OSS Distributions .element_idx = 0
888*c54f35caSApple OSS Distributions };
889*c54f35caSApple OSS Distributions
890*c54f35caSApple OSS Distributions return IOReporter::legendWith(channelIDs, channelNames, channelCount, channelType, unit);
891*c54f35caSApple OSS Distributions }
892