159 строки
5.6 KiB
C++
159 строки
5.6 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 "SocketAddr.h"
|
|
#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) );
|
|
|
|
if ( isCongestionControl( inSettings ) ) {
|
|
#ifdef TCP_CONGESTION
|
|
Socklen_t len = strlen( inSettings->mCongestion ) + 1;
|
|
int rc = setsockopt( inSettings->mSock, IPPROTO_TCP, TCP_CONGESTION,
|
|
inSettings->mCongestion, len);
|
|
if (rc == SOCKET_ERROR ) {
|
|
fprintf(stderr, "Attempt to set '%s' congestion control failed: %s\n",
|
|
inSettings->mCongestion, strerror(errno));
|
|
exit(1);
|
|
}
|
|
#else
|
|
fprintf( stderr, "The -Z option is not available on this operating system\n");
|
|
#endif
|
|
}
|
|
|
|
// check if we're sending multicast, and set TTL
|
|
if ( isMulticast( inSettings ) && ( inSettings->mTTL > 0 ) ) {
|
|
int val = inSettings->mTTL;
|
|
#ifdef HAVE_MULTICAST
|
|
if ( !SockAddr_isIPv6( &inSettings->local ) ) {
|
|
int rc = setsockopt( inSettings->mSock, IPPROTO_IP, IP_MULTICAST_TTL,
|
|
(const void*) &val, (Socklen_t) sizeof(val));
|
|
|
|
WARN_errno( rc == SOCKET_ERROR, "multicast ttl" );
|
|
}
|
|
#ifdef HAVE_IPV6_MULTICAST
|
|
else {
|
|
int rc = setsockopt( inSettings->mSock, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
|
|
(const void*) &val, (Socklen_t) sizeof(val));
|
|
WARN_errno( rc == SOCKET_ERROR, "multicast ttl" );
|
|
}
|
|
#endif
|
|
#endif
|
|
}
|
|
|
|
|
|
#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
|