1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/* This file is part of Owl.
*
* Copyright © 2019-2021 Sergey Bugaev <bugaevc@gmail.com>
*
* Owl is free software: you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published
* by the Free Software Foundation, either version 3 of the License,
* or (at your option) any later version.
*
* Owl is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* for more details.
*
* You should have received a copy of the GNU General Public License
* along with Owl. If not, see <http://www.gnu.org/licenses/>.
*/
#import "OwlZowlMachIpcPortV1.h"
#import <Cocoa/Cocoa.h>
#import <wayland-server.h>
#import "owl-mach-ipc-unstable-v1.h"
#import "owl-mach-ipc-unstable-v1-mig.h"
@implementation OwlZowlMachIpcPortV1
static NSMutableDictionary *secretToPort;
+ (void) initialize {
if (secretToPort == nil) {
secretToPort = [[NSMutableDictionary alloc] init];
}
}
static void mach_ipc_port_destroy_handler(
struct wl_client *client,
struct wl_resource *resource
) {
wl_resource_destroy(resource);
}
static const struct zowl_mach_ipc_port_v1_interface mach_ipc_port_impl = {
.destroy = mach_ipc_port_destroy_handler
};
static void mach_ipc_port_destroy(struct wl_resource *resource) {
OwlZowlMachIpcPortV1 *self = wl_resource_get_user_data(resource);
[secretToPort removeObjectsForKeys: [secretToPort allKeysForObject: self]];
[self release];
}
- (id) initWithResource: (struct wl_resource *) resource {
_resource = resource;
wl_resource_set_implementation(
resource,
&mach_ipc_port_impl,
[self retain],
mach_ipc_port_destroy
);
// Cannot use NSUUID, as it's 10.8+.
CFUUIDRef uuid = CFUUIDCreate(kCFAllocatorDefault);
NSString *secret = (NSString *) CFUUIDCreateString(kCFAllocatorDefault, uuid);
CFRelease(uuid);
zowl_mach_ipc_port_v1_send_secret(resource, [secret UTF8String]);
[secretToPort setObject: self forKey: secret];
[secret release];
return self;
}
- (void) setPort: (mach_port_t) port {
_port = port;
}
// FIXME: deallocate the port in dealloc?
kern_return_t owl_mach_ipc_v1_server_retrieve_port(
mach_port_t server_port,
secret_t raw_secret,
mach_port_t *port
) {
NSString *secret = [NSString stringWithUTF8String: raw_secret];
OwlZowlMachIpcPortV1 *self = [secretToPort objectForKey: secret];
if (self == nil) {
return KERN_INVALID_ARGUMENT;
}
[secretToPort removeObjectForKey: secret];
*port = self->_port;
return KERN_SUCCESS;
}
@end
|