diff options
Diffstat (limited to 'src/server/authserver/Server/RealmSocket.h')
-rw-r--r-- | src/server/authserver/Server/RealmSocket.h | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/src/server/authserver/Server/RealmSocket.h b/src/server/authserver/Server/RealmSocket.h new file mode 100644 index 0000000000..8ac52a801b --- /dev/null +++ b/src/server/authserver/Server/RealmSocket.h @@ -0,0 +1,80 @@ +/* + * Copyright (C) + * Copyright (C) + * + * 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, see <http://www.gnu.org/licenses/>. + */ + +#ifndef __REALMSOCKET_H__ +#define __REALMSOCKET_H__ + +#include <ace/Synch_Traits.h> +#include <ace/Svc_Handler.h> +#include <ace/SOCK_Stream.h> +#include <ace/Message_Block.h> +#include <ace/Basic_Types.h> +#include "Common.h" + +class RealmSocket : public ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_NULL_SYNCH> +{ +private: + typedef ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_NULL_SYNCH> Base; + +public: + class Session + { + public: + Session(void); + virtual ~Session(void); + + virtual void OnRead(void) = 0; + virtual void OnAccept(void) = 0; + virtual void OnClose(void) = 0; + }; + + RealmSocket(void); + virtual ~RealmSocket(void); + + size_t recv_len(void) const; + bool recv_soft(char *buf, size_t len); + bool recv(char *buf, size_t len); + void recv_skip(size_t len); + + bool send(const char *buf, size_t len); + + const std::string& getRemoteAddress(void) const; + + uint16 getRemotePort(void) const; + + virtual int open(void *); + + virtual int close(u_long); + + virtual int handle_input(ACE_HANDLE = ACE_INVALID_HANDLE); + virtual int handle_output(ACE_HANDLE = ACE_INVALID_HANDLE); + + virtual int handle_close(ACE_HANDLE = ACE_INVALID_HANDLE, ACE_Reactor_Mask = ACE_Event_Handler::ALL_EVENTS_MASK); + + void set_session(Session* session); + +private: + ssize_t noblk_send(ACE_Message_Block &message_block); + + ACE_Message_Block input_buffer_; + Session* session_; + std::string _remoteAddress; + uint16 _remotePort; +}; + +#endif /* __REALMSOCKET_H__ */ |