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