/**

  * $Id$

  *

  * OS X networking support functions. Currently this is just for activity callbacks.

  *

  * Copyright (c) 2009 Nathan Keynes.

  *

  * This program 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 2 of the License, or

  * (at your option) any later version.

  *

  * This program 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.

  */

 #include <CoreFoundation/CoreFoundation.h>

 #include "ioutil.h"

 struct io_osx_cbinfo {

     int fd;

     io_callback_t callback;

     void * cbdata;

     void (*cbdealloc)(void *);

     void *fdRef;

     CFRunLoopSourceRef sourceRef;

     struct io_osx_cbinfo *next;

 };

 static struct io_osx_cbinfo *cbinfo_list = NULL;

 void io_unregister_callback( struct io_osx_cbinfo *cbinfo )

 {

     CFRunLoopRemoveSource( CFRunLoopGetCurrent(), cbinfo->sourceRef, kCFRunLoopCommonModes );

     CFRelease(cbinfo->sourceRef);

     cbinfo->sourceRef = NULL;

     CFRelease(cbinfo->fdRef); /* Note this implicitly releases the cbinfo itself as well */

 }

 static void io_osx_net_callback( CFSocketRef s, CFSocketCallBackType type, CFDataRef address, const void *unused, void *data )

 {

     struct io_osx_cbinfo *cbinfo = (struct io_osx_cbinfo *)data;

     if(!cbinfo->callback( CFSocketGetNative(s), cbinfo->cbdata) ) {

         io_unregister_callback(cbinfo);

     }

 }

 static void io_osx_fd_callback( CFFileDescriptorRef f, CFOptionFlags type, void *data )

 {

     struct io_osx_cbinfo *cbinfo = (struct io_osx_cbinfo *)data;

     if(!cbinfo->callback( CFFileDescriptorGetNativeDescriptor(f), cbinfo->cbdata) ) {

         io_unregister_callback(cbinfo);

     }

 }

 static void io_osx_release( const void *data )

 {

     struct io_osx_cbinfo *cbinfo = (struct io_osx_cbinfo *)data;

     if( cbinfo->cbdealloc != NULL ) {

         cbinfo->cbdealloc(cbinfo->cbdata);

     }

     free( cbinfo );

 }

 /**

  * Register a TCP server socket listener on an already open (and listening) 

  * socket. The socket must not have been previously registered.

  * @return TRUE on success, FALSE on failure.

  * 

  * Defined in netutil.h

  */ 

 io_listener_t io_register_tcp_listener( int fd, io_callback_t callback, void *data, void (*dealloc)(void*) )

 {

     CFSocketContext socketContext;

     struct io_osx_cbinfo *cbinfo = malloc( sizeof(struct io_osx_cbinfo) );

     assert(cbinfo != NULL);

     cbinfo->callback = callback;

     cbinfo->cbdata = data;

     cbinfo->cbdealloc = dealloc;

     socketContext.version = 0;

     socketContext.info = cbinfo;

     socketContext.retain = NULL;

     socketContext.release = io_osx_release;

     socketContext.copyDescription = NULL;

     CFSocketRef ref = CFSocketCreateWithNative( kCFAllocatorDefault, fd, kCFSocketReadCallBack,

             io_osx_net_callback, &socketContext );

     cbinfo->fdRef = ref;

     cbinfo->sourceRef = CFSocketCreateRunLoopSource( kCFAllocatorDefault, ref, 0 );

     CFRunLoopAddSource( CFRunLoopGetCurrent(), cbinfo->sourceRef, kCFRunLoopCommonModes );

     return cbinfo;

 }

 /**

  * Register a file descriptor listener on an already open (and listening)

  * file descriptor. The file descriptor must not have been previously registered.

  * @return TRUE on success, FALSE on failure.

  *

  */

 io_listener_t io_register_listener( int fd, io_callback_t callback, void *data, void (*dealloc)(void *) )

 {

     CFFileDescriptorContext fdContext;

     struct io_osx_cbinfo *cbinfo = malloc( sizeof(struct io_osx_cbinfo) );

     assert(cbinfo != NULL);

     cbinfo->callback = callback;

     cbinfo->cbdata = data;

     cbinfo->cbdealloc = dealloc;

     fdContext.version = 0;

     fdContext.retain = NULL;

     fdContext.info = cbinfo;

     fdContext.release = io_osx_release;

     fdContext.copyDescription = NULL;

     CFFileDescriptorRef ref = CFFileDescriptorCreate( kCFAllocatorDefault, fd, FALSE,

             io_osx_fd_callback, &fdContext);

     cbinfo->fdRef = ref;

     cbinfo->sourceRef = CFFileDescriptorCreateRunLoopSource( kCFAllocatorDefault, ref, 0 );

     CFRunLoopAddSource( CFRunLoopGetCurrent(), cbinfo->sourceRef, kCFRunLoopCommonModes );

     return cbinfo;

 }

 void io_unregister_listener( io_listener_t data )

 {

     struct io_osx_cbinfo *cbinfo = (struct io_osx_cbinfo *)data;

     io_unregister_callback(cbinfo);

 }