122 lines
4.5 KiB
C++
122 lines
4.5 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
|
||
|
* ________________________________________________________________
|
||
|
*
|
||
|
* PerfSocket.cpp
|
||
|
* by Mark Gates <mgates@nlanr.net>
|
||
|
* Ajay Tirumala <tirumala@ncsa.uiuc.edu>
|
||
|
* -------------------------------------------------------------------
|
||
|
* Has routines the Client and Server classes use in common for
|
||
|
* performance testing the network.
|
||
|
* Changes in version 1.2.0
|
||
|
* for extracting data from files
|
||
|
* -------------------------------------------------------------------
|
||
|
* headers
|
||
|
* uses
|
||
|
* <stdlib.h>
|
||
|
* <stdio.h>
|
||
|
* <string.h>
|
||
|
*
|
||
|
* <sys/types.h>
|
||
|
* <sys/socket.h>
|
||
|
* <unistd.h>
|
||
|
*
|
||
|
* <arpa/inet.h>
|
||
|
* <netdb.h>
|
||
|
* <netinet/in.h>
|
||
|
* <sys/socket.h>
|
||
|
* ------------------------------------------------------------------- */
|
||
|
|
||
|
|
||
|
#define HEADERS()
|
||
|
|
||
|
#include "headers.h"
|
||
|
|
||
|
#include "PerfSocket.hpp"
|
||
|
#include "util.h"
|
||
|
|
||
|
/* -------------------------------------------------------------------
|
||
|
* Set socket options before the listen() or connect() calls.
|
||
|
* These are optional performance tuning factors.
|
||
|
* ------------------------------------------------------------------- */
|
||
|
|
||
|
void SetSocketOptions( thread_Settings *inSettings ) {
|
||
|
// set the TCP window size (socket buffer sizes)
|
||
|
// also the UDP buffer size
|
||
|
// must occur before call to accept() for large window sizes
|
||
|
setsock_tcp_windowsize( inSettings->mSock, inSettings->mTCPWin,
|
||
|
(inSettings->mThreadMode == kMode_Client ? 1 : 0) );
|
||
|
|
||
|
#ifdef IP_TOS
|
||
|
|
||
|
// set IP TOS (type-of-service) field
|
||
|
if ( inSettings->mTOS > 0 ) {
|
||
|
int tos = inSettings->mTOS;
|
||
|
Socklen_t len = sizeof(tos);
|
||
|
int rc = setsockopt( inSettings->mSock, IPPROTO_IP, IP_TOS,
|
||
|
(char*) &tos, len );
|
||
|
WARN_errno( rc == SOCKET_ERROR, "setsockopt IP_TOS" );
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
if ( !isUDP( inSettings ) ) {
|
||
|
// set the TCP maximum segment size
|
||
|
setsock_tcp_mss( inSettings->mSock, inSettings->mMSS );
|
||
|
|
||
|
#ifdef TCP_NODELAY
|
||
|
|
||
|
// set TCP nodelay option
|
||
|
if ( isNoDelay( inSettings ) ) {
|
||
|
int nodelay = 1;
|
||
|
Socklen_t len = sizeof(nodelay);
|
||
|
int rc = setsockopt( inSettings->mSock, IPPROTO_TCP, TCP_NODELAY,
|
||
|
(char*) &nodelay, len );
|
||
|
WARN_errno( rc == SOCKET_ERROR, "setsockopt TCP_NODELAY" );
|
||
|
}
|
||
|
#endif
|
||
|
}
|
||
|
}
|
||
|
// end SetSocketOptions
|