sockset.c File Reference

Socket Sets services definitions. More...

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include "rnr/rnrconfig.h"
#include "rnr/sock.h"
#include "rnr/sockset.h"
#include "rnr/log.h"
#include "rnr/new.h"

Go to the source code of this file.

Classes
struct	sockset_t
	Socket Set Structure. More...

Functions
static void	fdset_nowarn (int fd, fd_set *pset)
	FD_SET() wrapper with no annoying warnings. More...
static void	fdclr_nowarn (int fd, fd_set *pset)
	FD_CLR() wrapper with no annoying warnings. More...
static int	fdisset_nowarn (int fd, fd_set *pset)
	FD_ISSET() wrapper with no annoying warnings. More...
static ushort_t	ntohs_nowarn (ushort_t huShort)
	ntohs() wrapper with no annoying warnings. More...
static ushort_t	htons_nowarn (ushort_t huShort)
	htons() wrapper with no annoying warnings. More...
static int	SockSetFindHighestSd (SockSet_T *pSockSet)
	Find the highest socket descriptor number in the Active subset. More...
static int	SockSetChkSd (Socket_T *pSocket)
	Check if socket is open and within valid range. More...
SockSet_T *	SockSetNew ()
	Allocate a new SockSet. More...
void	SockSetDelete (SockSet_T *pSockSet)
	Delete an allocated SockSet. More...
int	SockSetAdd (SockSet_T *pSockSet, Socket_T *pSocket, bool_t bActivateRead, bool_t bActivateWrite)
	Add a Socket to a SockSet. More...
int	SockSetRemove (SockSet_T *pSockSet, Socket_T *pSocket)
	Remove Socket from SockSet. More...
int	SockSetActivate (SockSet_T *pSockSet, Socket_T *pSocket, bool_t bActivateRead, bool_t bActivateWrite)
	Activate Socket read/write operations in SockSet. More...
int	SockSetPutOnHold (SockSet_T *pSockSet, Socket_T *pSocket, bool_t bHoldRead, bool_t bHoldWrite)
	Put Socket read/write operations in SockSet on hold. More...
int	SockSetSelect (SockSet_T *pSockSet, struct timeval *pTimeOut)
	Perform select() on SockSet's Active sets. More...
Socket_T *	SockSetIterFirst (SockSet_T *pSockSet, int eSet, int eIO, SockSetIter_T *pIter)
	Start iteration of Sockets in the given set in SockSet. More...
Socket_T *	SockSetIterNext (SockSetIter_T *pIter)
	Next Socket in iteration over the initialized socket set of SockSet. More...

Detailed Description

Socket Sets services definitions.

LastChangedDate

2010-05-03 13:38:47 -0600 (Mon, 03 May 2010)

Rev

362

A socket set is composed of three open socket read/write subsets:

Active	SD's (socked descriptors) active for read/write I/O.
OnHold	SD's managed but are on hold (not select()'ed).
Select	The select() working subset. The Active subset is copied into the Select set and passed into select(). On select()'s return, the Select set is the subset SD's with I/O events.

Note

These socket functions were inspired by the camserv socket.c functions written by Jon Travis (jtrav.nosp@m.is@p.nosp@m.00p.o.nosp@m.rg)

Author

Robin Knight (robin.nosp@m..kni.nosp@m.ght@r.nosp@m.oadn.nosp@m.arrow.nosp@m.s.co.nosp@m.m)

Copyright

© 2005-2018. RoadNarrows LLC..
http://www.roadnarrows.com
All Rights Reserved

Definition in file sockset.c.

Function Documentation

static void fdclr_nowarn ( int  fd, fd_set *  pset  )

inlinestatic

FD_CLR() wrapper with no annoying warnings.

Parameters

fd	File descriptor to add to set.
pset	Pointer to fd set.

Definition at line 107 of file sockset.c.

Referenced by SockSetActivate(), SockSetPutOnHold(), and SockSetRemove().

{
  FD_CLR(fd, pset);
}

static int fdisset_nowarn ( int  fd, fd_set *  pset  )

inlinestatic

FD_ISSET() wrapper with no annoying warnings.

Parameters

fd	File descriptor to add to set.
pset	Pointer to fd set.

Returns

Returusn non-zero if set, else zero.

Definition at line 118 of file sockset.c.

