Socket.cpp

Ugrás a fájl dokumentációjához.

/*
    mySocket
    Copyright (C) 2006  Kornel Csernai <csko@csko.hu>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/
#include <Socket.h>
#include <SocketHandler.h>
#include <StringLib.h>

using namespace std;
using namespace StringLib;

// Constructor
Socket::Socket(){

#ifdef HAVE_SSL
  OpenSSL_add_ssl_algorithms();
  SSL_load_error_strings();
//  SSL_library_init();
  connected = false;
#endif

  is_SSL = false;
  sock = -1;
#ifdef _WIN32
  WSADATA WinsockData;
  if (WSAStartup(MAKEWORD(2, 2), &WinsockData) != 0) {

//    error("Failed to do the M$ stuff!");
    throw SocketException(WSA_FAILED);

    }

#endif
  }

// Constructor
Socket::Socket(socket_t _sock){

  sock = _sock;
  if(sock > -1) connected = true;
  else connected = false;
  is_SSL = false;

#ifdef _WIN32
  WSADATA WinsockData;
  if (WSAStartup(MAKEWORD(2, 2), &WinsockData) != 0) {

//    error("Failed to do the M$ stuff!");
    throw SocketException(WSA_FAILED);

    }

#endif

  }

// Destructor
Socket::~Socket(){
  if (sock != -1){
    Close();
    }
  }

void Socket::MakeSock(){

  if(sock!=-1) throw SocketException(ALREADY_HAVE_SOCKET);

  int sockettype;
  
  switch(type){
  
    case TYPE_TCP: sockettype = SOCK_STREAM; break;
    case TYPE_UDP: sockettype = SOCK_DGRAM; break;
//    case TYPE_ICMP: sockettype = SOCK_STREAM; break;
    default: throw SocketException(WRONG_SOCKET_TYPE);

    }

  sock = socket(AF_INET, sockettype, 0);
  if(sock < 0) throw SocketException(SOCKET_FAILED);

#ifdef HAVE_SSL

  if(is_SSL){
  
    SSL_library_init();
    if(!(ctx = SSL_CTX_new(SSLv23_method())) or !(ssl = SSL_new (ctx)) or
    SSL_set_fd(ssl, GetSock()) == 0) throw SocketException(SSL_FAILED);
    debug("SSL Created");
        
    }

#endif

  }

void Socket::Close(){

#ifdef _WIN32
  closesocket(sock);
#else
  close(sock);
#endif

  connected = false;
  sock = -1;

  __OnClose();

  }

// Return socket ID
socket_t Socket::GetSock(){

  return sock;

  }

// Return socket port
int Socket::GetPort(){
  
  return (int) ntohs(saddr.sin_port);
    
  }


// Return socket address
string Socket::GetAddr(){
  
  return (string) inet_ntoa(saddr.sin_addr);
  
  }

// Get last error message
string Socket::GetLastError(){

#ifdef _WIN32

  return itos(WSAGetLastError());

#else

  if(errno > 0) {

    char* err = strerror(errno);
    return (string) err + " (" + itos(errno) + ")";

    }
#endif
  return "";

  }


#ifdef HAVE_SSL
void Socket::SetSSL(bool _is_SSL){

  if(connected) return;

  is_SSL = _is_SSL;

  }
#endif

void Socket::OnConnected(){

  connected = true;
  __OnConnected();

  }


void Socket::OnReadError(){

  __OnReadError();
  Close();

  }

// Called on socket disconnect
void Socket::OnConnClosed(){

  SocketHandler* handler = SocketHandler::getInstance();
  if(!handler) throw SocketException(SOCKETHANDLING_FAILED);
  
  handler -> Remove(this);
  __OnConnClosed();

  }


string Resolv(string _addr){

  struct hostent *host;
//  u_long _saddr;
  struct in_addr destip;
#ifdef _WIN32 // sad :(
  if ((host = gethostbyname(_addr.c_str())) != NULL){

    memcpy(&(destip.s_addr),host->h_addr_list[0],sizeof(u_long));
          
    return (string) inet_ntoa(destip);

    }else{

    return "";

    }

#else

  if(inet_aton(_addr.c_str(), &destip)){ // ip

    return _addr;

    } else { // not ip

    if ((host = gethostbyname(_addr.c_str())) != NULL){

      memcpy(&(destip.s_addr),host->h_addr_list[0],sizeof(u_long));
          
          return (string) inet_ntoa(destip);

      }else{

      return "";

      }

    }

#endif
  }