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