]> git.saurik.com Git - wxWidgets.git/blob - src/mac/corefoundation/gsockosx.cpp
cleanup
[wxWidgets.git] / src / mac / corefoundation / gsockosx.cpp
1 /* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
3 * Name: src/mac/corefoundation/gsockosx.c
4 * Purpose: GSocket: Mac OS X mach-o part
5 * CVSID: $Id$
6 * Mac code by Brian Victor, February 2002. Email comments to bhv1@psu.edu
7 * ------------------------------------------------------------------------- */
8
9 #include "wx/wxprec.h"
10
11 #if wxUSE_SOCKETS
12
13 #include <stdlib.h>
14 #include "wx/gsocket.h"
15 #include "wx/unix/gsockunx.h"
16
17 #include <CoreFoundation/CoreFoundation.h>
18
19 #define ALL_CALLBACK_TYPES (kCFSocketReadCallBack | kCFSocketWriteCallBack | kCFSocketConnectCallBack)
20
21 struct MacGSocketData
22 {
23 CFSocketRef socket;
24 CFRunLoopSourceRef source;
25 };
26
27 // Sockets must use the event loop on the main thread
28 // We will store the main loop's reference when Initialize is called
29 static CFRunLoopRef s_mainRunLoop = NULL;
30
31 void Mac_Socket_Callback(CFSocketRef WXUNUSED(s), CFSocketCallBackType callbackType,
32 CFDataRef WXUNUSED(address), const void* data, void* info)
33 {
34 GSocket* socket = (GSocket*)info;
35 struct MacGSocketData* macdata;
36 macdata = (struct MacGSocketData*)socket->m_gui_dependent;
37 if (!macdata) return;
38 switch (callbackType)
39 {
40 case kCFSocketConnectCallBack:
41 assert(!socket->m_server);
42 // KH: If data is non-NULL, the connect failed, do not call Detected_Write,
43 // which will only end up creating a spurious connect event because the
44 // call to getsocketopt SO_ERROR inexplicably returns no error.
45 // The change in behavior cannot be traced to any particular commit or
46 // timeframe so I'm not sure what to think, but after so many hours,
47 // this seems to address the issue and it's time to move on.
48 if (data == NULL)
49 socket->Detected_Write();
50 break;
51 case kCFSocketReadCallBack:
52 socket->Detected_Read();
53 break;
54 case kCFSocketWriteCallBack:
55 socket->Detected_Write();
56 break;
57 default:
58 break; /* We shouldn't get here. */
59 }
60 }
61
62 struct MacGSocketData* _GSocket_Get_Mac_Socket(GSocket *socket)
63 {
64 /* If socket is already created, returns a pointer to the data */
65 /* Otherwise, creates socket and returns the pointer */
66 CFSocketContext cont;
67 struct MacGSocketData* data = (struct MacGSocketData*)socket->m_gui_dependent;
68
69 if (data && data->source) return data;
70
71 /* CFSocket has not been created, create it: */
72 if (socket->m_fd < 0 || !data) return NULL;
73 cont.version = 0; cont.retain = NULL;
74 cont.release = NULL; cont.copyDescription = NULL;
75 cont.info = socket;
76
77 CFSocketRef cf = CFSocketCreateWithNative(NULL, socket->m_fd,
78 ALL_CALLBACK_TYPES, Mac_Socket_Callback, &cont);
79 CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(NULL, cf, 0);
80 assert(source);
81 socket->m_gui_dependent = (char*)data;
82
83 /* Keep the source and the socket around. */
84 data->source = source;
85 data->socket = cf;
86
87 return data;
88 }
89
90 bool GSocketGUIFunctionsTableConcrete::CanUseEventLoop()
91 { return true; }
92
93 bool GSocketGUIFunctionsTableConcrete::OnInit(void)
94 {
95 // No need to store the main loop again
96 if (s_mainRunLoop != NULL)
97 return true;
98
99 // Get the loop for the main thread so our events will actually fire.
100 // The common socket.cpp code will assert if initialize is called from a
101 // secondary thread, otherwise Mac would have the same problems as MSW
102 s_mainRunLoop = CFRunLoopGetCurrent();
103 CFRetain(s_mainRunLoop);
104
105 return true;
106 }
107
108 void GSocketGUIFunctionsTableConcrete::OnExit(void)
109 {
110 // Release the reference count, and set the reference back to NULL
111 CFRelease(s_mainRunLoop);
112 s_mainRunLoop = NULL;
113 }
114
115 bool GSocketGUIFunctionsTableConcrete::Init_Socket(GSocket *socket)
116 {
117 struct MacGSocketData *data = (struct MacGSocketData *)malloc(sizeof(struct MacGSocketData));
118 if (data)
119 {
120 socket->m_gui_dependent = (char*)data;
121 data->socket = NULL;
122 data->source = NULL;
123 return 1;
124 }
125 return 0;
126 }
127
128 void GSocketGUIFunctionsTableConcrete::Destroy_Socket(GSocket *socket)
129 {
130 struct MacGSocketData *data = (struct MacGSocketData*)(socket->m_gui_dependent);
131 if (data)
132 {
133 if ( data->source )
134 CFRelease(data->source);
135 if ( data->socket )
136 CFRelease(data->socket);
137 free(data);
138 }
139 }
140
141 void GSocketGUIFunctionsTableConcrete::Install_Callback(GSocket *socket, GSocketEvent event)
142 {
143 int c;
144 struct MacGSocketData* data = _GSocket_Get_Mac_Socket(socket);
145 if (!data) return;
146 switch (event)
147 {
148 case GSOCK_CONNECTION:
149 if(socket->m_server)
150 c = kCFSocketReadCallBack;
151 else
152 c = kCFSocketConnectCallBack;
153 break;
154 case GSOCK_LOST:
155 case GSOCK_INPUT:
156 c = kCFSocketReadCallBack;
157 break;
158 case GSOCK_OUTPUT:
159 c = kCFSocketWriteCallBack;
160 break;
161 default:
162 c = 0;
163 }
164 CFSocketEnableCallBacks(data->socket, c);
165 }
166
167 void GSocketGUIFunctionsTableConcrete::Uninstall_Callback(GSocket *socket, GSocketEvent event)
168 {
169 int c;
170 struct MacGSocketData* data = _GSocket_Get_Mac_Socket(socket);
171 if (!data) return;
172 switch (event)
173 {
174 case GSOCK_CONNECTION:
175 if(socket->m_server)
176 c = kCFSocketReadCallBack;
177 else
178 c = kCFSocketConnectCallBack;
179 break;
180 case GSOCK_LOST:
181 case GSOCK_INPUT:
182 c = kCFSocketReadCallBack;
183 break;
184 case GSOCK_OUTPUT:
185 c = kCFSocketWriteCallBack;
186 break;
187 default:
188 c = 0;
189 }
190 CFSocketDisableCallBacks(data->socket, c);
191 }
192
193 void GSocketGUIFunctionsTableConcrete::Enable_Events(GSocket *socket)
194 {
195 struct MacGSocketData* data = _GSocket_Get_Mac_Socket(socket);
196 if (!data) return;
197
198 CFRunLoopAddSource(s_mainRunLoop, data->source, kCFRunLoopCommonModes);
199 }
200
201 void GSocketGUIFunctionsTableConcrete::Disable_Events(GSocket *socket)
202 {
203 struct MacGSocketData* data = _GSocket_Get_Mac_Socket(socket);
204 if (!data) return;
205
206 /* CFSocketInvalidate does CFRunLoopRemoveSource anyway */
207 CFRunLoopRemoveSource(s_mainRunLoop, data->source, kCFRunLoopCommonModes);
208 CFSocketInvalidate(data->socket);
209
210 // CFSocketInvalidate has closed the socket so we want to make sure GSocket knows this
211 socket->m_fd = -1 /*INVALID_SOCKET*/;
212 }
213
214 #endif // wxUSE_SOCKETS