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