framework/delibs/decpp/deSocket.hpp

#ifndef _DESOCKET_HPP
#define _DESOCKET_HPP
/*-------------------------------------------------------------------------
 * drawElements C++ Base Library
 * -----------------------------
 *
 * Copyright 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *//*!
 * \file
 * \brief deSocket C++ wrapper.
 *//*--------------------------------------------------------------------*/

#include "deDefs.hpp"
#include "deSocket.h"

#include <string>
#include <stdexcept>

namespace de
{

class SocketError : public std::runtime_error
{
public:
	SocketError (const std::string& message) : std::runtime_error(message) {}
};

class SocketAddress
{
public:
						SocketAddress		(void);
						~SocketAddress		(void);

	void				setHost				(const char* host);
	void				setPort				(int port);
	void				setFamily			(deSocketFamily family);
	void				setType				(deSocketType type);
	void				setProtocol			(deSocketProtocol protocol);

	const char*			getHost				(void) const				{ return deSocketAddress_getHost(m_address);					}
	int					getPort				(void) const				{ return deSocketAddress_getPort(m_address);					}
	deSocketFamily		getFamily			(void) const				{ return deSocketAddress_getFamily(m_address);					}
	deSocketType		getType				(void) const				{ return deSocketAddress_getType(m_address);					}
	deSocketProtocol	getProtocol			(void) const				{ return deSocketAddress_getProtocol(m_address);				}

	operator deSocketAddress*		()			{ return m_address; }
	operator const deSocketAddress* () const	{ return m_address; }

	deSocketAddress*	getPtr				(void)						{ return m_address; }


private:
						SocketAddress		(const SocketAddress& other);
	SocketAddress&		operator=			(const SocketAddress& other);

	deSocketAddress*	m_address;
};

class Socket
{
public:
						Socket				(void);
						~Socket				(void);

	void				setFlags			(deUint32 flags);

	deSocketState		getState			(void) const					{ return deSocket_getState(m_socket);				}
	bool				isConnected			(void) const					{ return getState() == DE_SOCKETSTATE_CONNECTED;	}

	void				listen				(const SocketAddress& address);
	Socket*				accept				(SocketAddress& clientAddress)	{ return accept(clientAddress.getPtr());			}
	Socket*				accept				(void)							{ return accept(DE_NULL);							}

	void				connect				(const SocketAddress& address);

	void				shutdown			(void);
	void				shutdownSend		(void);
	void				shutdownReceive		(void);

	bool				isSendOpen			(void)							{ return (deSocket_getOpenChannels(m_socket) & DE_SOCKETCHANNEL_SEND	) != 0;	}
	bool				isReceiveOpen		(void)							{ return (deSocket_getOpenChannels(m_socket) & DE_SOCKETCHANNEL_RECEIVE	) != 0;	}

	void				close				(void);

	deSocketResult		send				(const void* buf, size_t bufSize, size_t* numSent)	{ return deSocket_send(m_socket, buf, bufSize, numSent);	}
	deSocketResult		receive				(void* buf, size_t bufSize, size_t* numRecv)		{ return deSocket_receive(m_socket, buf, bufSize, numRecv);	}

private:
						Socket				(deSocket* socket) : m_socket(socket) {}
						Socket				(const Socket& other);
	Socket&				operator=			(const Socket& other);

	Socket*				accept				(deSocketAddress* clientAddress);

	deSocket*			m_socket;
};

} // de

#endif // _DESOCKET_HPP