Search  

---

C:/temp/src/j2k/Net/Socket.cpp

Go to the documentation of this file.

// Wolf for J2K Library
// LGPL Licensed.
// http://j2k.sourceforge.net/
// Revision #1: Fred P. - Oct 2nd, 2001

#ifndef __J2K__NET__SOCKET_CPP__
#define __J2K__NET__SOCKET_CPP__

#include <j2k/Net/Socket.hpp>

//
// CONSTRUCTORS
//

Socket::Socket( sockaddr* sa = NULL, 
                int domain   = PF_INET, 
                int type     = SOCK_STREAM, 
                int protocol = 0
) throw (int)
{
  s_domain   = domain;
  s_type     = type;
  s_protocol = protocol;

  rsa = NULL;
  lsa = NULL;

  sck = socket( domain, type, protocol );
  
  if(sck == -1) 
  {
    throw errno;
  }
  else 
  {
    if(sa != NULL) 
    {
      local_sa.sa_family = sa->sa_family;

      // NOTICE: Can't you use memcpy() instead ?!
      for ( register int i = 0; i < 14; i++ ) 
      {
        local_sa.sa_data[i] = sa->sa_data[i];
      }

      lsa = &local_sa;
    }
    else 
    {
/*    local_sa.sa_family = AF_INET;
      memset(&(local_sa.sa_data), '\0', 14);
      ((struct sockaddr_in*)local_sa).sin_port = htons(0);
      ((struct sockaddr_in)local_Sa).sin_addr.s_addr = htonl(INADDR_ANY);
      lsa = &local_sa; 
*/
    }
  }
}
  
Socket::Socket(Socket& original) throw (int)
{
  struct sockaddr* tmp;
  
  s_domain   = original.getDomain();
  s_type     = original.getType();
  s_protocol = original.getProtocol();
  
  sck = socket(s_domain,s_type,s_protocol);
  if(sck == -1) 
  {
    throw errno;
  }
  else 
  {
    tmp = original.getLocalSockAddr();
    local_sa.sa_family = tmp->sa_family;
    for (int i = 0;i<14;i++) local_sa.sa_data[i] = tmp->sa_data[i];
    tmp = original.getSockAddr();
    remote_sa.sa_family = tmp->sa_family;
    for (int i = 0;i<14;i++) remote_sa.sa_data[i] = tmp->sa_data[i];
    lsa = &local_sa;
    rsa = &remote_sa;
  }
}

Socket::Socket( int set_socket, 
                int domain   = PF_INET, 
                int type     = SOCK_STREAM, 
                int protocol = 0
) throw(int)
{
  s_domain   = domain;
  s_type     = type;
  s_protocol = protocol;

  lsa = NULL;
  rsa = NULL;
  sck = set_socket;
  int i;
  socklen_t socksize = sizeof(struct sockaddr);
  
  i = getsockname(sck,&local_sa,&socksize);
  if (i == -1) 
  {
    throw errno;
  }
  else 
  {
    lsa = &local_sa;
    socksize = sizeof(struct sockaddr);
    i = getpeername(sck,&remote_sa,&socksize);
    if ((i == -1) && (errno != ENOTCONN)) 
    {
      throw errno;
    }
    else 
    {
      rsa = &remote_sa;
    }
  }
}
      

//
// METHODS
//

int Socket::bind(struct sockaddr* sa = NULL)
{
  int ret;
  
  if (sa != NULL) 
  {
    local_sa.sa_family = sa->sa_family;
    for (int i = 0;i<14;i++) local_sa.sa_data[i] = sa->sa_data[i];
    rsa = &local_sa;
  }
    
  ret = ::bind(sck, lsa, sizeof(struct sockaddr));
  
  return ret;
}
  
int Socket::connect(struct sockaddr* sa = NULL)
{
  int ret;
  
  if (sa != NULL) 
  {
    remote_sa.sa_family = sa->sa_family;
    for (int i = 0;i<14;i++) remote_sa.sa_data[i] = sa->sa_data[i];
    rsa = &local_sa;
  }
  
  ret = ::connect(sck, rsa, sizeof(struct sockaddr));

  return ret;
}
  
void Socket::close()
{
  ::close(sck);
}
    
void Socket::shutdown(int how = 2)
{
  ::shutdown(sck, how);
}

void Socket::shutdownInput()
{
  ::shutdown(sck, 0);
}

void Socket::shutdownOutput()
{
  ::shutdown(sck, 1);
}

struct sockaddr* Socket::getSockAddr()
{
  int size = sizeof(struct sockaddr);
  getpeername(sck, rsa, (socklen_t *) &size);
  return rsa;
}

struct sockaddr* Socket::getLocalSockAddr()
{
  int size = sizeof(struct sockaddr);
  getsockname(sck, lsa, (socklen_t *) &size);
  return lsa;
}
    
int Socket::getSocket()
{
  return sck;
}

int Socket::getDomain()
{
  return s_domain;
}

int Socket::getType()
{
  return s_type;
}

int Socket::getProtocol()
{
  return s_protocol;
}

int Socket::setSockOpt(int level, int optname, void* optval, socklen_t optlen)
{
  int ret;
  ret = setsockopt(sck, level, optname, optval, optlen);
  return ret;
}

       
void* Socket::getSockOpt(int level, int optname, void* optval, socklen_t* optlen)
{
  int ret;
  ret = getsockopt(sck, level, optname, optval, optlen);
  if (ret != -1)
  {
    return optval;
  }
  else
  {
    return NULL;
  }
}

int Socket::read(void * buffer, size_t len, int flags = 0)
{
   int l  = len;
   l = recv(sck,buffer, len, flags);
   return l;
}

// return -1 and set errno on error 
int Socket::send(void * buffer, size_t len, int flags = 0)
{
   int l  = len;
   l = ::send(sck,buffer, len, flags);
   return l;
}

#endif //__J2K__NET__SOCKET_CPP__

---