1*33de042dSApple OSS Distributions #include "TestIOUserClient2022Entitlements.h"
2*33de042dSApple OSS Distributions #include <IOKit/IOService.h>
3*33de042dSApple OSS Distributions #include <IOKit/IOUserClient.h>
4*33de042dSApple OSS Distributions #include <IOKit/IOKitServer.h>
5*33de042dSApple OSS Distributions #include <kern/ipc_kobject.h>
6*33de042dSApple OSS Distributions
7*33de042dSApple OSS Distributions #if (DEVELOPMENT || DEBUG)
8*33de042dSApple OSS Distributions
9*33de042dSApple OSS Distributions OSDefineMetaClassAndStructors(TestIOUserClient2022Entitlements, IOService);
10*33de042dSApple OSS Distributions
11*33de042dSApple OSS Distributions OSDefineMetaClassAndStructors(TestIOUserClient2022EntitlementsUserClient, IOUserClient2022);
12*33de042dSApple OSS Distributions
13*33de042dSApple OSS Distributions bool
start(IOService * provider)14*33de042dSApple OSS Distributions TestIOUserClient2022Entitlements::start(IOService * provider)
15*33de042dSApple OSS Distributions {
16*33de042dSApple OSS Distributions OSString * str = OSString::withCStringNoCopy("TestIOUserClient2022EntitlementsUserClient");
17*33de042dSApple OSS Distributions bool ret = IOService::start(provider);
18*33de042dSApple OSS Distributions if (ret && str != NULL) {
19*33de042dSApple OSS Distributions setProperty(gIOUserClientClassKey, str);
20*33de042dSApple OSS Distributions registerService();
21*33de042dSApple OSS Distributions }
22*33de042dSApple OSS Distributions OSSafeReleaseNULL(str);
23*33de042dSApple OSS Distributions return ret;
24*33de042dSApple OSS Distributions }
25*33de042dSApple OSS Distributions
26*33de042dSApple OSS Distributions bool
start(IOService * provider)27*33de042dSApple OSS Distributions TestIOUserClient2022EntitlementsUserClient::start(IOService * provider)
28*33de042dSApple OSS Distributions {
29*33de042dSApple OSS Distributions if (!IOUserClient2022::start(provider)) {
30*33de042dSApple OSS Distributions return false;
31*33de042dSApple OSS Distributions }
32*33de042dSApple OSS Distributions setProperty(kIOUserClientDefaultLockingKey, kOSBooleanTrue);
33*33de042dSApple OSS Distributions setProperty(kIOUserClientDefaultLockingSetPropertiesKey, kOSBooleanTrue);
34*33de042dSApple OSS Distributions setProperty(kIOUserClientDefaultLockingSingleThreadExternalMethodKey, kOSBooleanTrue);
35*33de042dSApple OSS Distributions
36*33de042dSApple OSS Distributions setProperty(kIOUserClientEntitlementsKey, "com.apple.iokit.test-check-entitlement-open");
37*33de042dSApple OSS Distributions
38*33de042dSApple OSS Distributions return true;
39*33de042dSApple OSS Distributions }
40*33de042dSApple OSS Distributions
41*33de042dSApple OSS Distributions IOReturn
clientClose()42*33de042dSApple OSS Distributions TestIOUserClient2022EntitlementsUserClient::clientClose()
43*33de042dSApple OSS Distributions {
44*33de042dSApple OSS Distributions terminate();
45*33de042dSApple OSS Distributions return kIOReturnSuccess;
46*33de042dSApple OSS Distributions }
47*33de042dSApple OSS Distributions
48*33de042dSApple OSS Distributions IOReturn
extBasicMethod(OSObject * target,void * reference,IOExternalMethodArguments * arguments)49*33de042dSApple OSS Distributions TestIOUserClient2022EntitlementsUserClient::extBasicMethod(OSObject * target, void * reference, IOExternalMethodArguments * arguments)
50*33de042dSApple OSS Distributions {
51*33de042dSApple OSS Distributions return kIOReturnSuccess;
52*33de042dSApple OSS Distributions }
53*33de042dSApple OSS Distributions
54*33de042dSApple OSS Distributions IOReturn
extPerSelectorCheck(OSObject * target,void * reference,IOExternalMethodArguments * arguments)55*33de042dSApple OSS Distributions TestIOUserClient2022EntitlementsUserClient::extPerSelectorCheck(OSObject * target, void * reference, IOExternalMethodArguments * arguments)
56*33de042dSApple OSS Distributions {
57*33de042dSApple OSS Distributions return kIOReturnSuccess;
58*33de042dSApple OSS Distributions }
59*33de042dSApple OSS Distributions
60*33de042dSApple OSS Distributions IOReturn
externalMethod(uint32_t selector,IOExternalMethodArgumentsOpaque * args)61*33de042dSApple OSS Distributions TestIOUserClient2022EntitlementsUserClient::externalMethod(uint32_t selector, IOExternalMethodArgumentsOpaque * args)
62*33de042dSApple OSS Distributions {
63*33de042dSApple OSS Distributions static const IOExternalMethodDispatch2022 dispatchArray[] = {
64*33de042dSApple OSS Distributions [0] {
65*33de042dSApple OSS Distributions .function = &TestIOUserClient2022EntitlementsUserClient::extBasicMethod,
66*33de042dSApple OSS Distributions .checkScalarInputCount = 0,
67*33de042dSApple OSS Distributions .checkStructureInputSize = 0,
68*33de042dSApple OSS Distributions .checkScalarOutputCount = 0,
69*33de042dSApple OSS Distributions .checkStructureOutputSize = 0,
70*33de042dSApple OSS Distributions .allowAsync = false,
71*33de042dSApple OSS Distributions .checkEntitlement = NULL,
72*33de042dSApple OSS Distributions },
73*33de042dSApple OSS Distributions [1] {
74*33de042dSApple OSS Distributions .function = &TestIOUserClient2022EntitlementsUserClient::extPerSelectorCheck,
75*33de042dSApple OSS Distributions .checkScalarInputCount = 0,
76*33de042dSApple OSS Distributions .checkStructureInputSize = 0,
77*33de042dSApple OSS Distributions .checkScalarOutputCount = 0,
78*33de042dSApple OSS Distributions .checkStructureOutputSize = 0,
79*33de042dSApple OSS Distributions .allowAsync = false,
80*33de042dSApple OSS Distributions .checkEntitlement = "com.apple.iokit.test-check-entitlement-per-selector",
81*33de042dSApple OSS Distributions },
82*33de042dSApple OSS Distributions };
83*33de042dSApple OSS Distributions
84*33de042dSApple OSS Distributions return dispatchExternalMethod(selector, args, dispatchArray, sizeof(dispatchArray) / sizeof(dispatchArray[0]), this, NULL);
85*33de042dSApple OSS Distributions }
86*33de042dSApple OSS Distributions
87*33de042dSApple OSS Distributions #endif /* (DEVELOPMENT || DEBUG) */
88