236 строки
8.7 KiB
C++
236 строки
8.7 KiB
C++
/*---------------------------------------------------------------
|
|
* Copyright (c) 1999,2000,2001,2002,2003
|
|
* The Board of Trustees of the University of Illinois
|
|
* All Rights Reserved.
|
|
*---------------------------------------------------------------
|
|
* Permission is hereby granted, free of charge, to any person
|
|
* obtaining a copy of this software (Iperf) and associated
|
|
* documentation files (the "Software"), to deal in the Software
|
|
* without restriction, including without limitation the
|
|
* rights to use, copy, modify, merge, publish, distribute,
|
|
* sublicense, and/or sell copies of the Software, and to permit
|
|
* persons to whom the Software is furnished to do
|
|
* so, subject to the following conditions:
|
|
*
|
|
*
|
|
* Redistributions of source code must retain the above
|
|
* copyright notice, this list of conditions and
|
|
* the following disclaimers.
|
|
*
|
|
*
|
|
* Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimers in the documentation and/or other materials
|
|
* provided with the distribution.
|
|
*
|
|
*
|
|
* Neither the names of the University of Illinois, NCSA,
|
|
* nor the names of its contributors may be used to endorse
|
|
* or promote products derived from this Software without
|
|
* specific prior written permission.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
|
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE CONTIBUTORS OR COPYRIGHT
|
|
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
* ________________________________________________________________
|
|
* National Laboratory for Applied Network Research
|
|
* National Center for Supercomputing Applications
|
|
* University of Illinois at Urbana-Champaign
|
|
* http://www.ncsa.uiuc.edu
|
|
* ________________________________________________________________
|
|
*
|
|
* Server.cpp
|
|
* by Mark Gates <mgates@nlanr.net>
|
|
* Ajay Tirumala (tirumala@ncsa.uiuc.edu>.
|
|
* -------------------------------------------------------------------
|
|
* A server thread is initiated for each connection accept() returns.
|
|
* Handles sending and receiving data, and then closes socket.
|
|
* Changes to this version : The server can be run as a daemon
|
|
* ------------------------------------------------------------------- */
|
|
|
|
#define HEADERS()
|
|
|
|
#include "headers.h"
|
|
#include "Server.hpp"
|
|
#include "List.h"
|
|
#include "Extractor.h"
|
|
#include "Reporter.h"
|
|
#include "Locale.h"
|
|
|
|
/* -------------------------------------------------------------------
|
|
* Stores connected socket and socket info.
|
|
* ------------------------------------------------------------------- */
|
|
|
|
Server::Server( thread_Settings *inSettings ) {
|
|
mSettings = inSettings;
|
|
mBuf = NULL;
|
|
|
|
// initialize buffer
|
|
mBuf = new char[ mSettings->mBufLen ];
|
|
FAIL_errno( mBuf == NULL, "No memory for buffer\n", mSettings );
|
|
}
|
|
|
|
/* -------------------------------------------------------------------
|
|
* Destructor close socket.
|
|
* ------------------------------------------------------------------- */
|
|
|
|
Server::~Server() {
|
|
if ( mSettings->mSock != INVALID_SOCKET ) {
|
|
int rc = close( mSettings->mSock );
|
|
WARN_errno( rc == SOCKET_ERROR, "close" );
|
|
mSettings->mSock = INVALID_SOCKET;
|
|
}
|
|
DELETE_ARRAY( mBuf );
|
|
}
|
|
|
|
void Server::Sig_Int( int inSigno ) {
|
|
}
|
|
|
|
/* -------------------------------------------------------------------
|
|
* Receieve data from the (connected) TCP/UDP socket.
|
|
* Sends termination flag several times at the end.
|
|
* Does not close the socket.
|
|
* ------------------------------------------------------------------- */
|
|
void Server::Run( void ) {
|
|
long currLen;
|
|
max_size_t totLen = 0;
|
|
struct UDP_datagram* mBuf_UDP = (struct UDP_datagram*) mBuf;
|
|
|
|
ReportStruct *reportstruct = NULL;
|
|
|
|
reportstruct = new ReportStruct;
|
|
if ( reportstruct != NULL ) {
|
|
reportstruct->packetID = 0;
|
|
mSettings->reporthdr = InitReport( mSettings );
|
|
do {
|
|
// perform read
|
|
currLen = recv( mSettings->mSock, mBuf, mSettings->mBufLen, 0 );
|
|
|
|
if ( isUDP( mSettings ) ) {
|
|
// read the datagram ID and sentTime out of the buffer
|
|
reportstruct->packetID = ntohl( mBuf_UDP->id );
|
|
reportstruct->sentTime.tv_sec = ntohl( mBuf_UDP->tv_sec );
|
|
reportstruct->sentTime.tv_usec = ntohl( mBuf_UDP->tv_usec );
|
|
reportstruct->packetLen = currLen;
|
|
gettimeofday( &(reportstruct->packetTime), NULL );
|
|
} else {
|
|
totLen += currLen;
|
|
}
|
|
|
|
// terminate when datagram begins with negative index
|
|
// the datagram ID should be correct, just negated
|
|
if ( reportstruct->packetID < 0 ) {
|
|
reportstruct->packetID = -reportstruct->packetID;
|
|
currLen = -1;
|
|
}
|
|
if ( isUDP (mSettings))
|
|
ReportPacket( mSettings->reporthdr, reportstruct );
|
|
} while ( currLen > 0 );
|
|
|
|
// stop timing
|
|
gettimeofday( &(reportstruct->packetTime), NULL );
|
|
if ( !isUDP (mSettings)) {
|
|
reportstruct->packetLen = totLen;
|
|
ReportPacket( mSettings->reporthdr, reportstruct );
|
|
}
|
|
CloseReport( mSettings->reporthdr, reportstruct );
|
|
|
|
// send a acknowledgement back only if we're NOT receiving multicast
|
|
if ( isUDP( mSettings ) && !isMulticast( mSettings ) ) {
|
|
// send back an acknowledgement of the terminating datagram
|
|
write_UDP_AckFIN( );
|
|
}
|
|
} else {
|
|
FAIL(1, "Out of memory! Closing server thread\n", mSettings);
|
|
}
|
|
|
|
Mutex_Lock( &clients_mutex );
|
|
Iperf_delete( &(mSettings->peer), &clients );
|
|
Mutex_Unlock( &clients_mutex );
|
|
|
|
DELETE_PTR( reportstruct );
|
|
EndReport( mSettings->reporthdr );
|
|
}
|
|
// end Recv
|
|
|
|
/* -------------------------------------------------------------------
|
|
* Send an AckFIN (a datagram acknowledging a FIN) on the socket,
|
|
* then select on the socket for some time. If additional datagrams
|
|
* come in, probably our AckFIN was lost and they are re-transmitted
|
|
* termination datagrams, so re-transmit our AckFIN.
|
|
* ------------------------------------------------------------------- */
|
|
|
|
void Server::write_UDP_AckFIN( ) {
|
|
|
|
int rc;
|
|
|
|
fd_set readSet;
|
|
FD_ZERO( &readSet );
|
|
|
|
struct timeval timeout;
|
|
|
|
int count = 0;
|
|
while ( count < 10 ) {
|
|
count++;
|
|
|
|
UDP_datagram *UDP_Hdr;
|
|
server_hdr *hdr;
|
|
|
|
UDP_Hdr = (UDP_datagram*) mBuf;
|
|
|
|
if ( mSettings->mBufLen > (int) ( sizeof( UDP_datagram )
|
|
+ sizeof( server_hdr ) ) ) {
|
|
Transfer_Info *stats = GetReport( mSettings->reporthdr );
|
|
hdr = (server_hdr*) (UDP_Hdr+1);
|
|
|
|
hdr->flags = htonl( HEADER_VERSION1 );
|
|
hdr->total_len1 = htonl( (long) (stats->TotalLen >> 32) );
|
|
hdr->total_len2 = htonl( (long) (stats->TotalLen & 0xFFFFFFFF) );
|
|
hdr->stop_sec = htonl( (long) stats->endTime );
|
|
hdr->stop_usec = htonl( (long)((stats->endTime - (long)stats->endTime)
|
|
* rMillion));
|
|
hdr->error_cnt = htonl( stats->cntError );
|
|
hdr->outorder_cnt = htonl( stats->cntOutofOrder );
|
|
hdr->datagrams = htonl( stats->cntDatagrams );
|
|
hdr->jitter1 = htonl( (long) stats->jitter );
|
|
hdr->jitter2 = htonl( (long) ((stats->jitter - (long)stats->jitter)
|
|
* rMillion) );
|
|
|
|
}
|
|
|
|
// write data
|
|
write( mSettings->mSock, mBuf, mSettings->mBufLen );
|
|
|
|
// wait until the socket is readable, or our timeout expires
|
|
FD_SET( mSettings->mSock, &readSet );
|
|
timeout.tv_sec = 1;
|
|
timeout.tv_usec = 0;
|
|
|
|
rc = select( mSettings->mSock+1, &readSet, NULL, NULL, &timeout );
|
|
FAIL_errno( rc == SOCKET_ERROR, "select", mSettings );
|
|
|
|
if ( rc == 0 ) {
|
|
// select timed out
|
|
return;
|
|
} else {
|
|
// socket ready to read
|
|
rc = read( mSettings->mSock, mBuf, mSettings->mBufLen );
|
|
WARN_errno( rc < 0, "read" );
|
|
if ( rc <= 0 ) {
|
|
// Connection closed or errored
|
|
// Stop using it.
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
fprintf( stderr, warn_ack_failed, mSettings->mSock, count );
|
|
}
|
|
// end write_UDP_AckFIN
|
|
|