icinga2/lib/base/socket.hpp

82 lines
2.5 KiB
C++
Raw Normal View History

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
* *
* 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 *
2012-05-11 13:33:57 +02:00
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
#ifndef SOCKET_H
#define SOCKET_H
2012-03-28 13:24:49 +02:00
2014-05-25 16:23:35 +02:00
#include "base/i2-base.hpp"
#include "base/stream.hpp"
2013-03-15 18:21:29 +01:00
#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition_variable.hpp>
2013-03-16 21:18:53 +01:00
#include <boost/signals2.hpp>
2013-03-15 18:21:29 +01:00
namespace icinga
{
2012-03-28 13:24:49 +02:00
/**
2012-11-22 12:04:32 +01:00
* Base class for connection-oriented sockets.
2012-05-18 22:21:28 +02:00
*
* @ingroup base
*/
2013-04-04 16:08:02 +02:00
class I2_BASE_API Socket : public Object
2012-03-28 13:24:49 +02:00
{
public:
2014-11-07 12:32:25 +01:00
DECLARE_PTR_TYPEDEFS(Socket);
2012-03-28 13:24:49 +02:00
Socket(void);
2013-04-04 16:08:02 +02:00
Socket(SOCKET fd);
2012-03-28 13:24:49 +02:00
~Socket(void);
SOCKET GetFD(void) const;
2013-04-04 16:08:02 +02:00
void Close(void);
String GetClientAddress(void);
String GetPeerAddress(void);
2013-04-04 16:08:02 +02:00
size_t Read(void *buffer, size_t size);
size_t Write(const void *buffer, size_t size);
2012-11-22 12:04:32 +01:00
void Listen(void);
2013-04-04 16:08:02 +02:00
Socket::Ptr Accept(void);
2012-07-17 20:41:06 +02:00
bool Poll(bool read, bool write);
void MakeNonBlocking(void);
protected:
2012-06-24 02:56:48 +02:00
void SetFD(SOCKET fd);
2012-05-26 20:00:03 +02:00
int GetError(void) const;
2012-08-13 13:06:43 +02:00
mutable boost::mutex m_SocketMutex;
2012-08-06 10:01:21 +02:00
private:
SOCKET m_FD; /**< The socket descriptor. */
static String GetAddressFromSockaddr(sockaddr *address, socklen_t len);
2012-03-28 13:24:49 +02:00
};
2013-03-11 13:45:08 +01:00
class socket_error : virtual public std::exception, virtual public boost::exception { };
2012-05-26 20:00:03 +02:00
2012-03-28 13:24:49 +02:00
}
#endif /* SOCKET_H */