openmpi/ompi/mca/ptl/tcp/ptl_tcp_component.c

/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */
#include "ompi_config.h"
#include "include/ompi_socket_errno.h"
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <string.h>
#include <fcntl.h>
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#ifdef HAVE_SYS_SOCKET_H
#include <sys/socket.h>
#endif
#ifdef HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#ifdef HAVE_ARPA_INET_H
#include <arpa/inet.h>
#endif

#include "ompi/include/constants.h"
#include "opal/event/event.h"
#include "opal/util/if.h"
#include "opal/util/argv.h"
#include "opal/util/output.h"
#include "ompi/mca/pml/pml.h"
#include "ompi/mca/ptl/ptl.h"
#include "ompi/mca/pml/base/pml_base_sendreq.h"
#include "opal/mca/base/mca_base_param.h"
#include "ompi/mca/pml/base/pml_base_module_exchange.h"
#include "ptl_tcp.h"
#include "ptl_tcp_addr.h"
#include "ptl_tcp_proc.h"
#include "ptl_tcp_recvfrag.h"
#include "ptl_tcp_sendfrag.h"
#include "ptl_tcp_sendreq.h"

#define IMPORTANT_WINDOWS_COMMENT() \
            /* In windows, many of the socket functions return an EWOULDBLOCK instead of \
               things like EAGAIN, EINPROGRESS, etc. It has been verified that this will \
               not conflict with other error codes that are returned by these functions \
               under UNIX/Linux environments */

/*
 * Data structure for accepting connections.
 */

struct mca_ptl_tcp_event_t {
    opal_list_item_t item;
    opal_event_t event;
};
typedef struct mca_ptl_tcp_event_t mca_ptl_tcp_event_t;

static void mca_ptl_tcp_event_construct(mca_ptl_tcp_event_t* event)
{
    OPAL_THREAD_LOCK(&mca_ptl_tcp_component.tcp_lock);
    opal_list_append(&mca_ptl_tcp_component.tcp_events, &event->item);
    OPAL_THREAD_UNLOCK(&mca_ptl_tcp_component.tcp_lock);
}

static void mca_ptl_tcp_event_destruct(mca_ptl_tcp_event_t* event)
{
    OPAL_THREAD_LOCK(&mca_ptl_tcp_component.tcp_lock);
    opal_list_remove_item(&mca_ptl_tcp_component.tcp_events, &event->item);
    OPAL_THREAD_UNLOCK(&mca_ptl_tcp_component.tcp_lock);
}

OBJ_CLASS_INSTANCE(
    mca_ptl_tcp_event_t,
    opal_list_item_t,
    mca_ptl_tcp_event_construct,
    mca_ptl_tcp_event_destruct);


/*
 * The PTL TCP component
 */

mca_ptl_tcp_component_t mca_ptl_tcp_component = {
    {
      /* First, the mca_base_module_t struct containing meta
         information about the module itself */
      {
        /* Indicate that we are a pml v1.0.0 module (which also
           implies a specific MCA version) */

        MCA_PTL_BASE_VERSION_1_0_0,

        "tcp", /* MCA module name */
        OMPI_MAJOR_VERSION,  /* MCA module major version */
        OMPI_MINOR_VERSION,  /* MCA module minor version */
        OMPI_RELEASE_VERSION,  /* MCA module release version */
        mca_ptl_tcp_component_open,  /* module open */
        mca_ptl_tcp_component_close  /* module close */
      },
      
      /* Next the MCA v1.0.0 module meta data */
      
      {
        /* Whether the module is checkpointable or not */
        
        false
      },
      
      mca_ptl_tcp_component_init,  
      mca_ptl_tcp_component_control,
      NULL /*mca_ptl_tcp_component_progress*/,
    }
};

/*
 * functions for receiving event callbacks
 */

static void mca_ptl_tcp_component_recv_handler(int, short, void*);


/*
 * utility routines for parameter registration
 */

static inline char* mca_ptl_tcp_param_register_string(
    const char* param_name, 
    const char* default_value)
{
    char *param_value;
    int id = mca_base_param_register_string("ptl","tcp",param_name,NULL,default_value);
    mca_base_param_lookup_string(id, &param_value);
    return param_value;
}
                                                                                                                            
static inline int mca_ptl_tcp_param_register_int(
    const char* param_name, 
    int default_value)
{
    int id = mca_base_param_register_int("ptl","tcp",param_name,NULL,default_value);
    int param_value = default_value;
    mca_base_param_lookup_int(id,&param_value);
    return param_value;
}

