1/* 2 * 3 * @APPLE_LICENSE_HEADER_START@ 4 * 5 * Copyright (c) 1999-2013 Apple Computer, Inc. All Rights Reserved. 6 * 7 * This file contains Original Code and/or Modifications of Original Code 8 * as defined in and that are subject to the Apple Public Source License 9 * Version 2.0 (the 'License'). You may not use this file except in 10 * compliance with the License. Please obtain a copy of the License at 11 * http://www.opensource.apple.com/apsl/ and read it before using this 12 * file. 13 * 14 * The Original Code and all software distributed under the License are 15 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 16 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 17 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 18 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 19 * Please see the License for the specific language governing rights and 20 * limitations under the License. 21 * 22 * @APPLE_LICENSE_HEADER_END@ 23 */ 24 25#ifndef IOHIDFamily_OSStackRetain_h 26#define IOHIDFamily_OSStackRetain_h 27 28#include <IOKit/IOTypes.h> 29 30class OSStackRetain { 31 OSObject *_retainedObject; 32 OSStackRetain() : _retainedObject(NULL) {} 33public: 34 OSStackRetain(OSObject *object, bool retain = false) : _retainedObject(object) { 35 if (_retainedObject && retain) { 36 _retainedObject->retain(); 37 } 38 } 39 ~OSStackRetain() { 40 if (_retainedObject) { 41 _retainedObject->release(); 42 } 43 _retainedObject = NULL; 44 } 45}; 46 47#define CONVERT_TO_STACK_RETAIN(X) OSStackRetain _ ## X ## _retained(X, false) 48#define RETAIN_ON_STACK(X) OSStackRetain _ ## X ## _retained(X, true) 49 50#endif 51