sjero.net Git - iperf/blob - include/util.h

   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  * util.h
  48  * by Mark Gates <mgates@nlanr.net>
  49  * -------------------------------------------------------------------
  50  * various C utility functions.
  51  * ------------------------------------------------------------------- */
  52
  53 #ifndef UTIL_H
  54 #define UTIL_H
  55
  56 #ifdef HAVE_CONFIG_H
  57     #include "config.h"
  58 #endif
  59
  60 #ifdef __cplusplus
  61 extern "C" {
  62 #endif
  63
  64 /* -------------------------------------------------------------------
  65  * set/getsockopt wrappers for SO_RCVBUF and SO_SNDBUF; TCP_MAXSEG
  66  * socket.c
  67  * ------------------------------------------------------------------- */
  68 int setsock_tcp_windowsize( int inSock, int inTCPWin, int inSend );
  69 int getsock_tcp_windowsize( int inSock, int inSend );
  70
  71 void setsock_tcp_mss( int inSock, int inTCPWin );
  72 int  getsock_tcp_mss( int inSock );
  73
  74 /* -------------------------------------------------------------------
  75  * signal handlers
  76  * signal.c
  77  * ------------------------------------------------------------------- */
  78 typedef void Sigfunc(int);
  79 void sig_exit( int inSigno );
  80
  81 typedef Sigfunc *SigfuncPtr;
  82
  83 SigfuncPtr my_signal( int inSigno, SigfuncPtr inFunc );
  84
  85 /* -------------------------------------------------------------------
  86  * error handlers
  87  * error.c
  88  * ------------------------------------------------------------------- */
  89 void warn      ( const char *inMessage, const char *inFile, int inLine );
  90 void warn_errno( const char *inMessage, const char *inFile, int inLine );
  91
  92 #if defined( HAVE_POSIX_THREAD ) || defined( HAVE_WIN32_THREAD)
  93 #define FAIL( cond, msg, settings )             \
  94   do {                                          \
  95     if ( cond ) {                               \
  96       warn( msg, __FILE__, __LINE__ );          \
  97       thread_stop(settings);                    \
  98     }                                           \
  99   } while( 0 )
 100 #else
 101 #define FAIL( cond, msg, settings )             \
 102   do {                                          \
 103     if ( cond ) {                               \
 104       warn( msg, __FILE__, __LINE__ );          \
 105       exit( 1 );                                \
 106     }                                           \
 107   } while( 0 )
 108 #endif
 109
 110 #define WARN( cond, msg )                       \
 111   do {                                          \
 112     if ( cond ) {                               \
 113       warn( msg, __FILE__, __LINE__ );          \
 114     }                                           \
 115   } while( 0 )
 116
 117 #if defined( HAVE_POSIX_THREAD ) || defined( HAVE_WIN32_THREAD)
 118 #define FAIL_errno( cond, msg, settings )       \
 119   do {                                          \
 120     if ( cond ) {                               \
 121       warn_errno( msg, __FILE__, __LINE__ );    \
 122       thread_stop(settings);                    \
 123     }                                           \
 124   } while( 0 )
 125 #else
 126 #define FAIL_errno( cond, msg, settings )       \
 127   do {                                          \
 128     if ( cond ) {                               \
 129       warn_errno( msg, __FILE__, __LINE__ );    \
 130       exit( 1 );                                \
 131     }                                           \
 132   } while( 0 )
 133 #endif
 134
 135 #define WARN_errno( cond, msg )                 \
 136   do {                                          \
 137     if ( cond ) {                               \
 138       warn_errno( msg, __FILE__, __LINE__ );    \
 139     }                                           \
 140   } while( 0 )
 141
 142 /* -------------------------------------------------------------------
 143  * initialize buffer to a pattern
 144  * ------------------------------------------------------------------- */
 145 void pattern( char *outBuf, int inBytes );
 146
 147 /* -------------------------------------------------------------------
 148  * input and output numbers, converting with kilo, mega, giga
 149  * stdio.c
 150  * ------------------------------------------------------------------- */
 151 double byte_atof( const char *inString );
 152 max_size_t byte_atoi( const char  *inString );
 153 void byte_snprintf( char* outString, int inLen, double inNum, char inFormat );
 154
 155 /* -------------------------------------------------------------------
 156  * redirect the stdout to a specified file
 157  * stdio.c
 158  * ------------------------------------------------------------------- */
 159 void redirect(const char *inOutputFileName);
 160
 161 /* -------------------------------------------------------------------
 162  * delete macro
 163  * ------------------------------------------------------------------- */
 164 #define DELETE_PTR( ptr )                       \
 165   do {                                          \
 166     if ( ptr != NULL ) {                        \
 167       delete ptr;                               \
 168       ptr = NULL;                               \
 169     }                                           \
 170   } while( false )
 171
 172 #define DELETE_ARRAY( ptr )                     \
 173   do {                                          \
 174     if ( ptr != NULL ) {                        \
 175       delete [] ptr;                            \
 176       ptr = NULL;                               \
 177     }                                           \
 178   } while( false )
 179
 180 #ifdef __cplusplus
 181 } /* end extern "C" */
 182 #endif
 183
 184 #endif /* UTIL_H */
 185