/*
 *  Called by MCA framework to open the module, registers
 *  module parameters.
 */

int mca_ptl_tcp_component_open(void)
{
#ifdef WIN32
    WSADATA win_sock_data;
    if (WSAStartup(MAKEWORD(2,2), &win_sock_data) != 0) {
        opal_output (0, "mca_ptl_tcp_component_init: failed to initialise windows sockets:%d\n", WSAGetLastError());
        return OMPI_ERROR;
    }
#endif

    /* initialize state */
    mca_ptl_tcp_component.tcp_listen_sd = -1;
    mca_ptl_tcp_component.tcp_ptl_modules = NULL;
    mca_ptl_tcp_component.tcp_num_ptl_modules = 0;

    /* initialize objects */
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_lock, opal_mutex_t);
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_procs, opal_hash_table_t);
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_pending_acks, opal_list_t);
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_events, opal_list_t);
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_send_frags, ompi_free_list_t);
    OBJ_CONSTRUCT(&mca_ptl_tcp_component.tcp_recv_frags, ompi_free_list_t);
    opal_hash_table_init(&mca_ptl_tcp_component.tcp_procs, 256);

    /* register TCP module parameters */
    mca_ptl_tcp_component.tcp_if_include =
        mca_ptl_tcp_param_register_string("if_include", "");
    mca_ptl_tcp_component.tcp_if_exclude =
        mca_ptl_tcp_param_register_string("if_exclude", "lo");
    mca_ptl_tcp_component.tcp_free_list_num =
        mca_ptl_tcp_param_register_int("free_list_num", 256);
    mca_ptl_tcp_component.tcp_free_list_max =
        mca_ptl_tcp_param_register_int("free_list_max", -1);
    mca_ptl_tcp_component.tcp_free_list_inc =
        mca_ptl_tcp_param_register_int("free_list_inc", 256);
    mca_ptl_tcp_component.tcp_sndbuf =
        mca_ptl_tcp_param_register_int("sndbuf", 128*1024);
    mca_ptl_tcp_component.tcp_rcvbuf =
        mca_ptl_tcp_param_register_int("rcvbuf", 128*1024);
    mca_ptl_tcp_module.super.ptl_exclusivity =
        mca_ptl_tcp_param_register_int("exclusivity", 0);
    mca_ptl_tcp_module.super.ptl_first_frag_size =
        mca_ptl_tcp_param_register_int("first_frag_size", 64*1024);
    mca_ptl_tcp_module.super.ptl_min_frag_size = 
        mca_ptl_tcp_param_register_int("min_frag_size", 64*1024);
    mca_ptl_tcp_module.super.ptl_max_frag_size =
        mca_ptl_tcp_param_register_int("max_frag_size", -1);
    /* the tcp allocator will never allocate buffers with more than this size */
    mca_ptl_tcp_component.tcp_frag_size =
        mca_ptl_tcp_param_register_int("frag_size", 64*1024);
    /* adapt the first fragment size to fit with the allowed fragment size */
    if( (mca_ptl_tcp_component.tcp_frag_size != 0) &&
        (mca_ptl_tcp_module.super.ptl_first_frag_size > mca_ptl_tcp_component.tcp_frag_size) ) {
        mca_ptl_tcp_module.super.ptl_first_frag_size = mca_ptl_tcp_component.tcp_frag_size;
    }
    return OMPI_SUCCESS;
}

/*
 * module cleanup - sanity checking of queue lengths
 */

