socketstream.h revision f74420b3285b9fe04a7e00aa3b8c0ab07ea344bc
1f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch/* 2f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * libjingle 3f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Copyright 2005, Google Inc. 4f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 5f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Redistribution and use in source and binary forms, with or without 6f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * modification, are permitted provided that the following conditions are met: 7f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 8f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 1. Redistributions of source code must retain the above copyright notice, 9f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer. 10f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 2. Redistributions in binary form must reproduce the above copyright notice, 11f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer in the documentation 12f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * and/or other materials provided with the distribution. 13f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 3. The name of the author may not be used to endorse or promote products 14f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * derived from this software without specific prior written permission. 15f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 16f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch */ 27f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 28f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#ifndef TALK_BASE_SOCKET_STREAM_H__ 29f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#define TALK_BASE_SOCKET_STREAM_H__ 30f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 31f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/asyncsocket.h" 32f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/common.h" 33f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/stream.h" 34f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 35f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochnamespace talk_base { 36f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 37f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch/////////////////////////////////////////////////////////////////////////////// 38f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 39f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochclass SocketStream : public StreamInterface, public sigslot::has_slots<> { 40f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch public: 41f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SocketStream(AsyncSocket* socket) : socket_(NULL) { 42f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch Attach(socket); 43f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 44f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch virtual ~SocketStream() { delete socket_; } 45f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 46f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch void Attach(AsyncSocket* socket) { 47f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (socket_) 48f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch delete socket_; 49f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_ = socket; 50f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (socket_) { 51f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalConnectEvent.connect(this, &SocketStream::OnConnectEvent); 52f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalReadEvent.connect(this, &SocketStream::OnReadEvent); 53f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalWriteEvent.connect(this, &SocketStream::OnWriteEvent); 54f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalCloseEvent.connect(this, &SocketStream::OnCloseEvent); 55f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 56f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 57f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 58f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch AsyncSocket* Detach() { 59f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch AsyncSocket* socket = socket_; 60f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (socket_) { 61f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalConnectEvent.disconnect(this); 62f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalReadEvent.disconnect(this); 63f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalWriteEvent.disconnect(this); 64f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_->SignalCloseEvent.disconnect(this); 65f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch socket_ = NULL; 66f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 67f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return socket; 68f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 69f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 70f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch AsyncSocket* GetSocket() { return socket_; } 71f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 72f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch virtual StreamState GetState() const { 73f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket_ != NULL); 74f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch switch (socket_->GetState()) { 75f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch case Socket::CS_CONNECTED: 76f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SS_OPEN; 77f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch case Socket::CS_CONNECTING: 78f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SS_OPENING; 79f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch case Socket::CS_CLOSED: 80f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch default: 81f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SS_CLOSED; 82f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 83f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 84f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 85f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch virtual StreamResult Read(void* buffer, size_t buffer_len, 86f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch size_t* read, int* error) { 87f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket_ != NULL); 88f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch int result = socket_->Recv(buffer, buffer_len); 89f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (result < 0) { 90f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (socket_->IsBlocking()) 91f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_BLOCK; 92f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (error) 93f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *error = socket_->GetError(); 94f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_ERROR; 95f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 96f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if ((result > 0) || (buffer_len == 0)) { 97f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (read) 98f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *read = result; 99f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_SUCCESS; 100f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 101f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_EOS; 102f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 103f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 104f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch virtual StreamResult Write(const void* data, size_t data_len, 105f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch size_t* written, int* error) { 106f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket_ != NULL); 107f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch int result = socket_->Send(data, data_len); 108f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (result < 0) { 109f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (socket_->IsBlocking()) 110f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_BLOCK; 111f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (error) 112f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *error = socket_->GetError(); 113f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_ERROR; 114f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 115f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (written) 116f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *written = result; 117f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return SR_SUCCESS; 118f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 119f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 120f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch virtual void Close() { ASSERT(socket_ != NULL); socket_->Close(); } 121f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 122f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch private: 123f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch void OnConnectEvent(AsyncSocket* socket) { 124f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket == socket_); 125f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SignalEvent(this, SE_OPEN | SE_READ | SE_WRITE, 0); 126f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 127f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch void OnReadEvent(AsyncSocket* socket) { 128f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket == socket_); 129f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SignalEvent(this, SE_READ, 0); 130f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 131f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch void OnWriteEvent(AsyncSocket* socket) { 132f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket == socket_); 133f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SignalEvent(this, SE_WRITE, 0); 134f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 135f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch void OnCloseEvent(AsyncSocket* socket, int err) { 136f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ASSERT(socket == socket_); 137f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch SignalEvent(this, SE_CLOSE, err); 138f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 139f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 140f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch AsyncSocket* socket_; 141f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 142f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch DISALLOW_EVIL_CONSTRUCTORS(SocketStream); 143f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch}; 144f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 145f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch/////////////////////////////////////////////////////////////////////////////// 146f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 147f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} // namespace talk_base 148f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 149f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#endif // TALK_BASE_SOCKET_STREAM_H__ 150