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