int mca_ptl_tcp_component_close(void)
{
    opal_list_item_t* item;
#ifdef WIN32
    WSACleanup();
#endif
#if OMPI_ENABLE_DEBUG
    if (mca_ptl_tcp_component.tcp_send_frags.fl_num_allocated != 
        mca_ptl_tcp_component.tcp_send_frags.super.opal_list_length) {
        opal_output(0, "tcp send frags: %d allocated %d returned\n",
            mca_ptl_tcp_component.tcp_send_frags.fl_num_allocated, 
            mca_ptl_tcp_component.tcp_send_frags.super.opal_list_length);
    }
    if (mca_ptl_tcp_component.tcp_recv_frags.fl_num_allocated != 
        mca_ptl_tcp_component.tcp_recv_frags.super.opal_list_length) {
        opal_output(0, "tcp recv frags: %d allocated %d returned\n",
            mca_ptl_tcp_component.tcp_recv_frags.fl_num_allocated, 
            mca_ptl_tcp_component.tcp_recv_frags.super.opal_list_length);
    }
#endif

    if(NULL != mca_ptl_tcp_component.tcp_if_include)
        free(mca_ptl_tcp_component.tcp_if_include);
    if(NULL != mca_ptl_tcp_component.tcp_if_exclude)
       free(mca_ptl_tcp_component.tcp_if_exclude);
    if (NULL != mca_ptl_tcp_component.tcp_ptl_modules)
        free(mca_ptl_tcp_component.tcp_ptl_modules);
 
    if (mca_ptl_tcp_component.tcp_listen_sd >= 0) {
        opal_event_del(&mca_ptl_tcp_component.tcp_recv_event);
        close(mca_ptl_tcp_component.tcp_listen_sd);
        mca_ptl_tcp_component.tcp_listen_sd = -1;
    }

    /* cleanup any pending events */
    OPAL_THREAD_LOCK(&mca_ptl_tcp_component.tcp_lock);
    for(item =  opal_list_remove_first(&mca_ptl_tcp_component.tcp_events);
        item != NULL; 
        item =  opal_list_remove_first(&mca_ptl_tcp_component.tcp_events)) {
        mca_ptl_tcp_event_t* event = (mca_ptl_tcp_event_t*)item;
        opal_event_del(&event->event);
        OBJ_RELEASE(event);
    }
    OPAL_THREAD_UNLOCK(&mca_ptl_tcp_component.tcp_lock);

    /* release resources */
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_procs);
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_pending_acks);
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_events);
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_send_frags);
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_recv_frags);
    OBJ_DESTRUCT(&mca_ptl_tcp_component.tcp_lock);
    return OMPI_SUCCESS;
}


/*
 *  Create a ptl instance and add to modules list.
 */

static int mca_ptl_tcp_create(int if_index, const char* if_name)
{
    mca_ptl_tcp_module_t* ptl = (mca_ptl_tcp_module_t *)malloc(sizeof(mca_ptl_tcp_module_t));
    char param[256];
    if(NULL == ptl)
        return OMPI_ERR_OUT_OF_RESOURCE;
    memcpy(ptl, &mca_ptl_tcp_module, sizeof(mca_ptl_tcp_module));
    OBJ_CONSTRUCT(&ptl->ptl_peers, opal_list_t);
    mca_ptl_tcp_component.tcp_ptl_modules[mca_ptl_tcp_component.tcp_num_ptl_modules++] = ptl;

    /* initialize the ptl */
    ptl->ptl_ifindex = if_index;
#if MCA_PTL_TCP_STATISTICS
    ptl->ptl_bytes_recv = 0;
    ptl->ptl_bytes_sent = 0;
    ptl->ptl_send_handler = 0;
#endif
    opal_ifindextoaddr(if_index, (struct sockaddr*)&ptl->ptl_ifaddr, sizeof(ptl->ptl_ifaddr));
    opal_ifindextomask(if_index, (struct sockaddr*)&ptl->ptl_ifmask, sizeof(ptl->ptl_ifmask));

    /* allow user to specify interface bandwidth */
    sprintf(param, "bandwidth_%s", if_name);
    ptl->super.ptl_bandwidth = mca_ptl_tcp_param_register_int(param, 0);

    /* allow user to override/specify latency ranking */
    sprintf(param, "latency_%s", if_name);
    ptl->super.ptl_latency = mca_ptl_tcp_param_register_int(param, 0);

#if OMPI_ENABLE_DEBUG && 0
    opal_output(0,"interface: %s bandwidth %d latency %d\n", 
        if_name, ptl->super.ptl_bandwidth, ptl->super.ptl_latency);
#endif
    return OMPI_SUCCESS;
}

/*
 * Create a TCP PTL instance for either:
 * (1) all interfaces specified by the user
 * (2) all available interfaces 
 * (3) all available interfaces except for those excluded by the user
 */

