Socket.h

Go to the documentation of this file.

/* -*- Mode: C++; indent-tabs-mode: nil; c-basic-offset: 5 -*- */

/* Used to connect to a ServerSocket over TCP/IP. Able  to Send and recieve
 * data over this TCP/IP connection.
 */

/*
 * Socket.h
 * Utilities
 *
 * Created by Daniel Aarno on Sat May 11 2002.
 * Copyright (c) 2002-2003 Daniel Aarno - macbishop@sourceforge.net.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions are met:
 *
 *   *  Redistributions of source code must retain the above copyright notice,
 *      this list of conditions and the following disclaimer.
 *
 *   *  Redistributions in binary form must reproduce the above copyright 
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 *   *  Neither the name of the Royal Institute of Technology, KTH nor the
 *      names of its contributors may be used to endorse or promote products 
 *      derived from this software without specific prior written permission.
 *
 *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 
 *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 *   TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 *   PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 *   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 *   EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 *   PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, 
 *   OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 
 *   OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
 *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 *   THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

 
#ifndef _SOCKET_
#define _SOCKET_

#ifdef SOCKET_STAT
#include <time.h>
#include <iostream>
#endif

#include <sys/types.h>
#include <unistd.h>

class Socket
{
protected:
     bool connected;
     int sockID;
     int error;

#ifdef SOCKET_STAT
     mutable long sentBytes;
     mutable long readBytes;
     mutable time_t startTime;
#endif
     
public:
     Socket() : connected(false), sockID(-1), error(0) {
#ifdef SOCKET_STAT
          sentBytes = 0;
          readBytes = 0;
          startTime = time(NULL);
#endif     
     }
     
     Socket(int fd) : connected(true), sockID(fd), error(0) {
#ifdef SOCKET_STAT
          sentBytes = 0;
          readBytes = 0;
          startTime = time(NULL);
#endif     
     }

     int getError() { return error; }

     int close() { return ::close(sockID); }
     
     ssize_t read(void *buf, size_t nbytes) const 
     {
          size_t len = 0;
          ssize_t n;
          char* ptr = (char*)buf;
          
          while (len < nbytes) {
               n = ::read(sockID, ptr + len, nbytes - len);
               if(n < 0)
                    return n;
               
               len += n;
          }
          
#ifdef SOCKET_STAT
          readBytes += len;
#endif
          return len;
     }
     
     ssize_t write(const void *buf, size_t nbytes) const
     {
#ifdef SOCKET_STAT
          sentBytes += nbytes;
#endif
          return ::write(sockID, buf, nbytes);
     }

#ifdef SOCKET_STAT
     void writeStats(std::ostream& os) {
          static time_t lastTime = 0;
          static long lastRead = readBytes;
          static long lastSent = sentBytes;
          time_t now = time(NULL);

          os << "Total bytes sent: " << sentBytes << std::endl 
             << "Total bytes read: " << readBytes << std::endl 
             << "Total time connected: " << now - startTime << std::endl;
          
          if(lastTime == 0) {
               lastTime = now;
               lastRead = readBytes;
               lastSent = sentBytes;
               return;
          }

          if(lastTime == now)
               return;

          os << "RD Avg. since last stats: " 
             << 1.0f*(readBytes - lastRead)/(now - lastTime) << std::endl
             << "WR Avg. since last stats: " 
             << 1.0f*(sentBytes - lastSent)/(now - lastTime) << std::endl;

          lastTime = now;
          lastRead = readBytes;
          lastSent = sentBytes;
     }
#endif
};

#endif //_SOCKET_

---