References PRAGMA_IGNORED, and PRAGMA_WARNING.

Referenced by SockSetFindHighestSd(), and SockSetIterNext().

{
  return FD_ISSET(fd, pset);
}

static void fdset_nowarn ( int  fd, fd_set *  pset  )

inlinestatic

FD_SET() wrapper with no annoying warnings.

Parameters

fd	File descriptor to add to set.
pset	Pointer to fd set.

Definition at line 97 of file sockset.c.

Referenced by SockSetActivate(), SockSetAdd(), and SockSetPutOnHold().

{
  FD_SET(fd, pset);
}

static ushort_t htons_nowarn ( ushort_t  huShort )

inlinestatic

htons() wrapper with no annoying warnings.

Parameters

huShort

Unsigned short integer.

Returns

Network byte order.

Definition at line 140 of file sockset.c.

References PRAGMA_WARNING.

{
  return htons(huShort);
}

static ushort_t ntohs_nowarn ( ushort_t  huShort )

inlinestatic

ntohs() wrapper with no annoying warnings.

Parameters

huShort

Unsigned short integer.

Returns

Host byte order.

Definition at line 130 of file sockset.c.

{
  return ntohs(huShort);
}

int SockSetActivate	(	SockSet_T *	pSockSet,
		Socket_T *	pSocket,
		bool_t	bActivateRead,
		bool_t	bActivateWrite
	)

Activate Socket read/write operations in SockSet.

Socket is already part of the SockSet.

Parameters

pSockSet	Pointer to SockSet.
pSocket	Pointer to Socket.
bActivateRead	Do [not] add to active read set.
bActivateWrite	Do [not] add to active write set.

Returns

Returns OK(0) on success, ≤ 0 on error.

Definition at line 377 of file sockset.c.

References CHKPTR, fdclr_nowarn(), fdset_nowarn(), sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, OK, SOCK_IO_READ, SOCK_IO_WRITE, SOCK_RC_EBADSOCK, SockSetChkSd(), and SockSetFindHighestSd().

{
  int   sd;   // socket descriptor

  CHKPTR(pSockSet, SOCK_RC_EBADSOCK);

  // Check data and return socket descriptor
  if( (sd = SockSetChkSd(pSocket)) < 0 )
  {
    return sd;
  }

  if( bActivateRead )
  {
    fdclr_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_READ]));
    fdset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_READ]));
    if( sd > pSockSet->m_sdHighest )
    {
      pSockSet->m_sdHighest = sd;
    }
  }

  if( bActivateWrite )
  {
    fdclr_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_WRITE]));
    fdset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_WRITE]));
    if( sd > pSockSet->m_sdHighest )
    {
      pSockSet->m_sdHighest = sd;
    }
  }

  pSockSet->m_sdHighest = SockSetFindHighestSd(pSockSet);

  return OK;
}

int SockSetAdd	(	SockSet_T *	pSockSet,
		Socket_T *	pSocket,
		bool_t	bActivateRead,
		bool_t	bActivateWrite
	)

Add a Socket to a SockSet.

Socket is added to either the on-hold or active socket set, depending on the given boolean value.

Parameters

pSockSet	Pointer to SockSet.
pSocket	Pointer to Socket.
bActivateRead	Do [not] add to active read set.
bActivateWrite	Do [not] add to active write set.

Returns

Returns OK(0) on success, ≤ 0 on error.

Definition at line 272 of file sockset.c.

References _TBOOL, _TPTR, _TSTR, CHKPTR, fdset_nowarn(), LOGDIAG4CALL, sockset_t::m_pSocket, sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, OK, SOCK_IO_READ, SOCK_IO_WRITE, SOCK_RC_EBADSOCK, SocketAttrGetLocalName(), and SockSetChkSd().