static int mca_ptl_tcp_component_create_instances(void)
{
    int if_count = opal_ifcount();
    int if_index;
    char **include;
    char **exclude;
    char **argv;

    if(if_count <= 0)
        return OMPI_ERROR;

    /* allocate memory for ptls */
    mca_ptl_tcp_component.tcp_max_ptl_modules = if_count;
    mca_ptl_tcp_component.tcp_ptl_modules = (mca_ptl_tcp_module_t **)malloc(if_count * sizeof(mca_ptl_tcp_module_t*));
    if(NULL == mca_ptl_tcp_component.tcp_ptl_modules)
        return OMPI_ERR_OUT_OF_RESOURCE;

    /* if the user specified an interface list - use these exclusively */
    argv = include = opal_argv_split(mca_ptl_tcp_component.tcp_if_include,',');
    while(argv && *argv) {
        char* if_name = *argv;
        int if_index = opal_ifnametoindex(if_name);
        if(if_index < 0) {
            opal_output(0,"mca_ptl_tcp_component_init: invalid interface \"%s\"", if_name);
        } else {
            mca_ptl_tcp_create(if_index, if_name);
        }
        argv++;
    }
    opal_argv_free(include);
    if(mca_ptl_tcp_component.tcp_num_ptl_modules)
        return OMPI_SUCCESS;

    /* if the interface list was not specified by the user, create 
     * a PTL for each interface that was not excluded.
    */
    exclude = opal_argv_split(mca_ptl_tcp_component.tcp_if_exclude,',');
    for(if_index = opal_ifbegin(); if_index >= 0; if_index = opal_ifnext(if_index)) {
        char if_name[32];
        opal_ifindextoname(if_index, if_name, sizeof(if_name));

        /* check to see if this interface exists in the exclude list */
        if(opal_ifcount() > 1) {
            argv = exclude;
            while(argv && *argv) {
                if(strncmp(*argv,if_name,strlen(*argv)) == 0)
                    break;
                argv++;
            }
            /* if this interface was not found in the excluded list - create a PTL */
            if(argv == 0 || *argv == 0) {
                mca_ptl_tcp_create(if_index, if_name);
            }
        } else {
            mca_ptl_tcp_create(if_index, if_name);
        }
    }
    opal_argv_free(exclude);
    return OMPI_SUCCESS;
}

/*
 * Create a listen socket and bind to all interfaces
 */

static int mca_ptl_tcp_component_create_listen(void)
{
    int flags;
    struct sockaddr_in inaddr; 
    ompi_socklen_t addrlen;

    /* create a listen socket for incoming connections */
    mca_ptl_tcp_component.tcp_listen_sd = socket(AF_INET, SOCK_STREAM, 0);
    if(mca_ptl_tcp_component.tcp_listen_sd < 0) {
        opal_output(0,"mca_ptl_tcp_component_init: socket() failed with errno=%d", ompi_socket_errno);
        return OMPI_ERROR;
    }
    mca_ptl_tcp_set_socket_options(mca_ptl_tcp_component.tcp_listen_sd);
                                                                                                      
    /* bind to all addresses and dynamically assigned port */
    memset(&inaddr, 0, sizeof(inaddr));
    inaddr.sin_family = AF_INET;
    inaddr.sin_addr.s_addr = INADDR_ANY;
    inaddr.sin_port = 0;
                                                                                                      
    if(bind(mca_ptl_tcp_component.tcp_listen_sd, (struct sockaddr*)&inaddr, sizeof(inaddr)) < 0) {
        opal_output(0,"mca_ptl_tcp_component_init: bind() failed with errno=%d", ompi_socket_errno);
        return OMPI_ERROR;
    }
                                                                                                      
    /* resolve system assignend port */
    addrlen = sizeof(struct sockaddr_in);
    if(getsockname(mca_ptl_tcp_component.tcp_listen_sd, (struct sockaddr*)&inaddr, &addrlen) < 0) {
        opal_output(0, "mca_ptl_tcp_component_init: getsockname() failed with errno=%d", ompi_socket_errno);
        return OMPI_ERROR;
    }
    mca_ptl_tcp_component.tcp_listen_port = inaddr.sin_port;

    /* setup listen backlog to maximum allowed by kernel */
    if(listen(mca_ptl_tcp_component.tcp_listen_sd, SOMAXCONN) < 0) {
        opal_output(0, "mca_ptl_tcp_component_init: listen() failed with errno=%d", ompi_socket_errno);
        return OMPI_ERROR;
    }

    /* set socket up to be non-blocking, otherwise accept could block */
    if((flags = fcntl(mca_ptl_tcp_component.tcp_listen_sd, F_GETFL, 0)) < 0) {
        opal_output(0, "mca_ptl_tcp_component_init: fcntl(F_GETFL) failed with errno=%d", ompi_socket_errno);
        return OMPI_ERROR;
    } else {
        flags |= O_NONBLOCK;
        if(fcntl(mca_ptl_tcp_component.tcp_listen_sd, F_SETFL, flags) < 0) {
            opal_output(0, "mca_ptl_tcp_component_init: fcntl(F_SETFL) failed with errno=%d", ompi_socket_errno);
            return OMPI_ERROR;
        }
    }

    /* register listen port */
    opal_event_set(
        &mca_ptl_tcp_component.tcp_recv_event,
        mca_ptl_tcp_component.tcp_listen_sd, 
        OPAL_EV_READ|OPAL_EV_PERSIST, 
        mca_ptl_tcp_component_recv_handler, 
        0);
    return OMPI_SUCCESS;
}

