Socket.h

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
*/
#ifndef SOCKET_H
#define SOCKET_H

#ifdef _WIN32
#include <windows.h>
#define EINPROGRESS WSAEINPROGRESS
#define socket_t int
#define socklen_t int
#define sockaddr SOCKADDR
#else
#include <fcntl.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>
#include <arpa/inet.h>
#define socket_t int
#endif
#ifdef HAVE_SSL
#include <openssl/ssl.h>
#endif
#include <Locking.h>
#include <Errors.h>
#include <iostream>

using namespace std;

const int bufflen = 8192;

enum SocketType{

  TYPE_TCP=1,
  TYPE_UDP,
  TYPE_ICMP

  };

class Socket{ /* Finished, do not inherit this class! */

  friend class SocketHandler;
  friend class RealSocket; // needs to be removed
  friend class ListenServer; // Close();
#ifdef HAVE_SSL
  friend class ConnectSocket; // SSL
#endif

  public:

    Socket();
    Socket(socket_t _sock);
    virtual ~Socket();

    void MakeSock();
    socket_t GetSock();

    int GetPort();
    string GetAddr();
    string GetLastError();
#ifdef HAVE_SSL
    void SetSSL(bool _is_SSL);
#endif

  private:

    void OnReadError();
    void OnConnected();
    void OnConnClosed();

    virtual void __OnConnClosed() {}
    virtual void __OnConnected() {}
    virtual void __OnRead() = 0;
    virtual void __OnClose() = 0;
    virtual void __OnReadError() = 0;
    
    socket_t sock;
#ifdef HAVE_SSL
    SSL_CTX* ctx;
    SSL* ssl;
#endif

    bool is_SSL;
    CriticalSection cs;

  protected:

    void Close();
    SocketType type;
    bool connected; // should not be protected
    int port;
    string addr;

    struct sockaddr_in saddr;

  };

string Resolv(string _addr);

#endif