void PerfSocket::ReportPeriodicBW( void ) {
if ( mPReporting &&
mPacketTime.after( mPNextTime ) ) {
double inStart = mPLastTime.subSec( mStartTime );
double inStop = mPNextTime.subSec( mStartTime );
sReporting.Lock();
ReportBW( mTotalLen - mPLastTotalLen,
inStart,
inStop );
if ( ptr_parent ) {
ptr_parent->PeriodicUpdate( inStart,
inStop,
mTotalLen - mPLastTotalLen );
}
sReporting.Unlock();
mPLastTime = mPNextTime;
mPNextTime.add( mPInterval );
mPLastTotalLen = mTotalLen;
if ( mPacketTime.after( mPNextTime ) ) {
ReportPeriodicBW();
}
}
}
void PerfSocket::Server_Send_TCP(void) {
int currLen;
int writeLenCnt[8193];
if ( mSettings->mBufLen > 8193 )
mSettings->mBufLen = 8193;
for ( int i = 0; i <= mSettings->mBufLen; i++ ) {
writeLenCnt[ i ] = 0;
}
#ifndef WIN32
signal(SIGPIPE,SIG_IGN);
#endif
do {
// perform write
currLen = write( mSock, mBuf, mSettings->mBufLen );
mPacketTime.setnow();
mTotalLen += currLen;
// count number of reads of each size
if ( currLen <= mSettings->mBufLen ) {
writeLenCnt[ currLen ]++;
}
// periodically report bandwidths
ReportPeriodicBW();
} while ( currLen > 0 && sInterupted == false );
if ( sInterupted != false )
printf("The stream was interrupted\n");
// stop timing
mEndTime.setnow();
sReporting.Lock();
ReportBW( mTotalLen, 0.0, mEndTime.subSec( mStartTime ));
sReporting.Unlock();
if ( mSettings->mPrintMSS ) {
// read the socket option for MSS (maximum segment size)
ReportMSS( getsock_tcp_mss( mSock ));
// on WANs the most common read length is often the MSS
// on fast LANs it is much harder to detect
int totalWrites = 0;
for ( currLen = 0; currLen < mSettings->mBufLen+1; currLen++ ) {
totalWrites += writeLenCnt[ currLen ];
}
// print each read length that occured > 5% of reads
int thresh = (int) (0.05 * totalWrites);
printf( report_read_lengths, mSock );
for ( currLen = 0; currLen < mSettings->mBufLen+1; currLen++ ) {
if ( writeLenCnt[ currLen ] > thresh ) {
printf( report_read_length_times, mSock,
(int) currLen, writeLenCnt[ currLen ],
(100.0 * writeLenCnt[ currLen ]) / totalWrites );
}
}
}
}
void PerfSocket::Recv_TCP( void ) {
extern Mutex clients_mutex;
extern Iperf_ListEntry *clients;
// get the remote address and remove it later from the set of clients
SocketAddr remote = getRemoteAddress();
iperf_sockaddr peer = *(iperf_sockaddr *) (remote.get_sockaddr());
// keep track of read sizes -> gives some indication of MTU size
// on SGI this must be dynamically allocated to avoid seg faults
int currLen;
int *readLenCnt = new int[ mSettings->mBufLen+1 ];
for ( int i = 0; i <= mSettings->mBufLen; i++ ) {
readLenCnt[ i ] = 0;
}
InitTransfer();
#ifndef WIN32
signal (SIGPIPE, SIG_IGN);
#endif
do {
// perform read
currLen = read( mSock, mBuf, mSettings->mBufLen );
if ( false ) {
DELETE_ARRAY( readLenCnt );
Server_Send_TCP();
return;
}
mPacketTime.setnow();
// periodically report bandwidths
ReportPeriodicBW();
mTotalLen += currLen;
// count number of reads of each size
if ( currLen <= mSettings->mBufLen ) {
readLenCnt[ currLen ]++;
}
} while ( currLen > 0 && sInterupted == false );
// stop timing
mEndTime.setnow();
sReporting.Lock();
ReportBW( mTotalLen, 0.0, mEndTime.subSec( mStartTime ));
sReporting.Unlock();
if ( mSettings->mPrintMSS ) {
// read the socket option for MSS (maximum segment size)
ReportMSS( getsock_tcp_mss( mSock ));
// on WANs the most common read length is often the MSS
// on fast LANs it is much harder to detect
int totalReads = 0;
for ( currLen = 0; currLen < mSettings->mBufLen+1; currLen++ ) {
totalReads += readLenCnt[ currLen ];
}
// print each read length that occured > 5% of reads
int thresh = (int) (0.05 * totalReads);
printf( report_read_lengths, mSock );
for ( currLen = 0; currLen < mSettings->mBufLen+1; currLen++ ) {
if ( readLenCnt[ currLen ] > thresh ) {
printf( report_read_length_times, mSock,
(int) currLen, readLenCnt[ currLen ],
(100.0 * readLenCnt[ currLen ]) / totalReads );
}
}
}
DELETE_ARRAY( readLenCnt );
clients_mutex.Lock();
Iperf_delete ( &peer, &clients );
clients_mutex.Unlock();
}
void PerfSocket::Client_Recv_TCP(void) {
// terminate loop nicely on user interupts
sInterupted = false;
//my_signal( SIGINT, Sig_Interupt );
//my_signal( SIGPIPE, Sig_Interupt );
#ifndef WIN32
signal (SIGPIPE, SIG_IGN);
#endif
int currLen;
InitTransfer();
double fract = 0.0;
mStartTime.setnow();
long endSize = get_tcp_windowsize(mSock), startSize=endSize, loopLen =0, prevLen =0;
Timestamp prevTime;
prevTime.setnow();
/* Periodic reporting is done here in the loop itself, if Suggest Window Size option is set*/
mPReporting = false;
/* Send the first packet indicating that the server has to send data */
mBuf[0] = 'a';
currLen = write( mSock, mBuf, mSettings->mBufLen );
if ( currLen < 0 ) {
WARN_errno( currLen < 0, "write" );
return;
}
do {
// perform read
currLen = read( mSock, mBuf, mSettings->mBufLen );
mPacketTime.setnow();
if ( currLen < 0 ) {
WARN_errno( currLen < 0, "read" );
break;
}
mTotalLen += currLen;
loopLen +=currLen;
// periodically report bandwidths
ReportPeriodicBW();
double nFract = mStartTime.fraction(mPacketTime,mEndTime);
if ( nFract > (fract + 0.1) ) {
printf(seperator_line);
ReportWindowSize();
sReporting.Lock();
ReportBW( loopLen, prevTime.subSec(mStartTime), mPacketTime.subSec( mStartTime));
sReporting.Unlock();
fract +=0.1;
if ( startSize != endSize ) {
/* Change the window size only if the data transfer has changed at least by 5% */
if ( loopLen < prevLen ) {
if ( ( ((double)(prevLen - loopLen)) /
((double)prevLen)) > 0.05
) {
endSize = startSize + (endSize - startSize)/2;
}
} else {
if ( ( ((double)(loopLen - prevLen)) /
((double)prevLen) ) > 0.05
) {
startSize = endSize;
endSize = endSize*2;
prevLen = loopLen;
}
}
} else {
endSize = endSize*2;
prevLen = loopLen;
}
/** Reset the variables after setting new window size */
prevTime.setnow();
loopLen = 0 ;
//shutdown(mSock,SHUT_RDWR);
close(mSock);
mSock = -1;
Connect( mSettings->mHost, mSettings->mLocalhost );
mBuf[0] = 'a';
if ( set_tcp_windowsize(mSock,endSize) == -1 ) {
printf(unable_to_change_win);
}
if ( get_tcp_windowsize(mSock) != endSize ) {
printf(unable_to_change_win);
}
write( mSock, mBuf, mSettings->mBufLen );
}
} while ( ! (sInterupted ||
(mMode_time && mPacketTime.after( mEndTime )) ||
(!mMode_time && mTotalLen >= mAmount))
);
printf( seperator_line );
ReportWindowSize();
sReporting.Lock();
ReportBW( loopLen, prevTime.subSec(mStartTime), mPacketTime.subSec( mStartTime));
sReporting.Unlock();
printf( seperator_line);
printf( opt_estimate);
if ( loopLen > prevLen )
set_tcp_windowsize(mSock,endSize);
else
set_tcp_windowsize(mSock,startSize);
ReportWindowSize();
printf( seperator_line );
// stop timing
mEndTime.setnow();
sReporting.Lock();
ReportBW( mTotalLen, 0.0, mEndTime.subSec( mStartTime ));
sReporting.Unlock();
if ( mSettings->mPrintMSS ) {
// read the socket option for MSS (maximum segment size)
ReportMSS( getsock_tcp_mss( mSock ));
}
//close(mSock);
//mSock = -1;
}
void PerfSocket::Send_TCP( void ) {
if ( false ) {
Client_Recv_TCP();
return;
}
// terminate loop nicely on user interupts
sInterupted = false;
SigfuncPtr oldINT = my_signal( SIGINT, Sig_Interupt );
SigfuncPtr oldPIPE = my_signal( SIGPIPE, Sig_Interupt );
int currLen;
bool canRead;
InitTransfer();
do {
// If the input is from a
// file, fill the buffer with
// data from the file
if ( mSettings->mFileInput ) {
extractor->getNextDataBlock(mBuf);
// If the first character is 'a'
// change if to '0' so that the
// server does not mistake it for
// a Window Suggest option
if ( mBuf[0] == 'a' )
mBuf[0] = '0';
canRead = extractor->canRead();
} else
canRead = true;
// perform write
currLen = write( mSock, mBuf, mSettings->mBufLen );
mPacketTime.setnow();
if ( currLen < 0 ) {
WARN_errno( currLen < 0, "write" );
break;
}
// periodically report bandwidths
ReportPeriodicBW();
mTotalLen += currLen;
} while ( ! (sInterupted ||
(mMode_time && mPacketTime.after( mEndTime )) ||
(!mMode_time && mTotalLen >= mAmount)) && canRead );
if ( oldINT != Sig_Interupt ) {
// Return signal handlers to previous handlers
my_signal( SIGINT, oldINT );
my_signal( SIGPIPE, oldPIPE );
}
// shutdown sending connection and wait (read)
// for the other side to shutdown
shutdown( mSock, SHUT_WR );
currLen = read( mSock, mBuf, mSettings->mBufLen );
WARN_errno( currLen == SOCKET_ERROR, "read on server close" );
WARN( currLen > 0, "server sent unexpected data" );
// stop timing
mEndTime.setnow();
sReporting.Lock();
ReportBW( mTotalLen, 0.0, mEndTime.subSec( mStartTime ));
sReporting.Unlock();
if ( mSettings->mPrintMSS ) {
// read the socket option for MSS (maximum segment size)
ReportMSS( getsock_tcp_mss( mSock ));
}
}