/*
 *  Register TCP module addressing information. The MCA framework
 *  will make this available to all peers. 
 */

static int mca_ptl_tcp_component_exchange(void)
{
     int rc=0;
     size_t i=0;
     size_t size = mca_ptl_tcp_component.tcp_num_ptl_modules * sizeof(mca_ptl_tcp_addr_t);
     if(mca_ptl_tcp_component.tcp_num_ptl_modules != 0) {
         mca_ptl_tcp_addr_t *addrs = (mca_ptl_tcp_addr_t *)malloc(size);
         for(i=0; i<mca_ptl_tcp_component.tcp_num_ptl_modules; i++) {
             mca_ptl_tcp_module_t* ptl = mca_ptl_tcp_component.tcp_ptl_modules[i];
             addrs[i].addr_inet    = ptl->ptl_ifaddr.sin_addr;
             addrs[i].addr_port    = mca_ptl_tcp_component.tcp_listen_port;
             addrs[i].addr_inuse   = 0;
         }
         rc =  mca_pml_base_modex_send(&mca_ptl_tcp_component.super.ptlm_version, addrs, size);
         free(addrs);
     }
     return rc;
}

/*
 *  TCP module initialization:
 *  (1) read interface list from kernel and compare against module parameters
 *      then create a PTL instance for selected interfaces
 *  (2) setup TCP listen socket for incoming connection attempts
 *  (3) register PTL parameters with the MCA
 */
mca_ptl_base_module_t** mca_ptl_tcp_component_init(int *num_ptl_modules, 
                                                   bool enable_progress_threads,
                                                   bool enable_mpi_threads)
{
    mca_ptl_base_module_t **ptls;
    *num_ptl_modules = 0;

    ompi_free_list_init(&mca_ptl_tcp_component.tcp_send_frags, 
        sizeof(mca_ptl_tcp_send_frag_t),
        OBJ_CLASS(mca_ptl_tcp_send_frag_t),
        mca_ptl_tcp_component.tcp_free_list_num,
        mca_ptl_tcp_component.tcp_free_list_max,
        mca_ptl_tcp_component.tcp_free_list_inc,
        NULL); /* use default allocator */

    ompi_free_list_init(&mca_ptl_tcp_component.tcp_recv_frags, 
        sizeof(mca_ptl_tcp_recv_frag_t),
        OBJ_CLASS(mca_ptl_tcp_recv_frag_t),
        mca_ptl_tcp_component.tcp_free_list_num,
        mca_ptl_tcp_component.tcp_free_list_max,
        mca_ptl_tcp_component.tcp_free_list_inc,
        NULL); /* use default allocator */

    /* create a PTL TCP module for selected interfaces */
    if(mca_ptl_tcp_component_create_instances() != OMPI_SUCCESS)
        return 0;

    /* create a TCP listen socket for incoming connection attempts */
    if(mca_ptl_tcp_component_create_listen() != OMPI_SUCCESS)
        return 0;

    /* publish TCP parameters with the MCA framework */
    if(mca_ptl_tcp_component_exchange() != OMPI_SUCCESS)
        return 0;

    ptls = (mca_ptl_base_module_t **)malloc(mca_ptl_tcp_component.tcp_num_ptl_modules * 
                  sizeof(mca_ptl_base_module_t*));
    if(NULL == ptls)
        return NULL;

    memcpy(ptls, mca_ptl_tcp_component.tcp_ptl_modules, mca_ptl_tcp_component.tcp_num_ptl_modules*sizeof(mca_ptl_tcp_module_t*));
    *num_ptl_modules = mca_ptl_tcp_component.tcp_num_ptl_modules;

    return ptls;
}

