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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/*
* Copyright (c) 2015 Dmitry Bakhvalov
*
* SPDX-FileCopyrightText: 2024 Hiredict Contributors
* SPDX-FileCopyrightText: 2024 Dmitry Bakhvalov
*
* SPDX-License-Identifier: BSD-3-Clause
* SPDX-License-Identifier: LGPL-3.0-or-later
*
*/
#ifndef __HIREDICT_MACOSX_H__
#define __HIREDICT_MACOSX_H__
#include <CoreFoundation/CoreFoundation.h>
#include "../hiredict.h"
#include "../async.h"
typedef struct {
redictAsyncContext *context;
CFSocketRef socketRef;
CFRunLoopSourceRef sourceRef;
} RedictRunLoop;
static int freeRedictRunLoop(RedictRunLoop* redictRunLoop) {
if( redictRunLoop != NULL ) {
if( redictRunLoop->sourceRef != NULL ) {
CFRunLoopSourceInvalidate(redictRunLoop->sourceRef);
CFRelease(redictRunLoop->sourceRef);
}
if( redictRunLoop->socketRef != NULL ) {
CFSocketInvalidate(redictRunLoop->socketRef);
CFRelease(redictRunLoop->socketRef);
}
hi_free(redictRunLoop);
}
return REDICT_ERR;
}
static void redictMacOSAddRead(void *privdata) {
RedictRunLoop *redictRunLoop = (RedictRunLoop*)privdata;
CFSocketEnableCallBacks(redictRunLoop->socketRef, kCFSocketReadCallBack);
}
static void redictMacOSDelRead(void *privdata) {
RedictRunLoop *redictRunLoop = (RedictRunLoop*)privdata;
CFSocketDisableCallBacks(redictRunLoop->socketRef, kCFSocketReadCallBack);
}
static void redictMacOSAddWrite(void *privdata) {
RedictRunLoop *redictRunLoop = (RedictRunLoop*)privdata;
CFSocketEnableCallBacks(redictRunLoop->socketRef, kCFSocketWriteCallBack);
}
static void redictMacOSDelWrite(void *privdata) {
RedictRunLoop *redictRunLoop = (RedictRunLoop*)privdata;
CFSocketDisableCallBacks(redictRunLoop->socketRef, kCFSocketWriteCallBack);
}
static void redictMacOSCleanup(void *privdata) {
RedictRunLoop *redictRunLoop = (RedictRunLoop*)privdata;
freeRedictRunLoop(redictRunLoop);
}
static void redictMacOSAsyncCallback(CFSocketRef __unused s, CFSocketCallBackType callbackType, CFDataRef __unused address, const void __unused *data, void *info) {
redictAsyncContext* context = (redictAsyncContext*) info;
switch (callbackType) {
case kCFSocketReadCallBack:
redictAsyncHandleRead(context);
break;
case kCFSocketWriteCallBack:
redictAsyncHandleWrite(context);
break;
default:
break;
}
}
static int redictMacOSAttach(redictAsyncContext *redictAsyncCtx, CFRunLoopRef runLoop) {
redictContext *redictCtx = &(redictAsyncCtx->c);
/* Nothing should be attached when something is already attached */
if( redictAsyncCtx->ev.data != NULL ) return REDICT_ERR;
RedictRunLoop* redictRunLoop = (RedictRunLoop*) hi_calloc(1, sizeof(RedictRunLoop));
if (redictRunLoop == NULL)
return REDICT_ERR;
/* Setup redict stuff */
redictRunLoop->context = redictAsyncCtx;
redictAsyncCtx->ev.addRead = redictMacOSAddRead;
redictAsyncCtx->ev.delRead = redictMacOSDelRead;
redictAsyncCtx->ev.addWrite = redictMacOSAddWrite;
redictAsyncCtx->ev.delWrite = redictMacOSDelWrite;
redictAsyncCtx->ev.cleanup = redictMacOSCleanup;
redictAsyncCtx->ev.data = redictRunLoop;
/* Initialize and install read/write events */
CFSocketContext socketCtx = { 0, redictAsyncCtx, NULL, NULL, NULL };
redictRunLoop->socketRef = CFSocketCreateWithNative(NULL, redictCtx->fd,
kCFSocketReadCallBack | kCFSocketWriteCallBack,
redictMacOSAsyncCallback,
&socketCtx);
if( !redictRunLoop->socketRef ) return freeRedictRunLoop(redictRunLoop);
redictRunLoop->sourceRef = CFSocketCreateRunLoopSource(NULL, redictRunLoop->socketRef, 0);
if( !redictRunLoop->sourceRef ) return freeRedictRunLoop(redictRunLoop);
CFRunLoopAddSource(runLoop, redictRunLoop->sourceRef, kCFRunLoopDefaultMode);
return REDICT_OK;
}
#endif
|