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