---

dsSocket.h

Go to the documentation of this file.

/* 
 * $Id: dsSocket.h,v 1.25 2004/07/06 13:05:04 dsamersoff Exp $
 */
#ifndef dsSocket_h
#define dsSocket_h 

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h> 
#include <fcntl.h>
#include <errno.h>

#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif

#ifdef HAVE_SYS_TIME_H
# include <sys/time.h>
#endif

#ifdef HAVE_IO_H
# include <io.h>
#endif


#ifdef POSIX_OS
# include <sys/socket.h>
# include <sys/types.h>
# include <netinet/in.h>
# include <arpa/inet.h>
# include <netdb.h>
#endif

#ifdef Win32
# include <winsock.h>
#endif

#include <dsSmartException.h>
#include <dsform.h>

#define NO_FLAGS 0                   // No special flags specified

#ifdef Win32
 typedef int socklen_t;
#endif

DECLARE_EXCEPTION(dsSocket);

class dsSocketBase
{
protected:
    int           _sock;        // socket

    virtual size_t __read(char *buf,   size_t bytes) = 0;
    virtual size_t __write(const char *ptr,  size_t bytes) = 0;

public:
  
  dsSocketBase();

  virtual void Connect(const char *host, const int port, const int proto) = 0;
  virtual void Close() = 0;

  int fd() const { return _sock; };


  size_t Read(char *buffer, size_t bytes, int sec = 10, int usec = 0 );
  size_t Read(std::ostream& os, int sec = 10, int usec = 0);

  size_t Write(const char *buffer, size_t bytes, int sec = 10, int usec = 0 );
  size_t Write(std::istream& is, int sec = 10, int usec = 0);

  size_t Write(const char *str, int sec = 10, int usec = 0 )
  {
        return  Write(str,strlen(str), sec, usec);
  }

  size_t Printf(const char *format, ...);

         
  
  size_t ReadLine(std::ostream& os, int sec = 10);
  size_t ReadLine(char *buf, size_t size, int sec = 10);

  size_t WriteLine(std::istream& is, int sec = 10);
  size_t WriteLine(const char *str,  int sec = 10);
};


class dsSocket : public dsSocketBase
{
protected:
    struct sockaddr_in      _sockAddr; 
    struct hostent          *_pHostEnt;

#ifdef Win32
     WSADATA _ws;
#endif

public:
  
  dsSocket(const char *host, const int port, const int proto = IPPROTO_TCP);
  dsSocket();
  virtual ~dsSocket();

  void Open(const int proto = IPPROTO_UDP, const int fcntl_flags = 0); // O_NONBLOCK

  virtual void Connect(const char *host, const int port, const int proto = IPPROTO_TCP);
  virtual void Close();

  virtual size_t __read(char *buf,   size_t bytes);
  virtual size_t __write(const char *ptr,  size_t bytes);
  virtual size_t __sendto(const char *buf, size_t bytes, const sockaddr *to, socklen_t len);

  size_t Sendto(const char *ptr, size_t bytes, const char *host, const int port);

  unsigned long getip();
  unsigned long getpeerip();
  unsigned int  getport();

  char *getpeer_aip();
  char *get_aip();
};

class dsServerSocket : public dsSocket
{
  int                     _srv_sock;    // for listen
  struct sockaddr_in      _srv_sockAddr;

public:

/* listen and accept for fork based server */
  void Listen(int port);
#ifdef POSIX_OS
  pid_t FAccept();
#endif
  int   Accept();

  int  srvsocket(){ return _srv_sock; };
  virtual void Close();
};

#endif

---