sjero.net Git - iperf/blob - src/PerfSocket.cpp

   1 /*---------------------------------------------------------------
   2  * Copyright (c) 1999,2000,2001,2002,2003
   3  * The Board of Trustees of the University of Illinois
   4  * All Rights Reserved.
   5  *---------------------------------------------------------------
   6  * Permission is hereby granted, free of charge, to any person
   7  * obtaining a copy of this software (Iperf) and associated
   8  * documentation files (the "Software"), to deal in the Software
   9  * without restriction, including without limitation the
  10  * rights to use, copy, modify, merge, publish, distribute,
  11  * sublicense, and/or sell copies of the Software, and to permit
  12  * persons to whom the Software is furnished to do
  13  * so, subject to the following conditions:
  14  *
  15  *
  16  * Redistributions of source code must retain the above
  17  * copyright notice, this list of conditions and
  18  * the following disclaimers.
  19  *
  20  *
  21  * Redistributions in binary form must reproduce the above
  22  * copyright notice, this list of conditions and the following
  23  * disclaimers in the documentation and/or other materials
  24  * provided with the distribution.
  25  *
  26  *
  27  * Neither the names of the University of Illinois, NCSA,
  28  * nor the names of its contributors may be used to endorse
  29  * or promote products derived from this Software without
  30  * specific prior written permission.
  31  *
  32  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  33  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  34  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  35  * NONINFRINGEMENT. IN NO EVENT SHALL THE CONTIBUTORS OR COPYRIGHT
  36  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  37  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  38  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  39  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  40  * ________________________________________________________________
  41  * National Laboratory for Applied Network Research
  42  * National Center for Supercomputing Applications
  43  * University of Illinois at Urbana-Champaign
  44  * http://www.ncsa.uiuc.edu
  45  * ________________________________________________________________
  46  *
  47  * PerfSocket.cpp
  48  * by Mark Gates <mgates@nlanr.net>
  49  *    Ajay Tirumala <tirumala@ncsa.uiuc.edu>
  50  * -------------------------------------------------------------------
  51  * Has routines the Client and Server classes use in common for
  52  * performance testing the network.
  53  * Changes in version 1.2.0
  54  *     for extracting data from files
  55  * -------------------------------------------------------------------
  56  * headers
  57  * uses
  58  *   <stdlib.h>
  59  *   <stdio.h>
  60  *   <string.h>
  61  *
  62  *   <sys/types.h>
  63  *   <sys/socket.h>
  64  *   <unistd.h>
  65  *
  66  *   <arpa/inet.h>
  67  *   <netdb.h>
  68  *   <netinet/in.h>
  69  *   <sys/socket.h>
  70  * ------------------------------------------------------------------- */
  71 #define HEADERS()
  72
  73 #include "headers.h"
  74
  75 #include "PerfSocket.hpp"
  76 #include "SocketAddr.h"
  77 #include "util.h"
  78
  79 // create an internet socket
  80 void MakeSocket(thread_Settings *inSettings)
  81 {
  82    int type = 0, proto = 0, domain =
  83                 SockAddr_isIPv6(&inSettings->local) ? AF_INET6 :
  84                 AF_INET;
  85
  86     switch (inSettings->mProtocol) {
  87         case kProto_TCP:     type  = SOCK_STREAM;     break;
  88         case kProto_UDP:     type  = SOCK_DGRAM;      break;
  89         case kProto_DCCP:    type  = SOCK_DCCP;       break;
  90     }
  91     inSettings->mSock = socket( domain, type, proto );
  92
  93     WARN_errno( inSettings->mSock == INVALID_SOCKET, "socket" );
  94 }
  95
  96 /* -------------------------------------------------------------------
  97  * Set socket options before the listen() or connect() calls.
  98  * These are optional performance tuning factors.
  99  * ------------------------------------------------------------------- */
 100 void SetSocketOptions( thread_Settings *inSettings )
 101 {
 102     int rc, val;
 103     Socklen_t len = sizeof(int);
 104
 105     // check if we're sending multicast, and set TTL
 106     if ( isMulticast( inSettings ) && ( inSettings->mTTL > 0 ) ) {
 107         val = inSettings->mTTL;
 108 #ifdef HAVE_MULTICAST
 109         if ( !SockAddr_isIPv6( &inSettings->local ) ) {
 110             rc = setsockopt( inSettings->mSock, IPPROTO_IP, IP_MULTICAST_TTL,
 111                              &val, len);
 112
 113             WARN_errno( rc == SOCKET_ERROR, "multicast ttl" );
 114         }
 115 #ifdef HAVE_IPV6_MULTICAST
 116         else {
 117             rc = setsockopt( inSettings->mSock, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
 118                              &val, len);
 119             WARN_errno( rc == SOCKET_ERROR, "multicast ttl" );
 120         }
 121 #endif  /* IPV6_MULTICAST */
 122 #endif  /* MULTICAST      */
 123     }
 124
 125
 126     // set IP TOS (type-of-service) field
 127 #ifdef IP_TOS
 128     if ( inSettings->mTOS > 0 ) {
 129         val = inSettings->mTOS;
 130         rc = setsockopt( inSettings->mSock, IPPROTO_IP, IP_TOS, &val, len );
 131         WARN_errno( rc == SOCKET_ERROR, "setsockopt IP_TOS" );
 132     }
 133 #endif
 134
 135
 136     // TCP-specific options
 137     if ( inSettings->mProtocol == kProto_TCP ) {
 138
 139         // set the TCP window size (socket buffer sizes)
 140         // must occur before call to accept() for large window sizes
 141         setsock_tcp_windowsize(inSettings->mSock, inSettings->mWinSize,
 142                                inSettings->mThreadMode == kMode_Client);
 143         setsock_tcp_mss( inSettings->mSock, inSettings->mMSS );
 144
 145 #ifdef TCP_NODELAY
 146         if ( isNoDelay( inSettings ) ) {
 147             val = 1;
 148             rc = setsockopt( inSettings->mSock, IPPROTO_TCP, TCP_NODELAY,
 149                              &val, len );
 150             WARN_errno( rc == SOCKET_ERROR, "setsockopt TCP_NODELAY" );
 151         }
 152 #endif
 153     } else {
 154         rc = set_buffer_sock_size(inSettings->mSock, inSettings->mWinSize,
 155                                   inSettings->mThreadMode == kMode_Client);
 156         WARN_errno( rc < 0 , "setsockopt for buffer size" );
 157     }
 158     // DCCP-specific options
 159     if ( inSettings->mProtocol == kProto_DCCP ) {
 160         /*
 161          * We use the service code SC:PERF (0x50455246) from
 162          * draft-fairhurst-dccp-serv-codes to identify this service.
 163          */
 164         val = htonl(0x50455246);                        /* ALWAYS use htonl */
 165         rc = setsockopt( inSettings->mSock, SOL_DCCP, DCCP_SOCKOPT_SERVICE,
 166                          &val, len );
 167         WARN_errno( rc == SOCKET_ERROR, "setsockopt DCCP_SOCKOPT_SERVICE" );
 168     }
 169 }