{
  int   sd;   // socket descriptor

  CHKPTR(pSockSet, SOCK_RC_EBADSOCK);

  // Check data and return socket descriptor
  if( (sd = SockSetChkSd(pSocket)) < 0 )
  {
    return sd;
  }

  LOGDIAG4CALL(_TPTR(pSockSet), _TSTR(SocketAttrGetLocalName(pSocket)),
      _TBOOL(bActivateRead), _TBOOL(bActivateWrite));

  //
  // Add sd to appropriate read socket descriptor set
  //
  if( bActivateRead ) // place read in the active set
  {
    fdset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_READ]));
    if( pSockSet->m_sdHighest < sd )
    {
      pSockSet->m_sdHighest = sd;
    }
  }
  else  // place read in the on-hold set
  {
    fdset_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_READ]));
  }

  //
  // Add sd to appropriate write socket descriptor set
  //
  if( bActivateWrite ) // place write in the active set
  {
    fdset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_WRITE]));
    if( pSockSet->m_sdHighest < sd )
    {
      pSockSet->m_sdHighest = sd;
    }
  }
  else  // place write in the on-hold set
  {
    fdset_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_WRITE]));
  }

  // add socket 
  pSockSet->m_pSocket[sd] = pSocket;

  return OK;
}

static int SockSetChkSd ( Socket_T *  pSocket )

static

Check if socket is open and within valid range.

Parameters

pSocket

Pointer to socket.

Returns

Returns socket descriptor if valid.
Otherwise the appropriate socket return code < 0 is returned.

Definition at line 194 of file sockset.c.

References CHKPTR, LOGERROR, SOCK_RC_EBADSD, SOCK_RC_EBADSOCK, SocketAttrGetLocalName(), SocketAttrGetSd(), and SocketStateIsOpen().

Referenced by SockSetActivate(), SockSetAdd(), SockSetPutOnHold(), and SockSetRemove().

{
  int   sd;   // socket descriptor

  CHKPTR(pSocket, SOCK_RC_EBADSOCK);

  // this socket isn't open
  if( !SocketStateIsOpen(pSocket) )
  {
    LOGERROR("Socket not open");
    return SOCK_RC_EBADSD;
  }

  sd = SocketAttrGetSd(pSocket);

  if( sd >= FD_SETSIZE )
  {
    LOGERROR("Socket '%s' descriptor %d > %d: too big", 
        SocketAttrGetLocalName(pSocket), sd, FD_SETSIZE);
    return SOCK_RC_EBADSD;
  }

  return sd;
}

void SockSetDelete

(

SockSet_T *

pSockSet

)

Delete an allocated SockSet.

Parameters

pSockSet

Pointer to SockSet.

Definition at line 250 of file sockset.c.

References NULL.

{
  if( pSockSet == NULL )
  {
    return;
  }
  delete(pSockSet);
}

static int SockSetFindHighestSd ( SockSet_T *  pSockSet )

static

Find the highest socket descriptor number in the Active subset.

Parameters

pSockSet

Pointer to socket set.

Returns

Highest socket descriptor. -1 if none found.

Definition at line 165 of file sockset.c.

References fdisset_nowarn(), sockset_t::m_sdHighest, sockset_t::m_sdSetActive, SOCK_IO_READ, and SOCK_IO_WRITE.

Referenced by SockSetActivate(), SockSetPutOnHold(), and SockSetRemove().

{
  int   sd;
  int   sdHighest;

  for(sd=0, sdHighest=-1; sd <= pSockSet->m_sdHighest; ++sd)
  {
    if(   fdisset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_READ]))
       || fdisset_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_WRITE])) )
    {
      if( sd > sdHighest )
      {
        sdHighest = sd;
      }
    }
  }

  return sdHighest;
}

Socket_T* SockSetIterFirst	(	SockSet_T *	pSockSet,
		int	eSet,
		int	eIO,
		SockSetIter_T *	pIter
	)

Start iteration of Sockets in the given set in SockSet.

Parameters

pSockSet	Pointer to SockSet.
eSet	One of: SOCK_SET_ONHOLD SOCK_SET_ACTIVE SOCK_SET_SELECTED.
eIO	One of: SOCK_IO_READ SOCK_IO_WRITE.
pIter	SockSet interator.

Returns

Pointer to first Socket if found, else NULL.

Definition at line 548 of file sockset.c.

References CHKPTR, SockSetIter_T::m_eIO, SockSetIter_T::m_eSet, SockSetIter_T::m_pSockSet, SockSetIter_T::m_sdCur, SockSetIter_T::m_sdSet, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, sockset_t::m_sdSetSelect, NULL, SOCK_CHK_IO, SOCK_SET_ACTIVE, SOCK_SET_CHK_SET, SOCK_SET_ONHOLD, SOCK_SET_SELECTED, and SockSetIterNext().