/*
 *  TCP module control
 */

int mca_ptl_tcp_component_control(int param, void* value, size_t size)
{
    switch(param) {
        case MCA_PTL_ENABLE:
            if(*(int*)value) {
                opal_event_add(&mca_ptl_tcp_component.tcp_recv_event, 0);
                if(opal_hash_table_get_size(&mca_ptl_tcp_component.tcp_procs) > 0) {
                    opal_progress_events(OPAL_EVLOOP_NONBLOCK);
                }
            } else {
                opal_event_del(&mca_ptl_tcp_component.tcp_recv_event);
            }
            break;
        default:
            break;
    }
    return OMPI_SUCCESS;
}


/*
 *  TCP module progress.
 */

int mca_ptl_tcp_component_progress(mca_ptl_tstamp_t tstamp)
{
    return OMPI_SUCCESS;
}


/*
 *  Called by mca_ptl_tcp_component_recv() when the TCP listen
 *  socket has pending connection requests. Accept incoming
 *  requests and queue for completion of the connection handshake.
*/


static void mca_ptl_tcp_component_accept(void)
{
    while(true) {
        ompi_socklen_t addrlen = sizeof(struct sockaddr_in);
        struct sockaddr_in addr;
        mca_ptl_tcp_event_t *event;
        int sd = accept(mca_ptl_tcp_component.tcp_listen_sd, (struct sockaddr*)&addr, &addrlen);
        if(sd < 0) {
            IMPORTANT_WINDOWS_COMMENT();
            if(ompi_socket_errno == EINTR)
                continue;
            if(ompi_socket_errno != EAGAIN || ompi_socket_errno != EWOULDBLOCK)
                opal_output(0, "mca_ptl_tcp_component_accept: accept() failed with errno %d.", ompi_socket_errno);
            return;
        }
        mca_ptl_tcp_set_socket_options(sd);

        /* wait for receipt of peers process identifier to complete this connection */
         
        event = OBJ_NEW(mca_ptl_tcp_event_t);
        opal_event_set(&event->event, sd, OPAL_EV_READ, mca_ptl_tcp_component_recv_handler, event);
        opal_event_add(&event->event, 0);
    }
}


/*
 * Event callback when there is data available on the registered 
 * socket to recv.
 */
static void mca_ptl_tcp_component_recv_handler(int sd, short flags, void* user)
{
    orte_process_name_t guid;
    struct sockaddr_in addr;
    int retval;
    mca_ptl_tcp_proc_t* ptl_proc;
    ompi_socklen_t addr_len = sizeof(addr);
    mca_ptl_tcp_event_t *event = (mca_ptl_tcp_event_t *)user;

    /* accept new connections on the listen socket */
    if(mca_ptl_tcp_component.tcp_listen_sd == sd) {
        mca_ptl_tcp_component_accept();
        return;
    }
    OBJ_RELEASE(event);

    /* recv the process identifier */
    retval = recv(sd, (char *)&guid, sizeof(guid), 0);
    if(retval != sizeof(guid)) {
        close(sd);
        return;
    }

    /* now set socket up to be non-blocking */
    if((flags = fcntl(sd, F_GETFL, 0)) < 0) {
        opal_output(0, "mca_ptl_tcp_component_recv_handler: fcntl(F_GETFL) failed with errno=%d", ompi_socket_errno);
    } else {
        flags |= O_NONBLOCK;
        if(fcntl(sd, F_SETFL, flags) < 0) {
            opal_output(0, "mca_ptl_tcp_component_recv_handler: fcntl(F_SETFL) failed with errno=%d", ompi_socket_errno);
        }
    }
   
    /* lookup the corresponding process */
    ptl_proc = mca_ptl_tcp_proc_lookup(&guid);
    if(NULL == ptl_proc) {
        opal_output(0, "mca_ptl_tcp_component_recv_handler: unable to locate process");
        close(sd);
        return;
    }

    /* lookup peer address */
    if(getpeername(sd, (struct sockaddr*)&addr, &addr_len) != 0) {
        opal_output(0, "mca_ptl_tcp_component_recv_handler: getpeername() failed with errno=%d", ompi_socket_errno);
        close(sd);
        return;
    }

    /* are there any existing peer instances will to accept this connection */
    if(mca_ptl_tcp_proc_accept(ptl_proc, &addr, sd) == false) {
        close(sd);
        return;
    }
}