src/mac/carbon/gsockosx.cpp

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