{
  CHKPTR(pSockSet, NULL);
  SOCK_SET_CHK_SET(eSet, NULL);
  SOCK_CHK_IO(eIO, NULL);
  CHKPTR(pIter, NULL);

  pIter->m_pSockSet = pSockSet;
  pIter->m_eSet     = eSet;
  pIter->m_eIO      = eIO;
  pIter->m_sdCur    = -1;

  switch(eSet)
  {
    case SOCK_SET_ONHOLD:
      pIter->m_sdSet = pSockSet->m_sdSetOnHold[eIO];
      break;
    case SOCK_SET_ACTIVE:
      pIter->m_sdSet = pSockSet->m_sdSetActive[eIO];
      break;
    case SOCK_SET_SELECTED:
    default:
      pIter->m_sdSet = pSockSet->m_sdSetSelect[eIO];
      break;
  }

  return SockSetIterNext(pIter);
}

Socket_T* SockSetIterNext

(

SockSetIter_T *

pIter

)

Next Socket in iteration over the initialized socket set of SockSet.

Parameters

pIter

SockSet interator.

Returns

Pointer to next Socket if found, else NULL.

Definition at line 587 of file sockset.c.

References CHKPTR, fdisset_nowarn(), sockset_t::m_pSocket, SockSetIter_T::m_pSockSet, SockSetIter_T::m_sdCur, SockSetIter_T::m_sdSet, and NULL.

Referenced by SockSetIterFirst().

{
  int   sd;

  CHKPTR(pIter, NULL);
  CHKPTR(pIter->m_pSockSet, NULL);

  for(sd = pIter->m_sdCur + 1; sd <= pIter->m_pSockSet->m_sdHighest; ++sd)
  {
    if( fdisset_nowarn(sd, &(pIter->m_sdSet)) )
    {
      pIter->m_sdCur = sd;
      return pIter->m_pSockSet->m_pSocket[sd];
    }
  }

  // no more selected sockets found
  pIter->m_sdCur = sd;

  return NULL;
}

SockSet_T* SockSetNew

(

)

Allocate a new SockSet.

Returns

Pointer to allocated SockSet.

Definition at line 229 of file sockset.c.

References sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, sockset_t::m_sdSetSelect, NEW, SOCK_IO_READ, and SOCK_IO_WRITE.

{
  SockSet_T *pSockSet = NEW(SockSet_T);

  pSockSet->m_sdHighest = -1;

  FD_ZERO(&(pSockSet->m_sdSetSelect[SOCK_IO_READ]));
  FD_ZERO(&(pSockSet->m_sdSetSelect[SOCK_IO_WRITE]));
  FD_ZERO(&(pSockSet->m_sdSetActive[SOCK_IO_READ]));
  FD_ZERO(&(pSockSet->m_sdSetActive[SOCK_IO_WRITE]));
  FD_ZERO(&(pSockSet->m_sdSetOnHold[SOCK_IO_READ]));
  FD_ZERO(&(pSockSet->m_sdSetOnHold[SOCK_IO_WRITE]));

  return pSockSet;
}

int SockSetPutOnHold	(	SockSet_T *	pSockSet,
		Socket_T *	pSocket,
		bool_t	bHoldRead,
		bool_t	bHoldWrite
	)

Put Socket read/write operations in SockSet on hold.

Socket is already part of the SockSet.

Parameters

pSockSet	Pointer to SockSet.
pSocket	Pointer to Socket.
bHoldRead	Do [not] add to hold read set.
bHoldWrite	Do [not] add to hold write set.

Returns

Returns OK(0) on success, ≤ 0 on error.

Definition at line 429 of file sockset.c.

References CHKPTR, fdclr_nowarn(), fdset_nowarn(), sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, OK, SOCK_IO_READ, SOCK_IO_WRITE, SOCK_RC_EBADSOCK, SockSetChkSd(), and SockSetFindHighestSd().

{
  int   sd;   // socket descriptor

  CHKPTR(pSockSet, SOCK_RC_EBADSOCK);

  // Check data and return socket descriptor
  if( (sd = SockSetChkSd(pSocket)) < 0 )
  {
    return sd;
  }

  if( bHoldRead )
  {
    fdclr_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_READ]));
    fdset_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_READ]));
  }

  if( bHoldWrite )
  {
    fdclr_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_WRITE]));
    fdset_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_WRITE]));
  }

  pSockSet->m_sdHighest = SockSetFindHighestSd(pSockSet);

  return OK;
}

