1 files changed, 116 insertions, 0 deletions
diff --git a/src/lib/net/TCPSocket.h b/src/lib/net/TCPSocket.h
new file mode 100644
index 0000000..1006f88
--- /dev/null
+++ b/src/lib/net/TCPSocket.h
@@ -0,0 +1,116 @@
+/*
+ * barrier -- mouse and keyboard sharing utility
+ * Copyright (C) 2012-2016 Symless Ltd.
+ * Copyright (C) 2002 Chris Schoeneman
+ * 
+ * This package is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * found in the file LICENSE that should have accompanied this file.
+ * 
+ * This package 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/>.
+ */
+
+#pragma once
+
+#include "net/IDataSocket.h"
+#include "io/StreamBuffer.h"
+#include "mt/CondVar.h"
+#include "mt/Mutex.h"
+#include "arch/IArchNetwork.h"
+
+class Mutex;
+class Thread;
+class ISocketMultiplexerJob;
+class IEventQueue;
+class SocketMultiplexer;
+
+//! TCP data socket
+/*!
+A data socket using TCP.
+*/
+class TCPSocket : public IDataSocket {
+public:
+    TCPSocket(IEventQueue* events, SocketMultiplexer* socketMultiplexer, IArchNetwork::EAddressFamily family);
+    TCPSocket(IEventQueue* events, SocketMultiplexer* socketMultiplexer, ArchSocket socket);
+    virtual ~TCPSocket();
+
+    // ISocket overrides
+    virtual void        bind(const NetworkAddress&);
+    virtual void        close();
+    virtual void*        getEventTarget() const;
+
+    // IStream overrides
+    virtual UInt32        read(void* buffer, UInt32 n);
+    virtual void        write(const void* buffer, UInt32 n);
+    virtual void        flush();
+    virtual void        shutdownInput();
+    virtual void        shutdownOutput();
+    virtual bool        isReady() const;
+    virtual bool        isFatal() const;
+    virtual UInt32        getSize() const;
+
+    // IDataSocket overrides
+    virtual void        connect(const NetworkAddress&);
+
+    
+    virtual ISocketMultiplexerJob*
+                        newJob();
+
+protected:
+    enum EJobResult {
+        kBreak = -1,    //!< Break the Job chain
+        kRetry,            //!< Retry the same job
+        kNew            //!< Require a new job
+    };
+    
+    ArchSocket            getSocket() { return m_socket; }
+    IEventQueue*        getEvents() { return m_events; }
+    virtual EJobResult    doRead();
+    virtual EJobResult    doWrite();
+
+    void                setJob(ISocketMultiplexerJob*);
+    
+    bool                isReadable() { return m_readable; }
+    bool                isWritable() { return m_writable; }
+
+    Mutex&                getMutex() { return m_mutex; }
+
+    void                sendEvent(Event::Type);
+    void                discardWrittenData(int bytesWrote);
+
+private:
+    void                init();
+
+    void                sendConnectionFailedEvent(const char*);
+    void                onConnected();
+    void                onInputShutdown();
+    void                onOutputShutdown();
+    void                onDisconnected();
+
+    ISocketMultiplexerJob*
+                        serviceConnecting(ISocketMultiplexerJob*,
+                            bool, bool, bool);
+    ISocketMultiplexerJob*
+                        serviceConnected(ISocketMultiplexerJob*,
+                            bool, bool, bool);
+
+protected:
+    bool                m_readable;
+    bool                m_writable;
+    bool                m_connected;
+    IEventQueue*        m_events;
+    StreamBuffer        m_inputBuffer;
+    StreamBuffer        m_outputBuffer;
+    
+private:
+    Mutex                m_mutex;
+    ArchSocket            m_socket;
+    CondVar<bool>        m_flushed;
+    SocketMultiplexer*    m_socketMultiplexer;
+};