Several small improvements:
- consistent error message when something fails (via BTL_ERROR macro) - decrease the number of jumps. - cleanup some parts of the code. This commit was SVN r12719.
Этот коммит содержится в:
родитель
657168a74c
Коммит
658879232b
@ -629,15 +629,9 @@ static void mca_btl_tcp_endpoint_recv_handler(int sd, short flags, void* user)
|
|||||||
btl_endpoint->endpoint_recv_frag = frag;
|
btl_endpoint->endpoint_recv_frag = frag;
|
||||||
} else {
|
} else {
|
||||||
btl_endpoint->endpoint_recv_frag = NULL;
|
btl_endpoint->endpoint_recv_frag = NULL;
|
||||||
switch(frag->hdr.type) {
|
if( MCA_BTL_TCP_HDR_TYPE_SEND == frag->hdr.type ) {
|
||||||
case MCA_BTL_TCP_HDR_TYPE_SEND:
|
|
||||||
{
|
|
||||||
mca_btl_base_recv_reg_t* reg = frag->btl->tcp_reg + frag->hdr.base.tag;
|
mca_btl_base_recv_reg_t* reg = frag->btl->tcp_reg + frag->hdr.base.tag;
|
||||||
reg->cbfunc(&frag->btl->super, frag->hdr.base.tag, &frag->base, reg->cbdata);
|
reg->cbfunc(&frag->btl->super, frag->hdr.base.tag, &frag->base, reg->cbdata);
|
||||||
break;
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
||||||
if( 0 != btl_endpoint->endpoint_cache_length ) {
|
if( 0 != btl_endpoint->endpoint_cache_length ) {
|
||||||
|
@ -107,18 +107,17 @@ bool mca_btl_tcp_frag_send(mca_btl_tcp_frag_t* frag, int sd)
|
|||||||
case EINTR:
|
case EINTR:
|
||||||
continue;
|
continue;
|
||||||
case EWOULDBLOCK:
|
case EWOULDBLOCK:
|
||||||
/* opal_output(0, "mca_btl_tcp_frag_send: EWOULDBLOCK\n"); */
|
|
||||||
return false;
|
return false;
|
||||||
case EFAULT:
|
case EFAULT:
|
||||||
BTL_ERROR(("writev error (%p, %d)\n\t%s(%d)\n",
|
BTL_ERROR(("mca_btl_tcp_frag_send: writev error (%p, %d)\n\t%s(%d)\n",
|
||||||
frag->iov_ptr[0].iov_base, frag->iov_ptr[0].iov_len,
|
frag->iov_ptr[0].iov_base, frag->iov_ptr[0].iov_len,
|
||||||
strerror(opal_socket_errno), frag->iov_cnt));
|
strerror(opal_socket_errno), frag->iov_cnt));
|
||||||
default:
|
|
||||||
{
|
|
||||||
BTL_ERROR(("writev failed with errno=%d", opal_socket_errno));
|
|
||||||
mca_btl_tcp_endpoint_close(frag->endpoint);
|
mca_btl_tcp_endpoint_close(frag->endpoint);
|
||||||
return false;
|
return false;
|
||||||
}
|
default:
|
||||||
|
BTL_ERROR(("mca_btl_tcp_frag_send: writev failed with errno=%d", opal_socket_errno));
|
||||||
|
mca_btl_tcp_endpoint_close(frag->endpoint);
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -141,7 +140,6 @@ bool mca_btl_tcp_frag_send(mca_btl_tcp_frag_t* frag, int sd)
|
|||||||
return (frag->iov_cnt == 0);
|
return (frag->iov_cnt == 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
bool mca_btl_tcp_frag_recv(mca_btl_tcp_frag_t* frag, int sd)
|
bool mca_btl_tcp_frag_recv(mca_btl_tcp_frag_t* frag, int sd)
|
||||||
{
|
{
|
||||||
int cnt;
|
int cnt;
|
||||||
@ -152,12 +150,12 @@ bool mca_btl_tcp_frag_recv(mca_btl_tcp_frag_t* frag, int sd)
|
|||||||
num_vecs = frag->iov_cnt;
|
num_vecs = frag->iov_cnt;
|
||||||
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
||||||
if( 0 != btl_endpoint->endpoint_cache_length ) {
|
if( 0 != btl_endpoint->endpoint_cache_length ) {
|
||||||
size_t length = btl_endpoint->endpoint_cache_length;
|
size_t length;
|
||||||
/* It's strange at the first look but cnt have to be set to the full amount of data available.
|
/* It's strange at the first look but cnt have to be set to the full amount of data
|
||||||
* After going to advance_iov_position we will use cnt to detect if there is still some
|
* available. After going to advance_iov_position we will use cnt to detect if there
|
||||||
* data pending.
|
* is still some data pending.
|
||||||
*/
|
*/
|
||||||
cnt = btl_endpoint->endpoint_cache_length;
|
cnt = length = btl_endpoint->endpoint_cache_length;
|
||||||
for( i = 0; i < frag->iov_cnt; i++ ) {
|
for( i = 0; i < frag->iov_cnt; i++ ) {
|
||||||
if( length > frag->iov_ptr[i].iov_len )
|
if( length > frag->iov_ptr[i].iov_len )
|
||||||
length = frag->iov_ptr[0].iov_len;
|
length = frag->iov_ptr[0].iov_len;
|
||||||
@ -184,48 +182,45 @@ bool mca_btl_tcp_frag_recv(mca_btl_tcp_frag_t* frag, int sd)
|
|||||||
cnt = -1;
|
cnt = -1;
|
||||||
while( cnt < 0 ) {
|
while( cnt < 0 ) {
|
||||||
cnt = readv(sd, frag->iov_ptr, num_vecs);
|
cnt = readv(sd, frag->iov_ptr, num_vecs);
|
||||||
if(cnt < 0) {
|
if( 0 < cnt ) goto advance_iov_position;
|
||||||
|
if( cnt == 0 ) {
|
||||||
|
mca_btl_tcp_endpoint_close(btl_endpoint);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
switch(opal_socket_errno) {
|
switch(opal_socket_errno) {
|
||||||
case EINTR:
|
case EINTR:
|
||||||
continue;
|
continue;
|
||||||
case EWOULDBLOCK:
|
case EWOULDBLOCK:
|
||||||
return false;
|
return false;
|
||||||
case EFAULT:
|
case EFAULT:
|
||||||
opal_output( 0, "mca_btl_tcp_frag_send: writev error (%p, %d)\n\t%s(%d)\n",
|
BTL_ERROR(("mca_btl_tcp_frag_recv: readv error (%p, %d)\n\t%s(%d)\n",
|
||||||
frag->iov_ptr[0].iov_base, frag->iov_ptr[0].iov_len,
|
frag->iov_ptr[0].iov_base, frag->iov_ptr[0].iov_len,
|
||||||
strerror(opal_socket_errno), frag->iov_cnt );
|
strerror(opal_socket_errno), frag->iov_cnt));
|
||||||
|
mca_btl_tcp_endpoint_close(btl_endpoint);
|
||||||
|
return false;
|
||||||
default:
|
default:
|
||||||
opal_output(0, "mca_btl_tcp_frag_send: writev failed with errno=%d",
|
BTL_ERROR(("mca_btl_tcp_frag_recv: readv failed with errno=%d", opal_socket_errno));
|
||||||
opal_socket_errno);
|
|
||||||
mca_btl_tcp_endpoint_close(btl_endpoint);
|
mca_btl_tcp_endpoint_close(btl_endpoint);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
|
||||||
if( cnt == 0 ) {
|
|
||||||
mca_btl_tcp_endpoint_close(btl_endpoint);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
goto advance_iov_position;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
advance_iov_position:
|
advance_iov_position:
|
||||||
/* if the write didn't complete - update the iovec state */
|
/* if the write didn't complete - update the iovec state */
|
||||||
num_vecs = frag->iov_cnt;
|
num_vecs = frag->iov_cnt;
|
||||||
for( i = 0; i < num_vecs; i++ ) {
|
for( i = 0; i < num_vecs; i++ ) {
|
||||||
if( cnt >= (int)frag->iov_ptr->iov_len ) {
|
if( cnt < (int)frag->iov_ptr->iov_len ) {
|
||||||
cnt -= frag->iov_ptr->iov_len;
|
|
||||||
frag->iov_idx++;
|
|
||||||
frag->iov_ptr++;
|
|
||||||
frag->iov_cnt--;
|
|
||||||
} else {
|
|
||||||
frag->iov_ptr->iov_base = (ompi_iov_base_ptr_t)
|
frag->iov_ptr->iov_base = (ompi_iov_base_ptr_t)
|
||||||
(((unsigned char*)frag->iov_ptr->iov_base) + cnt);
|
(((unsigned char*)frag->iov_ptr->iov_base) + cnt);
|
||||||
frag->iov_ptr->iov_len -= cnt;
|
frag->iov_ptr->iov_len -= cnt;
|
||||||
cnt = 0;
|
cnt = 0;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
cnt -= frag->iov_ptr->iov_len;
|
||||||
|
frag->iov_idx++;
|
||||||
|
frag->iov_ptr++;
|
||||||
|
frag->iov_cnt--;
|
||||||
}
|
}
|
||||||
|
|
||||||
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
#if MCA_BTL_TCP_ENDPOINT_CACHE
|
||||||
btl_endpoint->endpoint_cache_length = cnt;
|
btl_endpoint->endpoint_cache_length = cnt;
|
||||||
#endif /* MCA_BTL_TCP_ENDPOINT_CACHE */
|
#endif /* MCA_BTL_TCP_ENDPOINT_CACHE */
|
||||||
|
@ -102,8 +102,7 @@ OBJ_CLASS_DECLARATION(mca_btl_tcp_frag_user_t);
|
|||||||
|
|
||||||
#define MCA_BTL_TCP_FRAG_RETURN(frag) \
|
#define MCA_BTL_TCP_FRAG_RETURN(frag) \
|
||||||
{ \
|
{ \
|
||||||
OMPI_FREE_LIST_RETURN(frag->my_list, \
|
OMPI_FREE_LIST_RETURN(frag->my_list, (ompi_free_list_item_t*)(frag)); \
|
||||||
(ompi_free_list_item_t*)(frag)); \
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#define MCA_BTL_TCP_FRAG_INIT_DST(frag,ep) \
|
#define MCA_BTL_TCP_FRAG_INIT_DST(frag,ep) \
|
||||||
|
Загрузка…
x
Ссылка в новой задаче
Block a user