int SockSetRemove	(	SockSet_T *	pSockSet,
		Socket_T *	pSocket
	)

Remove Socket from SockSet.

Parameters

pSockSet	Pointer to SockSet.
pSocket	Pointer to Socket.

Returns

Returns OK(0) on success, ≤ 0 on error.

Definition at line 336 of file sockset.c.

References _TPTR, _TSTR, CHKPTR, fdclr_nowarn(), LOGDIAG4CALL, sockset_t::m_pSocket, sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetOnHold, NULL, OK, SOCK_IO_READ, SOCK_IO_WRITE, SOCK_RC_EBADSOCK, SocketAttrGetLocalName(), SockSetChkSd(), and SockSetFindHighestSd().

{
  int   sd;   // socket descriptor

  CHKPTR(pSockSet, SOCK_RC_EBADSOCK);

  // Check data and return socket descriptor
  if( (sd = SockSetChkSd(pSocket)) < 0 )
  {
    return sd;
  }

  LOGDIAG4CALL(_TPTR(pSockSet), _TSTR(SocketAttrGetLocalName(pSocket)));

  // remove socket descriptor from all sets
  fdclr_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_READ]));
  fdclr_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_READ]));
  fdclr_nowarn(sd, &(pSockSet->m_sdSetActive[SOCK_IO_WRITE]));
  fdclr_nowarn(sd, &(pSockSet->m_sdSetOnHold[SOCK_IO_WRITE]));

  // remove socket
  pSockSet->m_pSocket[sd] = NULL;

  // find new highest socket descriptor
  pSockSet->m_sdHighest = SockSetFindHighestSd(pSockSet);

  return OK;
}

int SockSetSelect	(	SockSet_T *	pSockSet,
		struct timeval *	pTimeOut
	)

Perform select() on SockSet's Active sets.

Parameters

pSockSet	Pointer to SockSet.
pTimeOut	Select time out.

Returns

On success, returns number of socket descriptors found in the active state.
On time out, returns 0.
On error, returns ≤ 0.

Definition at line 472 of file sockset.c.

References CHKPTR, LOGERROR, LOGSYSERROR, sockset_t::m_sdHighest, sockset_t::m_sdSetActive, sockset_t::m_sdSetSelect, NULL, SOCK_IO_READ, SOCK_IO_WRITE, SOCK_RC_EBADSOCK, SOCK_RC_EFAIL, and SOCK_RC_ESYSERR.

{
  struct timeval  tvSave = {0, 0};  // saved timeout value
  int             nSd;              // number of socket descriptors selected
  int             nTries = 0;       // number of tries when interrupted

  CHKPTR(pSockSet, SOCK_RC_EBADSOCK);

  // save timeout value - can get munged by select()
  if( pTimeOut != NULL )
  {
    tvSave = *pTimeOut;
  }

  //
  // Try at most three times to select.
  //
  while(nTries++ < 3)
  {
    // build select read/write sets
    pSockSet->m_sdSetSelect[SOCK_IO_READ] = 
                                      pSockSet->m_sdSetActive[SOCK_IO_READ];
    pSockSet->m_sdSetSelect[SOCK_IO_WRITE] = 
                                      pSockSet->m_sdSetActive[SOCK_IO_WRITE];

    //
    // Wait with timeout (optional) for read/write availability
    //
    nSd = select(pSockSet->m_sdHighest+1,
                 &(pSockSet->m_sdSetSelect[SOCK_IO_READ]),
                 &(pSockSet->m_sdSetSelect[SOCK_IO_WRITE]),
                 NULL,
                 pTimeOut);

    // restore mangled time out
    if( pTimeOut != NULL )
    {
      *pTimeOut = tvSave;
    }

    // system error occurred
    if( nSd < 0 )
    {
      switch(errno)
      {
        case EINTR: // interrupted - try again
          break;
        default:
          LOGSYSERROR("select()");
          return SOCK_RC_ESYSERR;
      }
    }

    // read/write sets available or timeout
    else
    {
      return nSd;
    }
  }

  LOGERROR("Failed %d times to select()", nTries);

  return SOCK_RC_EFAIL;
}